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 <string.h>
18 
19 #include "gmid.h"
20 
21 int
22 main(int argc, char **argv)
23 {
24 	struct iri iri;
25 	struct tls_config *conf;
26 	struct tls *ctx;
27 	char iribuf[GEMINI_URL_LEN], req[GEMINI_URL_LEN], buf[1024];
28 	const char *parse_err = "unknown error", *port = "1965";
29 	char *t;
30 	int ch, flag2, flag3, bflag, cflag, hflag, Nflag, Vflag;
31 	ssize_t len;
32 
33 	flag2 = flag3 = bflag = cflag = hflag = Nflag = Vflag = 0;
34 	while ((ch = getopt(argc, argv, "23cbhNV")) != -1) {
35 		switch (ch) {
36 		case '2':
37 			flag2 = 1;
38 			break;
39 		case '3':
40 			flag3 = 1;
41 			break;
42 		case 'b':
43 			bflag = 1;
44 			break;
45 		case 'c':
46 			cflag = 1;
47 			break;
48 		case 'h':
49 			hflag = 1;
50 			break;
51 		case 'N':
52 			Nflag = 1;
53 			break;
54 		case 'V':
55 			Vflag = 1;
56 			break;
57 		default:
58 			fprintf(stderr, "USAGE: %s [-23cbhNV]", *argv);
59 			return 1;
60 		}
61 	}
62 	argc -= optind;
63 	argv += optind;
64 
65 	if ((bflag + cflag + hflag + Vflag) > 1)
66 		errx(1, "only one of bchr flags can be used.");
67 
68 	if (flag2 + flag3 > 1)
69 		errx(1, "only -2 or -3 can be specified at the same time.");
70 
71 	if (argc != 1)
72 		errx(1, "missing IRI");
73 
74 	if (strlcpy(iribuf, argv[0], sizeof(iribuf)) >= sizeof(iribuf))
75 		errx(1, "request too long: %s", argv[0]);
76 	if (strlcpy(req, argv[0], sizeof(req)) >= sizeof(iribuf))
77 		errx(1, "request too long: %s", argv[0]);
78 	if (strlcat(req, "\r\n", sizeof(req)) >= sizeof(req))
79 		errx(1, "request too long: %s", argv[0]);
80 
81 	if (!parse_iri(iribuf, &iri, &parse_err))
82 		errx(1, "invalid IRI: %s", parse_err);
83 
84 	if (Vflag)
85 		errx(0, "IRI: OK");
86 
87 	if ((conf = tls_config_new()) == NULL)
88 		errx(1, "tls_config_new");
89 
90 	tls_config_insecure_noverifycert(conf);
91 	if (Nflag)
92 		tls_config_insecure_noverifyname(conf);
93 
94 	if (flag2 && tls_config_set_protocols(conf, TLS_PROTOCOL_TLSv1_2) == -1)
95 		errx(1, "cannot set TLSv1.2");
96 	if (flag3 && tls_config_set_protocols(conf, TLS_PROTOCOL_TLSv1_3) == -1)
97 		errx(1, "cannot set TLSv1.3");
98 
99 	if ((ctx = tls_client()) == NULL)
100 		errx(1, "tls_client creation failed");
101 
102 	if (tls_configure(ctx, conf) == -1)
103 		errx(1, "tls_configure: %s", tls_error(ctx));
104 
105 	if (*iri.port == '\0')
106 		iri.port = (char*)port;
107 	if (tls_connect(ctx, iri.host, iri.port) == -1)
108 		errx(1, "tls_connect: %s", tls_error(ctx));
109 
110 	tls_write(ctx, req, strlen(req));
111 	/* if (tls_write(ctx, req, strlen(req)) != -1) */
112 	/*	errx(1, "tls_write: %s", tls_error(ctx)); */
113 
114 	for (;;) {
115 		len = tls_read(ctx, buf, sizeof(buf));
116 		if (len == 0 || len == -1)
117 			break;
118 
119 		if (bflag) {
120 			bflag = 0;
121 			if ((t = strchr(buf, '\r')) != NULL)
122 				t += 2;
123 			else if ((t = strchr(buf, '\n')) != NULL)
124 				t += 1;
125 			else
126 				continue;
127 			len -= t - buf;
128 			write(1, t, len);
129 			continue;
130 		}
131 
132 		if (cflag) {
133 			write(1, buf, 2);
134 			write(1, "\n", 1);
135 			break;
136 		}
137 
138 		if (hflag) {
139 			t = strchr(buf, '\r');
140 			if (t == NULL)
141 				t = strchr(buf, '\n');
142 			if (t == NULL)
143 				t = &buf[len];
144                         write(1, buf, t - buf);
145 			write(1, "\n", 1);
146 			break;
147 		}
148 
149 		write(1, buf, len);
150 	}
151 
152 	tls_close(ctx);
153 	tls_free(ctx);
154 
155 	return 0;
156 }