1 c36d43ac 2021-04-25 op /*
  2 c36d43ac 2021-04-25 op  * Copyright (c) 2021 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 c36d43ac 2021-04-25 op #include "telescope.h"
 18 c36d43ac 2021-04-25 op 
 19 c36d43ac 2021-04-25 op #include <stdlib.h>
 20 c36d43ac 2021-04-25 op #include <string.h>
 21 c36d43ac 2021-04-25 op 
 22 c36d43ac 2021-04-25 op static void	*hash_alloc(size_t, void*);
 23 c36d43ac 2021-04-25 op static void	*hash_calloc(size_t, size_t, void*);
 24 c36d43ac 2021-04-25 op static void	 hash_free(void*, void*);
 25 c36d43ac 2021-04-25 op 
 26 c36d43ac 2021-04-25 op static void *
 27 c36d43ac 2021-04-25 op hash_alloc(size_t len, void *d)
 28 c36d43ac 2021-04-25 op {
 29 c36d43ac 2021-04-25 op 	if ((d = malloc(len)) == NULL)
 30 c36d43ac 2021-04-25 op 		abort();
 31 c36d43ac 2021-04-25 op 	return d;
 32 c36d43ac 2021-04-25 op }
 33 c36d43ac 2021-04-25 op 
 34 c36d43ac 2021-04-25 op static void *
 35 c36d43ac 2021-04-25 op hash_calloc(size_t nmemb, size_t size, void *d)
 36 c36d43ac 2021-04-25 op {
 37 c36d43ac 2021-04-25 op 	if ((d = calloc(nmemb, size)) == NULL)
 38 c36d43ac 2021-04-25 op 		abort();
 39 c36d43ac 2021-04-25 op 	return d;
 40 c36d43ac 2021-04-25 op }
 41 c36d43ac 2021-04-25 op 
 42 c36d43ac 2021-04-25 op static void
 43 c36d43ac 2021-04-25 op hash_free(void *ptr, void *d)
 44 c36d43ac 2021-04-25 op {
 45 c36d43ac 2021-04-25 op 	free(ptr);
 46 c36d43ac 2021-04-25 op }
 47 c36d43ac 2021-04-25 op 
 48 c36d43ac 2021-04-25 op void
 49 c36d43ac 2021-04-25 op tofu_init(struct ohash *h, unsigned int sz, ptrdiff_t ko)
 50 c36d43ac 2021-04-25 op {
 51 c36d43ac 2021-04-25 op 	struct ohash_info info = {
 52 c36d43ac 2021-04-25 op 		.key_offset = ko,
 53 c36d43ac 2021-04-25 op 		.calloc = hash_calloc,
 54 c36d43ac 2021-04-25 op 		.free = hash_free,
 55 c36d43ac 2021-04-25 op 		.alloc = hash_alloc,
 56 c36d43ac 2021-04-25 op 	};
 57 c36d43ac 2021-04-25 op 
 58 c36d43ac 2021-04-25 op 	ohash_init(h, sz, &info);
 59 c36d43ac 2021-04-25 op }
 60 c36d43ac 2021-04-25 op 
 61 c36d43ac 2021-04-25 op struct tofu_entry *
 62 c36d43ac 2021-04-25 op tofu_lookup(struct ohash *h, const char *domain, const char *port)
 63 c36d43ac 2021-04-25 op {
 64 c36d43ac 2021-04-25 op 	char		buf[GEMINI_URL_LEN];
 65 c36d43ac 2021-04-25 op 	unsigned int	slot;
 66 c36d43ac 2021-04-25 op 
 67 c36d43ac 2021-04-25 op 	strlcpy(buf, domain, sizeof(buf));
 68 c36d43ac 2021-04-25 op 	if (port != NULL && *port != '\0' && strcmp(port, "1965")) {
 69 c36d43ac 2021-04-25 op 		strlcat(buf, ":", sizeof(buf));
 70 c36d43ac 2021-04-25 op 		strlcat(buf, port, sizeof(buf));
 71 c36d43ac 2021-04-25 op 	}
 72 c36d43ac 2021-04-25 op 
 73 c36d43ac 2021-04-25 op 	slot = ohash_qlookup(h, buf);
 74 c36d43ac 2021-04-25 op 	return ohash_find(h, slot);
 75 c36d43ac 2021-04-25 op }
 76 c36d43ac 2021-04-25 op 
 77 c36d43ac 2021-04-25 op void
 78 c36d43ac 2021-04-25 op tofu_add(struct ohash *h, struct tofu_entry *e)
 79 c36d43ac 2021-04-25 op {
 80 c36d43ac 2021-04-25 op 	unsigned int	slot;
 81 c36d43ac 2021-04-25 op 
 82 c36d43ac 2021-04-25 op 	slot = ohash_qlookup(h, e->domain);
 83 c36d43ac 2021-04-25 op 	ohash_insert(h, slot, e);
 84 c36d43ac 2021-04-25 op }
 85 288fd238 2021-04-25 op 
 86 288fd238 2021-04-25 op void
 87 288fd238 2021-04-25 op tofu_update(struct ohash *h, struct tofu_entry *e)
 88 288fd238 2021-04-25 op {
 89 288fd238 2021-04-25 op 	struct tofu_entry *t;
 90 288fd238 2021-04-25 op 
 91 288fd238 2021-04-25 op 	if ((t = tofu_lookup(h, e->domain, NULL)) == NULL)
 92 288fd238 2021-04-25 op 		tofu_add(h, e);
 93 288fd238 2021-04-25 op 	else {
 94 288fd238 2021-04-25 op 		strlcpy(t->hash, e->hash, sizeof(t->hash));
 95 288fd238 2021-04-25 op 		t->verified = e->verified;
 96 288fd238 2021-04-25 op 		free(e);
 97 288fd238 2021-04-25 op 	}
 98 288fd238 2021-04-25 op }
 99 a2fd3805 2021-07-06 op 
100 a2fd3805 2021-07-06 op void
101 a2fd3805 2021-07-06 op tofu_temp_trust(struct ohash *h, const char *host, const char *port,
102 a2fd3805 2021-07-06 op     const char *hash)
103 a2fd3805 2021-07-06 op {
104 a2fd3805 2021-07-06 op 	struct tofu_entry *e;
105 a2fd3805 2021-07-06 op 
106 a2fd3805 2021-07-06 op 	if ((e = calloc(1, sizeof(*e))) == NULL)
107 a2fd3805 2021-07-06 op                 abort();
108 a2fd3805 2021-07-06 op 
109 a2fd3805 2021-07-06 op         strlcpy(e->domain, host, sizeof(e->domain));
110 a2fd3805 2021-07-06 op 	if (*port != '\0' && strcmp(port, "1965")) {
111 a2fd3805 2021-07-06 op 		strlcat(e->domain, ":", sizeof(e->domain));
112 a2fd3805 2021-07-06 op 		strlcat(e->domain, port, sizeof(e->domain));
113 a2fd3805 2021-07-06 op 	}
114 a2fd3805 2021-07-06 op 	strlcpy(e->hash, hash, sizeof(e->hash));
115 a2fd3805 2021-07-06 op 	e->verified = -1;
116 a2fd3805 2021-07-06 op 
117 a2fd3805 2021-07-06 op 	tofu_update(h, e);
118 a2fd3805 2021-07-06 op }