f098acb5fd0510285fa5d4aa63460a57e8183bda
[git/git.git] / send-pack.c
1 #include "cache.h"
2 #include "pkt-line.h"
3
4 static const char send_pack_usage[] = "git-send-pack [--exec=other] destination [heads]*";
5 static const char *exec = "git-receive-pack";
6
7 struct ref {
8 struct ref *next;
9 unsigned char old_sha1[20];
10 unsigned char new_sha1[20];
11 char name[0];
12 };
13
14 static void exec_pack_objects(void)
15 {
16 static char *args[] = {
17 "git-pack-objects",
18 "--stdout",
19 NULL
20 };
21 execvp("git-pack-objects", args);
22 die("git-pack-objects exec failed (%s)", strerror(errno));
23 }
24
25 static void exec_rev_list(struct ref *refs)
26 {
27 static char *args[1000];
28 int i = 0;
29
30 args[i++] = "git-rev-list"; /* 0 */
31 args[i++] = "--objects"; /* 1 */
32 while (refs) {
33 char *buf = malloc(100);
34 if (i > 900)
35 die("git-rev-list environment overflow");
36 args[i++] = buf;
37 snprintf(buf, 50, "^%s", sha1_to_hex(refs->old_sha1));
38 buf += 50;
39 args[i++] = buf;
40 snprintf(buf, 50, "%s", sha1_to_hex(refs->new_sha1));
41 refs = refs->next;
42 }
43 args[i] = NULL;
44 execvp("git-rev-list", args);
45 die("git-rev-list exec failed (%s)", strerror(errno));
46 }
47
48 static void rev_list(int fd, struct ref *refs)
49 {
50 int pipe_fd[2];
51 pid_t pack_objects_pid;
52
53 if (pipe(pipe_fd) < 0)
54 die("rev-list setup: pipe failed");
55 pack_objects_pid = fork();
56 if (!pack_objects_pid) {
57 dup2(pipe_fd[0], 0);
58 dup2(fd, 1);
59 close(pipe_fd[0]);
60 close(pipe_fd[1]);
61 close(fd);
62 exec_pack_objects();
63 die("pack-objects setup failed");
64 }
65 if (pack_objects_pid < 0)
66 die("pack-objects fork failed");
67 dup2(pipe_fd[1], 1);
68 close(pipe_fd[0]);
69 close(pipe_fd[1]);
70 close(fd);
71 exec_rev_list(refs);
72 }
73
74 static int pack_objects(int fd, struct ref *refs)
75 {
76 pid_t rev_list_pid;
77
78 rev_list_pid = fork();
79 if (!rev_list_pid) {
80 rev_list(fd, refs);
81 die("rev-list setup failed");
82 }
83 if (rev_list_pid < 0)
84 die("rev-list fork failed");
85 /*
86 * We don't wait for the rev-list pipeline in the parent:
87 * we end up waiting for the other end instead
88 */
89 return 0;
90 }
91
92 static int read_ref(const char *ref, unsigned char *sha1)
93 {
94 int fd, ret;
95 static char pathname[PATH_MAX];
96 char buffer[60];
97 const char *git_dir = gitenv(GIT_DIR_ENVIRONMENT) ? : DEFAULT_GIT_DIR_ENVIRONMENT;
98
99 snprintf(pathname, sizeof(pathname), "%s/%s", git_dir, ref);
100 fd = open(pathname, O_RDONLY);
101 if (fd < 0)
102 return -1;
103 ret = -1;
104 if (read(fd, buffer, sizeof(buffer)) >= 40)
105 ret = get_sha1_hex(buffer, sha1);
106 close(fd);
107 return ret;
108 }
109
110 static int send_pack(int in, int out, int nr_match, char **match)
111 {
112 struct ref *ref_list = NULL, **last_ref = &ref_list;
113 struct ref *ref;
114
115 for (;;) {
116 unsigned char old_sha1[20];
117 unsigned char new_sha1[20];
118 static char buffer[1000];
119 char *name;
120 int len;
121
122 len = packet_read_line(in, buffer, sizeof(buffer));
123 if (!len)
124 break;
125 if (buffer[len-1] == '\n')
126 buffer[--len] = 0;
127
128 if (len < 42 || get_sha1_hex(buffer, old_sha1) || buffer[40] != ' ')
129 die("protocol error: expected sha/ref, got '%s'", buffer);
130 name = buffer + 41;
131 if (nr_match && !path_match(name, nr_match, match))
132 continue;
133 if (read_ref(name, new_sha1) < 0)
134 return error("no such local reference '%s'", name);
135 if (!has_sha1_file(old_sha1))
136 return error("remote '%s' points to object I don't have", name);
137 if (!memcmp(old_sha1, new_sha1, 20)) {
138 fprintf(stderr, "'%s' unchanged\n", name);
139 continue;
140 }
141 ref = xmalloc(sizeof(*ref) + len - 40);
142 memcpy(ref->old_sha1, old_sha1, 20);
143 memcpy(ref->new_sha1, new_sha1, 20);
144 memcpy(ref->name, buffer + 41, len - 40);
145 ref->next = NULL;
146 *last_ref = ref;
147 last_ref = &ref->next;
148 }
149
150 for (ref = ref_list; ref; ref = ref->next) {
151 char old_hex[60], *new_hex;
152 strcpy(old_hex, sha1_to_hex(ref->old_sha1));
153 new_hex = sha1_to_hex(ref->new_sha1);
154 packet_write(out, "%s %s %s", old_hex, new_hex, ref->name);
155 fprintf(stderr, "'%s': updating from %s to %s\n", ref->name, old_hex, new_hex);
156 }
157
158 packet_flush(out);
159 if (ref_list)
160 pack_objects(out, ref_list);
161 close(out);
162 return 0;
163 }
164
165 int main(int argc, char **argv)
166 {
167 int i, nr_heads = 0;
168 char *dest = NULL;
169 char **heads = NULL;
170 int fd[2], ret;
171 pid_t pid;
172
173 argv++;
174 for (i = 1; i < argc; i++) {
175 char *arg = *argv++;
176
177 if (*arg == '-') {
178 if (!strncmp(arg, "--exec=", 7)) {
179 exec = arg + 7;
180 continue;
181 }
182 usage(send_pack_usage);
183 }
184 dest = arg;
185 heads = argv;
186 nr_heads = argc - i -1;
187 break;
188 }
189 if (!dest)
190 usage(send_pack_usage);
191 pid = git_connect(fd, dest, exec);
192 if (pid < 0)
193 return 1;
194 ret = send_pack(fd[0], fd[1], nr_heads, heads);
195 close(fd[0]);
196 close(fd[1]);
197 finish_connect(pid);
198 return ret;
199 }