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 33bf1ca4 2023-04-08 benni #include <sys/types.h>
21 33bf1ca4 2023-04-08 benni #include <sys/disklabel.h>
22 33bf1ca4 2023-04-08 benni #include <sys/sysctl.h>
23 33bf1ca4 2023-04-08 benni #include <sys/ioctl.h>
24 33bf1ca4 2023-04-08 benni #include <sys/mount.h>
25 33bf1ca4 2023-04-08 benni #include <sys/dkio.h>
26 33bf1ca4 2023-04-08 benni #include <inttypes.h>
27 33bf1ca4 2023-04-08 benni #include <string.h>
28 33bf1ca4 2023-04-08 benni #include <stdarg.h>
29 33bf1ca4 2023-04-08 benni #include <stdint.h>
30 33bf1ca4 2023-04-08 benni #include <stdlib.h>
31 33bf1ca4 2023-04-08 benni #include <unistd.h>
32 33bf1ca4 2023-04-08 benni #include <stdio.h>
33 33bf1ca4 2023-04-08 benni #include <fcntl.h>
34 33bf1ca4 2023-04-08 benni #include <errno.h>
36 33bf1ca4 2023-04-08 benni static void die (const char *fmt, ...)
38 33bf1ca4 2023-04-08 benni va_list ap;
40 33bf1ca4 2023-04-08 benni va_start (ap, fmt);
41 33bf1ca4 2023-04-08 benni fputs ("Error: ", stderr);
42 33bf1ca4 2023-04-08 benni vfprintf (stderr, fmt, ap);
44 33bf1ca4 2023-04-08 benni if (errno != 0) {
45 33bf1ca4 2023-04-08 benni fprintf (stderr, ": %s\n", strerror (errno));
47 33bf1ca4 2023-04-08 benni fputc ('\n', stderr);
53 33bf1ca4 2023-04-08 benni static int diskcount (void)
55 33bf1ca4 2023-04-08 benni const int mib[2] = { CTL_HW, HW_DISKCOUNT };
56 33bf1ca4 2023-04-08 benni int diskcount;
57 33bf1ca4 2023-04-08 benni size_t len = sizeof diskcount;
59 33bf1ca4 2023-04-08 benni if (sysctl (mib, 2, &diskcount, &len, NULL, 0) == -1)
60 33bf1ca4 2023-04-08 benni die ("sysctl(hw.diskcount)");
62 33bf1ca4 2023-04-08 benni return diskcount;
65 33bf1ca4 2023-04-08 benni static char *disknames (void)
67 33bf1ca4 2023-04-08 benni const int num = diskcount ();
68 33bf1ca4 2023-04-08 benni const int mib[2] = { CTL_HW, HW_DISKNAMES };
69 33bf1ca4 2023-04-08 benni size_t len = 32 * num;
70 33bf1ca4 2023-04-08 benni char *buffer = malloc (len);
72 33bf1ca4 2023-04-08 benni if (sysctl (mib, 2, buffer, &len, NULL, 0) == -1)
73 33bf1ca4 2023-04-08 benni die ("sysctl(hw.disknames)");
75 33bf1ca4 2023-04-08 benni return buffer;
78 33bf1ca4 2023-04-08 benni static void printsize (uint64_t sz)
80 33bf1ca4 2023-04-08 benni struct unit {
82 33bf1ca4 2023-04-08 benni uint64_t factor;
85 33bf1ca4 2023-04-08 benni const struct unit units[] = {
86 33bf1ca4 2023-04-08 benni { 'P', 1ull << 50 },
87 33bf1ca4 2023-04-08 benni { 'T', 1ull << 40 },
88 33bf1ca4 2023-04-08 benni { 'G', 1ull << 30 },
89 33bf1ca4 2023-04-08 benni { 'M', 1ull << 20 },
90 33bf1ca4 2023-04-08 benni { 'K', 1ull << 10 },
91 33bf1ca4 2023-04-08 benni { '0', 0 },
94 33bf1ca4 2023-04-08 benni char sym = 'B';
95 33bf1ca4 2023-04-08 benni uint64_t factor = 1;
97 33bf1ca4 2023-04-08 benni for (const struct unit *u = &units[0]; u->factor; ++u) {
98 33bf1ca4 2023-04-08 benni if (sz >= (u->factor * 9 / 10)) {
99 33bf1ca4 2023-04-08 benni sym = u->sym;
100 33bf1ca4 2023-04-08 benni factor = u->factor;
105 33bf1ca4 2023-04-08 benni const unsigned scaled10 = sz * 10 / factor;
106 33bf1ca4 2023-04-08 benni const unsigned scaled = sz / factor;
107 33bf1ca4 2023-04-08 benni if (scaled10 >= 1000) {
108 33bf1ca4 2023-04-08 benni printf (" %u", scaled);
109 33bf1ca4 2023-04-08 benni } else if (scaled10 >= 100) {
110 33bf1ca4 2023-04-08 benni printf ("%u.%u", scaled, scaled10 % 10);
112 33bf1ca4 2023-04-08 benni printf (" %u.%u", scaled, scaled10 % 10);
115 33bf1ca4 2023-04-08 benni putchar (sym);
118 33bf1ca4 2023-04-08 benni static int usage (void)
120 33bf1ca4 2023-04-08 benni fputs ("Usage: lsblk [-V]\n", stderr);
124 33bf1ca4 2023-04-08 benni int main (int argc, char *argv[])
126 33bf1ca4 2023-04-08 benni int option;
127 33bf1ca4 2023-04-08 benni while ((option = getopt (argc, argv, ":V")) != -1) {
128 33bf1ca4 2023-04-08 benni switch (option) {
130 33bf1ca4 2023-04-08 benni puts ("lsblk-" VERSION);
133 33bf1ca4 2023-04-08 benni return usage ();
137 33bf1ca4 2023-04-08 benni if (argc != optind)
138 33bf1ca4 2023-04-08 benni return usage ();
140 33bf1ca4 2023-04-08 benni struct statfs *mounts;
141 33bf1ca4 2023-04-08 benni char *names = disknames ();
143 33bf1ca4 2023-04-08 benni if (pledge ("stdio rpath disklabel", NULL) == -1)
144 33bf1ca4 2023-04-08 benni die ("pledge()");
146 33bf1ca4 2023-04-08 benni const int n_mounts = getmntinfo (&mounts, MNT_NOWAIT);
147 33bf1ca4 2023-04-08 benni if (n_mounts == 0)
148 33bf1ca4 2023-04-08 benni die ("getmntinfo()");
150 33bf1ca4 2023-04-08 benni for (char *disk; (disk = strsep (&names, ",")) != NULL;) {
151 33bf1ca4 2023-04-08 benni struct disklabel label;
152 33bf1ca4 2023-04-08 benni char *colon, *name = NULL;
155 33bf1ca4 2023-04-08 benni colon = strchr (disk, ':');
156 33bf1ca4 2023-04-08 benni if (colon)
157 33bf1ca4 2023-04-08 benni *colon = '\0';
159 33bf1ca4 2023-04-08 benni asprintf (&name, "/dev/%sc", disk);
161 33bf1ca4 2023-04-08 benni fd = open (name, O_RDONLY);
162 33bf1ca4 2023-04-08 benni if (fd < 0)
163 33bf1ca4 2023-04-08 benni die ("open(%s)", name);
165 33bf1ca4 2023-04-08 benni if (ioctl (fd, DIOCGDINFO, &label) < 0)
166 33bf1ca4 2023-04-08 benni die ("ioctl(%s, IOCGDINFO)", name);
168 33bf1ca4 2023-04-08 benni close (fd);
170 33bf1ca4 2023-04-08 benni const uint32_t ss = label.d_secsize;
172 33bf1ca4 2023-04-08 benni char *letter = name + strlen (name) - 1;
174 33bf1ca4 2023-04-08 benni printf ("%s ", disk);
175 bb3f9866 2023-04-08 benni printsize ((((uint64_t)label.d_secperunith << 32) | label.d_secperunit) * ss);
176 33bf1ca4 2023-04-08 benni printf (" %s\n", label.d_typename);
177 33bf1ca4 2023-04-08 benni for (uint16_t i = 0; i < label.d_npartitions; ++i) {
178 33bf1ca4 2023-04-08 benni const struct partition *p = &label.d_partitions[i];
179 33bf1ca4 2023-04-08 benni const uint64_t size = (((uint64_t)p->p_sizeh << 32) | p->p_size) * ss;
180 33bf1ca4 2023-04-08 benni if (size == 0)
183 33bf1ca4 2023-04-08 benni *letter = i + 'a';
184 33bf1ca4 2023-04-08 benni const char *mp = "";
185 33bf1ca4 2023-04-08 benni for (int i = 0; i < n_mounts; ++i) {
186 33bf1ca4 2023-04-08 benni const struct statfs *m = &mounts[i];
187 33bf1ca4 2023-04-08 benni if (!strcmp (name, m->f_mntfromname)) {
188 33bf1ca4 2023-04-08 benni mp = m->f_mntonname;
193 33bf1ca4 2023-04-08 benni printf (" %s%c ", disk, *letter);
194 33bf1ca4 2023-04-08 benni printsize (size);
195 33bf1ca4 2023-04-08 benni printf (" %s %s\n", fstypenames[p->p_fstype], mp);
199 33bf1ca4 2023-04-08 benni free (names);
