#ifndef __FLOW_DIVERT_H__
#define __FLOW_DIVERT_H__
#include <sys/mbuf.h>
struct flow_divert_group;
struct flow_divert_pcb {
decl_lck_mtx_data(, mtx);
socket_t so;
RB_ENTRY(flow_divert_pcb) rb_link;
uint32_t hash;
mbuf_t connect_token;
struct sockaddr *local_address;
struct sockaddr *remote_address;
uint32_t flags;
uint32_t send_window;
uint32_t sb_size;
struct flow_divert_group *group;
uint32_t control_group_unit;
int32_t ref_count;
uint32_t bytes_written_by_app;
uint32_t bytes_read_by_app;
uint32_t bytes_sent;
uint32_t bytes_received;
uint8_t log_level;
SLIST_ENTRY(flow_divert_pcb) tmp_list_entry;
};
RB_HEAD(fd_pcb_tree, flow_divert_pcb);
struct flow_divert_group {
decl_lck_rw_data(, lck);
struct fd_pcb_tree pcb_tree;
uint32_t ctl_unit;
uint8_t atomic_bits;
MBUFQ_HEAD(send_queue_head) send_queue;
uint8_t *token_key;
size_t token_key_size;
};
void flow_divert_init(void);
void flow_divert_detach(struct socket *so);
errno_t flow_divert_token_set(struct socket *so, struct sockopt *sopt);
errno_t flow_divert_token_get(struct socket *so, struct sockopt *sopt);
errno_t flow_divert_pcb_init(struct socket *so, uint32_t ctl_unit);
errno_t flow_divert_check_policy(struct socket *so, proc_t p, boolean_t match_delegate, uint32_t *ctl_unit);
errno_t flow_divert_connect_out(struct socket *so, struct sockaddr *to, proc_t p);
void flow_divert_so_init(struct socket *so, proc_t p);
boolean_t flow_divert_is_dns_service(struct socket *so);
#endif