]>
Commit | Line | Data |
---|---|---|
ed09aef0 JS |
1 | #include "cache.h" |
2 | #include "commit.h" | |
abef3a16 | 3 | #include "tag.h" |
ed09aef0 JS |
4 | |
5 | static int is_shallow = -1; | |
6035d6aa NTND |
6 | static struct stat shallow_stat; |
7 | static char *alternate_shallow_file; | |
8 | ||
9 | void set_alternate_shallow_file(const char *path) | |
10 | { | |
11 | if (is_shallow != -1) | |
12 | die("BUG: is_repository_shallow must not be called before set_alternate_shallow_file"); | |
13 | free(alternate_shallow_file); | |
14 | alternate_shallow_file = path ? xstrdup(path) : NULL; | |
15 | } | |
ed09aef0 JS |
16 | |
17 | int register_shallow(const unsigned char *sha1) | |
18 | { | |
19 | struct commit_graft *graft = | |
20 | xmalloc(sizeof(struct commit_graft)); | |
21 | struct commit *commit = lookup_commit(sha1); | |
22 | ||
23 | hashcpy(graft->sha1, sha1); | |
24 | graft->nr_parent = -1; | |
25 | if (commit && commit->object.parsed) | |
26 | commit->parents = NULL; | |
27 | return register_commit_graft(graft, 0); | |
28 | } | |
29 | ||
f43117a6 | 30 | int is_repository_shallow(void) |
ed09aef0 JS |
31 | { |
32 | FILE *fp; | |
33 | char buf[1024]; | |
6035d6aa | 34 | const char *path = alternate_shallow_file; |
ed09aef0 JS |
35 | |
36 | if (is_shallow >= 0) | |
37 | return is_shallow; | |
38 | ||
6035d6aa NTND |
39 | if (!path) |
40 | path = git_path("shallow"); | |
41 | /* | |
42 | * fetch-pack sets '--shallow-file ""' as an indicator that no | |
43 | * shallow file should be used. We could just open it and it | |
44 | * will likely fail. But let's do an explicit check instead. | |
45 | */ | |
46 | if (!*path || | |
47 | stat(path, &shallow_stat) || | |
48 | (fp = fopen(path, "r")) == NULL) { | |
ed09aef0 JS |
49 | is_shallow = 0; |
50 | return is_shallow; | |
51 | } | |
52 | is_shallow = 1; | |
53 | ||
54 | while (fgets(buf, sizeof(buf), fp)) { | |
55 | unsigned char sha1[20]; | |
56 | if (get_sha1_hex(buf, sha1)) | |
57 | die("bad shallow line: %s", buf); | |
58 | register_shallow(sha1); | |
59 | } | |
60 | fclose(fp); | |
61 | return is_shallow; | |
62 | } | |
63 | ||
f53514bc JS |
64 | struct commit_list *get_shallow_commits(struct object_array *heads, int depth, |
65 | int shallow_flag, int not_shallow_flag) | |
ed09aef0 JS |
66 | { |
67 | int i = 0, cur_depth = 0; | |
68 | struct commit_list *result = NULL; | |
3cd47459 | 69 | struct object_array stack = OBJECT_ARRAY_INIT; |
ed09aef0 JS |
70 | struct commit *commit = NULL; |
71 | ||
72 | while (commit || i < heads->nr || stack.nr) { | |
73 | struct commit_list *p; | |
74 | if (!commit) { | |
75 | if (i < heads->nr) { | |
76 | commit = (struct commit *) | |
abef3a16 | 77 | deref_tag(heads->objects[i++].item, NULL, 0); |
affeef12 | 78 | if (!commit || commit->object.type != OBJ_COMMIT) { |
ed09aef0 JS |
79 | commit = NULL; |
80 | continue; | |
81 | } | |
d64d6c9f AJ |
82 | if (!commit->util) |
83 | commit->util = xmalloc(sizeof(int)); | |
84 | *(int *)commit->util = 0; | |
ed09aef0 JS |
85 | cur_depth = 0; |
86 | } else { | |
87 | commit = (struct commit *) | |
88 | stack.objects[--stack.nr].item; | |
89 | cur_depth = *(int *)commit->util; | |
90 | } | |
91 | } | |
dec38c81 MK |
92 | if (parse_commit(commit)) |
93 | die("invalid commit"); | |
ed09aef0 | 94 | cur_depth++; |
682c7d2f NTND |
95 | if (cur_depth >= depth) { |
96 | commit_list_insert(commit, &result); | |
97 | commit->object.flags |= shallow_flag; | |
98 | commit = NULL; | |
99 | continue; | |
100 | } | |
101 | commit->object.flags |= not_shallow_flag; | |
ed09aef0 JS |
102 | for (p = commit->parents, commit = NULL; p; p = p->next) { |
103 | if (!p->item->util) { | |
104 | int *pointer = xmalloc(sizeof(int)); | |
105 | p->item->util = pointer; | |
106 | *pointer = cur_depth; | |
107 | } else { | |
108 | int *pointer = p->item->util; | |
109 | if (cur_depth >= *pointer) | |
110 | continue; | |
111 | *pointer = cur_depth; | |
112 | } | |
4b796951 MK |
113 | if (p->next) |
114 | add_object_array(&p->item->object, | |
115 | NULL, &stack); | |
116 | else { | |
117 | commit = p->item; | |
118 | cur_depth = *(int *)commit->util; | |
f53514bc | 119 | } |
ed09aef0 JS |
120 | } |
121 | } | |
122 | ||
123 | return result; | |
124 | } | |
6035d6aa NTND |
125 | |
126 | void check_shallow_file_for_update(void) | |
127 | { | |
128 | struct stat st; | |
129 | ||
130 | if (!is_shallow) | |
131 | return; | |
132 | else if (is_shallow == -1) | |
133 | die("BUG: shallow must be initialized by now"); | |
134 | ||
135 | if (stat(git_path("shallow"), &st)) | |
136 | die("shallow file was removed during fetch"); | |
137 | else if (st.st_mtime != shallow_stat.st_mtime | |
138 | #ifdef USE_NSEC | |
139 | || ST_MTIME_NSEC(st) != ST_MTIME_NSEC(shallow_stat) | |
140 | #endif | |
141 | ) | |
142 | die("shallow file was changed during fetch"); | |
143 | } |