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