mirror of https://github.com/sheumann/hush.git
137 lines
2.9 KiB
C
137 lines
2.9 KiB
C
#include "internal.h"
|
|
#include <stdio.h>
|
|
#include <mntent.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/vfs.h>
|
|
#include <fstab.h>
|
|
|
|
static const char df_usage[] = "df [filesystem ...]\n"
|
|
"\n"
|
|
"\tPrint the filesystem space used and space available.\n";
|
|
|
|
|
|
static int
|
|
df(char* device, const char * mountPoint)
|
|
{
|
|
struct statfs s;
|
|
long blocks_used;
|
|
long blocks_percent_used;
|
|
|
|
if ( statfs(mountPoint, &s) != 0 ) {
|
|
perror(mountPoint);
|
|
return 1;
|
|
}
|
|
|
|
if ( s.f_blocks > 0 ) {
|
|
blocks_used = s.f_blocks - s.f_bfree;
|
|
blocks_percent_used = (long)
|
|
(blocks_used * 100.0 / (blocks_used + s.f_bavail) + 0.5);
|
|
if ( strcmp(device, "/dev/root")==0)
|
|
device=(getfsfile ("/"))->fs_spec;
|
|
|
|
printf(
|
|
"%-20s %9ld %9ld %9ld %3ld%% %s\n",
|
|
device,
|
|
(long)(s.f_blocks * (s.f_bsize / 1024.0)),
|
|
(long)((s.f_blocks - s.f_bfree) * (s.f_bsize / 1024.0)),
|
|
(long)(s.f_bavail * (s.f_bsize / 1024.0)),
|
|
blocks_percent_used,
|
|
mountPoint);
|
|
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Given a block device, find the mount table entry if that block device
|
|
* is mounted.
|
|
*
|
|
* Given any other file (or directory), find the mount table entry for its
|
|
* filesystem.
|
|
*/
|
|
extern struct mntent *
|
|
findMountPoint(const char* name, const char* table)
|
|
{
|
|
struct stat s;
|
|
dev_t mountDevice;
|
|
FILE * mountTable;
|
|
struct mntent * mountEntry;
|
|
|
|
if ( stat(name, &s) != 0 )
|
|
return 0;
|
|
|
|
if ( (s.st_mode & S_IFMT) == S_IFBLK )
|
|
mountDevice = s.st_rdev;
|
|
else
|
|
mountDevice = s.st_dev;
|
|
|
|
|
|
if ( (mountTable = setmntent(table, "r")) == 0 )
|
|
return 0;
|
|
|
|
while ( (mountEntry = getmntent(mountTable)) != 0 ) {
|
|
if ( strcmp(name, mountEntry->mnt_dir) == 0
|
|
|| strcmp(name, mountEntry->mnt_fsname) == 0 ) /* String match. */
|
|
break;
|
|
if ( stat(mountEntry->mnt_fsname, &s) == 0
|
|
&& s.st_rdev == mountDevice ) /* Match the device. */
|
|
break;
|
|
if ( stat(mountEntry->mnt_dir, &s) == 0
|
|
&& s.st_dev == mountDevice ) /* Match the directory's mount point. */
|
|
break;
|
|
}
|
|
endmntent(mountTable);
|
|
return mountEntry;
|
|
}
|
|
|
|
|
|
|
|
extern int
|
|
df_main(int argc, char * * argv)
|
|
{
|
|
printf("%-20s %-14s %s %s %s %s\n", "Filesystem",
|
|
"1k-blocks", "Used", "Available", "Use%", "Mounted on");
|
|
|
|
if ( argc > 1 ) {
|
|
struct mntent* mountEntry;
|
|
int status;
|
|
|
|
while ( argc > 1 ) {
|
|
if ( (mountEntry = findMountPoint(argv[1], "/proc/mounts")) == 0 )
|
|
{
|
|
fprintf(stderr, "%s: can't find mount point.\n" ,argv[1]);
|
|
return 1;
|
|
}
|
|
status = df(mountEntry->mnt_fsname, mountEntry->mnt_dir);
|
|
if ( status != 0 )
|
|
return status;
|
|
argc--;
|
|
argv++;
|
|
}
|
|
return 0;
|
|
}
|
|
else {
|
|
FILE * mountTable;
|
|
struct mntent * mountEntry;
|
|
|
|
mountTable = setmntent("/proc/mounts", "r");
|
|
if ( mountTable == 0) {
|
|
perror("/proc/mounts");
|
|
exit( FALSE);
|
|
}
|
|
|
|
while ( (mountEntry = getmntent (mountTable))) {
|
|
int status=df(mountEntry->mnt_fsname, mountEntry->mnt_dir);
|
|
if (status)
|
|
return status;
|
|
}
|
|
endmntent(mountTable);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|