builtin-fetch--tool.c

   1 #include "cache.h"
   2 #include "refs.h"
   3 #include "commit.h"
   4
   5 static void show_new(char *type, unsigned char *sha1_new)
   6 {
   7         fprintf(stderr, "  %s: %s\n", type,
   8                 find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
   9 }
  10
  11 static int update_ref(const char *action,
  12                       const char *refname,
  13                       unsigned char *sha1,
  14                       unsigned char *oldval)
  15 {
  16         int len;
  17         char msg[1024];
  18         char *rla = getenv("GIT_REFLOG_ACTION");
  19         static struct ref_lock *lock;
  20
  21         if (!rla)
  22                 rla = "(reflog update)";
  23         len = snprintf(msg, sizeof(msg), "%s: %s", rla, action);
  24         if (sizeof(msg) <= len)
  25                 die("insanely long action");
  26         lock = lock_any_ref_for_update(refname, oldval);
  27         if (!lock)
  28                 return 1;
  29         if (write_ref_sha1(lock, sha1, msg) < 0)
  30                 return 1;
  31         return 0;
  32 }
  33
  34 static int update_local_ref(const char *name,
  35                             const char *new_head,
  36                             const char *note,
  37                             int verbose, int force)
  38 {
  39         char type[20];
  40         unsigned char sha1_old[20], sha1_new[20];
  41         char oldh[41], newh[41];
  42         struct commit *current, *updated;
  43
  44         if (get_sha1_hex(new_head, sha1_new))
  45                 die("malformed object name %s", new_head);
  46         if (sha1_object_info(sha1_new, type, NULL))
  47                 die("object %s not found", new_head);
  48
  49         if (!*name) {
  50                 /* Not storing */
  51                 if (verbose) {
  52                         fprintf(stderr, "* fetched %s\n", note);
  53                         show_new(type, sha1_new);
  54                 }
  55                 return 0;
  56         }
  57
  58         if (get_sha1(name, sha1_old)) {
  59                 char *msg;
  60         just_store:
  61                 /* new ref */
  62                 if (!strncmp(name, "refs/tags/", 10))
  63                         msg = "storing tag";
  64                 else
  65                         msg = "storing head";
  66                 fprintf(stderr, "* %s: storing %s\n",
  67                         name, note);
  68                 show_new(type, sha1_new);
  69                 return update_ref(msg, name, sha1_new, NULL);
  70         }
  71
  72         if (!hashcmp(sha1_old, sha1_new)) {
  73                 if (verbose) {
  74                         fprintf(stderr, "* %s: same as %s\n", name, note);
  75                         show_new(type, sha1_new);
  76                 }
  77                 return 0;
  78         }
  79
  80         if (!strncmp(name, "refs/tags/", 10)) {
  81                 fprintf(stderr, "* %s: updating with %s\n", name, note);
  82                 show_new(type, sha1_new);
  83                 return update_ref("updating tag", name, sha1_new, NULL);
  84         }
  85
  86         current = lookup_commit_reference(sha1_old);
  87         updated = lookup_commit_reference(sha1_new);
  88         if (!current || !updated)
  89                 goto just_store;
  90
  91         strcpy(oldh, find_unique_abbrev(current->object.sha1, DEFAULT_ABBREV));
  92         strcpy(newh, find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
  93
  94         if (in_merge_bases(current, &updated, 1)) {
  95                 fprintf(stderr, "* %s: fast forward to %s\n",
  96                         name, note);
  97                 fprintf(stderr, "  old..new: %s..%s\n", oldh, newh);
  98                 return update_ref("fast forward", name, sha1_new, sha1_old);
  99         }
 100         if (!force) {
 101                 fprintf(stderr,
 102                         "* %s: not updating to non-fast forward %s\n",
 103                         name, note);
 104                 fprintf(stderr,
 105                         "  old...new: %s...%s\n", oldh, newh);
 106                 return 1;
 107         }
 108         fprintf(stderr,
 109                 "* %s: forcing update to non-fast forward %s\n",
 110                 name, note);
 111         fprintf(stderr, "  old...new: %s...%s\n", oldh, newh);
 112         return update_ref("forced-update", name, sha1_new, sha1_old);
 113 }
 114
 115 static int append_fetch_head(FILE *fp,
 116                              const char *head, const char *remote,
 117                              const char *remote_name, const char *remote_nick,
 118                              const char *local_name, int not_for_merge,
 119                              int verbose, int force)
 120 {
 121         struct commit *commit;
 122         int remote_len, i, note_len;
 123         unsigned char sha1[20];
 124         char note[1024];
 125         const char *what, *kind;
 126
 127         if (get_sha1(head, sha1))
 128                 return error("Not a valid object name: %s", head);
 129         commit = lookup_commit_reference(sha1);
 130         if (!commit)
 131                 not_for_merge = 1;
 132
 133         if (!strcmp(remote_name, "HEAD")) {
 134                 kind = "";
 135                 what = "";
 136         }
 137         else if (!strncmp(remote_name, "refs/heads/", 11)) {
 138                 kind = "branch";
 139                 what = remote_name + 11;
 140         }
 141         else if (!strncmp(remote_name, "refs/tags/", 10)) {
 142                 kind = "tag";
 143                 what = remote_name + 10;
 144         }
 145         else if (!strncmp(remote_name, "refs/remotes/", 13)) {
 146                 kind = "remote branch";
 147                 what = remote_name + 13;
 148         }
 149         else {
 150                 kind = "";
 151                 what = remote_name;
 152         }
 153
 154         remote_len = strlen(remote);
 155         for (i = remote_len - 1; remote[i] == '/' && 0 <= i; i--)
 156                 ;
 157         remote_len = i + 1;
 158         if (4 < i && !strncmp(".git", remote + i - 3, 4))
 159                 remote_len = i - 3;
 160         note_len = sprintf(note, "%s\t%s\t",
 161                            sha1_to_hex(commit ? commit->object.sha1 : sha1),
 162                            not_for_merge ? "not-for-merge" : "");
 163         if (*what) {
 164                 if (*kind)
 165                         note_len += sprintf(note + note_len, "%s ", kind);
 166                 note_len += sprintf(note + note_len, "'%s' of ", what);
 167         }
 168         note_len += sprintf(note + note_len, "%.*s", remote_len, remote);
 169         fprintf(fp, "%s\n", note);
 170         return update_local_ref(local_name, head, note, verbose, force);
 171 }
 172
 173 static char *keep;
 174 static void remove_keep(void)
 175 {
 176         if (keep && *keep)
 177                 unlink(keep);
 178 }
 179
 180 static void remove_keep_on_signal(int signo)
 181 {
 182         remove_keep();
 183         signal(SIGINT, SIG_DFL);
 184         raise(signo);
 185 }
 186
 187 static char *find_local_name(const char *remote_name, const char *refs,
 188                              int *force_p, int *not_for_merge_p)
 189 {
 190         const char *ref = refs;
 191         int len = strlen(remote_name);
 192
 193         while (ref) {
 194                 const char *next;
 195                 int single_force, not_for_merge;
 196
 197                 while (*ref == '\n')
 198                         ref++;
 199                 if (!*ref)
 200                         break;
 201                 next = strchr(ref, '\n');
 202
 203                 single_force = not_for_merge = 0;
 204                 if (*ref == '+') {
 205                         single_force = 1;
 206                         ref++;
 207                 }
 208                 if (*ref == '.') {
 209                         not_for_merge = 1;
 210                         ref++;
 211                         if (*ref == '+') {
 212                                 single_force = 1;
 213                                 ref++;
 214                         }
 215                 }
 216                 if (!strncmp(remote_name, ref, len) && ref[len] == ':') {
 217                         const char *local_part = ref + len + 1;
 218                         char *ret;
 219                         int retlen;
 220
 221                         if (!next)
 222                                 retlen = strlen(local_part);
 223                         else
 224                                 retlen = next - local_part;
 225                         ret = xmalloc(retlen + 1);
 226                         memcpy(ret, local_part, retlen);
 227                         ret[retlen] = 0;
 228                         *force_p = single_force;
 229                         *not_for_merge_p = not_for_merge;
 230                         return ret;
 231                 }
 232                 ref = next;
 233         }
 234         return NULL;
 235 }
 236
 237 static int fetch_native_store(FILE *fp,
 238                               const char *remote,
 239                               const char *remote_nick,
 240                               const char *refs,
 241                               int verbose, int force)
 242 {
 243         char buffer[1024];
 244         int err = 0;
 245
 246         signal(SIGINT, remove_keep_on_signal);
 247         atexit(remove_keep);
 248
 249         while (fgets(buffer, sizeof(buffer), stdin)) {
 250                 int len;
 251                 char *cp;
 252                 char *local_name;
 253                 int single_force, not_for_merge;
 254
 255                 for (cp = buffer; *cp && !isspace(*cp); cp++)
 256                         ;
 257                 if (*cp)
 258                         *cp++ = 0;
 259                 len = strlen(cp);
 260                 if (len && cp[len-1] == '\n')
 261                         cp[--len] = 0;
 262                 if (!strcmp(buffer, "failed"))
 263                         die("Fetch failure: %s", remote);
 264                 if (!strcmp(buffer, "pack"))
 265                         continue;
 266                 if (!strcmp(buffer, "keep")) {
 267                         char *od = get_object_directory();
 268                         int len = strlen(od) + strlen(cp) + 50;
 269                         keep = xmalloc(len);
 270                         sprintf(keep, "%s/pack/pack-%s.keep", od, cp);
 271                         continue;
 272                 }
 273
 274                 local_name = find_local_name(cp, refs,
 275                                              &single_force, &not_for_merge);
 276                 if (!local_name)
 277                         continue;
 278                 err |= append_fetch_head(fp,
 279                                          buffer, remote, cp, remote_nick,
 280                                          local_name, not_for_merge,
 281                                          verbose, force || single_force);
 282         }
 283         return err;
 284 }
 285
 286 static int parse_reflist(const char *reflist)
 287 {
 288         const char *ref;
 289
 290         printf("refs='");
 291         for (ref = reflist; ref; ) {
 292                 const char *next;
 293                 while (*ref && isspace(*ref))
 294                         ref++;
 295                 if (!*ref)
 296                         break;
 297                 for (next = ref; *next && !isspace(*next); next++)
 298                         ;
 299                 printf("\n%.*s", (int)(next - ref), ref);
 300                 ref = next;
 301         }
 302         printf("'\n");
 303
 304         printf("rref='");
 305         for (ref = reflist; ref; ) {
 306                 const char *next, *colon;
 307                 while (*ref && isspace(*ref))
 308                         ref++;
 309                 if (!*ref)
 310                         break;
 311                 for (next = ref; *next && !isspace(*next); next++)
 312                         ;
 313                 if (*ref == '.')
 314                         ref++;
 315                 if (*ref == '+')
 316                         ref++;
 317                 colon = strchr(ref, ':');
 318                 putchar('\n');
 319                 printf("%.*s", (int)((colon ? colon : next) - ref), ref);
 320                 ref = next;
 321         }
 322         printf("'\n");
 323         return 0;
 324 }
 325
 326 int cmd_fetch__tool(int argc, const char **argv, const char *prefix)
 327 {
 328         int verbose = 0;
 329         int force = 0;
 330
 331         while (1 < argc) {
 332                 const char *arg = argv[1];
 333                 if (!strcmp("-v", arg))
 334                         verbose = 1;
 335                 else if (!strcmp("-f", arg))
 336                         force = 1;
 337                 else
 338                         break;
 339                 argc--;
 340                 argv++;
 341         }
 342
 343         if (argc <= 1)
 344                 return error("Missing subcommand");
 345
 346         if (!strcmp("append-fetch-head", argv[1])) {
 347                 int result;
 348                 FILE *fp;
 349
 350                 if (argc != 8)
 351                         return error("append-fetch-head takes 6 args");
 352                 fp = fopen(git_path("FETCH_HEAD"), "a");
 353                 result = append_fetch_head(fp, argv[2], argv[3],
 354                                            argv[4], argv[5],
 355                                            argv[6], !!argv[7][0],
 356                                            verbose, force);
 357                 fclose(fp);
 358                 return result;
 359         }
 360         if (!strcmp("update-local-ref", argv[1])) {
 361                 if (argc != 5)
 362                         return error("update-local-ref takes 3 args");
 363                 return update_local_ref(argv[2], argv[3], argv[4],
 364                                         verbose, force);
 365         }
 366         if (!strcmp("native-store", argv[1])) {
 367                 int result;
 368                 FILE *fp;
 369
 370                 if (argc != 5)
 371                         return error("fetch-native-store takes 3 args");
 372                 fp = fopen(git_path("FETCH_HEAD"), "a");
 373                 result = fetch_native_store(fp, argv[2], argv[3], argv[4],
 374                                             verbose, force);
 375                 fclose(fp);
 376                 return result;
 377         }
 378         if (!strcmp("parse-reflist", argv[1])) {
 379                 if (argc != 3)
 380                         return error("parse-reflist takes 1 arg");
 381                 return parse_reflist(argv[2]);
 382         }
 383
 384         return error("Unknown subcommand: %s", argv[1]);
 385 }