reachable.c

   1 #include "cache.h"
   2 #include "refs.h"
   3 #include "tag.h"
   4 #include "commit.h"
   5 #include "blob.h"
   6 #include "diff.h"
   7 #include "revision.h"
   8 #include "reachable.h"
   9 #include "cache-tree.h"
  10 #include "progress.h"
  11 #include "list-objects.h"
  12 #include "packfile.h"
  13
  14 struct connectivity_progress {
  15         struct progress *progress;
  16         unsigned long count;
  17 };
  18
  19 static void update_progress(struct connectivity_progress *cp)
  20 {
  21         cp->count++;
  22         if ((cp->count & 1023) == 0)
  23                 display_progress(cp->progress, cp->count);
  24 }
  25
  26 static int add_one_ref(const char *path, const struct object_id *oid,
  27                        int flag, void *cb_data)
  28 {
  29         struct rev_info *revs = (struct rev_info *)cb_data;
  30         struct object *object;
  31
  32         if ((flag & REF_ISSYMREF) && (flag & REF_ISBROKEN)) {
  33                 warning("symbolic ref is dangling: %s", path);
  34                 return 0;
  35         }
  36
  37         object = parse_object_or_die(oid, path);
  38         add_pending_object(revs, object, "");
  39
  40         return 0;
  41 }
  42
  43 /*
  44  * The traversal will have already marked us as SEEN, so we
  45  * only need to handle any progress reporting here.
  46  */
  47 static void mark_object(struct object *obj, const char *name, void *data)
  48 {
  49         update_progress(data);
  50 }
  51
  52 static void mark_commit(struct commit *c, void *data)
  53 {
  54         mark_object(&c->object, NULL, data);
  55 }
  56
  57 struct recent_data {
  58         struct rev_info *revs;
  59         timestamp_t timestamp;
  60 };
  61
  62 static void add_recent_object(const struct object_id *oid,
  63                               timestamp_t mtime,
  64                               struct recent_data *data)
  65 {
  66         struct object *obj;
  67         enum object_type type;
  68
  69         if (mtime <= data->timestamp)
  70                 return;
  71
  72         /*
  73          * We do not want to call parse_object here, because
  74          * inflating blobs and trees could be very expensive.
  75          * However, we do need to know the correct type for
  76          * later processing, and the revision machinery expects
  77          * commits and tags to have been parsed.
  78          */
  79         type = sha1_object_info(oid->hash, NULL);
  80         if (type < 0)
  81                 die("unable to get object info for %s", oid_to_hex(oid));
  82
  83         switch (type) {
  84         case OBJ_TAG:
  85         case OBJ_COMMIT:
  86                 obj = parse_object_or_die(oid, NULL);
  87                 break;
  88         case OBJ_TREE:
  89                 obj = (struct object *)lookup_tree(oid);
  90                 break;
  91         case OBJ_BLOB:
  92                 obj = (struct object *)lookup_blob(oid);
  93                 break;
  94         default:
  95                 die("unknown object type for %s: %s",
  96                     oid_to_hex(oid), typename(type));
  97         }
  98
  99         if (!obj)
 100                 die("unable to lookup %s", oid_to_hex(oid));
 101
 102         add_pending_object(data->revs, obj, "");
 103 }
 104
 105 static int add_recent_loose(const struct object_id *oid,
 106                             const char *path, void *data)
 107 {
 108         struct stat st;
 109         struct object *obj = lookup_object(oid->hash);
 110
 111         if (obj && obj->flags & SEEN)
 112                 return 0;
 113
 114         if (stat(path, &st) < 0) {
 115                 /*
 116                  * It's OK if an object went away during our iteration; this
 117                  * could be due to a simultaneous repack. But anything else
 118                  * we should abort, since we might then fail to mark objects
 119                  * which should not be pruned.
 120                  */
 121                 if (errno == ENOENT)
 122                         return 0;
 123                 return error_errno("unable to stat %s", oid_to_hex(oid));
 124         }
 125
 126         add_recent_object(oid, st.st_mtime, data);
 127         return 0;
 128 }
 129
 130 static int add_recent_packed(const struct object_id *oid,
 131                              struct packed_git *p, uint32_t pos,
 132                              void *data)
 133 {
 134         struct object *obj = lookup_object(oid->hash);
 135
 136         if (obj && obj->flags & SEEN)
 137                 return 0;
 138         add_recent_object(oid, p->mtime, data);
 139         return 0;
 140 }
 141
 142 int add_unseen_recent_objects_to_traversal(struct rev_info *revs,
 143                                            timestamp_t timestamp)
 144 {
 145         struct recent_data data;
 146         int r;
 147
 148         data.revs = revs;
 149         data.timestamp = timestamp;
 150
 151         r = for_each_loose_object(add_recent_loose, &data,
 152                                   FOR_EACH_OBJECT_LOCAL_ONLY);
 153         if (r)
 154                 return r;
 155         return for_each_packed_object(add_recent_packed, &data,
 156                                       FOR_EACH_OBJECT_LOCAL_ONLY);
 157 }
 158
 159 void mark_reachable_objects(struct rev_info *revs, int mark_reflog,
 160                             timestamp_t mark_recent, struct progress *progress)
 161 {
 162         struct connectivity_progress cp;
 163
 164         /*
 165          * Set up revision parsing, and mark us as being interested
 166          * in all object types, not just commits.
 167          */
 168         revs->tag_objects = 1;
 169         revs->blob_objects = 1;
 170         revs->tree_objects = 1;
 171
 172         /* Add all refs from the index file */
 173         add_index_objects_to_pending(revs, 0);
 174
 175         /* Add all external refs */
 176         for_each_ref(add_one_ref, revs);
 177
 178         /* detached HEAD is not included in the list above */
 179         head_ref(add_one_ref, revs);
 180
 181         /* Add all reflog info */
 182         if (mark_reflog)
 183                 add_reflogs_to_pending(revs, 0);
 184
 185         cp.progress = progress;
 186         cp.count = 0;
 187
 188         /*
 189          * Set up the revision walk - this will move all commits
 190          * from the pending list to the commit walking list.
 191          */
 192         if (prepare_revision_walk(revs))
 193                 die("revision walk setup failed");
 194         traverse_commit_list(revs, mark_commit, mark_object, &cp);
 195
 196         if (mark_recent) {
 197                 revs->ignore_missing_links = 1;
 198                 if (add_unseen_recent_objects_to_traversal(revs, mark_recent))
 199                         die("unable to mark recent objects");
 200                 if (prepare_revision_walk(revs))
 201                         die("revision walk setup failed");
 202                 traverse_commit_list(revs, mark_commit, mark_object, &cp);
 203         }
 204
 205         display_progress(cp.progress, cp.count);
 206 }