1 44ee1bac 2021-01-27 op /*
  2 d2b941f3 2021-01-28 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 44ee1bac 2021-01-27 op  *
  4 44ee1bac 2021-01-27 op  * Permission to use, copy, modify, and distribute this software for any
  5 44ee1bac 2021-01-27 op  * purpose with or without fee is hereby granted, provided that the above
  6 44ee1bac 2021-01-27 op  * copyright notice and this permission notice appear in all copies.
  7 44ee1bac 2021-01-27 op  *
  8 44ee1bac 2021-01-27 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 44ee1bac 2021-01-27 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 44ee1bac 2021-01-27 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 44ee1bac 2021-01-27 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 44ee1bac 2021-01-27 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 44ee1bac 2021-01-27 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 44ee1bac 2021-01-27 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 44ee1bac 2021-01-27 op  */
 16 44ee1bac 2021-01-27 op 
 17 52418c8d 2021-02-12 op #include "gmid.h"
 18 52418c8d 2021-02-12 op 
 19 44ee1bac 2021-01-27 op #include <errno.h>
 20 44ee1bac 2021-01-27 op #include <string.h>
 21 44ee1bac 2021-01-27 op 
 22 4283d65f 2021-02-10 op #include <openssl/bn.h>
 23 3abf91b0 2021-02-07 op #include <openssl/pem.h>
 24 02be96c6 2021-02-09 op #include <openssl/x509_vfy.h>
 25 02be96c6 2021-02-09 op #include <openssl/x509v3.h>
 26 3abf91b0 2021-02-07 op 
 27 eae52ad4 2023-06-06 op #include "log.h"
 28 df5058c9 2023-06-05 op 
 29 44ee1bac 2021-01-27 op int
 30 44ee1bac 2021-01-27 op starts_with(const char *str, const char *prefix)
 31 44ee1bac 2021-01-27 op {
 32 44ee1bac 2021-01-27 op 	size_t i;
 33 44ee1bac 2021-01-27 op 
 34 44ee1bac 2021-01-27 op 	if (prefix == NULL)
 35 44ee1bac 2021-01-27 op 		return 0;
 36 44ee1bac 2021-01-27 op 
 37 44ee1bac 2021-01-27 op 	for (i = 0; prefix[i] != '\0'; ++i)
 38 44ee1bac 2021-01-27 op 		if (str[i] != prefix[i])
 39 44ee1bac 2021-01-27 op 			return 0;
 40 44ee1bac 2021-01-27 op 	return 1;
 41 44ee1bac 2021-01-27 op }
 42 44ee1bac 2021-01-27 op 
 43 44ee1bac 2021-01-27 op int
 44 44ee1bac 2021-01-27 op ends_with(const char *str, const char *sufx)
 45 44ee1bac 2021-01-27 op {
 46 44ee1bac 2021-01-27 op 	size_t i, j;
 47 44ee1bac 2021-01-27 op 
 48 44ee1bac 2021-01-27 op 	i = strlen(str);
 49 44ee1bac 2021-01-27 op 	j = strlen(sufx);
 50 44ee1bac 2021-01-27 op 
 51 44ee1bac 2021-01-27 op 	if (j > i)
 52 44ee1bac 2021-01-27 op 		return 0;
 53 44ee1bac 2021-01-27 op 
 54 44ee1bac 2021-01-27 op 	i -= j;
 55 44ee1bac 2021-01-27 op 	for (j = 0; str[i] != '\0'; i++, j++)
 56 44ee1bac 2021-01-27 op 		if (str[i] != sufx[j])
 57 44ee1bac 2021-01-27 op 			return 0;
 58 44ee1bac 2021-01-27 op 	return 1;
 59 44ee1bac 2021-01-27 op }
 60 44ee1bac 2021-01-27 op 
 61 44ee1bac 2021-01-27 op ssize_t
 62 44ee1bac 2021-01-27 op filesize(int fd)
 63 44ee1bac 2021-01-27 op {
 64 44ee1bac 2021-01-27 op 	ssize_t len;
 65 44ee1bac 2021-01-27 op 
 66 44ee1bac 2021-01-27 op 	if ((len = lseek(fd, 0, SEEK_END)) == -1)
 67 44ee1bac 2021-01-27 op 		return -1;
 68 44ee1bac 2021-01-27 op 	if (lseek(fd, 0, SEEK_SET) == -1)
 69 44ee1bac 2021-01-27 op 		return -1;
 70 44ee1bac 2021-01-27 op 	return len;
 71 44ee1bac 2021-01-27 op }
 72 bcf5d929 2021-02-01 op 
 73 bcf5d929 2021-02-01 op char *
 74 bcf5d929 2021-02-01 op absolutify_path(const char *path)
 75 bcf5d929 2021-02-01 op {
 76 bcf5d929 2021-02-01 op 	char *wd, *r;
 77 bcf5d929 2021-02-01 op 
 78 bcf5d929 2021-02-01 op 	if (*path == '/') {
 79 bcf5d929 2021-02-01 op 		if ((r = strdup(path)) == NULL)
 80 bcf5d929 2021-02-01 op 			err(1, "strdup");
 81 bcf5d929 2021-02-01 op 		return r;
 82 bcf5d929 2021-02-01 op 	}
 83 bcf5d929 2021-02-01 op 
 84 bcf5d929 2021-02-01 op 	wd = getcwd(NULL, 0);
 85 bcf5d929 2021-02-01 op 	if (asprintf(&r, "%s/%s", wd, path) == -1)
 86 bcf5d929 2021-02-01 op 		err(1, "asprintf");
 87 bcf5d929 2021-02-01 op 	free(wd);
 88 bcf5d929 2021-02-01 op 	return r;
 89 bcf5d929 2021-02-01 op }
 90 ca21e100 2021-02-04 op 
 91 ca21e100 2021-02-04 op char *
 92 ca21e100 2021-02-04 op xstrdup(const char *s)
 93 ca21e100 2021-02-04 op {
 94 ca21e100 2021-02-04 op 	char *d;
 95 ca21e100 2021-02-04 op 
 96 ca21e100 2021-02-04 op 	if ((d = strdup(s)) == NULL)
 97 ca21e100 2021-02-04 op 		err(1, "strdup");
 98 b8e64ccd 2021-03-31 op 	return d;
 99 b8e64ccd 2021-03-31 op }
100 b8e64ccd 2021-03-31 op 
101 b8e64ccd 2021-03-31 op void *
102 b8e64ccd 2021-03-31 op xcalloc(size_t nmemb, size_t size)
103 b8e64ccd 2021-03-31 op {
104 b8e64ccd 2021-03-31 op 	void *d;
105 b8e64ccd 2021-03-31 op 
106 b8e64ccd 2021-03-31 op 	if ((d = calloc(nmemb, size)) == NULL)
107 b8e64ccd 2021-03-31 op 		err(1, "calloc");
108 ca21e100 2021-02-04 op 	return d;
109 ca21e100 2021-02-04 op }
110 3abf91b0 2021-02-07 op 
111 3abf91b0 2021-02-07 op void
112 52418c8d 2021-02-12 op gen_certificate(const char *hostname, const char *certpath, const char *keypath)
113 3abf91b0 2021-02-07 op {
114 4283d65f 2021-02-10 op 	BIGNUM		*e;
115 3abf91b0 2021-02-07 op 	EVP_PKEY	*pkey;
116 3abf91b0 2021-02-07 op 	RSA		*rsa;
117 3abf91b0 2021-02-07 op 	X509		*x509;
118 3abf91b0 2021-02-07 op 	X509_NAME	*name;
119 3abf91b0 2021-02-07 op 	FILE		*f;
120 52418c8d 2021-02-12 op 	const unsigned char *host = (const unsigned char*)hostname;
121 3abf91b0 2021-02-07 op 
122 eae52ad4 2023-06-06 op 	log_info("generating new certificate for %s (it could take a while)",
123 3abf91b0 2021-02-07 op 	    host);
124 3abf91b0 2021-02-07 op 
125 3abf91b0 2021-02-07 op 	if ((pkey = EVP_PKEY_new()) == NULL)
126 df5058c9 2023-06-05 op 		fatalx("couldn't create a new private key");
127 3abf91b0 2021-02-07 op 
128 3abf91b0 2021-02-07 op 	if ((rsa = RSA_new()) == NULL)
129 df5058c9 2023-06-05 op 		fatalx("couldn't generate rsa");
130 3abf91b0 2021-02-07 op 
131 4283d65f 2021-02-10 op 	if ((e = BN_new()) == NULL)
132 df5058c9 2023-06-05 op 		fatalx("couldn't allocate a bignum");
133 4283d65f 2021-02-10 op 
134 1eb3631d 2021-05-12 op 	BN_set_word(e, RSA_F4);
135 4283d65f 2021-02-10 op 	if (!RSA_generate_key_ex(rsa, 4096, e, NULL))
136 df5058c9 2023-06-05 op 		fatalx("couldn't generate a rsa key");
137 3abf91b0 2021-02-07 op 
138 3abf91b0 2021-02-07 op 	if (!EVP_PKEY_assign_RSA(pkey, rsa))
139 df5058c9 2023-06-05 op 		fatalx("couldn't assign the key");
140 3abf91b0 2021-02-07 op 
141 3abf91b0 2021-02-07 op 	if ((x509 = X509_new()) == NULL)
142 df5058c9 2023-06-05 op 		fatalx("couldn't generate the X509 certificate");
143 3abf91b0 2021-02-07 op 
144 1eb3631d 2021-05-12 op 	ASN1_INTEGER_set(X509_get_serialNumber(x509), 0);
145 3abf91b0 2021-02-07 op 	X509_gmtime_adj(X509_get_notBefore(x509), 0);
146 3abf91b0 2021-02-07 op 	X509_gmtime_adj(X509_get_notAfter(x509), 315360000L); /* 10 years */
147 1eb3631d 2021-05-12 op 	X509_set_version(x509, 3);
148 3abf91b0 2021-02-07 op 
149 3abf91b0 2021-02-07 op 	if (!X509_set_pubkey(x509, pkey))
150 df5058c9 2023-06-05 op 		fatalx("couldn't set the public key");
151 3abf91b0 2021-02-07 op 
152 3abf91b0 2021-02-07 op 	name = X509_get_subject_name(x509);
153 3abf91b0 2021-02-07 op 	if (!X509_NAME_add_entry_by_txt(name, "CN", MBSTRING_ASC, host, -1, -1, 0))
154 df5058c9 2023-06-05 op 		fatalx("couldn't add CN to cert");
155 3abf91b0 2021-02-07 op 	X509_set_issuer_name(x509, name);
156 3abf91b0 2021-02-07 op 
157 3abf91b0 2021-02-07 op 	if (!X509_sign(x509, pkey, EVP_sha256()))
158 df5058c9 2023-06-05 op 		fatalx("couldn't sign the certificate");
159 3abf91b0 2021-02-07 op 
160 3abf91b0 2021-02-07 op 	if ((f = fopen(keypath, "w")) == NULL)
161 df5058c9 2023-06-05 op 		fatal("can't open %s", keypath);
162 3abf91b0 2021-02-07 op 	if (!PEM_write_PrivateKey(f, pkey, NULL, NULL, 0, NULL, NULL))
163 df5058c9 2023-06-05 op 		fatalx("couldn't write private key");
164 3abf91b0 2021-02-07 op 	fclose(f);
165 3abf91b0 2021-02-07 op 
166 3abf91b0 2021-02-07 op 	if ((f = fopen(certpath, "w")) == NULL)
167 df5058c9 2023-06-05 op 		fatal("can't open %s", certpath);
168 3abf91b0 2021-02-07 op 	if (!PEM_write_X509(f, x509))
169 df5058c9 2023-06-05 op 		fatalx("couldn't write cert");
170 3abf91b0 2021-02-07 op 	fclose(f);
171 3abf91b0 2021-02-07 op 
172 4283d65f 2021-02-10 op 	BN_free(e);
173 3abf91b0 2021-02-07 op 	X509_free(x509);
174 3abf91b0 2021-02-07 op 	RSA_free(rsa);
175 7b27af83 2022-04-12 op 
176 eae52ad4 2023-06-06 op 	log_info("certificate successfully generated");
177 3abf91b0 2021-02-07 op }
178 02be96c6 2021-02-09 op 
179 02be96c6 2021-02-09 op X509_STORE *
180 02be96c6 2021-02-09 op load_ca(const char *path)
181 02be96c6 2021-02-09 op {
182 02be96c6 2021-02-09 op 	FILE		*f = NULL;
183 02be96c6 2021-02-09 op 	X509		*x = NULL;
184 02be96c6 2021-02-09 op 	X509_STORE	*store;
185 02be96c6 2021-02-09 op 
186 02be96c6 2021-02-09 op 	if ((store = X509_STORE_new()) == NULL)
187 02be96c6 2021-02-09 op 		return NULL;
188 02be96c6 2021-02-09 op 
189 02be96c6 2021-02-09 op 	if ((f = fopen(path, "r")) == NULL)
190 02be96c6 2021-02-09 op 		goto err;
191 02be96c6 2021-02-09 op 
192 02be96c6 2021-02-09 op 	if ((x = PEM_read_X509(f, NULL, NULL, NULL)) == NULL)
193 02be96c6 2021-02-09 op 		goto err;
194 02be96c6 2021-02-09 op 
195 02be96c6 2021-02-09 op 	if (X509_check_ca(x) == 0)
196 02be96c6 2021-02-09 op 		goto err;
197 02be96c6 2021-02-09 op 
198 02be96c6 2021-02-09 op 	if (!X509_STORE_add_cert(store, x))
199 02be96c6 2021-02-09 op 		goto err;
200 02be96c6 2021-02-09 op 
201 02be96c6 2021-02-09 op 	X509_free(x);
202 02be96c6 2021-02-09 op 	fclose(f);
203 02be96c6 2021-02-09 op 	return store;
204 02be96c6 2021-02-09 op 
205 02be96c6 2021-02-09 op err:
206 02be96c6 2021-02-09 op 	X509_STORE_free(store);
207 02be96c6 2021-02-09 op 	if (x != NULL)
208 02be96c6 2021-02-09 op 		X509_free(x);
209 02be96c6 2021-02-09 op 	if (f != NULL)
210 02be96c6 2021-02-09 op 		fclose(f);
211 02be96c6 2021-02-09 op 	return NULL;
212 02be96c6 2021-02-09 op }
213 02be96c6 2021-02-09 op 
214 02be96c6 2021-02-09 op int
215 02be96c6 2021-02-09 op validate_against_ca(X509_STORE *ca, const uint8_t *chain, size_t len)
216 02be96c6 2021-02-09 op {
217 02be96c6 2021-02-09 op 	X509		*client;
218 02be96c6 2021-02-09 op 	BIO		*m;
219 02be96c6 2021-02-09 op 	X509_STORE_CTX	*ctx = NULL;
220 02be96c6 2021-02-09 op 	int		 ret = 0;
221 02be96c6 2021-02-09 op 
222 02be96c6 2021-02-09 op 	if ((m = BIO_new_mem_buf(chain, len)) == NULL)
223 02be96c6 2021-02-09 op 		return 0;
224 02be96c6 2021-02-09 op 
225 02be96c6 2021-02-09 op 	if ((client = PEM_read_bio_X509(m, NULL, NULL, NULL)) == NULL)
226 02be96c6 2021-02-09 op 		goto end;
227 02be96c6 2021-02-09 op 
228 02be96c6 2021-02-09 op 	if ((ctx = X509_STORE_CTX_new()) == NULL)
229 02be96c6 2021-02-09 op 		goto end;
230 02be96c6 2021-02-09 op 
231 02be96c6 2021-02-09 op 	if (!X509_STORE_CTX_init(ctx, ca, client, NULL))
232 02be96c6 2021-02-09 op 		goto end;
233 02be96c6 2021-02-09 op 
234 02be96c6 2021-02-09 op 	ret = X509_verify_cert(ctx);
235 02be96c6 2021-02-09 op 
236 02be96c6 2021-02-09 op end:
237 02be96c6 2021-02-09 op 	BIO_free(m);
238 02be96c6 2021-02-09 op 	if (client != NULL)
239 02be96c6 2021-02-09 op 		X509_free(client);
240 02be96c6 2021-02-09 op 	if (ctx != NULL)
241 02be96c6 2021-02-09 op 		X509_STORE_CTX_free(ctx);
242 02be96c6 2021-02-09 op 	return ret;
243 bc99d868 2021-03-19 op }
244 bc99d868 2021-03-19 op 
245 bc99d868 2021-03-19 op void
246 bc99d868 2021-03-19 op dispatch_imsg(struct imsgbuf *ibuf, imsg_handlerfn **handlers, size_t size)
247 bc99d868 2021-03-19 op {
248 bc99d868 2021-03-19 op 	struct imsg	imsg;
249 bc99d868 2021-03-19 op 	size_t		datalen, i;
250 bc99d868 2021-03-19 op 	ssize_t		n;
251 bc99d868 2021-03-19 op 
252 bc99d868 2021-03-19 op 	if ((n = imsg_read(ibuf)) == -1) {
253 bc99d868 2021-03-19 op 		if (errno == EAGAIN || errno == EWOULDBLOCK)
254 bc99d868 2021-03-19 op 			return;
255 bc99d868 2021-03-19 op 		_exit(1);
256 bc99d868 2021-03-19 op 	}
257 bc99d868 2021-03-19 op 
258 bc99d868 2021-03-19 op 	if (n == 0)
259 bc99d868 2021-03-19 op 		_exit(1);
260 bc99d868 2021-03-19 op 
261 bc99d868 2021-03-19 op 	for (;;) {
262 bc99d868 2021-03-19 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
263 bc99d868 2021-03-19 op 			_exit(1);
264 bc99d868 2021-03-19 op 		if (n == 0)
265 bc99d868 2021-03-19 op 			return;
266 bc99d868 2021-03-19 op 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
267 bc99d868 2021-03-19 op 		i = imsg.hdr.type;
268 bc99d868 2021-03-19 op 		if (i > (size / sizeof(imsg_handlerfn*)) || handlers[i] == NULL)
269 bc99d868 2021-03-19 op 			abort();
270 bc99d868 2021-03-19 op 		handlers[i](ibuf, &imsg, datalen);
271 bc99d868 2021-03-19 op 		imsg_free(&imsg);
272 bc99d868 2021-03-19 op 	}
273 02be96c6 2021-02-09 op }