Blame


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 44ee1bac 2021-01-27 op int
28 44ee1bac 2021-01-27 op starts_with(const char *str, const char *prefix)
29 44ee1bac 2021-01-27 op {
30 44ee1bac 2021-01-27 op size_t i;
31 44ee1bac 2021-01-27 op
32 44ee1bac 2021-01-27 op if (prefix == NULL)
33 44ee1bac 2021-01-27 op return 0;
34 44ee1bac 2021-01-27 op
35 44ee1bac 2021-01-27 op for (i = 0; prefix[i] != '\0'; ++i)
36 44ee1bac 2021-01-27 op if (str[i] != prefix[i])
37 44ee1bac 2021-01-27 op return 0;
38 44ee1bac 2021-01-27 op return 1;
39 44ee1bac 2021-01-27 op }
40 44ee1bac 2021-01-27 op
41 44ee1bac 2021-01-27 op int
42 44ee1bac 2021-01-27 op ends_with(const char *str, const char *sufx)
43 44ee1bac 2021-01-27 op {
44 44ee1bac 2021-01-27 op size_t i, j;
45 44ee1bac 2021-01-27 op
46 44ee1bac 2021-01-27 op i = strlen(str);
47 44ee1bac 2021-01-27 op j = strlen(sufx);
48 44ee1bac 2021-01-27 op
49 44ee1bac 2021-01-27 op if (j > i)
50 44ee1bac 2021-01-27 op return 0;
51 44ee1bac 2021-01-27 op
52 44ee1bac 2021-01-27 op i -= j;
53 44ee1bac 2021-01-27 op for (j = 0; str[i] != '\0'; i++, j++)
54 44ee1bac 2021-01-27 op if (str[i] != sufx[j])
55 44ee1bac 2021-01-27 op return 0;
56 44ee1bac 2021-01-27 op return 1;
57 44ee1bac 2021-01-27 op }
58 44ee1bac 2021-01-27 op
59 44ee1bac 2021-01-27 op ssize_t
60 44ee1bac 2021-01-27 op filesize(int fd)
61 44ee1bac 2021-01-27 op {
62 44ee1bac 2021-01-27 op ssize_t len;
63 44ee1bac 2021-01-27 op
64 44ee1bac 2021-01-27 op if ((len = lseek(fd, 0, SEEK_END)) == -1)
65 44ee1bac 2021-01-27 op return -1;
66 44ee1bac 2021-01-27 op if (lseek(fd, 0, SEEK_SET) == -1)
67 44ee1bac 2021-01-27 op return -1;
68 44ee1bac 2021-01-27 op return len;
69 44ee1bac 2021-01-27 op }
70 bcf5d929 2021-02-01 op
71 bcf5d929 2021-02-01 op char *
72 bcf5d929 2021-02-01 op absolutify_path(const char *path)
73 bcf5d929 2021-02-01 op {
74 bcf5d929 2021-02-01 op char *wd, *r;
75 bcf5d929 2021-02-01 op
76 bcf5d929 2021-02-01 op if (*path == '/') {
77 bcf5d929 2021-02-01 op if ((r = strdup(path)) == NULL)
78 bcf5d929 2021-02-01 op err(1, "strdup");
79 bcf5d929 2021-02-01 op return r;
80 bcf5d929 2021-02-01 op }
81 bcf5d929 2021-02-01 op
82 bcf5d929 2021-02-01 op wd = getcwd(NULL, 0);
83 bcf5d929 2021-02-01 op if (asprintf(&r, "%s/%s", wd, path) == -1)
84 bcf5d929 2021-02-01 op err(1, "asprintf");
85 bcf5d929 2021-02-01 op free(wd);
86 bcf5d929 2021-02-01 op return r;
87 bcf5d929 2021-02-01 op }
88 ca21e100 2021-02-04 op
89 ca21e100 2021-02-04 op char *
90 ca21e100 2021-02-04 op xstrdup(const char *s)
91 ca21e100 2021-02-04 op {
92 ca21e100 2021-02-04 op char *d;
93 ca21e100 2021-02-04 op
94 ca21e100 2021-02-04 op if ((d = strdup(s)) == NULL)
95 ca21e100 2021-02-04 op err(1, "strdup");
96 b8e64ccd 2021-03-31 op return d;
97 b8e64ccd 2021-03-31 op }
98 b8e64ccd 2021-03-31 op
99 b8e64ccd 2021-03-31 op void *
100 b8e64ccd 2021-03-31 op xcalloc(size_t nmemb, size_t size)
101 b8e64ccd 2021-03-31 op {
102 b8e64ccd 2021-03-31 op void *d;
103 b8e64ccd 2021-03-31 op
104 b8e64ccd 2021-03-31 op if ((d = calloc(nmemb, size)) == NULL)
105 b8e64ccd 2021-03-31 op err(1, "calloc");
106 ca21e100 2021-02-04 op return d;
107 ca21e100 2021-02-04 op }
108 3abf91b0 2021-02-07 op
109 3abf91b0 2021-02-07 op void
110 52418c8d 2021-02-12 op gen_certificate(const char *hostname, const char *certpath, const char *keypath)
111 3abf91b0 2021-02-07 op {
112 4283d65f 2021-02-10 op BIGNUM *e;
113 3abf91b0 2021-02-07 op EVP_PKEY *pkey;
114 3abf91b0 2021-02-07 op RSA *rsa;
115 3abf91b0 2021-02-07 op X509 *x509;
116 3abf91b0 2021-02-07 op X509_NAME *name;
117 3abf91b0 2021-02-07 op FILE *f;
118 52418c8d 2021-02-12 op const unsigned char *host = (const unsigned char*)hostname;
119 3abf91b0 2021-02-07 op
120 3abf91b0 2021-02-07 op log_notice(NULL,
121 3abf91b0 2021-02-07 op "generating new certificate for %s (it could take a while)",
122 3abf91b0 2021-02-07 op host);
123 3abf91b0 2021-02-07 op
124 3abf91b0 2021-02-07 op if ((pkey = EVP_PKEY_new()) == NULL)
125 4842c72d 2021-10-18 op fatal("couldn't create a new private key");
126 3abf91b0 2021-02-07 op
127 3abf91b0 2021-02-07 op if ((rsa = RSA_new()) == NULL)
128 4283d65f 2021-02-10 op fatal("couldn't generate rsa");
129 3abf91b0 2021-02-07 op
130 4283d65f 2021-02-10 op if ((e = BN_new()) == NULL)
131 4283d65f 2021-02-10 op fatal("couldn't allocate a bignum");
132 4283d65f 2021-02-10 op
133 1eb3631d 2021-05-12 op BN_set_word(e, RSA_F4);
134 4283d65f 2021-02-10 op if (!RSA_generate_key_ex(rsa, 4096, e, NULL))
135 3abf91b0 2021-02-07 op fatal("couldn't generate a rsa key");
136 3abf91b0 2021-02-07 op
137 3abf91b0 2021-02-07 op if (!EVP_PKEY_assign_RSA(pkey, rsa))
138 3abf91b0 2021-02-07 op fatal("couldn't assign the key");
139 3abf91b0 2021-02-07 op
140 3abf91b0 2021-02-07 op if ((x509 = X509_new()) == NULL)
141 3abf91b0 2021-02-07 op fatal("couldn't generate the X509 certificate");
142 3abf91b0 2021-02-07 op
143 1eb3631d 2021-05-12 op ASN1_INTEGER_set(X509_get_serialNumber(x509), 0);
144 3abf91b0 2021-02-07 op X509_gmtime_adj(X509_get_notBefore(x509), 0);
145 3abf91b0 2021-02-07 op X509_gmtime_adj(X509_get_notAfter(x509), 315360000L); /* 10 years */
146 1eb3631d 2021-05-12 op X509_set_version(x509, 3);
147 3abf91b0 2021-02-07 op
148 3abf91b0 2021-02-07 op if (!X509_set_pubkey(x509, pkey))
149 3abf91b0 2021-02-07 op fatal("couldn't set the public key");
150 3abf91b0 2021-02-07 op
151 3abf91b0 2021-02-07 op name = X509_get_subject_name(x509);
152 3abf91b0 2021-02-07 op if (!X509_NAME_add_entry_by_txt(name, "CN", MBSTRING_ASC, host, -1, -1, 0))
153 3abf91b0 2021-02-07 op fatal("couldn't add CN to cert");
154 3abf91b0 2021-02-07 op X509_set_issuer_name(x509, name);
155 3abf91b0 2021-02-07 op
156 3abf91b0 2021-02-07 op if (!X509_sign(x509, pkey, EVP_sha256()))
157 4842c72d 2021-10-18 op fatal("couldn't sign the certificate");
158 3abf91b0 2021-02-07 op
159 3abf91b0 2021-02-07 op if ((f = fopen(keypath, "w")) == NULL)
160 3abf91b0 2021-02-07 op fatal("fopen(%s): %s", keypath, strerror(errno));
161 3abf91b0 2021-02-07 op if (!PEM_write_PrivateKey(f, pkey, NULL, NULL, 0, NULL, NULL))
162 3abf91b0 2021-02-07 op fatal("couldn't write private key");
163 3abf91b0 2021-02-07 op fclose(f);
164 3abf91b0 2021-02-07 op
165 3abf91b0 2021-02-07 op if ((f = fopen(certpath, "w")) == NULL)
166 3abf91b0 2021-02-07 op fatal("fopen(%s): %s", certpath, strerror(errno));
167 3abf91b0 2021-02-07 op if (!PEM_write_X509(f, x509))
168 3abf91b0 2021-02-07 op fatal("couldn't write cert");
169 3abf91b0 2021-02-07 op fclose(f);
170 3abf91b0 2021-02-07 op
171 4283d65f 2021-02-10 op BN_free(e);
172 3abf91b0 2021-02-07 op X509_free(x509);
173 3abf91b0 2021-02-07 op RSA_free(rsa);
174 3abf91b0 2021-02-07 op }
175 02be96c6 2021-02-09 op
176 02be96c6 2021-02-09 op X509_STORE *
177 02be96c6 2021-02-09 op load_ca(const char *path)
178 02be96c6 2021-02-09 op {
179 02be96c6 2021-02-09 op FILE *f = NULL;
180 02be96c6 2021-02-09 op X509 *x = NULL;
181 02be96c6 2021-02-09 op X509_STORE *store;
182 02be96c6 2021-02-09 op
183 02be96c6 2021-02-09 op if ((store = X509_STORE_new()) == NULL)
184 02be96c6 2021-02-09 op return NULL;
185 02be96c6 2021-02-09 op
186 02be96c6 2021-02-09 op if ((f = fopen(path, "r")) == NULL)
187 02be96c6 2021-02-09 op goto err;
188 02be96c6 2021-02-09 op
189 02be96c6 2021-02-09 op if ((x = PEM_read_X509(f, NULL, NULL, NULL)) == NULL)
190 02be96c6 2021-02-09 op goto err;
191 02be96c6 2021-02-09 op
192 02be96c6 2021-02-09 op if (X509_check_ca(x) == 0)
193 02be96c6 2021-02-09 op goto err;
194 02be96c6 2021-02-09 op
195 02be96c6 2021-02-09 op if (!X509_STORE_add_cert(store, x))
196 02be96c6 2021-02-09 op goto err;
197 02be96c6 2021-02-09 op
198 02be96c6 2021-02-09 op X509_free(x);
199 02be96c6 2021-02-09 op fclose(f);
200 02be96c6 2021-02-09 op return store;
201 02be96c6 2021-02-09 op
202 02be96c6 2021-02-09 op err:
203 02be96c6 2021-02-09 op X509_STORE_free(store);
204 02be96c6 2021-02-09 op if (x != NULL)
205 02be96c6 2021-02-09 op X509_free(x);
206 02be96c6 2021-02-09 op if (f != NULL)
207 02be96c6 2021-02-09 op fclose(f);
208 02be96c6 2021-02-09 op return NULL;
209 02be96c6 2021-02-09 op }
210 02be96c6 2021-02-09 op
211 02be96c6 2021-02-09 op int
212 02be96c6 2021-02-09 op validate_against_ca(X509_STORE *ca, const uint8_t *chain, size_t len)
213 02be96c6 2021-02-09 op {
214 02be96c6 2021-02-09 op X509 *client;
215 02be96c6 2021-02-09 op BIO *m;
216 02be96c6 2021-02-09 op X509_STORE_CTX *ctx = NULL;
217 02be96c6 2021-02-09 op int ret = 0;
218 02be96c6 2021-02-09 op
219 02be96c6 2021-02-09 op if ((m = BIO_new_mem_buf(chain, len)) == NULL)
220 02be96c6 2021-02-09 op return 0;
221 02be96c6 2021-02-09 op
222 02be96c6 2021-02-09 op if ((client = PEM_read_bio_X509(m, NULL, NULL, NULL)) == NULL)
223 02be96c6 2021-02-09 op goto end;
224 02be96c6 2021-02-09 op
225 02be96c6 2021-02-09 op if ((ctx = X509_STORE_CTX_new()) == NULL)
226 02be96c6 2021-02-09 op goto end;
227 02be96c6 2021-02-09 op
228 02be96c6 2021-02-09 op if (!X509_STORE_CTX_init(ctx, ca, client, NULL))
229 02be96c6 2021-02-09 op goto end;
230 02be96c6 2021-02-09 op
231 02be96c6 2021-02-09 op ret = X509_verify_cert(ctx);
232 02be96c6 2021-02-09 op
233 02be96c6 2021-02-09 op end:
234 02be96c6 2021-02-09 op BIO_free(m);
235 02be96c6 2021-02-09 op if (client != NULL)
236 02be96c6 2021-02-09 op X509_free(client);
237 02be96c6 2021-02-09 op if (ctx != NULL)
238 02be96c6 2021-02-09 op X509_STORE_CTX_free(ctx);
239 02be96c6 2021-02-09 op return ret;
240 bc99d868 2021-03-19 op }
241 bc99d868 2021-03-19 op
242 bc99d868 2021-03-19 op void
243 bc99d868 2021-03-19 op dispatch_imsg(struct imsgbuf *ibuf, imsg_handlerfn **handlers, size_t size)
244 bc99d868 2021-03-19 op {
245 bc99d868 2021-03-19 op struct imsg imsg;
246 bc99d868 2021-03-19 op size_t datalen, i;
247 bc99d868 2021-03-19 op ssize_t n;
248 bc99d868 2021-03-19 op
249 bc99d868 2021-03-19 op if ((n = imsg_read(ibuf)) == -1) {
250 bc99d868 2021-03-19 op if (errno == EAGAIN || errno == EWOULDBLOCK)
251 bc99d868 2021-03-19 op return;
252 bc99d868 2021-03-19 op _exit(1);
253 bc99d868 2021-03-19 op }
254 bc99d868 2021-03-19 op
255 bc99d868 2021-03-19 op if (n == 0)
256 bc99d868 2021-03-19 op _exit(1);
257 bc99d868 2021-03-19 op
258 bc99d868 2021-03-19 op for (;;) {
259 bc99d868 2021-03-19 op if ((n = imsg_get(ibuf, &imsg)) == -1)
260 bc99d868 2021-03-19 op _exit(1);
261 bc99d868 2021-03-19 op if (n == 0)
262 bc99d868 2021-03-19 op return;
263 bc99d868 2021-03-19 op datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
264 bc99d868 2021-03-19 op i = imsg.hdr.type;
265 bc99d868 2021-03-19 op if (i > (size / sizeof(imsg_handlerfn*)) || handlers[i] == NULL)
266 bc99d868 2021-03-19 op abort();
267 bc99d868 2021-03-19 op handlers[i](ibuf, &imsg, datalen);
268 bc99d868 2021-03-19 op imsg_free(&imsg);
269 bc99d868 2021-03-19 op }
270 02be96c6 2021-02-09 op }