1 | /* |
2 | * Copyright (c) 2012-2017 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 | struct sockaddr *local_address; |
44 | struct sockaddr *remote_address; |
45 | uint32_t flags; |
46 | uint32_t send_window; |
47 | uint32_t sb_size; |
48 | struct flow_divert_group *group; |
49 | uint32_t control_group_unit; |
50 | int32_t ref_count; |
51 | uint32_t bytes_written_by_app; |
52 | uint32_t bytes_read_by_app; |
53 | uint32_t bytes_sent; |
54 | uint32_t bytes_received; |
55 | uint8_t log_level; |
56 | SLIST_ENTRY(flow_divert_pcb) tmp_list_entry; |
57 | mbuf_t connect_packet; |
58 | uint8_t *app_data; |
59 | size_t app_data_length; |
60 | }; |
61 | |
62 | RB_HEAD(fd_pcb_tree, flow_divert_pcb); |
63 | |
64 | struct flow_divert_trie |
65 | { |
66 | struct flow_divert_trie_node *nodes; |
67 | uint16_t *child_maps; |
68 | uint8_t *bytes; |
69 | void *memory; |
70 | size_t nodes_count; |
71 | size_t child_maps_count; |
72 | size_t bytes_count; |
73 | size_t nodes_free_next; |
74 | size_t child_maps_free_next; |
75 | size_t bytes_free_next; |
76 | uint16_t root; |
77 | }; |
78 | |
79 | struct flow_divert_group { |
80 | decl_lck_rw_data(, lck); |
81 | struct fd_pcb_tree pcb_tree; |
82 | uint32_t ctl_unit; |
83 | uint8_t atomic_bits; |
84 | MBUFQ_HEAD(send_queue_head) send_queue; |
85 | uint8_t *token_key; |
86 | size_t token_key_size; |
87 | uint32_t flags; |
88 | struct flow_divert_trie signing_id_trie; |
89 | }; |
90 | |
91 | void flow_divert_init(void); |
92 | void flow_divert_detach(struct socket *so); |
93 | errno_t flow_divert_token_set(struct socket *so, struct sockopt *sopt); |
94 | errno_t flow_divert_token_get(struct socket *so, struct sockopt *sopt); |
95 | errno_t flow_divert_pcb_init(struct socket *so, uint32_t ctl_unit); |
96 | errno_t flow_divert_connect_out(struct socket *so, struct sockaddr *to, proc_t p); |
97 | errno_t flow_divert_implicit_data_out(struct socket *so, int flags, mbuf_t data, struct sockaddr *to, mbuf_t control, struct proc *p); |
98 | |
99 | #endif /* __FLOW_DIVERT_H__ */ |
100 | |