test/fastbulk/getpkgsrcdir.c

   1 /*
   2  *
   3  */
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <unistd.h>
   8 #include <string.h>
   9 #include <sys/stat.h>
  10
  11 static void usage(void);
  12
  13 /*
  14  * Process paths on stdin and generate a directory path if it looks
  15  * like a pkgsrc directory.
  16  *
  17  * av[1]
  18  */
  19 int
  20 main(int ac, char **av)
  21 {
  22         char buf[1024];
  23         char *path;
  24         char *bpath;
  25         struct stat st;
  26         size_t len;
  27         size_t blen;
  28         int count;
  29         int ch;
  30         int stripOpt = 0;
  31
  32         while ((ch = getopt(ac, av, "s")) != -1) {
  33                 switch(ch) {
  34                 case 's':
  35                         stripOpt = 1;
  36                         break;
  37                 default:
  38                         usage();
  39                         /* NOT REACHED */
  40                 }
  41         }
  42         ac -= optind;
  43         av += optind;
  44         if (ac != 1) {
  45                 fprintf(stderr, "requires base directory as first argument\n");
  46                 exit(1);
  47         }
  48
  49         /*
  50          * Base dir
  51          */
  52         bpath = strdup(av[0]);
  53         blen = strlen(bpath);
  54         while (blen && bpath[blen-1] == '/')
  55                 --blen;
  56         bpath[blen] = 0;
  57
  58         /*
  59          * Process lines
  60          */
  61         while (fgets(buf, sizeof(buf) - 32, stdin) != NULL) {
  62                 path = strtok(buf, " \t\r\n");
  63                 if (path == NULL || *path == 0)
  64                         continue;
  65                 len = strlen(path);
  66                 if (len < blen || bcmp(path, bpath, blen) != 0)
  67                         continue;
  68                 if (stat(path, &st) != 0)
  69                         continue;
  70                 len = strlen(path);
  71                 if (!S_ISDIR(st.st_mode)) {
  72                         while (len && path[len-1] != '/')
  73                                 --len;
  74                 }
  75                 while (len && path[len-1] == '/')
  76                         --len;
  77                 strcpy(path + len, "/Makefile");
  78                 if (stat(path, &st) != 0)
  79                         continue;
  80                 strcpy(path + len, "/DESCR");
  81                 if (stat(path, &st) != 0)
  82                         continue;
  83                 path[len] = 0;
  84
  85                 /*
  86                  * Must be at least one sub-directory
  87                  */
  88                 count = 0;
  89                 for (len = blen; path[len]; ++len) {
  90                         if (path[len] == '/')
  91                                 ++count;
  92                 }
  93                 if (count < 2)
  94                         continue;
  95                 if (stripOpt)
  96                         printf("%s\n", path + blen + 1);
  97                 else
  98                         printf("%s\n", path);
  99         }
 100         return(0);
 101 }
 102
 103 static void
 104 usage(void)
 105 {
 106         fprintf(stderr, "getpkgsrcdir: unsupported option\n");
 107         exit(1);
 108 }