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