1 /*- 2 * BSD LICENSE 3 * 4 * Copyright (c) Intel Corporation. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 11 * * Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * * Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in 15 * the documentation and/or other materials provided with the 16 * distribution. 17 * * Neither the name of Intel Corporation nor the names of its 18 * contributors may be used to endorse or promote products derived 19 * from this software without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 22 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 24 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 25 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 26 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 27 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 28 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 29 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 30 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 31 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 32 */ 33 34 #include "spdk/fd.h" 35 36 #include <sys/ioctl.h> 37 #include <sys/types.h> 38 #include <sys/stat.h> 39 #include <unistd.h> 40 #include <stdint.h> 41 42 #ifdef __linux__ 43 #include <linux/fs.h> 44 #endif 45 46 static uint64_t 47 dev_get_size(int fd) 48 { 49 #if defined(DIOCGMEDIASIZE) /* FreeBSD */ 50 off_t size; 51 52 if (ioctl(fd, DIOCGMEDIASIZE, &size) == 0) { 53 return size; 54 } 55 #elif defined(__linux__) && defined(BLKGETSIZE64) 56 uint64_t size; 57 58 if (ioctl(fd, BLKGETSIZE64, &size) == 0) { 59 return size; 60 } 61 #endif 62 63 return 0; 64 } 65 66 uint32_t 67 spdk_fd_get_blocklen(int fd) 68 { 69 #if defined(DKIOCGETBLOCKSIZE) /* FreeBSD */ 70 uint32_t blocklen; 71 72 if (ioctl(fd, DKIOCGETBLOCKSIZE, &blocklen) == 0) { 73 return blocklen; 74 } 75 #elif defined(__linux__) && defined(BLKSSZGET) 76 uint32_t blocklen; 77 78 if (ioctl(fd, BLKSSZGET, &blocklen) == 0) { 79 return blocklen; 80 } 81 #endif 82 83 /* Assume 512 if none of the other methods worked */ 84 return 512; 85 } 86 87 uint64_t 88 spdk_fd_get_size(int fd) 89 { 90 struct stat st; 91 92 if (fstat(fd, &st) != 0) { 93 return 0; 94 } 95 96 if (S_ISLNK(st.st_mode)) { 97 return 0; 98 } 99 100 if (S_ISBLK(st.st_mode) || S_ISCHR(st.st_mode)) { 101 return dev_get_size(fd); 102 } else if (S_ISREG(st.st_mode)) { 103 return st.st_size; 104 } 105 106 /* Not REG, CHR or BLK */ 107 return 0; 108 } 109