Junio C Hamano | 215a7ad | 2005-09-08 00:26:23 | [diff] [blame] | 1 | #include "fetch.h" |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 2 | |
| 3 | #include "cache.h" |
| 4 | #include "commit.h" |
| 5 | #include "tree.h" |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 6 | #include "tag.h" |
| 7 | #include "blob.h" |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 8 | #include "refs.h" |
| 9 | |
| 10 | const char *write_ref = NULL; |
| 11 | |
| 12 | const unsigned char *current_ref = NULL; |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 13 | |
| 14 | int get_tree = 0; |
| 15 | int get_history = 0; |
| 16 | int get_all = 0; |
Junio C Hamano | e78d977 | 2005-05-06 08:37:21 | [diff] [blame] | 17 | int get_verbosely = 0; |
Junio C Hamano | b2d62f1 | 2005-05-04 08:26:24 | [diff] [blame] | 18 | static unsigned char current_commit_sha1[20]; |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 19 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 20 | void pull_say(const char *fmt, const char *hex) |
| 21 | { |
Junio C Hamano | e78d977 | 2005-05-06 08:37:21 | [diff] [blame] | 22 | if (get_verbosely) |
| 23 | fprintf(stderr, fmt, hex); |
| 24 | } |
| 25 | |
Junio C Hamano | b2d62f1 | 2005-05-04 08:26:24 | [diff] [blame] | 26 | static void report_missing(const char *what, const unsigned char *missing) |
Junio C Hamano | ee4f439 | 2005-05-02 04:07:40 | [diff] [blame] | 27 | { |
Junio C Hamano | b2d62f1 | 2005-05-04 08:26:24 | [diff] [blame] | 28 | char missing_hex[41]; |
| 29 | |
| 30 | strcpy(missing_hex, sha1_to_hex(missing));; |
| 31 | fprintf(stderr, |
| 32 | "Cannot obtain needed %s %s\nwhile processing commit %s.\n", |
| 33 | what, missing_hex, sha1_to_hex(current_commit_sha1)); |
| 34 | } |
| 35 | |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 36 | static int process(struct object *obj); |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 37 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 38 | static int process_tree(struct tree *tree) |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 39 | { |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 40 | struct tree_entry_list *entry; |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 41 | |
| 42 | if (parse_tree(tree)) |
| 43 | return -1; |
| 44 | |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 45 | entry = tree->entries; |
| 46 | tree->entries = NULL; |
| 47 | while (entry) { |
| 48 | struct tree_entry_list *next = entry->next; |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 49 | if (process(entry->item.any)) |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 50 | return -1; |
Sergey Vlasov | d35bbe0 | 2005-09-23 12:28:18 | [diff] [blame] | 51 | free(entry->name); |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 52 | free(entry); |
| 53 | entry = next; |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 54 | } |
| 55 | return 0; |
| 56 | } |
| 57 | |
Sergey Vlasov | 24451c3 | 2005-09-21 16:34:24 | [diff] [blame] | 58 | #define COMPLETE (1U << 0) |
| 59 | #define SEEN (1U << 1) |
| 60 | #define TO_SCAN (1U << 2) |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 61 | |
Junio C Hamano | d0ac30f | 2005-09-16 21:30:29 | [diff] [blame] | 62 | static struct commit_list *complete = NULL; |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 63 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 64 | static int process_commit(struct commit *commit) |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 65 | { |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 66 | if (parse_commit(commit)) |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 67 | return -1; |
| 68 | |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 69 | while (complete && complete->item->date >= commit->date) { |
Junio C Hamano | d0ac30f | 2005-09-16 21:30:29 | [diff] [blame] | 70 | pop_most_recent_commit(&complete, COMPLETE); |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 71 | } |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 72 | |
Junio C Hamano | d0ac30f | 2005-09-16 21:30:29 | [diff] [blame] | 73 | if (commit->object.flags & COMPLETE) |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 74 | return 0; |
| 75 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 76 | memcpy(current_commit_sha1, commit->object.sha1, 20); |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 77 | |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 78 | pull_say("walk %s\n", sha1_to_hex(commit->object.sha1)); |
| 79 | |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 80 | if (get_tree) { |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 81 | if (process(&commit->tree->object)) |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 82 | return -1; |
| 83 | if (!get_all) |
| 84 | get_tree = 0; |
| 85 | } |
| 86 | if (get_history) { |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 87 | struct commit_list *parents = commit->parents; |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 88 | for (; parents; parents = parents->next) { |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 89 | if (process(&parents->item->object)) |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 90 | return -1; |
| 91 | } |
| 92 | } |
| 93 | return 0; |
| 94 | } |
| 95 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 96 | static int process_tag(struct tag *tag) |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 97 | { |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 98 | if (parse_tag(tag)) |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 99 | return -1; |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 100 | return process(tag->tagged); |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 101 | } |
| 102 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 103 | static struct object_list *process_queue = NULL; |
| 104 | static struct object_list **process_queue_end = &process_queue; |
| 105 | |
Daniel Barkalow | f88fcf8 | 2005-08-11 23:38:09 | [diff] [blame] | 106 | static int process_object(struct object *obj) |
| 107 | { |
| 108 | if (obj->type == commit_type) { |
| 109 | if (process_commit((struct commit *)obj)) |
| 110 | return -1; |
| 111 | return 0; |
| 112 | } |
| 113 | if (obj->type == tree_type) { |
| 114 | if (process_tree((struct tree *)obj)) |
| 115 | return -1; |
| 116 | return 0; |
| 117 | } |
| 118 | if (obj->type == blob_type) { |
| 119 | return 0; |
| 120 | } |
| 121 | if (obj->type == tag_type) { |
| 122 | if (process_tag((struct tag *)obj)) |
| 123 | return -1; |
| 124 | return 0; |
| 125 | } |
| 126 | return error("Unable to determine requirements " |
| 127 | "of type %s for %s", |
| 128 | obj->type, sha1_to_hex(obj->sha1)); |
| 129 | } |
| 130 | |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 131 | static int process(struct object *obj) |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 132 | { |
Sergey Vlasov | a82d07e | 2005-09-21 16:33:59 | [diff] [blame] | 133 | if (obj->flags & SEEN) |
| 134 | return 0; |
| 135 | obj->flags |= SEEN; |
| 136 | |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 137 | if (has_sha1_file(obj->sha1)) { |
Daniel Barkalow | f88fcf8 | 2005-08-11 23:38:09 | [diff] [blame] | 138 | /* We already have it, so we should scan it now. */ |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 139 | obj->flags |= TO_SCAN; |
Sergey Vlasov | 7b64d06 | 2005-09-21 16:34:14 | [diff] [blame] | 140 | } else { |
| 141 | if (obj->flags & COMPLETE) |
| 142 | return 0; |
| 143 | prefetch(obj->sha1); |
Daniel Barkalow | f88fcf8 | 2005-08-11 23:38:09 | [diff] [blame] | 144 | } |
Sergey Vlasov | 7b64d06 | 2005-09-21 16:34:14 | [diff] [blame] | 145 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 146 | object_list_insert(obj, process_queue_end); |
| 147 | process_queue_end = &(*process_queue_end)->next; |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 148 | return 0; |
| 149 | } |
| 150 | |
| 151 | static int loop(void) |
| 152 | { |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 153 | struct object_list *elem; |
| 154 | |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 155 | while (process_queue) { |
| 156 | struct object *obj = process_queue->item; |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 157 | elem = process_queue; |
| 158 | process_queue = elem->next; |
| 159 | free(elem); |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 160 | if (!process_queue) |
| 161 | process_queue_end = &process_queue; |
| 162 | |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 163 | /* If we are not scanning this object, we placed it in |
| 164 | * the queue because we needed to fetch it first. |
| 165 | */ |
| 166 | if (! (obj->flags & TO_SCAN)) { |
Junio C Hamano | 029f6de | 2005-09-18 21:11:53 | [diff] [blame] | 167 | if (!has_sha1_file(obj->sha1) && fetch(obj->sha1)) { |
Junio C Hamano | 85d106c | 2005-09-18 08:01:07 | [diff] [blame] | 168 | report_missing(obj->type |
| 169 | ? obj->type |
| 170 | : "object", obj->sha1); |
| 171 | return -1; |
| 172 | } |
| 173 | } |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 174 | if (!obj->type) |
| 175 | parse_object(obj->sha1); |
Daniel Barkalow | f88fcf8 | 2005-08-11 23:38:09 | [diff] [blame] | 176 | if (process_object(obj)) |
| 177 | return -1; |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 178 | } |
| 179 | return 0; |
Daniel Barkalow | 3173bd4 | 2005-06-22 00:35:53 | [diff] [blame] | 180 | } |
| 181 | |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 182 | static int interpret_target(char *target, unsigned char *sha1) |
| 183 | { |
| 184 | if (!get_sha1_hex(target, sha1)) |
| 185 | return 0; |
| 186 | if (!check_ref_format(target)) { |
| 187 | if (!fetch_ref(target, sha1)) { |
| 188 | return 0; |
| 189 | } |
| 190 | } |
| 191 | return -1; |
| 192 | } |
| 193 | |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 194 | static int mark_complete(const char *path, const unsigned char *sha1) |
| 195 | { |
Junio C Hamano | d0ac30f | 2005-09-16 21:30:29 | [diff] [blame] | 196 | struct commit *commit = lookup_commit_reference_gently(sha1, 1); |
| 197 | if (commit) { |
| 198 | commit->object.flags |= COMPLETE; |
| 199 | insert_by_date(commit, &complete); |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 200 | } |
| 201 | return 0; |
| 202 | } |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 203 | |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 204 | int pull(char *target) |
| 205 | { |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 206 | unsigned char sha1[20]; |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 207 | int fd = -1; |
| 208 | |
Junio C Hamano | 98533b9 | 2005-09-15 22:06:39 | [diff] [blame] | 209 | save_commit_buffer = 0; |
Sergey Vlasov | a95cb6f | 2005-09-23 12:28:13 | [diff] [blame] | 210 | track_object_refs = 0; |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 211 | if (write_ref && current_ref) { |
| 212 | fd = lock_ref_sha1(write_ref, current_ref); |
| 213 | if (fd < 0) |
| 214 | return -1; |
| 215 | } |
| 216 | |
Daniel Barkalow | 22c6e1d | 2005-09-15 01:31:42 | [diff] [blame] | 217 | for_each_ref(mark_complete); |
| 218 | |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 219 | if (interpret_target(target, sha1)) |
| 220 | return error("Could not interpret %s as something to pull", |
| 221 | target); |
Sergey Vlasov | 80077f0 | 2005-09-21 16:33:54 | [diff] [blame] | 222 | if (process(lookup_unknown_object(sha1))) |
barkalow@iabervon.org | 1e8be59 | 2005-08-02 23:46:10 | [diff] [blame] | 223 | return -1; |
| 224 | if (loop()) |
Daniel Barkalow | cd541a6 | 2005-06-06 20:38:26 | [diff] [blame] | 225 | return -1; |
| 226 | |
| 227 | if (write_ref) { |
| 228 | if (current_ref) { |
| 229 | write_ref_sha1(write_ref, fd, sha1); |
| 230 | } else { |
| 231 | write_ref_sha1_unlocked(write_ref, sha1); |
| 232 | } |
| 233 | } |
| 234 | return 0; |
Daniel Barkalow | 4250a5e | 2005-04-30 23:53:56 | [diff] [blame] | 235 | } |