int main(int argc, char *argv[]) { struct blk_arguments arguments; /* set the random seed value */ srand(time(NULL)); /* set default values */ memset(&arguments, 0, sizeof (struct blk_arguments)); arguments.block_size = 512; arguments.num_ops = 100; arguments.file_size = (PMEMBLK_MIN_POOL / 1024) / 1024; if (argp_parse(&argp, argc, argv, 0, 0, &arguments) != 0) { exit(1); } struct worker_info worker_params[arguments.thread_count]; memset(worker_params, 0, sizeof (struct worker_info)); /* set common values */ worker_params[0].block_size = arguments.block_size; worker_params[0].num_ops = arguments.num_ops; worker_params[0].file_lanes = arguments.thread_count; /* file_size is provided in MB */ unsigned long long file_size_bytes = arguments.file_size * 1024 * 1024; worker *thread_workers = NULL; /* prepare parameters specific for file/pmem */ if (arguments.file_io) { /* prepare open flags */ int flags = O_RDWR | O_CREAT | O_SYNC; /* create file on PMEM-aware file system */ if ((worker_params[0].file_desc = open(arguments.file_path, flags, FILE_MODE)) < 0) { perror(arguments.file_path); exit(1); } /* pre-allocate file_size MB of persistent memory */ if ((errno = posix_fallocate(worker_params[0].file_desc, (off_t)0, (off_t)file_size_bytes)) != 0) { warn("posix_fallocate"); close(worker_params[0].file_desc); exit(1); } worker_params[0].num_blocks = file_size_bytes / worker_params[0].block_size; thread_workers = file_workers; } else { worker_params[0].file_desc = -1; if (arguments.prep_blk_file) { if ((worker_params[0].handle = pmemblk_create( arguments.file_path, worker_params[0].block_size, (off_t)file_size_bytes, FILE_MODE)) == NULL) { err(1, "%s: pmemblk_open", argv[2]); } } else { if ((worker_params[0].handle = pmemblk_open( arguments.file_path, worker_params[0].block_size)) == NULL) { err(1, "%s: pmemblk_open", argv[2]); } } worker_params[0].num_blocks = pmemblk_nblock( worker_params[0].handle); thread_workers = pmem_workers; } /* propagate params to each info_t */ for (int i = 1; i < arguments.thread_count; ++i) { memcpy(&worker_params[i], &worker_params[0], sizeof (struct worker_info)); worker_params[i].thread_index = i; worker_params[i].seed = rand(); } /* The blk mode file prep */ if (arguments.prep_blk_file) { if (worker_params[0].file_desc >= 0) close(worker_params[0].file_desc); return run_threads(prep_worker, arguments.thread_count, worker_params); } struct measurements perf_meas; perf_meas.total_ops = arguments.thread_count * worker_params[0].num_ops; /* perform PMEMBLK warmup */ if (!arguments.file_io) { if (run_threads(warmup_worker, arguments.thread_count, worker_params) != 0) { if (worker_params[0].file_desc >= 0) close(worker_params[0].file_desc); exit(1); } } for (int i = 0; i < WORKER_COUNT_MAX; ++i) { clock_gettime(CLOCK_MONOTONIC, &perf_meas.start_time); if (run_threads(thread_workers[i], arguments.thread_count, worker_params) != 0) { if (worker_params[0].file_desc >= 0) close(worker_params[0].file_desc); exit(1); } clock_gettime(CLOCK_MONOTONIC, &perf_meas.stop_time); calculate_stats(&perf_meas); printf("%f;%f;", perf_meas.total_run_time, perf_meas.ops_per_second); } printf("\n"); if (worker_params[0].file_desc >= 0) close(worker_params[0].file_desc); /* cleanup and check pmem file */ if (!arguments.file_io) { pmemblk_close(worker_params[0].handle); /* not really necessary, but check consistency */ int result = pmemblk_check(arguments.file_path); if (result < 0) { warn("%s: pmemblk_check", arguments.file_path); } else if (result == 0) { warnx("%s: pmemblk_check: not consistent", arguments.file_path); } } exit(0); }
int main(int argc, char *argv[]) { START(argc, argv, "blk_non_zero"); if (argc < 5) UT_FATAL("usage: %s bsize file func [file_size] op:lba...", argv[0]); int read_arg = 1; Bsize = strtoul(argv[read_arg++], NULL, 0); const char *path = argv[read_arg++]; PMEMblkpool *handle = NULL; switch (*argv[read_arg++]) { case 'c': { size_t fsize = strtoul(argv[read_arg++], NULL, 0); handle = pmemblk_create(path, Bsize, fsize, S_IRUSR | S_IWUSR); if (handle == NULL) UT_FATAL("!%s: pmemblk_create", path); break; } case 'o': handle = pmemblk_open(path, Bsize); if (handle == NULL) UT_FATAL("!%s: pmemblk_open", path); break; default: UT_FATAL("unrecognized command %s", argv[read_arg - 1]); } UT_OUT("%s block size %zu usable blocks %zu", argv[1], Bsize, pmemblk_nblock(handle)); UT_OUT("is zeroed:\t%d", is_zeroed(path)); unsigned char *buf = MALLOC(Bsize); if (buf == NULL) UT_FATAL("cannot allocate buf"); /* map each file argument with the given map type */ for (; read_arg < argc; read_arg++) { if (strchr("rwze", argv[read_arg][0]) == NULL || argv[read_arg][1] != ':') UT_FATAL("op must be r: or w: or z: or e:"); off_t lba = strtoul(&argv[read_arg][2], NULL, 0); switch (argv[read_arg][0]) { case 'r': if (pmemblk_read(handle, buf, lba) < 0) UT_OUT("!read lba %zu", lba); else UT_OUT("read lba %zu: %s", lba, ident(buf)); break; case 'w': construct(buf); if (pmemblk_write(handle, buf, lba) < 0) UT_OUT("!write lba %zu", lba); else UT_OUT("write lba %zu: %s", lba, ident(buf)); break; case 'z': if (pmemblk_set_zero(handle, lba) < 0) UT_OUT("!set_zero lba %zu", lba); else UT_OUT("set_zero lba %zu", lba); break; case 'e': if (pmemblk_set_error(handle, lba) < 0) UT_OUT("!set_error lba %zu", lba); else UT_OUT("set_error lba %zu", lba); break; } } FREE(buf); pmemblk_close(handle); int result = pmemblk_check(path, Bsize); if (result < 0) UT_OUT("!%s: pmemblk_check", path); else if (result == 0) UT_OUT("%s: pmemblk_check: not consistent", path); DONE(NULL); }
int main(int argc, char *argv[]) { START(argc, argv, "out_err_mt"); if (argc != 5) UT_FATAL("usage: %s filename1 filename2 filename3 dir", argv[0]); PMEMobjpool *pop = pmemobj_create(argv[1], "test", PMEMOBJ_MIN_POOL, 0666); PMEMlogpool *plp = pmemlog_create(argv[2], PMEMLOG_MIN_POOL, 0666); PMEMblkpool *pbp = pmemblk_create(argv[3], 128, PMEMBLK_MIN_POOL, 0666); VMEM *vmp = vmem_create(argv[4], VMEM_MIN_POOL); util_init(); pmem_check_version(10000, 0); pmemobj_check_version(10001, 0); pmemlog_check_version(10002, 0); pmemblk_check_version(10003, 0); vmem_check_version(10004, 0); pmempool_check_version(10005, 0); print_errors("version check"); void *ptr = NULL; /* * We are testing library error reporting and we don't want this test * to fail under memcheck. */ VALGRIND_DO_DISABLE_ERROR_REPORTING; pmem_msync(ptr, 1); VALGRIND_DO_ENABLE_ERROR_REPORTING; print_errors("pmem_msync"); pmemlog_append(plp, NULL, PMEMLOG_MIN_POOL); print_errors("pmemlog_append"); size_t nblock = pmemblk_nblock(pbp); pmemblk_set_error(pbp, nblock + 1); print_errors("pmemblk_set_error"); VMEM *vmp2 = vmem_create_in_region(NULL, 1); UT_ASSERTeq(vmp2, NULL); print_errors("vmem_create_in_region"); run_mt_test(do_test); pmemobj_close(pop); pmemlog_close(plp); pmemblk_close(pbp); vmem_delete(vmp); PMEMpoolcheck *ppc; struct pmempool_check_args args = {0, }; ppc = pmempool_check_init(&args, sizeof(args) / 2); UT_ASSERTeq(ppc, NULL); print_errors("pmempool_check_init"); DONE(NULL); }