|
|
@@ -0,0 +1,95 @@
|
|
|
+
|
|
|
+ * Copyright © 2010
|
|
|
+ * Waldemar Brodkorb <wbx@openadk.org>
|
|
|
+ * Thorsten Glaser <tg@mirbsd.org>
|
|
|
+ *
|
|
|
+ * Provided that these terms and disclaimer and all copyright notices
|
|
|
+ * are retained or reproduced in an accompanying document, permission
|
|
|
+ * is granted to deal in this work without restriction, including un‐
|
|
|
+ * limited rights to use, publicly perform, distribute, sell, modify,
|
|
|
+ * merge, give away, or sublicence.
|
|
|
+ *
|
|
|
+ * This work is provided “AS IS” and WITHOUT WARRANTY of any kind, to
|
|
|
+ * the utmost extent permitted by applicable law, neither express nor
|
|
|
+ * implied; without malicious intent or gross negligence. In no event
|
|
|
+ * may a licensor, author or contributor be held liable for indirect,
|
|
|
+ * direct, other damage, loss, or other issues arising in any way out
|
|
|
+ * of dealing in the work, even if advised of the possibility of such
|
|
|
+ * damage or existence of a defect, except proven that it results out
|
|
|
+ * of said person’s immediate fault when using the work as intended.
|
|
|
+ *
|
|
|
+ * Alternatively, this work may be distributed under the terms of the
|
|
|
+ * General Public License, any version, as published by the Free Soft-
|
|
|
+ * ware Foundation.
|
|
|
+ *-
|
|
|
+ * Display the size of a block device (e.g. USB stick, CF/SF/MMC card
|
|
|
+ * or hard disc) in 512-byte sectors.
|
|
|
+ */
|
|
|
+
|
|
|
+#define _FILE_OFFSET_BITS 64
|
|
|
+
|
|
|
+#include <sys/param.h>
|
|
|
+#include <sys/types.h>
|
|
|
+#include <sys/ioctl.h>
|
|
|
+#include <sys/mount.h>
|
|
|
+
|
|
|
+#if defined(__APPLE__)
|
|
|
+#include <sys/disk.h>
|
|
|
+#endif
|
|
|
+
|
|
|
+#if defined(DIOCGDINFO)
|
|
|
+#include <sys/disklabel.h>
|
|
|
+#endif
|
|
|
+
|
|
|
+#include <err.h>
|
|
|
+#include <fcntl.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <unistd.h>
|
|
|
+
|
|
|
+unsigned long long numsecs(int);
|
|
|
+
|
|
|
+int
|
|
|
+main(int argc, char *argv[]) {
|
|
|
+ int fd;
|
|
|
+
|
|
|
+ if (argc != 2)
|
|
|
+ errx(255, "Syntax: dkgetsz /dev/sda");
|
|
|
+
|
|
|
+ if ((fd = open(argv[1], O_RDONLY)) == -1)
|
|
|
+ err(1, "open");
|
|
|
+ printf("%llu\n", numsecs(fd));
|
|
|
+ close(fd);
|
|
|
+ return (0);
|
|
|
+}
|
|
|
+
|
|
|
+unsigned long long
|
|
|
+numsecs(int fd)
|
|
|
+{
|
|
|
+#if defined(BLKGETSIZE) || defined(DKIOCGETBLOCKCOUNT)
|
|
|
+
|
|
|
+ * note: BLKGETSIZE64 returns bytes, not sectors, but the return
|
|
|
+ * type is size_t which is 32 bits on an ILP32 platform, so it
|
|
|
+ * fails interestingly here… thus we use BLKGETSIZE instead.
|
|
|
+ */
|
|
|
+#if defined(DKIOCGETBLOCKCOUNT)
|
|
|
+ uint64_t nsecs;
|
|
|
+#define THEIOCTL DKIOCGETBLOCKCOUNT
|
|
|
+#define STRIOCTL "DKIOCGETBLOCKCOUNT"
|
|
|
+#else
|
|
|
+ unsigned long nsecs;
|
|
|
+#define THEIOCTL BLKGETSIZE
|
|
|
+#define STRIOCTL "BLKGETSIZE"
|
|
|
+#endif
|
|
|
+ if (ioctl(fd, THEIOCTL, &nsecs) == -1)
|
|
|
+ err(1, "ioctl %s", STRIOCTL);
|
|
|
+ return ((unsigned long long)nsecs);
|
|
|
+#elif defined(DIOCGDINFO)
|
|
|
+ struct disklabel dl;
|
|
|
+
|
|
|
+ if (ioctl(fd, DIOCGDINFO, &dl) == -1)
|
|
|
+ err(1, "ioctl DIOCGDINFO");
|
|
|
+ return ((unsigned long long)dl.d_secperunit);
|
|
|
+#else
|
|
|
+#error PLEASE DO IMPLEMENT numsecs FOR THIS PLATFORM.
|
|
|
+#endif
|
|
|
+}
|
Thorsten Glaser