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