int main(int ac, char **av) { int c, i, nnodes=0; long node=-1; char *end; char shortopts[array_len(opts)*2 + 1]; struct bitmask *mask = NULL; get_short_opts(opts,shortopts); while ((c = getopt_long(ac, av, shortopts, opts, NULL)) != -1) { switch (c) { case 's': /* --show */ show(); exit(0); case 'H': /* --hardware */ nopolicy(); hardware(); exit(0); case 'i': /* --interleave */ checknuma(); mask = numactl_parse_nodestring(optarg); if (!mask) { printf ("<%s> is invalid\n", optarg); usage(); } errno = 0; setpolicy(MPOL_INTERLEAVE); if (shmfd >= 0) numa_interleave_memory(shmptr, shmlen, mask); else numa_set_interleave_mask(mask); checkerror("setting interleave mask"); break; case 'N': /* --cpunodebind */ case 'c': /* --cpubind */ dontshm("-c/--cpubind/--cpunodebind"); checknuma(); mask = numactl_parse_nodestring(optarg); if (!mask) { printf ("<%s> is invalid\n", optarg); usage(); } errno = 0; check_cpubind(do_shm); did_cpubind = 1; numa_run_on_node_mask(mask); checkerror("sched_setaffinity"); break; case 'C': /* --physcpubind */ { struct bitmask *cpubuf; dontshm("-C/--physcpubind"); cpubuf = numa_parse_cpustring(optarg); if (!cpubuf) { printf ("<%s> is invalid\n", optarg); usage(); } errno = 0; check_cpubind(do_shm); did_cpubind = 1; numa_sched_setaffinity(0, cpubuf); checkerror("sched_setaffinity"); free(cpubuf); break; } case 'm': /* --membind */ checknuma(); setpolicy(MPOL_BIND); mask = numactl_parse_nodestring(optarg); if (!mask) { printf ("<%s> is invalid\n", optarg); usage(); } errno = 0; numa_set_bind_policy(1); if (shmfd >= 0) { numa_tonodemask_memory(shmptr, shmlen, mask); } else { numa_set_membind(mask); } numa_set_bind_policy(0); checkerror("setting membind"); break; case 'p': /* --preferred */ checknuma(); setpolicy(MPOL_PREFERRED); mask = numactl_parse_nodestring(optarg); if (!mask) { printf ("<%s> is invalid\n", optarg); usage(); } for (i=0; i<mask->size; i++) { if (numa_bitmask_isbitset(mask, i)) { node = i; nnodes++; } } if (nnodes != 1) usage(); numa_bitmask_free(mask); errno = 0; numa_set_bind_policy(0); if (shmfd >= 0) numa_tonode_memory(shmptr, shmlen, node); else numa_set_preferred(node); checkerror("setting preferred node"); break; case 'l': /* --local */ checknuma(); setpolicy(MPOL_DEFAULT); errno = 0; if (shmfd >= 0) numa_setlocal_memory(shmptr, shmlen); else numa_set_localalloc(); checkerror("local allocation"); break; case 'S': /* --shm */ check_cpubind(did_cpubind); nopolicy(); attach_sysvshm(optarg, "--shm"); shmattached = 1; break; case 'f': /* --file */ check_cpubind(did_cpubind); nopolicy(); attach_shared(optarg, "--file"); shmattached = 1; break; case 'L': /* --length */ noshm("--length"); shmlen = memsize(optarg); break; case 'M': /* --shmmode */ noshm("--shmmode"); shmmode = strtoul(optarg, &end, 8); if (end == optarg || *end) usage(); break; case 'd': /* --dump */ if (shmfd < 0) complain( "Cannot do --dump without shared memory.\n"); dump_shm(); do_dump = 1; break; case 'D': /* --dump-nodes */ if (shmfd < 0) complain( "Cannot do --dump-nodes without shared memory.\n"); dump_shm_nodes(); do_dump = 1; break; case 't': /* --strict */ did_strict = 1; numa_set_strict(1); break; case 'I': /* --shmid */ shmid = strtoul(optarg, &end, 0); if (end == optarg || *end) usage(); break; case 'u': /* --huge */ noshm("--huge"); shmflags |= SHM_HUGETLB; break; case 'o': /* --offset */ noshm("--offset"); shmoffset = memsize(optarg); break; case 'T': /* --touch */ needshm("--touch"); check_shmbeyond("--touch"); numa_police_memory(shmptr, shmlen); break; case 'V': /* --verify */ needshm("--verify"); if (set_policy < 0) complain("Need a policy first to verify"); check_shmbeyond("--verify"); numa_police_memory(shmptr, shmlen); if (!mask) complain("Need a mask to verify"); else verify_shm(set_policy, mask); break; default: usage(); } } av += optind; ac -= optind; if (shmfd >= 0) { if (*av) usage(); exit(exitcode); } if (did_strict) fprintf(stderr, "numactl: warning. Strict flag for process ignored.\n"); if (do_dump) usage_msg("cannot do --dump|--dump-shm for process"); if (shmoption) usage_msg("shm related option %s for process", shmoption); if (*av == NULL) usage(); execvp(*av, av); complain("execution of `%s': %s\n", av[0], strerror(errno)); return 0; /* not reached */ }
static int parse_opts(int argc, char **argv, int *status) { struct option *long_option_arr = NULL; int option, opt_num, n, i = 0, ret = 0; char buf[256] = { 0 }; char *endptr; long int tmp; if (argc <= 1) { show_usage(); *status = -1; ret = 1; goto out; } get_short_opts(buf, 256 - strlen(short_option)); sprintf(buf + strlen(buf), "%s", short_option); opt_num = get_opt_num() + ARRAY_SIZE(long_option); long_option_arr = calloc(1, opt_num * sizeof(*long_option_arr)); if (!long_option_arr) { fprintf(stderr, "ERROR - unable to allocate memory for parser\n"); *status = -1; ret = 1; goto out; } n = get_long_opts(long_option_arr, opt_num); while (long_option[i].name && n + i < opt_num) { long_option_arr[n + i] = long_option[i]; i++; } do { option = getopt_long(argc, argv, buf, long_option_arr, NULL); switch (option) { case 'l': tmp = strtol(optarg, &endptr, 10); if (endptr == optarg) { fprintf(stderr, "ERROR - no digits were found in option -%c\n", option); ret = 1; goto out; } if (errno == ERANGE && (tmp == LONG_MAX || tmp == LONG_MIN) ) { fprintf(stderr, "ERROR - out of range in option -%c\n", option); ret = 1; goto out; } if (tmp <0 || tmp >= IB_LID_MCAST_START) { fprintf(stderr, "ERROR - invalid lid %ld in option -l\n", tmp); ret = 1; goto out; } dest_lid = tmp; break; case 'g': dest_gid = optarg; break; case 'v': show_version(); *status = 0; ret = 1; goto out; case 'd': ca_name = optarg; break; case 'P': tmp = strtol(optarg, &endptr, 10); if (endptr == optarg) { fprintf(stderr, "ERROR - no digits were found in option -%c\n", option); ret = 1; goto out; } if (errno == ERANGE && (tmp == LONG_MAX || tmp == LONG_MIN) ) { fprintf(stderr, "ERROR - out of range in option -%c\n", option); ret = 1; goto out; } if (tmp < 0) { fprintf(stderr, "ERROR - invalid value %ld in option -%c\n", tmp, option); ret = 1; goto out; } src_port = tmp; break; case 't': tmp = strtol(optarg, &endptr, 10); if (endptr == optarg) { fprintf(stderr, "ERROR - no digits were found in option -%c\n", option); ret = 1; goto out; } if (errno == ERANGE && (tmp = LONG_MAX || tmp == LONG_MIN) ) { fprintf(stderr, "ERROR - out of range in option -%c\n", option); ret = 1; goto out; } if (tmp < 0) fprintf(stderr, "WARNING - infinite timeout is used\n"); else if (tmp == 0) fprintf(stderr, "WARNING - operation timeout is 0\n"); timeout = tmp; break; case 'p': pkey = (uint16_t) strtoul(optarg, NULL, 0); break; case 'a': admin_port = atoi(optarg); break; case 'r': if (!optarg || !strcmp(optarg, "d")) { recursive = ADMIN_RECURSION_DOWN; } else if (!strcmp(optarg, "u")) { recursive = ADMIN_RECURSION_UP; } else { fprintf(stderr, "ERROR - out of range in option -%c\n", option); ret = 1; goto out; } break; case '?': case 'h': show_usage(); *status = 0; ret = 1; goto out; default: break; } } while (option != -1); if (dest_lid && dest_gid) { fprintf(stderr, "Destination address ambiguity: " "both GID and LID are specified\n"); *status = -1; ret = 1; goto out; } if (optind == argc) { fprintf(stderr, "No command specified\n"); show_usage(); *status = -1; ret = 1; goto out; } out: if (long_option_arr) free(long_option_arr); return ret; }