commit_ref_update(): new function, extracted from write_ref_sha1()
[git/raj.git] / builtin / fsck.c
blob0c757862e8cd414c087b27a608281b66c8608a4b
1 #include "builtin.h"
2 #include "cache.h"
3 #include "commit.h"
4 #include "tree.h"
5 #include "blob.h"
6 #include "tag.h"
7 #include "refs.h"
8 #include "pack.h"
9 #include "cache-tree.h"
10 #include "tree-walk.h"
11 #include "fsck.h"
12 #include "parse-options.h"
13 #include "dir.h"
14 #include "progress.h"
15 #include "streaming.h"
17 #define REACHABLE 0x0001
18 #define SEEN 0x0002
19 #define HAS_OBJ 0x0004
21 static int show_root;
22 static int show_tags;
23 static int show_unreachable;
24 static int include_reflogs = 1;
25 static int check_full = 1;
26 static int check_strict;
27 static int keep_cache_objects;
28 static unsigned char head_sha1[20];
29 static const char *head_points_at;
30 static int errors_found;
31 static int write_lost_and_found;
32 static int verbose;
33 static int show_progress = -1;
34 static int show_dangling = 1;
35 #define ERROR_OBJECT 01
36 #define ERROR_REACHABLE 02
37 #define ERROR_PACK 04
39 #ifdef NO_D_INO_IN_DIRENT
40 #define SORT_DIRENT 0
41 #define DIRENT_SORT_HINT(de) 0
42 #else
43 #define SORT_DIRENT 1
44 #define DIRENT_SORT_HINT(de) ((de)->d_ino)
45 #endif
47 static void objreport(struct object *obj, const char *severity,
48 const char *err, va_list params)
50 fprintf(stderr, "%s in %s %s: ",
51 severity, typename(obj->type), sha1_to_hex(obj->sha1));
52 vfprintf(stderr, err, params);
53 fputs("\n", stderr);
56 __attribute__((format (printf, 2, 3)))
57 static int objerror(struct object *obj, const char *err, ...)
59 va_list params;
60 va_start(params, err);
61 errors_found |= ERROR_OBJECT;
62 objreport(obj, "error", err, params);
63 va_end(params);
64 return -1;
67 __attribute__((format (printf, 3, 4)))
68 static int fsck_error_func(struct object *obj, int type, const char *err, ...)
70 va_list params;
71 va_start(params, err);
72 objreport(obj, (type == FSCK_WARN) ? "warning" : "error", err, params);
73 va_end(params);
74 return (type == FSCK_WARN) ? 0 : 1;
77 static struct object_array pending;
79 static int mark_object(struct object *obj, int type, void *data)
81 struct object *parent = data;
84 * The only case data is NULL or type is OBJ_ANY is when
85 * mark_object_reachable() calls us. All the callers of
86 * that function has non-NULL obj hence ...
88 if (!obj) {
89 /* ... these references to parent->fld are safe here */
90 printf("broken link from %7s %s\n",
91 typename(parent->type), sha1_to_hex(parent->sha1));
92 printf("broken link from %7s %s\n",
93 (type == OBJ_ANY ? "unknown" : typename(type)), "unknown");
94 errors_found |= ERROR_REACHABLE;
95 return 1;
98 if (type != OBJ_ANY && obj->type != type)
99 /* ... and the reference to parent is safe here */
100 objerror(parent, "wrong object type in link");
102 if (obj->flags & REACHABLE)
103 return 0;
104 obj->flags |= REACHABLE;
105 if (!(obj->flags & HAS_OBJ)) {
106 if (parent && !has_sha1_file(obj->sha1)) {
107 printf("broken link from %7s %s\n",
108 typename(parent->type), sha1_to_hex(parent->sha1));
109 printf(" to %7s %s\n",
110 typename(obj->type), sha1_to_hex(obj->sha1));
111 errors_found |= ERROR_REACHABLE;
113 return 1;
116 add_object_array(obj, NULL, &pending);
117 return 0;
120 static void mark_object_reachable(struct object *obj)
122 mark_object(obj, OBJ_ANY, NULL);
125 static int traverse_one_object(struct object *obj)
127 int result;
128 struct tree *tree = NULL;
130 if (obj->type == OBJ_TREE) {
131 tree = (struct tree *)obj;
132 if (parse_tree(tree) < 0)
133 return 1; /* error already displayed */
135 result = fsck_walk(obj, mark_object, obj);
136 if (tree)
137 free_tree_buffer(tree);
138 return result;
141 static int traverse_reachable(void)
143 struct progress *progress = NULL;
144 unsigned int nr = 0;
145 int result = 0;
146 if (show_progress)
147 progress = start_progress_delay(_("Checking connectivity"), 0, 0, 2);
148 while (pending.nr) {
149 struct object_array_entry *entry;
150 struct object *obj;
152 entry = pending.objects + --pending.nr;
153 obj = entry->item;
154 result |= traverse_one_object(obj);
155 display_progress(progress, ++nr);
157 stop_progress(&progress);
158 return !!result;
161 static int mark_used(struct object *obj, int type, void *data)
163 if (!obj)
164 return 1;
165 obj->used = 1;
166 return 0;
170 * Check a single reachable object
172 static void check_reachable_object(struct object *obj)
175 * We obviously want the object to be parsed,
176 * except if it was in a pack-file and we didn't
177 * do a full fsck
179 if (!(obj->flags & HAS_OBJ)) {
180 if (has_sha1_pack(obj->sha1))
181 return; /* it is in pack - forget about it */
182 printf("missing %s %s\n", typename(obj->type), sha1_to_hex(obj->sha1));
183 errors_found |= ERROR_REACHABLE;
184 return;
189 * Check a single unreachable object
191 static void check_unreachable_object(struct object *obj)
194 * Missing unreachable object? Ignore it. It's not like
195 * we miss it (since it can't be reached), nor do we want
196 * to complain about it being unreachable (since it does
197 * not exist).
199 if (!obj->parsed)
200 return;
203 * Unreachable object that exists? Show it if asked to,
204 * since this is something that is prunable.
206 if (show_unreachable) {
207 printf("unreachable %s %s\n", typename(obj->type), sha1_to_hex(obj->sha1));
208 return;
212 * "!used" means that nothing at all points to it, including
213 * other unreachable objects. In other words, it's the "tip"
214 * of some set of unreachable objects, usually a commit that
215 * got dropped.
217 * Such starting points are more interesting than some random
218 * set of unreachable objects, so we show them even if the user
219 * hasn't asked for _all_ unreachable objects. If you have
220 * deleted a branch by mistake, this is a prime candidate to
221 * start looking at, for example.
223 if (!obj->used) {
224 if (show_dangling)
225 printf("dangling %s %s\n", typename(obj->type),
226 sha1_to_hex(obj->sha1));
227 if (write_lost_and_found) {
228 char *filename = git_path("lost-found/%s/%s",
229 obj->type == OBJ_COMMIT ? "commit" : "other",
230 sha1_to_hex(obj->sha1));
231 FILE *f;
233 if (safe_create_leading_directories(filename)) {
234 error("Could not create lost-found");
235 return;
237 if (!(f = fopen(filename, "w")))
238 die_errno("Could not open '%s'", filename);
239 if (obj->type == OBJ_BLOB) {
240 if (stream_blob_to_fd(fileno(f), obj->sha1, NULL, 1))
241 die_errno("Could not write '%s'", filename);
242 } else
243 fprintf(f, "%s\n", sha1_to_hex(obj->sha1));
244 if (fclose(f))
245 die_errno("Could not finish '%s'",
246 filename);
248 return;
252 * Otherwise? It's there, it's unreachable, and some other unreachable
253 * object points to it. Ignore it - it's not interesting, and we showed
254 * all the interesting cases above.
258 static void check_object(struct object *obj)
260 if (verbose)
261 fprintf(stderr, "Checking %s\n", sha1_to_hex(obj->sha1));
263 if (obj->flags & REACHABLE)
264 check_reachable_object(obj);
265 else
266 check_unreachable_object(obj);
269 static void check_connectivity(void)
271 int i, max;
273 /* Traverse the pending reachable objects */
274 traverse_reachable();
276 /* Look up all the requirements, warn about missing objects.. */
277 max = get_max_object_index();
278 if (verbose)
279 fprintf(stderr, "Checking connectivity (%d objects)\n", max);
281 for (i = 0; i < max; i++) {
282 struct object *obj = get_indexed_object(i);
284 if (obj)
285 check_object(obj);
289 static int fsck_obj(struct object *obj)
291 if (obj->flags & SEEN)
292 return 0;
293 obj->flags |= SEEN;
295 if (verbose)
296 fprintf(stderr, "Checking %s %s\n",
297 typename(obj->type), sha1_to_hex(obj->sha1));
299 if (fsck_walk(obj, mark_used, NULL))
300 objerror(obj, "broken links");
301 if (fsck_object(obj, NULL, 0, check_strict, fsck_error_func))
302 return -1;
304 if (obj->type == OBJ_TREE) {
305 struct tree *item = (struct tree *) obj;
307 free_tree_buffer(item);
310 if (obj->type == OBJ_COMMIT) {
311 struct commit *commit = (struct commit *) obj;
313 free_commit_buffer(commit);
315 if (!commit->parents && show_root)
316 printf("root %s\n", sha1_to_hex(commit->object.sha1));
319 if (obj->type == OBJ_TAG) {
320 struct tag *tag = (struct tag *) obj;
322 if (show_tags && tag->tagged) {
323 printf("tagged %s %s", typename(tag->tagged->type), sha1_to_hex(tag->tagged->sha1));
324 printf(" (%s) in %s\n", tag->tag, sha1_to_hex(tag->object.sha1));
328 return 0;
331 static int fsck_sha1(const unsigned char *sha1)
333 struct object *obj = parse_object(sha1);
334 if (!obj) {
335 errors_found |= ERROR_OBJECT;
336 return error("%s: object corrupt or missing",
337 sha1_to_hex(sha1));
339 obj->flags |= HAS_OBJ;
340 return fsck_obj(obj);
343 static int fsck_obj_buffer(const unsigned char *sha1, enum object_type type,
344 unsigned long size, void *buffer, int *eaten)
346 struct object *obj;
347 obj = parse_object_buffer(sha1, type, size, buffer, eaten);
348 if (!obj) {
349 errors_found |= ERROR_OBJECT;
350 return error("%s: object corrupt or missing", sha1_to_hex(sha1));
352 obj->flags = HAS_OBJ;
353 return fsck_obj(obj);
357 * This is the sorting chunk size: make it reasonably
358 * big so that we can sort well..
360 #define MAX_SHA1_ENTRIES (1024)
362 struct sha1_entry {
363 unsigned long ino;
364 unsigned char sha1[20];
367 static struct {
368 unsigned long nr;
369 struct sha1_entry *entry[MAX_SHA1_ENTRIES];
370 } sha1_list;
372 static int ino_compare(const void *_a, const void *_b)
374 const struct sha1_entry *a = _a, *b = _b;
375 unsigned long ino1 = a->ino, ino2 = b->ino;
376 return ino1 < ino2 ? -1 : ino1 > ino2 ? 1 : 0;
379 static void fsck_sha1_list(void)
381 int i, nr = sha1_list.nr;
383 if (SORT_DIRENT)
384 qsort(sha1_list.entry, nr,
385 sizeof(struct sha1_entry *), ino_compare);
386 for (i = 0; i < nr; i++) {
387 struct sha1_entry *entry = sha1_list.entry[i];
388 unsigned char *sha1 = entry->sha1;
390 sha1_list.entry[i] = NULL;
391 if (fsck_sha1(sha1))
392 errors_found |= ERROR_OBJECT;
393 free(entry);
395 sha1_list.nr = 0;
398 static void add_sha1_list(unsigned char *sha1, unsigned long ino)
400 struct sha1_entry *entry = xmalloc(sizeof(*entry));
401 int nr;
403 entry->ino = ino;
404 hashcpy(entry->sha1, sha1);
405 nr = sha1_list.nr;
406 if (nr == MAX_SHA1_ENTRIES) {
407 fsck_sha1_list();
408 nr = 0;
410 sha1_list.entry[nr] = entry;
411 sha1_list.nr = ++nr;
414 static inline int is_loose_object_file(struct dirent *de,
415 char *name, unsigned char *sha1)
417 if (strlen(de->d_name) != 38)
418 return 0;
419 memcpy(name + 2, de->d_name, 39);
420 return !get_sha1_hex(name, sha1);
423 static void fsck_dir(int i, char *path)
425 DIR *dir = opendir(path);
426 struct dirent *de;
427 char name[100];
429 if (!dir)
430 return;
432 if (verbose)
433 fprintf(stderr, "Checking directory %s\n", path);
435 sprintf(name, "%02x", i);
436 while ((de = readdir(dir)) != NULL) {
437 unsigned char sha1[20];
439 if (is_dot_or_dotdot(de->d_name))
440 continue;
441 if (is_loose_object_file(de, name, sha1)) {
442 add_sha1_list(sha1, DIRENT_SORT_HINT(de));
443 continue;
445 if (starts_with(de->d_name, "tmp_obj_"))
446 continue;
447 fprintf(stderr, "bad sha1 file: %s/%s\n", path, de->d_name);
449 closedir(dir);
452 static int default_refs;
454 static int fsck_handle_reflog_ent(unsigned char *osha1, unsigned char *nsha1,
455 const char *email, unsigned long timestamp, int tz,
456 const char *message, void *cb_data)
458 struct object *obj;
460 if (verbose)
461 fprintf(stderr, "Checking reflog %s->%s\n",
462 sha1_to_hex(osha1), sha1_to_hex(nsha1));
464 if (!is_null_sha1(osha1)) {
465 obj = lookup_object(osha1);
466 if (obj) {
467 obj->used = 1;
468 mark_object_reachable(obj);
471 obj = lookup_object(nsha1);
472 if (obj) {
473 obj->used = 1;
474 mark_object_reachable(obj);
476 return 0;
479 static int fsck_handle_reflog(const char *logname, const unsigned char *sha1, int flag, void *cb_data)
481 for_each_reflog_ent(logname, fsck_handle_reflog_ent, NULL);
482 return 0;
485 static int fsck_handle_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
487 struct object *obj;
489 obj = parse_object(sha1);
490 if (!obj) {
491 error("%s: invalid sha1 pointer %s", refname, sha1_to_hex(sha1));
492 errors_found |= ERROR_REACHABLE;
493 /* We'll continue with the rest despite the error.. */
494 return 0;
496 if (obj->type != OBJ_COMMIT && is_branch(refname))
497 error("%s: not a commit", refname);
498 default_refs++;
499 obj->used = 1;
500 mark_object_reachable(obj);
502 return 0;
505 static void get_default_heads(void)
507 if (head_points_at && !is_null_sha1(head_sha1))
508 fsck_handle_ref("HEAD", head_sha1, 0, NULL);
509 for_each_rawref(fsck_handle_ref, NULL);
510 if (include_reflogs)
511 for_each_reflog(fsck_handle_reflog, NULL);
514 * Not having any default heads isn't really fatal, but
515 * it does mean that "--unreachable" no longer makes any
516 * sense (since in this case everything will obviously
517 * be unreachable by definition.
519 * Showing dangling objects is valid, though (as those
520 * dangling objects are likely lost heads).
522 * So we just print a warning about it, and clear the
523 * "show_unreachable" flag.
525 if (!default_refs) {
526 fprintf(stderr, "notice: No default references\n");
527 show_unreachable = 0;
531 static void fsck_object_dir(const char *path)
533 int i;
534 struct progress *progress = NULL;
536 if (verbose)
537 fprintf(stderr, "Checking object directory\n");
539 if (show_progress)
540 progress = start_progress(_("Checking object directories"), 256);
541 for (i = 0; i < 256; i++) {
542 static char dir[4096];
543 sprintf(dir, "%s/%02x", path, i);
544 fsck_dir(i, dir);
545 display_progress(progress, i+1);
547 stop_progress(&progress);
548 fsck_sha1_list();
551 static int fsck_head_link(void)
553 int flag;
554 int null_is_error = 0;
556 if (verbose)
557 fprintf(stderr, "Checking HEAD link\n");
559 head_points_at = resolve_ref_unsafe("HEAD", 0, head_sha1, &flag);
560 if (!head_points_at)
561 return error("Invalid HEAD");
562 if (!strcmp(head_points_at, "HEAD"))
563 /* detached HEAD */
564 null_is_error = 1;
565 else if (!starts_with(head_points_at, "refs/heads/"))
566 return error("HEAD points to something strange (%s)",
567 head_points_at);
568 if (is_null_sha1(head_sha1)) {
569 if (null_is_error)
570 return error("HEAD: detached HEAD points at nothing");
571 fprintf(stderr, "notice: HEAD points to an unborn branch (%s)\n",
572 head_points_at + 11);
574 return 0;
577 static int fsck_cache_tree(struct cache_tree *it)
579 int i;
580 int err = 0;
582 if (verbose)
583 fprintf(stderr, "Checking cache tree\n");
585 if (0 <= it->entry_count) {
586 struct object *obj = parse_object(it->sha1);
587 if (!obj) {
588 error("%s: invalid sha1 pointer in cache-tree",
589 sha1_to_hex(it->sha1));
590 return 1;
592 obj->used = 1;
593 mark_object_reachable(obj);
594 if (obj->type != OBJ_TREE)
595 err |= objerror(obj, "non-tree in cache-tree");
597 for (i = 0; i < it->subtree_nr; i++)
598 err |= fsck_cache_tree(it->down[i]->cache_tree);
599 return err;
602 static char const * const fsck_usage[] = {
603 N_("git fsck [<options>] [<object>...]"),
604 NULL
607 static struct option fsck_opts[] = {
608 OPT__VERBOSE(&verbose, N_("be verbose")),
609 OPT_BOOL(0, "unreachable", &show_unreachable, N_("show unreachable objects")),
610 OPT_BOOL(0, "dangling", &show_dangling, N_("show dangling objects")),
611 OPT_BOOL(0, "tags", &show_tags, N_("report tags")),
612 OPT_BOOL(0, "root", &show_root, N_("report root nodes")),
613 OPT_BOOL(0, "cache", &keep_cache_objects, N_("make index objects head nodes")),
614 OPT_BOOL(0, "reflogs", &include_reflogs, N_("make reflogs head nodes (default)")),
615 OPT_BOOL(0, "full", &check_full, N_("also consider packs and alternate objects")),
616 OPT_BOOL(0, "strict", &check_strict, N_("enable more strict checking")),
617 OPT_BOOL(0, "lost-found", &write_lost_and_found,
618 N_("write dangling objects in .git/lost-found")),
619 OPT_BOOL(0, "progress", &show_progress, N_("show progress")),
620 OPT_END(),
623 int cmd_fsck(int argc, const char **argv, const char *prefix)
625 int i, heads;
626 struct alternate_object_database *alt;
628 errors_found = 0;
629 check_replace_refs = 0;
631 argc = parse_options(argc, argv, prefix, fsck_opts, fsck_usage, 0);
633 if (show_progress == -1)
634 show_progress = isatty(2);
635 if (verbose)
636 show_progress = 0;
638 if (write_lost_and_found) {
639 check_full = 1;
640 include_reflogs = 0;
643 fsck_head_link();
644 fsck_object_dir(get_object_directory());
646 prepare_alt_odb();
647 for (alt = alt_odb_list; alt; alt = alt->next) {
648 char namebuf[PATH_MAX];
649 int namelen = alt->name - alt->base;
650 memcpy(namebuf, alt->base, namelen);
651 namebuf[namelen - 1] = 0;
652 fsck_object_dir(namebuf);
655 if (check_full) {
656 struct packed_git *p;
657 uint32_t total = 0, count = 0;
658 struct progress *progress = NULL;
660 prepare_packed_git();
662 if (show_progress) {
663 for (p = packed_git; p; p = p->next) {
664 if (open_pack_index(p))
665 continue;
666 total += p->num_objects;
669 progress = start_progress(_("Checking objects"), total);
671 for (p = packed_git; p; p = p->next) {
672 /* verify gives error messages itself */
673 if (verify_pack(p, fsck_obj_buffer,
674 progress, count))
675 errors_found |= ERROR_PACK;
676 count += p->num_objects;
678 stop_progress(&progress);
681 heads = 0;
682 for (i = 0; i < argc; i++) {
683 const char *arg = argv[i];
684 unsigned char sha1[20];
685 if (!get_sha1(arg, sha1)) {
686 struct object *obj = lookup_object(sha1);
688 /* Error is printed by lookup_object(). */
689 if (!obj)
690 continue;
692 obj->used = 1;
693 mark_object_reachable(obj);
694 heads++;
695 continue;
697 error("invalid parameter: expected sha1, got '%s'", arg);
701 * If we've not been given any explicit head information, do the
702 * default ones from .git/refs. We also consider the index file
703 * in this case (ie this implies --cache).
705 if (!heads) {
706 get_default_heads();
707 keep_cache_objects = 1;
710 if (keep_cache_objects) {
711 read_cache();
712 for (i = 0; i < active_nr; i++) {
713 unsigned int mode;
714 struct blob *blob;
715 struct object *obj;
717 mode = active_cache[i]->ce_mode;
718 if (S_ISGITLINK(mode))
719 continue;
720 blob = lookup_blob(active_cache[i]->sha1);
721 if (!blob)
722 continue;
723 obj = &blob->object;
724 obj->used = 1;
725 mark_object_reachable(obj);
727 if (active_cache_tree)
728 fsck_cache_tree(active_cache_tree);
731 check_connectivity();
732 return errors_found;