url.c
No OneTemporary
Actions

Size

17 KB

Referenced Files

None

Subscribers

None

url.c
View Options

	#include <compat.h> // IWYU pragma: keep

	#include <stdlib.h>
	#include <stdio.h>
	#include <string.h>
	#include <ctype.h>

	#include "ant.h"
	#include "runtime.h"
	#include "modules/url.h"

	typedef struct {
	char *protocol;
	char *username;
	char *password;
	char *hostname;
	char *port;
	char *pathname;
	char *search;
	char *hash;
	} parsed_url_t;

	static void free_parsed_url(parsed_url_t *p) {
	if (p->protocol) free(p->protocol);
	if (p->username) free(p->username);
	if (p->password) free(p->password);
	if (p->hostname) free(p->hostname);
	if (p->port) free(p->port);
	if (p->pathname) free(p->pathname);
	if (p->search) free(p->search);
	if (p->hash) free(p->hash);
	}

	static char url_encode_component(const char str) {
	if (!str) return strdup("");
	size_t len = strlen(str);
	char out = malloc(len 3 + 1);
	if (!out) return NULL;
	size_t j = 0;
	for (size_t i = 0; i < len; i++) {
	unsigned char c = (unsigned char)str[i];
	if (isalnum(c) \|\| c == '-' \|\| c == '_' \|\| c == '.' \|\| c == '~') {
	out[j++] = c;
	} else {
	snprintf(out + j, 4, "%%%02X", c);
	j += 3;
	}
	}
	out[j] = '\0';
	return out;
	}

	static char url_decode_component(const char str) {
	if (!str) return strdup("");
	size_t len = strlen(str);
	char *out = malloc(len + 1);
	if (!out) return NULL;
	size_t j = 0;
	for (size_t i = 0; i < len; i++) {
	if (str[i] == '%' && i + 2 < len && isxdigit(str[i+1]) && isxdigit(str[i+2])) {
	int hi = isdigit(str[i+1]) ? str[i+1] - '0' : tolower(str[i+1]) - 'a' + 10;
	int lo = isdigit(str[i+2]) ? str[i+2] - '0' : tolower(str[i+2]) - 'a' + 10;
	out[j++] = (char)((hi << 4) \| lo);
	i += 2;
	} else if (str[i] == '+') {
	out[j++] = ' ';
	} else {
	out[j++] = str[i];
	}
	}
	out[j] = '\0';
	return out;
	}

	static int parse_url(const char url_str, const char base_str, parsed_url_t *out) {
	memset(out, 0, sizeof(*out));

	const char *p = url_str;
	const char *end = url_str + strlen(url_str);

	if (base_str && *p == '/') {
	parsed_url_t base = {0};
	if (parse_url(base_str, NULL, &base) < 0) return -1;
	out->protocol = base.protocol ? strdup(base.protocol) : NULL;
	out->username = base.username ? strdup(base.username) : strdup("");
	out->password = base.password ? strdup(base.password) : strdup("");
	out->hostname = base.hostname ? strdup(base.hostname) : NULL;
	out->port = base.port ? strdup(base.port) : strdup("");

	char *path_copy = strdup(p);
	char *q = strchr(path_copy, '?');
	char *h = strchr(path_copy, '#');

	if (h) {
	out->hash = strdup(h);
	*h = '\0';
	} else {
	out->hash = strdup("");
	}

	if (q) {
	out->search = strdup(q);
	*q = '\0';
	} else {
	out->search = strdup("");
	}

	out->pathname = strdup(path_copy);
	free(path_copy);

	free(base.protocol); free(base.username); free(base.password);
	free(base.hostname); free(base.port); free(base.pathname);
	free(base.search); free(base.hash);
	return 0;
	}

	const char *scheme_end = strstr(p, "://");
	if (!scheme_end) return -1;

	size_t scheme_len = scheme_end - p;
	out->protocol = malloc(scheme_len + 2);
	memcpy(out->protocol, p, scheme_len);
	out->protocol[scheme_len] = ':';
	out->protocol[scheme_len + 1] = '\0';
	p = scheme_end + 3;

	out->username = strdup("");
	out->password = strdup("");

	const char *at = strchr(p, '@');
	const char *slash = strchr(p, '/');
	if (at && (!slash \|\| at < slash)) {
	const char *colon = strchr(p, ':');
	if (colon && colon < at) {
	free(out->username);
	out->username = strndup(p, colon - p);
	free(out->password);
	out->password = strndup(colon + 1, at - colon - 1);
	} else {
	free(out->username);
	out->username = strndup(p, at - p);
	}
	p = at + 1;
	}

	slash = strchr(p, '/');
	const char *qmark = strchr(p, '?');
	const char *hash = strchr(p, '#');

	const char *host_end = slash ? slash : (qmark ? qmark : (hash ? hash : end));
	char *host_part = strndup(p, host_end - p);

	char *port_sep = strrchr(host_part, ':');
	if (port_sep && strchr(port_sep, ']') == NULL) {
	*port_sep = '\0';
	out->port = strdup(port_sep + 1);
	out->hostname = strdup(host_part);
	} else {
	out->hostname = strdup(host_part);
	out->port = strdup("");
	}
	free(host_part);
	p = host_end;

	if (slash) {
	const char *path_end = qmark ? qmark : (hash ? hash : end);
	out->pathname = strndup(slash, path_end - slash);
	p = path_end;
	} else {
	out->pathname = strdup("/");
	}

	if (qmark && (!hash \|\| qmark < hash)) {
	const char *search_end = hash ? hash : end;
	out->search = strndup(qmark, search_end - qmark);
	p = search_end;
	} else {
	out->search = strdup("");
	}

	if (hash) {
	out->hash = strdup(hash);
	} else {
	out->hash = strdup("");
	}

	return 0;
	}

	static char *build_href(
	const char protocol, const char username, const char *password,
	const char hostname, const char port, const char *pathname,
	const char search, const char hash
	) {
	size_t len = strlen(protocol) + 2 + strlen(hostname) + strlen(pathname) + strlen(search) + strlen(hash) + 32;
	if (username && *username) len += strlen(username) + strlen(password) + 2;
	if (port && *port) len += strlen(port) + 1;

	char *href = malloc(len);
	if (!href) return NULL;

	char *w = href;
	w += sprintf(w, "%s//", protocol);
	if (username && *username) {
	w += sprintf(w, "%s", username);
	if (password && *password) w += sprintf(w, ":%s", password);
	w += sprintf(w, "@");
	}
	w += sprintf(w, "%s", hostname);
	if (port && *port) w += sprintf(w, ":%s", port);
	w += sprintf(w, "%s%s%s", pathname, search, hash);
	return href;
	}

	static void update_url_href(struct js *js, jsval_t url_obj) {
	char *protocol = js_getstr(js, js_get(js, url_obj, "protocol"), NULL);
	char *username = js_getstr(js, js_get(js, url_obj, "username"), NULL);
	char *password = js_getstr(js, js_get(js, url_obj, "password"), NULL);
	char *hostname = js_getstr(js, js_get(js, url_obj, "hostname"), NULL);
	char *port = js_getstr(js, js_get(js, url_obj, "port"), NULL);
	char *pathname = js_getstr(js, js_get(js, url_obj, "pathname"), NULL);
	char *search = js_getstr(js, js_get(js, url_obj, "search"), NULL);
	char *hash = js_getstr(js, js_get(js, url_obj, "hash"), NULL);

	char *host;
	if (port && *port) {
	size_t hlen = strlen(hostname) + strlen(port) + 2;
	host = malloc(hlen);
	snprintf(host, hlen, "%s:%s", hostname, port);
	} else {
	host = strdup(hostname ? hostname : "");
	}
	js_set(js, url_obj, "host", js_mkstr(js, host, strlen(host)));

	char *origin;
	if (port && *port) {
	size_t olen = strlen(protocol) + strlen(hostname) + strlen(port) + 8;
	origin = malloc(olen);
	snprintf(origin, olen, "%s//%s:%s", protocol ? protocol : "", hostname ? hostname : "", port);
	} else {
	size_t olen = strlen(protocol) + strlen(hostname) + 8;
	origin = malloc(olen);
	snprintf(origin, olen, "%s//%s", protocol ? protocol : "", hostname ? hostname : "");
	}
	js_set(js, url_obj, "origin", js_mkstr(js, origin, strlen(origin)));

	char *href = build_href(
	protocol ? protocol : "", username ? username : "",
	password ? password : "", hostname ? hostname : "",
	port ? port : "", pathname ? pathname : "/",
	search ? search : "", hash ? hash : ""
	);

	js_set(js, url_obj, "href", js_mkstr(js, href, strlen(href)));

	free(host);
	free(origin);
	free(href);
	}

	static jsval_t url_toString(struct js js, jsval_t args, int nargs) {
	(void)args; (void)nargs;
	return js_get(js, js_getthis(js), "href");
	}

	static jsval_t js_URL(struct js js, jsval_t args, int nargs) {
	if (nargs < 1) return js_mkerr(js, "TypeError: URL requires at least 1 argument");

	char *url_str = js_getstr(js, args[0], NULL);
	char *base_str = (nargs > 1) ? js_getstr(js, args[1], NULL) : NULL;
	if (!url_str) return js_mkerr(js, "TypeError: Invalid URL");

	parsed_url_t parsed;
	if (parse_url(url_str, base_str, &parsed) < 0) {
	return js_mkerr(js, "TypeError: Invalid URL");
	}

	jsval_t url_obj = js_mkobj(js);

	js_set(js, url_obj, "protocol", js_mkstr(js, parsed.protocol, strlen(parsed.protocol)));
	js_set(js, url_obj, "username", js_mkstr(js, parsed.username, strlen(parsed.username)));
	js_set(js, url_obj, "password", js_mkstr(js, parsed.password, strlen(parsed.password)));
	js_set(js, url_obj, "hostname", js_mkstr(js, parsed.hostname, strlen(parsed.hostname)));
	js_set(js, url_obj, "port", js_mkstr(js, parsed.port, strlen(parsed.port)));
	js_set(js, url_obj, "pathname", js_mkstr(js, parsed.pathname, strlen(parsed.pathname)));
	js_set(js, url_obj, "search", js_mkstr(js, parsed.search, strlen(parsed.search)));
	js_set(js, url_obj, "hash", js_mkstr(js, parsed.hash, strlen(parsed.hash)));

	update_url_href(js, url_obj);

	jsval_t search_params;
	if (parsed.search && *parsed.search) {
	const char *qs = parsed.search[0] == '?' ? parsed.search + 1 : parsed.search;
	char code[1024];
	snprintf(code, sizeof(code), "new URLSearchParams('%s')", qs);
	search_params = js_eval(js, code, strlen(code));
	} else {
	search_params = js_eval(js, "new URLSearchParams()", 21);
	}
	js_set(js, search_params, "_url", url_obj);
	js_set(js, url_obj, "searchParams", search_params);

	js_set(js, url_obj, "toString", js_mkfun(url_toString));

	free_parsed_url(&parsed);
	return url_obj;
	}

	static jsval_t usp_get(struct js js, jsval_t args, int nargs) {
	if (nargs < 1) return js_mknull();
	jsval_t this_val = js_getthis(js);
	char *key = js_getstr(js, args[0], NULL);
	if (!key) return js_mknull();

	jsval_t entries = js_get(js, this_val, "_entries");
	if (js_type(entries) != JS_OBJ) return js_mknull();

	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	jsval_t k = js_get(js, entry, "0");
	char *ks = js_getstr(js, k, NULL);
	if (ks && strcmp(ks, key) == 0) {
	return js_get(js, entry, "1");
	}
	}
	return js_mknull();
	}

	static jsval_t usp_getAll(struct js js, jsval_t args, int nargs) {
	if (nargs < 1) return js_mkarr(js);
	jsval_t this_val = js_getthis(js);
	char *key = js_getstr(js, args[0], NULL);
	if (!key) return js_mkarr(js);

	jsval_t result = js_mkarr(js);
	jsval_t entries = js_get(js, this_val, "_entries");
	if (js_type(entries) != JS_OBJ) return result;

	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	jsval_t k = js_get(js, entry, "0");
	char *ks = js_getstr(js, k, NULL);
	if (ks && strcmp(ks, key) == 0) {
	js_arr_push(js, result, js_get(js, entry, "1"));
	}
	}
	return result;
	}

	static jsval_t usp_has(struct js js, jsval_t args, int nargs) {
	if (nargs < 1) return js_mkfalse();
	jsval_t this_val = js_getthis(js);
	char *key = js_getstr(js, args[0], NULL);
	if (!key) return js_mkfalse();

	jsval_t entries = js_get(js, this_val, "_entries");
	if (js_type(entries) != JS_OBJ) return js_mkfalse();

	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	jsval_t k = js_get(js, entry, "0");
	char *ks = js_getstr(js, k, NULL);
	if (ks && strcmp(ks, key) == 0) return js_mktrue();
	}
	return js_mkfalse();
	}

	static void usp_sync_url(struct js *js, jsval_t this_val) {
	jsval_t url_obj = js_get(js, this_val, "_url");
	if (js_type(url_obj) != JS_OBJ) return;

	jsval_t entries = js_get(js, this_val, "_entries");
	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	size_t buf_size = 1024;
	char *buf = malloc(buf_size);
	buf[0] = '?';
	size_t pos = 1;

	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	char *k = js_getstr(js, js_get(js, entry, "0"), NULL);
	char *v = js_getstr(js, js_get(js, entry, "1"), NULL);
	if (!k) continue;
	char *ek = url_encode_component(k);
	char *ev = url_encode_component(v ? v : "");
	size_t needed = strlen(ek) + strlen(ev) + 3;
	if (pos + needed >= buf_size) {
	buf_size = buf_size * 2 + needed;
	buf = realloc(buf, buf_size);
	}
	if (pos > 1) buf[pos++] = '&';
	pos += sprintf(buf + pos, "%s=%s", ek, ev);
	free(ek);
	free(ev);
	}
	buf[pos] = '\0';

	js_set(js, url_obj, "search", js_mkstr(js, buf, pos));
	update_url_href(js, url_obj);
	free(buf);
	}

	static jsval_t usp_set(struct js js, jsval_t args, int nargs) {
	if (nargs < 2) return js_mkundef();
	jsval_t this_val = js_getthis(js);
	char *key = js_getstr(js, args[0], NULL);
	if (!key) return js_mkundef();

	jsval_t entries = js_get(js, this_val, "_entries");
	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	jsval_t new_entries = js_mkarr(js);
	int found = 0;

	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	char *ks = js_getstr(js, js_get(js, entry, "0"), NULL);
	if (ks && strcmp(ks, key) == 0) {
	if (!found) {
	jsval_t new_entry = js_mkarr(js);
	js_arr_push(js, new_entry, args[0]);
	js_arr_push(js, new_entry, args[1]);
	js_arr_push(js, new_entries, new_entry);
	found = 1;
	}
	} else {
	js_arr_push(js, new_entries, entry);
	}
	}

	if (!found) {
	jsval_t new_entry = js_mkarr(js);
	js_arr_push(js, new_entry, args[0]);
	js_arr_push(js, new_entry, args[1]);
	js_arr_push(js, new_entries, new_entry);
	}

	js_set(js, this_val, "_entries", new_entries);
	usp_sync_url(js, this_val);
	return js_mkundef();
	}

	static jsval_t usp_append(struct js js, jsval_t args, int nargs) {
	if (nargs < 2) return js_mkundef();
	jsval_t this_val = js_getthis(js);

	jsval_t entries = js_get(js, this_val, "_entries");
	jsval_t entry = js_mkarr(js);
	js_arr_push(js, entry, args[0]);
	js_arr_push(js, entry, args[1]);
	js_arr_push(js, entries, entry);

	usp_sync_url(js, this_val);
	return js_mkundef();
	}

	static jsval_t usp_delete(struct js js, jsval_t args, int nargs) {
	if (nargs < 1) return js_mkundef();
	jsval_t this_val = js_getthis(js);
	char *key = js_getstr(js, args[0], NULL);
	if (!key) return js_mkundef();

	jsval_t entries = js_get(js, this_val, "_entries");
	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	jsval_t new_entries = js_mkarr(js);
	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	char *ks = js_getstr(js, js_get(js, entry, "0"), NULL);
	if (!ks \|\| strcmp(ks, key) != 0) {
	js_arr_push(js, new_entries, entry);
	}
	}

	js_set(js, this_val, "_entries", new_entries);
	usp_sync_url(js, this_val);
	js_set_needs_gc(js, true);
	return js_mkundef();
	}

	static jsval_t usp_toString(struct js js, jsval_t args, int nargs) {
	(void)args; (void)nargs;
	jsval_t this_val = js_getthis(js);
	jsval_t entries = js_get(js, this_val, "_entries");
	jsval_t len_val = js_get(js, entries, "length");
	int len = (js_type(len_val) == JS_NUM) ? (int)js_getnum(len_val) : 0;

	size_t buf_size = 1024;
	char *buf = malloc(buf_size);
	size_t pos = 0;

	for (int i = 0; i < len; i++) {
	char idx[16];
	snprintf(idx, sizeof(idx), "%d", i);
	jsval_t entry = js_get(js, entries, idx);
	char *k = js_getstr(js, js_get(js, entry, "0"), NULL);
	char *v = js_getstr(js, js_get(js, entry, "1"), NULL);
	if (!k) continue;
	char *ek = url_encode_component(k);
	char *ev = url_encode_component(v ? v : "");
	size_t needed = strlen(ek) + strlen(ev) + 3;
	if (pos + needed >= buf_size) {
	buf_size = buf_size * 2 + needed;
	buf = realloc(buf, buf_size);
	}
	if (pos > 0) buf[pos++] = '&';
	pos += sprintf(buf + pos, "%s=%s", ek, ev);
	free(ek);
	free(ev);
	}
	buf[pos] = '\0';
	jsval_t ret = js_mkstr(js, buf, pos);
	free(buf);
	return ret;
	}

	static jsval_t js_URLSearchParams(struct js js, jsval_t args, int nargs) {
	jsval_t usp = js_mkobj(js);
	jsval_t entries = js_mkarr(js);
	js_set(js, usp, "_entries", entries);

	if (nargs < 1 \|\| js_type(args[0]) != JS_STR) goto done_parse;
	char *init = js_getstr(js, args[0], NULL);
	if (!init) goto done_parse;

	const char *p = init;
	if (*p == '?') p++;

	parse_pair:
	if (!*p) goto done_parse;
	const char *amp = strchr(p, '&');
	size_t plen = amp ? (size_t)(amp - p) : strlen(p);
	char *pair = strndup(p, plen);
	char *eq = strchr(pair, '=');
	char key = pair, val = eq ? (eq[0] = '\0', eq + 1) : "";
	char *dk = url_decode_component(key);
	char *dv = url_decode_component(val);
	jsval_t entry = js_mkarr(js);
	js_arr_push(js, entry, js_mkstr(js, dk, strlen(dk)));
	js_arr_push(js, entry, js_mkstr(js, dv, strlen(dv)));
	js_arr_push(js, entries, entry);
	free(pair); free(dk); free(dv);
	if (!amp) goto done_parse;
	p = amp + 1;
	goto parse_pair;

	done_parse:

	js_set(js, usp, "get", js_mkfun(usp_get));
	js_set(js, usp, "getAll", js_mkfun(usp_getAll));
	js_set(js, usp, "has", js_mkfun(usp_has));
	js_set(js, usp, "set", js_mkfun(usp_set));
	js_set(js, usp, "append", js_mkfun(usp_append));
	js_set(js, usp, "delete", js_mkfun(usp_delete));
	js_set(js, usp, "toString", js_mkfun(usp_toString));

	return usp;
	}

	void init_url_module(void) {
	struct js *js = rt->js;
	jsval_t glob = js_glob(js);

	js_set(js, glob, "URL", js_mkfun(js_URL));
	js_set(js, glob, "URLSearchParams", js_mkfun(js_URLSearchParams));
	}

File Metadata

Mime Type: text/x-c
Expires: Fri, Mar 27, 12:20 PM (2 d)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 513090
Default Alt Text: url.c (17 KB)

url.cNo OneTemporaryActions

url.cView Options

File Metadata

Event Timeline

url.c
No OneTemporary
Actions

url.c
View Options