@@ -131,6 +131,9 @@ int run_consumer(int port, umf_memory_provider_ops_t *provider_ops,
131
131
return -1 ;
132
132
}
133
133
134
+ umf_memory_pool_handle_t pool ;
135
+ umf_result = umfPoolCreate (umfScalablePoolOps (), provider , NULL , 0 , & pool );
136
+
134
137
producer_socket = consumer_connect (port );
135
138
if (producer_socket < 0 ) {
136
139
goto err_umfMemoryProviderDestroy ;
@@ -183,7 +186,7 @@ int run_consumer(int port, umf_memory_provider_ops_t *provider_ops,
183
186
len );
184
187
185
188
void * SHM_ptr ;
186
- umf_result = umfMemoryProviderOpenIPCHandle ( provider , IPC_handle , & SHM_ptr );
189
+ umf_result = umfOpenIPCHandle ( pool , IPC_handle , & SHM_ptr );
187
190
if (umf_result == UMF_RESULT_ERROR_NOT_SUPPORTED ) {
188
191
fprintf (stderr ,
189
192
"[consumer] SKIP: opening the IPC handle is not supported\n" );
@@ -240,8 +243,7 @@ int run_consumer(int port, umf_memory_provider_ops_t *provider_ops,
240
243
241
244
err_closeIPCHandle :
242
245
// we do not know the exact size of the remote shared memory
243
- umf_result = umfMemoryProviderCloseIPCHandle (provider , SHM_ptr ,
244
- sizeof (unsigned long long ));
246
+ umf_result = umfCloseIPCHandle (pool , SHM_ptr , sizeof (unsigned long long ));
245
247
if (umf_result != UMF_RESULT_SUCCESS ) {
246
248
fprintf (stderr , "[consumer] ERROR: closing the IPC handle failed\n" );
247
249
}
@@ -254,6 +256,7 @@ int run_consumer(int port, umf_memory_provider_ops_t *provider_ops,
254
256
255
257
err_umfMemoryProviderDestroy :
256
258
umfMemoryProviderDestroy (provider );
259
+ umfPoolDestroy (pool );
257
260
258
261
if (ret == 0 ) {
259
262
fprintf (stderr , "[consumer] Shutting down (status OK) ...\n" );
@@ -321,6 +324,9 @@ int run_producer(int port, umf_memory_provider_ops_t *provider_ops,
321
324
return -1 ;
322
325
}
323
326
327
+ umf_memory_pool_handle_t pool ;
328
+ umf_result = umfPoolCreate (umfScalablePoolOps (), provider , NULL , 0 , & pool );
329
+
324
330
size_t page_size ;
325
331
umf_result = umfMemoryProviderGetMinPageSize (provider , NULL , & page_size );
326
332
if (umf_result != UMF_RESULT_SUCCESS ) {
@@ -335,28 +341,27 @@ int run_producer(int port, umf_memory_provider_ops_t *provider_ops,
335
341
size_t ptr2_size = 2 * page_size ;
336
342
size_t size_IPC_shared_memory = 3 * page_size ;
337
343
338
- umf_result = umfMemoryProviderAlloc ( provider , ptr1_size , 0 , & ptr1 );
339
- if (umf_result != UMF_RESULT_SUCCESS ) {
344
+ ptr1 = umfPoolMalloc ( pool , ptr1_size );
345
+ if (ptr1 == NULL ) {
340
346
fprintf (stderr , "[producer] ERROR: allocating 1 page failed\n" );
341
347
goto err_umfMemoryProviderDestroy ;
342
348
}
343
349
344
- umf_result = umfMemoryProviderAlloc ( provider , ptr2_size , 0 , & ptr2 );
345
- if (umf_result != UMF_RESULT_SUCCESS ) {
350
+ ptr2 = umfPoolMalloc ( pool , ptr2_size );
351
+ if (ptr2 == NULL ) {
346
352
fprintf (stderr , "[producer] ERROR: allocating 2 pages failed\n" );
347
353
goto err_free_ptr1 ;
348
354
}
349
355
350
- umf_result = umfMemoryProviderAlloc (provider , size_IPC_shared_memory , 0 ,
351
- & IPC_shared_memory );
352
- if (umf_result != UMF_RESULT_SUCCESS ) {
356
+ IPC_shared_memory = umfPoolMalloc (pool , size_IPC_shared_memory );
357
+ if (IPC_shared_memory == NULL ) {
353
358
fprintf (stderr , "[producer] ERROR: allocating 3 pages failed\n" );
354
359
goto err_free_ptr2 ;
355
360
}
356
361
357
362
// get size of the IPC handle
358
363
size_t IPC_handle_size ;
359
- umf_result = umfMemoryProviderGetIPCHandleSize ( provider , & IPC_handle_size );
364
+ umf_result = umfPoolGetIPCHandleSize ( pool , & IPC_handle_size );
360
365
if (umf_result != UMF_RESULT_SUCCESS ) {
361
366
fprintf (stderr ,
362
367
"[producer] ERROR: getting size of the IPC handle failed\n" );
@@ -383,8 +388,8 @@ int run_producer(int port, umf_memory_provider_ops_t *provider_ops,
383
388
SHM_number_1 );
384
389
385
390
// get the IPC handle from the OS memory provider
386
- umf_result = umfMemoryProviderGetIPCHandle (
387
- provider , IPC_shared_memory , size_IPC_shared_memory , IPC_handle );
391
+ umf_result = umfGetIPCHandle ( pool , IPC_shared_memory ,
392
+ size_IPC_shared_memory , IPC_handle );
388
393
if (umf_result != UMF_RESULT_SUCCESS ) {
389
394
fprintf (stderr ,
390
395
"[producer] ERROR: getting the IPC handle from the OS memory "
@@ -494,7 +499,7 @@ int run_producer(int port, umf_memory_provider_ops_t *provider_ops,
494
499
close (producer_socket );
495
500
496
501
err_PutIPCHandle :
497
- umf_result = umfMemoryProviderPutIPCHandle (provider , IPC_handle );
502
+ umf_result = umfPutIPCHandle (provider , IPC_handle );
498
503
if (umf_result != UMF_RESULT_SUCCESS ) {
499
504
fprintf (stderr , "[producer] ERROR: putting the IPC handle failed\n" );
500
505
}
@@ -503,15 +508,19 @@ int run_producer(int port, umf_memory_provider_ops_t *provider_ops,
503
508
504
509
err_free_IPC_handle :
505
510
free (IPC_handle );
511
+
506
512
err_free_IPC_shared_memory :
507
- (void )umfMemoryProviderFree ( provider , IPC_shared_memory ,
508
- size_IPC_shared_memory );
513
+ (void )umfFree ( IPC_shared_memory );
514
+
509
515
err_free_ptr2 :
510
- (void )umfMemoryProviderFree (provider , ptr2 , ptr2_size );
516
+ (void )umfFree (ptr2 );
517
+
511
518
err_free_ptr1 :
512
- (void )umfMemoryProviderFree (provider , ptr1 , ptr1_size );
519
+ (void )umfFree (ptr1 );
520
+
513
521
err_umfMemoryProviderDestroy :
514
522
umfMemoryProviderDestroy (provider );
523
+ umfPoolDestroy (pool );
515
524
516
525
if (ret == 0 ) {
517
526
fprintf (stderr , "[producer] Shutting down (status OK) ...\n" );
0 commit comments