From: Francois Tigeot <ftigeot@wolfpond.org>
Date: Sat, 6 Aug 2011 20:57:36 +0000 (+0200)
Subject: Add a vquota(8) utility
X-Git-Tag: v3.0.0~503
X-Git-Url: https://gitweb.dragonflybsd.org/~nant/dragonfly.git/commitdiff_plain/6a4c3e18112f181025eb8fb5c1e67926366f43ec

Add a vquota(8) utility
---

diff --git a/sbin/Makefile b/sbin/Makefile
index 846ac44f73..31f0f00389 100644
--- a/sbin/Makefile
+++ b/sbin/Makefile
@@ -96,7 +96,8 @@ SUBDIR=	adjkerntz \
 	tunefs \
 	udevd \
 	umount \
-	vinum
+	vinum \
+	vquota
 
 .if ${MACHINE_ARCH} == i386 || ${MACHINE_ARCH} == x86_64
 .if ${MACHINE_ARCH} == i386
diff --git a/sbin/vquota/Makefile b/sbin/vquota/Makefile
new file mode 100644
index 0000000000..216afa90e3
--- /dev/null
+++ b/sbin/vquota/Makefile
@@ -0,0 +1,5 @@
+PROG=	vquota
+SRCS=	vquota.c
+MAN=	vquota.8
+
+.include <bsd.prog.mk>
diff --git a/sbin/vquota/vquota.c b/sbin/vquota/vquota.c
new file mode 100644
index 0000000000..49147cb896
--- /dev/null
+++ b/sbin/vquota/vquota.c
@@ -0,0 +1,101 @@
+/*
+ * Copyright (c) 2011 François Tigeot <ftigeot@wolfpond.org>
+ * All rights reserved.
+ * 
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in
+ *    the documentation and/or other materials provided with the
+ *    distribution.
+ * 3. Neither the name of The DragonFly Project nor the names of its
+ *    contributors may be used to endorse or promote products derived
+ *    from this software without specific, prior written permission.
+ * 
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
+ * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
+ * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <sys/stat.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <err.h>
+#include <string.h>
+#include <fts.h>
+
+static void usage(int);
+static int get_dirsize( char* );
+
+static void usage(int retcode) {
+	fprintf( stderr, "usage: vquota check directory\n" );
+	exit(retcode);
+}
+
+static int get_dirsize( char* dirname ) {
+	FTS			*fts;
+	FTSENT		*p;
+	char*		fts_args[2];
+	uint64_t	size_of_files = 0;
+	int			retval = 0;
+
+	/* TODO: check directory name sanity */
+	fts_args[0] = dirname;
+	fts_args[1] = NULL;
+
+	if ((fts = fts_open( fts_args, FTS_PHYSICAL, NULL)) == NULL)
+		err(1, "fts_open() failed");
+
+	while ((p = fts_read(fts)) != NULL) {
+		switch (p->fts_info) {
+		/* directories, ignore them */
+		case FTS_D:
+		case FTS_DC:
+		case FTS_DP:
+			break;
+		/* read errors, warn, continue and flag */
+		case FTS_DNR:
+		case FTS_ERR:
+		case FTS_NS:
+			warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
+			retval = 1;
+			break;
+		default:
+			/* files with more than one link, undecided */
+			// if (p->fts_statp->st_nlink > 1 && linkchk(p))
+			if (p->fts_statp->st_nlink > 1)
+				fprintf( stderr, "%s has more than one link\n", p->fts_name );
+				
+			size_of_files += p->fts_statp->st_size;
+		}
+	}
+
+	printf( "%lu\n", size_of_files );
+	return retval;
+}
+
+int main( int argc, char **argv ) {
+	
+	if (argc < 3)
+		usage(1);
+	
+	if (strcmp(argv[1], "check") == 0) {
+		return get_dirsize( argv[2] );
+	}
+
+	return 0;
+}