1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
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
}
|
