2 9ddaf4ab 2023-04-08 benni * Copyright (c) 2023 Benjamin Stürz <benni@stuerz.xyz>
4 9ddaf4ab 2023-04-08 benni * Permission to use, copy, modify, and distribute this software for any
5 9ddaf4ab 2023-04-08 benni * purpose with or without fee is hereby granted, provided that the above
6 9ddaf4ab 2023-04-08 benni * copyright notice and this permission notice appear in all copies.
8 9ddaf4ab 2023-04-08 benni * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 9ddaf4ab 2023-04-08 benni * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 9ddaf4ab 2023-04-08 benni * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 9ddaf4ab 2023-04-08 benni * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 9ddaf4ab 2023-04-08 benni * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 9ddaf4ab 2023-04-08 benni * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 9ddaf4ab 2023-04-08 benni * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 33bf1ca4 2023-04-08 benni #define _XOPEN_SOURCE 700
17 33bf1ca4 2023-04-08 benni #define _BSD_SOURCE 1
18 33bf1ca4 2023-04-08 benni #define DKTYPENAMES
19 33bf1ca4 2023-04-08 benni #include <stddef.h>
20 9c5d8ecc 2023-05-02 benni #include <sys/cdefs.h>
21 33bf1ca4 2023-04-08 benni #include <sys/types.h>
22 33bf1ca4 2023-04-08 benni #include <sys/disklabel.h>
23 33bf1ca4 2023-04-08 benni #include <sys/sysctl.h>
24 33bf1ca4 2023-04-08 benni #include <sys/ioctl.h>
25 33bf1ca4 2023-04-08 benni #include <sys/mount.h>
26 33bf1ca4 2023-04-08 benni #include <sys/dkio.h>
27 33bf1ca4 2023-04-08 benni #include <inttypes.h>
28 8380cc51 2023-05-01 benni #include <stdbool.h>
29 33bf1ca4 2023-04-08 benni #include <string.h>
30 33bf1ca4 2023-04-08 benni #include <stdarg.h>
31 33bf1ca4 2023-04-08 benni #include <stdint.h>
32 33bf1ca4 2023-04-08 benni #include <stdlib.h>
33 33bf1ca4 2023-04-08 benni #include <unistd.h>
34 5691dc7b 2023-05-05 benni #include <libgen.h>
35 33bf1ca4 2023-04-08 benni #include <stdio.h>
36 33bf1ca4 2023-04-08 benni #include <fcntl.h>
37 33bf1ca4 2023-04-08 benni #include <errno.h>
39 9c5d8ecc 2023-05-02 benni static __dead void die (const char *fmt, ...)
41 33bf1ca4 2023-04-08 benni va_list ap;
43 33bf1ca4 2023-04-08 benni va_start (ap, fmt);
44 33bf1ca4 2023-04-08 benni fputs ("Error: ", stderr);
45 33bf1ca4 2023-04-08 benni vfprintf (stderr, fmt, ap);
47 33bf1ca4 2023-04-08 benni if (errno != 0) {
48 33bf1ca4 2023-04-08 benni fprintf (stderr, ": %s\n", strerror (errno));
50 33bf1ca4 2023-04-08 benni fputc ('\n', stderr);
56 33bf1ca4 2023-04-08 benni static int diskcount (void)
58 33bf1ca4 2023-04-08 benni const int mib[2] = { CTL_HW, HW_DISKCOUNT };
59 33bf1ca4 2023-04-08 benni int diskcount;
60 33bf1ca4 2023-04-08 benni size_t len = sizeof diskcount;
62 33bf1ca4 2023-04-08 benni if (sysctl (mib, 2, &diskcount, &len, NULL, 0) == -1)
63 33bf1ca4 2023-04-08 benni die ("sysctl(hw.diskcount)");
65 33bf1ca4 2023-04-08 benni return diskcount;
68 33bf1ca4 2023-04-08 benni static char *disknames (void)
70 33bf1ca4 2023-04-08 benni const int num = diskcount ();
71 33bf1ca4 2023-04-08 benni const int mib[2] = { CTL_HW, HW_DISKNAMES };
72 33bf1ca4 2023-04-08 benni size_t len = 32 * num;
73 33bf1ca4 2023-04-08 benni char *buffer = malloc (len);
75 33bf1ca4 2023-04-08 benni if (sysctl (mib, 2, buffer, &len, NULL, 0) == -1)
76 33bf1ca4 2023-04-08 benni die ("sysctl(hw.disknames)");
78 33bf1ca4 2023-04-08 benni return buffer;
81 9c5d8ecc 2023-05-02 benni static char *stripdisk (char *n)
83 9c5d8ecc 2023-05-02 benni const char sufx[] = " disk";
84 9c5d8ecc 2023-05-02 benni const size_t ln = strnlen (n, 16);
85 9c5d8ecc 2023-05-02 benni const size_t ls = sizeof sufx - 1;
87 9c5d8ecc 2023-05-02 benni if (!memcmp (n + ln - ls, sufx, ls)) {
88 9c5d8ecc 2023-05-02 benni n[ln - ls] = '\0';
94 9c5d8ecc 2023-05-02 benni static void print_size (uint64_t sz)
96 33bf1ca4 2023-04-08 benni struct unit {
98 33bf1ca4 2023-04-08 benni uint64_t factor;
101 33bf1ca4 2023-04-08 benni const struct unit units[] = {
102 33bf1ca4 2023-04-08 benni { 'P', 1ull << 50 },
103 33bf1ca4 2023-04-08 benni { 'T', 1ull << 40 },
104 33bf1ca4 2023-04-08 benni { 'G', 1ull << 30 },
105 33bf1ca4 2023-04-08 benni { 'M', 1ull << 20 },
106 33bf1ca4 2023-04-08 benni { 'K', 1ull << 10 },
107 33bf1ca4 2023-04-08 benni { '0', 0 },
110 33bf1ca4 2023-04-08 benni char sym = 'B';
111 33bf1ca4 2023-04-08 benni uint64_t factor = 1;
113 33bf1ca4 2023-04-08 benni for (const struct unit *u = &units[0]; u->factor; ++u) {
114 33bf1ca4 2023-04-08 benni if (sz >= (u->factor * 9 / 10)) {
115 33bf1ca4 2023-04-08 benni sym = u->sym;
116 33bf1ca4 2023-04-08 benni factor = u->factor;
121 33bf1ca4 2023-04-08 benni const unsigned scaled10 = sz * 10 / factor;
122 33bf1ca4 2023-04-08 benni const unsigned scaled = sz / factor;
123 33bf1ca4 2023-04-08 benni if (scaled10 >= 1000) {
124 8a374bef 2023-04-08 benni printf ("%u", scaled);
125 33bf1ca4 2023-04-08 benni } else if (scaled10 >= 100) {
126 8a374bef 2023-04-08 benni printf (" %u", scaled);
128 8a374bef 2023-04-08 benni printf ("%u.%u", scaled, scaled10 % 10);
131 33bf1ca4 2023-04-08 benni putchar (sym);
132 9c5d8ecc 2023-05-02 benni putchar (' ');
136 0c961f7d 2023-05-04 benni FIELD_NAME = 0x01,
137 0c961f7d 2023-05-04 benni FIELD_SIZE = 0x02,
138 0c961f7d 2023-05-04 benni FIELD_USED = 0x04,
139 0c961f7d 2023-05-04 benni FIELD_FREE = 0x08,
140 0c961f7d 2023-05-04 benni FIELD_TYPE = 0x10,
141 0c961f7d 2023-05-04 benni FIELD_LMNT = 0x20,
143 2b650bba 2023-05-04 benni FIELD_DEFAULT = FIELD_NAME | FIELD_SIZE | FIELD_TYPE | FIELD_LMNT,
147 0c961f7d 2023-05-04 benni OPT_NOHEADER = 0x01,
148 0c961f7d 2023-05-04 benni OPT_NOUNICODE = 0x02,
151 9c5d8ecc 2023-05-02 benni struct my_partinfo {
152 9c5d8ecc 2023-05-02 benni char name[5];
153 9c5d8ecc 2023-05-02 benni uint64_t size;
154 9c5d8ecc 2023-05-02 benni uint64_t fssize;
155 9c5d8ecc 2023-05-02 benni uint64_t free;
156 9c5d8ecc 2023-05-02 benni const char *fstype;
157 9c5d8ecc 2023-05-02 benni const char *mount;
160 9c5d8ecc 2023-05-02 benni struct my_diskinfo {
161 9c5d8ecc 2023-05-02 benni char type[16];
162 9c5d8ecc 2023-05-02 benni char label[16];
163 9c5d8ecc 2023-05-02 benni char name[4];
164 9c5d8ecc 2023-05-02 benni uint64_t size;
165 9c5d8ecc 2023-05-02 benni uint64_t used;
166 9c5d8ecc 2023-05-02 benni uint8_t num_parts;
167 9c5d8ecc 2023-05-02 benni struct my_partinfo parts[MAXPARTITIONS];
170 9c5d8ecc 2023-05-02 benni static void print_header (int fields)
172 9c5d8ecc 2023-05-02 benni if (fields & FIELD_NAME)
173 9c5d8ecc 2023-05-02 benni printf ("%-6s ", "NAME");
175 9c5d8ecc 2023-05-02 benni if (fields & FIELD_SIZE)
176 9c5d8ecc 2023-05-02 benni printf ("%-4s ", "SIZE");
178 9c5d8ecc 2023-05-02 benni if (fields & FIELD_USED)
179 9c5d8ecc 2023-05-02 benni printf ("%-4s ", "USED");
181 9c5d8ecc 2023-05-02 benni if (fields & FIELD_FREE)
182 9c5d8ecc 2023-05-02 benni printf ("%-4s ", "FREE");
184 9c5d8ecc 2023-05-02 benni if (fields & FIELD_TYPE)
185 9c5d8ecc 2023-05-02 benni printf ("%-8s ", "TYPE");
187 2b650bba 2023-05-04 benni if (fields & FIELD_LMNT)
188 2b650bba 2023-05-04 benni printf ("LABEL/MOUNT ");
190 9c5d8ecc 2023-05-02 benni putchar ('\n');
193 0c961f7d 2023-05-04 benni static void print_part (const struct my_partinfo *part, int fields, int options, bool last)
195 0c961f7d 2023-05-04 benni if (fields & FIELD_NAME) {
196 0c961f7d 2023-05-04 benni const char *prefix = (options & OPT_NOUNICODE) ? " " : (last ? "└─" : "├─");
197 0c961f7d 2023-05-04 benni printf ("%s%s ", prefix, part->name);
200 9c5d8ecc 2023-05-02 benni if (fields & FIELD_SIZE)
201 9c5d8ecc 2023-05-02 benni print_size (part->size);
203 9c5d8ecc 2023-05-02 benni if (fields & FIELD_USED) {
204 9c5d8ecc 2023-05-02 benni if (part->fssize) {
205 9c5d8ecc 2023-05-02 benni print_size (part->fssize - part->free);
207 9c5d8ecc 2023-05-02 benni printf (" N/A ");
211 9c5d8ecc 2023-05-02 benni if (fields & FIELD_FREE) {
212 9c5d8ecc 2023-05-02 benni if (part->fssize) {
213 9c5d8ecc 2023-05-02 benni print_size (part->free);
215 9c5d8ecc 2023-05-02 benni printf (" N/A ");
219 9c5d8ecc 2023-05-02 benni if (fields & FIELD_TYPE)
220 9c5d8ecc 2023-05-02 benni printf ("%-8.16s ", part->fstype);
222 2b650bba 2023-05-04 benni if (fields & FIELD_LMNT && part->mount)
223 9c5d8ecc 2023-05-02 benni printf ("%s ", part->mount);
225 9c5d8ecc 2023-05-02 benni putchar ('\n');
228 0c961f7d 2023-05-04 benni static void print_disk (const struct my_diskinfo *disk, int fields, int options)
230 9c5d8ecc 2023-05-02 benni if (fields & FIELD_NAME)
231 9c5d8ecc 2023-05-02 benni printf ("%s ", disk->name);
233 9c5d8ecc 2023-05-02 benni if (fields & FIELD_SIZE)
234 9c5d8ecc 2023-05-02 benni print_size (disk->size);
236 9c5d8ecc 2023-05-02 benni if (fields & FIELD_USED)
237 9c5d8ecc 2023-05-02 benni print_size (disk->used);
239 9c5d8ecc 2023-05-02 benni if (fields & FIELD_FREE)
240 9c5d8ecc 2023-05-02 benni print_size (disk->size - disk->used);
242 9c5d8ecc 2023-05-02 benni // Pad only upto 8 characters because most disk types are <=8 bytes long.
243 9c5d8ecc 2023-05-02 benni if (fields & FIELD_TYPE)
244 9c5d8ecc 2023-05-02 benni printf ("%-8.16s ", disk->type);
246 2b650bba 2023-05-04 benni if (fields & FIELD_LMNT)
247 2b650bba 2023-05-04 benni printf ("%.16s ", disk->label);
249 9c5d8ecc 2023-05-02 benni putchar ('\n');
251 9c5d8ecc 2023-05-02 benni for (uint8_t i = 0; i < disk->num_parts; ++i)
252 0c961f7d 2023-05-04 benni print_part (&disk->parts[i], fields, options, i == (disk->num_parts - 1));
255 9c5d8ecc 2023-05-02 benni static const struct statfs *find_mount (const char *dev)
257 9c5d8ecc 2023-05-02 benni static struct statfs *mounts = NULL;
258 9c5d8ecc 2023-05-02 benni static int n_mounts;
260 9c5d8ecc 2023-05-02 benni if (!mounts) {
261 9c5d8ecc 2023-05-02 benni n_mounts = getmntinfo (&mounts, MNT_NOWAIT);
262 9c5d8ecc 2023-05-02 benni if (n_mounts == 0)
263 9c5d8ecc 2023-05-02 benni die ("getmntinfo()");
266 9c5d8ecc 2023-05-02 benni for (int i = 0; i < n_mounts; ++i) {
267 9c5d8ecc 2023-05-02 benni if (!strcmp (dev, mounts[i].f_mntfromname))
268 9c5d8ecc 2023-05-02 benni return &mounts[i];
271 9c5d8ecc 2023-05-02 benni return NULL;
274 9c5d8ecc 2023-05-02 benni static struct my_diskinfo read_disk (const char *name)
276 9c5d8ecc 2023-05-02 benni struct my_diskinfo disk;
277 9c5d8ecc 2023-05-02 benni struct disklabel label;
279 9c5d8ecc 2023-05-02 benni bzero (&disk, sizeof disk);
281 9c5d8ecc 2023-05-02 benni { // Read disklabel.
282 5691dc7b 2023-05-05 benni char *path = NULL;
285 5691dc7b 2023-05-05 benni asprintf (&path, "/dev/%sc", name);
286 9c5d8ecc 2023-05-02 benni fd = open (path, O_RDONLY);
287 9c5d8ecc 2023-05-02 benni if (fd < 0)
288 9c5d8ecc 2023-05-02 benni die ("open(%s)", path);
289 9c5d8ecc 2023-05-02 benni if (ioctl (fd, DIOCGDINFO, &label) < 0)
290 9c5d8ecc 2023-05-02 benni die ("ioctl(%s, DIOCGDINFO)", path);
291 9c5d8ecc 2023-05-02 benni close (fd);
292 5691dc7b 2023-05-05 benni free (path);
296 9c5d8ecc 2023-05-02 benni memcpy (disk.name, name, 3);
297 9c5d8ecc 2023-05-02 benni disk.name[3] = '\0';
298 9c5d8ecc 2023-05-02 benni disk.size = DL_GETDSIZE (&label) * label.d_secsize;
299 9c5d8ecc 2023-05-02 benni memcpy (disk.type, label.d_typename, 16);
300 9c5d8ecc 2023-05-02 benni stripdisk (disk.type);
301 9c5d8ecc 2023-05-02 benni memcpy (disk.label, label.d_packname, 16);
302 9c5d8ecc 2023-05-02 benni disk.num_parts = 0;
304 9c5d8ecc 2023-05-02 benni for (uint16_t i = 0; i < label.d_npartitions; ++i) {
305 9c5d8ecc 2023-05-02 benni const struct partition *p = &label.d_partitions[i];
306 9c5d8ecc 2023-05-02 benni const struct statfs *mnt;
307 9c5d8ecc 2023-05-02 benni struct my_partinfo part;
308 9c5d8ecc 2023-05-02 benni char path[5 + 4 + 1];
310 9c5d8ecc 2023-05-02 benni bzero (&part, sizeof part);
312 9c5d8ecc 2023-05-02 benni part.size = DL_GETPSIZE (p) * label.d_secsize;
314 9c5d8ecc 2023-05-02 benni if (!part.size)
317 9c5d8ecc 2023-05-02 benni memcpy (part.name, disk.name, 3);
318 9c5d8ecc 2023-05-02 benni part.name[3] = 'a' + i;
319 9c5d8ecc 2023-05-02 benni part.name[4] = '\0';
321 9c5d8ecc 2023-05-02 benni snprintf (path, sizeof path, "/dev/%s", part.name);
322 9c5d8ecc 2023-05-02 benni mnt = find_mount (path);
324 9c5d8ecc 2023-05-02 benni if (mnt) {
325 9c5d8ecc 2023-05-02 benni const uint64_t bs = mnt->f_bsize;
326 9c5d8ecc 2023-05-02 benni part.mount = mnt->f_mntonname;
327 9c5d8ecc 2023-05-02 benni part.fssize = mnt->f_blocks * bs;
328 9c5d8ecc 2023-05-02 benni part.free = mnt->f_bfree * bs;
331 9c5d8ecc 2023-05-02 benni part.fstype = fstypenames[p->p_fstype];
332 9c5d8ecc 2023-05-02 benni if (i != 2)
333 9c5d8ecc 2023-05-02 benni disk.used += part.size;
334 9c5d8ecc 2023-05-02 benni disk.parts[disk.num_parts++] = part;
337 9c5d8ecc 2023-05-02 benni return disk;
340 9c5d8ecc 2023-05-02 benni static int usage (void)
342 5691dc7b 2023-05-05 benni fputs ("Usage: lsblk [-Vainu] [disk...]\n", stderr);
346 33bf1ca4 2023-04-08 benni int main (int argc, char *argv[])
348 8380cc51 2023-05-01 benni int option;
349 9c5d8ecc 2023-05-02 benni int fields = FIELD_DEFAULT;
350 0c961f7d 2023-05-04 benni int options = 0;
352 aff15c23 2023-04-30 benni if (unveil ("/dev", "r") == -1)
353 aff15c23 2023-04-30 benni die ("unveil(/dev)");
355 aff15c23 2023-04-30 benni if (unveil (NULL, NULL) == -1)
356 aff15c23 2023-04-30 benni die ("unveil()");
358 0c961f7d 2023-05-04 benni while ((option = getopt (argc, argv, ":Vainu")) != -1) {
359 33bf1ca4 2023-04-08 benni switch (option) {
361 33bf1ca4 2023-04-08 benni puts ("lsblk-" VERSION);
364 bf0e6461 2023-05-02 benni fields = -1;
367 0c961f7d 2023-05-04 benni options |= OPT_NOUNICODE;
370 0c961f7d 2023-05-04 benni options |= OPT_NOHEADER;
373 9c5d8ecc 2023-05-02 benni fields |= FIELD_USED | FIELD_FREE;
376 33bf1ca4 2023-04-08 benni return usage ();
380 5691dc7b 2023-05-05 benni argv += optind;
381 5691dc7b 2023-05-05 benni argc -= optind;
383 5691dc7b 2023-05-05 benni char *names = argc == 0 ? disknames () : NULL;
385 33bf1ca4 2023-04-08 benni if (pledge ("stdio rpath disklabel", NULL) == -1)
386 33bf1ca4 2023-04-08 benni die ("pledge()");
388 5691dc7b 2023-05-05 benni size_t cap_disks;
389 5691dc7b 2023-05-05 benni if (argc == 0) {
390 5691dc7b 2023-05-05 benni const char *s = names;
391 5691dc7b 2023-05-05 benni cap_disks = 0;
392 5691dc7b 2023-05-05 benni while ((s = strchr (s, ',')) != NULL)
393 5691dc7b 2023-05-05 benni ++cap_disks, ++s;
395 5691dc7b 2023-05-05 benni cap_disks = argc;
398 9c5d8ecc 2023-05-02 benni struct my_diskinfo *disks = calloc (cap_disks, sizeof (struct my_diskinfo));
399 5691dc7b 2023-05-05 benni size_t num_disks = 0;
401 5691dc7b 2023-05-05 benni if (argc == 0) {
402 5691dc7b 2023-05-05 benni for (char *disk; (disk = strsep (&names, ",")) != NULL; ) {
403 5691dc7b 2023-05-05 benni char *colon = strchr (disk, ':');
404 5691dc7b 2023-05-05 benni if (colon)
405 5691dc7b 2023-05-05 benni *colon = '\0';
406 5691dc7b 2023-05-05 benni disks[num_disks++] = read_disk (disk);
407 5691dc7b 2023-05-05 benni if (colon)
408 5691dc7b 2023-05-05 benni *colon = ':';
411 5691dc7b 2023-05-05 benni for (int i = 0; i < argc; ++i) {
412 5691dc7b 2023-05-05 benni char *disk = basename (argv[i]);
413 5691dc7b 2023-05-05 benni disks[num_disks++] = read_disk (disk);
417 33bf1ca4 2023-04-08 benni free (names);
419 0c961f7d 2023-05-04 benni if (!(options & OPT_NOHEADER))
420 9c5d8ecc 2023-05-02 benni print_header (fields);
422 9c5d8ecc 2023-05-02 benni for (size_t i = 0; i < num_disks; ++i) {
423 0c961f7d 2023-05-04 benni print_disk (&disks[i], fields, options);
