static void pool_create(const char *path, size_t bsize, size_t poolsize, unsigned mode) { PMEMblkpool *pbp = pmemblk_create(path, bsize, poolsize, mode); if (pbp == NULL) UT_OUT("!%s: pmemblk_create", path); else { struct stat stbuf; STAT(path, &stbuf); UT_OUT("%s: file size %zu usable blocks %zu mode 0%o", path, stbuf.st_size, pmemblk_nblock(pbp), stbuf.st_mode & 0777); pmemblk_close(pbp); 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); else UT_ASSERTeq(pmemblk_check(path, bsize * 2), -1); } }
int main(int argc, char *argv[]) { START(argc, argv, "blk_nblock"); if (argc < 2) FATAL("usage: %s bsize:file...", argv[0]); /* map each file argument with the given map type */ for (int arg = 1; arg < argc; arg++) { char *fname; size_t bsize = strtoul(argv[arg], &fname, 0); if (*fname != ':') FATAL("usage: %s bsize:file...", argv[0]); fname++; PMEMblkpool *handle; handle = pmemblk_create(fname, bsize, 0, S_IWUSR | S_IRUSR); if (handle == NULL) { OUT("!%s: pmemblk_create", fname); } else { OUT("%s: block size %zu usable blocks: %zu", fname, bsize, pmemblk_nblock(handle)); ASSERTeq(pmemblk_bsize(handle), bsize); pmemblk_close(handle); int result = pmemblk_check(fname, bsize); if (result < 0) OUT("!%s: pmemblk_check", fname); else if (result == 0) OUT("%s: pmemblk_check: not consistent", fname); else { ASSERTeq(pmemblk_check(fname, bsize + 1), -1); ASSERTeq(pmemblk_check(fname, 0), 1); handle = pmemblk_open(fname, 0); ASSERTeq(pmemblk_bsize(handle), bsize); pmemblk_close(handle); } } } DONE(NULL); }
/* * blk_exit -- function for de-initialization benchmark */ static int blk_exit(struct benchmark *bench, struct benchmark_args *args) { struct blk_bench *bb = (struct blk_bench *)pmembench_get_priv(bench); struct blk_args *ba = (struct blk_args *)args->opts; if (ba->file_io) { os_close(bb->fd); } else { pmemblk_close(bb->pbp); int result = pmemblk_check(args->fname, args->dsize); if (result < 0) { perror("pmemblk_check error"); return -1; } else if (result == 0) { perror("pmemblk_check: not consistent"); return -1; } } free(bb); return 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)); /* 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); unsigned char buf[Bsize]; 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; } } 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, "blk_recovery"); if (argc != 5) FATAL("usage: %s bsize file first_lba lba", argv[0]); Bsize = strtoul(argv[1], NULL, 0); const char *path = argv[2]; PMEMblkpool *handle; if ((handle = pmemblk_create(path, Bsize, 0, S_IWUSR | S_IRUSR)) == NULL) FATAL("!%s: pmemblk_create", path); OUT("%s block size %zu usable blocks %zu", argv[1], Bsize, pmemblk_nblock(handle)); /* write the first lba */ off_t lba = strtoul(argv[3], NULL, 0); unsigned char buf[Bsize]; construct(buf); if (pmemblk_write(handle, buf, lba) < 0) FATAL("!write lba %zu", lba); OUT("write lba %zu: %s", lba, ident(buf)); /* reach into the layout and write-protect the map */ struct btt_info *infop = (void *)handle + roundup(sizeof (struct pmemblk), BLK_FORMAT_DATA_ALIGN); void *mapaddr = (void *)infop + le32toh(infop->mapoff); void *flogaddr = (void *)infop + le32toh(infop->flogoff); OUT("write-protecting map, length %zu", (size_t)(flogaddr - mapaddr)); MPROTECT(mapaddr, (size_t)(flogaddr - mapaddr), PROT_READ); /* arrange to catch SEGV */ struct sigaction v; sigemptyset(&v.sa_mask); v.sa_flags = 0; v.sa_handler = signal_handler; SIGACTION(SIGSEGV, &v, NULL); /* map each file argument with the given map type */ lba = strtoul(argv[4], NULL, 0); construct(buf); if (!sigsetjmp(Jmp, 1)) { if (pmemblk_write(handle, buf, lba) < 0) FATAL("!write lba %zu", lba); else FATAL("write lba %zu: %s", lba, ident(buf)); } pmemblk_close(handle); int result = pmemblk_check(path); if (result < 0) OUT("!%s: pmemblk_check", path); else if (result == 0) OUT("%s: pmemblk_check: not consistent", path); else OUT("%s: consistent", path); DONE(NULL); }
int main(int argc, char *argv[]) { START(argc, argv, "blk_rw"); if (argc < 5) FATAL("usage: %s bsize file func op:lba...", argv[0]); Bsize = strtoul(argv[1], NULL, 0); const char *path = argv[2]; PMEMblkpool *handle; switch (*argv[3]) { case 'c': handle = pmemblk_create(path, Bsize, 0, S_IWUSR | S_IRUSR); if (handle == NULL) FATAL("!%s: pmemblk_create", path); break; case 'o': handle = pmemblk_open(path, Bsize); if (handle == NULL) FATAL("!%s: pmemblk_open", path); break; } OUT("%s block size %zu usable blocks %zu", argv[1], Bsize, pmemblk_nblock(handle)); /* map each file argument with the given map type */ for (int arg = 4; arg < argc; arg++) { if (strchr("rwze", argv[arg][0]) == NULL || argv[arg][1] != ':') FATAL("op must be r: or w: or z: or e:"); off_t lba = strtol(&argv[arg][2], NULL, 0); unsigned char buf[Bsize]; switch (argv[arg][0]) { case 'r': if (pmemblk_read(handle, buf, lba) < 0) OUT("!read lba %jd", lba); else OUT("read lba %jd: %s", lba, ident(buf)); break; case 'w': construct(buf); if (pmemblk_write(handle, buf, lba) < 0) OUT("!write lba %jd", lba); else OUT("write lba %jd: %s", lba, ident(buf)); break; case 'z': if (pmemblk_set_zero(handle, lba) < 0) OUT("!set_zero lba %jd", lba); else OUT("set_zero lba %jd", lba); break; case 'e': if (pmemblk_set_error(handle, lba) < 0) OUT("!set_error lba %jd", lba); else OUT("set_error lba %jd", lba); break; } } pmemblk_close(handle); int result = pmemblk_check(path, Bsize); if (result < 0) OUT("!%s: pmemblk_check", path); else if (result == 0) OUT("%s: pmemblk_check: not consistent", path); DONE(NULL); }
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); }