1 c36d43ac 2021-04-25 op /*
  2 12472189 2024-02-05 op  * Copyright (c) 2021, 2022, 2024 Omar Polo <op@omarpolo.com>
  3 c36d43ac 2021-04-25 op  *
  4 c36d43ac 2021-04-25 op  * Permission to use, copy, modify, and distribute this software for any
  5 c36d43ac 2021-04-25 op  * purpose with or without fee is hereby granted, provided that the above
  6 c36d43ac 2021-04-25 op  * copyright notice and this permission notice appear in all copies.
  7 c36d43ac 2021-04-25 op  *
  8 c36d43ac 2021-04-25 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 c36d43ac 2021-04-25 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 c36d43ac 2021-04-25 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 c36d43ac 2021-04-25 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 c36d43ac 2021-04-25 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 c36d43ac 2021-04-25 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 c36d43ac 2021-04-25 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 c36d43ac 2021-04-25 op  */
 16 c36d43ac 2021-04-25 op 
 17 036c104e 2022-01-11 op #include "compat.h"
 18 c36d43ac 2021-04-25 op 
 19 4fa88f57 2022-04-24 op #include <limits.h>
 20 d163c210 2024-02-06 op #include <stdio.h>
 21 c36d43ac 2021-04-25 op #include <stdlib.h>
 22 c36d43ac 2021-04-25 op #include <string.h>
 23 3078e1bc 2022-04-24 op #include <unistd.h>
 24 c36d43ac 2021-04-25 op 
 25 4fa88f57 2022-04-24 op #include "fs.h"
 26 d163c210 2024-02-06 op #include "tofu.h"
 27 9d65b1d9 2022-01-11 op #include "utils.h"
 28 3d89457c 2024-06-18 thomas.ad #include "xwrapper.h"
 29 c36d43ac 2021-04-25 op 
 30 c36d43ac 2021-04-25 op void
 31 c36d43ac 2021-04-25 op tofu_init(struct ohash *h, unsigned int sz, ptrdiff_t ko)
 32 c36d43ac 2021-04-25 op {
 33 c36d43ac 2021-04-25 op 	struct ohash_info info = {
 34 c36d43ac 2021-04-25 op 		.key_offset = ko,
 35 c36d43ac 2021-04-25 op 		.calloc = hash_calloc,
 36 c36d43ac 2021-04-25 op 		.free = hash_free,
 37 c36d43ac 2021-04-25 op 		.alloc = hash_alloc,
 38 c36d43ac 2021-04-25 op 	};
 39 c36d43ac 2021-04-25 op 
 40 c36d43ac 2021-04-25 op 	ohash_init(h, sz, &info);
 41 c36d43ac 2021-04-25 op }
 42 c36d43ac 2021-04-25 op 
 43 c36d43ac 2021-04-25 op struct tofu_entry *
 44 c36d43ac 2021-04-25 op tofu_lookup(struct ohash *h, const char *domain, const char *port)
 45 c36d43ac 2021-04-25 op {
 46 d163c210 2024-02-06 op 	char		buf[TOFU_URL_MAX_LEN];
 47 c36d43ac 2021-04-25 op 	unsigned int	slot;
 48 c36d43ac 2021-04-25 op 
 49 c36d43ac 2021-04-25 op 	strlcpy(buf, domain, sizeof(buf));
 50 c36d43ac 2021-04-25 op 	if (port != NULL && *port != '\0' && strcmp(port, "1965")) {
 51 c36d43ac 2021-04-25 op 		strlcat(buf, ":", sizeof(buf));
 52 c36d43ac 2021-04-25 op 		strlcat(buf, port, sizeof(buf));
 53 c36d43ac 2021-04-25 op 	}
 54 c36d43ac 2021-04-25 op 
 55 c36d43ac 2021-04-25 op 	slot = ohash_qlookup(h, buf);
 56 c36d43ac 2021-04-25 op 	return ohash_find(h, slot);
 57 c36d43ac 2021-04-25 op }
 58 c36d43ac 2021-04-25 op 
 59 c36d43ac 2021-04-25 op void
 60 c36d43ac 2021-04-25 op tofu_add(struct ohash *h, struct tofu_entry *e)
 61 c36d43ac 2021-04-25 op {
 62 c36d43ac 2021-04-25 op 	unsigned int	slot;
 63 c36d43ac 2021-04-25 op 
 64 c36d43ac 2021-04-25 op 	slot = ohash_qlookup(h, e->domain);
 65 c36d43ac 2021-04-25 op 	ohash_insert(h, slot, e);
 66 c36d43ac 2021-04-25 op }
 67 288fd238 2021-04-25 op 
 68 4fa88f57 2022-04-24 op int
 69 4fa88f57 2022-04-24 op tofu_save(struct ohash *h, struct tofu_entry *e)
 70 4fa88f57 2022-04-24 op {
 71 4fa88f57 2022-04-24 op 	FILE *fp;
 72 4fa88f57 2022-04-24 op 
 73 4fa88f57 2022-04-24 op 	tofu_add(h, e);
 74 4fa88f57 2022-04-24 op 
 75 4fa88f57 2022-04-24 op 	if ((fp = fopen(known_hosts_file, "a")) == NULL)
 76 4fa88f57 2022-04-24 op 		return -1;
 77 4fa88f57 2022-04-24 op 	fprintf(fp, "%s %s %d\n", e->domain, e->hash, e->verified);
 78 4fa88f57 2022-04-24 op 	fclose(fp);
 79 4fa88f57 2022-04-24 op 	return 0;
 80 4fa88f57 2022-04-24 op }
 81 4fa88f57 2022-04-24 op 
 82 288fd238 2021-04-25 op void
 83 288fd238 2021-04-25 op tofu_update(struct ohash *h, struct tofu_entry *e)
 84 288fd238 2021-04-25 op {
 85 288fd238 2021-04-25 op 	struct tofu_entry *t;
 86 288fd238 2021-04-25 op 
 87 288fd238 2021-04-25 op 	if ((t = tofu_lookup(h, e->domain, NULL)) == NULL)
 88 288fd238 2021-04-25 op 		tofu_add(h, e);
 89 288fd238 2021-04-25 op 	else {
 90 288fd238 2021-04-25 op 		strlcpy(t->hash, e->hash, sizeof(t->hash));
 91 288fd238 2021-04-25 op 		t->verified = e->verified;
 92 288fd238 2021-04-25 op 		free(e);
 93 288fd238 2021-04-25 op 	}
 94 288fd238 2021-04-25 op }
 95 a2fd3805 2021-07-06 op 
 96 3078e1bc 2022-04-24 op int
 97 3078e1bc 2022-04-24 op tofu_update_persist(struct ohash *h, struct tofu_entry *e)
 98 3078e1bc 2022-04-24 op {
 99 3078e1bc 2022-04-24 op 	FILE	*tmp, *fp;
100 3078e1bc 2022-04-24 op 	char	 sfn[PATH_MAX], *line = NULL;
101 3078e1bc 2022-04-24 op 	size_t	 l, linesize = 0;
102 3078e1bc 2022-04-24 op 	ssize_t	 linelen;
103 3078e1bc 2022-04-24 op 	int	 fd, err;
104 3078e1bc 2022-04-24 op 
105 3078e1bc 2022-04-24 op 	tofu_update(h, e);
106 3078e1bc 2022-04-24 op 
107 3078e1bc 2022-04-24 op 	strlcpy(sfn, known_hosts_tmp, sizeof(sfn));
108 3078e1bc 2022-04-24 op 	if ((fd = mkstemp(sfn)) == -1 ||
109 3078e1bc 2022-04-24 op 	    (tmp = fdopen(fd, "w")) == NULL) {
110 3078e1bc 2022-04-24 op 		if (fd != -1) {
111 3078e1bc 2022-04-24 op 			unlink(sfn);
112 3078e1bc 2022-04-24 op 			close(fd);
113 3078e1bc 2022-04-24 op 		}
114 3078e1bc 2022-04-24 op 		return -1;
115 3078e1bc 2022-04-24 op 	}
116 3078e1bc 2022-04-24 op 
117 3078e1bc 2022-04-24 op 	if ((fp = fopen(known_hosts_file, "r")) == NULL) {
118 3078e1bc 2022-04-24 op 		unlink(sfn);
119 3078e1bc 2022-04-24 op 		fclose(tmp);
120 3078e1bc 2022-04-24 op 		return -1;
121 3078e1bc 2022-04-24 op 	}
122 3078e1bc 2022-04-24 op 
123 3078e1bc 2022-04-24 op 	l = strlen(e->domain);
124 3078e1bc 2022-04-24 op 	while ((linelen = getline(&line, &linesize, fp)) != -1) {
125 3078e1bc 2022-04-24 op 		if (!strncmp(line, e->domain, l))
126 3078e1bc 2022-04-24 op 			continue;
127 3078e1bc 2022-04-24 op 		if (linesize > 0 && line[linesize-1] == '\n')
128 3078e1bc 2022-04-24 op 			line[linesize-1] = '\0';
129 3078e1bc 2022-04-24 op 		fprintf(tmp, "%s\n", line);
130 3078e1bc 2022-04-24 op 	}
131 3078e1bc 2022-04-24 op 	fprintf(tmp, "%s %s %d\n", e->domain, e->hash, e->verified);
132 3078e1bc 2022-04-24 op 
133 3078e1bc 2022-04-24 op 	free(line);
134 3078e1bc 2022-04-24 op 	err = ferror(tmp);
135 3078e1bc 2022-04-24 op 	fclose(tmp);
136 3078e1bc 2022-04-24 op 	fclose(fp);
137 3078e1bc 2022-04-24 op 
138 3078e1bc 2022-04-24 op 	if (err) {
139 3078e1bc 2022-04-24 op 		unlink(sfn);
140 3078e1bc 2022-04-24 op 		return -1;
141 3078e1bc 2022-04-24 op 	}
142 3078e1bc 2022-04-24 op 
143 12472189 2024-02-05 op 	if (rename(sfn, known_hosts_file) == -1)
144 3078e1bc 2022-04-24 op 		return -1;
145 3078e1bc 2022-04-24 op 	return 0;
146 3078e1bc 2022-04-24 op }
147 3078e1bc 2022-04-24 op 
148 a2fd3805 2021-07-06 op void
149 a2fd3805 2021-07-06 op tofu_temp_trust(struct ohash *h, const char *host, const char *port,
150 a2fd3805 2021-07-06 op     const char *hash)
151 a2fd3805 2021-07-06 op {
152 a2fd3805 2021-07-06 op 	struct tofu_entry *e;
153 a2fd3805 2021-07-06 op 
154 3d89457c 2024-06-18 thomas.ad 	e = xcalloc(1, sizeof(*e));
155 a2fd3805 2021-07-06 op 
156 95a8c791 2021-08-26 op 	strlcpy(e->domain, host, sizeof(e->domain));
157 a2fd3805 2021-07-06 op 	if (*port != '\0' && strcmp(port, "1965")) {
158 a2fd3805 2021-07-06 op 		strlcat(e->domain, ":", sizeof(e->domain));
159 a2fd3805 2021-07-06 op 		strlcat(e->domain, port, sizeof(e->domain));
160 a2fd3805 2021-07-06 op 	}
161 a2fd3805 2021-07-06 op 	strlcpy(e->hash, hash, sizeof(e->hash));
162 a2fd3805 2021-07-06 op 	e->verified = -1;
163 a2fd3805 2021-07-06 op 
164 a2fd3805 2021-07-06 op 	tofu_update(h, e);
165 a2fd3805 2021-07-06 op }