From 7317e77da344d1a5eff16e28771dcda91acd4ae3 Mon Sep 17 00:00:00 2001
From: rsc <devnull@localhost>
Date: Sun, 25 Apr 2004 22:24:56 +0000
Subject: display correct size of block devices for linux. that was
 surprisingly easy.

---
 src/lib9/_p9dir.c | 31 +++++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/lib9/_p9dir.c b/src/lib9/_p9dir.c
index 6611751a..18e899b6 100644
--- a/src/lib9/_p9dir.c
+++ b/src/lib9/_p9dir.c
@@ -1,7 +1,6 @@
 #include <u.h>
 #define NOPLAN9DEFINES
 #include <libc.h>
-
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <dirent.h>
@@ -43,6 +42,28 @@ isdisk(struct stat *st)
 #define _HAVEDISKLABEL
 #endif
 
+#if defined(__linux__)
+#include <linux/hdreg.h>
+#include <linux/fs.h>
+#include <sys/ioctl.h>
+static vlong
+disksize(int fd, int dev)
+{
+	u64int u64;
+	long l;
+	struct hd_geometry geo;
+
+	if(ioctl(fd, BLKGETSIZE64, &u64) >= 0)
+		return u64;
+	if(ioctl(fd, BLKGETSIZE, &l) >= 0)
+		return l*512;
+	if(ioctl(fd, HDIO_GETGEO, &geo) >= 0)
+		return (vlong)geo.heads*geo.sectors*geo.cylinders*512;
+	return 0;
+}
+#define _HAVEDISKSIZE
+#endif
+
 #if !defined(__linux__) && !defined(__sun__)
 #define _HAVESTGEN
 #endif
@@ -61,7 +82,7 @@ _p9dir(struct stat *st, char *name, Dir *d, char **str, char *estr)
 	static struct group *g;
 	static struct passwd *p;
 	static int gid, uid;
-	int sz;
+	int sz, fd;
 
 	sz = 0;
 	if(d)
@@ -154,6 +175,12 @@ _p9dir(struct stat *st, char *name, Dir *d, char **str, char *estr)
 		}
 
 		/* fetch real size for disks */
+#ifdef _HAVEDISKSIZE
+		if((fd = open(name, O_RDONLY)) >= 0){
+			d->length = disksize(fd, major(st->st_dev));
+			close(fd);
+		}
+#endif
 #ifdef _HAVEDISKLABEL
 		if(isdisk(st)){
 			int fd, n;
-- 
cgit v1.2.3