blob: 5ceec433fd590e8bf6a51700ea69c37f9af30fa7 [file] [log] [blame]
Franck Bui-Huu4df096a2006-09-07 15:12:02 +02001/*
2 * Copyright (c) 2006 Franck Bui-Huu
3 * Copyright (c) 2006 Rene Scharfe
4 */
Franck Bui-Huu4df096a2006-09-07 15:12:02 +02005#include "cache.h"
6#include "builtin.h"
7#include "archive.h"
Franck Bui-Huu4df096a2006-09-07 15:12:02 +02008#include "pkt-line.h"
Junio C Hamano23d6d112006-09-10 03:33:34 -07009#include "sideband.h"
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020010
Junio C Hamano37f94432006-09-09 23:48:03 -070011static int run_remote_archiver(const char *remote, int argc,
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020012 const char **argv)
13{
Junio C Hamano23d6d112006-09-10 03:33:34 -070014 char *url, buf[LARGE_PACKET_MAX];
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020015 int fd[2], i, len, rv;
Johannes Sixt98158e92007-10-19 21:47:53 +020016 struct child_process *conn;
Junio C Hamanofe5ab762006-09-10 04:02:57 -070017 const char *exec = "git-upload-archive";
Rene Scharfe819b2b52008-07-25 12:41:25 +020018 int exec_at = 0, exec_value_at = 0;
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020019
Junio C Hamanofe5ab762006-09-10 04:02:57 -070020 for (i = 1; i < argc; i++) {
21 const char *arg = argv[i];
Junio C Hamano599065a2007-02-20 01:54:00 -080022 if (!prefixcmp(arg, "--exec=")) {
Junio C Hamanofe5ab762006-09-10 04:02:57 -070023 if (exec_at)
24 die("multiple --exec specified");
25 exec = arg + 7;
26 exec_at = i;
Rene Scharfe819b2b52008-07-25 12:41:25 +020027 } else if (!strcmp(arg, "--exec")) {
28 if (exec_at)
29 die("multiple --exec specified");
30 if (i + 1 >= argc)
31 die("option --exec requires a value");
32 exec = argv[i + 1];
33 exec_at = i;
34 exec_value_at = ++i;
Junio C Hamanofe5ab762006-09-10 04:02:57 -070035 }
36 }
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020037
Junio C Hamano37f94432006-09-09 23:48:03 -070038 url = xstrdup(remote);
Johannes Sixt98158e92007-10-19 21:47:53 +020039 conn = git_connect(fd, url, exec, 0);
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020040
Junio C Hamanofe5ab762006-09-10 04:02:57 -070041 for (i = 1; i < argc; i++) {
Rene Scharfe819b2b52008-07-25 12:41:25 +020042 if (i == exec_at || i == exec_value_at)
Junio C Hamanofe5ab762006-09-10 04:02:57 -070043 continue;
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020044 packet_write(fd[1], "argument %s\n", argv[i]);
Junio C Hamanofe5ab762006-09-10 04:02:57 -070045 }
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020046 packet_flush(fd[1]);
47
48 len = packet_read_line(fd[0], buf, sizeof(buf));
49 if (!len)
Heikki Orsila34baebc2008-08-30 14:12:53 +030050 die("git archive: expected ACK/NAK, got EOF");
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020051 if (buf[len-1] == '\n')
52 buf[--len] = 0;
53 if (strcmp(buf, "ACK")) {
Junio C Hamanocc44c762007-02-20 01:53:29 -080054 if (len > 5 && !prefixcmp(buf, "NACK "))
Heikki Orsila34baebc2008-08-30 14:12:53 +030055 die("git archive: NACK %s", buf + 5);
56 die("git archive: protocol error");
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020057 }
58
59 len = packet_read_line(fd[0], buf, sizeof(buf));
60 if (len)
Heikki Orsila34baebc2008-08-30 14:12:53 +030061 die("git archive: expected a flush");
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020062
63 /* Now, start reading from fd[0] and spit it out to stdout */
Nicolas Pitre9ac13ec2006-10-11 11:49:15 -040064 rv = recv_sideband("archive", fd[0], 1, 2);
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020065 close(fd[0]);
Junio C Hamanoec587fd2007-01-21 17:10:51 -080066 close(fd[1]);
Johannes Sixt98158e92007-10-19 21:47:53 +020067 rv |= finish_connect(conn);
Franck Bui-Huu4df096a2006-09-07 15:12:02 +020068
69 return !!rv;
70}
71
Junio C Hamanod7518642006-09-12 22:42:31 -070072static const char *extract_remote_arg(int *ac, const char **av)
Junio C Hamano37f94432006-09-09 23:48:03 -070073{
74 int ix, iy, cnt = *ac;
75 int no_more_options = 0;
76 const char *remote = NULL;
77
78 for (ix = iy = 1; ix < cnt; ix++) {
79 const char *arg = av[ix];
80 if (!strcmp(arg, "--"))
81 no_more_options = 1;
82 if (!no_more_options) {
Junio C Hamanocc44c762007-02-20 01:53:29 -080083 if (!prefixcmp(arg, "--remote=")) {
Junio C Hamano37f94432006-09-09 23:48:03 -070084 if (remote)
85 die("Multiple --remote specified");
86 remote = arg + 9;
87 continue;
Rene Scharfe819b2b52008-07-25 12:41:25 +020088 } else if (!strcmp(arg, "--remote")) {
89 if (remote)
90 die("Multiple --remote specified");
91 if (++ix >= cnt)
92 die("option --remote requires a value");
93 remote = av[ix];
94 continue;
Junio C Hamano37f94432006-09-09 23:48:03 -070095 }
96 if (arg[0] != '-')
97 no_more_options = 1;
98 }
99 if (ix != iy)
100 av[iy] = arg;
101 iy++;
102 }
103 if (remote) {
104 av[--cnt] = NULL;
105 *ac = cnt;
106 }
107 return remote;
108}
109
Franck Bui-Huu4df096a2006-09-07 15:12:02 +0200110int cmd_archive(int argc, const char **argv, const char *prefix)
111{
Junio C Hamano37f94432006-09-09 23:48:03 -0700112 const char *remote = NULL;
Franck Bui-Huu4df096a2006-09-07 15:12:02 +0200113
Junio C Hamanod7518642006-09-12 22:42:31 -0700114 remote = extract_remote_arg(&argc, argv);
Junio C Hamano37f94432006-09-09 23:48:03 -0700115 if (remote)
116 return run_remote_archiver(remote, argc, argv);
117
Michal Rokosaa909862006-11-21 23:19:28 +0100118 setvbuf(stderr, NULL, _IOLBF, BUFSIZ);
Junio C Hamano8142f602006-09-10 04:16:39 -0700119
Rene Scharfe6e94e682008-07-25 12:41:21 +0200120 return write_archive(argc, argv, prefix, 1);
Franck Bui-Huu4df096a2006-09-07 15:12:02 +0200121}