Merge branch 'master' of git://bogomips.org/git-svn
[git] / 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
12 struct connectivity_progress {
13         struct progress *progress;
14         unsigned long count;
15 };
16
17 static void update_progress(struct connectivity_progress *cp)
18 {
19         cp->count++;
20         if ((cp->count & 1023) == 0)
21                 display_progress(cp->progress, cp->count);
22 }
23
24 static void process_blob(struct blob *blob,
25                          struct object_array *p,
26                          struct name_path *path,
27                          const char *name,
28                          struct connectivity_progress *cp)
29 {
30         struct object *obj = &blob->object;
31
32         if (!blob)
33                 die("bad blob object");
34         if (obj->flags & SEEN)
35                 return;
36         obj->flags |= SEEN;
37         update_progress(cp);
38         /* Nothing to do, really .. The blob lookup was the important part */
39 }
40
41 static void process_gitlink(const unsigned char *sha1,
42                             struct object_array *p,
43                             struct name_path *path,
44                             const char *name)
45 {
46         /* I don't think we want to recurse into this, really. */
47 }
48
49 static void process_tree(struct tree *tree,
50                          struct object_array *p,
51                          struct name_path *path,
52                          const char *name,
53                          struct connectivity_progress *cp)
54 {
55         struct object *obj = &tree->object;
56         struct tree_desc desc;
57         struct name_entry entry;
58         struct name_path me;
59
60         if (!tree)
61                 die("bad tree object");
62         if (obj->flags & SEEN)
63                 return;
64         obj->flags |= SEEN;
65         update_progress(cp);
66         if (parse_tree(tree) < 0)
67                 die("bad tree object %s", sha1_to_hex(obj->sha1));
68         add_object(obj, p, path, name);
69         me.up = path;
70         me.elem = name;
71         me.elem_len = strlen(name);
72
73         init_tree_desc(&desc, tree->buffer, tree->size);
74
75         while (tree_entry(&desc, &entry)) {
76                 if (S_ISDIR(entry.mode))
77                         process_tree(lookup_tree(entry.sha1), p, &me, entry.path, cp);
78                 else if (S_ISGITLINK(entry.mode))
79                         process_gitlink(entry.sha1, p, &me, entry.path);
80                 else
81                         process_blob(lookup_blob(entry.sha1), p, &me, entry.path, cp);
82         }
83         free(tree->buffer);
84         tree->buffer = NULL;
85 }
86
87 static void process_tag(struct tag *tag, struct object_array *p,
88                         const char *name, struct connectivity_progress *cp)
89 {
90         struct object *obj = &tag->object;
91
92         if (obj->flags & SEEN)
93                 return;
94         obj->flags |= SEEN;
95         update_progress(cp);
96
97         if (parse_tag(tag) < 0)
98                 die("bad tag object %s", sha1_to_hex(obj->sha1));
99         if (tag->tagged)
100                 add_object(tag->tagged, p, NULL, name);
101 }
102
103 static void walk_commit_list(struct rev_info *revs,
104                              struct connectivity_progress *cp)
105 {
106         int i;
107         struct commit *commit;
108         struct object_array objects = OBJECT_ARRAY_INIT;
109
110         /* Walk all commits, process their trees */
111         while ((commit = get_revision(revs)) != NULL) {
112                 process_tree(commit->tree, &objects, NULL, "", cp);
113                 update_progress(cp);
114         }
115
116         /* Then walk all the pending objects, recursively processing them too */
117         for (i = 0; i < revs->pending.nr; i++) {
118                 struct object_array_entry *pending = revs->pending.objects + i;
119                 struct object *obj = pending->item;
120                 const char *name = pending->name;
121                 if (obj->type == OBJ_TAG) {
122                         process_tag((struct tag *) obj, &objects, name, cp);
123                         continue;
124                 }
125                 if (obj->type == OBJ_TREE) {
126                         process_tree((struct tree *)obj, &objects, NULL, name, cp);
127                         continue;
128                 }
129                 if (obj->type == OBJ_BLOB) {
130                         process_blob((struct blob *)obj, &objects, NULL, name, cp);
131                         continue;
132                 }
133                 die("unknown pending object %s (%s)", sha1_to_hex(obj->sha1), name);
134         }
135 }
136
137 static int add_one_reflog_ent(unsigned char *osha1, unsigned char *nsha1,
138                 const char *email, unsigned long timestamp, int tz,
139                 const char *message, void *cb_data)
140 {
141         struct object *object;
142         struct rev_info *revs = (struct rev_info *)cb_data;
143
144         object = parse_object(osha1);
145         if (object)
146                 add_pending_object(revs, object, "");
147         object = parse_object(nsha1);
148         if (object)
149                 add_pending_object(revs, object, "");
150         return 0;
151 }
152
153 static int add_one_ref(const char *path, const unsigned char *sha1, int flag, void *cb_data)
154 {
155         struct object *object = parse_object(sha1);
156         struct rev_info *revs = (struct rev_info *)cb_data;
157
158         if (!object)
159                 die("bad object ref: %s:%s", path, sha1_to_hex(sha1));
160         add_pending_object(revs, object, "");
161
162         return 0;
163 }
164
165 static int add_one_reflog(const char *path, const unsigned char *sha1, int flag, void *cb_data)
166 {
167         for_each_reflog_ent(path, add_one_reflog_ent, cb_data);
168         return 0;
169 }
170
171 static void add_one_tree(const unsigned char *sha1, struct rev_info *revs)
172 {
173         struct tree *tree = lookup_tree(sha1);
174         if (tree)
175                 add_pending_object(revs, &tree->object, "");
176 }
177
178 static void add_cache_tree(struct cache_tree *it, struct rev_info *revs)
179 {
180         int i;
181
182         if (it->entry_count >= 0)
183                 add_one_tree(it->sha1, revs);
184         for (i = 0; i < it->subtree_nr; i++)
185                 add_cache_tree(it->down[i]->cache_tree, revs);
186 }
187
188 static void add_cache_refs(struct rev_info *revs)
189 {
190         int i;
191
192         read_cache();
193         for (i = 0; i < active_nr; i++) {
194                 /*
195                  * The index can contain blobs and GITLINKs, GITLINKs are hashes
196                  * that don't actually point to objects in the repository, it's
197                  * almost guaranteed that they are NOT blobs, so we don't call
198                  * lookup_blob() on them, to avoid populating the hash table
199                  * with invalid information
200                  */
201                 if (S_ISGITLINK(active_cache[i]->ce_mode))
202                         continue;
203
204                 lookup_blob(active_cache[i]->sha1);
205                 /*
206                  * We could add the blobs to the pending list, but quite
207                  * frankly, we don't care. Once we've looked them up, and
208                  * added them as objects, we've really done everything
209                  * there is to do for a blob
210                  */
211         }
212         if (active_cache_tree)
213                 add_cache_tree(active_cache_tree, revs);
214 }
215
216 void mark_reachable_objects(struct rev_info *revs, int mark_reflog,
217                             struct progress *progress)
218 {
219         struct connectivity_progress cp;
220
221         /*
222          * Set up revision parsing, and mark us as being interested
223          * in all object types, not just commits.
224          */
225         revs->tag_objects = 1;
226         revs->blob_objects = 1;
227         revs->tree_objects = 1;
228
229         /* Add all refs from the index file */
230         add_cache_refs(revs);
231
232         /* Add all external refs */
233         for_each_ref(add_one_ref, revs);
234
235         /* Add all reflog info */
236         if (mark_reflog)
237                 for_each_reflog(add_one_reflog, revs);
238
239         cp.progress = progress;
240         cp.count = 0;
241
242         /*
243          * Set up the revision walk - this will move all commits
244          * from the pending list to the commit walking list.
245          */
246         if (prepare_revision_walk(revs))
247                 die("revision walk setup failed");
248         walk_commit_list(revs, &cp);
249         display_progress(cp.progress, cp.count);
250 }