2 * Copyright (c) 2018, 2019 Stefan Sperling <stsp@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>
31 #include "got_compat.h"
33 #include "got_error.h"
34 #include "got_object.h"
36 #include "got_lib_delta.h"
37 #include "got_lib_inflate.h"
38 #include "got_lib_object.h"
39 #include "got_lib_object_parse.h"
40 #include "got_lib_privsep.h"
41 #include "got_lib_sha1.h"
43 static volatile sig_atomic_t sigint_received
;
46 catch_sigint(int signo
)
52 main(int argc
, char *argv
[])
54 const struct got_error
*err
= NULL
;
58 signal(SIGINT
, catch_sigint
);
60 imsg_init(&ibuf
, GOT_IMSG_FD_CHILD
);
63 /* revoke access to most system calls */
64 if (pledge("stdio recvfd", NULL
) == -1) {
65 err
= got_error_from_errno("pledge");
66 got_privsep_send_error(&ibuf
, err
);
72 struct imsg imsg
, imsg_outfd
;
75 struct got_object
*obj
= NULL
;
77 struct got_object_id id
;
78 struct got_object_id expected_id
;
79 struct got_inflate_checksum csum
;
83 memset(&csum
, 0, sizeof(csum
));
84 csum
.output_sha1
= &sha1_ctx
;
86 memset(&imsg
, 0, sizeof(imsg
));
88 memset(&imsg_outfd
, 0, sizeof(imsg_outfd
));
91 if (sigint_received
) {
92 err
= got_error(GOT_ERR_CANCELLED
);
96 err
= got_privsep_recv_imsg(&imsg
, &ibuf
, 0);
98 if (err
->code
== GOT_ERR_PRIVSEP_PIPE
)
103 if (imsg
.hdr
.type
== GOT_IMSG_STOP
)
106 if (imsg
.hdr
.type
!= GOT_IMSG_BLOB_REQUEST
) {
107 err
= got_error(GOT_ERR_PRIVSEP_MSG
);
111 datalen
= imsg
.hdr
.len
- IMSG_HEADER_SIZE
;
112 if (datalen
!= sizeof(expected_id
)) {
113 err
= got_error(GOT_ERR_PRIVSEP_LEN
);
116 memcpy(&expected_id
, imsg
.data
, sizeof(expected_id
));
119 err
= got_error(GOT_ERR_PRIVSEP_NO_FD
);
123 err
= got_privsep_recv_imsg(&imsg_outfd
, &ibuf
, 0);
125 if (imsg
.hdr
.len
== 0)
130 if (imsg_outfd
.hdr
.type
== GOT_IMSG_STOP
)
133 if (imsg_outfd
.hdr
.type
!= GOT_IMSG_BLOB_OUTFD
) {
134 err
= got_error(GOT_ERR_PRIVSEP_MSG
);
138 datalen
= imsg_outfd
.hdr
.len
- IMSG_HEADER_SIZE
;
140 err
= got_error(GOT_ERR_PRIVSEP_LEN
);
143 if (imsg_outfd
.fd
== -1) {
144 err
= got_error(GOT_ERR_PRIVSEP_NO_FD
);
148 err
= got_object_read_header(&obj
, imsg
.fd
);
152 if (lseek(imsg
.fd
, SEEK_SET
, 0) == -1) {
153 err
= got_error_from_errno("lseek");
157 f
= fdopen(imsg
.fd
, "rb");
159 err
= got_error_from_errno("fdopen");
163 if (obj
->size
+ obj
->hdrlen
<=
164 GOT_PRIVSEP_INLINE_BLOB_DATA_MAX
) {
165 err
= got_inflate_to_mem(&buf
, &size
, NULL
, &csum
, f
);
169 err
= got_inflate_to_fd(&size
, f
, &csum
, imsg_outfd
.fd
);
173 SHA1Final(id
.sha1
, &sha1_ctx
);
174 if (memcmp(expected_id
.sha1
, id
.sha1
, SHA1_DIGEST_LENGTH
) != 0) {
175 char buf
[SHA1_DIGEST_STRING_LENGTH
];
176 err
= got_error_fmt(GOT_ERR_OBJ_CSUM
,
177 "checksum failure for object %s",
178 got_sha1_digest_to_str(expected_id
.sha1
, buf
,
183 if (size
< obj
->hdrlen
) {
184 err
= got_error(GOT_ERR_BAD_OBJ_HDR
);
188 err
= got_privsep_send_blob(&ibuf
, size
, obj
->hdrlen
, buf
);
192 if (fclose(f
) == EOF
&& err
== NULL
)
193 err
= got_error_from_errno("fclose");
194 } else if (imsg
.fd
!= -1) {
195 if (close(imsg
.fd
) == -1 && err
== NULL
)
196 err
= got_error_from_errno("close");
198 if (imsg_outfd
.fd
!= -1) {
199 if (close(imsg_outfd
.fd
) == -1 && err
== NULL
)
200 err
= got_error_from_errno("close");
204 imsg_free(&imsg_outfd
);
206 got_object_close(obj
);
213 if (!sigint_received
&& err
->code
!= GOT_ERR_PRIVSEP_PIPE
) {
214 fprintf(stderr
, "%s: %s\n", getprogname(), err
->msg
);
215 got_privsep_send_error(&ibuf
, err
);
218 if (close(GOT_IMSG_FD_CHILD
) == -1 && err
== NULL
)
219 err
= got_error_from_errno("close");