1 /*
2  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3  *
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.
7  *
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.
15  */
16 
17 #include "../gmid.h"
18 
19 #include <string.h>
20 
21 int flag2, flag3, bflag, cflag, hflag, Nflag, Vflag, vflag;
22 const char *cert, *key;
23 
24 static void
25 timeout(int signo)
26 {
27 	dprintf(2, "%s: timer expired\n", getprogname());
28 	exit(1);
29 }
30 
31 int
32 main(int argc, char **argv)
33 {
34 	struct iri iri;
35 	struct tls_config *conf;
36 	struct tls *ctx;
37 	char iribuf[GEMINI_URL_LEN], buf[GEMINI_URL_LEN];
38 	const char *parse_err = "unknown error", *port = "1965", *errstr;
39 	const char *hostname;
40 	char *t;
41 	int ch, handshake, timer;
42 	ssize_t len;
43 
44 	hostname = NULL;
45 	while ((ch = getopt(argc, argv, "23C:cbH:hK:NT:Vv")) != -1) {
46 		switch (ch) {
47 		case '2':
48 			flag2 = 1;
49 			break;
50 		case '3':
51 			flag3 = 1;
52 			break;
53 		case 'b':
54 			bflag = 1;
55 			break;
56 		case 'C':
57 			cert = optarg;
58 			break;
59 		case 'c':
60 			cflag = 1;
61 			break;
62 		case 'H':
63 			hostname = optarg;
64 			break;
65 		case 'h':
66 			hflag = 1;
67 			break;
68 		case 'K':
69 			key = optarg;
70 			break;
71 		case 'N':
72 			Nflag = 1;
73 			break;
74 		case 'T':
75 			timer = strtonum(optarg, 1, 1000, &errstr);
76 			if (errstr != NULL)
77 				errx(1, "timeout is %s: %s", errstr, optarg);
78 			signal(SIGALRM, timeout);
79 			alarm(timer);
80 			break;
81 		case 'V':
82 			Vflag = 1;
83 			break;
84 		case 'v':
85 			vflag = 1;
86 			break;
87 		default:
88 			fprintf(stderr, "USAGE: %s [-23cbhNVv] [-H hostname]\n",
89 			    *argv);
90 			return 1;
91 		}
92 	}
93 	argc -= optind;
94 	argv += optind;
95 
96 	if ((bflag + cflag + hflag + Vflag) > 1)
97 		errx(1, "only one of bchr flags can be used.");
98 
99 	if (flag2 + flag3 > 1)
100 		errx(1, "only -2 or -3 can be specified at the same time.");
101 
102 	if ((cert != NULL && key == NULL) || (cert == NULL && key != NULL))
103 		errx(1, "missing certificate or key");
104 
105 	if (argc != 1)
106 		errx(1, "missing IRI");
107 
108 	if (strlcpy(iribuf, argv[0], sizeof(iribuf)) >= sizeof(iribuf))
109 		errx(1, "request too long: %s", argv[0]);
110 	if (strlcpy(buf, argv[0], sizeof(buf)) >= sizeof(iribuf))
111 		errx(1, "request too long: %s", argv[0]);
112 	if (strlcat(buf, "\r\n", sizeof(buf)) >= sizeof(buf))
113 		errx(1, "request too long: %s", argv[0]);
114 
115 	if (!parse_iri(iribuf, &iri, &parse_err))
116 		errx(1, "invalid IRI: %s", parse_err);
117 
118 	if (Vflag)
119 		errx(0, "IRI: OK");
120 
121 	if ((conf = tls_config_new()) == NULL)
122 		errx(1, "tls_config_new");
123 
124 	tls_config_insecure_noverifycert(conf);
125 	if (Nflag)
126 		tls_config_insecure_noverifyname(conf);
127 
128 	if (flag2 && tls_config_set_protocols(conf, TLS_PROTOCOL_TLSv1_2) == -1)
129 		errx(1, "cannot set TLSv1.2");
130 	if (flag3 && tls_config_set_protocols(conf, TLS_PROTOCOL_TLSv1_3) == -1)
131 		errx(1, "cannot set TLSv1.3");
132 
133 	if (cert != NULL && tls_config_set_keypair_file(conf, cert, key))
134 		errx(1, "couldn't load cert: %s", cert);
135 
136 	if ((ctx = tls_client()) == NULL)
137 		errx(1, "tls_client creation failed");
138 
139 	if (tls_configure(ctx, conf) == -1)
140 		errx(1, "tls_configure: %s", tls_error(ctx));
141 
142 	if (*iri.port != '\0')
143 		port = iri.port;
144 
145 	if (hostname == NULL)
146 		hostname = iri.host;
147 
148 	if (tls_connect_servername(ctx, iri.host, port, hostname) == -1)
149 		errx(1, "tls_connect: %s", tls_error(ctx));
150 
151 	for (handshake = 0; !handshake;) {
152 		switch (tls_handshake(ctx)) {
153 		case 0:
154 		case -1:
155 			handshake = 1;
156 			break;
157 		}
158 	}
159 
160 	if (vflag)
161 		printf("%s", buf);
162 	if (tls_write(ctx, buf, strlen(buf)) == -1)
163 		errx(1, "tls_write: %s", tls_error(ctx));
164 
165 	for (;;) {
166 		switch (len = tls_read(ctx, buf, sizeof(buf))) {
167 		case 0:
168 		case -1:
169 			goto end;
170 		case TLS_WANT_POLLIN:
171 		case TLS_WANT_POLLOUT:
172 			continue;
173 		}
174 
175 		if (bflag) {
176 			bflag = 0;
177 			if ((t = strchr(buf, '\r')) != NULL)
178 				t += 2;
179 			else if ((t = strchr(buf, '\n')) != NULL)
180 				t += 1;
181 			else
182 				continue;
183 			len -= t - buf;
184 			write(1, t, len);
185 			continue;
186 		}
187 
188 		if (cflag) {
189 			write(1, buf, 2);
190 			write(1, "\n", 1);
191 			break;
192 		}
193 
194 		if (hflag) {
195 			t = strchr(buf, '\r');
196 			if (t == NULL)
197 				t = strchr(buf, '\n');
198 			if (t == NULL)
199 				t = &buf[len];
200                         write(1, buf, t - buf);
201 			write(1, "\n", 1);
202 			break;
203 		}
204 
205 		write(1, buf, len);
206 	}
207 end:
208 
209 	tls_close(ctx);
210 	tls_free(ctx);
211 
212 	return 0;
213 }