upload-pack.con commit Merge branch 'pw/fast-import-dataref-parsing' into maint (b9f0662)
   1#include "cache.h"
   2#include "refs.h"
   3#include "pkt-line.h"
   4#include "sideband.h"
   5#include "tag.h"
   6#include "object.h"
   7#include "commit.h"
   8#include "exec_cmd.h"
   9#include "diff.h"
  10#include "revision.h"
  11#include "list-objects.h"
  12#include "run-command.h"
  13#include "sigchain.h"
  14
  15static const char upload_pack_usage[] = "git upload-pack [--strict] [--timeout=<n>] <dir>";
  16
  17/* bits #0..7 in revision.h, #8..10 in commit.c */
  18#define THEY_HAVE       (1u << 11)
  19#define OUR_REF         (1u << 12)
  20#define WANTED          (1u << 13)
  21#define COMMON_KNOWN    (1u << 14)
  22#define REACHABLE       (1u << 15)
  23
  24#define SHALLOW         (1u << 16)
  25#define NOT_SHALLOW     (1u << 17)
  26#define CLIENT_SHALLOW  (1u << 18)
  27
  28static unsigned long oldest_have;
  29
  30static int multi_ack, nr_our_refs;
  31static int no_done;
  32static int use_thin_pack, use_ofs_delta, use_include_tag;
  33static int no_progress, daemon_mode;
  34static int shallow_nr;
  35static struct object_array have_obj;
  36static struct object_array want_obj;
  37static struct object_array extra_edge_obj;
  38static unsigned int timeout;
  39/* 0 for no sideband,
  40 * otherwise maximum packet size (up to 65520 bytes).
  41 */
  42static int use_sideband;
  43static int debug_fd;
  44static int advertise_refs;
  45static int stateless_rpc;
  46
  47static void reset_timeout(void)
  48{
  49        alarm(timeout);
  50}
  51
  52static int strip(char *line, int len)
  53{
  54        if (len && line[len-1] == '\n')
  55                line[--len] = 0;
  56        return len;
  57}
  58
  59static ssize_t send_client_data(int fd, const char *data, ssize_t sz)
  60{
  61        if (use_sideband)
  62                return send_sideband(1, fd, data, sz, use_sideband);
  63        if (fd == 3)
  64                /* emergency quit */
  65                fd = 2;
  66        if (fd == 2) {
  67                /* XXX: are we happy to lose stuff here? */
  68                xwrite(fd, data, sz);
  69                return sz;
  70        }
  71        return safe_write(fd, data, sz);
  72}
  73
  74static FILE *pack_pipe = NULL;
  75static void show_commit(struct commit *commit, void *data)
  76{
  77        if (commit->object.flags & BOUNDARY)
  78                fputc('-', pack_pipe);
  79        if (fputs(sha1_to_hex(commit->object.sha1), pack_pipe) < 0)
  80                die("broken output pipe");
  81        fputc('\n', pack_pipe);
  82        fflush(pack_pipe);
  83        free(commit->buffer);
  84        commit->buffer = NULL;
  85}
  86
  87static void show_object(struct object *obj,
  88                        const struct name_path *path, const char *component,
  89                        void *cb_data)
  90{
  91        show_object_with_name(pack_pipe, obj, path, component);
  92}
  93
  94static void show_edge(struct commit *commit)
  95{
  96        fprintf(pack_pipe, "-%s\n", sha1_to_hex(commit->object.sha1));
  97}
  98
  99static int do_rev_list(int in, int out, void *user_data)
 100{
 101        int i;
 102        struct rev_info revs;
 103
 104        pack_pipe = xfdopen(out, "w");
 105        init_revisions(&revs, NULL);
 106        revs.tag_objects = 1;
 107        revs.tree_objects = 1;
 108        revs.blob_objects = 1;
 109        if (use_thin_pack)
 110                revs.edge_hint = 1;
 111
 112        for (i = 0; i < want_obj.nr; i++) {
 113                struct object *o = want_obj.objects[i].item;
 114                /* why??? */
 115                o->flags &= ~UNINTERESTING;
 116                add_pending_object(&revs, o, NULL);
 117        }
 118        for (i = 0; i < have_obj.nr; i++) {
 119                struct object *o = have_obj.objects[i].item;
 120                o->flags |= UNINTERESTING;
 121                add_pending_object(&revs, o, NULL);
 122        }
 123        setup_revisions(0, NULL, &revs, NULL);
 124        if (prepare_revision_walk(&revs))
 125                die("revision walk setup failed");
 126        mark_edges_uninteresting(revs.commits, &revs, show_edge);
 127        if (use_thin_pack)
 128                for (i = 0; i < extra_edge_obj.nr; i++)
 129                        fprintf(pack_pipe, "-%s\n", sha1_to_hex(
 130                                        extra_edge_obj.objects[i].item->sha1));
 131        traverse_commit_list(&revs, show_commit, show_object, NULL);
 132        fflush(pack_pipe);
 133        fclose(pack_pipe);
 134        return 0;
 135}
 136
 137static void create_pack_file(void)
 138{
 139        struct async rev_list;
 140        struct child_process pack_objects;
 141        int create_full_pack = (nr_our_refs == want_obj.nr && !have_obj.nr);
 142        char data[8193], progress[128];
 143        char abort_msg[] = "aborting due to possible repository "
 144                "corruption on the remote side.";
 145        int buffered = -1;
 146        ssize_t sz;
 147        const char *argv[10];
 148        int arg = 0;
 149
 150        argv[arg++] = "pack-objects";
 151        if (!shallow_nr) {
 152                argv[arg++] = "--revs";
 153                if (create_full_pack)
 154                        argv[arg++] = "--all";
 155                else if (use_thin_pack)
 156                        argv[arg++] = "--thin";
 157        }
 158
 159        argv[arg++] = "--stdout";
 160        if (!no_progress)
 161                argv[arg++] = "--progress";
 162        if (use_ofs_delta)
 163                argv[arg++] = "--delta-base-offset";
 164        if (use_include_tag)
 165                argv[arg++] = "--include-tag";
 166        argv[arg++] = NULL;
 167
 168        memset(&pack_objects, 0, sizeof(pack_objects));
 169        pack_objects.in = -1;
 170        pack_objects.out = -1;
 171        pack_objects.err = -1;
 172        pack_objects.git_cmd = 1;
 173        pack_objects.argv = argv;
 174
 175        if (start_command(&pack_objects))
 176                die("git upload-pack: unable to fork git-pack-objects");
 177
 178        if (shallow_nr) {
 179                memset(&rev_list, 0, sizeof(rev_list));
 180                rev_list.proc = do_rev_list;
 181                rev_list.out = pack_objects.in;
 182                if (start_async(&rev_list))
 183                        die("git upload-pack: unable to fork git-rev-list");
 184        }
 185        else {
 186                FILE *pipe_fd = xfdopen(pack_objects.in, "w");
 187                if (!create_full_pack) {
 188                        int i;
 189                        for (i = 0; i < want_obj.nr; i++)
 190                                fprintf(pipe_fd, "%s\n", sha1_to_hex(want_obj.objects[i].item->sha1));
 191                        fprintf(pipe_fd, "--not\n");
 192                        for (i = 0; i < have_obj.nr; i++)
 193                                fprintf(pipe_fd, "%s\n", sha1_to_hex(have_obj.objects[i].item->sha1));
 194                }
 195
 196                fprintf(pipe_fd, "\n");
 197                fflush(pipe_fd);
 198                fclose(pipe_fd);
 199        }
 200
 201
 202        /* We read from pack_objects.err to capture stderr output for
 203         * progress bar, and pack_objects.out to capture the pack data.
 204         */
 205
 206        while (1) {
 207                struct pollfd pfd[2];
 208                int pe, pu, pollsize;
 209
 210                reset_timeout();
 211
 212                pollsize = 0;
 213                pe = pu = -1;
 214
 215                if (0 <= pack_objects.out) {
 216                        pfd[pollsize].fd = pack_objects.out;
 217                        pfd[pollsize].events = POLLIN;
 218                        pu = pollsize;
 219                        pollsize++;
 220                }
 221                if (0 <= pack_objects.err) {
 222                        pfd[pollsize].fd = pack_objects.err;
 223                        pfd[pollsize].events = POLLIN;
 224                        pe = pollsize;
 225                        pollsize++;
 226                }
 227
 228                if (!pollsize)
 229                        break;
 230
 231                if (poll(pfd, pollsize, -1) < 0) {
 232                        if (errno != EINTR) {
 233                                error("poll failed, resuming: %s",
 234                                      strerror(errno));
 235                                sleep(1);
 236                        }
 237                        continue;
 238                }
 239                if (0 <= pe && (pfd[pe].revents & (POLLIN|POLLHUP))) {
 240                        /* Status ready; we ship that in the side-band
 241                         * or dump to the standard error.
 242                         */
 243                        sz = xread(pack_objects.err, progress,
 244                                  sizeof(progress));
 245                        if (0 < sz)
 246                                send_client_data(2, progress, sz);
 247                        else if (sz == 0) {
 248                                close(pack_objects.err);
 249                                pack_objects.err = -1;
 250                        }
 251                        else
 252                                goto fail;
 253                        /* give priority to status messages */
 254                        continue;
 255                }
 256                if (0 <= pu && (pfd[pu].revents & (POLLIN|POLLHUP))) {
 257                        /* Data ready; we keep the last byte to ourselves
 258                         * in case we detect broken rev-list, so that we
 259                         * can leave the stream corrupted.  This is
 260                         * unfortunate -- unpack-objects would happily
 261                         * accept a valid packdata with trailing garbage,
 262                         * so appending garbage after we pass all the
 263                         * pack data is not good enough to signal
 264                         * breakage to downstream.
 265                         */
 266                        char *cp = data;
 267                        ssize_t outsz = 0;
 268                        if (0 <= buffered) {
 269                                *cp++ = buffered;
 270                                outsz++;
 271                        }
 272                        sz = xread(pack_objects.out, cp,
 273                                  sizeof(data) - outsz);
 274                        if (0 < sz)
 275                                ;
 276                        else if (sz == 0) {
 277                                close(pack_objects.out);
 278                                pack_objects.out = -1;
 279                        }
 280                        else
 281                                goto fail;
 282                        sz += outsz;
 283                        if (1 < sz) {
 284                                buffered = data[sz-1] & 0xFF;
 285                                sz--;
 286                        }
 287                        else
 288                                buffered = -1;
 289                        sz = send_client_data(1, data, sz);
 290                        if (sz < 0)
 291                                goto fail;
 292                }
 293        }
 294
 295        if (finish_command(&pack_objects)) {
 296                error("git upload-pack: git-pack-objects died with error.");
 297                goto fail;
 298        }
 299        if (shallow_nr && finish_async(&rev_list))
 300                goto fail;      /* error was already reported */
 301
 302        /* flush the data */
 303        if (0 <= buffered) {
 304                data[0] = buffered;
 305                sz = send_client_data(1, data, 1);
 306                if (sz < 0)
 307                        goto fail;
 308                fprintf(stderr, "flushed.\n");
 309        }
 310        if (use_sideband)
 311                packet_flush(1);
 312        return;
 313
 314 fail:
 315        send_client_data(3, abort_msg, sizeof(abort_msg));
 316        die("git upload-pack: %s", abort_msg);
 317}
 318
 319static int got_sha1(char *hex, unsigned char *sha1)
 320{
 321        struct object *o;
 322        int we_knew_they_have = 0;
 323
 324        if (get_sha1_hex(hex, sha1))
 325                die("git upload-pack: expected SHA1 object, got '%s'", hex);
 326        if (!has_sha1_file(sha1))
 327                return -1;
 328
 329        o = lookup_object(sha1);
 330        if (!(o && o->parsed))
 331                o = parse_object(sha1);
 332        if (!o)
 333                die("oops (%s)", sha1_to_hex(sha1));
 334        if (o->type == OBJ_COMMIT) {
 335                struct commit_list *parents;
 336                struct commit *commit = (struct commit *)o;
 337                if (o->flags & THEY_HAVE)
 338                        we_knew_they_have = 1;
 339                else
 340                        o->flags |= THEY_HAVE;
 341                if (!oldest_have || (commit->date < oldest_have))
 342                        oldest_have = commit->date;
 343                for (parents = commit->parents;
 344                     parents;
 345                     parents = parents->next)
 346                        parents->item->object.flags |= THEY_HAVE;
 347        }
 348        if (!we_knew_they_have) {
 349                add_object_array(o, NULL, &have_obj);
 350                return 1;
 351        }
 352        return 0;
 353}
 354
 355static int reachable(struct commit *want)
 356{
 357        struct commit_list *work = NULL;
 358
 359        commit_list_insert_by_date(want, &work);
 360        while (work) {
 361                struct commit_list *list = work->next;
 362                struct commit *commit = work->item;
 363                free(work);
 364                work = list;
 365
 366                if (commit->object.flags & THEY_HAVE) {
 367                        want->object.flags |= COMMON_KNOWN;
 368                        break;
 369                }
 370                if (!commit->object.parsed)
 371                        parse_object(commit->object.sha1);
 372                if (commit->object.flags & REACHABLE)
 373                        continue;
 374                commit->object.flags |= REACHABLE;
 375                if (commit->date < oldest_have)
 376                        continue;
 377                for (list = commit->parents; list; list = list->next) {
 378                        struct commit *parent = list->item;
 379                        if (!(parent->object.flags & REACHABLE))
 380                                commit_list_insert_by_date(parent, &work);
 381                }
 382        }
 383        want->object.flags |= REACHABLE;
 384        clear_commit_marks(want, REACHABLE);
 385        free_commit_list(work);
 386        return (want->object.flags & COMMON_KNOWN);
 387}
 388
 389static int ok_to_give_up(void)
 390{
 391        int i;
 392
 393        if (!have_obj.nr)
 394                return 0;
 395
 396        for (i = 0; i < want_obj.nr; i++) {
 397                struct object *want = want_obj.objects[i].item;
 398
 399                if (want->flags & COMMON_KNOWN)
 400                        continue;
 401                want = deref_tag(want, "a want line", 0);
 402                if (!want || want->type != OBJ_COMMIT) {
 403                        /* no way to tell if this is reachable by
 404                         * looking at the ancestry chain alone, so
 405                         * leave a note to ourselves not to worry about
 406                         * this object anymore.
 407                         */
 408                        want_obj.objects[i].item->flags |= COMMON_KNOWN;
 409                        continue;
 410                }
 411                if (!reachable((struct commit *)want))
 412                        return 0;
 413        }
 414        return 1;
 415}
 416
 417static int get_common_commits(void)
 418{
 419        static char line[1000];
 420        unsigned char sha1[20];
 421        char last_hex[41];
 422        int got_common = 0;
 423        int got_other = 0;
 424        int sent_ready = 0;
 425
 426        save_commit_buffer = 0;
 427
 428        for (;;) {
 429                int len = packet_read_line(0, line, sizeof(line));
 430                reset_timeout();
 431
 432                if (!len) {
 433                        if (multi_ack == 2 && got_common
 434                            && !got_other && ok_to_give_up()) {
 435                                sent_ready = 1;
 436                                packet_write(1, "ACK %s ready\n", last_hex);
 437                        }
 438                        if (have_obj.nr == 0 || multi_ack)
 439                                packet_write(1, "NAK\n");
 440
 441                        if (no_done && sent_ready) {
 442                                packet_write(1, "ACK %s\n", last_hex);
 443                                return 0;
 444                        }
 445                        if (stateless_rpc)
 446                                exit(0);
 447                        got_common = 0;
 448                        got_other = 0;
 449                        continue;
 450                }
 451                strip(line, len);
 452                if (!prefixcmp(line, "have ")) {
 453                        switch (got_sha1(line+5, sha1)) {
 454                        case -1: /* they have what we do not */
 455                                got_other = 1;
 456                                if (multi_ack && ok_to_give_up()) {
 457                                        const char *hex = sha1_to_hex(sha1);
 458                                        if (multi_ack == 2) {
 459                                                sent_ready = 1;
 460                                                packet_write(1, "ACK %s ready\n", hex);
 461                                        } else
 462                                                packet_write(1, "ACK %s continue\n", hex);
 463                                }
 464                                break;
 465                        default:
 466                                got_common = 1;
 467                                memcpy(last_hex, sha1_to_hex(sha1), 41);
 468                                if (multi_ack == 2)
 469                                        packet_write(1, "ACK %s common\n", last_hex);
 470                                else if (multi_ack)
 471                                        packet_write(1, "ACK %s continue\n", last_hex);
 472                                else if (have_obj.nr == 1)
 473                                        packet_write(1, "ACK %s\n", last_hex);
 474                                break;
 475                        }
 476                        continue;
 477                }
 478                if (!strcmp(line, "done")) {
 479                        if (have_obj.nr > 0) {
 480                                if (multi_ack)
 481                                        packet_write(1, "ACK %s\n", last_hex);
 482                                return 0;
 483                        }
 484                        packet_write(1, "NAK\n");
 485                        return -1;
 486                }
 487                die("git upload-pack: expected SHA1 list, got '%s'", line);
 488        }
 489}
 490
 491static void check_non_tip(void)
 492{
 493        static const char *argv[] = {
 494                "rev-list", "--stdin", NULL,
 495        };
 496        static struct child_process cmd;
 497        struct object *o;
 498        char namebuf[42]; /* ^ + SHA-1 + LF */
 499        int i;
 500
 501        /* In the normal in-process case non-tip request can never happen */
 502        if (!stateless_rpc)
 503                goto error;
 504
 505        cmd.argv = argv;
 506        cmd.git_cmd = 1;
 507        cmd.no_stderr = 1;
 508        cmd.in = -1;
 509        cmd.out = -1;
 510
 511        if (start_command(&cmd))
 512                goto error;
 513
 514        /*
 515         * If rev-list --stdin encounters an unknown commit, it
 516         * terminates, which will cause SIGPIPE in the write loop
 517         * below.
 518         */
 519        sigchain_push(SIGPIPE, SIG_IGN);
 520
 521        namebuf[0] = '^';
 522        namebuf[41] = '\n';
 523        for (i = get_max_object_index(); 0 < i; ) {
 524                o = get_indexed_object(--i);
 525                if (!o)
 526                        continue;
 527                if (!(o->flags & OUR_REF))
 528                        continue;
 529                memcpy(namebuf + 1, sha1_to_hex(o->sha1), 40);
 530                if (write_in_full(cmd.in, namebuf, 42) < 0)
 531                        goto error;
 532        }
 533        namebuf[40] = '\n';
 534        for (i = 0; i < want_obj.nr; i++) {
 535                o = want_obj.objects[i].item;
 536                if (o->flags & OUR_REF)
 537                        continue;
 538                memcpy(namebuf, sha1_to_hex(o->sha1), 40);
 539                if (write_in_full(cmd.in, namebuf, 41) < 0)
 540                        goto error;
 541        }
 542        close(cmd.in);
 543
 544        sigchain_pop(SIGPIPE);
 545
 546        /*
 547         * The commits out of the rev-list are not ancestors of
 548         * our ref.
 549         */
 550        i = read_in_full(cmd.out, namebuf, 1);
 551        if (i)
 552                goto error;
 553        close(cmd.out);
 554
 555        /*
 556         * rev-list may have died by encountering a bad commit
 557         * in the history, in which case we do want to bail out
 558         * even when it showed no commit.
 559         */
 560        if (finish_command(&cmd))
 561                goto error;
 562
 563        /* All the non-tip ones are ancestors of what we advertised */
 564        return;
 565
 566error:
 567        /* Pick one of them (we know there at least is one) */
 568        for (i = 0; i < want_obj.nr; i++) {
 569                o = want_obj.objects[i].item;
 570                if (!(o->flags & OUR_REF))
 571                        die("git upload-pack: not our ref %s",
 572                            sha1_to_hex(o->sha1));
 573        }
 574}
 575
 576static void receive_needs(void)
 577{
 578        struct object_array shallows = OBJECT_ARRAY_INIT;
 579        static char line[1000];
 580        int len, depth = 0;
 581        int has_non_tip = 0;
 582
 583        shallow_nr = 0;
 584        if (debug_fd)
 585                write_str_in_full(debug_fd, "#S\n");
 586        for (;;) {
 587                struct object *o;
 588                const char *features;
 589                unsigned char sha1_buf[20];
 590                len = packet_read_line(0, line, sizeof(line));
 591                reset_timeout();
 592                if (!len)
 593                        break;
 594                if (debug_fd)
 595                        write_in_full(debug_fd, line, len);
 596
 597                if (!prefixcmp(line, "shallow ")) {
 598                        unsigned char sha1[20];
 599                        struct object *object;
 600                        if (get_sha1(line + 8, sha1))
 601                                die("invalid shallow line: %s", line);
 602                        object = parse_object(sha1);
 603                        if (!object)
 604                                die("did not find object for %s", line);
 605                        object->flags |= CLIENT_SHALLOW;
 606                        add_object_array(object, NULL, &shallows);
 607                        continue;
 608                }
 609                if (!prefixcmp(line, "deepen ")) {
 610                        char *end;
 611                        depth = strtol(line + 7, &end, 0);
 612                        if (end == line + 7 || depth <= 0)
 613                                die("Invalid deepen: %s", line);
 614                        continue;
 615                }
 616                if (prefixcmp(line, "want ") ||
 617                    get_sha1_hex(line+5, sha1_buf))
 618                        die("git upload-pack: protocol error, "
 619                            "expected to get sha, not '%s'", line);
 620
 621                features = line + 45;
 622
 623                if (parse_feature_request(features, "multi_ack_detailed"))
 624                        multi_ack = 2;
 625                else if (parse_feature_request(features, "multi_ack"))
 626                        multi_ack = 1;
 627                if (parse_feature_request(features, "no-done"))
 628                        no_done = 1;
 629                if (parse_feature_request(features, "thin-pack"))
 630                        use_thin_pack = 1;
 631                if (parse_feature_request(features, "ofs-delta"))
 632                        use_ofs_delta = 1;
 633                if (parse_feature_request(features, "side-band-64k"))
 634                        use_sideband = LARGE_PACKET_MAX;
 635                else if (parse_feature_request(features, "side-band"))
 636                        use_sideband = DEFAULT_PACKET_MAX;
 637                if (parse_feature_request(features, "no-progress"))
 638                        no_progress = 1;
 639                if (parse_feature_request(features, "include-tag"))
 640                        use_include_tag = 1;
 641
 642                o = lookup_object(sha1_buf);
 643                if (!o)
 644                        die("git upload-pack: not our ref %s",
 645                            sha1_to_hex(sha1_buf));
 646                if (!(o->flags & WANTED)) {
 647                        o->flags |= WANTED;
 648                        if (!(o->flags & OUR_REF))
 649                                has_non_tip = 1;
 650                        add_object_array(o, NULL, &want_obj);
 651                }
 652        }
 653        if (debug_fd)
 654                write_str_in_full(debug_fd, "#E\n");
 655
 656        /*
 657         * We have sent all our refs already, and the other end
 658         * should have chosen out of them. When we are operating
 659         * in the stateless RPC mode, however, their choice may
 660         * have been based on the set of older refs advertised
 661         * by another process that handled the initial request.
 662         */
 663        if (has_non_tip)
 664                check_non_tip();
 665
 666        if (!use_sideband && daemon_mode)
 667                no_progress = 1;
 668
 669        if (depth == 0 && shallows.nr == 0)
 670                return;
 671        if (depth > 0) {
 672                struct commit_list *result, *backup;
 673                int i;
 674                backup = result = get_shallow_commits(&want_obj, depth,
 675                        SHALLOW, NOT_SHALLOW);
 676                while (result) {
 677                        struct object *object = &result->item->object;
 678                        if (!(object->flags & (CLIENT_SHALLOW|NOT_SHALLOW))) {
 679                                packet_write(1, "shallow %s",
 680                                                sha1_to_hex(object->sha1));
 681                                register_shallow(object->sha1);
 682                                shallow_nr++;
 683                        }
 684                        result = result->next;
 685                }
 686                free_commit_list(backup);
 687                for (i = 0; i < shallows.nr; i++) {
 688                        struct object *object = shallows.objects[i].item;
 689                        if (object->flags & NOT_SHALLOW) {
 690                                struct commit_list *parents;
 691                                packet_write(1, "unshallow %s",
 692                                        sha1_to_hex(object->sha1));
 693                                object->flags &= ~CLIENT_SHALLOW;
 694                                /* make sure the real parents are parsed */
 695                                unregister_shallow(object->sha1);
 696                                object->parsed = 0;
 697                                if (parse_commit((struct commit *)object))
 698                                        die("invalid commit");
 699                                parents = ((struct commit *)object)->parents;
 700                                while (parents) {
 701                                        add_object_array(&parents->item->object,
 702                                                        NULL, &want_obj);
 703                                        parents = parents->next;
 704                                }
 705                                add_object_array(object, NULL, &extra_edge_obj);
 706                        }
 707                        /* make sure commit traversal conforms to client */
 708                        register_shallow(object->sha1);
 709                }
 710                packet_flush(1);
 711        } else
 712                if (shallows.nr > 0) {
 713                        int i;
 714                        for (i = 0; i < shallows.nr; i++)
 715                                register_shallow(shallows.objects[i].item->sha1);
 716                }
 717
 718        shallow_nr += shallows.nr;
 719        free(shallows.objects);
 720}
 721
 722static int send_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
 723{
 724        static const char *capabilities = "multi_ack thin-pack side-band"
 725                " side-band-64k ofs-delta shallow no-progress"
 726                " include-tag multi_ack_detailed";
 727        struct object *o = lookup_unknown_object(sha1);
 728        const char *refname_nons = strip_namespace(refname);
 729
 730        if (o->type == OBJ_NONE) {
 731                o->type = sha1_object_info(sha1, NULL);
 732                if (o->type < 0)
 733                    die("git upload-pack: cannot find object %s:", sha1_to_hex(sha1));
 734        }
 735
 736        if (capabilities)
 737                packet_write(1, "%s %s%c%s%s\n", sha1_to_hex(sha1), refname_nons,
 738                             0, capabilities,
 739                             stateless_rpc ? " no-done" : "");
 740        else
 741                packet_write(1, "%s %s\n", sha1_to_hex(sha1), refname_nons);
 742        capabilities = NULL;
 743        if (!(o->flags & OUR_REF)) {
 744                o->flags |= OUR_REF;
 745                nr_our_refs++;
 746        }
 747        if (o->type == OBJ_TAG) {
 748                o = deref_tag_noverify(o);
 749                if (o)
 750                        packet_write(1, "%s %s^{}\n", sha1_to_hex(o->sha1), refname_nons);
 751        }
 752        return 0;
 753}
 754
 755static int mark_our_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
 756{
 757        struct object *o = parse_object(sha1);
 758        if (!o)
 759                die("git upload-pack: cannot find object %s:", sha1_to_hex(sha1));
 760        if (!(o->flags & OUR_REF)) {
 761                o->flags |= OUR_REF;
 762                nr_our_refs++;
 763        }
 764        return 0;
 765}
 766
 767static void upload_pack(void)
 768{
 769        if (advertise_refs || !stateless_rpc) {
 770                reset_timeout();
 771                head_ref_namespaced(send_ref, NULL);
 772                for_each_namespaced_ref(send_ref, NULL);
 773                packet_flush(1);
 774        } else {
 775                head_ref_namespaced(mark_our_ref, NULL);
 776                for_each_namespaced_ref(mark_our_ref, NULL);
 777        }
 778        if (advertise_refs)
 779                return;
 780
 781        receive_needs();
 782        if (want_obj.nr) {
 783                get_common_commits();
 784                create_pack_file();
 785        }
 786}
 787
 788int main(int argc, char **argv)
 789{
 790        char *dir;
 791        int i;
 792        int strict = 0;
 793
 794        git_setup_gettext();
 795
 796        packet_trace_identity("upload-pack");
 797        git_extract_argv0_path(argv[0]);
 798        read_replace_refs = 0;
 799
 800        for (i = 1; i < argc; i++) {
 801                char *arg = argv[i];
 802
 803                if (arg[0] != '-')
 804                        break;
 805                if (!strcmp(arg, "--advertise-refs")) {
 806                        advertise_refs = 1;
 807                        continue;
 808                }
 809                if (!strcmp(arg, "--stateless-rpc")) {
 810                        stateless_rpc = 1;
 811                        continue;
 812                }
 813                if (!strcmp(arg, "--strict")) {
 814                        strict = 1;
 815                        continue;
 816                }
 817                if (!prefixcmp(arg, "--timeout=")) {
 818                        timeout = atoi(arg+10);
 819                        daemon_mode = 1;
 820                        continue;
 821                }
 822                if (!strcmp(arg, "--")) {
 823                        i++;
 824                        break;
 825                }
 826        }
 827
 828        if (i != argc-1)
 829                usage(upload_pack_usage);
 830
 831        setup_path();
 832
 833        dir = argv[i];
 834
 835        if (!enter_repo(dir, strict))
 836                die("'%s' does not appear to be a git repository", dir);
 837        if (is_repository_shallow())
 838                die("attempt to fetch/clone from a shallow repository");
 839        if (getenv("GIT_DEBUG_SEND_PACK"))
 840                debug_fd = atoi(getenv("GIT_DEBUG_SEND_PACK"));
 841        upload_pack();
 842        return 0;
 843}