2 * Copyright (c) 2018, 2019 Ori Bernstein <ori@openbsd.org>
4 * Permission to use, copy, modify, and distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice appear in all copies.
8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 #include <sys/types.h>
19 #include <sys/queue.h>
21 #include <sys/socket.h>
23 #include <sys/syslimits.h>
24 #include <sys/resource.h>
25 #include <sys/socket.h>
41 #include <netinet/in.h>
43 #include "got_error.h"
44 #include "got_reference.h"
45 #include "got_repository.h"
47 #include "got_cancel.h"
48 #include "got_worktree.h"
49 #include "got_object.h"
50 #include "got_opentemp.h"
52 #include "got_lib_delta.h"
53 #include "got_lib_inflate.h"
54 #include "got_lib_object.h"
55 #include "got_lib_object_parse.h"
56 #include "got_lib_object_create.h"
57 #include "got_lib_pack.h"
58 #include "got_lib_sha1.h"
59 #include "got_lib_privsep.h"
60 #include "got_lib_object_cache.h"
61 #include "got_lib_repository.h"
63 #define GOT_PROTOMAX 64
64 #define GOT_HOSTMAX 256
65 #define GOT_PATHMAX 512
66 #define GOT_REPOMAX 256
67 #define GOT_PORTMAX 16
68 #define GOT_URIMAX 1024
73 char *p, namebuf[PATH_MAX];
79 if (strlcpy(namebuf, path, sizeof(namebuf)) >= sizeof(namebuf)) {
91 if (mkdir(namebuf, 0755) != 0) {
92 int mkdir_errno = errno;
93 if (stat(path, &sb) == -1) {
94 /* Not there; use mkdir()s errno */
98 if (!S_ISDIR(sb.st_mode)) {
99 /* Is there, but isn't a directory */
114 hassuffix(char *base, char *suf)
120 if (ns <= nb && strcmp(base + (nb - ns), suf) == 0)
126 grab(char *dst, int n, char *p, char *e)
132 errno = ENAMETOOLONG;
135 return strlcpy(dst, p, l + 1);
139 got_dial_ssh(char *host, char *port, char *path, char *direction)
153 snprintf(cmd, sizeof(cmd), "git-%s-pack", direction);
154 execlp("ssh", "ssh", host, cmd, path, NULL);
163 got_dial_git(char *host, char *port, char *path, char *direction)
165 struct addrinfo hints, *servinfo, *p;
169 memset(&hints, 0, sizeof hints);
170 hints.ai_family = AF_UNSPEC;
171 hints.ai_socktype = SOCK_STREAM;
172 if (getaddrinfo(host, port, &hints, &servinfo) != 0)
175 for (p = servinfo; p != NULL; p = p->ai_next) {
176 if ((fd = socket(p->ai_family, p->ai_socktype,
177 p->ai_protocol)) == -1)
179 if (connect(fd, p->ai_addr, p->ai_addrlen) == 0)
186 if ((l = asprintf(&cmd, "git-%s-pack %s\n", direction, path)) == -1)
188 if ((l = asprintf(&pkt, "%04x%s", l+4, cmd)) == -1)
190 r = write(fd, pkt, l);
201 got_parse_uri(char *uri, char *proto, char *host, char *port, char *path, char *repo)
206 p = strstr(uri, "://");
208 //werrstr("missing protocol");
211 if (grab(proto, GOT_PROTOMAX, uri, p) == -1)
213 hasport = (strcmp(proto, "git") == 0 || strstr(proto, "http") == proto);
223 if (p == NULL || strlen(p) == 1) {
224 //werrstr("missing path");
228 q = memchr(s, ':', p - s);
230 grab(host, GOT_HOSTMAX, s, q);
231 grab(port, GOT_PORTMAX, q + 1, p);
233 grab(host, GOT_HOSTMAX, s, p);
234 snprintf(port, GOT_PORTMAX, "9418");
237 snprintf(path, GOT_PATHMAX, "%s", p);
238 p = strrchr(p, '/') + 1;
239 if (!p || strlen(p) == 0) {
240 //werrstr("missing repository in uri");
244 if (hassuffix(p, ".git"))
246 grab(repo, GOT_REPOMAX, p, p + n);
250 const struct got_error*
251 got_clone(char *uri, char *branch_filter, char *dirname)
253 char proto[GOT_PROTOMAX], host[GOT_HOSTMAX], port[GOT_PORTMAX];
254 char repo[GOT_REPOMAX], path[GOT_PATHMAX];
255 int imsg_fetchfds[2], imsg_idxfds[2], fetchfd;
256 int packfd = -1, npackfd, idxfd = -1, nidxfd, status;
257 struct got_object_id packhash;
258 const struct got_error *err;
261 char *packpath = NULL, *idxpath = NULL;
264 if (got_parse_uri(uri, proto, host, port, path, repo) == -1)
265 return got_error(GOT_ERR_PARSE_URI);
268 err = got_repo_init(dirname);
272 return got_error_from_errno("enter new repo");
273 if (mkpath(".git/objects/pack") == -1)
274 return got_error_from_errno("mkpath");
275 err = got_opentemp_named_fd(&packpath, &packfd,
276 ".git/objects/pack/fetching.pack");
279 npackfd = dup(packfd);
281 return got_error_from_errno("dup");
282 err = got_opentemp_named_fd(&idxpath, &idxfd,
283 ".git/objects/pack/fetching.idx");
288 return got_error_from_errno("dup");
290 if (strcmp(proto, "ssh") == 0 || strcmp(proto, "git+ssh") == 0)
291 fetchfd = got_dial_ssh(host, port, path, "upload");
292 else if (strcmp(proto, "git") == 0)
293 fetchfd = got_dial_git(host, port, path, "upload");
294 else if (strcmp(proto, "http") == 0 || strcmp(proto, "git+http") == 0)
295 err = got_error(GOT_ERR_BAD_PROTO);
297 err = got_error(GOT_ERR_BAD_PROTO);
300 err = got_error_from_errno("dial uri");
301 if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fetchfds) == -1)
302 return got_error_from_errno("socketpair");
306 return got_error_from_errno("fork");
308 got_privsep_exec_child(imsg_fetchfds, GOT_PATH_PROG_FETCH_PACK, ".");
311 if (close(imsg_fetchfds[1]) != 0)
312 return got_error_from_errno("close");
313 imsg_init(&ibuf, imsg_fetchfds[0]);
314 err = got_privsep_send_fetch_req(&ibuf, fetchfd);
317 err = got_privsep_wait_ack(&ibuf);
320 err = got_privsep_send_tmpfd(&ibuf, npackfd);
323 npackfd = dup(packfd);
325 return got_error_from_errno("dup");
326 err = got_privsep_wait_fetch_done(&ibuf, &packhash);
329 if (waitpid(pid, &status, 0) == -1)
330 return got_error_from_errno("child exit");
332 if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_idxfds) == -1)
333 return got_error_from_errno("socketpair");
336 return got_error_from_errno("fork");
338 got_privsep_exec_child(imsg_idxfds, GOT_PATH_PROG_INDEX_PACK, ".");
339 if (close(imsg_idxfds[1]) != 0)
340 return got_error_from_errno("close");
341 imsg_init(&ibuf, imsg_idxfds[0]);
343 err = got_privsep_send_index_pack_req(&ibuf, npackfd, packhash);
346 err = got_privsep_wait_ack(&ibuf);
349 err = got_privsep_send_tmpfd(&ibuf, nidxfd);
352 err = got_privsep_wait_index_pack_done(&ibuf);
356 if (close(imsg_idxfds[0]) == -1)
357 return got_error_from_errno("close child");
358 if (waitpid(pid, &status, 0) == -1)
359 return got_error_from_errno("child exit");