| /* |
| * Copyright (c) 2020 Cisco and/or its affiliates. |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at: |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #ifndef included_gro_h |
| #define included_gro_h |
| |
| #include <vlib/vlib.h> |
| #include <vppinfra/error.h> |
| #include <vnet/ip/ip46_address.h> |
| |
| #define GRO_FLOW_TABLE_MAX_SIZE 16 |
| #define GRO_FLOW_TABLE_FLUSH 1e-5 |
| #define GRO_FLOW_N_BUFFERS 64 |
| #define GRO_FLOW_TIMEOUT 1e-5 /* 10 micro-seconds */ |
| #define GRO_TO_VECTOR_SIZE(X) (X + GRO_FLOW_TABLE_MAX_SIZE) |
| |
| typedef union |
| { |
| struct |
| { |
| u32 sw_if_index[VLIB_N_RX_TX]; |
| ip46_address_t src_address; |
| ip46_address_t dst_address; |
| u16 src_port; |
| u16 dst_port; |
| }; |
| |
| u64 flow_data[5]; |
| u32 flow_data_u32; |
| } gro_flow_key_t; |
| |
| typedef struct |
| { |
| gro_flow_key_t flow_key; |
| f64 next_timeout_ts; |
| u32 last_ack_number; |
| u32 buffer_index; |
| u16 n_buffers; |
| } gro_flow_t; |
| |
| typedef struct |
| { |
| f64 timeout_ts; |
| u64 total_vectors; |
| u32 n_vectors; |
| u32 node_index; |
| u8 is_enable; |
| u8 is_l2; |
| u8 flow_table_size; |
| gro_flow_t gro_flow[GRO_FLOW_TABLE_MAX_SIZE]; |
| } gro_flow_table_t; |
| |
| static_always_inline void |
| gro_flow_set_flow_key (gro_flow_t * to, gro_flow_key_t * from) |
| { |
| to->flow_key.flow_data[0] = from->flow_data[0]; |
| to->flow_key.flow_data[1] = from->flow_data[1]; |
| to->flow_key.flow_data[2] = from->flow_data[2]; |
| to->flow_key.flow_data[3] = from->flow_data[3]; |
| to->flow_key.flow_data[4] = from->flow_data[4]; |
| to->flow_key.flow_data_u32 = from->flow_data_u32; |
| } |
| |
| static_always_inline u8 |
| gro_flow_is_equal (gro_flow_key_t * first, gro_flow_key_t * second) |
| { |
| if (first->flow_data[0] == second->flow_data[0] && |
| first->flow_data[1] == second->flow_data[1] && |
| first->flow_data[2] == second->flow_data[2] && |
| first->flow_data[3] == second->flow_data[3] && |
| first->flow_data[4] == second->flow_data[4] && |
| first->flow_data_u32 == second->flow_data_u32) |
| return 1; |
| |
| return 0; |
| } |
| |
| /** |
| * timeout_expire is in between 3 to 10 microseconds |
| * 3e-6 1e-5 |
| */ |
| static_always_inline void |
| gro_flow_set_timeout (vlib_main_t * vm, gro_flow_t * gro_flow, |
| f64 timeout_expire) |
| { |
| gro_flow->next_timeout_ts = vlib_time_now (vm) + timeout_expire; |
| } |
| |
| static_always_inline u8 |
| gro_flow_is_timeout (vlib_main_t * vm, gro_flow_t * gro_flow) |
| { |
| if (gro_flow->next_timeout_ts < vlib_time_now (vm)) |
| return 1; |
| return 0; |
| } |
| |
| static_always_inline void |
| gro_flow_store_packet (gro_flow_t * gro_flow, u32 bi0) |
| { |
| if (gro_flow->n_buffers == 0) |
| { |
| gro_flow->buffer_index = bi0; |
| } |
| gro_flow->n_buffers++; |
| } |
| |
| static_always_inline u32 |
| gro_flow_table_init (gro_flow_table_t ** flow_table, u8 is_l2, u32 node_index) |
| { |
| if (*flow_table) |
| return 0; |
| |
| gro_flow_table_t *flow_table_temp = 0; |
| flow_table_temp = |
| (gro_flow_table_t *) clib_mem_alloc (sizeof (gro_flow_table_t)); |
| if (!flow_table_temp) |
| return 0; |
| clib_memset (flow_table_temp, 0, sizeof (gro_flow_table_t)); |
| flow_table_temp->node_index = node_index; |
| flow_table_temp->is_enable = 1; |
| flow_table_temp->is_l2 = is_l2; |
| *flow_table = flow_table_temp; |
| return 1; |
| } |
| |
| static_always_inline void |
| gro_flow_table_set_timeout (vlib_main_t * vm, gro_flow_table_t * flow_table, |
| f64 timeout_expire) |
| { |
| if (flow_table) |
| flow_table->timeout_ts = vlib_time_now (vm) + timeout_expire; |
| } |
| |
| static_always_inline u8 |
| gro_flow_table_is_timeout (vlib_main_t * vm, gro_flow_table_t * flow_table) |
| { |
| if (flow_table && (flow_table->timeout_ts < vlib_time_now (vm))) |
| return 1; |
| return 0; |
| } |
| |
| static_always_inline u8 |
| gro_flow_table_is_enable (gro_flow_table_t * flow_table) |
| { |
| if (flow_table) |
| return flow_table->is_enable; |
| |
| return 0; |
| } |
| |
| static_always_inline void |
| gro_flow_table_set_is_enable (gro_flow_table_t * flow_table, u8 is_enable) |
| { |
| if (flow_table) |
| { |
| if (is_enable) |
| { |
| flow_table->is_enable = 1; |
| } |
| else |
| { |
| flow_table->is_enable = 0; |
| } |
| } |
| } |
| |
| static_always_inline void |
| gro_flow_table_free (gro_flow_table_t * flow_table) |
| { |
| if (flow_table) |
| clib_mem_free (flow_table); |
| } |
| |
| static_always_inline void |
| gro_flow_table_set_node_index (gro_flow_table_t * flow_table, u32 node_index) |
| { |
| if (flow_table) |
| flow_table->node_index = node_index; |
| } |
| |
| static_always_inline gro_flow_t * |
| gro_flow_table_new_flow (gro_flow_table_t * flow_table) |
| { |
| if (PREDICT_TRUE (flow_table->flow_table_size < GRO_FLOW_TABLE_MAX_SIZE)) |
| { |
| gro_flow_t *gro_flow; |
| u32 i = 0; |
| while (i < GRO_FLOW_TABLE_MAX_SIZE) |
| { |
| gro_flow = &flow_table->gro_flow[i]; |
| if (gro_flow->n_buffers == 0) |
| { |
| flow_table->flow_table_size++; |
| return gro_flow; |
| } |
| i++; |
| } |
| } |
| |
| return (0); |
| } |
| |
| static_always_inline gro_flow_t * |
| gro_flow_table_get_flow (gro_flow_table_t * flow_table, |
| gro_flow_key_t * flow_key) |
| { |
| gro_flow_t *gro_flow = 0; |
| u32 i = 0; |
| while (i < GRO_FLOW_TABLE_MAX_SIZE) |
| { |
| gro_flow = &flow_table->gro_flow[i]; |
| if (gro_flow_is_equal (flow_key, &gro_flow->flow_key)) |
| return gro_flow; |
| i++; |
| } |
| return (0); |
| } |
| |
| static_always_inline gro_flow_t * |
| gro_flow_table_find_or_add_flow (gro_flow_table_t * flow_table, |
| gro_flow_key_t * flow_key) |
| { |
| gro_flow_t *gro_flow = 0; |
| |
| gro_flow = gro_flow_table_get_flow (flow_table, flow_key); |
| if (gro_flow) |
| return gro_flow; |
| |
| gro_flow = gro_flow_table_new_flow (flow_table); |
| |
| if (gro_flow) |
| { |
| gro_flow_set_flow_key (gro_flow, flow_key); |
| return gro_flow; |
| } |
| |
| return (0); |
| } |
| |
| static_always_inline void |
| gro_flow_table_reset_flow (gro_flow_table_t * flow_table, |
| gro_flow_t * gro_flow) |
| { |
| if (PREDICT_TRUE (flow_table->flow_table_size > 0)) |
| { |
| clib_memset (gro_flow, 0, sizeof (gro_flow_t)); |
| flow_table->flow_table_size--; |
| } |
| } |
| |
| static_always_inline u8 * |
| gro_flow_table_format (u8 * s, va_list * args) |
| { |
| gro_flow_table_t *flow_table = va_arg (*args, gro_flow_table_t *); |
| |
| s = |
| format (s, |
| "flow-table: size %u gro-total-vectors %lu gro-n-vectors %u", |
| flow_table->flow_table_size, flow_table->total_vectors, |
| flow_table->n_vectors); |
| if (flow_table->n_vectors) |
| { |
| double average_rate = |
| (double) flow_table->total_vectors / (double) flow_table->n_vectors; |
| s = format (s, " gro-average-rate %.2f", average_rate); |
| } |
| else |
| s = format (s, " gro-average-rate 0.00"); |
| |
| return s; |
| } |
| #endif /* included_gro_h */ |
| |
| /* |
| * fd.io coding-style-patch-verification: ON |
| * |
| * Local Variables: |
| * eval: (c-set-style "gnu") |
| * End: |
| */ |