*
* See examples/pmemblk.fio for more.
*
- * libpmemblk.so
- * By default, the pmemblk engine will let the system find the libpmemblk.so
- * that it uses. You can use an alternative libpmemblk by setting the
- * FIO_PMEMBLK_LIB environment variable to the full path to the desired
- * libpmemblk.so.
- *
*/
#include <stdio.h>
#include <sys/uio.h>
#include <errno.h>
#include <assert.h>
-#include <dlfcn.h>
#include <string.h>
+#include <libpmem.h>
+#include <libpmemblk.h>
#include "../fio.h"
/*
* libpmemblk
*/
-struct PMEMblkpool_s;
-typedef struct PMEMblkpool_s PMEMblkpool;
-
-PMEMblkpool *(*pmemblk_create) (const char *, size_t, size_t, mode_t) = NULL;
-PMEMblkpool *(*pmemblk_open) (const char *, size_t) = NULL;
-void (*pmemblk_close) (PMEMblkpool *) = NULL;
-size_t(*pmemblk_nblock) (PMEMblkpool *) = NULL;
-size_t(*pmemblk_bsize) (PMEMblkpool *) = NULL;
-int (*pmemblk_read) (PMEMblkpool *, void *, off_t) = NULL;
-int (*pmemblk_write) (PMEMblkpool *, const void *, off_t) = NULL;
-
-int load_libpmemblk(const char *path)
-{
- void *dl;
-
- if (NULL == path)
- path = "libpmemblk.so";
-
- dl = dlopen(path, RTLD_NOW | RTLD_NODELETE);
- if (NULL == dl)
- goto errorout;
-
- if (NULL == (pmemblk_create = dlsym(dl, "pmemblk_create")))
- goto errorout;
- if (NULL == (pmemblk_open = dlsym(dl, "pmemblk_open")))
- goto errorout;
- if (NULL == (pmemblk_close = dlsym(dl, "pmemblk_close")))
- goto errorout;
- if (NULL == (pmemblk_nblock = dlsym(dl, "pmemblk_nblock")))
- goto errorout;
- if (NULL == (pmemblk_bsize = dlsym(dl, "pmemblk_bsize")))
- goto errorout;
- if (NULL == (pmemblk_read = dlsym(dl, "pmemblk_read")))
- goto errorout;
- if (NULL == (pmemblk_write = dlsym(dl, "pmemblk_write")))
- goto errorout;
-
- return 0;
-
-errorout:
- log_err("fio: unable to load libpmemblk: %s\n", dlerror());
- if (NULL != dl)
- dlclose(dl);
-
- return (-1);
-
-} /* load_libpmemblk() */
-
typedef struct fio_pmemblk_file *fio_pmemblk_file_t;
+
struct fio_pmemblk_file {
fio_pmemblk_file_t pmb_next;
char *pmb_filename;
} while(0)
#define FIOFILEPMBGET(_f) ((fio_pmemblk_file_t)((_f)->engine_data))
-static fio_pmemblk_file_t Cache = NULL;
+static fio_pmemblk_file_t Cache;
static pthread_mutex_t CacheLock = PTHREAD_MUTEX_INITIALIZER;
fio_pmemblk_file_t i;
for (i = Cache; i != NULL; i = i->pmb_next)
- if (0 == strcmp(filename, i->pmb_filename))
+ if (!strcmp(filename, i->pmb_filename))
return i;
return NULL;
-
-} /* fio_pmemblk_cache_lookup() */
+}
static void fio_pmemblk_cache_insert(fio_pmemblk_file_t pmb)
{
pmb->pmb_next = Cache;
Cache = pmb;
-
- return;
-
-} /* fio_pmemblk_cache_insert() */
+}
static void fio_pmemblk_cache_remove(fio_pmemblk_file_t pmb)
{
pmb->pmb_next = NULL;
return;
}
-
- return;
-
-} /* fio_pmemblk_cache_remove() */
+}
/*
* to control block size and gross file size at the libpmemblk
* note that the user should specify the file size in MiB, but
* we return bytes from here.
*/
-static void
-pmb_parse_path(const char *pathspec,
- char **ppath, uint64_t * pbsize, uint64_t * pfsize)
+static void pmb_parse_path(const char *pathspec, char **ppath, uint64_t *pbsize,
+ uint64_t *pfsize)
{
char *path;
char *s;
uint64_t fsizemb;
path = strdup(pathspec);
- if (NULL == path) {
+ if (!path) {
*ppath = NULL;
return;
}
*ppath = path;
*pbsize = 0;
*pfsize = 0;
- return;
-
-} /* pmb_parse_path() */
+}
-static
- fio_pmemblk_file_t pmb_open(const char *pathspec, int flags)
+static fio_pmemblk_file_t pmb_open(const char *pathspec, int flags)
{
fio_pmemblk_file_t pmb;
char *path = NULL;
uint64_t fsize = 0;
pmb_parse_path(pathspec, &path, &bsize, &fsize);
- if (NULL == path)
+ if (!path)
return NULL;
pthread_mutex_lock(&CacheLock);
pmb = fio_pmemblk_cache_lookup(path);
-
- if (NULL == pmb) {
- /* load libpmemblk if needed */
- if (NULL == pmemblk_open)
- if (0 != load_libpmemblk(getenv("FIO_PMEMBLK_LIB")))
- goto error;
-
+ if (!pmb) {
pmb = malloc(sizeof(*pmb));
- if (NULL == pmb)
+ if (!pmb)
goto error;
/* try opening existing first, create it if needed */
pmb->pmb_pool = pmemblk_open(path, bsize);
- if ((NULL == pmb->pmb_pool) &&
- (ENOENT == errno) &&
+ if (!pmb->pmb_pool && (errno == ENOENT) &&
(flags & PMB_CREATE) && (0 < fsize) && (0 < bsize)) {
pmb->pmb_pool =
pmemblk_create(path, bsize, fsize, 0644);
}
- if (NULL == pmb->pmb_pool) {
+ if (!pmb->pmb_pool) {
log_err
("fio: enable to open pmemblk pool file (errno %d)\n",
errno);
return pmb;
error:
- if (NULL != pmb) {
- if (NULL != pmb->pmb_pool)
+ if (pmb) {
+ if (pmb->pmb_pool)
pmemblk_close(pmb->pmb_pool);
pmb->pmb_pool = NULL;
pmb->pmb_filename = NULL;
free(pmb);
}
- if (NULL != path)
+ if (path)
free(path);
pthread_mutex_unlock(&CacheLock);
return NULL;
+}
-} /* pmb_open() */
-
-static void pmb_close(fio_pmemblk_file_t pmb, const int keep)
+static void pmb_close(fio_pmemblk_file_t pmb, const bool keep)
{
pthread_mutex_lock(&CacheLock);
pmb->pmb_refcnt--;
- if (!keep && (0 == pmb->pmb_refcnt)) {
+ if (!keep && !pmb->pmb_refcnt) {
pmemblk_close(pmb->pmb_pool);
pmb->pmb_pool = NULL;
free(pmb->pmb_filename);
}
pthread_mutex_unlock(&CacheLock);
+}
-} /* pmb_close() */
-
-static int pmb_get_flags(struct thread_data *td, uint64_t * pflags)
+static int pmb_get_flags(struct thread_data *td, uint64_t *pflags)
{
static int thread_warned = 0;
static int odirect_warned = 0;
(*pflags) = flags;
return 0;
-
-} /* pmb_get_flags() */
+}
static int fio_pmemblk_open_file(struct thread_data *td, struct fio_file *f)
{
uint64_t flags = 0;
fio_pmemblk_file_t pmb;
- if (0 != pmb_get_flags(td, &flags))
+ if (pmb_get_flags(td, &flags))
return 1;
pmb = pmb_open(f->file_name, flags);
- if (NULL == pmb)
+ if (!pmb)
return 1;
FIOFILEPMBSET(f, pmb);
-
return 0;
+}
-} /* fio_pmemblk_open_file() */
-
-static int
-fio_pmemblk_close_file(struct thread_data fio_unused * td, struct fio_file *f)
+static int fio_pmemblk_close_file(struct thread_data fio_unused *td,
+ struct fio_file *f)
{
fio_pmemblk_file_t pmb = FIOFILEPMBGET(f);
if (pmb)
- pmb_close(pmb, 0);
+ pmb_close(pmb, false);
FIOFILEPMBSET(f, NULL);
-
return 0;
-
-} /* fio_pmemblk_close_file() */
+}
static int fio_pmemblk_get_file_size(struct thread_data *td, struct fio_file *f)
{
if (fio_file_size_known(f))
return 0;
- if (NULL == pmb) {
- if (0 != pmb_get_flags(td, &flags))
+ if (!pmb) {
+ if (pmb_get_flags(td, &flags))
return 1;
pmb = pmb_open(f->file_name, flags);
- if (NULL == pmb)
+ if (!pmb)
return 1;
}
fio_file_set_size_known(f);
- if (NULL == FIOFILEPMBGET(f))
- pmb_close(pmb, 1);
+ if (!FIOFILEPMBGET(f))
+ pmb_close(pmb, true);
return 0;
-
-} /* fio_pmemblk_get_file_size() */
+}
static int fio_pmemblk_queue(struct thread_data *td, struct io_u *io_u)
{
unsigned long long off;
unsigned long len;
void *buf;
- int (*blkop) (PMEMblkpool *, void *, off_t) = (void *)pmemblk_write;
fio_ro_check(td, io_u);
switch (io_u->ddir) {
case DDIR_READ:
- blkop = pmemblk_read;
- /* fall through */
case DDIR_WRITE:
off = io_u->offset;
len = io_u->xfer_buflen;
io_u->error = EINVAL;
- if (0 != (off % pmb->pmb_bsize))
+ if (off % pmb->pmb_bsize)
break;
- if (0 != (len % pmb->pmb_bsize))
+ if (len % pmb->pmb_bsize)
break;
if ((off + len) / pmb->pmb_bsize > pmb->pmb_nblocks)
break;
off /= pmb->pmb_bsize;
len /= pmb->pmb_bsize;
while (0 < len) {
- if (0 != blkop(pmb->pmb_pool, buf, off)) {
+ if (io_u->ddir == DDIR_READ &&
+ 0 != pmemblk_read(pmb->pmb_pool, buf, off)) {
+ io_u->error = errno;
+ break;
+ } else if (0 != pmemblk_write(pmb->pmb_pool, buf, off)) {
io_u->error = errno;
break;
}
}
return FIO_Q_COMPLETED;
-
-} /* fio_pmemblk_queue() */
+}
static int fio_pmemblk_unlink_file(struct thread_data *td, struct fio_file *f)
{
*/
pmb_parse_path(f->file_name, &path, &bsize, &fsize);
- if (NULL == path)
- return 1;
+ if (!path)
+ return ENOENT;
unlink(path);
free(path);
-
return 0;
+}
-} /* fio_pmemblk_unlink_file() */
-
-struct ioengine_ops ioengine = {
+static struct ioengine_ops ioengine = {
.name = "pmemblk",
.version = FIO_IOOPS_VERSION,
.queue = fio_pmemblk_queue,
.flags = FIO_SYNCIO | FIO_DISKLESSIO | FIO_NOEXTEND | FIO_NODISKUTIL,
};
-static void
-fio_init fio_pmemblk_register(void)
+static void fio_init fio_pmemblk_register(void)
{
register_ioengine(&ioengine);
}
-static void
-fio_exit fio_pmemblk_unregister(void)
+static void fio_exit fio_pmemblk_unregister(void)
{
unregister_ioengine(&ioengine);
}