1/*
2 * Copyright (c) 2012-2013 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;
35
36struct flow_divert_pcb {
37    decl_lck_mtx_data(, mtx);
38    socket_t						so;
39    RB_ENTRY(flow_divert_pcb)		rb_link;
40    uint32_t						hash;
41    mbuf_t							connect_token;
42    struct sockaddr					*local_address;
43    struct sockaddr					*remote_address;
44    uint32_t						flags;
45    uint32_t						send_window;
46    uint32_t						sb_size;
47    struct flow_divert_group		*group;
48    uint32_t						control_group_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};
57
58RB_HEAD(fd_pcb_tree, flow_divert_pcb);
59
60struct flow_divert_group {
61    decl_lck_rw_data(, lck);
62    struct fd_pcb_tree				pcb_tree;
63    uint32_t						ctl_unit;
64    uint8_t							atomic_bits;
65    MBUFQ_HEAD(send_queue_head)		send_queue;
66    uint8_t							*token_key;
67    size_t							token_key_size;
68};
69
70void		flow_divert_init(void);
71void		flow_divert_detach(struct socket *so);
72errno_t		flow_divert_token_set(struct socket *so, struct sockopt *sopt);
73errno_t		flow_divert_token_get(struct socket *so, struct sockopt *sopt);
74errno_t		flow_divert_pcb_init(struct socket *so, uint32_t ctl_unit);
75errno_t		flow_divert_connect_out(struct socket *so, struct sockaddr *to, proc_t p);
76
77#endif /* __FLOW_DIVERT_H__ */
78