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
34struct flow_divert_group;
35struct flow_divert_trie_node;
36
37struct 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
62RB_HEAD(fd_pcb_tree, flow_divert_pcb);
63
64struct 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
79struct 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
91void flow_divert_init(void);
92void flow_divert_detach(struct socket *so);
93errno_t flow_divert_token_set(struct socket *so, struct sockopt *sopt);
94errno_t flow_divert_token_get(struct socket *so, struct sockopt *sopt);
95errno_t flow_divert_pcb_init(struct socket *so, uint32_t ctl_unit);
96errno_t flow_divert_connect_out(struct socket *so, struct sockaddr *to, proc_t p);
97errno_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