From aaf0555e032edece4705110cd3f4d4692edaea21 Mon Sep 17 00:00:00 2001 From: Daniel Verkamp Date: Mon, 19 Oct 2015 15:47:56 -0700 Subject: [PATCH] util: add file size and block size functions Change-Id: I0d3269805470768ce75e81955db601c48ae0e737 Signed-off-by: Daniel Verkamp --- include/spdk/file.h | 42 ++++++++++++++++++ lib/util/Makefile | 2 +- lib/util/file.c | 106 ++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 149 insertions(+), 1 deletion(-) create mode 100644 include/spdk/file.h create mode 100644 lib/util/file.c diff --git a/include/spdk/file.h b/include/spdk/file.h new file mode 100644 index 000000000..08e0b752d --- /dev/null +++ b/include/spdk/file.h @@ -0,0 +1,42 @@ +/*- + * BSD LICENSE + * + * Copyright(c) 2015 Intel Corporation. All rights reserved. + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in + * the documentation and/or other materials provided with the + * distribution. + * * Neither the name of Intel Corporation nor the names of its + * contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#ifndef SPDK_FILE_H +#define SPDK_FILE_H + +#include + +uint64_t file_get_size(int fd); +uint32_t dev_get_blocklen(int fd); + +#endif diff --git a/lib/util/Makefile b/lib/util/Makefile index 00376fa00..86d76e4ba 100644 --- a/lib/util/Makefile +++ b/lib/util/Makefile @@ -35,7 +35,7 @@ include $(SPDK_ROOT_DIR)/mk/spdk.common.mk CFLAGS += $(DPDK_INC) -C_SRCS = string.c pci.c +C_SRCS = file.c string.c pci.c C_OBJS := $(patsubst %.c,%.o,$(C_SRCS)) diff --git a/lib/util/file.c b/lib/util/file.c new file mode 100644 index 000000000..b6b0b68e8 --- /dev/null +++ b/lib/util/file.c @@ -0,0 +1,106 @@ +/*- + * BSD LICENSE + * + * Copyright(c) 2015 Intel Corporation. All rights reserved. + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in + * the documentation and/or other materials provided with the + * distribution. + * * Neither the name of Intel Corporation nor the names of its + * contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include +#include +#include +#include +#include + +#ifdef __linux__ +#include +#endif + +static uint64_t +dev_get_size(int fd) +{ +#if defined(DIOCGMEDIASIZE) /* FreeBSD */ + off_t size; + + if (ioctl(fd, DIOCGMEDIASIZE, &size) == 0) { + return size; + } +#elif defined(__linux__) && defined(BLKGETSIZE64) + uint64_t size; + + if (ioctl(fd, BLKGETSIZE64, &size) == 0) { + return size; + } +#endif + + return 0; +} + +uint32_t +dev_get_blocklen(int fd) +{ +#if defined(DKIOCGETBLOCKSIZE) /* FreeBSD */ + uint32_t blocklen; + + if (ioctl(fd, DKIOCGETBLOCKSIZE, &blocklen) != 0) { + return blocklen; + } +#elif defined(__linux__) && defined(BLKSSZGET) + uint32_t blocklen; + + if (ioctl(fd, BLKSSZGET, &blocklen) != 0) { + return blocklen; + } +#endif + + /* Assume 512 if none of the other methods worked */ + return 512; +} + +uint64_t +file_get_size(int fd) +{ + struct stat st; + + if (fstat(fd, &st) != 0) { + return 0; + } + + if (S_ISLNK(st.st_mode)) { + return 0; + } + + if (S_ISBLK(st.st_mode) || S_ISCHR(st.st_mode)) { + return dev_get_size(fd); + } else if (S_ISREG(st.st_mode)) { + return st.st_size; + } + + /* Not REG, CHR or BLK */ + return 0; +}