Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 1 | #include "cache.h" |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 2 | #include "commit.h" |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 3 | #include "diff.h" |
| 4 | #include "revision.h" |
| 5 | #include "builtin.h" |
Junio C Hamano | 9442147 | 2007-01-06 02:16:17 -0800 | [diff] [blame] | 6 | #include "reachable.h" |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 7 | #include "parse-options.h" |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 8 | |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 9 | static const char * const prune_usage[] = { |
Michael J Gruber | b35ddf4 | 2008-09-29 18:49:52 +0200 | [diff] [blame] | 10 | "git prune [-n] [-v] [--expire <time>] [--] [<head>...]", |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 11 | NULL |
| 12 | }; |
David Rientjes | 96f1e58 | 2006-08-15 10:23:48 -0700 | [diff] [blame] | 13 | static int show_only; |
Michael J Gruber | b35ddf4 | 2008-09-29 18:49:52 +0200 | [diff] [blame] | 14 | static int verbose; |
Johannes Schindelin | f01913e | 2007-11-29 20:59:55 +0000 | [diff] [blame] | 15 | static unsigned long expire; |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 16 | |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 17 | static int prune_tmp_object(const char *path, const char *filename) |
Brandon Casey | 0e8316c | 2008-07-24 17:41:12 -0500 | [diff] [blame] | 18 | { |
| 19 | const char *fullpath = mkpath("%s/%s", path, filename); |
| 20 | if (expire) { |
| 21 | struct stat st; |
| 22 | if (lstat(fullpath, &st)) |
| 23 | return error("Could not stat '%s'", fullpath); |
| 24 | if (st.st_mtime > expire) |
| 25 | return 0; |
| 26 | } |
| 27 | printf("Removing stale temporary file %s\n", fullpath); |
| 28 | if (!show_only) |
| 29 | unlink(fullpath); |
| 30 | return 0; |
| 31 | } |
| 32 | |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 33 | static int prune_object(char *path, const char *filename, const unsigned char *sha1) |
| 34 | { |
Johannes Schindelin | f01913e | 2007-11-29 20:59:55 +0000 | [diff] [blame] | 35 | const char *fullpath = mkpath("%s/%s", path, filename); |
| 36 | if (expire) { |
| 37 | struct stat st; |
| 38 | if (lstat(fullpath, &st)) |
| 39 | return error("Could not stat '%s'", fullpath); |
| 40 | if (st.st_mtime > expire) |
| 41 | return 0; |
| 42 | } |
Michael J Gruber | b35ddf4 | 2008-09-29 18:49:52 +0200 | [diff] [blame] | 43 | if (show_only || verbose) { |
Nicolas Pitre | 21666f1 | 2007-02-26 14:55:59 -0500 | [diff] [blame] | 44 | enum object_type type = sha1_object_info(sha1, NULL); |
| 45 | printf("%s %s\n", sha1_to_hex(sha1), |
| 46 | (type > 0) ? typename(type) : "unknown"); |
Michael J Gruber | b35ddf4 | 2008-09-29 18:49:52 +0200 | [diff] [blame] | 47 | } |
| 48 | if (!show_only) |
Johannes Schindelin | f01913e | 2007-11-29 20:59:55 +0000 | [diff] [blame] | 49 | unlink(fullpath); |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 50 | return 0; |
| 51 | } |
| 52 | |
| 53 | static int prune_dir(int i, char *path) |
| 54 | { |
| 55 | DIR *dir = opendir(path); |
| 56 | struct dirent *de; |
| 57 | |
| 58 | if (!dir) |
| 59 | return 0; |
| 60 | |
| 61 | while ((de = readdir(dir)) != NULL) { |
| 62 | char name[100]; |
| 63 | unsigned char sha1[20]; |
| 64 | int len = strlen(de->d_name); |
| 65 | |
| 66 | switch (len) { |
| 67 | case 2: |
| 68 | if (de->d_name[1] != '.') |
| 69 | break; |
| 70 | case 1: |
| 71 | if (de->d_name[0] != '.') |
| 72 | break; |
| 73 | continue; |
| 74 | case 38: |
| 75 | sprintf(name, "%02x", i); |
| 76 | memcpy(name+2, de->d_name, len+1); |
| 77 | if (get_sha1_hex(name, sha1) < 0) |
| 78 | break; |
| 79 | |
| 80 | /* |
| 81 | * Do we know about this object? |
| 82 | * It must have been reachable |
| 83 | */ |
| 84 | if (lookup_object(sha1)) |
| 85 | continue; |
| 86 | |
| 87 | prune_object(path, de->d_name, sha1); |
| 88 | continue; |
| 89 | } |
Brandon Casey | 3d32a46 | 2008-08-05 13:01:50 -0500 | [diff] [blame] | 90 | if (!prefixcmp(de->d_name, "tmp_obj_")) { |
| 91 | prune_tmp_object(path, de->d_name); |
| 92 | continue; |
| 93 | } |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 94 | fprintf(stderr, "bad sha1 file: %s/%s\n", path, de->d_name); |
| 95 | } |
Nicolas Pitre | 3254d21 | 2007-03-20 23:32:13 -0400 | [diff] [blame] | 96 | if (!show_only) |
| 97 | rmdir(path); |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 98 | closedir(dir); |
| 99 | return 0; |
| 100 | } |
| 101 | |
| 102 | static void prune_object_dir(const char *path) |
| 103 | { |
| 104 | int i; |
| 105 | for (i = 0; i < 256; i++) { |
| 106 | static char dir[4096]; |
| 107 | sprintf(dir, "%s/%02x", path, i); |
| 108 | prune_dir(i, dir); |
| 109 | } |
| 110 | } |
| 111 | |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 112 | /* |
| 113 | * Write errors (particularly out of space) can result in |
| 114 | * failed temporary packs (and more rarely indexes and other |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 115 | * files begining with "tmp_") accumulating in the object |
| 116 | * and the pack directories. |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 117 | */ |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 118 | static void remove_temporary_files(const char *path) |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 119 | { |
| 120 | DIR *dir; |
| 121 | struct dirent *de; |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 122 | |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 123 | dir = opendir(path); |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 124 | if (!dir) { |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 125 | fprintf(stderr, "Unable to open directory %s\n", path); |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 126 | return; |
| 127 | } |
Brandon Casey | 0e8316c | 2008-07-24 17:41:12 -0500 | [diff] [blame] | 128 | while ((de = readdir(dir)) != NULL) |
| 129 | if (!prefixcmp(de->d_name, "tmp_")) |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 130 | prune_tmp_object(path, de->d_name); |
David Steven Tweed | 8464010 | 2008-02-07 02:55:14 +0000 | [diff] [blame] | 131 | closedir(dir); |
| 132 | } |
| 133 | |
Linus Torvalds | a633fca | 2006-07-28 22:44:25 -0700 | [diff] [blame] | 134 | int cmd_prune(int argc, const char **argv, const char *prefix) |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 135 | { |
Junio C Hamano | 2430481 | 2007-01-06 02:16:10 -0800 | [diff] [blame] | 136 | struct rev_info revs; |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 137 | const struct option options[] = { |
| 138 | OPT_BOOLEAN('n', NULL, &show_only, |
| 139 | "do not remove, show only"), |
Michael J Gruber | b35ddf4 | 2008-09-29 18:49:52 +0200 | [diff] [blame] | 140 | OPT_BOOLEAN('v', NULL, &verbose, |
| 141 | "report pruned objects"), |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 142 | OPT_DATE(0, "expire", &expire, |
| 143 | "expire objects older than <time>"), |
| 144 | OPT_END() |
| 145 | }; |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 146 | char *s; |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 147 | |
Junio C Hamano | 16157b8 | 2007-01-05 13:31:43 -0800 | [diff] [blame] | 148 | save_commit_buffer = 0; |
Linus Torvalds | a633fca | 2006-07-28 22:44:25 -0700 | [diff] [blame] | 149 | init_revisions(&revs, prefix); |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 150 | |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 151 | argc = parse_options(argc, argv, options, prune_usage, 0); |
Junio C Hamano | fe308f5 | 2008-03-24 23:20:51 -0700 | [diff] [blame] | 152 | while (argc--) { |
| 153 | unsigned char sha1[20]; |
| 154 | const char *name = *argv++; |
| 155 | |
| 156 | if (!get_sha1(name, sha1)) { |
| 157 | struct object *object = parse_object(sha1); |
| 158 | if (!object) |
| 159 | die("bad object: %s", name); |
| 160 | add_pending_object(&revs, object, ""); |
| 161 | } |
| 162 | else |
| 163 | die("unrecognized argument: %s", name); |
| 164 | } |
Michele Ballabio | 629de47 | 2008-03-23 21:50:29 +0100 | [diff] [blame] | 165 | mark_reachable_objects(&revs, 1); |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 166 | prune_object_dir(get_object_directory()); |
| 167 | |
J. Bruce Fields | 2eb53e6 | 2006-10-22 19:01:23 -0400 | [diff] [blame] | 168 | prune_packed_objects(show_only); |
Brandon Casey | db87e39 | 2008-09-22 18:34:26 -0500 | [diff] [blame] | 169 | remove_temporary_files(get_object_directory()); |
| 170 | s = xstrdup(mkpath("%s/pack", get_object_directory())); |
| 171 | remove_temporary_files(s); |
| 172 | free(s); |
Linus Torvalds | ba84a79 | 2006-07-06 10:16:22 -0700 | [diff] [blame] | 173 | return 0; |
| 174 | } |