1/*
2* Copyright (c) 2006 Franck Bui-Huu
3* Copyright (c) 2006 Rene Scharfe
4*/
5#include "cache.h"
6#include "builtin.h"
7#include "archive.h"
8#include "transport.h"
9#include "parse-options.h"
10#include "pkt-line.h"
11#include "sideband.h"
1213
static void create_output_file(const char *output_file)
14{
15int output_fd = open(output_file, O_CREAT | O_WRONLY | O_TRUNC, 0666);
16if (output_fd < 0)
17die_errno("could not create archive file '%s'", output_file);
18if (output_fd != 1) {
19if (dup2(output_fd, 1) < 0)
20die_errno("could not redirect output");
21else
22close(output_fd);
23}
24}
2526
static int run_remote_archiver(int argc, const char **argv,
27const char *remote, const char *exec)
28{
29char buf[LARGE_PACKET_MAX];
30int fd[2], i, len, rv;
31struct transport *transport;
32struct remote *_remote;
3334
_remote = remote_get(remote);
35if (!_remote->url[0])
36die("git archive: Remote with no URL");
37transport = transport_get(_remote, _remote->url[0]);
38transport_connect(transport, "git-upload-archive", exec, fd);
3940
for (i = 1; i < argc; i++)
41packet_write(fd[1], "argument %s\n", argv[i]);
42packet_flush(fd[1]);
4344
len = packet_read_line(fd[0], buf, sizeof(buf));
45if (!len)
46die("git archive: expected ACK/NAK, got EOF");
47if (buf[len-1] == '\n')
48buf[--len] = 0;
49if (strcmp(buf, "ACK")) {
50if (len > 5 && !prefixcmp(buf, "NACK "))
51die("git archive: NACK %s", buf + 5);
52die("git archive: protocol error");
53}
5455
len = packet_read_line(fd[0], buf, sizeof(buf));
56if (len)
57die("git archive: expected a flush");
5859
/* Now, start reading from fd[0] and spit it out to stdout */
60rv = recv_sideband("archive", fd[0], 1);
61rv |= transport_disconnect(transport);
6263
return !!rv;
64}
6566
static const char *format_from_name(const char *filename)
67{
68const char *ext = strrchr(filename, '.');
69if (!ext)
70return NULL;
71ext++;
72if (!strcasecmp(ext, "zip"))
73return "zip";
74return NULL;
75}
7677
#define PARSE_OPT_KEEP_ALL ( PARSE_OPT_KEEP_DASHDASH | \
78PARSE_OPT_KEEP_ARGV0 | \
79PARSE_OPT_KEEP_UNKNOWN | \
80PARSE_OPT_NO_INTERNAL_HELP )
8182
int cmd_archive(int argc, const char **argv, const char *prefix)
83{
84const char *exec = "git-upload-archive";
85const char *output = NULL;
86const char *remote = NULL;
87const char *format = NULL;
88struct option local_opts[] = {
89OPT_STRING('o', "output", &output, "file",
90"write the archive to this file"),
91OPT_STRING(0, "remote", &remote, "repo",
92"retrieve the archive from remote repository <repo>"),
93OPT_STRING(0, "exec", &exec, "cmd",
94"path to the remote git-upload-archive command"),
95OPT_STRING(0, "format", &format, "fmt", "archive format"),
96OPT_END()
97};
98char fmt_opt[32];
99100
argc = parse_options(argc, argv, prefix, local_opts, NULL,
101PARSE_OPT_KEEP_ALL);
102103
if (output) {
104create_output_file(output);
105if (!format)
106format = format_from_name(output);
107}
108109
if (format) {
110sprintf(fmt_opt, "--format=%s", format);
111/*
112* This is safe because either --format and/or --output must
113* have been given on the original command line if we get to
114* this point, and parse_options() must have eaten at least
115* one argument, i.e. we have enough room to append to argv[].
116*/
117argv[argc++] = fmt_opt;
118argv[argc] = NULL;
119}
120121
if (remote)
122return run_remote_archiver(argc, argv, remote, exec);
123124
setvbuf(stderr, NULL, _IOLBF, BUFSIZ);
125126
return write_archive(argc, argv, prefix, 1);
127}