2 * Copyright (c) 2021, 2022 Omar Polo <op@omarpolo.com>
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.
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.
25 #include "telescope.h"
29 tofu_init(struct ohash *h, unsigned int sz, ptrdiff_t ko)
31 struct ohash_info info = {
33 .calloc = hash_calloc,
38 ohash_init(h, sz, &info);
42 tofu_lookup(struct ohash *h, const char *domain, const char *port)
44 char buf[GEMINI_URL_LEN];
47 strlcpy(buf, domain, sizeof(buf));
48 if (port != NULL && *port != '\0' && strcmp(port, "1965")) {
49 strlcat(buf, ":", sizeof(buf));
50 strlcat(buf, port, sizeof(buf));
53 slot = ohash_qlookup(h, buf);
54 return ohash_find(h, slot);
58 tofu_add(struct ohash *h, struct tofu_entry *e)
62 slot = ohash_qlookup(h, e->domain);
63 ohash_insert(h, slot, e);
67 tofu_save(struct ohash *h, struct tofu_entry *e)
73 if ((fp = fopen(known_hosts_file, "a")) == NULL)
75 fprintf(fp, "%s %s %d\n", e->domain, e->hash, e->verified);
81 tofu_update(struct ohash *h, struct tofu_entry *e)
85 if ((t = tofu_lookup(h, e->domain, NULL)) == NULL)
88 strlcpy(t->hash, e->hash, sizeof(t->hash));
89 t->verified = e->verified;
95 tofu_update_persist(struct ohash *h, struct tofu_entry *e)
98 char sfn[PATH_MAX], *line = NULL;
99 size_t l, linesize = 0;
105 strlcpy(sfn, known_hosts_tmp, sizeof(sfn));
106 if ((fd = mkstemp(sfn)) == -1 ||
107 (tmp = fdopen(fd, "w")) == NULL) {
115 if ((fp = fopen(known_hosts_file, "r")) == NULL) {
121 l = strlen(e->domain);
122 while ((linelen = getline(&line, &linesize, fp)) != -1) {
123 if (!strncmp(line, e->domain, l))
125 if (linesize > 0 && line[linesize-1] == '\n')
126 line[linesize-1] = '\0';
127 fprintf(tmp, "%s\n", line);
129 fprintf(tmp, "%s %s %d\n", e->domain, e->hash, e->verified);
141 if (rename(sfn, known_hosts_file))
147 tofu_temp_trust(struct ohash *h, const char *host, const char *port,
150 struct tofu_entry *e;
152 if ((e = calloc(1, sizeof(*e))) == NULL)
155 strlcpy(e->domain, host, sizeof(e->domain));
156 if (*port != '\0' && strcmp(port, "1965")) {
157 strlcat(e->domain, ":", sizeof(e->domain));
158 strlcat(e->domain, port, sizeof(e->domain));
160 strlcpy(e->hash, hash, sizeof(e->hash));