Blame


1 497977d5 2021-01-23 op /*
2 497977d5 2021-01-23 op * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3 497977d5 2021-01-23 op *
4 497977d5 2021-01-23 op * Permission to use, copy, modify, and distribute this software for any
5 497977d5 2021-01-23 op * purpose with or without fee is hereby granted, provided that the above
6 497977d5 2021-01-23 op * copyright notice and this permission notice appear in all copies.
7 497977d5 2021-01-23 op *
8 497977d5 2021-01-23 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 497977d5 2021-01-23 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 497977d5 2021-01-23 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 497977d5 2021-01-23 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 497977d5 2021-01-23 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 497977d5 2021-01-23 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 497977d5 2021-01-23 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 497977d5 2021-01-23 op */
16 497977d5 2021-01-23 op
17 497977d5 2021-01-23 op #include "gmid.h"
18 497977d5 2021-01-23 op
19 52418c8d 2021-02-12 op #include <string.h>
20 52418c8d 2021-02-12 op
21 58173ca2 2021-01-27 op int flag2, flag3, bflag, cflag, hflag, Nflag, Vflag, vflag;
22 31b3662c 2021-02-09 op const char *cert, *key;
23 58173ca2 2021-01-27 op
24 497977d5 2021-01-23 op int
25 497977d5 2021-01-23 op main(int argc, char **argv)
26 497977d5 2021-01-23 op {
27 497977d5 2021-01-23 op struct iri iri;
28 497977d5 2021-01-23 op struct tls_config *conf;
29 497977d5 2021-01-23 op struct tls *ctx;
30 9adde3d8 2021-01-23 op char iribuf[GEMINI_URL_LEN], buf[GEMINI_URL_LEN];
31 497977d5 2021-01-23 op const char *parse_err = "unknown error", *port = "1965";
32 58173ca2 2021-01-27 op const char *hostname;
33 497977d5 2021-01-23 op char *t;
34 58173ca2 2021-01-27 op int ch;
35 58173ca2 2021-01-27 op int handshake;
36 497977d5 2021-01-23 op ssize_t len;
37 497977d5 2021-01-23 op
38 58173ca2 2021-01-27 op hostname = NULL;
39 31b3662c 2021-02-09 op while ((ch = getopt(argc, argv, "23C:cbH:hK:NVv")) != -1) {
40 497977d5 2021-01-23 op switch (ch) {
41 497977d5 2021-01-23 op case '2':
42 497977d5 2021-01-23 op flag2 = 1;
43 497977d5 2021-01-23 op break;
44 497977d5 2021-01-23 op case '3':
45 497977d5 2021-01-23 op flag3 = 1;
46 497977d5 2021-01-23 op break;
47 497977d5 2021-01-23 op case 'b':
48 497977d5 2021-01-23 op bflag = 1;
49 497977d5 2021-01-23 op break;
50 31b3662c 2021-02-09 op case 'C':
51 31b3662c 2021-02-09 op cert = optarg;
52 31b3662c 2021-02-09 op break;
53 497977d5 2021-01-23 op case 'c':
54 497977d5 2021-01-23 op cflag = 1;
55 497977d5 2021-01-23 op break;
56 58173ca2 2021-01-27 op case 'H':
57 58173ca2 2021-01-27 op hostname = optarg;
58 58173ca2 2021-01-27 op break;
59 497977d5 2021-01-23 op case 'h':
60 497977d5 2021-01-23 op hflag = 1;
61 497977d5 2021-01-23 op break;
62 31b3662c 2021-02-09 op case 'K':
63 31b3662c 2021-02-09 op key = optarg;
64 31b3662c 2021-02-09 op break;
65 497977d5 2021-01-23 op case 'N':
66 497977d5 2021-01-23 op Nflag = 1;
67 497977d5 2021-01-23 op break;
68 497977d5 2021-01-23 op case 'V':
69 497977d5 2021-01-23 op Vflag = 1;
70 497977d5 2021-01-23 op break;
71 58173ca2 2021-01-27 op case 'v':
72 58173ca2 2021-01-27 op vflag = 1;
73 58173ca2 2021-01-27 op break;
74 497977d5 2021-01-23 op default:
75 58173ca2 2021-01-27 op fprintf(stderr, "USAGE: %s [-23cbhNVv] [-H hostname]\n",
76 58173ca2 2021-01-27 op *argv);
77 497977d5 2021-01-23 op return 1;
78 497977d5 2021-01-23 op }
79 497977d5 2021-01-23 op }
80 497977d5 2021-01-23 op argc -= optind;
81 497977d5 2021-01-23 op argv += optind;
82 497977d5 2021-01-23 op
83 497977d5 2021-01-23 op if ((bflag + cflag + hflag + Vflag) > 1)
84 497977d5 2021-01-23 op errx(1, "only one of bchr flags can be used.");
85 497977d5 2021-01-23 op
86 497977d5 2021-01-23 op if (flag2 + flag3 > 1)
87 497977d5 2021-01-23 op errx(1, "only -2 or -3 can be specified at the same time.");
88 497977d5 2021-01-23 op
89 31b3662c 2021-02-09 op if ((cert != NULL && key == NULL) || (cert == NULL && key != NULL))
90 31b3662c 2021-02-09 op errx(1, "missing certificate or key");
91 31b3662c 2021-02-09 op
92 497977d5 2021-01-23 op if (argc != 1)
93 497977d5 2021-01-23 op errx(1, "missing IRI");
94 497977d5 2021-01-23 op
95 497977d5 2021-01-23 op if (strlcpy(iribuf, argv[0], sizeof(iribuf)) >= sizeof(iribuf))
96 497977d5 2021-01-23 op errx(1, "request too long: %s", argv[0]);
97 9adde3d8 2021-01-23 op if (strlcpy(buf, argv[0], sizeof(buf)) >= sizeof(iribuf))
98 497977d5 2021-01-23 op errx(1, "request too long: %s", argv[0]);
99 9adde3d8 2021-01-23 op if (strlcat(buf, "\r\n", sizeof(buf)) >= sizeof(buf))
100 497977d5 2021-01-23 op errx(1, "request too long: %s", argv[0]);
101 497977d5 2021-01-23 op
102 497977d5 2021-01-23 op if (!parse_iri(iribuf, &iri, &parse_err))
103 497977d5 2021-01-23 op errx(1, "invalid IRI: %s", parse_err);
104 497977d5 2021-01-23 op
105 497977d5 2021-01-23 op if (Vflag)
106 497977d5 2021-01-23 op errx(0, "IRI: OK");
107 497977d5 2021-01-23 op
108 497977d5 2021-01-23 op if ((conf = tls_config_new()) == NULL)
109 497977d5 2021-01-23 op errx(1, "tls_config_new");
110 497977d5 2021-01-23 op
111 497977d5 2021-01-23 op tls_config_insecure_noverifycert(conf);
112 497977d5 2021-01-23 op if (Nflag)
113 497977d5 2021-01-23 op tls_config_insecure_noverifyname(conf);
114 497977d5 2021-01-23 op
115 497977d5 2021-01-23 op if (flag2 && tls_config_set_protocols(conf, TLS_PROTOCOL_TLSv1_2) == -1)
116 497977d5 2021-01-23 op errx(1, "cannot set TLSv1.2");
117 497977d5 2021-01-23 op if (flag3 && tls_config_set_protocols(conf, TLS_PROTOCOL_TLSv1_3) == -1)
118 497977d5 2021-01-23 op errx(1, "cannot set TLSv1.3");
119 497977d5 2021-01-23 op
120 31b3662c 2021-02-09 op if (cert != NULL && tls_config_set_keypair_file(conf, cert, key))
121 31b3662c 2021-02-09 op errx(1, "couldn't load cert: %s", cert);
122 31b3662c 2021-02-09 op
123 497977d5 2021-01-23 op if ((ctx = tls_client()) == NULL)
124 497977d5 2021-01-23 op errx(1, "tls_client creation failed");
125 497977d5 2021-01-23 op
126 497977d5 2021-01-23 op if (tls_configure(ctx, conf) == -1)
127 497977d5 2021-01-23 op errx(1, "tls_configure: %s", tls_error(ctx));
128 497977d5 2021-01-23 op
129 d760973a 2021-01-23 op if (*iri.port != '\0')
130 d760973a 2021-01-23 op port = iri.port;
131 58173ca2 2021-01-27 op
132 58173ca2 2021-01-27 op if (hostname == NULL)
133 58173ca2 2021-01-27 op hostname = iri.host;
134 58173ca2 2021-01-27 op
135 58173ca2 2021-01-27 op if (tls_connect_servername(ctx, iri.host, port, hostname) == -1)
136 497977d5 2021-01-23 op errx(1, "tls_connect: %s", tls_error(ctx));
137 497977d5 2021-01-23 op
138 58173ca2 2021-01-27 op for (handshake = 0; !handshake;) {
139 58173ca2 2021-01-27 op switch (tls_handshake(ctx)) {
140 58173ca2 2021-01-27 op case 0:
141 58173ca2 2021-01-27 op case -1:
142 58173ca2 2021-01-27 op handshake = 1;
143 58173ca2 2021-01-27 op break;
144 58173ca2 2021-01-27 op }
145 58173ca2 2021-01-27 op }
146 497977d5 2021-01-23 op
147 58173ca2 2021-01-27 op if (vflag)
148 58173ca2 2021-01-27 op printf("%s", buf);
149 58173ca2 2021-01-27 op if (tls_write(ctx, buf, strlen(buf)) == -1)
150 58173ca2 2021-01-27 op errx(1, "tls_write: %s", tls_error(ctx));
151 58173ca2 2021-01-27 op
152 497977d5 2021-01-23 op for (;;) {
153 f62aab51 2021-01-23 op switch (len = tls_read(ctx, buf, sizeof(buf))) {
154 f62aab51 2021-01-23 op case 0:
155 f62aab51 2021-01-23 op case -1:
156 f62aab51 2021-01-23 op goto end;
157 f62aab51 2021-01-23 op case TLS_WANT_POLLIN:
158 f62aab51 2021-01-23 op case TLS_WANT_POLLOUT:
159 f62aab51 2021-01-23 op continue;
160 f62aab51 2021-01-23 op }
161 497977d5 2021-01-23 op
162 497977d5 2021-01-23 op if (bflag) {
163 497977d5 2021-01-23 op bflag = 0;
164 497977d5 2021-01-23 op if ((t = strchr(buf, '\r')) != NULL)
165 497977d5 2021-01-23 op t += 2;
166 497977d5 2021-01-23 op else if ((t = strchr(buf, '\n')) != NULL)
167 497977d5 2021-01-23 op t += 1;
168 497977d5 2021-01-23 op else
169 497977d5 2021-01-23 op continue;
170 497977d5 2021-01-23 op len -= t - buf;
171 497977d5 2021-01-23 op write(1, t, len);
172 497977d5 2021-01-23 op continue;
173 497977d5 2021-01-23 op }
174 497977d5 2021-01-23 op
175 497977d5 2021-01-23 op if (cflag) {
176 497977d5 2021-01-23 op write(1, buf, 2);
177 497977d5 2021-01-23 op write(1, "\n", 1);
178 497977d5 2021-01-23 op break;
179 497977d5 2021-01-23 op }
180 497977d5 2021-01-23 op
181 497977d5 2021-01-23 op if (hflag) {
182 497977d5 2021-01-23 op t = strchr(buf, '\r');
183 497977d5 2021-01-23 op if (t == NULL)
184 497977d5 2021-01-23 op t = strchr(buf, '\n');
185 497977d5 2021-01-23 op if (t == NULL)
186 497977d5 2021-01-23 op t = &buf[len];
187 497977d5 2021-01-23 op write(1, buf, t - buf);
188 497977d5 2021-01-23 op write(1, "\n", 1);
189 497977d5 2021-01-23 op break;
190 497977d5 2021-01-23 op }
191 497977d5 2021-01-23 op
192 497977d5 2021-01-23 op write(1, buf, len);
193 497977d5 2021-01-23 op }
194 f62aab51 2021-01-23 op end:
195 497977d5 2021-01-23 op
196 497977d5 2021-01-23 op tls_close(ctx);
197 497977d5 2021-01-23 op tls_free(ctx);
198 497977d5 2021-01-23 op
199 497977d5 2021-01-23 op return 0;
200 497977d5 2021-01-23 op }