Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2006 Franck Bui-Huu |
| 3 | */ |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 4 | #include "builtin.h" |
| 5 | #include "archive.h" |
Elijah Newren | c339932 | 2023-05-16 06:33:59 +0000 | [diff] [blame] | 6 | #include "path.h" |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 7 | #include "pkt-line.h" |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 8 | #include "sideband.h" |
Elijah Newren | d1cbe1e | 2023-04-22 20:17:20 +0000 | [diff] [blame] | 9 | #include "repository.h" |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 10 | #include "run-command.h" |
Jeff King | dbbcd44 | 2020-07-28 16:23:39 -0400 | [diff] [blame] | 11 | #include "strvec.h" |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 12 | |
| 13 | static const char upload_archive_usage[] = |
Ævar Arnfjörð Bjarmason | f6a8ef0 | 2022-10-13 17:39:10 +0200 | [diff] [blame] | 14 | "git upload-archive <repository>"; |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 15 | |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 16 | static const char deadchild[] = |
Stephan Beyer | 1b1dd23 | 2008-07-13 15:36:15 +0200 | [diff] [blame] | 17 | "git upload-archive: archiver died with error"; |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 18 | |
Rene Scharfe | 7f4d051 | 2008-07-25 12:41:23 +0200 | [diff] [blame] | 19 | #define MAX_ARGS (64) |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 20 | |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 21 | int cmd_upload_archive_writer(int argc, const char **argv, const char *prefix) |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 22 | { |
Jeff King | 22f9b7f | 2020-07-28 16:24:27 -0400 | [diff] [blame] | 23 | struct strvec sent_argv = STRVEC_INIT; |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 24 | const char *arg_cmd = "argument "; |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 25 | |
Jeff King | 619b6c1 | 2017-05-30 01:13:43 -0400 | [diff] [blame] | 26 | if (argc != 2 || !strcmp(argv[1], "-h")) |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 27 | usage(upload_archive_usage); |
| 28 | |
Jeff King | 6379dd0 | 2013-02-20 15:00:59 -0500 | [diff] [blame] | 29 | if (!enter_repo(argv[1], 0)) |
| 30 | die("'%s' does not appear to be a git repository", argv[1]); |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 31 | |
Josh Steadmon | 00436bf | 2018-10-25 13:32:14 -0700 | [diff] [blame] | 32 | init_archivers(); |
| 33 | |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 34 | /* put received options in sent_argv[] */ |
Jeff King | 22f9b7f | 2020-07-28 16:24:27 -0400 | [diff] [blame] | 35 | strvec_push(&sent_argv, "git-upload-archive"); |
Jeff King | 090fd4f | 2013-02-20 15:01:26 -0500 | [diff] [blame] | 36 | for (;;) { |
Jeff King | 74543a0 | 2013-02-20 15:02:57 -0500 | [diff] [blame] | 37 | char *buf = packet_read_line(0, NULL); |
| 38 | if (!buf) |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 39 | break; /* got a flush */ |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 40 | if (sent_argv.nr > MAX_ARGS) |
Jeff King | 090fd4f | 2013-02-20 15:01:26 -0500 | [diff] [blame] | 41 | die("Too many options (>%d)", MAX_ARGS - 1); |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 42 | |
Christian Couder | 5955654 | 2013-11-30 21:55:40 +0100 | [diff] [blame] | 43 | if (!starts_with(buf, arg_cmd)) |
Jeff King | 090fd4f | 2013-02-20 15:01:26 -0500 | [diff] [blame] | 44 | die("'argument' token or flush expected"); |
Jeff King | 22f9b7f | 2020-07-28 16:24:27 -0400 | [diff] [blame] | 45 | strvec_push(&sent_argv, buf + strlen(arg_cmd)); |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 46 | } |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 47 | |
| 48 | /* parse all options sent by the client */ |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 49 | return write_archive(sent_argv.nr, sent_argv.v, prefix, |
Nguyễn Thái Ngọc Duy | b612ee2 | 2018-08-13 18:14:35 +0200 | [diff] [blame] | 50 | the_repository, NULL, 1); |
Franck Bui-Huu | 39345a2 | 2006-09-07 15:12:05 +0200 | [diff] [blame] | 51 | } |
| 52 | |
Tarmigan Casebolt | 28bea9e | 2009-11-14 13:33:13 -0800 | [diff] [blame] | 53 | __attribute__((format (printf, 1, 2))) |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 54 | static void error_clnt(const char *fmt, ...) |
| 55 | { |
Jeff King | 0cb9d6d | 2015-09-24 17:07:25 -0400 | [diff] [blame] | 56 | struct strbuf buf = STRBUF_INIT; |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 57 | va_list params; |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 58 | |
| 59 | va_start(params, fmt); |
Jeff King | 0cb9d6d | 2015-09-24 17:07:25 -0400 | [diff] [blame] | 60 | strbuf_vaddf(&buf, fmt, params); |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 61 | va_end(params); |
Jeff King | 0cb9d6d | 2015-09-24 17:07:25 -0400 | [diff] [blame] | 62 | send_sideband(1, 3, buf.buf, buf.len, LARGE_PACKET_MAX); |
| 63 | die("sent error to the client: %s", buf.buf); |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 64 | } |
| 65 | |
René Scharfe | 1b19fa4 | 2009-06-17 12:11:10 +0200 | [diff] [blame] | 66 | static ssize_t process_input(int child_fd, int band) |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 67 | { |
| 68 | char buf[16384]; |
| 69 | ssize_t sz = read(child_fd, buf, sizeof(buf)); |
| 70 | if (sz < 0) { |
Andy Whitcroft | 93d26e4 | 2007-01-08 15:58:08 +0000 | [diff] [blame] | 71 | if (errno != EAGAIN && errno != EINTR) |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 72 | error_clnt("read error: %s\n", strerror(errno)); |
René Scharfe | 1b19fa4 | 2009-06-17 12:11:10 +0200 | [diff] [blame] | 73 | return sz; |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 74 | } |
| 75 | send_sideband(1, band, buf, sz, LARGE_PACKET_MAX); |
René Scharfe | 1b19fa4 | 2009-06-17 12:11:10 +0200 | [diff] [blame] | 76 | return sz; |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 77 | } |
| 78 | |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 79 | int cmd_upload_archive(int argc, const char **argv, const char *prefix) |
| 80 | { |
Ævar Arnfjörð Bjarmason | c8a4cd55 | 2021-11-25 23:52:17 +0100 | [diff] [blame] | 81 | struct child_process writer = CHILD_PROCESS_INIT; |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 82 | |
Jeff King | 7915691 | 2023-03-28 16:55:17 -0400 | [diff] [blame] | 83 | BUG_ON_NON_EMPTY_PREFIX(prefix); |
| 84 | |
Jeff King | 619b6c1 | 2017-05-30 01:13:43 -0400 | [diff] [blame] | 85 | if (argc == 2 && !strcmp(argv[1], "-h")) |
| 86 | usage(upload_archive_usage); |
| 87 | |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 88 | /* |
| 89 | * Set up sideband subprocess. |
| 90 | * |
| 91 | * We (parent) monitor and read from child, sending its fd#1 and fd#2 |
| 92 | * multiplexed out to our fd#1. If the child dies, we tell the other |
| 93 | * end over channel #3. |
| 94 | */ |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 95 | writer.out = writer.err = -1; |
| 96 | writer.git_cmd = 1; |
Ævar Arnfjörð Bjarmason | c8a4cd55 | 2021-11-25 23:52:17 +0100 | [diff] [blame] | 97 | strvec_push(&writer.args, "upload-archive--writer"); |
| 98 | strvec_pushv(&writer.args, argv + 1); |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 99 | if (start_command(&writer)) { |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 100 | int err = errno; |
Lars Schneider | 81c634e | 2016-10-16 16:20:29 -0700 | [diff] [blame] | 101 | packet_write_fmt(1, "NACK unable to spawn subprocess\n"); |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 102 | die("upload-archive: %s", strerror(err)); |
| 103 | } |
Junio C Hamano | f0c7fd4 | 2011-11-15 15:39:33 -0800 | [diff] [blame] | 104 | |
Lars Schneider | 81c634e | 2016-10-16 16:20:29 -0700 | [diff] [blame] | 105 | packet_write_fmt(1, "ACK\n"); |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 106 | packet_flush(1); |
| 107 | |
| 108 | while (1) { |
| 109 | struct pollfd pfd[2]; |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 110 | |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 111 | pfd[0].fd = writer.out; |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 112 | pfd[0].events = POLLIN; |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 113 | pfd[1].fd = writer.err; |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 114 | pfd[1].events = POLLIN; |
| 115 | if (poll(pfd, 2, -1) < 0) { |
| 116 | if (errno != EINTR) { |
Nguyễn Thái Ngọc Duy | 17bef17 | 2016-05-08 16:47:33 +0700 | [diff] [blame] | 117 | error_errno("poll failed resuming"); |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 118 | sleep(1); |
| 119 | } |
| 120 | continue; |
| 121 | } |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 122 | if (pfd[1].revents & POLLIN) |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 123 | /* Status stream ready */ |
Nicolas Pitre | 6b59f51 | 2009-11-11 17:24:42 -0500 | [diff] [blame] | 124 | if (process_input(pfd[1].fd, 2)) |
| 125 | continue; |
| 126 | if (pfd[0].revents & POLLIN) |
| 127 | /* Data stream ready */ |
| 128 | if (process_input(pfd[0].fd, 1)) |
| 129 | continue; |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 130 | |
Jeff King | 1bc01ef | 2011-11-19 02:40:04 -0500 | [diff] [blame] | 131 | if (finish_command(&writer)) |
Junio C Hamano | d3788e1 | 2006-09-12 00:26:57 -0700 | [diff] [blame] | 132 | error_clnt("%s", deadchild); |
Junio C Hamano | 23d6d11 | 2006-09-10 03:33:34 -0700 | [diff] [blame] | 133 | packet_flush(1); |
| 134 | break; |
| 135 | } |
| 136 | return 0; |
| 137 | } |