builtin-fetch--tool.c

   1 #include "cache.h"
   2 #include "refs.h"
   3 #include "commit.h"
   4
   5 #define CHUNK_SIZE 1024
   6
   7 static char *get_stdin(void)
   8 {
   9         size_t offset = 0;
  10         char *data = xmalloc(CHUNK_SIZE);
  11
  12         while (1) {
  13                 ssize_t cnt = xread(0, data + offset, CHUNK_SIZE);
  14                 if (cnt < 0)
  15                         die("error reading standard input: %s",
  16                             strerror(errno));
  17                 if (cnt == 0) {
  18                         data[offset] = 0;
  19                         break;
  20                 }
  21                 offset += cnt;
  22                 data = xrealloc(data, offset + CHUNK_SIZE);
  23         }
  24         return data;
  25 }
  26
  27 static void show_new(enum object_type type, unsigned char *sha1_new)
  28 {
  29         fprintf(stderr, "  %s: %s\n", typename(type),
  30                 find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
  31 }
  32
  33 static int update_ref(const char *action,
  34                       const char *refname,
  35                       unsigned char *sha1,
  36                       unsigned char *oldval)
  37 {
  38         char msg[1024];
  39         char *rla = getenv("GIT_REFLOG_ACTION");
  40         static struct ref_lock *lock;
  41
  42         if (!rla)
  43                 rla = "(reflog update)";
  44         snprintf(msg, sizeof(msg), "%s: %s", rla, action);
  45         lock = lock_any_ref_for_update(refname, oldval, 0);
  46         if (!lock)
  47                 return 1;
  48         if (write_ref_sha1(lock, sha1, msg) < 0)
  49                 return 1;
  50         return 0;
  51 }
  52
  53 static int update_local_ref(const char *name,
  54                             const char *new_head,
  55                             const char *note,
  56                             int verbose, int force)
  57 {
  58         unsigned char sha1_old[20], sha1_new[20];
  59         char oldh[41], newh[41];
  60         struct commit *current, *updated;
  61         enum object_type type;
  62
  63         if (get_sha1_hex(new_head, sha1_new))
  64                 die("malformed object name %s", new_head);
  65
  66         type = sha1_object_info(sha1_new, NULL);
  67         if (type < 0)
  68                 die("object %s not found", new_head);
  69
  70         if (!*name) {
  71                 /* Not storing */
  72                 if (verbose) {
  73                         fprintf(stderr, "* fetched %s\n", note);
  74                         show_new(type, sha1_new);
  75                 }
  76                 return 0;
  77         }
  78
  79         if (get_sha1(name, sha1_old)) {
  80                 char *msg;
  81         just_store:
  82                 /* new ref */
  83                 if (!strncmp(name, "refs/tags/", 10))
  84                         msg = "storing tag";
  85                 else
  86                         msg = "storing head";
  87                 fprintf(stderr, "* %s: storing %s\n",
  88                         name, note);
  89                 show_new(type, sha1_new);
  90                 return update_ref(msg, name, sha1_new, NULL);
  91         }
  92
  93         if (!hashcmp(sha1_old, sha1_new)) {
  94                 if (verbose) {
  95                         fprintf(stderr, "* %s: same as %s\n", name, note);
  96                         show_new(type, sha1_new);
  97                 }
  98                 return 0;
  99         }
 100
 101         if (!strncmp(name, "refs/tags/", 10)) {
 102                 fprintf(stderr, "* %s: updating with %s\n", name, note);
 103                 show_new(type, sha1_new);
 104                 return update_ref("updating tag", name, sha1_new, NULL);
 105         }
 106
 107         current = lookup_commit_reference(sha1_old);
 108         updated = lookup_commit_reference(sha1_new);
 109         if (!current || !updated)
 110                 goto just_store;
 111
 112         strcpy(oldh, find_unique_abbrev(current->object.sha1, DEFAULT_ABBREV));
 113         strcpy(newh, find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
 114
 115         if (in_merge_bases(current, &updated, 1)) {
 116                 fprintf(stderr, "* %s: fast forward to %s\n",
 117                         name, note);
 118                 fprintf(stderr, "  old..new: %s..%s\n", oldh, newh);
 119                 return update_ref("fast forward", name, sha1_new, sha1_old);
 120         }
 121         if (!force) {
 122                 fprintf(stderr,
 123                         "* %s: not updating to non-fast forward %s\n",
 124                         name, note);
 125                 fprintf(stderr,
 126                         "  old...new: %s...%s\n", oldh, newh);
 127                 return 1;
 128         }
 129         fprintf(stderr,
 130                 "* %s: forcing update to non-fast forward %s\n",
 131                 name, note);
 132         fprintf(stderr, "  old...new: %s...%s\n", oldh, newh);
 133         return update_ref("forced-update", name, sha1_new, sha1_old);
 134 }
 135
 136 static int append_fetch_head(FILE *fp,
 137                              const char *head, const char *remote,
 138                              const char *remote_name, const char *remote_nick,
 139                              const char *local_name, int not_for_merge,
 140                              int verbose, int force)
 141 {
 142         struct commit *commit;
 143         int remote_len, i, note_len;
 144         unsigned char sha1[20];
 145         char note[1024];
 146         const char *what, *kind;
 147
 148         if (get_sha1(head, sha1))
 149                 return error("Not a valid object name: %s", head);
 150         commit = lookup_commit_reference(sha1);
 151         if (!commit)
 152                 not_for_merge = 1;
 153
 154         if (!strcmp(remote_name, "HEAD")) {
 155                 kind = "";
 156                 what = "";
 157         }
 158         else if (!strncmp(remote_name, "refs/heads/", 11)) {
 159                 kind = "branch";
 160                 what = remote_name + 11;
 161         }
 162         else if (!strncmp(remote_name, "refs/tags/", 10)) {
 163                 kind = "tag";
 164                 what = remote_name + 10;
 165         }
 166         else if (!strncmp(remote_name, "refs/remotes/", 13)) {
 167                 kind = "remote branch";
 168                 what = remote_name + 13;
 169         }
 170         else {
 171                 kind = "";
 172                 what = remote_name;
 173         }
 174
 175         remote_len = strlen(remote);
 176         for (i = remote_len - 1; remote[i] == '/' && 0 <= i; i--)
 177                 ;
 178         remote_len = i + 1;
 179         if (4 < i && !strncmp(".git", remote + i - 3, 4))
 180                 remote_len = i - 3;
 181
 182         note_len = 0;
 183         if (*what) {
 184                 if (*kind)
 185                         note_len += sprintf(note + note_len, "%s ", kind);
 186                 note_len += sprintf(note + note_len, "'%s' of ", what);
 187         }
 188         note_len += sprintf(note + note_len, "%.*s", remote_len, remote);
 189         fprintf(fp, "%s\t%s\t%s\n",
 190                 sha1_to_hex(commit ? commit->object.sha1 : sha1),
 191                 not_for_merge ? "not-for-merge" : "",
 192                 note);
 193         return update_local_ref(local_name, head, note, verbose, force);
 194 }
 195
 196 static char *keep;
 197 static void remove_keep(void)
 198 {
 199         if (keep && *keep)
 200                 unlink(keep);
 201 }
 202
 203 static void remove_keep_on_signal(int signo)
 204 {
 205         remove_keep();
 206         signal(SIGINT, SIG_DFL);
 207         raise(signo);
 208 }
 209
 210 static char *find_local_name(const char *remote_name, const char *refs,
 211                              int *force_p, int *not_for_merge_p)
 212 {
 213         const char *ref = refs;
 214         int len = strlen(remote_name);
 215
 216         while (ref) {
 217                 const char *next;
 218                 int single_force, not_for_merge;
 219
 220                 while (*ref == '\n')
 221                         ref++;
 222                 if (!*ref)
 223                         break;
 224                 next = strchr(ref, '\n');
 225
 226                 single_force = not_for_merge = 0;
 227                 if (*ref == '+') {
 228                         single_force = 1;
 229                         ref++;
 230                 }
 231                 if (*ref == '.') {
 232                         not_for_merge = 1;
 233                         ref++;
 234                         if (*ref == '+') {
 235                                 single_force = 1;
 236                                 ref++;
 237                         }
 238                 }
 239                 if (!strncmp(remote_name, ref, len) && ref[len] == ':') {
 240                         const char *local_part = ref + len + 1;
 241                         char *ret;
 242                         int retlen;
 243
 244                         if (!next)
 245                                 retlen = strlen(local_part);
 246                         else
 247                                 retlen = next - local_part;
 248                         ret = xmalloc(retlen + 1);
 249                         memcpy(ret, local_part, retlen);
 250                         ret[retlen] = 0;
 251                         *force_p = single_force;
 252                         *not_for_merge_p = not_for_merge;
 253                         return ret;
 254                 }
 255                 ref = next;
 256         }
 257         return NULL;
 258 }
 259
 260 static int fetch_native_store(FILE *fp,
 261                               const char *remote,
 262                               const char *remote_nick,
 263                               const char *refs,
 264                               int verbose, int force)
 265 {
 266         char buffer[1024];
 267         int err = 0;
 268
 269         signal(SIGINT, remove_keep_on_signal);
 270         atexit(remove_keep);
 271
 272         while (fgets(buffer, sizeof(buffer), stdin)) {
 273                 int len;
 274                 char *cp;
 275                 char *local_name;
 276                 int single_force, not_for_merge;
 277
 278                 for (cp = buffer; *cp && !isspace(*cp); cp++)
 279                         ;
 280                 if (*cp)
 281                         *cp++ = 0;
 282                 len = strlen(cp);
 283                 if (len && cp[len-1] == '\n')
 284                         cp[--len] = 0;
 285                 if (!strcmp(buffer, "failed"))
 286                         die("Fetch failure: %s", remote);
 287                 if (!strcmp(buffer, "pack"))
 288                         continue;
 289                 if (!strcmp(buffer, "keep")) {
 290                         char *od = get_object_directory();
 291                         int len = strlen(od) + strlen(cp) + 50;
 292                         keep = xmalloc(len);
 293                         sprintf(keep, "%s/pack/pack-%s.keep", od, cp);
 294                         continue;
 295                 }
 296
 297                 local_name = find_local_name(cp, refs,
 298                                              &single_force, &not_for_merge);
 299                 if (!local_name)
 300                         continue;
 301                 err |= append_fetch_head(fp,
 302                                          buffer, remote, cp, remote_nick,
 303                                          local_name, not_for_merge,
 304                                          verbose, force || single_force);
 305         }
 306         return err;
 307 }
 308
 309 static int parse_reflist(const char *reflist)
 310 {
 311         const char *ref;
 312
 313         printf("refs='");
 314         for (ref = reflist; ref; ) {
 315                 const char *next;
 316                 while (*ref && isspace(*ref))
 317                         ref++;
 318                 if (!*ref)
 319                         break;
 320                 for (next = ref; *next && !isspace(*next); next++)
 321                         ;
 322                 printf("\n%.*s", (int)(next - ref), ref);
 323                 ref = next;
 324         }
 325         printf("'\n");
 326
 327         printf("rref='");
 328         for (ref = reflist; ref; ) {
 329                 const char *next, *colon;
 330                 while (*ref && isspace(*ref))
 331                         ref++;
 332                 if (!*ref)
 333                         break;
 334                 for (next = ref; *next && !isspace(*next); next++)
 335                         ;
 336                 if (*ref == '.')
 337                         ref++;
 338                 if (*ref == '+')
 339                         ref++;
 340                 colon = strchr(ref, ':');
 341                 putchar('\n');
 342                 printf("%.*s", (int)((colon ? colon : next) - ref), ref);
 343                 ref = next;
 344         }
 345         printf("'\n");
 346         return 0;
 347 }
 348
 349 static int expand_refs_wildcard(const char *ls_remote_result, int numrefs,
 350                                 const char **refs)
 351 {
 352         int i, matchlen, replacelen;
 353         int found_one = 0;
 354         const char *remote = *refs++;
 355         numrefs--;
 356
 357         if (numrefs == 0) {
 358                 fprintf(stderr, "Nothing specified for fetching with remote.%s.fetch\n",
 359                         remote);
 360                 printf("empty\n");
 361         }
 362
 363         for (i = 0; i < numrefs; i++) {
 364                 const char *ref = refs[i];
 365                 const char *lref = ref;
 366                 const char *colon;
 367                 const char *tail;
 368                 const char *ls;
 369                 const char *next;
 370
 371                 if (*lref == '+')
 372                         lref++;
 373                 colon = strchr(lref, ':');
 374                 tail = lref + strlen(lref);
 375                 if (!(colon &&
 376                       2 < colon - lref &&
 377                       colon[-1] == '*' &&
 378                       colon[-2] == '/' &&
 379                       2 < tail - (colon + 1) &&
 380                       tail[-1] == '*' &&
 381                       tail[-2] == '/')) {
 382                         /* not a glob */
 383                         if (!found_one++)
 384                                 printf("explicit\n");
 385                         printf("%s\n", ref);
 386                         continue;
 387                 }
 388
 389                 /* glob */
 390                 if (!found_one++)
 391                         printf("glob\n");
 392
 393                 /* lref to colon-2 is remote hierarchy name;
 394                  * colon+1 to tail-2 is local.
 395                  */
 396                 matchlen = (colon-1) - lref;
 397                 replacelen = (tail-1) - (colon+1);
 398                 for (ls = ls_remote_result; ls; ls = next) {
 399                         const char *eol;
 400                         unsigned char sha1[20];
 401                         int namelen;
 402
 403                         while (*ls && isspace(*ls))
 404                                 ls++;
 405                         next = strchr(ls, '\n');
 406                         eol = !next ? (ls + strlen(ls)) : next;
 407                         if (!memcmp("^{}", eol-3, 3))
 408                                 continue;
 409                         if (eol - ls < 40)
 410                                 continue;
 411                         if (get_sha1_hex(ls, sha1))
 412                                 continue;
 413                         ls += 40;
 414                         while (ls < eol && isspace(*ls))
 415                                 ls++;
 416                         /* ls to next (or eol) is the name.
 417                          * is it identical to lref to colon-2?
 418                          */
 419                         if ((eol - ls) <= matchlen ||
 420                             strncmp(ls, lref, matchlen))
 421                                 continue;
 422
 423                         /* Yes, it is a match */
 424                         namelen = eol - ls;
 425                         if (lref != ref)
 426                                 putchar('+');
 427                         printf("%.*s:%.*s%.*s\n",
 428                                namelen, ls,
 429                                replacelen, colon + 1,
 430                                namelen - matchlen, ls + matchlen);
 431                 }
 432         }
 433         return 0;
 434 }
 435
 436 static int pick_rref(int sha1_only, const char *rref, const char *ls_remote_result)
 437 {
 438         int err = 0;
 439         int lrr_count = lrr_count, i, pass;
 440         const char *cp;
 441         struct lrr {
 442                 const char *line;
 443                 const char *name;
 444                 int namelen;
 445                 int shown;
 446         } *lrr_list = lrr_list;
 447
 448         for (pass = 0; pass < 2; pass++) {
 449                 /* pass 0 counts and allocates, pass 1 fills... */
 450                 cp = ls_remote_result;
 451                 i = 0;
 452                 while (1) {
 453                         const char *np;
 454                         while (*cp && isspace(*cp))
 455                                 cp++;
 456                         if (!*cp)
 457                                 break;
 458                         np = strchr(cp, '\n');
 459                         if (!np)
 460                                 np = cp + strlen(cp);
 461                         if (pass) {
 462                                 lrr_list[i].line = cp;
 463                                 lrr_list[i].name = cp + 41;
 464                                 lrr_list[i].namelen = np - (cp + 41);
 465                         }
 466                         i++;
 467                         cp = np;
 468                 }
 469                 if (!pass) {
 470                         lrr_count = i;
 471                         lrr_list = xcalloc(lrr_count, sizeof(*lrr_list));
 472                 }
 473         }
 474
 475         while (1) {
 476                 const char *next;
 477                 int rreflen;
 478                 int i;
 479
 480                 while (*rref && isspace(*rref))
 481                         rref++;
 482                 if (!*rref)
 483                         break;
 484                 next = strchr(rref, '\n');
 485                 if (!next)
 486                         next = rref + strlen(rref);
 487                 rreflen = next - rref;
 488
 489                 for (i = 0; i < lrr_count; i++) {
 490                         struct lrr *lrr = &(lrr_list[i]);
 491
 492                         if (rreflen == lrr->namelen &&
 493                             !memcmp(lrr->name, rref, rreflen)) {
 494                                 if (!lrr->shown)
 495                                         printf("%.*s\n",
 496                                                sha1_only ? 40 : lrr->namelen + 41,
 497                                                lrr->line);
 498                                 lrr->shown = 1;
 499                                 break;
 500                         }
 501                 }
 502                 if (lrr_count <= i) {
 503                         error("pick-rref: %.*s not found", rreflen, rref);
 504                         err = 1;
 505                 }
 506                 rref = next;
 507         }
 508         free(lrr_list);
 509         return err;
 510 }
 511
 512 int cmd_fetch__tool(int argc, const char **argv, const char *prefix)
 513 {
 514         int verbose = 0;
 515         int force = 0;
 516         int sopt = 0;
 517
 518         while (1 < argc) {
 519                 const char *arg = argv[1];
 520                 if (!strcmp("-v", arg))
 521                         verbose = 1;
 522                 else if (!strcmp("-f", arg))
 523                         force = 1;
 524                 else if (!strcmp("-s", arg))
 525                         sopt = 1;
 526                 else
 527                         break;
 528                 argc--;
 529                 argv++;
 530         }
 531
 532         if (argc <= 1)
 533                 return error("Missing subcommand");
 534
 535         if (!strcmp("append-fetch-head", argv[1])) {
 536                 int result;
 537                 FILE *fp;
 538
 539                 if (argc != 8)
 540                         return error("append-fetch-head takes 6 args");
 541                 fp = fopen(git_path("FETCH_HEAD"), "a");
 542                 result = append_fetch_head(fp, argv[2], argv[3],
 543                                            argv[4], argv[5],
 544                                            argv[6], !!argv[7][0],
 545                                            verbose, force);
 546                 fclose(fp);
 547                 return result;
 548         }
 549         if (!strcmp("native-store", argv[1])) {
 550                 int result;
 551                 FILE *fp;
 552
 553                 if (argc != 5)
 554                         return error("fetch-native-store takes 3 args");
 555                 fp = fopen(git_path("FETCH_HEAD"), "a");
 556                 result = fetch_native_store(fp, argv[2], argv[3], argv[4],
 557                                             verbose, force);
 558                 fclose(fp);
 559                 return result;
 560         }
 561         if (!strcmp("parse-reflist", argv[1])) {
 562                 const char *reflist;
 563                 if (argc != 3)
 564                         return error("parse-reflist takes 1 arg");
 565                 reflist = argv[2];
 566                 if (!strcmp(reflist, "-"))
 567                         reflist = get_stdin();
 568                 return parse_reflist(reflist);
 569         }
 570         if (!strcmp("pick-rref", argv[1])) {
 571                 const char *ls_remote_result;
 572                 if (argc != 4)
 573                         return error("pick-rref takes 2 args");
 574                 ls_remote_result = argv[3];
 575                 if (!strcmp(ls_remote_result, "-"))
 576                         ls_remote_result = get_stdin();
 577                 return pick_rref(sopt, argv[2], ls_remote_result);
 578         }
 579         if (!strcmp("expand-refs-wildcard", argv[1])) {
 580                 const char *reflist;
 581                 if (argc < 4)
 582                         return error("expand-refs-wildcard takes at least 2 args");
 583                 reflist = argv[2];
 584                 if (!strcmp(reflist, "-"))
 585                         reflist = get_stdin();
 586                 return expand_refs_wildcard(reflist, argc - 3, argv + 3);
 587         }
 588
 589         return error("Unknown subcommand: %s", argv[1]);
 590 }