proplib/prop_du.c

   1 #include <err.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <sys/wait.h>    /* for waitpid() macros */
   6 #include <prop/proplib.h>
   7
   8 #define INIT_CAPACITY 100
   9 #define MAX_STR 100
  10 #define MAX_TOKENS 3
  11
  12 int main(void)
  13 {
  14     char str[MAX_STR];
  15     char *tokens[MAX_TOKENS];    /* for ``du'' output parse */
  16     char *last, *p;
  17     int i, ret;
  18     FILE *fp;
  19     prop_dictionary_t pd;
  20     prop_number_t pn;
  21
  22     /* Initiate pipe stream to ``du'' */
  23     fp = popen("du", "r");
  24     if (fp == NULL) {
  25         perror("popen()");
  26         exit(EXIT_FAILURE);
  27     }
  28
  29     /* Create dictionary */
  30     pd = prop_dictionary_create_with_capacity(INIT_CAPACITY);
  31     if (pd == NULL) {
  32         pclose(fp);
  33         errx(EXIT_FAILURE, "prop_dictionary_create_with_capacity()");
  34     }
  35
  36     /* Read from stream */
  37     while (fgets(str, MAX_STR, fp) != NULL) {
  38         /* Parse output of ``du'' command */
  39         i = 0;
  40         for ((p = strtok_r(str, "\t", &last)); p;
  41              (p = strtok_r(NULL, "\t", &last)), i++) {
  42             if (i < MAX_TOKENS - 1)
  43                 tokens[i] = p;
  44         }
  45         tokens[i] = NULL;
  46
  47         /* Trim '\n' from tokens[1] */
  48         (tokens[1])[strlen(tokens[1]) - 1] = '\0';
  49
  50         /*
  51          * We use a signed prop_number_t object, so that
  52          * when externalized it will be represented as decimal
  53          * (unsigned numbers are externalized in base-16)
  54          */
  55         pn = prop_number_create_integer(atoi(tokens[0]));
  56         prop_dictionary_set(pd, tokens[1], pn);
  57         prop_object_release(pn);
  58     }
  59
  60     prop_dictionary_externalize_to_file(pd, "./data.xml");
  61     prop_object_release(pd);
  62
  63     /* Close pipe stream */
  64     ret = pclose(fp);
  65     if  (ret == -1) {
  66         perror("pclose()");
  67         exit(EXIT_FAILURE);
  68     }
  69
  70     return EXIT_SUCCESS;
  71 }