sha1_name.c

   1 #include "cache.h"
   2 #include "tag.h"
   3 #include "commit.h"
   4 #include "tree.h"
   5 #include "blob.h"
   6 #include "tree-walk.h"
   7 #include "refs.h"
   8 #include "remote.h"
   9
  10 static int find_short_object_filename(int len, const char *name, unsigned char *sha1)
  11 {
  12         struct alternate_object_database *alt;
  13         char hex[40];
  14         int found = 0;
  15         static struct alternate_object_database *fakeent;
  16
  17         if (!fakeent) {
  18                 const char *objdir = get_object_directory();
  19                 int objdir_len = strlen(objdir);
  20                 int entlen = objdir_len + 43;
  21                 fakeent = xmalloc(sizeof(*fakeent) + entlen);
  22                 memcpy(fakeent->base, objdir, objdir_len);
  23                 fakeent->name = fakeent->base + objdir_len + 1;
  24                 fakeent->name[-1] = '/';
  25         }
  26         fakeent->next = alt_odb_list;
  27
  28         sprintf(hex, "%.2s", name);
  29         for (alt = fakeent; alt && found < 2; alt = alt->next) {
  30                 struct dirent *de;
  31                 DIR *dir;
  32                 sprintf(alt->name, "%.2s/", name);
  33                 dir = opendir(alt->base);
  34                 if (!dir)
  35                         continue;
  36                 while ((de = readdir(dir)) != NULL) {
  37                         if (strlen(de->d_name) != 38)
  38                                 continue;
  39                         if (memcmp(de->d_name, name + 2, len - 2))
  40                                 continue;
  41                         if (!found) {
  42                                 memcpy(hex + 2, de->d_name, 38);
  43                                 found++;
  44                         }
  45                         else if (memcmp(hex + 2, de->d_name, 38)) {
  46                                 found = 2;
  47                                 break;
  48                         }
  49                 }
  50                 closedir(dir);
  51         }
  52         if (found == 1)
  53                 return get_sha1_hex(hex, sha1) == 0;
  54         return found;
  55 }
  56
  57 static int match_sha(unsigned len, const unsigned char *a, const unsigned char *b)
  58 {
  59         do {
  60                 if (*a != *b)
  61                         return 0;
  62                 a++;
  63                 b++;
  64                 len -= 2;
  65         } while (len > 1);
  66         if (len)
  67                 if ((*a ^ *b) & 0xf0)
  68                         return 0;
  69         return 1;
  70 }
  71
  72 static int find_short_packed_object(int len, const unsigned char *match, unsigned char *sha1)
  73 {
  74         struct packed_git *p;
  75         const unsigned char *found_sha1 = NULL;
  76         int found = 0;
  77
  78         prepare_packed_git();
  79         for (p = packed_git; p && found < 2; p = p->next) {
  80                 uint32_t num, last;
  81                 uint32_t first = 0;
  82                 open_pack_index(p);
  83                 num = p->num_objects;
  84                 last = num;
  85                 while (first < last) {
  86                         uint32_t mid = (first + last) / 2;
  87                         const unsigned char *now;
  88                         int cmp;
  89
  90                         now = nth_packed_object_sha1(p, mid);
  91                         cmp = hashcmp(match, now);
  92                         if (!cmp) {
  93                                 first = mid;
  94                                 break;
  95                         }
  96                         if (cmp > 0) {
  97                                 first = mid+1;
  98                                 continue;
  99                         }
 100                         last = mid;
 101                 }
 102                 if (first < num) {
 103                         const unsigned char *now, *next;
 104                        now = nth_packed_object_sha1(p, first);
 105                         if (match_sha(len, match, now)) {
 106                                 next = nth_packed_object_sha1(p, first+1);
 107                                if (!next|| !match_sha(len, match, next)) {
 108                                         /* unique within this pack */
 109                                         if (!found) {
 110                                                 found_sha1 = now;
 111                                                 found++;
 112                                         }
 113                                         else if (hashcmp(found_sha1, now)) {
 114                                                 found = 2;
 115                                                 break;
 116                                         }
 117                                 }
 118                                 else {
 119                                         /* not even unique within this pack */
 120                                         found = 2;
 121                                         break;
 122                                 }
 123                         }
 124                 }
 125         }
 126         if (found == 1)
 127                 hashcpy(sha1, found_sha1);
 128         return found;
 129 }
 130
 131 #define SHORT_NAME_NOT_FOUND (-1)
 132 #define SHORT_NAME_AMBIGUOUS (-2)
 133
 134 static int find_unique_short_object(int len, char *canonical,
 135                                     unsigned char *res, unsigned char *sha1)
 136 {
 137         int has_unpacked, has_packed;
 138         unsigned char unpacked_sha1[20], packed_sha1[20];
 139
 140         prepare_alt_odb();
 141         has_unpacked = find_short_object_filename(len, canonical, unpacked_sha1);
 142         has_packed = find_short_packed_object(len, res, packed_sha1);
 143         if (!has_unpacked && !has_packed)
 144                 return SHORT_NAME_NOT_FOUND;
 145         if (1 < has_unpacked || 1 < has_packed)
 146                 return SHORT_NAME_AMBIGUOUS;
 147         if (has_unpacked != has_packed) {
 148                 hashcpy(sha1, (has_packed ? packed_sha1 : unpacked_sha1));
 149                 return 0;
 150         }
 151         /* Both have unique ones -- do they match? */
 152         if (hashcmp(packed_sha1, unpacked_sha1))
 153                 return SHORT_NAME_AMBIGUOUS;
 154         hashcpy(sha1, packed_sha1);
 155         return 0;
 156 }
 157
 158 static int get_short_sha1(const char *name, int len, unsigned char *sha1,
 159                           int quietly)
 160 {
 161         int i, status;
 162         char canonical[40];
 163         unsigned char res[20];
 164
 165         if (len < MINIMUM_ABBREV || len > 40)
 166                 return -1;
 167         hashclr(res);
 168         memset(canonical, 'x', 40);
 169         for (i = 0; i < len ;i++) {
 170                 unsigned char c = name[i];
 171                 unsigned char val;
 172                 if (c >= '0' && c <= '9')
 173                         val = c - '0';
 174                 else if (c >= 'a' && c <= 'f')
 175                         val = c - 'a' + 10;
 176                 else if (c >= 'A' && c <='F') {
 177                         val = c - 'A' + 10;
 178                         c -= 'A' - 'a';
 179                 }
 180                 else
 181                         return -1;
 182                 canonical[i] = c;
 183                 if (!(i & 1))
 184                         val <<= 4;
 185                 res[i >> 1] |= val;
 186         }
 187
 188         status = find_unique_short_object(i, canonical, res, sha1);
 189         if (!quietly && (status == SHORT_NAME_AMBIGUOUS))
 190                 return error("short SHA1 %.*s is ambiguous.", len, canonical);
 191         return status;
 192 }
 193
 194 const char *find_unique_abbrev(const unsigned char *sha1, int len)
 195 {
 196         int status, exists;
 197         static char hex[41];
 198
 199         exists = has_sha1_file(sha1);
 200         memcpy(hex, sha1_to_hex(sha1), 40);
 201         if (len == 40 || !len)
 202                 return hex;
 203         while (len < 40) {
 204                 unsigned char sha1_ret[20];
 205                 status = get_short_sha1(hex, len, sha1_ret, 1);
 206                 if (exists
 207                     ? !status
 208                     : status == SHORT_NAME_NOT_FOUND) {
 209                         hex[len] = 0;
 210                         return hex;
 211                 }
 212                 len++;
 213         }
 214         return hex;
 215 }
 216
 217 static int ambiguous_path(const char *path, int len)
 218 {
 219         int slash = 1;
 220         int cnt;
 221
 222         for (cnt = 0; cnt < len; cnt++) {
 223                 switch (*path++) {
 224                 case '\0':
 225                         break;
 226                 case '/':
 227                         if (slash)
 228                                 break;
 229                         slash = 1;
 230                         continue;
 231                 case '.':
 232                         continue;
 233                 default:
 234                         slash = 0;
 235                         continue;
 236                 }
 237                 break;
 238         }
 239         return slash;
 240 }
 241
 242 /*
 243  * *string and *len will only be substituted, and *string returned (for
 244  * later free()ing) if the string passed in is a magic short-hand form
 245  * to name a branch.
 246  */
 247 static char *substitute_branch_name(const char **string, int *len)
 248 {
 249         struct strbuf buf = STRBUF_INIT;
 250         int ret = interpret_branch_name(*string, &buf);
 251
 252         if (ret == *len) {
 253                 size_t size;
 254                 *string = strbuf_detach(&buf, &size);
 255                 *len = size;
 256                 return (char *)*string;
 257         }
 258
 259         return NULL;
 260 }
 261
 262 int dwim_ref(const char *str, int len, unsigned char *sha1, char **ref)
 263 {
 264         char *last_branch = substitute_branch_name(&str, &len);
 265         const char **p, *r;
 266         int refs_found = 0;
 267
 268         *ref = NULL;
 269         for (p = ref_rev_parse_rules; *p; p++) {
 270                 char fullref[PATH_MAX];
 271                 unsigned char sha1_from_ref[20];
 272                 unsigned char *this_result;
 273                 int flag;
 274
 275                 this_result = refs_found ? sha1_from_ref : sha1;
 276                 mksnpath(fullref, sizeof(fullref), *p, len, str);
 277                 r = resolve_ref(fullref, this_result, 1, &flag);
 278                 if (r) {
 279                         if (!refs_found++)
 280                                 *ref = xstrdup(r);
 281                         if (!warn_ambiguous_refs)
 282                                 break;
 283                 } else if ((flag & REF_ISSYMREF) && strcmp(fullref, "HEAD"))
 284                         warning("ignoring dangling symref %s.", fullref);
 285         }
 286         free(last_branch);
 287         return refs_found;
 288 }
 289
 290 int dwim_log(const char *str, int len, unsigned char *sha1, char **log)
 291 {
 292         char *last_branch = substitute_branch_name(&str, &len);
 293         const char **p;
 294         int logs_found = 0;
 295
 296         *log = NULL;
 297         for (p = ref_rev_parse_rules; *p; p++) {
 298                 struct stat st;
 299                 unsigned char hash[20];
 300                 char path[PATH_MAX];
 301                 const char *ref, *it;
 302
 303                 mksnpath(path, sizeof(path), *p, len, str);
 304                 ref = resolve_ref(path, hash, 1, NULL);
 305                 if (!ref)
 306                         continue;
 307                 if (!stat(git_path("logs/%s", path), &st) &&
 308                     S_ISREG(st.st_mode))
 309                         it = path;
 310                 else if (strcmp(ref, path) &&
 311                          !stat(git_path("logs/%s", ref), &st) &&
 312                          S_ISREG(st.st_mode))
 313                         it = ref;
 314                 else
 315                         continue;
 316                 if (!logs_found++) {
 317                         *log = xstrdup(it);
 318                         hashcpy(sha1, hash);
 319                 }
 320                 if (!warn_ambiguous_refs)
 321                         break;
 322         }
 323         free(last_branch);
 324         return logs_found;
 325 }
 326
 327 static inline int upstream_mark(const char *string, int len)
 328 {
 329         const char *suffix[] = { "@{upstream}", "@{u}" };
 330         int i;
 331
 332         for (i = 0; i < ARRAY_SIZE(suffix); i++) {
 333                 int suffix_len = strlen(suffix[i]);
 334                 if (suffix_len <= len
 335                     && !memcmp(string, suffix[i], suffix_len))
 336                         return suffix_len;
 337         }
 338         return 0;
 339 }
 340
 341 static int get_sha1_1(const char *name, int len, unsigned char *sha1);
 342
 343 static int get_sha1_basic(const char *str, int len, unsigned char *sha1)
 344 {
 345         static const char *warning = "warning: refname '%.*s' is ambiguous.\n";
 346         char *real_ref = NULL;
 347         int refs_found = 0;
 348         int at, reflog_len;
 349
 350         if (len == 40 && !get_sha1_hex(str, sha1))
 351                 return 0;
 352
 353         /* basic@{time or number or -number} format to query ref-log */
 354         reflog_len = at = 0;
 355         if (len && str[len-1] == '}') {
 356                 for (at = len-2; at >= 0; at--) {
 357                         if (str[at] == '@' && str[at+1] == '{') {
 358                                 if (!upstream_mark(str + at, len - at)) {
 359                                         reflog_len = (len-1) - (at+2);
 360                                         len = at;
 361                                 }
 362                                 break;
 363                         }
 364                 }
 365         }
 366
 367         /* Accept only unambiguous ref paths. */
 368         if (len && ambiguous_path(str, len))
 369                 return -1;
 370
 371         if (!len && reflog_len) {
 372                 struct strbuf buf = STRBUF_INIT;
 373                 int ret;
 374                 /* try the @{-N} syntax for n-th checkout */
 375                 ret = interpret_branch_name(str+at, &buf);
 376                 if (ret > 0) {
 377                         /* substitute this branch name and restart */
 378                         return get_sha1_1(buf.buf, buf.len, sha1);
 379                 } else if (ret == 0) {
 380                         return -1;
 381                 }
 382                 /* allow "@{...}" to mean the current branch reflog */
 383                 refs_found = dwim_ref("HEAD", 4, sha1, &real_ref);
 384         } else if (reflog_len)
 385                 refs_found = dwim_log(str, len, sha1, &real_ref);
 386         else
 387                 refs_found = dwim_ref(str, len, sha1, &real_ref);
 388
 389         if (!refs_found)
 390                 return -1;
 391
 392         if (warn_ambiguous_refs && refs_found > 1)
 393                 fprintf(stderr, warning, len, str);
 394
 395         if (reflog_len) {
 396                 int nth, i;
 397                 unsigned long at_time;
 398                 unsigned long co_time;
 399                 int co_tz, co_cnt;
 400
 401                 /* a @{-N} placed anywhere except the start is an error */
 402                 if (str[at+2] == '-')
 403                         return -1;
 404
 405                 /* Is it asking for N-th entry, or approxidate? */
 406                 for (i = nth = 0; 0 <= nth && i < reflog_len; i++) {
 407                         char ch = str[at+2+i];
 408                         if ('0' <= ch && ch <= '9')
 409                                 nth = nth * 10 + ch - '0';
 410                         else
 411                                 nth = -1;
 412                 }
 413                 if (100000000 <= nth) {
 414                         at_time = nth;
 415                         nth = -1;
 416                 } else if (0 <= nth)
 417                         at_time = 0;
 418                 else {
 419                         int errors = 0;
 420                         char *tmp = xstrndup(str + at + 2, reflog_len);
 421                         at_time = approxidate_careful(tmp, &errors);
 422                         free(tmp);
 423                         if (errors)
 424                                 return -1;
 425                 }
 426                 if (read_ref_at(real_ref, at_time, nth, sha1, NULL,
 427                                 &co_time, &co_tz, &co_cnt)) {
 428                         if (at_time)
 429                                 fprintf(stderr,
 430                                         "warning: Log for '%.*s' only goes "
 431                                         "back to %s.\n", len, str,
 432                                         show_date(co_time, co_tz, DATE_RFC2822));
 433                         else
 434                                 fprintf(stderr,
 435                                         "warning: Log for '%.*s' only has "
 436                                         "%d entries.\n", len, str, co_cnt);
 437                 }
 438         }
 439
 440         free(real_ref);
 441         return 0;
 442 }
 443
 444 static int get_parent(const char *name, int len,
 445                       unsigned char *result, int idx)
 446 {
 447         unsigned char sha1[20];
 448         int ret = get_sha1_1(name, len, sha1);
 449         struct commit *commit;
 450         struct commit_list *p;
 451
 452         if (ret)
 453                 return ret;
 454         commit = lookup_commit_reference(sha1);
 455         if (!commit)
 456                 return -1;
 457         if (parse_commit(commit))
 458                 return -1;
 459         if (!idx) {
 460                 hashcpy(result, commit->object.sha1);
 461                 return 0;
 462         }
 463         p = commit->parents;
 464         while (p) {
 465                 if (!--idx) {
 466                         hashcpy(result, p->item->object.sha1);
 467                         return 0;
 468                 }
 469                 p = p->next;
 470         }
 471         return -1;
 472 }
 473
 474 static int get_nth_ancestor(const char *name, int len,
 475                             unsigned char *result, int generation)
 476 {
 477         unsigned char sha1[20];
 478         struct commit *commit;
 479         int ret;
 480
 481         ret = get_sha1_1(name, len, sha1);
 482         if (ret)
 483                 return ret;
 484         commit = lookup_commit_reference(sha1);
 485         if (!commit)
 486                 return -1;
 487
 488         while (generation--) {
 489                 if (parse_commit(commit) || !commit->parents)
 490                         return -1;
 491                 commit = commit->parents->item;
 492         }
 493         hashcpy(result, commit->object.sha1);
 494         return 0;
 495 }
 496
 497 struct object *peel_to_type(const char *name, int namelen,
 498                             struct object *o, enum object_type expected_type)
 499 {
 500         if (name && !namelen)
 501                 namelen = strlen(name);
 502         if (!o) {
 503                 unsigned char sha1[20];
 504                 if (get_sha1_1(name, namelen, sha1))
 505                         return NULL;
 506                 o = parse_object(sha1);
 507         }
 508         while (1) {
 509                 if (!o || (!o->parsed && !parse_object(o->sha1)))
 510                         return NULL;
 511                 if (o->type == expected_type)
 512                         return o;
 513                 if (o->type == OBJ_TAG)
 514                         o = ((struct tag*) o)->tagged;
 515                 else if (o->type == OBJ_COMMIT)
 516                         o = &(((struct commit *) o)->tree->object);
 517                 else {
 518                         if (name)
 519                                 error("%.*s: expected %s type, but the object "
 520                                       "dereferences to %s type",
 521                                       namelen, name, typename(expected_type),
 522                                       typename(o->type));
 523                         return NULL;
 524                 }
 525         }
 526 }
 527
 528 static int peel_onion(const char *name, int len, unsigned char *sha1)
 529 {
 530         unsigned char outer[20];
 531         const char *sp;
 532         unsigned int expected_type = 0;
 533         struct object *o;
 534
 535         /*
 536          * "ref^{type}" dereferences ref repeatedly until you cannot
 537          * dereference anymore, or you get an object of given type,
 538          * whichever comes first.  "ref^{}" means just dereference
 539          * tags until you get a non-tag.  "ref^0" is a shorthand for
 540          * "ref^{commit}".  "commit^{tree}" could be used to find the
 541          * top-level tree of the given commit.
 542          */
 543         if (len < 4 || name[len-1] != '}')
 544                 return -1;
 545
 546         for (sp = name + len - 1; name <= sp; sp--) {
 547                 int ch = *sp;
 548                 if (ch == '{' && name < sp && sp[-1] == '^')
 549                         break;
 550         }
 551         if (sp <= name)
 552                 return -1;
 553
 554         sp++; /* beginning of type name, or closing brace for empty */
 555         if (!strncmp(commit_type, sp, 6) && sp[6] == '}')
 556                 expected_type = OBJ_COMMIT;
 557         else if (!strncmp(tree_type, sp, 4) && sp[4] == '}')
 558                 expected_type = OBJ_TREE;
 559         else if (!strncmp(blob_type, sp, 4) && sp[4] == '}')
 560                 expected_type = OBJ_BLOB;
 561         else if (sp[0] == '}')
 562                 expected_type = OBJ_NONE;
 563         else
 564                 return -1;
 565
 566         if (get_sha1_1(name, sp - name - 2, outer))
 567                 return -1;
 568
 569         o = parse_object(outer);
 570         if (!o)
 571                 return -1;
 572         if (!expected_type) {
 573                 o = deref_tag(o, name, sp - name - 2);
 574                 if (!o || (!o->parsed && !parse_object(o->sha1)))
 575                         return -1;
 576                 hashcpy(sha1, o->sha1);
 577         }
 578         else {
 579                 /*
 580                  * At this point, the syntax look correct, so
 581                  * if we do not get the needed object, we should
 582                  * barf.
 583                  */
 584                 o = peel_to_type(name, len, o, expected_type);
 585                 if (o) {
 586                         hashcpy(sha1, o->sha1);
 587                         return 0;
 588                 }
 589                 return -1;
 590         }
 591         return 0;
 592 }
 593
 594 static int get_describe_name(const char *name, int len, unsigned char *sha1)
 595 {
 596         const char *cp;
 597
 598         for (cp = name + len - 1; name + 2 <= cp; cp--) {
 599                 char ch = *cp;
 600                 if (hexval(ch) & ~0377) {
 601                         /* We must be looking at g in "SOMETHING-g"
 602                          * for it to be describe output.
 603                          */
 604                         if (ch == 'g' && cp[-1] == '-') {
 605                                 cp++;
 606                                 len -= cp - name;
 607                                 return get_short_sha1(cp, len, sha1, 1);
 608                         }
 609                 }
 610         }
 611         return -1;
 612 }
 613
 614 static int get_sha1_1(const char *name, int len, unsigned char *sha1)
 615 {
 616         int ret, has_suffix;
 617         const char *cp;
 618
 619         /*
 620          * "name~3" is "name^^^", "name~" is "name~1", and "name^" is "name^1".
 621          */
 622         has_suffix = 0;
 623         for (cp = name + len - 1; name <= cp; cp--) {
 624                 int ch = *cp;
 625                 if ('0' <= ch && ch <= '9')
 626                         continue;
 627                 if (ch == '~' || ch == '^')
 628                         has_suffix = ch;
 629                 break;
 630         }
 631
 632         if (has_suffix) {
 633                 int num = 0;
 634                 int len1 = cp - name;
 635                 cp++;
 636                 while (cp < name + len)
 637                         num = num * 10 + *cp++ - '0';
 638                 if (!num && len1 == len - 1)
 639                         num = 1;
 640                 if (has_suffix == '^')
 641                         return get_parent(name, len1, sha1, num);
 642                 /* else if (has_suffix == '~') -- goes without saying */
 643                 return get_nth_ancestor(name, len1, sha1, num);
 644         }
 645
 646         ret = peel_onion(name, len, sha1);
 647         if (!ret)
 648                 return 0;
 649
 650         ret = get_sha1_basic(name, len, sha1);
 651         if (!ret)
 652                 return 0;
 653
 654         /* It could be describe output that is "SOMETHING-gXXXX" */
 655         ret = get_describe_name(name, len, sha1);
 656         if (!ret)
 657                 return 0;
 658
 659         return get_short_sha1(name, len, sha1, 0);
 660 }
 661
 662 /*
 663  * This interprets names like ':/Initial revision of "git"' by searching
 664  * through history and returning the first commit whose message starts
 665  * with the given string.
 666  *
 667  * For future extension, ':/!' is reserved. If you want to match a message
 668  * beginning with a '!', you have to repeat the exclamation mark.
 669  */
 670 #define ONELINE_SEEN (1u<<20)
 671
 672 static int handle_one_ref(const char *path,
 673                 const unsigned char *sha1, int flag, void *cb_data)
 674 {
 675         struct commit_list **list = cb_data;
 676         struct object *object = parse_object(sha1);
 677         if (!object)
 678                 return 0;
 679         if (object->type == OBJ_TAG) {
 680                 object = deref_tag(object, path, strlen(path));
 681                 if (!object)
 682                         return 0;
 683         }
 684         if (object->type != OBJ_COMMIT)
 685                 return 0;
 686         insert_by_date((struct commit *)object, list);
 687         object->flags |= ONELINE_SEEN;
 688         return 0;
 689 }
 690
 691 static int get_sha1_oneline(const char *prefix, unsigned char *sha1)
 692 {
 693         struct commit_list *list = NULL, *backup = NULL, *l;
 694         int retval = -1;
 695         char *temp_commit_buffer = NULL;
 696
 697         if (prefix[0] == '!') {
 698                 if (prefix[1] != '!')
 699                         die ("Invalid search pattern: %s", prefix);
 700                 prefix++;
 701         }
 702         for_each_ref(handle_one_ref, &list);
 703         for (l = list; l; l = l->next)
 704                 commit_list_insert(l->item, &backup);
 705         while (list) {
 706                 char *p;
 707                 struct commit *commit;
 708                 enum object_type type;
 709                 unsigned long size;
 710
 711                 commit = pop_most_recent_commit(&list, ONELINE_SEEN);
 712                 if (!parse_object(commit->object.sha1))
 713                         continue;
 714                 free(temp_commit_buffer);
 715                 if (commit->buffer)
 716                         p = commit->buffer;
 717                 else {
 718                         p = read_sha1_file(commit->object.sha1, &type, &size);
 719                         if (!p)
 720                                 continue;
 721                         temp_commit_buffer = p;
 722                 }
 723                 if (!(p = strstr(p, "\n\n")))
 724                         continue;
 725                 if (!prefixcmp(p + 2, prefix)) {
 726                         hashcpy(sha1, commit->object.sha1);
 727                         retval = 0;
 728                         break;
 729                 }
 730         }
 731         free(temp_commit_buffer);
 732         free_commit_list(list);
 733         for (l = backup; l; l = l->next)
 734                 clear_commit_marks(l->item, ONELINE_SEEN);
 735         return retval;
 736 }
 737
 738 struct grab_nth_branch_switch_cbdata {
 739         long cnt, alloc;
 740         struct strbuf *buf;
 741 };
 742
 743 static int grab_nth_branch_switch(unsigned char *osha1, unsigned char *nsha1,
 744                                   const char *email, unsigned long timestamp, int tz,
 745                                   const char *message, void *cb_data)
 746 {
 747         struct grab_nth_branch_switch_cbdata *cb = cb_data;
 748         const char *match = NULL, *target = NULL;
 749         size_t len;
 750         int nth;
 751
 752         if (!prefixcmp(message, "checkout: moving from ")) {
 753                 match = message + strlen("checkout: moving from ");
 754                 target = strstr(match, " to ");
 755         }
 756
 757         if (!match || !target)
 758                 return 0;
 759
 760         len = target - match;
 761         nth = cb->cnt++ % cb->alloc;
 762         strbuf_reset(&cb->buf[nth]);
 763         strbuf_add(&cb->buf[nth], match, len);
 764         return 0;
 765 }
 766
 767 /*
 768  * Parse @{-N} syntax, return the number of characters parsed
 769  * if successful; otherwise signal an error with negative value.
 770  */
 771 static int interpret_nth_prior_checkout(const char *name, struct strbuf *buf)
 772 {
 773         long nth;
 774         int i, retval;
 775         struct grab_nth_branch_switch_cbdata cb;
 776         const char *brace;
 777         char *num_end;
 778
 779         if (name[0] != '@' || name[1] != '{' || name[2] != '-')
 780                 return -1;
 781         brace = strchr(name, '}');
 782         if (!brace)
 783                 return -1;
 784         nth = strtol(name+3, &num_end, 10);
 785         if (num_end != brace)
 786                 return -1;
 787         if (nth <= 0)
 788                 return -1;
 789         cb.alloc = nth;
 790         cb.buf = xmalloc(nth * sizeof(struct strbuf));
 791         for (i = 0; i < nth; i++)
 792                 strbuf_init(&cb.buf[i], 20);
 793         cb.cnt = 0;
 794         retval = 0;
 795         for_each_recent_reflog_ent("HEAD", grab_nth_branch_switch, 40960, &cb);
 796         if (cb.cnt < nth) {
 797                 cb.cnt = 0;
 798                 for_each_reflog_ent("HEAD", grab_nth_branch_switch, &cb);
 799         }
 800         if (cb.cnt < nth)
 801                 goto release_return;
 802         i = cb.cnt % nth;
 803         strbuf_reset(buf);
 804         strbuf_add(buf, cb.buf[i].buf, cb.buf[i].len);
 805         retval = brace-name+1;
 806
 807 release_return:
 808         for (i = 0; i < nth; i++)
 809                 strbuf_release(&cb.buf[i]);
 810         free(cb.buf);
 811
 812         return retval;
 813 }
 814
 815 int get_sha1_mb(const char *name, unsigned char *sha1)
 816 {
 817         struct commit *one, *two;
 818         struct commit_list *mbs;
 819         unsigned char sha1_tmp[20];
 820         const char *dots;
 821         int st;
 822
 823         dots = strstr(name, "...");
 824         if (!dots)
 825                 return get_sha1(name, sha1);
 826         if (dots == name)
 827                 st = get_sha1("HEAD", sha1_tmp);
 828         else {
 829                 struct strbuf sb;
 830                 strbuf_init(&sb, dots - name);
 831                 strbuf_add(&sb, name, dots - name);
 832                 st = get_sha1(sb.buf, sha1_tmp);
 833                 strbuf_release(&sb);
 834         }
 835         if (st)
 836                 return st;
 837         one = lookup_commit_reference_gently(sha1_tmp, 0);
 838         if (!one)
 839                 return -1;
 840
 841         if (get_sha1(dots[3] ? (dots + 3) : "HEAD", sha1_tmp))
 842                 return -1;
 843         two = lookup_commit_reference_gently(sha1_tmp, 0);
 844         if (!two)
 845                 return -1;
 846         mbs = get_merge_bases(one, two, 1);
 847         if (!mbs || mbs->next)
 848                 st = -1;
 849         else {
 850                 st = 0;
 851                 hashcpy(sha1, mbs->item->object.sha1);
 852         }
 853         free_commit_list(mbs);
 854         return st;
 855 }
 856
 857 /*
 858  * This reads short-hand syntax that not only evaluates to a commit
 859  * object name, but also can act as if the end user spelled the name
 860  * of the branch from the command line.
 861  *
 862  * - "@{-N}" finds the name of the Nth previous branch we were on, and
 863  *   places the name of the branch in the given buf and returns the
 864  *   number of characters parsed if successful.
 865  *
 866  * - "<branch>@{upstream}" finds the name of the other ref that
 867  *   <branch> is configured to merge with (missing <branch> defaults
 868  *   to the current branch), and places the name of the branch in the
 869  *   given buf and returns the number of characters parsed if
 870  *   successful.
 871  *
 872  * If the input is not of the accepted format, it returns a negative
 873  * number to signal an error.
 874  *
 875  * If the input was ok but there are not N branch switches in the
 876  * reflog, it returns 0.
 877  */
 878 int interpret_branch_name(const char *name, struct strbuf *buf)
 879 {
 880         char *cp;
 881         struct branch *upstream;
 882         int namelen = strlen(name);
 883         int len = interpret_nth_prior_checkout(name, buf);
 884         int tmp_len;
 885
 886         if (!len)
 887                 return len; /* syntax Ok, not enough switches */
 888         if (0 < len && len == namelen)
 889                 return len; /* consumed all */
 890         else if (0 < len) {
 891                 /* we have extra data, which might need further processing */
 892                 struct strbuf tmp = STRBUF_INIT;
 893                 int used = buf->len;
 894                 int ret;
 895
 896                 strbuf_add(buf, name + len, namelen - len);
 897                 ret = interpret_branch_name(buf->buf, &tmp);
 898                 /* that data was not interpreted, remove our cruft */
 899                 if (ret < 0) {
 900                         strbuf_setlen(buf, used);
 901                         return len;
 902                 }
 903                 strbuf_reset(buf);
 904                 strbuf_addbuf(buf, &tmp);
 905                 strbuf_release(&tmp);
 906                 /* tweak for size of {-N} versus expanded ref name */
 907                 return ret - used + len;
 908         }
 909
 910         cp = strchr(name, '@');
 911         if (!cp)
 912                 return -1;
 913         tmp_len = upstream_mark(cp, namelen - (cp - name));
 914         if (!tmp_len)
 915                 return -1;
 916         len = cp + tmp_len - name;
 917         cp = xstrndup(name, cp - name);
 918         upstream = branch_get(*cp ? cp : NULL);
 919         if (!upstream
 920             || !upstream->merge
 921             || !upstream->merge[0]->dst)
 922                 return error("No upstream branch found for '%s'", cp);
 923         free(cp);
 924         cp = shorten_unambiguous_ref(upstream->merge[0]->dst, 0);
 925         strbuf_reset(buf);
 926         strbuf_addstr(buf, cp);
 927         free(cp);
 928         return len;
 929 }
 930
 931 /*
 932  * This is like "get_sha1_basic()", except it allows "sha1 expressions",
 933  * notably "xyz^" for "parent of xyz"
 934  */
 935 int get_sha1(const char *name, unsigned char *sha1)
 936 {
 937         unsigned unused;
 938         return get_sha1_with_mode(name, sha1, &unused);
 939 }
 940
 941 /* Must be called only when object_name:filename doesn't exist. */
 942 static void diagnose_invalid_sha1_path(const char *prefix,
 943                                        const char *filename,
 944                                        const unsigned char *tree_sha1,
 945                                        const char *object_name)
 946 {
 947         struct stat st;
 948         unsigned char sha1[20];
 949         unsigned mode;
 950
 951         if (!prefix)
 952                 prefix = "";
 953
 954         if (!lstat(filename, &st))
 955                 die("Path '%s' exists on disk, but not in '%s'.",
 956                     filename, object_name);
 957         if (errno == ENOENT || errno == ENOTDIR) {
 958                 char *fullname = xmalloc(strlen(filename)
 959                                              + strlen(prefix) + 1);
 960                 strcpy(fullname, prefix);
 961                 strcat(fullname, filename);
 962
 963                 if (!get_tree_entry(tree_sha1, fullname,
 964                                     sha1, &mode)) {
 965                         die("Path '%s' exists, but not '%s'.\n"
 966                             "Did you mean '%s:%s'?",
 967                             fullname,
 968                             filename,
 969                             object_name,
 970                             fullname);
 971                 }
 972                 die("Path '%s' does not exist in '%s'",
 973                     filename, object_name);
 974         }
 975 }
 976
 977 /* Must be called only when :stage:filename doesn't exist. */
 978 static void diagnose_invalid_index_path(int stage,
 979                                         const char *prefix,
 980                                         const char *filename)
 981 {
 982         struct stat st;
 983         struct cache_entry *ce;
 984         int pos;
 985         unsigned namelen = strlen(filename);
 986         unsigned fullnamelen;
 987         char *fullname;
 988
 989         if (!prefix)
 990                 prefix = "";
 991
 992         /* Wrong stage number? */
 993         pos = cache_name_pos(filename, namelen);
 994         if (pos < 0)
 995                 pos = -pos - 1;
 996         if (pos < active_nr) {
 997                 ce = active_cache[pos];
 998                 if (ce_namelen(ce) == namelen &&
 999                     !memcmp(ce->name, filename, namelen))
1000                         die("Path '%s' is in the index, but not at stage %d.\n"
1001                             "Did you mean ':%d:%s'?",
1002                             filename, stage,
1003                             ce_stage(ce), filename);
1004         }
1005
1006         /* Confusion between relative and absolute filenames? */
1007         fullnamelen = namelen + strlen(prefix);
1008         fullname = xmalloc(fullnamelen + 1);
1009         strcpy(fullname, prefix);
1010         strcat(fullname, filename);
1011         pos = cache_name_pos(fullname, fullnamelen);
1012         if (pos < 0)
1013                 pos = -pos - 1;
1014         if (pos < active_nr) {
1015                 ce = active_cache[pos];
1016                 if (ce_namelen(ce) == fullnamelen &&
1017                     !memcmp(ce->name, fullname, fullnamelen))
1018                         die("Path '%s' is in the index, but not '%s'.\n"
1019                             "Did you mean ':%d:%s'?",
1020                             fullname, filename,
1021                             ce_stage(ce), fullname);
1022         }
1023
1024         if (!lstat(filename, &st))
1025                 die("Path '%s' exists on disk, but not in the index.", filename);
1026         if (errno == ENOENT || errno == ENOTDIR)
1027                 die("Path '%s' does not exist (neither on disk nor in the index).",
1028                     filename);
1029
1030         free(fullname);
1031 }
1032
1033
1034 int get_sha1_with_mode_1(const char *name, unsigned char *sha1, unsigned *mode, int gently, const char *prefix)
1035 {
1036         int ret, bracket_depth;
1037         int namelen = strlen(name);
1038         const char *cp;
1039
1040         *mode = S_IFINVALID;
1041         ret = get_sha1_1(name, namelen, sha1);
1042         if (!ret)
1043                 return ret;
1044         /* sha1:path --> object name of path in ent sha1
1045          * :path -> object name of path in index
1046          * :[0-3]:path -> object name of path in index at stage
1047          */
1048         if (name[0] == ':') {
1049                 int stage = 0;
1050                 struct cache_entry *ce;
1051                 int pos;
1052                 if (namelen > 2 && name[1] == '/')
1053                         return get_sha1_oneline(name + 2, sha1);
1054                 if (namelen < 3 ||
1055                     name[2] != ':' ||
1056                     name[1] < '0' || '3' < name[1])
1057                         cp = name + 1;
1058                 else {
1059                         stage = name[1] - '0';
1060                         cp = name + 3;
1061                 }
1062                 namelen = namelen - (cp - name);
1063                 if (!active_cache)
1064                         read_cache();
1065                 pos = cache_name_pos(cp, namelen);
1066                 if (pos < 0)
1067                         pos = -pos - 1;
1068                 while (pos < active_nr) {
1069                         ce = active_cache[pos];
1070                         if (ce_namelen(ce) != namelen ||
1071                             memcmp(ce->name, cp, namelen))
1072                                 break;
1073                         if (ce_stage(ce) == stage) {
1074                                 hashcpy(sha1, ce->sha1);
1075                                 *mode = ce->ce_mode;
1076                                 return 0;
1077                         }
1078                         pos++;
1079                 }
1080                 if (!gently)
1081                         diagnose_invalid_index_path(stage, prefix, cp);
1082                 return -1;
1083         }
1084         for (cp = name, bracket_depth = 0; *cp; cp++) {
1085                 if (*cp == '{')
1086                         bracket_depth++;
1087                 else if (bracket_depth && *cp == '}')
1088                         bracket_depth--;
1089                 else if (!bracket_depth && *cp == ':')
1090                         break;
1091         }
1092         if (*cp == ':') {
1093                 unsigned char tree_sha1[20];
1094                 char *object_name = NULL;
1095                 if (!gently) {
1096                         object_name = xmalloc(cp-name+1);
1097                         strncpy(object_name, name, cp-name);
1098                         object_name[cp-name] = '\0';
1099                 }
1100                 if (!get_sha1_1(name, cp-name, tree_sha1)) {
1101                         const char *filename = cp+1;
1102                         ret = get_tree_entry(tree_sha1, filename, sha1, mode);
1103                         if (!gently) {
1104                                 diagnose_invalid_sha1_path(prefix, filename,
1105                                                            tree_sha1, object_name);
1106                                 free(object_name);
1107                         }
1108                         return ret;
1109                 } else {
1110                         if (!gently)
1111                                 die("Invalid object name '%s'.", object_name);
1112                 }
1113         }
1114         return ret;
1115 }