skip duplicate util-vserver dir
[util-vserver.git] / src / vdu.c
diff --git a/src/vdu.c b/src/vdu.c
new file mode 100644 (file)
index 0000000..480784f
--- /dev/null
+++ b/src/vdu.c
@@ -0,0 +1,334 @@
+// $Id$    --*- c -*--
+
+// Copyright (C) 2006 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
+//  
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; version 2 of the License.
+//  
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//  
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+
+#ifdef HAVE_CONFIG_H
+#  include <config.h>
+#endif
+
+#include "util.h"
+#include <lib/vserver.h>
+#include <lib/fmt.h>
+
+#include <stdlib.h>
+#include <getopt.h>
+#include <stdint.h>
+#include <errno.h>
+#include <sys/stat.h>
+#include <dirent.h>
+#include <fcntl.h>
+
+#define ENSC_WRAPPERS_PREFIX   "vdu: "
+#define ENSC_WRAPPERS_VSERVER  1
+#define ENSC_WRAPPERS_UNISTD   1
+#define ENSC_WRAPPERS_DIRENT   1
+#define ENSC_WRAPPERS_FCNTL    1
+#define ENSC_WRAPPERS_STAT     1
+#include <wrappers.h>
+
+#define CMD_HELP               0x1000
+#define CMD_VERSION            0x1001
+#define CMD_XID                        0x2000
+#define CMD_SPACE              0x2001
+#define CMD_INODES             0x2002
+#define CMD_SCRIPT             0x2003
+#define CMD_BLOCKSIZE          0x2005
+
+int                    wrapper_exit_code = 1;
+
+struct option const
+CMDLINE_OPTIONS[] = {
+  { "help",           no_argument,       0, CMD_HELP },
+  { "version",        no_argument,       0, CMD_VERSION },
+  { "xid",            required_argument, 0, CMD_XID },
+  { "space",          no_argument,       0, CMD_SPACE },
+  { "inodes",         no_argument,       0, CMD_INODES },
+  { "script",         no_argument,       0, CMD_SCRIPT },
+  { "blocksize",      required_argument, 0, CMD_BLOCKSIZE },
+  {0,0,0,0}
+};
+
+struct Arguments {
+    xid_t              xid;
+    bool               space;
+    bool               inodes;
+    bool               script;
+    unsigned long      blocksize;
+};
+
+struct Result {
+    uint_least64_t     blocks;
+    uint_least64_t     inodes;
+};
+
+struct TraversalParams {
+    struct Arguments const * const     args;
+    struct Result * const              result;
+};
+
+static void
+showHelp(int fd, char const *cmd, int res)
+{
+  WRITE_MSG(fd, "Usage:\n    ");
+  WRITE_STR(fd, cmd);
+  WRITE_MSG(fd,
+           " --xid <xid> (--space|--inodes) [--blocksize <blocksize>] [--script] <directory>*\n"
+           "\n"
+           "Please report bugs to " PACKAGE_BUGREPORT "\n");
+
+  exit(res);
+}
+
+static void
+showVersion()
+{
+  WRITE_MSG(1,
+           "vdu " VERSION " -- calculates the size of a directory\n"
+           "This program is part of " PACKAGE_STRING "\n\n"
+           "Copyright (C) 2006 Enrico Scholz\n"
+           VERSION_COPYRIGHT_DISCLAIMER);
+  exit(0);
+}
+
+/* basic hash table implementation for inode tracking */
+#define HASH_SIZE 103
+typedef struct hash_entry {
+  struct hash_entry *next;
+  ino_t inode;
+} hash_entry;
+
+typedef struct hash_table {
+  hash_entry *entries[HASH_SIZE];
+} hash_table;
+
+static hash_table ht;
+
+static void
+hash_init(void)
+{
+  memset(&ht, 0, sizeof(hash_table));
+}
+
+static void
+hash_free(void)
+{
+  int i;
+  hash_entry *e, *p;
+  for (i = 0; i < HASH_SIZE; i++) {
+    for (e = ht.entries[i], p = NULL; e; e = e->next) {
+      free(p);
+      p = e;
+    }
+    free(p);
+  }
+}
+
+static int
+hash_insert(ino_t inode)
+{
+  hash_entry *e, *p;
+  unsigned int hashval = inode % HASH_SIZE;
+
+  /* no one else here */
+  if (ht.entries[hashval] == NULL) {
+    ht.entries[hashval]        = malloc(sizeof(hash_entry));
+    ht.entries[hashval]->next  = NULL;
+    ht.entries[hashval]->inode = inode;
+    return 0;
+  }
+
+  for (e = ht.entries[hashval], p = NULL; e; e = e->next) {
+    /* already in the hash table */
+    if (e->inode == inode)
+      return -1;
+    else if (e->inode > inode) {
+      /* we're first */
+      if (p == NULL) {
+       ht.entries[hashval]        = malloc(sizeof(hash_entry));
+       ht.entries[hashval]->next  = e;
+       ht.entries[hashval]->inode = inode;
+      }
+      /* we're in the middle */
+      else {
+       p->next        = malloc(sizeof(hash_entry));
+       p->next->next  = e;
+       p->next->inode = inode;
+      }
+      return 0;
+    }
+    p = e;
+  }
+  /* we're last */
+  p->next        = malloc(sizeof(hash_entry));
+  p->next->next  = NULL;
+  p->next->inode = inode;
+
+  return 0;
+}
+
+static void
+visitDirEntry(char const *name, dev_t const dir_dev,
+             struct TraversalParams *params);
+
+static void
+visitDir(char const *name, struct stat const *expected_stat, struct TraversalParams *params)
+{
+  int          fd = Eopen(".", O_RDONLY|O_DIRECTORY, 0);
+  DIR *                dir;
+
+  EsafeChdir(name, expected_stat);
+
+  dir = Eopendir(".");
+
+  for (;;) {
+    struct dirent              *ent = Ereaddir(dir);
+    if (ent==0) break;
+
+    if (isDotfile(ent->d_name)) continue;
+    visitDirEntry(ent->d_name, expected_stat->st_dev, params);
+  }
+
+  Eclosedir(dir);
+
+  Efchdir(fd);
+  Eclose(fd);
+}
+
+static void
+visitDirEntry(char const *name, dev_t const dir_dev,
+             struct TraversalParams *params)
+{
+  struct stat          st;
+  xid_t                        xid;
+  
+  ElstatD(name, &st);
+
+  xid = vc_getfilecontext(name);
+  if (xid == params->args->xid &&
+      (st.st_nlink == 1 || hash_insert(st.st_ino) != -1)) {
+    params->result->blocks += st.st_blocks;
+    params->result->inodes += 1;
+  }
+
+  if (S_ISDIR(st.st_mode) && dir_dev == st.st_dev)
+    visitDir(name, &st, params);
+}
+
+static void
+visitDirStart(char const *name, struct TraversalParams *params)
+{
+  struct stat  st;
+  int          fd = Eopen(".", O_RDONLY|O_DIRECTORY, 0);
+
+  Estat(name, &st);
+  Echdir(name);
+
+  visitDirEntry(".", st.st_dev, params);
+
+  Efchdir(fd);
+  Eclose(fd);  
+}
+
+int main(int argc, char *argv[])
+{
+  struct Arguments             args = {
+    .xid       = VC_NOCTX,
+    .space     = false,
+    .inodes    = false,
+    .script    = false,
+    .blocksize = 1024,
+  };
+  
+  while (1) {
+    int                c = getopt_long(argc, argv, "+", CMDLINE_OPTIONS, 0);
+    if (c==-1) break;
+
+    switch (c) {
+      case CMD_HELP    :  showHelp(1, argv[0], 0);
+      case CMD_VERSION :  showVersion();
+      case CMD_XID     :  args.xid = Evc_xidopt2xid(optarg,true); break;
+      case CMD_SPACE   :  args.space  = true;                     break;
+      case CMD_INODES  :  args.inodes = true;                     break;
+      case CMD_SCRIPT  :  args.script = true;                     break;
+      case CMD_BLOCKSIZE:
+       if (!isNumberUnsigned(optarg, &args.blocksize, false)) {
+         WRITE_MSG(2, "Invalid block size argument: '");
+         WRITE_STR(2, optarg);
+         WRITE_MSG(2, "'; try '--help' for more information\n");
+         return EXIT_FAILURE;
+       }
+       break;
+      default          :
+       WRITE_MSG(2, "Try '");
+       WRITE_STR(2, argv[0]);
+       WRITE_MSG(2, " --help' for more information.\n");
+       return 255;
+       break;
+    }
+  }
+
+  if (args.xid==VC_NOCTX)
+    WRITE_MSG(2, "No xid specified; try '--help' for more information\n");
+  else if (!args.space && !args.inodes)
+    WRITE_MSG(2, "Must specify --space or --inodes; try '--help' for more information\n");
+  else if (optind==argc)
+    WRITE_MSG(2, "No directory specified; try '--help' for more information\n");
+  else {
+    int                i;
+    size_t     len;
+    struct Result              result;
+    struct TraversalParams     params   = {
+      .args   = &args,
+      .result = &result
+    };
+
+    for (i = optind; i < argc; i++) {
+      uint_least64_t           size;
+      char                     buf[sizeof(size)*3 + 3];
+      char const *             delim = "";
+      
+      result.blocks = 0;
+      result.inodes = 0;
+
+      hash_init();
+      visitDirStart(argv[i], &params);
+      hash_free();
+
+      if (!args.script) {
+       WRITE_STR(1, argv[i]);
+       WRITE_MSG(1, " ");
+      }
+
+      if (args.space) {
+       len = utilvserver_fmt_uint64(buf, result.blocks*512 / args.blocksize);
+       if (*delim) WRITE_STR(1, delim);
+       Vwrite(1, buf, len);
+       delim = " ";
+      }
+      if (args.inodes) {
+       len = utilvserver_fmt_uint64(buf, result.inodes);
+       if (*delim) WRITE_STR(1, delim);
+       Vwrite(1, buf, len);
+       delim = " ";
+      }        
+      WRITE_MSG(1, "\n");
+    }
+    return EXIT_SUCCESS;
+  }
+
+  return EXIT_FAILURE;
+}