builtin-prune-packed.c

   1 #include "builtin.h"
   2 #include "cache.h"
   3 #include "progress.h"
   4
   5 static const char prune_packed_usage[] =
   6 "git-prune-packed [-n] [-q]";
   7
   8 #define DRY_RUN 01
   9 #define VERBOSE 02
  10
  11 static struct progress *progress;
  12
  13 static void prune_dir(int i, DIR *dir, char *pathname, int len, int opts)
  14 {
  15         struct dirent *de;
  16         char hex[40];
  17
  18         if (opts == VERBOSE)
  19                 display_progress(progress, i + 1);
  20
  21         sprintf(hex, "%02x", i);
  22         while ((de = readdir(dir)) != NULL) {
  23                 unsigned char sha1[20];
  24                 if (strlen(de->d_name) != 38)
  25                         continue;
  26                 memcpy(hex+2, de->d_name, 38);
  27                 if (get_sha1_hex(hex, sha1))
  28                         continue;
  29                 if (!has_sha1_pack(sha1, NULL))
  30                         continue;
  31                 memcpy(pathname + len, de->d_name, 38);
  32                 if (opts & DRY_RUN)
  33                         printf("rm -f %s\n", pathname);
  34                 else if (unlink(pathname) < 0)
  35                         error("unable to unlink %s", pathname);
  36         }
  37         pathname[len] = 0;
  38         rmdir(pathname);
  39 }
  40
  41 void prune_packed_objects(int opts)
  42 {
  43         int i;
  44         static char pathname[PATH_MAX];
  45         const char *dir = get_object_directory();
  46         int len = strlen(dir);
  47
  48         if (opts == VERBOSE)
  49                 progress = start_progress_delay("Removing duplicate objects",
  50                         256, 95, 2);
  51
  52         if (len > PATH_MAX - 42)
  53                 die("impossible object directory");
  54         memcpy(pathname, dir, len);
  55         if (len && pathname[len-1] != '/')
  56                 pathname[len++] = '/';
  57         for (i = 0; i < 256; i++) {
  58                 DIR *d;
  59
  60                 sprintf(pathname + len, "%02x/", i);
  61                 d = opendir(pathname);
  62                 if (!d)
  63                         continue;
  64                 prune_dir(i, d, pathname, len + 3, opts);
  65                 closedir(d);
  66         }
  67         if (opts == VERBOSE)
  68                 stop_progress(&progress);
  69 }
  70
  71 int cmd_prune_packed(int argc, const char **argv, const char *prefix)
  72 {
  73         int i;
  74         int opts = VERBOSE;
  75
  76         for (i = 1; i < argc; i++) {
  77                 const char *arg = argv[i];
  78
  79                 if (*arg == '-') {
  80                         if (!strcmp(arg, "-n"))
  81                                 opts |= DRY_RUN;
  82                         else if (!strcmp(arg, "-q"))
  83                                 opts &= ~VERBOSE;
  84                         else
  85                                 usage(prune_packed_usage);
  86                         continue;
  87                 }
  88                 /* Handle arguments here .. */
  89                 usage(prune_packed_usage);
  90         }
  91         sync();
  92         prune_packed_objects(opts);
  93         return 0;
  94 }