Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 1 | /* |
Florin Coras | 288eaab | 2019-02-03 15:26:14 -0800 | [diff] [blame] | 2 | * Copyright (c) 2017-2019 Cisco and/or its affiliates. |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 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: |
| 6 | * |
| 7 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | * |
| 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. |
| 14 | */ |
| 15 | |
| 16 | #include <vnet/session/segment_manager.h> |
| 17 | #include <vnet/session/session.h> |
| 18 | #include <vnet/session/application.h> |
| 19 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 20 | segment_manager_main_t segment_manager_main; |
| 21 | |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 22 | /** |
| 23 | * Counter used to build segment names |
| 24 | */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 25 | static u32 segment_name_counter = 0; |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 26 | |
| 27 | /** |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 28 | * Default fifo and segment size. TODO config. |
| 29 | */ |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 30 | static u32 default_fifo_size = 1 << 12; |
| 31 | static u32 default_segment_size = 1 << 20; |
| 32 | static u32 default_app_evt_queue_size = 128; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 33 | |
Florin Coras | ad0c77f | 2017-11-09 18:00:15 -0800 | [diff] [blame] | 34 | segment_manager_properties_t * |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 35 | segment_manager_properties_get (segment_manager_t * sm) |
Florin Coras | ad0c77f | 2017-11-09 18:00:15 -0800 | [diff] [blame] | 36 | { |
Florin Coras | ab2f6db | 2018-08-31 14:31:41 -0700 | [diff] [blame] | 37 | app_worker_t *app_wrk = app_worker_get (sm->app_wrk_index); |
| 38 | return application_get_segment_manager_properties (app_wrk->app_index); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 39 | } |
| 40 | |
| 41 | segment_manager_properties_t * |
| 42 | segment_manager_properties_init (segment_manager_properties_t * props) |
| 43 | { |
Florin Coras | b384b54 | 2018-01-15 01:08:33 -0800 | [diff] [blame] | 44 | props->add_segment_size = default_segment_size; |
| 45 | props->rx_fifo_size = default_fifo_size; |
| 46 | props->tx_fifo_size = default_fifo_size; |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 47 | props->evt_q_size = default_app_evt_queue_size; |
Florin Coras | ad0c77f | 2017-11-09 18:00:15 -0800 | [diff] [blame] | 48 | return props; |
| 49 | } |
| 50 | |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 51 | static u8 |
| 52 | segment_manager_app_detached (segment_manager_t * sm) |
| 53 | { |
Florin Coras | 1553197 | 2018-08-12 23:50:53 -0700 | [diff] [blame] | 54 | return (sm->app_wrk_index == SEGMENT_MANAGER_INVALID_APP_INDEX); |
Dave Wallace | 7b749fe | 2017-07-05 14:30:46 -0400 | [diff] [blame] | 55 | } |
| 56 | |
Florin Coras | 4e4531e | 2017-11-06 23:27:56 -0800 | [diff] [blame] | 57 | void |
| 58 | segment_manager_app_detach (segment_manager_t * sm) |
| 59 | { |
Florin Coras | 1553197 | 2018-08-12 23:50:53 -0700 | [diff] [blame] | 60 | sm->app_wrk_index = SEGMENT_MANAGER_INVALID_APP_INDEX; |
Florin Coras | 4e4531e | 2017-11-06 23:27:56 -0800 | [diff] [blame] | 61 | } |
| 62 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 63 | always_inline u32 |
| 64 | segment_manager_segment_index (segment_manager_t * sm, |
| 65 | svm_fifo_segment_private_t * seg) |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 66 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 67 | return (seg - sm->segments); |
| 68 | } |
| 69 | |
| 70 | /** |
| 71 | * Remove segment without lock |
| 72 | */ |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 73 | void |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 74 | segment_manager_del_segment (segment_manager_t * sm, |
| 75 | svm_fifo_segment_private_t * fs) |
| 76 | { |
| 77 | segment_manager_main_t *smm = &segment_manager_main; |
| 78 | |
| 79 | if (ssvm_type (&fs->ssvm) != SSVM_SEGMENT_PRIVATE) |
Florin Coras | a0dbf9e | 2019-03-01 17:12:02 -0800 | [diff] [blame] | 80 | { |
| 81 | clib_valloc_free (&smm->va_allocator, fs->ssvm.requested_va); |
| 82 | |
| 83 | if (sm->app_wrk_index != SEGMENT_MANAGER_INVALID_APP_INDEX) |
| 84 | { |
| 85 | app_worker_t *app_wrk; |
| 86 | u64 segment_handle; |
| 87 | app_wrk = app_worker_get (sm->app_wrk_index); |
| 88 | segment_handle = segment_manager_segment_handle (sm, fs); |
| 89 | app_worker_del_segment_notify (app_wrk, segment_handle); |
| 90 | } |
| 91 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 92 | |
| 93 | ssvm_delete (&fs->ssvm); |
| 94 | |
| 95 | if (CLIB_DEBUG) |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 96 | clib_memset (fs, 0xfb, sizeof (*fs)); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 97 | pool_put (sm->segments, fs); |
| 98 | } |
| 99 | |
| 100 | /** |
| 101 | * Removes segment after acquiring writer lock |
| 102 | */ |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 103 | static inline void |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 104 | segment_manager_lock_and_del_segment (segment_manager_t * sm, u32 fs_index) |
| 105 | { |
| 106 | svm_fifo_segment_private_t *fs; |
| 107 | u8 is_prealloc; |
| 108 | |
| 109 | clib_rwlock_writer_lock (&sm->segments_rwlock); |
| 110 | fs = segment_manager_get_segment (sm, fs_index); |
| 111 | is_prealloc = svm_fifo_segment_flags (fs) & FIFO_SEGMENT_F_IS_PREALLOCATED; |
| 112 | if (is_prealloc && !segment_manager_app_detached (sm)) |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 113 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 114 | clib_rwlock_writer_unlock (&sm->segments_rwlock); |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 115 | return; |
| 116 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 117 | |
| 118 | segment_manager_del_segment (sm, fs); |
| 119 | clib_rwlock_writer_unlock (&sm->segments_rwlock); |
| 120 | } |
| 121 | |
| 122 | /** |
| 123 | * Reads a segment from the segment manager's pool without lock |
| 124 | */ |
| 125 | svm_fifo_segment_private_t * |
| 126 | segment_manager_get_segment (segment_manager_t * sm, u32 segment_index) |
| 127 | { |
| 128 | return pool_elt_at_index (sm->segments, segment_index); |
| 129 | } |
| 130 | |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 131 | u64 |
| 132 | segment_manager_segment_handle (segment_manager_t * sm, |
| 133 | svm_fifo_segment_private_t * segment) |
| 134 | { |
| 135 | u32 segment_index = segment_manager_segment_index (sm, segment); |
| 136 | return (((u64) segment_manager_index (sm) << 32) | segment_index); |
| 137 | } |
| 138 | |
| 139 | void |
| 140 | segment_manager_parse_segment_handle (u64 segment_handle, u32 * sm_index, |
| 141 | u32 * segment_index) |
| 142 | { |
| 143 | *sm_index = segment_handle >> 32; |
| 144 | *segment_index = segment_handle & 0xFFFFFFFF; |
| 145 | } |
| 146 | |
| 147 | svm_fifo_segment_private_t * |
| 148 | segment_manager_get_segment_w_handle (u64 segment_handle) |
| 149 | { |
| 150 | u32 sm_index, segment_index; |
| 151 | segment_manager_t *sm; |
| 152 | |
| 153 | segment_manager_parse_segment_handle (segment_handle, &sm_index, |
| 154 | &segment_index); |
| 155 | sm = segment_manager_get (sm_index); |
| 156 | if (!sm || pool_is_free_index (sm->segments, segment_index)) |
| 157 | return 0; |
| 158 | return pool_elt_at_index (sm->segments, segment_index); |
| 159 | } |
| 160 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 161 | /** |
| 162 | * Reads a segment from the segment manager's pool and acquires reader lock |
| 163 | * |
| 164 | * Caller must drop the reader's lock by calling |
| 165 | * @ref segment_manager_segment_reader_unlock once it finishes working with |
| 166 | * the segment. |
| 167 | */ |
| 168 | svm_fifo_segment_private_t * |
| 169 | segment_manager_get_segment_w_lock (segment_manager_t * sm, u32 segment_index) |
| 170 | { |
| 171 | clib_rwlock_reader_lock (&sm->segments_rwlock); |
| 172 | return pool_elt_at_index (sm->segments, segment_index); |
| 173 | } |
| 174 | |
| 175 | void |
| 176 | segment_manager_segment_reader_unlock (segment_manager_t * sm) |
| 177 | { |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 178 | ASSERT (sm->segments_rwlock->n_readers > 0); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 179 | clib_rwlock_reader_unlock (&sm->segments_rwlock); |
| 180 | } |
| 181 | |
| 182 | void |
| 183 | segment_manager_segment_writer_unlock (segment_manager_t * sm) |
| 184 | { |
| 185 | clib_rwlock_writer_unlock (&sm->segments_rwlock); |
| 186 | } |
| 187 | |
| 188 | /** |
| 189 | * Adds segment to segment manager's pool |
| 190 | * |
| 191 | * If needed a writer's lock is acquired before allocating a new segment |
| 192 | * to avoid affecting any of the segments pool readers. |
| 193 | */ |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 194 | int |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 195 | segment_manager_add_segment (segment_manager_t * sm, u32 segment_size) |
| 196 | { |
| 197 | segment_manager_main_t *smm = &segment_manager_main; |
Florin Coras | 5da96a7 | 2018-07-12 01:45:13 -0700 | [diff] [blame] | 198 | u32 rnd_margin = 128 << 10, seg_index, page_size; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 199 | segment_manager_properties_t *props; |
David Johnson | d9818dd | 2018-12-14 14:53:41 -0500 | [diff] [blame] | 200 | uword baseva = (uword) ~ 0ULL, alloc_size; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 201 | svm_fifo_segment_private_t *seg; |
| 202 | u8 *seg_name; |
| 203 | int rv; |
| 204 | |
| 205 | props = segment_manager_properties_get (sm); |
| 206 | |
| 207 | /* Not configured for addition of new segments and not first */ |
| 208 | if (!props->add_segment && !segment_size) |
| 209 | { |
| 210 | clib_warning ("cannot allocate new segment"); |
| 211 | return VNET_API_ERROR_INVALID_VALUE; |
| 212 | } |
| 213 | |
| 214 | /* |
| 215 | * Allocate fifo segment and lock if needed |
| 216 | */ |
| 217 | if (vlib_num_workers ()) |
| 218 | { |
| 219 | clib_rwlock_writer_lock (&sm->segments_rwlock); |
| 220 | pool_get (sm->segments, seg); |
| 221 | } |
| 222 | else |
| 223 | { |
| 224 | pool_get (sm->segments, seg); |
| 225 | } |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 226 | clib_memset (seg, 0, sizeof (*seg)); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 227 | |
| 228 | /* |
| 229 | * Initialize ssvm segment and svm fifo private header |
| 230 | */ |
| 231 | segment_size = segment_size ? segment_size : props->add_segment_size; |
Florin Coras | 5da96a7 | 2018-07-12 01:45:13 -0700 | [diff] [blame] | 232 | page_size = clib_mem_get_page_size (); |
| 233 | segment_size = (segment_size + page_size - 1) & ~(page_size - 1); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 234 | if (props->segment_type != SSVM_SEGMENT_PRIVATE) |
| 235 | { |
| 236 | seg_name = format (0, "%d-%d%c", getpid (), segment_name_counter++, 0); |
Florin Coras | 86f0450 | 2018-09-12 16:08:01 -0700 | [diff] [blame] | 237 | alloc_size = (uword) segment_size + rnd_margin; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 238 | baseva = clib_valloc_alloc (&smm->va_allocator, alloc_size, 0); |
| 239 | if (!baseva) |
| 240 | { |
| 241 | clib_warning ("out of space for segments"); |
| 242 | return -1; |
| 243 | } |
| 244 | } |
| 245 | else |
| 246 | seg_name = format (0, "%s%c", "process-private-segment", 0); |
| 247 | |
| 248 | seg->ssvm.ssvm_size = segment_size; |
| 249 | seg->ssvm.name = seg_name; |
| 250 | seg->ssvm.requested_va = baseva; |
| 251 | |
| 252 | if ((rv = ssvm_master_init (&seg->ssvm, props->segment_type))) |
| 253 | { |
| 254 | clib_warning ("svm_master_init ('%v', %u) failed", seg_name, |
| 255 | segment_size); |
| 256 | |
| 257 | if (props->segment_type != SSVM_SEGMENT_PRIVATE) |
| 258 | clib_valloc_free (&smm->va_allocator, baseva); |
| 259 | pool_put (sm->segments, seg); |
| 260 | return (rv); |
| 261 | } |
| 262 | |
| 263 | svm_fifo_segment_init (seg); |
| 264 | |
| 265 | /* |
| 266 | * Save segment index before dropping lock, if any held |
| 267 | */ |
| 268 | seg_index = seg - sm->segments; |
| 269 | |
| 270 | if (vlib_num_workers ()) |
| 271 | clib_rwlock_writer_unlock (&sm->segments_rwlock); |
| 272 | |
| 273 | return seg_index; |
| 274 | } |
| 275 | |
| 276 | segment_manager_t * |
| 277 | segment_manager_new () |
| 278 | { |
| 279 | segment_manager_main_t *smm = &segment_manager_main; |
| 280 | segment_manager_t *sm; |
| 281 | pool_get (smm->segment_managers, sm); |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 282 | clib_memset (sm, 0, sizeof (*sm)); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 283 | clib_rwlock_init (&sm->segments_rwlock); |
| 284 | return sm; |
| 285 | } |
| 286 | |
| 287 | /** |
| 288 | * Initializes segment manager based on options provided. |
| 289 | * Returns error if ssvm segment(s) allocation fails. |
| 290 | */ |
| 291 | int |
| 292 | segment_manager_init (segment_manager_t * sm, u32 first_seg_size, |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 293 | u32 prealloc_fifo_pairs) |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 294 | { |
| 295 | u32 rx_fifo_size, tx_fifo_size, pair_size; |
| 296 | u32 rx_rounded_data_size, tx_rounded_data_size; |
Florin Coras | 86f0450 | 2018-09-12 16:08:01 -0700 | [diff] [blame] | 297 | u64 approx_total_size, max_seg_size = ((u64) 1 << 32) - (128 << 10); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 298 | segment_manager_properties_t *props; |
| 299 | svm_fifo_segment_private_t *segment; |
| 300 | u32 approx_segment_count; |
| 301 | int seg_index, i; |
| 302 | |
| 303 | props = segment_manager_properties_get (sm); |
| 304 | first_seg_size = clib_max (first_seg_size, default_segment_size); |
| 305 | |
| 306 | if (prealloc_fifo_pairs) |
| 307 | { |
| 308 | /* Figure out how many segments should be preallocated */ |
| 309 | rx_rounded_data_size = (1 << (max_log2 (props->rx_fifo_size))); |
| 310 | tx_rounded_data_size = (1 << (max_log2 (props->tx_fifo_size))); |
| 311 | |
| 312 | rx_fifo_size = sizeof (svm_fifo_t) + rx_rounded_data_size; |
| 313 | tx_fifo_size = sizeof (svm_fifo_t) + tx_rounded_data_size; |
| 314 | pair_size = rx_fifo_size + tx_fifo_size; |
| 315 | |
| 316 | approx_total_size = (u64) prealloc_fifo_pairs *pair_size; |
| 317 | if (first_seg_size > approx_total_size) |
| 318 | max_seg_size = first_seg_size; |
| 319 | approx_segment_count = (approx_total_size + (max_seg_size - 1)) |
| 320 | / max_seg_size; |
| 321 | |
| 322 | /* Allocate the segments */ |
| 323 | for (i = 0; i < approx_segment_count + 1; i++) |
| 324 | { |
| 325 | seg_index = segment_manager_add_segment (sm, max_seg_size); |
| 326 | if (seg_index < 0) |
| 327 | { |
| 328 | clib_warning ("Failed to preallocate segment %d", i); |
| 329 | return seg_index; |
| 330 | } |
| 331 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 332 | segment = segment_manager_get_segment (sm, seg_index); |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 333 | if (i == 0) |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 334 | sm->event_queue = segment_manager_alloc_queue (segment, props); |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 335 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 336 | svm_fifo_segment_preallocate_fifo_pairs (segment, |
| 337 | props->rx_fifo_size, |
| 338 | props->tx_fifo_size, |
| 339 | &prealloc_fifo_pairs); |
| 340 | svm_fifo_segment_flags (segment) = FIFO_SEGMENT_F_IS_PREALLOCATED; |
| 341 | if (prealloc_fifo_pairs == 0) |
| 342 | break; |
| 343 | } |
| 344 | } |
| 345 | else |
| 346 | { |
| 347 | seg_index = segment_manager_add_segment (sm, first_seg_size); |
| 348 | if (seg_index) |
| 349 | { |
| 350 | clib_warning ("Failed to allocate segment"); |
| 351 | return seg_index; |
| 352 | } |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 353 | segment = segment_manager_get_segment (sm, seg_index); |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 354 | sm->event_queue = segment_manager_alloc_queue (segment, props); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 355 | } |
| 356 | |
| 357 | return 0; |
| 358 | } |
| 359 | |
| 360 | u8 |
| 361 | segment_manager_has_fifos (segment_manager_t * sm) |
| 362 | { |
| 363 | svm_fifo_segment_private_t *seg; |
| 364 | u8 first = 1; |
| 365 | |
| 366 | /* *INDENT-OFF* */ |
| 367 | segment_manager_foreach_segment_w_lock (seg, sm, ({ |
| 368 | if (CLIB_DEBUG && !first && !svm_fifo_segment_has_fifos (seg) |
| 369 | && !(svm_fifo_segment_flags (seg) & FIFO_SEGMENT_F_IS_PREALLOCATED)) |
| 370 | { |
| 371 | clib_warning ("segment %d has no fifos!", |
| 372 | segment_manager_segment_index (sm, seg)); |
| 373 | first = 0; |
| 374 | } |
| 375 | if (svm_fifo_segment_has_fifos (seg)) |
| 376 | { |
| 377 | segment_manager_segment_reader_unlock (sm); |
| 378 | return 1; |
| 379 | } |
| 380 | })); |
| 381 | /* *INDENT-ON* */ |
| 382 | |
| 383 | return 0; |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 384 | } |
| 385 | |
| 386 | /** |
| 387 | * Initiate disconnects for all sessions 'owned' by a segment manager |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 388 | */ |
| 389 | void |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 390 | segment_manager_del_sessions (segment_manager_t * sm) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 391 | { |
Dave Wallace | 7b749fe | 2017-07-05 14:30:46 -0400 | [diff] [blame] | 392 | svm_fifo_segment_private_t *fifo_segment; |
Florin Coras | 2b81e3c | 2019-02-27 07:55:46 -0800 | [diff] [blame] | 393 | session_handle_t *handles = 0, *handle; |
Florin Coras | 288eaab | 2019-02-03 15:26:14 -0800 | [diff] [blame] | 394 | session_t *session; |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 395 | svm_fifo_t *fifo; |
| 396 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 397 | ASSERT (pool_elts (sm->segments) != 0); |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 398 | |
| 399 | /* Across all fifo segments used by the server */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 400 | /* *INDENT-OFF* */ |
| 401 | segment_manager_foreach_segment_w_lock (fifo_segment, sm, ({ |
| 402 | fifo = svm_fifo_segment_get_fifo_list (fifo_segment); |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 403 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 404 | /* |
| 405 | * Remove any residual sessions from the session lookup table |
| 406 | * Don't bother deleting the individual fifos, we're going to |
| 407 | * throw away the fifo segment in a minute. |
| 408 | */ |
| 409 | while (fifo) |
| 410 | { |
Florin Coras | 653e43f | 2019-03-04 10:56:23 -0800 | [diff] [blame] | 411 | session = session_get_if_valid (fifo->master_session_index, |
| 412 | fifo->master_thread_index); |
| 413 | if (session) |
| 414 | vec_add1 (handles, session_handle (session)); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 415 | fifo = fifo->next; |
| 416 | } |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 417 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 418 | /* Instead of removing the segment, test when cleaning up disconnected |
| 419 | * sessions if the segment can be removed. |
| 420 | */ |
| 421 | })); |
| 422 | /* *INDENT-ON* */ |
Florin Coras | 2b81e3c | 2019-02-27 07:55:46 -0800 | [diff] [blame] | 423 | |
| 424 | vec_foreach (handle, handles) |
| 425 | session_close (session_get_from_handle (*handle)); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 426 | } |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 427 | |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 428 | /** |
| 429 | * Removes segment manager. |
| 430 | * |
| 431 | * Since the fifos allocated in the segment keep backpointers to the sessions |
| 432 | * prior to removing the segment, we call session disconnect. This |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 433 | * subsequently propagates into transport. |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 434 | */ |
| 435 | void |
| 436 | segment_manager_del (segment_manager_t * sm) |
| 437 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 438 | segment_manager_main_t *smm = &segment_manager_main; |
| 439 | svm_fifo_segment_private_t *fifo_segment; |
Dave Wallace | 7b749fe | 2017-07-05 14:30:46 -0400 | [diff] [blame] | 440 | |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 441 | ASSERT (!segment_manager_has_fifos (sm) |
| 442 | && segment_manager_app_detached (sm)); |
| 443 | |
| 444 | /* If we have empty preallocated segments that haven't been removed, remove |
| 445 | * them now. Apart from that, the first segment in the first segment manager |
| 446 | * is not removed when all fifos are removed. It can only be removed when |
| 447 | * the manager is explicitly deleted/detached by the app. */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 448 | clib_rwlock_writer_lock (&sm->segments_rwlock); |
| 449 | |
| 450 | /* *INDENT-OFF* */ |
| 451 | pool_foreach (fifo_segment, sm->segments, ({ |
| 452 | segment_manager_del_segment (sm, fifo_segment); |
| 453 | })); |
| 454 | /* *INDENT-ON* */ |
| 455 | |
| 456 | clib_rwlock_writer_unlock (&sm->segments_rwlock); |
| 457 | |
| 458 | clib_rwlock_free (&sm->segments_rwlock); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 459 | if (CLIB_DEBUG) |
Dave Barach | b7b9299 | 2018-10-17 10:38:51 -0400 | [diff] [blame] | 460 | clib_memset (sm, 0xfe, sizeof (*sm)); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 461 | pool_put (smm->segment_managers, sm); |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 462 | } |
| 463 | |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 464 | void |
| 465 | segment_manager_init_del (segment_manager_t * sm) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 466 | { |
Florin Coras | 4e4531e | 2017-11-06 23:27:56 -0800 | [diff] [blame] | 467 | segment_manager_app_detach (sm); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 468 | if (segment_manager_has_fifos (sm)) |
| 469 | segment_manager_del_sessions (sm); |
| 470 | else |
| 471 | { |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 472 | ASSERT (!sm->first_is_protected || segment_manager_app_detached (sm)); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 473 | segment_manager_del (sm); |
| 474 | } |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 475 | } |
| 476 | |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 477 | int |
| 478 | segment_manager_try_alloc_fifos (svm_fifo_segment_private_t * fifo_segment, |
| 479 | u32 rx_fifo_size, u32 tx_fifo_size, |
| 480 | svm_fifo_t ** rx_fifo, svm_fifo_t ** tx_fifo) |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 481 | { |
| 482 | rx_fifo_size = clib_max (rx_fifo_size, default_fifo_size); |
| 483 | *rx_fifo = svm_fifo_segment_alloc_fifo (fifo_segment, rx_fifo_size, |
| 484 | FIFO_SEGMENT_RX_FREELIST); |
| 485 | |
| 486 | tx_fifo_size = clib_max (tx_fifo_size, default_fifo_size); |
| 487 | *tx_fifo = svm_fifo_segment_alloc_fifo (fifo_segment, tx_fifo_size, |
| 488 | FIFO_SEGMENT_TX_FREELIST); |
| 489 | |
| 490 | if (*rx_fifo == 0) |
| 491 | { |
| 492 | /* This would be very odd, but handle it... */ |
| 493 | if (*tx_fifo != 0) |
| 494 | { |
| 495 | svm_fifo_segment_free_fifo (fifo_segment, *tx_fifo, |
| 496 | FIFO_SEGMENT_TX_FREELIST); |
| 497 | *tx_fifo = 0; |
| 498 | } |
| 499 | return -1; |
| 500 | } |
| 501 | if (*tx_fifo == 0) |
| 502 | { |
| 503 | if (*rx_fifo != 0) |
| 504 | { |
| 505 | svm_fifo_segment_free_fifo (fifo_segment, *rx_fifo, |
| 506 | FIFO_SEGMENT_RX_FREELIST); |
| 507 | *rx_fifo = 0; |
| 508 | } |
| 509 | return -1; |
| 510 | } |
| 511 | |
| 512 | return 0; |
| 513 | } |
| 514 | |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 515 | int |
| 516 | segment_manager_alloc_session_fifos (segment_manager_t * sm, |
Florin Coras | b384b54 | 2018-01-15 01:08:33 -0800 | [diff] [blame] | 517 | svm_fifo_t ** rx_fifo, |
| 518 | svm_fifo_t ** tx_fifo, |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 519 | u32 * fifo_segment_index) |
| 520 | { |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 521 | svm_fifo_segment_private_t *fifo_segment = 0; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 522 | int alloc_fail = 1, rv = 0, new_fs_index; |
Florin Coras | ad0c77f | 2017-11-09 18:00:15 -0800 | [diff] [blame] | 523 | segment_manager_properties_t *props; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 524 | u8 added_a_segment = 0; |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 525 | u64 segment_handle; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 526 | u32 sm_index; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 527 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 528 | props = segment_manager_properties_get (sm); |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 529 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 530 | /* |
| 531 | * Find the first free segment to allocate the fifos in |
| 532 | */ |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 533 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 534 | /* *INDENT-OFF* */ |
| 535 | segment_manager_foreach_segment_w_lock (fifo_segment, sm, ({ |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 536 | alloc_fail = segment_manager_try_alloc_fifos (fifo_segment, |
| 537 | props->rx_fifo_size, |
| 538 | props->tx_fifo_size, |
| 539 | rx_fifo, tx_fifo); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 540 | /* Exit with lock held, drop it after notifying app */ |
| 541 | if (!alloc_fail) |
| 542 | goto alloc_success; |
| 543 | })); |
| 544 | /* *INDENT-ON* */ |
| 545 | |
| 546 | alloc_check: |
| 547 | |
| 548 | if (!alloc_fail) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 549 | { |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 550 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 551 | alloc_success: |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 552 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 553 | ASSERT (rx_fifo && tx_fifo); |
| 554 | sm_index = segment_manager_index (sm); |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 555 | *fifo_segment_index = segment_manager_segment_index (sm, fifo_segment); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 556 | (*tx_fifo)->segment_manager = sm_index; |
| 557 | (*rx_fifo)->segment_manager = sm_index; |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 558 | (*tx_fifo)->segment_index = *fifo_segment_index; |
| 559 | (*rx_fifo)->segment_index = *fifo_segment_index; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 560 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 561 | if (added_a_segment) |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 562 | { |
Florin Coras | 2b81e3c | 2019-02-27 07:55:46 -0800 | [diff] [blame] | 563 | app_worker_t *app_wrk; |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 564 | segment_handle = segment_manager_segment_handle (sm, fifo_segment); |
Florin Coras | 2b81e3c | 2019-02-27 07:55:46 -0800 | [diff] [blame] | 565 | app_wrk = app_worker_get (sm->app_wrk_index); |
| 566 | rv = app_worker_add_segment_notify (app_wrk, segment_handle); |
Florin Coras | fa76a76 | 2018-11-29 12:40:10 -0800 | [diff] [blame] | 567 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 568 | /* Drop the lock after app is notified */ |
| 569 | segment_manager_segment_reader_unlock (sm); |
| 570 | return rv; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 571 | } |
| 572 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 573 | /* |
| 574 | * Allocation failed, see if we can add a new segment |
| 575 | */ |
| 576 | if (props->add_segment) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 577 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 578 | if (added_a_segment) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 579 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 580 | clib_warning ("Added a segment, still can't allocate a fifo"); |
| 581 | segment_manager_segment_reader_unlock (sm); |
| 582 | return SESSION_ERROR_NEW_SEG_NO_SPACE; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 583 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 584 | if ((new_fs_index = segment_manager_add_segment (sm, 0)) < 0) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 585 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 586 | clib_warning ("Failed to add new segment"); |
| 587 | return SESSION_ERROR_SEG_CREATE; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 588 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 589 | fifo_segment = segment_manager_get_segment_w_lock (sm, new_fs_index); |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 590 | alloc_fail = segment_manager_try_alloc_fifos (fifo_segment, |
| 591 | props->rx_fifo_size, |
| 592 | props->tx_fifo_size, |
| 593 | rx_fifo, tx_fifo); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 594 | added_a_segment = 1; |
| 595 | goto alloc_check; |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 596 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 597 | else |
| 598 | { |
| 599 | clib_warning ("Can't add new seg and no space to allocate fifos!"); |
| 600 | return SESSION_ERROR_NO_SPACE; |
| 601 | } |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 602 | } |
| 603 | |
| 604 | void |
Florin Coras | 19223e0 | 2019-03-03 14:56:05 -0800 | [diff] [blame] | 605 | segment_manager_dealloc_fifos (svm_fifo_t * rx_fifo, svm_fifo_t * tx_fifo) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 606 | { |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 607 | svm_fifo_segment_private_t *fifo_segment; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 608 | segment_manager_t *sm; |
Florin Coras | 19223e0 | 2019-03-03 14:56:05 -0800 | [diff] [blame] | 609 | u32 segment_index; |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 610 | |
Florin Coras | 58a93e8 | 2019-01-14 23:33:46 -0800 | [diff] [blame] | 611 | if (!rx_fifo || !tx_fifo) |
| 612 | return; |
| 613 | |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 614 | /* It's possible to have no segment manager if the session was removed |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 615 | * as result of a detach. */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 616 | if (!(sm = segment_manager_get_if_valid (rx_fifo->segment_manager))) |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 617 | return; |
| 618 | |
Florin Coras | 19223e0 | 2019-03-03 14:56:05 -0800 | [diff] [blame] | 619 | segment_index = rx_fifo->segment_index; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 620 | fifo_segment = segment_manager_get_segment_w_lock (sm, segment_index); |
Dave Barach | 10d8cc6 | 2017-05-30 09:30:07 -0400 | [diff] [blame] | 621 | svm_fifo_segment_free_fifo (fifo_segment, rx_fifo, |
| 622 | FIFO_SEGMENT_RX_FREELIST); |
| 623 | svm_fifo_segment_free_fifo (fifo_segment, tx_fifo, |
| 624 | FIFO_SEGMENT_TX_FREELIST); |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 625 | |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 626 | /* |
| 627 | * Try to remove svm segment if it has no fifos. This can be done only if |
| 628 | * the segment is not the first in the segment manager or if it is first |
| 629 | * and it is not protected. Moreover, if the segment is first and the app |
| 630 | * has detached from the segment manager, remove the segment manager. |
| 631 | */ |
| 632 | if (!svm_fifo_segment_has_fifos (fifo_segment)) |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 633 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 634 | segment_manager_segment_reader_unlock (sm); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 635 | |
| 636 | /* Remove segment if it holds no fifos or first but not protected */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 637 | if (segment_index != 0 || !sm->first_is_protected) |
| 638 | segment_manager_lock_and_del_segment (sm, segment_index); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 639 | |
| 640 | /* Remove segment manager if no sessions and detached from app */ |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 641 | if (segment_manager_app_detached (sm) |
| 642 | && !segment_manager_has_fifos (sm)) |
Florin Coras | 568ebc7 | 2018-09-18 16:12:50 -0700 | [diff] [blame] | 643 | { |
| 644 | segment_manager_del (sm); |
| 645 | } |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 646 | } |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 647 | else |
| 648 | segment_manager_segment_reader_unlock (sm); |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 649 | } |
| 650 | |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 651 | u32 |
| 652 | segment_manager_evt_q_expected_size (u32 q_len) |
| 653 | { |
| 654 | u32 fifo_evt_size, notif_q_size, q_hdrs; |
| 655 | u32 msg_q_sz, fifo_evt_ring_sz, session_ntf_ring_sz; |
| 656 | |
Florin Coras | 52207f1 | 2018-07-12 14:48:06 -0700 | [diff] [blame] | 657 | fifo_evt_size = 1 << max_log2 (sizeof (session_event_t)); |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 658 | notif_q_size = clib_max (16, q_len >> 4); |
| 659 | |
| 660 | msg_q_sz = q_len * sizeof (svm_msg_q_msg_t); |
| 661 | fifo_evt_ring_sz = q_len * fifo_evt_size; |
| 662 | session_ntf_ring_sz = notif_q_size * 256; |
| 663 | q_hdrs = sizeof (svm_queue_t) + sizeof (svm_msg_q_t); |
| 664 | |
| 665 | return (msg_q_sz + fifo_evt_ring_sz + session_ntf_ring_sz + q_hdrs); |
| 666 | } |
| 667 | |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 668 | /** |
| 669 | * Allocates shm queue in the first segment |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 670 | * |
| 671 | * Must be called with lock held |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 672 | */ |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 673 | svm_msg_q_t * |
Florin Coras | f8f516a | 2018-02-08 15:10:09 -0800 | [diff] [blame] | 674 | segment_manager_alloc_queue (svm_fifo_segment_private_t * segment, |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 675 | segment_manager_properties_t * props) |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 676 | { |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 677 | u32 fifo_evt_size, session_evt_size = 256, notif_q_size; |
| 678 | svm_msg_q_cfg_t _cfg, *cfg = &_cfg; |
| 679 | svm_msg_q_t *q; |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 680 | void *oldheap; |
| 681 | |
Florin Coras | 52207f1 | 2018-07-12 14:48:06 -0700 | [diff] [blame] | 682 | fifo_evt_size = sizeof (session_event_t); |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 683 | notif_q_size = clib_max (16, props->evt_q_size >> 4); |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 684 | /* *INDENT-OFF* */ |
| 685 | svm_msg_q_ring_cfg_t rc[SESSION_MQ_N_RINGS] = { |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 686 | {props->evt_q_size, fifo_evt_size, 0}, |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 687 | {notif_q_size, session_evt_size, 0} |
| 688 | }; |
| 689 | /* *INDENT-ON* */ |
| 690 | cfg->consumer_pid = 0; |
| 691 | cfg->n_rings = 2; |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 692 | cfg->q_nitems = props->evt_q_size; |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 693 | cfg->ring_cfgs = rc; |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 694 | |
Florin Coras | 3c2fed5 | 2018-07-04 04:15:05 -0700 | [diff] [blame] | 695 | oldheap = ssvm_push_heap (segment->ssvm.sh); |
| 696 | q = svm_msg_q_alloc (cfg); |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 697 | ssvm_pop_heap (oldheap); |
Florin Coras | 9936831 | 2018-08-02 10:45:44 -0700 | [diff] [blame] | 698 | |
| 699 | if (props->use_mq_eventfd) |
| 700 | { |
| 701 | if (svm_msg_q_alloc_producer_eventfd (q)) |
| 702 | clib_warning ("failed to alloc eventfd"); |
| 703 | } |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 704 | return q; |
| 705 | } |
| 706 | |
| 707 | /** |
| 708 | * Frees shm queue allocated in the first segment |
| 709 | */ |
| 710 | void |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 711 | segment_manager_dealloc_queue (segment_manager_t * sm, svm_queue_t * q) |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 712 | { |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 713 | svm_fifo_segment_private_t *segment; |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 714 | ssvm_shared_header_t *sh; |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 715 | void *oldheap; |
| 716 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 717 | ASSERT (!pool_is_free_index (sm->segments, 0)); |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 718 | |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 719 | segment = segment_manager_get_segment_w_lock (sm, 0); |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 720 | sh = segment->ssvm.sh; |
| 721 | |
| 722 | oldheap = ssvm_push_heap (sh); |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 723 | svm_queue_free (q); |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 724 | ssvm_pop_heap (oldheap); |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 725 | segment_manager_segment_reader_unlock (sm); |
| 726 | } |
| 727 | |
| 728 | /* |
| 729 | * Init segment vm address allocator |
| 730 | */ |
| 731 | void |
| 732 | segment_manager_main_init (segment_manager_main_init_args_t * a) |
| 733 | { |
| 734 | segment_manager_main_t *sm = &segment_manager_main; |
| 735 | clib_valloc_chunk_t _ip, *ip = &_ip; |
| 736 | |
| 737 | ip->baseva = a->baseva; |
| 738 | ip->size = a->size; |
| 739 | |
| 740 | clib_valloc_init (&sm->va_allocator, ip, 1 /* lock */ ); |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 741 | } |
| 742 | |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 743 | static clib_error_t * |
| 744 | segment_manager_show_fn (vlib_main_t * vm, unformat_input_t * input, |
| 745 | vlib_cli_command_t * cmd) |
| 746 | { |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 747 | segment_manager_main_t *smm = &segment_manager_main; |
| 748 | svm_fifo_segment_private_t *seg; |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 749 | segment_manager_t *sm; |
Florin Coras | b384b54 | 2018-01-15 01:08:33 -0800 | [diff] [blame] | 750 | u8 show_segments = 0, verbose = 0; |
David Johnson | d9818dd | 2018-12-14 14:53:41 -0500 | [diff] [blame] | 751 | char *address; |
| 752 | size_t size; |
Dave Barach | 91f3e74 | 2017-09-01 19:12:11 -0400 | [diff] [blame] | 753 | u32 active_fifos; |
| 754 | u32 free_fifos; |
| 755 | |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 756 | while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) |
| 757 | { |
| 758 | if (unformat (input, "segments")) |
| 759 | show_segments = 1; |
| 760 | else if (unformat (input, "verbose")) |
| 761 | verbose = 1; |
| 762 | else |
| 763 | return clib_error_return (0, "unknown input `%U'", |
| 764 | format_unformat_error, input); |
| 765 | } |
| 766 | vlib_cli_output (vm, "%d segment managers allocated", |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 767 | pool_elts (smm->segment_managers)); |
| 768 | if (verbose && pool_elts (smm->segment_managers)) |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 769 | { |
| 770 | vlib_cli_output (vm, "%-10s%=15s%=12s", "Index", "App Index", |
| 771 | "Segments"); |
| 772 | |
| 773 | /* *INDENT-OFF* */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 774 | pool_foreach (sm, smm->segment_managers, ({ |
Florin Coras | 1553197 | 2018-08-12 23:50:53 -0700 | [diff] [blame] | 775 | vlib_cli_output (vm, "%-10d%=15d%=12d", segment_manager_index (sm), |
| 776 | sm->app_wrk_index, pool_elts (sm->segments)); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 777 | })); |
| 778 | /* *INDENT-ON* */ |
| 779 | |
| 780 | } |
| 781 | if (show_segments) |
| 782 | { |
Florin Coras | 2f8d8fa | 2018-01-26 06:36:04 -0800 | [diff] [blame] | 783 | vlib_cli_output (vm, "%-15s%15s%15s%15s%15s%15s", "Name", "Type", |
Dave Barach | 91f3e74 | 2017-09-01 19:12:11 -0400 | [diff] [blame] | 784 | "HeapSize (M)", "ActiveFifos", "FreeFifos", "Address"); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 785 | |
| 786 | /* *INDENT-OFF* */ |
Florin Coras | a332c46 | 2018-01-31 06:52:17 -0800 | [diff] [blame] | 787 | pool_foreach (sm, smm->segment_managers, ({ |
| 788 | segment_manager_foreach_segment_w_lock (seg, sm, ({ |
| 789 | svm_fifo_segment_info (seg, &address, &size); |
| 790 | active_fifos = svm_fifo_segment_num_fifos (seg); |
| 791 | free_fifos = svm_fifo_segment_num_free_fifos (seg, ~0 /* size */); |
| 792 | vlib_cli_output (vm, "%-15v%15U%15llu%15u%15u%15llx", |
| 793 | ssvm_name (&seg->ssvm), format_svm_fifo_segment_type, |
| 794 | seg, size >> 20ULL, active_fifos, free_fifos, |
| 795 | address); |
| 796 | if (verbose) |
| 797 | vlib_cli_output (vm, "%U", format_svm_fifo_segment, seg, verbose); |
| 798 | })); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 799 | })); |
| 800 | /* *INDENT-ON* */ |
| 801 | |
| 802 | } |
| 803 | return 0; |
| 804 | } |
| 805 | |
Florin Coras | ad0c77f | 2017-11-09 18:00:15 -0800 | [diff] [blame] | 806 | /* *INDENT-OFF* */ |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 807 | VLIB_CLI_COMMAND (segment_manager_show_command, static) = |
| 808 | { |
| 809 | .path = "show segment-manager", |
Dave Barach | 91f3e74 | 2017-09-01 19:12:11 -0400 | [diff] [blame] | 810 | .short_help = "show segment-manager [segments][verbose]", |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 811 | .function = segment_manager_show_fn, |
| 812 | }; |
| 813 | /* *INDENT-ON* */ |
| 814 | |
Florin Coras | 6cf30ad | 2017-04-04 23:08:23 -0700 | [diff] [blame] | 815 | /* |
| 816 | * fd.io coding-style-patch-verification: ON |
| 817 | * |
| 818 | * Local Variables: |
| 819 | * eval: (c-set-style "gnu") |
| 820 | * End: |
| 821 | */ |