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