Line data    Source code

       1             : #include "../fd_util.h"
       2             : 
       3             : #if FD_HAS_HOSTED
       4             : 
       5             : #include <stdio.h>
       6             : #include <errno.h>
       7             : #include <sys/stat.h>
       8             :  
       9             : FD_IMPORT_CSTR( fd_shmem_ctl_help, "src/util/shmem/fd_shmem_ctl_help" );
      10             : 
      11             : int
      12             : main( int     argc,
      13         141 :       char ** argv ) {
      14         141 :   fd_boot( &argc, &argv );
      15         354 : # define SHIFT(n) argv+=(n),argc-=(n)
      16             : 
      17         141 :   if( FD_UNLIKELY( argc<1 ) ) FD_LOG_ERR(( "No arguments" ));
      18         141 :   char const * bin = argv[0];
      19         141 :   SHIFT(1);
      20             :   
      21         141 :   umask( (mode_t)0 ); /* So mode setting gets respected */
      22             : 
      23         141 :   int cnt = 0;
      24         198 :   while( argc ) {
      25         168 :     char const * cmd = argv[0];
      26         168 :     SHIFT(1);
      27             : 
      28         168 :     if( !strcmp( cmd, "help" ) ) {
      29             : 
      30           3 :       fputs( fd_shmem_ctl_help, stdout );
      31             : 
      32           3 :       FD_LOG_NOTICE(( "%i: %s: success", cnt, cmd ));
      33             : 
      34         165 :     } else if( !strcmp( cmd, "cpu-cnt" ) ) {
      35             : 
      36           3 :       printf( "%lu\n", fd_shmem_cpu_cnt() );
      37             : 
      38           3 :       FD_LOG_NOTICE(( "%i: %s: success", cnt, cmd ));
      39             : 
      40         162 :     } else if( !strcmp( cmd, "numa-cnt" ) ) {
      41             : 
      42           6 :       printf( "%lu\n", fd_shmem_numa_cnt() );
      43             : 
      44           6 :       FD_LOG_NOTICE(( "%i: %s: success", cnt, cmd ));
      45             : 
      46         156 :     } else if( !strcmp( cmd, "cpu-idx" ) ) {
      47             : 
      48           6 :       if( FD_UNLIKELY( argc<1 ) ) FD_LOG_ERR(( "%i: %s: too few arguments\n\tDo %s help for help", cnt, cmd, bin ));
      49             : 
      50           3 :       ulong numa_idx = fd_cstr_to_ulong( argv[0] );
      51             : 
      52           3 :       ulong cpu_idx = fd_shmem_cpu_idx( numa_idx );
      53             : 
      54           3 :       if( FD_LIKELY( cpu_idx<ULONG_MAX ) ) printf( "%lu\n", cpu_idx );
      55           0 :       else                                 printf( "-\n" );
      56             : 
      57           3 :       FD_LOG_NOTICE(( "%i: %s %lu: success", cnt, cmd, numa_idx ));
      58           3 :       SHIFT(1);
      59             : 
      60         150 :     } else if( !strcmp( cmd, "numa-idx" ) ) {
      61             : 
      62           6 :       if( FD_UNLIKELY( argc<1 ) ) FD_LOG_ERR(( "%i: %s: too few arguments\n\tDo %s help for help", cnt, cmd, bin ));
      63             : 
      64           3 :       ulong cpu_idx = fd_cstr_to_ulong( argv[0] );
      65             : 
      66           3 :       ulong numa_idx = fd_shmem_numa_idx( cpu_idx );
      67             : 
      68           3 :       if( FD_LIKELY( numa_idx<ULONG_MAX ) ) printf( "%lu\n", numa_idx );
      69           0 :       else                                  printf( "-\n" );
      70             : 
      71           3 :       FD_LOG_NOTICE(( "%i: %s %lu: success", cnt, cmd, cpu_idx ));
      72           3 :       SHIFT(1);
      73             : 
      74         144 :     } else if( !strcmp( cmd, "create" ) ) {
      75             : 
      76          84 :       if( FD_UNLIKELY( argc<5 ) ) FD_LOG_ERR(( "%i: %s: too few arguments\n\tDo %s help for help", cnt, cmd, bin ));
      77             : 
      78          69 :       char const * name     =                           argv[0];
      79          69 :       ulong        page_cnt = fd_cstr_to_ulong        ( argv[1] );
      80          69 :       ulong        page_sz  = fd_cstr_to_shmem_page_sz( argv[2] );
      81          69 :       char const * seq      =                           argv[3];
      82          69 :       ulong        mode     = fd_cstr_to_ulong_octal  ( argv[4] );
      83             : 
      84          69 :       ulong sub_page_cnt[ 512 ];
      85          69 :       ulong sub_cpu_idx [ 512 ];
      86          69 :       ulong sub_cnt = fd_cstr_to_ulong_seq( seq, sub_cpu_idx, 512UL );
      87             : 
      88          69 :       if( FD_UNLIKELY( !sub_cnt ) )
      89          42 :         FD_LOG_ERR(( "%i: %s %s %lu %s %s 0%03lo: empty or invalid cpu sequence\n\t"
      90          69 :                      "Do %s help for help", cnt, cmd, name, page_cnt, argv[2], seq, mode, bin ));
      91             : 
      92          27 :       if( FD_UNLIKELY( sub_cnt>512UL ) )
      93           3 :         FD_LOG_ERR(( "%i: %s %s %lu %s %s 0%03lo: sequence too long, increase limit in fd_shmem_ctl.c\n\t"
      94          27 :                      "Do %s help for help", cnt, cmd, name, page_cnt, argv[2], seq, mode, bin ));
      95             : 
      96          24 :       ulong sub_page_min = page_cnt / sub_cnt;
      97          24 :       ulong sub_page_rem = page_cnt % sub_cnt;
      98          48 :       for( ulong sub_idx=0UL; sub_idx<sub_cnt; sub_idx++ ) sub_page_cnt[ sub_idx ] = sub_page_min + (ulong)(sub_idx<sub_page_rem);
      99             : 
     100          24 :       if( FD_UNLIKELY( fd_shmem_create_multi( name, page_sz, sub_cnt, sub_page_cnt, sub_cpu_idx, mode ) ) )
     101          18 :         FD_LOG_ERR(( "%i: %s %s %lu %s %s 0%03lo: FAIL\n\t"
     102          24 :                      "Do %s help for help", cnt, cmd, name, page_cnt, argv[2], seq, mode, bin ));
     103             : 
     104           6 :       FD_LOG_NOTICE(( "%i: %s %s %lu %s %s 0%03lo: success", cnt, cmd, name, page_cnt, argv[2], seq, mode ));
     105           6 :       SHIFT(5);
     106             : 
     107          60 :     } else if( !strcmp( cmd, "unlink" ) ) {
     108             : 
     109          21 :       if( FD_UNLIKELY( argc<2 ) ) FD_LOG_ERR(( "%i: %s: too few arguments\n\tDo %s help for help", cnt, cmd, bin ));
     110             : 
     111          12 :       char const * name     =                           argv[0];
     112          12 :       ulong        page_sz  = fd_cstr_to_shmem_page_sz( argv[1] );
     113             : 
     114          12 :       if( !page_sz ) {
     115           6 :         fd_shmem_info_t info[1];
     116           6 :         if( FD_UNLIKELY( fd_shmem_info( name, page_sz, info ) ) )
     117           3 :           FD_LOG_ERR(( "%i: %s %s %s: not found or bad permissions\n\tDo %s help for help", cnt, cmd, name, argv[1], bin ));
     118           3 :         page_sz = info->page_sz;
     119           3 :       }
     120             : 
     121           9 :       if( FD_UNLIKELY( fd_shmem_unlink( name, page_sz ) ) )
     122           3 :         FD_LOG_ERR(( "%i: %s %s %s: FAIL\n\tDo %s help for help", cnt, cmd, name, argv[1], bin ));
     123             : 
     124           6 :       FD_LOG_NOTICE(( "%i: %s %s %s: success", cnt, cmd, name, argv[1]));
     125           6 :       SHIFT(2);
     126             : 
     127          39 :     } else if( !strcmp( cmd, "query" ) ) {
     128             : 
     129             :       /* FIXME: MAKE THIS MORE LIKE POD QUERY WITH "WHAT" FIELDS */
     130             : 
     131          33 :       if( FD_UNLIKELY( argc<2 ) ) FD_LOG_ERR(( "%i: %s: too few arguments\n\tDo %s help for help", cnt, cmd, bin ));
     132             : 
     133          27 :       char const * name     =                           argv[0];
     134          27 :       ulong        page_sz  = fd_cstr_to_shmem_page_sz( argv[1] );
     135             : 
     136          27 :       fd_shmem_info_t info[1];
     137          27 :       int err = fd_shmem_info( name, page_sz, info );
     138          27 :       if( FD_UNLIKELY( err ) ) printf( "%i 0 0\n",    err );
     139          12 :       else                     printf( "0 %lu %lu\n", info->page_cnt, info->page_sz );
     140             : 
     141          27 :       FD_LOG_NOTICE(( "%i: %s %s %s: success", cnt, cmd, name, argv[1]));
     142          27 :       SHIFT(2);
     143             : 
     144          27 :     } else {
     145             : 
     146           6 :       FD_LOG_ERR(( "%i: %s: unknown command\n\t"
     147           6 :                    "Do %s help for help", cnt, cmd, bin ));
     148             : 
     149           6 :     }
     150          57 :     cnt++;
     151          57 :   }
     152             : 
     153          30 :   FD_LOG_NOTICE(( "processed %i commands", cnt ));
     154             : 
     155          30 : # undef SHIFT
     156          30 :   fd_halt();
     157          30 :   return 0;
     158         141 : }
     159             : 
     160             : #else
     161             : 
     162             : int
     163             : main( int     argc,
     164             :       char ** argv ) {
     165             :   fd_boot( &argc, &argv );
     166             :   if( FD_UNLIKELY( argc<1 ) ) FD_LOG_ERR(( "No arguments" ));
     167             :   if( FD_UNLIKELY( argc>1 ) ) FD_LOG_ERR(( "fd_shmem_ctl not supported on this platform" ));
     168             :   FD_LOG_NOTICE(( "processed 0 commands" ));
     169             :   fd_halt();
     170             :   return 0;
     171             : }
     172             : 
     173             : #endif