2 * Copyright (c) 2019 Ori Bernstein <ori@openbsd.org>
3 * Copyright (c) 2021 Stefan Sperling <stsp@openbsd.org>
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
25 #include "got_error.h"
27 const struct got_error *
28 got_pkt_readn(ssize_t *off, int fd, void *buf, size_t n)
34 r = read(fd, buf + *off, n - *off);
36 return got_error_from_errno("read");
44 const struct got_error *
45 got_pkt_flushpkt(int fd, int chattygot)
50 fprintf(stderr, "%s: writepkt: 0000\n", getprogname());
52 w = write(fd, "0000", 4);
54 return got_error_from_errno("write");
56 return got_error(GOT_ERR_IO);
61 * Packet header contains a 4-byte hexstring which specifies the length
62 * of data which follows.
64 const struct got_error *
65 got_pkt_readhdr(int *datalen, int fd, int chattygot)
67 static const struct got_error *err = NULL;
76 err = got_pkt_readn(&r, fd, lenstr, 4);
82 fprintf(stderr, "%s: readpkt: 0000\n", getprogname());
86 return got_error_msg(GOT_ERR_BAD_PACKET,
87 "wrong packet header length");
90 for (i = 0; i < 4; i++) {
91 if (!isprint((unsigned char)lenstr[i]))
92 return got_error_msg(GOT_ERR_BAD_PACKET,
93 "unprintable character in packet length field");
95 for (i = 0; i < 4; i++) {
96 if (!isxdigit((unsigned char)lenstr[i])) {
98 fprintf(stderr, "%s: bad length: '%s'\n",
99 getprogname(), lenstr);
100 return got_error_msg(GOT_ERR_BAD_PACKET,
101 "packet length not specified in hex");
105 len = strtol(lenstr, &e, 16);
106 if (lenstr[0] == '\0' || *e != '\0')
107 return got_error(GOT_ERR_BAD_PACKET);
108 if (errno == ERANGE && (len == LONG_MAX || len == LONG_MIN))
109 return got_error_msg(GOT_ERR_BAD_PACKET, "bad packet length");
110 if (len > INT_MAX || len < INT_MIN)
111 return got_error_msg(GOT_ERR_BAD_PACKET, "bad packet length");
116 return got_error_msg(GOT_ERR_BAD_PACKET, "packet too short");
123 const struct got_error *
124 got_pkt_readpkt(int *outlen, int fd, char *buf, int buflen, int chattygot)
126 const struct got_error *err = NULL;
130 err = got_pkt_readhdr(&datalen, fd, chattygot);
134 if (datalen > buflen)
135 return got_error(GOT_ERR_NO_SPACE);
137 err = got_pkt_readn(&n, fd, buf, datalen);
141 return got_error_msg(GOT_ERR_BAD_PACKET, "short packet");
144 fprintf(stderr, "%s: readpkt: %zd:\t", getprogname(), n);
145 for (i = 0; i < n; i++) {
147 fputc(buf[i], stderr);
149 fprintf(stderr, "[0x%.2x]", buf[i]);
158 const struct got_error *
159 got_pkt_writepkt(int fd, char *buf, int nbuf, int chattygot)
165 if (snprintf(len, sizeof(len), "%04x", nbuf + 4) >= sizeof(len))
166 return got_error(GOT_ERR_NO_SPACE);
167 w = write(fd, len, 4);
169 return got_error_from_errno("write");
171 return got_error(GOT_ERR_IO);
172 w = write(fd, buf, nbuf);
174 return got_error_from_errno("write");
176 return got_error(GOT_ERR_IO);
178 fprintf(stderr, "%s: writepkt: %s:\t", getprogname(), len);
179 for (i = 0; i < nbuf; i++) {
181 fputc(buf[i], stderr);
183 fprintf(stderr, "[0x%.2x]", buf[i]);