Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | /* * Copyright (c) 2012-2017, 2020, 2022, 2024 Apple Inc. All rights reserved. * * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ * * This file contains Original Code and/or Modifications of Original Code * as defined in and that are subject to the Apple Public Source License * Version 2.0 (the 'License'). You may not use this file except in * compliance with the License. The rights granted to you under the License * may not be used to create, or enable the creation or redistribution of, * unlawful or unlicensed copies of an Apple operating system, or to * circumvent, violate, or enable the circumvention or violation of, any * terms of an Apple operating system software license agreement. * * Please obtain a copy of the License at * http://www.opensource.apple.com/apsl/ and read it before using this file. * * The Original Code and all software distributed under the License are * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. * Please see the License for the specific language governing rights and * limitations under the License. * * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ */ #ifndef __FLOW_DIVERT_H__ #define __FLOW_DIVERT_H__ #include <sys/mbuf.h> #define FLOW_DIVERT_ORDER_LAST INT_MAX struct flow_divert_group; struct flow_divert_trie_node { uint16_t start; uint16_t length; uint16_t child_map; }; struct flow_divert_pcb { decl_lck_mtx_data(, mtx); socket_t so; RB_ENTRY(flow_divert_pcb) rb_link; uint32_t hash; mbuf_t connect_token; uint32_t flags; uint32_t send_window; struct flow_divert_group *group; uint32_t control_group_unit; uint32_t aggregate_unit; uint32_t policy_control_unit; int32_t ref_count; uint64_t bytes_written_by_app; uint64_t bytes_sent; uint64_t bytes_received; uint8_t log_level; SLIST_ENTRY(flow_divert_pcb) tmp_list_entry; mbuf_t connect_packet; uint8_t *app_data __counted_by(app_data_length); size_t app_data_length; union sockaddr_in_4_6 local_endpoint; struct sockaddr *original_remote_endpoint; struct ifnet *original_last_outifp6; struct ifnet *original_last_outifp; uint8_t original_vflag; }; RB_HEAD(fd_pcb_tree, flow_divert_pcb); struct flow_divert_trie { struct flow_divert_trie_node *nodes __counted_by(nodes_count); uint16_t *child_maps __sized_by(child_maps_size); uint8_t *bytes __counted_by(bytes_count); void *memory __sized_by(memory_size); uint16_t nodes_count; uint16_t child_maps_count; uint16_t bytes_count; uint16_t nodes_free_next; uint16_t child_maps_free_next; uint16_t bytes_free_next; uint16_t root; size_t memory_size; size_t child_maps_size; }; struct flow_divert_group { decl_lck_rw_data(, lck); TAILQ_ENTRY(flow_divert_group) chain; struct fd_pcb_tree pcb_tree; uint32_t ctl_unit; uint8_t atomic_bits; MBUFQ_HEAD(send_queue_head) send_queue; uint8_t *token_key __counted_by(token_key_size); size_t token_key_size; uint32_t flags; struct flow_divert_trie signing_id_trie; int32_t ref_count; pid_t in_process_pid; int32_t order; }; void flow_divert_init(void); void flow_divert_detach(struct socket *so); errno_t flow_divert_token_set(struct socket *so, struct sockopt *sopt); errno_t flow_divert_token_get(struct socket *so, struct sockopt *sopt); errno_t flow_divert_pcb_init(struct socket *so); errno_t flow_divert_connect_out(struct socket *so, struct sockaddr *to, proc_t p); errno_t flow_divert_implicit_data_out(struct socket *so, int flags, mbuf_t data, struct sockaddr *to, mbuf_t control, struct proc *p); #endif /* __FLOW_DIVERT_H__ */ |