]>
Commit | Line | Data |
---|---|---|
39345a21 FBH |
1 | /* |
2 | * Copyright (c) 2006 Franck Bui-Huu | |
3 | */ | |
39345a21 FBH |
4 | #include "cache.h" |
5 | #include "builtin.h" | |
6 | #include "archive.h" | |
7 | #include "pkt-line.h" | |
23d6d112 | 8 | #include "sideband.h" |
1bc01efe | 9 | #include "run-command.h" |
090fd4fe | 10 | #include "argv-array.h" |
39345a21 FBH |
11 | |
12 | static const char upload_archive_usage[] = | |
1b1dd23f | 13 | "git upload-archive <repo>"; |
39345a21 | 14 | |
23d6d112 | 15 | static const char deadchild[] = |
1b1dd23f | 16 | "git upload-archive: archiver died with error"; |
39345a21 | 17 | |
7f4d0511 | 18 | #define MAX_ARGS (64) |
23d6d112 | 19 | |
1bc01efe | 20 | int cmd_upload_archive_writer(int argc, const char **argv, const char *prefix) |
39345a21 | 21 | { |
090fd4fe | 22 | struct argv_array sent_argv = ARGV_ARRAY_INIT; |
39345a21 | 23 | const char *arg_cmd = "argument "; |
39345a21 | 24 | |
f0c7fd49 JH |
25 | if (argc != 2) |
26 | usage(upload_archive_usage); | |
27 | ||
6379dd05 JK |
28 | if (!enter_repo(argv[1], 0)) |
29 | die("'%s' does not appear to be a git repository", argv[1]); | |
f0c7fd49 | 30 | |
39345a21 | 31 | /* put received options in sent_argv[] */ |
090fd4fe JK |
32 | argv_array_push(&sent_argv, "git-upload-archive"); |
33 | for (;;) { | |
74543a04 JK |
34 | char *buf = packet_read_line(0, NULL); |
35 | if (!buf) | |
39345a21 | 36 | break; /* got a flush */ |
090fd4fe JK |
37 | if (sent_argv.argc > MAX_ARGS) |
38 | die("Too many options (>%d)", MAX_ARGS - 1); | |
39345a21 | 39 | |
59556548 | 40 | if (!starts_with(buf, arg_cmd)) |
090fd4fe JK |
41 | die("'argument' token or flush expected"); |
42 | argv_array_push(&sent_argv, buf + strlen(arg_cmd)); | |
39345a21 | 43 | } |
f0c7fd49 JH |
44 | |
45 | /* parse all options sent by the client */ | |
090fd4fe | 46 | return write_archive(sent_argv.argc, sent_argv.argv, prefix, 0, NULL, 1); |
23d6d112 JH |
47 | } |
48 | ||
28bea9e5 | 49 | __attribute__((format (printf, 1, 2))) |
d3788e19 JH |
50 | static void error_clnt(const char *fmt, ...) |
51 | { | |
52 | char buf[1024]; | |
53 | va_list params; | |
54 | int len; | |
55 | ||
56 | va_start(params, fmt); | |
57 | len = vsprintf(buf, fmt, params); | |
58 | va_end(params); | |
59 | send_sideband(1, 3, buf, len, LARGE_PACKET_MAX); | |
60 | die("sent error to the client: %s", buf); | |
61 | } | |
62 | ||
1b19fa46 | 63 | static ssize_t process_input(int child_fd, int band) |
d3788e19 JH |
64 | { |
65 | char buf[16384]; | |
66 | ssize_t sz = read(child_fd, buf, sizeof(buf)); | |
67 | if (sz < 0) { | |
93d26e4c | 68 | if (errno != EAGAIN && errno != EINTR) |
d3788e19 | 69 | error_clnt("read error: %s\n", strerror(errno)); |
1b19fa46 | 70 | return sz; |
d3788e19 JH |
71 | } |
72 | send_sideband(1, band, buf, sz, LARGE_PACKET_MAX); | |
1b19fa46 | 73 | return sz; |
d3788e19 JH |
74 | } |
75 | ||
23d6d112 JH |
76 | int cmd_upload_archive(int argc, const char **argv, const char *prefix) |
77 | { | |
1bc01efe JK |
78 | struct child_process writer = { argv }; |
79 | ||
f0c7fd49 JH |
80 | /* |
81 | * Set up sideband subprocess. | |
82 | * | |
83 | * We (parent) monitor and read from child, sending its fd#1 and fd#2 | |
84 | * multiplexed out to our fd#1. If the child dies, we tell the other | |
85 | * end over channel #3. | |
86 | */ | |
1bc01efe JK |
87 | argv[0] = "upload-archive--writer"; |
88 | writer.out = writer.err = -1; | |
89 | writer.git_cmd = 1; | |
90 | if (start_command(&writer)) { | |
23d6d112 | 91 | int err = errno; |
1bc01efe | 92 | packet_write(1, "NACK unable to spawn subprocess\n"); |
23d6d112 JH |
93 | die("upload-archive: %s", strerror(err)); |
94 | } | |
23d6d112 | 95 | |
39345a21 FBH |
96 | packet_write(1, "ACK\n"); |
97 | packet_flush(1); | |
98 | ||
23d6d112 JH |
99 | while (1) { |
100 | struct pollfd pfd[2]; | |
23d6d112 | 101 | |
1bc01efe | 102 | pfd[0].fd = writer.out; |
23d6d112 | 103 | pfd[0].events = POLLIN; |
1bc01efe | 104 | pfd[1].fd = writer.err; |
23d6d112 JH |
105 | pfd[1].events = POLLIN; |
106 | if (poll(pfd, 2, -1) < 0) { | |
107 | if (errno != EINTR) { | |
108 | error("poll failed resuming: %s", | |
109 | strerror(errno)); | |
110 | sleep(1); | |
111 | } | |
112 | continue; | |
113 | } | |
d3788e19 | 114 | if (pfd[1].revents & POLLIN) |
23d6d112 | 115 | /* Status stream ready */ |
6b59f51b NP |
116 | if (process_input(pfd[1].fd, 2)) |
117 | continue; | |
118 | if (pfd[0].revents & POLLIN) | |
119 | /* Data stream ready */ | |
120 | if (process_input(pfd[0].fd, 1)) | |
121 | continue; | |
d3788e19 | 122 | |
1bc01efe | 123 | if (finish_command(&writer)) |
d3788e19 | 124 | error_clnt("%s", deadchild); |
23d6d112 JH |
125 | packet_flush(1); |
126 | break; | |
127 | } | |
128 | return 0; | |
129 | } |