builtin-push.c

   1 /*
   2  * "git push"
   3  */
   4 #include "cache.h"
   5 #include "refs.h"
   6 #include "run-command.h"
   7 #include "builtin.h"
   8
   9 #define MAX_URI (16)
  10
  11 static const char push_usage[] = "git-push [--all] [--tags] [-f | --force] <repository> [<refspec>...]";
  12
  13 static int all = 0, tags = 0, force = 0, thin = 1;
  14 static const char *execute = NULL;
  15
  16 #define BUF_SIZE (2084)
  17 static char buffer[BUF_SIZE];
  18
  19 static const char **refspec = NULL;
  20 static int refspec_nr = 0;
  21
  22 static void add_refspec(const char *ref)
  23 {
  24         int nr = refspec_nr + 1;
  25         refspec = xrealloc(refspec, nr * sizeof(char *));
  26         refspec[nr-1] = ref;
  27         refspec_nr = nr;
  28 }
  29
  30 static int expand_one_ref(const char *ref, const unsigned char *sha1)
  31 {
  32         /* Ignore the "refs/" at the beginning of the refname */
  33         ref += 5;
  34
  35         if (!strncmp(ref, "tags/", 5))
  36                 add_refspec(strdup(ref));
  37         return 0;
  38 }
  39
  40 static void expand_refspecs(void)
  41 {
  42         if (all) {
  43                 if (refspec_nr)
  44                         die("cannot mix '--all' and a refspec");
  45
  46                 /*
  47                  * No need to expand "--all" - we'll just use
  48                  * the "--all" flag to send-pack
  49                  */
  50                 return;
  51         }
  52         if (!tags)
  53                 return;
  54         for_each_ref(expand_one_ref);
  55 }
  56
  57 static void set_refspecs(const char **refs, int nr)
  58 {
  59         if (nr) {
  60                 size_t bytes = nr * sizeof(char *);
  61
  62                 refspec = xrealloc(refspec, bytes);
  63                 memcpy(refspec, refs, bytes);
  64                 refspec_nr = nr;
  65         }
  66         expand_refspecs();
  67 }
  68
  69 static int get_remotes_uri(const char *repo, const char *uri[MAX_URI])
  70 {
  71         int n = 0;
  72         FILE *f = fopen(git_path("remotes/%s", repo), "r");
  73         int has_explicit_refspec = refspec_nr || all || tags;
  74
  75         if (!f)
  76                 return -1;
  77         while (fgets(buffer, BUF_SIZE, f)) {
  78                 int is_refspec;
  79                 char *s, *p;
  80
  81                 if (!strncmp("URL: ", buffer, 5)) {
  82                         is_refspec = 0;
  83                         s = buffer + 5;
  84                 } else if (!strncmp("Push: ", buffer, 6)) {
  85                         is_refspec = 1;
  86                         s = buffer + 6;
  87                 } else
  88                         continue;
  89
  90                 /* Remove whitespace at the head.. */
  91                 while (isspace(*s))
  92                         s++;
  93                 if (!*s)
  94                         continue;
  95
  96                 /* ..and at the end */
  97                 p = s + strlen(s);
  98                 while (isspace(p[-1]))
  99                         *--p = 0;
 100
 101                 if (!is_refspec) {
 102                         if (n < MAX_URI)
 103                                 uri[n++] = strdup(s);
 104                         else
 105                                 error("more than %d URL's specified, ignoring the rest", MAX_URI);
 106                 }
 107                 else if (is_refspec && !has_explicit_refspec)
 108                         add_refspec(strdup(s));
 109         }
 110         fclose(f);
 111         if (!n)
 112                 die("remote '%s' has no URL", repo);
 113         return n;
 114 }
 115
 116 static const char **config_uri;
 117 static const char *config_repo;
 118 static int config_repo_len;
 119 static int config_current_uri;
 120 static int config_get_refspecs;
 121
 122 static int get_remote_config(const char* key, const char* value)
 123 {
 124         if (!strncmp(key, "remote.", 7) &&
 125             !strncmp(key + 7, config_repo, config_repo_len)) {
 126                 if (!strcmp(key + 7 + config_repo_len, ".url")) {
 127                         if (config_current_uri < MAX_URI)
 128                                 config_uri[config_current_uri++] = strdup(value);
 129                         else
 130                                 error("more than %d URL's specified, ignoring the rest", MAX_URI);
 131                 }
 132                 else if (config_get_refspecs &&
 133                          !strcmp(key + 7 + config_repo_len, ".push"))
 134                         add_refspec(strdup(value));
 135         }
 136         return 0;
 137 }
 138
 139 static int get_config_remotes_uri(const char *repo, const char *uri[MAX_URI])
 140 {
 141         config_repo_len = strlen(repo);
 142         config_repo = repo;
 143         config_current_uri = 0;
 144         config_uri = uri;
 145         config_get_refspecs = !(refspec_nr || all || tags);
 146
 147         git_config(get_remote_config);
 148         return config_current_uri;
 149 }
 150
 151 static int get_branches_uri(const char *repo, const char *uri[MAX_URI])
 152 {
 153         const char *slash = strchr(repo, '/');
 154         int n = slash ? slash - repo : 1000;
 155         FILE *f = fopen(git_path("branches/%.*s", n, repo), "r");
 156         char *s, *p;
 157         int len;
 158
 159         if (!f)
 160                 return 0;
 161         s = fgets(buffer, BUF_SIZE, f);
 162         fclose(f);
 163         if (!s)
 164                 return 0;
 165         while (isspace(*s))
 166                 s++;
 167         if (!*s)
 168                 return 0;
 169         p = s + strlen(s);
 170         while (isspace(p[-1]))
 171                 *--p = 0;
 172         len = p - s;
 173         if (slash)
 174                 len += strlen(slash);
 175         p = xmalloc(len + 1);
 176         strcpy(p, s);
 177         if (slash)
 178                 strcat(p, slash);
 179         uri[0] = p;
 180         return 1;
 181 }
 182
 183 /*
 184  * Read remotes and branches file, fill the push target URI
 185  * list.  If there is no command line refspecs, read Push: lines
 186  * to set up the *refspec list as well.
 187  * return the number of push target URIs
 188  */
 189 static int read_config(const char *repo, const char *uri[MAX_URI])
 190 {
 191         int n;
 192
 193         if (*repo != '/') {
 194                 n = get_remotes_uri(repo, uri);
 195                 if (n > 0)
 196                         return n;
 197
 198                 n = get_config_remotes_uri(repo, uri);
 199                 if (n > 0)
 200                         return n;
 201
 202                 n = get_branches_uri(repo, uri);
 203                 if (n > 0)
 204                         return n;
 205         }
 206
 207         uri[0] = repo;
 208         return 1;
 209 }
 210
 211 static int do_push(const char *repo)
 212 {
 213         const char *uri[MAX_URI];
 214         int i, n;
 215         int common_argc;
 216         const char **argv;
 217         int argc;
 218
 219         n = read_config(repo, uri);
 220         if (n <= 0)
 221                 die("bad repository '%s'", repo);
 222
 223         argv = xmalloc((refspec_nr + 10) * sizeof(char *));
 224         argv[0] = "dummy-send-pack";
 225         argc = 1;
 226         if (all)
 227                 argv[argc++] = "--all";
 228         if (force)
 229                 argv[argc++] = "--force";
 230         if (execute)
 231                 argv[argc++] = execute;
 232         common_argc = argc;
 233
 234         for (i = 0; i < n; i++) {
 235                 int error;
 236                 int dest_argc = common_argc;
 237                 int dest_refspec_nr = refspec_nr;
 238                 const char **dest_refspec = refspec;
 239                 const char *dest = uri[i];
 240                 const char *sender = "git-send-pack";
 241                 if (!strncmp(dest, "http://", 7) ||
 242                     !strncmp(dest, "https://", 8))
 243                         sender = "git-http-push";
 244                 else if (thin)
 245                         argv[dest_argc++] = "--thin";
 246                 argv[0] = sender;
 247                 argv[dest_argc++] = dest;
 248                 while (dest_refspec_nr--)
 249                         argv[dest_argc++] = *dest_refspec++;
 250                 argv[dest_argc] = NULL;
 251                 error = run_command_v(argc, argv);
 252                 if (!error)
 253                         continue;
 254                 switch (error) {
 255                 case -ERR_RUN_COMMAND_FORK:
 256                         die("unable to fork for %s", sender);
 257                 case -ERR_RUN_COMMAND_EXEC:
 258                         die("unable to exec %s", sender);
 259                 case -ERR_RUN_COMMAND_WAITPID:
 260                 case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
 261                 case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
 262                 case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
 263                         die("%s died with strange error", sender);
 264                 default:
 265                         return -error;
 266                 }
 267         }
 268         return 0;
 269 }
 270
 271 int cmd_push(int argc, const char **argv, const char *prefix)
 272 {
 273         int i;
 274         const char *repo = "origin";    /* default repository */
 275
 276         for (i = 1; i < argc; i++) {
 277                 const char *arg = argv[i];
 278
 279                 if (arg[0] != '-') {
 280                         repo = arg;
 281                         i++;
 282                         break;
 283                 }
 284                 if (!strcmp(arg, "--all")) {
 285                         all = 1;
 286                         continue;
 287                 }
 288                 if (!strcmp(arg, "--tags")) {
 289                         tags = 1;
 290                         continue;
 291                 }
 292                 if (!strcmp(arg, "--force") || !strcmp(arg, "-f")) {
 293                         force = 1;
 294                         continue;
 295                 }
 296                 if (!strcmp(arg, "--thin")) {
 297                         thin = 1;
 298                         continue;
 299                 }
 300                 if (!strcmp(arg, "--no-thin")) {
 301                         thin = 0;
 302                         continue;
 303                 }
 304                 if (!strncmp(arg, "--exec=", 7)) {
 305                         execute = arg;
 306                         continue;
 307                 }
 308                 usage(push_usage);
 309         }
 310         set_refspecs(argv + i, argc - i);
 311         return do_push(repo);
 312 }