builtin-fetch--tool.c

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