builtin-prune.c

   1 #include "cache.h"
   2 #include "commit.h"
   3 #include "diff.h"
   4 #include "revision.h"
   5 #include "builtin.h"
   6 #include "reachable.h"
   7
   8 static const char prune_usage[] = "git-prune [-n]";
   9 static int show_only;
  10 static unsigned long expire;
  11
  12 static int prune_object(char *path, const char *filename, const unsigned char *sha1)
  13 {
  14         const char *fullpath = mkpath("%s/%s", path, filename);
  15         if (expire) {
  16                 struct stat st;
  17                 if (lstat(fullpath, &st))
  18                         return error("Could not stat '%s'", fullpath);
  19                 if (st.st_mtime > expire)
  20                         return 0;
  21         }
  22         if (show_only) {
  23                 enum object_type type = sha1_object_info(sha1, NULL);
  24                 printf("%s %s\n", sha1_to_hex(sha1),
  25                        (type > 0) ? typename(type) : "unknown");
  26         } else
  27                 unlink(fullpath);
  28         return 0;
  29 }
  30
  31 static int prune_dir(int i, char *path)
  32 {
  33         DIR *dir = opendir(path);
  34         struct dirent *de;
  35
  36         if (!dir)
  37                 return 0;
  38
  39         while ((de = readdir(dir)) != NULL) {
  40                 char name[100];
  41                 unsigned char sha1[20];
  42                 int len = strlen(de->d_name);
  43
  44                 switch (len) {
  45                 case 2:
  46                         if (de->d_name[1] != '.')
  47                                 break;
  48                 case 1:
  49                         if (de->d_name[0] != '.')
  50                                 break;
  51                         continue;
  52                 case 38:
  53                         sprintf(name, "%02x", i);
  54                         memcpy(name+2, de->d_name, len+1);
  55                         if (get_sha1_hex(name, sha1) < 0)
  56                                 break;
  57
  58                         /*
  59                          * Do we know about this object?
  60                          * It must have been reachable
  61                          */
  62                         if (lookup_object(sha1))
  63                                 continue;
  64
  65                         prune_object(path, de->d_name, sha1);
  66                         continue;
  67                 }
  68                 fprintf(stderr, "bad sha1 file: %s/%s\n", path, de->d_name);
  69         }
  70         if (!show_only)
  71                 rmdir(path);
  72         closedir(dir);
  73         return 0;
  74 }
  75
  76 static void prune_object_dir(const char *path)
  77 {
  78         int i;
  79         for (i = 0; i < 256; i++) {
  80                 static char dir[4096];
  81                 sprintf(dir, "%s/%02x", path, i);
  82                 prune_dir(i, dir);
  83         }
  84 }
  85
  86 /*
  87  * Write errors (particularly out of space) can result in
  88  * failed temporary packs (and more rarely indexes and other
  89  * files begining with "tmp_") accumulating in the
  90  * object directory.
  91  */
  92 static void remove_temporary_files(void)
  93 {
  94         DIR *dir;
  95         struct dirent *de;
  96         char* dirname=get_object_directory();
  97
  98         dir = opendir(dirname);
  99         if (!dir) {
 100                 fprintf(stderr, "Unable to open object directory %s\n",
 101                         dirname);
 102                 return;
 103         }
 104         while ((de = readdir(dir)) != NULL) {
 105                 if (!prefixcmp(de->d_name, "tmp_")) {
 106                         char name[PATH_MAX];
 107                         int c = snprintf(name, PATH_MAX, "%s/%s",
 108                                          dirname, de->d_name);
 109                         if (c < 0 || c >= PATH_MAX)
 110                                 continue;
 111                         if (expire) {
 112                                 struct stat st;
 113                                 if (stat(name, &st) != 0 || st.st_mtime >= expire)
 114                                         continue;
 115                         }
 116                         printf("Removing stale temporary file %s\n", name);
 117                         if (!show_only)
 118                                 unlink(name);
 119                 }
 120         }
 121         closedir(dir);
 122 }
 123
 124 int cmd_prune(int argc, const char **argv, const char *prefix)
 125 {
 126         int i;
 127         struct rev_info revs;
 128
 129         for (i = 1; i < argc; i++) {
 130                 const char *arg = argv[i];
 131                 if (!strcmp(arg, "-n")) {
 132                         show_only = 1;
 133                         continue;
 134                 }
 135                 if (!strcmp(arg, "--expire")) {
 136                         if (++i < argc) {
 137                                 expire = approxidate(argv[i]);
 138                                 continue;
 139                         }
 140                 }
 141                 else if (!prefixcmp(arg, "--expire=")) {
 142                         expire = approxidate(arg + 9);
 143                         continue;
 144                 }
 145                 usage(prune_usage);
 146         }
 147
 148         save_commit_buffer = 0;
 149         init_revisions(&revs, prefix);
 150         mark_reachable_objects(&revs, 1);
 151
 152         prune_object_dir(get_object_directory());
 153
 154         sync();
 155         prune_packed_objects(show_only);
 156         remove_temporary_files();
 157         return 0;
 158 }