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 _NETINET_MP_PCB_H_ |
30 | #define _NETINET_MP_PCB_H_ |
31 | |
32 | #ifdef BSD_KERNEL_PRIVATE |
33 | #include <sys/domain.h> |
34 | #include <sys/protosw.h> |
35 | #include <sys/socketvar.h> |
36 | #include <sys/types.h> |
37 | #include <sys/queue.h> |
38 | #include <kern/locks.h> |
39 | |
40 | /* Keep in sync with bsd/dev/dtrace/scripts/mptcp.d */ |
41 | typedef enum mppcb_state { |
42 | MPPCB_STATE_INUSE = 1, |
43 | MPPCB_STATE_DEAD = 2, |
44 | } mppcb_state_t; |
45 | |
46 | /* |
47 | * Multipath Protocol Control Block |
48 | */ |
49 | struct mppcb { |
50 | TAILQ_ENTRY(mppcb) mpp_entry; /* glue to all PCBs */ |
51 | decl_lck_mtx_data(, mpp_lock); /* per PCB lock */ |
52 | struct mppcbinfo *mpp_pcbinfo; /* PCB info */ |
53 | struct mptses *mpp_pcbe; /* ptr to MPTCP-session */ |
54 | struct socket *mpp_socket; /* back pointer to socket */ |
55 | uint32_t mpp_flags; /* PCB flags */ |
56 | mppcb_state_t mpp_state; /* PCB state */ |
57 | |
58 | #if NECP |
59 | uuid_t necp_client_uuid; |
60 | void (*necp_cb)(void *, int, uint32_t, uint32_t, bool *); |
61 | #endif |
62 | }; |
63 | |
64 | static inline struct mppcb * |
65 | mpsotomppcb(struct socket *mp_so) |
66 | { |
67 | VERIFY(SOCK_DOM(mp_so) == PF_MULTIPATH); |
68 | return ((struct mppcb *)mp_so->so_pcb); |
69 | } |
70 | |
71 | /* valid values for mpp_flags */ |
72 | #define MPP_ATTACHED 0x001 |
73 | #define MPP_INSIDE_OUTPUT 0x002 /* MPTCP-stack is inside mptcp_subflow_output */ |
74 | #define MPP_INSIDE_INPUT 0x004 /* MPTCP-stack is inside mptcp_subflow_input */ |
75 | #define MPP_RUPCALL 0x008 /* MPTCP-stack is handling a read upcall */ |
76 | #define MPP_WUPCALL 0x010 /* MPTCP-stack is handling a read upcall */ |
77 | #define MPP_SHOULD_WORKLOOP 0x020 /* MPTCP-stack should call the workloop function */ |
78 | #define MPP_SHOULD_RWAKEUP 0x040 /* MPTCP-stack should call sorwakeup */ |
79 | #define MPP_SHOULD_WWAKEUP 0x080 /* MPTCP-stack should call sowwakeup */ |
80 | #define MPP_CREATE_SUBFLOWS 0x100 /* This connection needs to create subflows */ |
81 | #define MPP_SET_CELLICON 0x200 /* Set the cellicon (deferred) */ |
82 | #define MPP_UNSET_CELLICON 0x400 /* Unset the cellicon (deferred) */ |
83 | |
84 | static inline boolean_t |
85 | mptcp_should_defer_upcall(struct mppcb *mpp) |
86 | { |
87 | return !!(mpp->mpp_flags & (MPP_INSIDE_OUTPUT | MPP_INSIDE_INPUT | MPP_RUPCALL | MPP_WUPCALL)); |
88 | } |
89 | |
90 | /* |
91 | * Multipath PCB Information |
92 | */ |
93 | struct mppcbinfo { |
94 | TAILQ_ENTRY(mppcbinfo) mppi_entry; /* glue to all PCB info */ |
95 | TAILQ_HEAD(, mppcb) mppi_pcbs; /* list of PCBs */ |
96 | uint32_t mppi_count; /* # of PCBs in list */ |
97 | struct zone *mppi_zone; /* zone for this PCB */ |
98 | uint32_t mppi_size; /* size of PCB structure */ |
99 | lck_grp_t *mppi_lock_grp; /* lock grp */ |
100 | lck_attr_t *mppi_lock_attr; /* lock attr */ |
101 | lck_grp_attr_t *mppi_lock_grp_attr; /* lock grp attr */ |
102 | decl_lck_mtx_data(, mppi_lock); /* global PCB lock */ |
103 | uint32_t (*mppi_gc)(struct mppcbinfo *); /* garbage collector func */ |
104 | uint32_t (*mppi_timer)(struct mppcbinfo *); /* timer func */ |
105 | }; |
106 | |
107 | __BEGIN_DECLS |
108 | extern void mp_pcbinit(void); |
109 | extern void mp_pcbinfo_attach(struct mppcbinfo *); |
110 | extern int mp_pcbinfo_detach(struct mppcbinfo *); |
111 | extern int mp_pcballoc(struct socket *, struct mppcbinfo *); |
112 | extern void mp_pcbdetach(struct socket *); |
113 | extern void mp_pcbdispose(struct mppcb *); |
114 | extern void mp_gc_sched(void); |
115 | extern void mptcp_timer_sched(void); |
116 | extern void mptcp_handle_deferred_upcalls(struct mppcb *mpp, uint32_t flag); |
117 | extern int mp_getsockaddr(struct socket *mp_so, struct sockaddr **nam); |
118 | extern int mp_getpeeraddr(struct socket *mp_so, struct sockaddr **nam); |
119 | #if NECP |
120 | extern int necp_client_register_multipath_cb(pid_t pid, uuid_t client_id, struct mppcb *mpp); |
121 | extern void necp_mppcb_dispose(struct mppcb *mpp); |
122 | #endif |
123 | __END_DECLS |
124 | |
125 | #endif /* BSD_KERNEL_PRIVATE */ |
126 | #endif /* !_NETINET_MP_PCB_H_ */ |
127 | |