2 * Copyright (c) 2017-2019 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
7 * http://www.apache.org/licenses/LICENSE-2.0
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
16 #ifndef SRC_VNET_SESSION_TRANSPORT_H_
17 #define SRC_VNET_SESSION_TRANSPORT_H_
19 #include <vnet/vnet.h>
20 #include <vnet/session/transport_types.h>
22 typedef struct _transport_options_t
24 transport_tx_fn_type_t tx_type;
25 transport_service_type_t service_type;
26 } transport_options_t;
29 * Transport protocol virtual function table
32 typedef struct _transport_proto_vft
37 u32 (*start_listen) (u32 session_index, transport_endpoint_t * lcl);
38 u32 (*stop_listen) (u32 conn_index);
39 int (*connect) (transport_endpoint_cfg_t * rmt);
40 void (*close) (u32 conn_index, u32 thread_index);
41 void (*cleanup) (u32 conn_index, u32 thread_index);
42 clib_error_t *(*enable) (vlib_main_t * vm, u8 is_en);
48 u32 (*push_header) (transport_connection_t * tconn, vlib_buffer_t * b);
49 u16 (*send_mss) (transport_connection_t * tc);
50 u32 (*send_space) (transport_connection_t * tc);
51 u32 (*tx_fifo_offset) (transport_connection_t * tc);
52 void (*update_time) (f64 time_now, u8 thread_index);
53 void (*flush_data) (transport_connection_t *tconn);
54 int (*custom_tx) (void *session);
55 int (*app_rx_evt) (transport_connection_t *tconn);
58 * Connection retrieval
60 transport_connection_t *(*get_connection) (u32 conn_idx, u32 thread_idx);
61 transport_connection_t *(*get_listener) (u32 conn_index);
62 transport_connection_t *(*get_half_open) (u32 conn_index);
67 u8 *(*format_connection) (u8 * s, va_list * args);
68 u8 *(*format_listener) (u8 * s, va_list * args);
69 u8 *(*format_half_open) (u8 * s, va_list * args);
72 * Properties retrieval
74 void (*get_transport_endpoint) (u32 conn_index, u32 thread_index,
75 transport_endpoint_t *tep, u8 is_lcl);
76 void (*get_transport_listener_endpoint) (u32 conn_index,
77 transport_endpoint_t *tep,
83 transport_options_t transport_options;
84 } transport_proto_vft_t;
87 extern transport_proto_vft_t *tp_vfts;
89 #define transport_proto_foreach(VAR, BODY) \
91 for (VAR = 0; VAR < vec_len (tp_vfts); VAR++) \
92 if (tp_vfts[VAR].push_header != 0) \
93 do { BODY; } while (0); \
96 int transport_connect (transport_proto_t tp, transport_endpoint_cfg_t * tep);
97 void transport_close (transport_proto_t tp, u32 conn_index, u8 thread_index);
98 u32 transport_start_listen (transport_proto_t tp, u32 session_index,
99 transport_endpoint_t * tep);
100 u32 transport_stop_listen (transport_proto_t tp, u32 conn_index);
101 void transport_cleanup (transport_proto_t tp, u32 conn_index,
103 void transport_get_endpoint (transport_proto_t tp, u32 conn_index,
104 u32 thread_index, transport_endpoint_t * tep,
106 void transport_get_listener_endpoint (transport_proto_t tp, u32 conn_index,
107 transport_endpoint_t * tep, u8 is_lcl);
109 static inline transport_connection_t *
110 transport_get_connection (transport_proto_t tp, u32 conn_index,
113 return tp_vfts[tp].get_connection (conn_index, thread_index);
116 static inline transport_connection_t *
117 transport_get_listener (transport_proto_t tp, u32 conn_index)
119 return tp_vfts[tp].get_listener (conn_index);
122 static inline transport_connection_t *
123 transport_get_half_open (transport_proto_t tp, u32 conn_index)
125 return tp_vfts[tp].get_half_open (conn_index);
129 transport_custom_tx (transport_proto_t tp, void *s)
131 return tp_vfts[tp].custom_tx (s);
135 transport_app_rx_evt (transport_proto_t tp, u32 conn_index, u32 thread_index)
137 transport_connection_t *tc;
138 if (!tp_vfts[tp].app_rx_evt)
140 tc = transport_get_connection (tp, conn_index, thread_index);
141 return tp_vfts[tp].app_rx_evt (tc);
144 void transport_register_protocol (transport_proto_t transport_proto,
145 const transport_proto_vft_t * vft,
146 fib_protocol_t fib_proto, u32 output_node);
147 transport_proto_vft_t *transport_protocol_get_vft (transport_proto_t tp);
148 void transport_update_time (f64 time_now, u8 thread_index);
150 int transport_alloc_local_port (u8 proto, ip46_address_t * ip);
151 int transport_alloc_local_endpoint (u8 proto, transport_endpoint_cfg_t * rmt,
152 ip46_address_t * lcl_addr,
154 void transport_endpoint_cleanup (u8 proto, ip46_address_t * lcl_ip, u16 port);
155 void transport_enable_disable (vlib_main_t * vm, u8 is_en);
156 void transport_init (void);
159 transport_elog_track_index (transport_connection_t * tc)
162 return tc->elog_track.track_index_plus_one - 1;
168 void transport_connection_tx_pacer_reset (transport_connection_t * tc,
169 u32 rate_bytes_per_sec,
170 u32 initial_bucket, u64 time_now);
172 * Initialize tx pacer for connection
174 * @param tc transport connection
175 * @param rate_bytes_per_second initial byte rate
176 * @param burst_bytes initial burst size in bytes
178 void transport_connection_tx_pacer_init (transport_connection_t * tc,
179 u32 rate_bytes_per_sec,
183 * Update tx pacer pacing rate
185 * @param tc transport connection
186 * @param bytes_per_sec new pacing rate
188 void transport_connection_tx_pacer_update (transport_connection_t * tc,
192 * Get maximum tx burst allowed for transport connection
194 * @param tc transport connection
195 * @param time_now current cpu time as returned by @ref clib_cpu_time_now
196 * @param mss transport's mss
198 u32 transport_connection_snd_space (transport_connection_t * tc,
199 u64 time_now, u16 mss);
202 * Get tx pacer max burst
204 * @param tc transport connection
205 * @param time_now current cpu time
206 * @return max burst for connection
208 u32 transport_connection_tx_pacer_burst (transport_connection_t * tc,
212 * Get tx pacer current rate
214 * @param tc transport connection
215 * @return rate for connection in bytes/s
217 u64 transport_connection_tx_pacer_rate (transport_connection_t * tc);
220 * Initialize period for tx pacers
222 * Defines a unit of time with respect to number of cpu cycles that is to
223 * be used by all tx pacers.
225 void transport_init_tx_pacers_period (void);
228 * Check if transport connection is paced
231 transport_connection_is_tx_paced (transport_connection_t * tc)
233 return (tc->flags & TRANSPORT_CONNECTION_F_IS_TX_PACED);
236 u8 *format_transport_pacer (u8 * s, va_list * args);
239 * Update tx byte stats for transport connection
241 * If tx pacing is enabled, this also updates pacer bucket to account for the
242 * amount of bytes that have been sent.
244 * @param tc transport connection
245 * @param pkts packets recently sent
246 * @param bytes bytes recently sent
248 void transport_connection_update_tx_stats (transport_connection_t * tc,
252 transport_connection_tx_pacer_update_bytes (transport_connection_t * tc,
255 #endif /* SRC_VNET_SESSION_TRANSPORT_H_ */
258 * fd.io coding-style-patch-verification: ON
261 * eval: (c-set-style "gnu")