Andrew's git - gitweb.git/blob - builtin/fetch-pack.c

   1#include "builtin.h"
   2#include "pkt-line.h"
   3#include "fetch-pack.h"
   4
   5static const char fetch_pack_usage[] =
   6"git fetch-pack [--all] [--stdin] [--quiet|-q] [--keep|-k] [--thin] "
   7"[--include-tag] [--upload-pack=<git-upload-pack>] [--depth=<n>] "
   8"[--no-progress] [-v] [<host>:]<directory> [<refs>...]";
   9
  10int cmd_fetch_pack(int argc, const char **argv, const char *prefix)
  11{
  12        int i, ret;
  13        struct ref *ref = NULL;
  14        const char *dest = NULL;
  15        struct string_list sought = STRING_LIST_INIT_DUP;
  16        int fd[2];
  17        char *pack_lockfile = NULL;
  18        char **pack_lockfile_ptr = NULL;
  19        struct child_process *conn;
  20        struct fetch_pack_args args;
  21
  22        packet_trace_identity("fetch-pack");
  23
  24        memset(&args, 0, sizeof(args));
  25        args.uploadpack = "git-upload-pack";
  26
  27        for (i = 1; i < argc && *argv[i] == '-'; i++) {
  28                const char *arg = argv[i];
  29
  30                if (!prefixcmp(arg, "--upload-pack=")) {
  31                        args.uploadpack = arg + 14;
  32                        continue;
  33                }
  34                if (!prefixcmp(arg, "--exec=")) {
  35                        args.uploadpack = arg + 7;
  36                        continue;
  37                }
  38                if (!strcmp("--quiet", arg) || !strcmp("-q", arg)) {
  39                        args.quiet = 1;
  40                        continue;
  41                }
  42                if (!strcmp("--keep", arg) || !strcmp("-k", arg)) {
  43                        args.lock_pack = args.keep_pack;
  44                        args.keep_pack = 1;
  45                        continue;
  46                }
  47                if (!strcmp("--thin", arg)) {
  48                        args.use_thin_pack = 1;
  49                        continue;
  50                }
  51                if (!strcmp("--include-tag", arg)) {
  52                        args.include_tag = 1;
  53                        continue;
  54                }
  55                if (!strcmp("--all", arg)) {
  56                        args.fetch_all = 1;
  57                        continue;
  58                }
  59                if (!strcmp("--stdin", arg)) {
  60                        args.stdin_refs = 1;
  61                        continue;
  62                }
  63                if (!strcmp("-v", arg)) {
  64                        args.verbose = 1;
  65                        continue;
  66                }
  67                if (!prefixcmp(arg, "--depth=")) {
  68                        args.depth = strtol(arg + 8, NULL, 0);
  69                        continue;
  70                }
  71                if (!strcmp("--no-progress", arg)) {
  72                        args.no_progress = 1;
  73                        continue;
  74                }
  75                if (!strcmp("--stateless-rpc", arg)) {
  76                        args.stateless_rpc = 1;
  77                        continue;
  78                }
  79                if (!strcmp("--lock-pack", arg)) {
  80                        args.lock_pack = 1;
  81                        pack_lockfile_ptr = &pack_lockfile;
  82                        continue;
  83                }
  84                usage(fetch_pack_usage);
  85        }
  86
  87        if (i < argc)
  88                dest = argv[i++];
  89        else
  90                usage(fetch_pack_usage);
  91
  92        /*
  93         * Copy refs from cmdline to growable list, then append any
  94         * refs from the standard input:
  95         */
  96        for (; i < argc; i++)
  97                string_list_append(&sought, xstrdup(argv[i]));
  98        if (args.stdin_refs) {
  99                if (args.stateless_rpc) {
 100                        /* in stateless RPC mode we use pkt-line to read
 101                         * from stdin, until we get a flush packet
 102                         */
 103                        static char line[1000];
 104                        for (;;) {
 105                                int n = packet_read_line(0, line, sizeof(line));
 106                                if (!n)
 107                                        break;
 108                                if (line[n-1] == '\n')
 109                                        n--;
 110                                string_list_append(&sought, xmemdupz(line, n));
 111                        }
 112                }
 113                else {
 114                        /* read from stdin one ref per line, until EOF */
 115                        struct strbuf line = STRBUF_INIT;
 116                        while (strbuf_getline(&line, stdin, '\n') != EOF)
 117                                string_list_append(&sought, strbuf_detach(&line, NULL));
 118                        strbuf_release(&line);
 119                }
 120        }
 121
 122        if (args.stateless_rpc) {
 123                conn = NULL;
 124                fd[0] = 0;
 125                fd[1] = 1;
 126        } else {
 127                conn = git_connect(fd, dest, args.uploadpack,
 128                                   args.verbose ? CONNECT_VERBOSE : 0);
 129        }
 130
 131        get_remote_heads(fd[0], &ref, 0, NULL);
 132
 133        ref = fetch_pack(&args, fd, conn, ref, dest,
 134                         &sought, pack_lockfile_ptr);
 135        if (pack_lockfile) {
 136                printf("lock %s\n", pack_lockfile);
 137                fflush(stdout);
 138        }
 139        close(fd[0]);
 140        close(fd[1]);
 141        if (finish_connect(conn))
 142                return 1;
 143
 144        ret = !ref || sought.nr;
 145
 146        /*
 147         * If the heads to pull were given, we should have consumed
 148         * all of them by matching the remote.  Otherwise, 'git fetch
 149         * remote no-such-ref' would silently succeed without issuing
 150         * an error.
 151         */
 152        for (i = 0; i < sought.nr; i++)
 153                error("no such remote ref %s", sought.items[i].string);
 154        while (ref) {
 155                printf("%s %s\n",
 156                       sha1_to_hex(ref->old_sha1), ref->name);
 157                ref = ref->next;
 158        }
 159
 160        return ret;
 161}