/*
 * Copyright (c) 2017, 2018, 2019 Reyk Floeter <reyk@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/wait.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>

#include <net/if.h>
#include <netinet/in.h>
#include <netinet/if_ether.h>

#include <limits.h>
#include <stdio.h>
#include <syslog.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <signal.h>
#include <unistd.h>
#include <resolv.h>
#include <netdb.h>
#include <ctype.h>
#include <fcntl.h>
#include <errno.h>
#include <poll.h>
#include <pwd.h>
#include <err.h>

#include "main.h"
#include "xml.h"

__dead void	 usage(void);
static struct system_config
		*agent_init(const char *, const char *, int, int);
static int	 agent_configure(struct system_config *);
static int	 agent_network(struct system_config *);
static void	 agent_free(struct system_config *);
static int	 agent_pf(struct system_config *, int);
static int	 agent_userdata(struct system_config *,
		    const unsigned char *, size_t);
static void	 agent_unconfigure(void);
static char	*metadata_parse(char *, size_t, enum strtype);

static int	 agent_timeout;
static char	*cloudnames[] = CLOUDNAMES;

int
shell(const char *arg, ...)
{
	const char	**argv, *a;
	int		 argc, i = 0, status;
	va_list		 ap;
	pid_t		 pid, child_pid;
	struct sigaction sigint, sigquit;
	sigset_t	 mask, omask;

	/* create arguments */
	va_start(ap, arg);
	for (argc = 2; va_arg(ap, const char *) != NULL; argc++)
		;
	va_end(ap);

	if ((argv = calloc(argc, sizeof(const char *))) == NULL)
		fatal("%s: calloc", __func__);
	argv[i++] = arg;

	va_start(ap, arg);
	while ((a = va_arg(ap, char *)) != NULL)
		argv[i++] = a;
	va_end(ap);

	sigemptyset(&mask);
	sigaddset(&mask, SIGCHLD);
	sigprocmask(SIG_BLOCK, &mask, &omask);

	/* run command in forked process */
	switch (child_pid = fork()) {
	case -1:
		sigprocmask(SIG_SETMASK, &omask, NULL);
		free(argv);
		return (-1);
	case 0:
		sigprocmask(SIG_SETMASK, &omask, NULL);
		execvp(argv[0], (char *const *)(caddr_t)argv);
		_exit(127);
	}

	free(argv);
	sigaction(SIGINT, NULL, &sigint);
	sigaction(SIGQUIT, NULL, &sigquit);

	do {
		pid = waitpid(child_pid, &status, 0);
	} while (pid == -1 && errno == EINTR);

	sigprocmask(SIG_SETMASK, &omask, NULL);
	sigaction(SIGINT, &sigint, NULL);
	sigaction(SIGQUIT, &sigquit, NULL);

	/* Simplified return value: returns 0 on success and -1 on error */
	if (pid != -1 && WIFEXITED(status) && WEXITSTATUS(status) == 0)
		return (0);

	return (-1);
}

int
shellout(const char *in, char **out, const char *arg, ...)
{
	const char	**argv = NULL, *a;
	int		 argc, i = 0, status;
	va_list		 ap;
	pid_t		 pid, child_pid;
	struct sigaction sigint, sigquit;
	sigset_t	 mask, omask;
	FILE		*outfp = NULL, *fp = NULL;
	char		*outbuf;
	size_t		 outbufsz;
	char		 buf[BUFSIZ];
	int		 fdi[2], fdo[2];

	if (out)
		*out = NULL;

	/* create arguments */
	va_start(ap, arg);
	for (argc = 2; va_arg(ap, const char *) != NULL; argc++)
		;
	va_end(ap);

	if ((argv = calloc(argc, sizeof(const char *))) == NULL)
		fatal("%s: calloc", __func__);
	argv[i++] = arg;

	va_start(ap, arg);
	while ((a = va_arg(ap, char *)) != NULL)
		argv[i++] = a;
	va_end(ap);

	if (in && socketpair(AF_UNIX,
	    SOCK_STREAM|SOCK_CLOEXEC, AF_UNSPEC, fdi) == -1)
		goto fail;

	if (out && socketpair(AF_UNIX,
	    SOCK_STREAM|SOCK_CLOEXEC, AF_UNSPEC, fdo) == -1)
		goto fail;

	sigemptyset(&mask);
	sigaddset(&mask, SIGCHLD);
	sigprocmask(SIG_BLOCK, &mask, &omask);

	/* run command in forked process */
	switch (child_pid = fork()) {
	case -1:
		sigprocmask(SIG_SETMASK, &omask, NULL);
		goto fail;
	case 0:
		if (in) {
			close(fdi[1]);
			if (dup2(fdi[0], STDIN_FILENO) == -1)
				_exit(127);
		}
		if (out) {
			close(fdo[1]);
			if (dup2(fdo[0], STDOUT_FILENO) == -1)
				_exit(127);
		}
		sigprocmask(SIG_SETMASK, &omask, NULL);
		execvp(argv[0], (char *const *)(caddr_t)argv);
		_exit(127);
	}

	free(argv);
	argv = NULL;
	sigaction(SIGINT, NULL, &sigint);
	sigaction(SIGQUIT, NULL, &sigquit);

	if (in) {
		close(fdi[0]);
		if ((fp = fdopen(fdi[1], "w")) != NULL) {
			fputs(in, fp);
			fflush(fp);
			fclose(fp);
		}
		close(fdi[1]);
	}

	if (out) {
		close(fdo[0]);
		if ((fp = fdopen(fdo[1], "r")) != NULL &&
		    (outfp = open_memstream(&outbuf, &outbufsz)) != NULL) {
			while (fgets(buf, sizeof(buf), fp) != NULL) {
				fputs(buf, outfp);
			}
			fclose(outfp);
			*out = outbuf;
		}
		fclose(fp);
		close(fdo[1]);
	}

	do {
		pid = waitpid(child_pid, &status, 0);
	} while (pid == -1 && errno == EINTR);

	sigprocmask(SIG_SETMASK, &omask, NULL);
	sigaction(SIGINT, &sigint, NULL);
	sigaction(SIGQUIT, &sigquit, NULL);

	/* Simplified return value: returns 0 on success and -1 on error */
	if (pid != -1 && WIFEXITED(status) && WEXITSTATUS(status) == 0)
		return (0);

 fail:
	free(argv);
	if (out) {
		free(*out);
		*out = NULL;
	}
	return (-1);
}

int
disable_output(struct system_config *sc, int fd)
{
	int	 oldfd;

	if (log_getverbose() > 2)
		return (-1);

	if ((oldfd = dup(fd)) == -1 ||
	    dup2(sc->sc_nullfd, fd) == -1)
		return (-1);

	return (oldfd);
}

int
enable_output(struct system_config *sc, int fd, int oldfd)
{
	if (oldfd == -1)
		return (0);

	close(fd);
	if (dup2(oldfd, fd) == -1)
		return (-1);

	return (0);
}

char *
get_string(const unsigned char *ptr, size_t len)
{
	size_t	 i;

	/*
	 * We don't use vis(3) here because the string should not be
	 * modified and only validated for printable characters and proper
	 * NUL-termination.  From relayd.
	 */
	for (i = 0; i < len; i++)
		if (!(isprint((unsigned char)ptr[i]) ||
		    isspace((unsigned char)ptr[i])))
			break;

	return strndup(ptr, i);
}

char *
get_line(const unsigned char *ptr, size_t len)
{
	size_t	 i;

	/* Like the previous, but without newlines */
	for (i = 0; i < len; i++)
		if (!isprint((unsigned char)ptr[i]) ||
		    (isspace((unsigned char)ptr[i]) &&
		    !isblank((unsigned char)ptr[i])))
			break;

	return strndup(ptr, i);
}

char *
get_word(const unsigned char *ptr, size_t len)
{
	size_t	 i;

	/* Like the previous, but without spaces and newlines */
	for (i = 0; i < len; i++)
		if (!isprint((unsigned char)ptr[i]) ||
		    isspace((unsigned char)ptr[i]))
			break;

	return strndup(ptr, i);
}

static struct system_config *
agent_init(const char *ifname, const char *rootdisk, int dryrun, int timeout)
{
	struct system_config	*sc;
	int			 fd, ret;

	if ((sc = calloc(1, sizeof(*sc))) == NULL)
		return (NULL);

	sc->sc_interface = ifname;
	sc->sc_cdrom = "/dev/cd0c";
	sc->sc_dryrun = dryrun ? 1 : 0;
	sc->sc_timeout = agent_timeout = timeout < 1 ? -1 : timeout * 1000;
	sc->sc_rootdisk = rootdisk;
	TAILQ_INIT(&sc->sc_pubkeys);
	TAILQ_INIT(&sc->sc_netaddrs);

	if ((sc->sc_nullfd = open("/dev/null", O_RDWR)) == -1) {
		free(sc);
		return (NULL);
	}

	/* Silently try to mount the cdrom */
	fd = disable_output(sc, STDERR_FILENO);
	ret = shell("mount", "-r", sc->sc_cdrom, "/mnt", NULL);
	enable_output(sc, STDERR_FILENO, fd);
	if (ret == 0) {
		log_debug("%s: mounted %s", __func__, sc->sc_cdrom);
		sc->sc_mount = 1;
	}

	if (sc->sc_dryrun)
		return (sc);

	if (agent_pf(sc, 1) != 0)
		fatalx("pf");
	if (http_init() == -1)
		fatalx("http_init");

	return (sc);
}

static void
agent_free(struct system_config *sc)
{
	struct ssh_pubkey	*ssh;
	struct net_addr		*net;

	/* unmount if we mounted the cdrom before */
	if (sc->sc_mount && shell("umount", "/mnt", NULL) == 0) {
		log_debug("%s: unmounted %s", __func__, sc->sc_cdrom);
	}

	if (sc->sc_password_plain != NULL) {
		/* XXX can be removed with calloc_conceal() post-6.6 */
		explicit_bzero(sc->sc_password_plain,
		    strlen(sc->sc_password_plain));
		free(sc->sc_password_plain);
	}
	free(sc->sc_password_hash);
	free(sc->sc_hostname);
	free(sc->sc_username);
	free(sc->sc_userdata);
	free(sc->sc_endpoint);
	free(sc->sc_instance);
	free(sc->sc_args);
	close(sc->sc_nullfd);

	while ((ssh = TAILQ_FIRST(&sc->sc_pubkeys))) {
		free(ssh->ssh_keyval);
		free(ssh->ssh_keyfp);
		TAILQ_REMOVE(&sc->sc_pubkeys, ssh, ssh_entry);
		free(ssh);
	}

	while ((net = TAILQ_FIRST(&sc->sc_netaddrs))) {
		TAILQ_REMOVE(&sc->sc_netaddrs, net, net_entry);
		free(net->net_value);
		free(net);
	}
}

int
agent_addpubkey(struct system_config *sc, const char *sshval, const char *sshfp)
{
	struct ssh_pubkey	*ssh;

	/* Ignore if neither key nor fingerprint is available */
	if (sshval == NULL && sshfp == NULL)
		return (0);

	if ((ssh = calloc(1, sizeof(*ssh))) == NULL)
		return (-1);

	if (sshfp != NULL && (ssh->ssh_keyfp = strdup(sshfp)) == NULL) {
		free(ssh);
		return (-1);
	}

	if (sshval != NULL && (ssh->ssh_keyval = strdup(sshval)) == NULL) {
		free(ssh->ssh_keyfp);
		free(ssh);
		return (-1);
	}

	TAILQ_INSERT_TAIL(&sc->sc_pubkeys, ssh, ssh_entry);

	return (0);
}

int
agent_setpubkey(struct system_config *sc, const char *sshval, const char *sshfp)
{
	struct ssh_pubkey	*ssh;
	int			 ret = 0;
	char			*v = NULL;

	TAILQ_FOREACH(ssh, &sc->sc_pubkeys, ssh_entry) {
		if (sshfp && ssh->ssh_keyfp &&
		    strcasecmp(ssh->ssh_keyfp, sshfp) == 0) {
			if ((sshval == NULL) ||
			    (sshval && (v = strdup(sshval)) == NULL))
				break;
			v[strcspn(v, "\r\n")] = '\0';
			free(ssh->ssh_keyval);
			ssh->ssh_keyval = v;
			ret++;
		}
	}

	return (ret);
}

struct net_addr *
agent_getnetaddr(struct system_config *sc, struct net_addr *net)
{
	struct net_addr		*na;

	TAILQ_FOREACH(na, &sc->sc_netaddrs, net_entry) {
		if (na->net_type != net->net_type)
			continue;
		if (na->net_ifunit != net->net_ifunit)
			continue;
		if (na->net_type == NET_DNS_DOMAIN &&
		    strcasecmp(na->net_value, net->net_value) != 0)
			continue;
		if (net->net_addr.ss_family != AF_UNSPEC) {
			if (na->net_addr.ss_family !=
			    net->net_addr.ss_family)
				continue;
			if (memcmp(&na->net_addr, &net->net_addr,
			    na->net_addr.ss_len) != 0)
				continue;
		}

		return (na);
	}

	return (NULL);
}

int
agent_addnetaddr(struct system_config *sc, unsigned int unit,
    const char *value, int af, enum net_type type)
{
	const char		*errstr;
	struct addrinfo		 hints, *res;
	struct net_addr		*net, *na;

	if ((net = calloc(1, sizeof(*net))) == NULL) {
		log_debug("%s: calloc", __func__);
		return (-1);
	}
	net->net_ifunit = unit;
	net->net_type = type;

	switch (type) {
	case NET_DNS_DOMAIN:
		if (strlen(value) >= NI_MAXHOST) {
			log_debug("%s: if%u domain %s", __func__, unit, value);
			free(net);
			return (-1);
		}
		break;
	case NET_MAC:
		if (ether_aton(value) == NULL) {
			log_debug("%s: if%u mac %s", __func__, unit, value);
			free(net);
			return (-1);
		}
		break;
	case NET_MTU:
	case NET_PREFIX:
		net->net_num = strtonum(value, 0, UINT32_MAX, &errstr);
		if (errstr != NULL) {
			log_debug("%s: if%u %s", __func__, unit, value);
			free(net);
			return (-1);
		}
		break;
	default:
		memset(&hints, 0, sizeof(hints));
		hints.ai_family = af;
		hints.ai_socktype = SOCK_DGRAM;
		hints.ai_flags = AI_NUMERICHOST;
		if (getaddrinfo(value, "0", &hints, &res) != 0) {
			log_debug("%s: invalid address %s",
			    __func__, value);
			free(net);
			return (-1);
		}

		if (res->ai_addrlen > sizeof(net->net_addr)) {
			log_debug("%s: address too long",
			    __func__);
			free(net);
			freeaddrinfo(res);
			return (-1);
		}
		memcpy(&net->net_addr, res->ai_addr, res->ai_addrlen);
		net->net_addr.ss_len = res->ai_addrlen;
		net->net_addr.ss_family = res->ai_family;
	}

	if ((net->net_value = strdup(value)) == NULL) {
		free(net);
		return (-1);
	}

	/* Address already exists, ignore new entry */
	if ((na = agent_getnetaddr(sc, net)) != NULL) {
		free(net->net_value);
		free(net);
		return (0);
	}

	TAILQ_INSERT_TAIL(&sc->sc_netaddrs, net, net_entry);

	return (0);
}

static int
fileout(const char *str, const char *mode, const char *fmt, ...)
{
	FILE	*fp;
	va_list	 ap;
	char	*path;
	int	 ret;

	va_start(ap, fmt);
	ret = vasprintf(&path, fmt, ap);
	va_end(ap);

	if (ret == -1)
		return (-1);
	if ((fp = fopen(path, mode)) == NULL) {
		free(path);
		return (-1);
	}
	if (str != NULL) {
		fputs(str, fp);
		if (strpbrk(str, "\r\n") == NULL)
			fputs("\n", fp);
	}
	fclose(fp);

	free(path);

	return (0);
}

static char *
filein(const char *mode, const char *fmt, ...)
{
	FILE	*fp;
	va_list	 ap;
	char	*path;
	int	 ret;
	char	 buf[BUFSIZ];
	FILE	*infp;
	char	*inbuf;
	size_t	 inbufsz;

	va_start(ap, fmt);
	ret = vasprintf(&path, fmt, ap);
	va_end(ap);

	if (ret == -1)
		return (NULL);
	if ((fp = fopen(path, mode)) == NULL) {
		free(path);
		return (NULL);
	}
	free(path);
	if ((infp = open_memstream(&inbuf, &inbufsz)) == NULL) {
		fclose(fp);
		return (NULL);
	}
	while (fgets(buf, sizeof(buf), fp) != NULL) {
		fputs(buf, infp);
	}
	fclose(fp);
	fclose(infp);

	return (inbuf);
}

static int
agent_pf(struct system_config *sc, int open)
{
	int	 ret;

	if (shell("rcctl", "get", "pf", "status", NULL) != 0)
		return (0);

	if (open)
		ret = shellout("pass out proto tcp from (egress) to port www\n",
		    NULL, "pfctl", "-f", "-", NULL);
	else
		ret = shellout("\n", NULL, "pfctl", "-f", "-", NULL);

	return (ret);
}

static int
agent_configure(struct system_config *sc)
{
	char			 pwbuf[_PASSWORD_LEN + 2];
	struct ssh_pubkey	*ssh;
	char			*str1, *str2;

	/* Skip configuration on the same instance */
	if ((str1 = filein("r", "/var/db/cloud-instance")) != NULL) {
		str1[strcspn(str1, "\r\n")] = '\0';
		if (strcmp(sc->sc_instance, str1) == 0) {
			free(str1);
			return (0);
		}
	}
	free(str1);

	if (fileout(sc->sc_instance, "w", "/var/db/cloud-instance") != 0)
		log_warnx("instance failed");

	/* Set default username if not set */
	if ((sc->sc_username == NULL) &&
	    (sc->sc_username = strdup("puffy")) == NULL) {
		log_warn("default username");
		return (-1);
	}

	/* hostname */
	log_debug("%s: hostname %s", __func__, sc->sc_hostname);
	if (fileout(sc->sc_hostname, "w", "/etc/myname") != 0)
		log_warnx("hostname failed");
	else
		(void)shell("hostname", sc->sc_hostname, NULL);

	/* username */
	log_debug("%s: username %s", __func__, sc->sc_username);
	if (strcmp("root", sc->sc_username) != 0) {
		if (shell("useradd", "-L", "staff", "-G", "wheel",
		    "-m", sc->sc_username, NULL) != 0)
			log_warnx("username failed");
		if (fileout(sc->sc_username, "w", "/root/.forward") != 0)
			log_warnx(".forward failed");
	}

	/* password */
	if (sc->sc_password_hash == NULL) {
		if (asprintf(&str2, "permit keepenv nopass %s as root\n"
		    "permit keepenv nopass root\n", sc->sc_username) == -1)
			str2 = NULL;
	} else {
		if (shell("usermod", "-p", sc->sc_password_hash,
		    sc->sc_username, NULL) != 0)
			log_warnx("password failed");

		if (asprintf(&str2, "permit keepenv persist %s as root\n"
		    "permit keepenv nopass root\n", sc->sc_username) == -1)
			str2 = NULL;

		/* write generated password as comment to authorized_keys */
		if (sc->sc_password_plain != NULL) {
			snprintf(pwbuf, sizeof(pwbuf), "# %s",
			    sc->sc_password_plain);
			if (fileout(pwbuf, "w",
			    "%s/%s/.ssh/authorized_keys",
			    strcmp("root", sc->sc_username) == 0 ? "" : "/home",
			    sc->sc_username) != 0)
				log_warnx("password comment failed");
			explicit_bzero(pwbuf, sizeof(pwbuf));
		}
	}

	/* doas */
	if ((strcmp("root", sc->sc_username) != 0) &&
	    (str2 == NULL || fileout(str2, "w", "/etc/doas.conf")) != 0)
		log_warnx("doas failed");
	free(str2);

	/* ssh configuration */
	if (sc->sc_password_hash == NULL && !TAILQ_EMPTY(&sc->sc_pubkeys))
		str1 = "/PasswordAuthentication/"
		    "s/.*/PasswordAuthentication no/";
	else
		str1 = "/PasswordAuthentication/"
		    "s/.*/PasswordAuthentication yes/";
	shell("sed", "-i", "-e", str1,
	    "-e", "/ClientAliveInterval/s/.*/ClientAliveInterval 180/",
	    "/etc/ssh/sshd_config",
	    NULL);

	/* ssh public keys */
	TAILQ_FOREACH(ssh, &sc->sc_pubkeys, ssh_entry) {
		if (ssh->ssh_keyval == NULL)
			continue;
		log_debug("%s: key %s", __func__, ssh->ssh_keyval);
		if (fileout(ssh->ssh_keyval, "a",
		    "%s/%s/.ssh/authorized_keys",
		    strcmp("root", sc->sc_username) == 0 ? "" : "/home",
		    sc->sc_username) != 0)
			log_warnx("public key failed");
	}

	if (sc->sc_userdata) {
		if (agent_userdata(sc, sc->sc_userdata,
		    strlen(sc->sc_userdata)) != 0)
			log_warnx("user-data failed");
	}

	if (agent_network(sc) != 0)
		log_warnx("network configuration failed");

	log_debug("%s: %s", __func__, "/etc/rc.firsttime");
	if (fileout("logger -s -t cloud-agent <<EOF\n"
	    "#############################################################\n"
	    "-----BEGIN SSH HOST KEY FINGERPRINTS-----\n"
	    "$(for _f in /etc/ssh/ssh_host_*_key.pub;"
	    " do ssh-keygen -lf ${_f}; done)\n"
	    "-----END SSH HOST KEY FINGERPRINTS-----\n"
	    "#############################################################\n"
	    "EOF\n",
	    "a", "/etc/rc.firsttime") != 0)
		log_warnx("ssh fingerprints failed");

	return (0);
}

static int
agent_userdata(struct system_config *sc,
    const unsigned char *userdata, size_t len)
{
	char		*shebang = NULL, *str = NULL, *line = NULL;
	const char	*file;
	int		 ret = -1;

	if (len <= 2) {
		log_warnx("user-data too short");
		goto fail;
	}

	if (userdata[0] == 0x1f && userdata[1] == 0x8b) {
		log_warnx("gzip-compressed user-data is not supported");
		goto fail;
	} else if (userdata[0] == '#') {
		if ((shebang = get_line(userdata, len)) == NULL) {
			log_warnx("failed to decode shebang from user-data");
			goto fail;
		}
	} else if (isprint(userdata[0]) && isprint(userdata[1])) {
		/* Decode user-data and call the function again */
		if ((str = calloc(1, len + 1)) == NULL ||
		    (len = b64_pton(userdata, str, len)) < 1 ||
		    agent_userdata(sc, str, len) != 0) {
			log_warnx("failed to decode user-data");
			goto fail;
		}
		goto done;
	}

	log_debug("%s: user-data: %s", __func__, shebang);

	if (strlen(shebang) <= 2 || strncmp("#!", shebang, 2) != 0) {
		log_warnx("unsupported user-data type");
		goto fail;
	}

	/* now get the whole script */
	if ((str = get_string(userdata, len)) == NULL) {
		log_warnx("invalid user-data script");
		goto fail;
	}

	if (sc->sc_dryrun)
		goto done;

	/* write user-data script into file */
	file = "/etc/rc.user-data";
	if (fileout(str, "w", file) != 0) {
		log_warnx("failed to write user-data");
		goto fail;
	}

	/* and call it from rc.firsttime later on boot */
	if (asprintf(&line,
	    "logger -s -t cloud-agent \"running user-data\"\n"
	    "%s %s\nrm %s\n", shebang + 2, file, file) == -1 ||
	    fileout(line, "a", "/etc/rc.firsttime") != 0)
		log_warnx("failed to add user-data script");

 done:
	ret = 0;
 fail:
	free(line);
	free(str);
	free(shebang);

	return (ret);
}

static int
agent_network(struct system_config *sc)
{
	struct net_addr	*net;
	char		 ift[16], ifname[16], line[1024];
	const char	*family;
	char		 domain[(NI_MAXHOST + 1) * 6 + 8]; /* up to 6 domains */
	int		 has_domain = 0;
	char		 ifidx[UINT16_MAX];
	const char	*comment = "# Generated by cloud-agent";

	if (!sc->sc_network)
		return (0);

	if (strlcpy(ift, sc->sc_interface, sizeof(ift)) >= sizeof(ift))
		return (-1);
	ift[strcspn(ift, "0123456789")] = '\0';

	memset(ifidx, 0, sizeof(ifidx));
	snprintf(domain, sizeof(domain), "search ");
	fileout(comment, "w", "/etc/mygate");
	fileout(comment, "w", "/etc/resolv.conf");

	TAILQ_FOREACH(net, &sc->sc_netaddrs, net_entry) {
		snprintf(ifname, sizeof(ifname), "%s%u", ift, net->net_ifunit);
		switch (net->net_type) {
		case NET_IP:
			family = net->net_addr.ss_family == AF_INET ?
			    "inet" : "inet6";
			/* XXX prefix or mask */

			/* hostname.if startup configuration */
			if (!ifidx[net->net_ifunit])
				fileout(comment, "w",
				    "/etc/hostname.%s", ifname);

			snprintf(line, sizeof(line), "%s alias %s",
			    family, net->net_value);
			fileout(line, "a", "/etc/hostname.%s", ifname);

			if (!ifidx[net->net_ifunit]++ &&
			    net->net_ifunit == 0) {
				snprintf(line, sizeof(line),
				    "!%s", sc->sc_args);
				fileout(line, "a", "/etc/hostname.%s", ifname);
			}

			/* runtime configuration */
			(void)shell("ifconfig", ifname, family,
			    "alias", net->net_value, NULL);
			break;
		case NET_GATEWAY:
			fileout(net->net_value, "a", "/etc/mygate");
			break;
		case NET_DNS:
			snprintf(line, sizeof(line), "nameserver %s",
			    net->net_value);
			fileout(line, "a", "/etc/resolv.conf");
			break;
		case NET_DNS_DOMAIN:
			if (!has_domain++) {
				/* use the first search domain as our own */
				snprintf(line, sizeof(line), "domain %s",
				    net->net_value);
				fileout(line, "a", "/etc/resolv.conf");
			} else
				(void)strlcat(domain, " ", sizeof(domain));
			(void)strlcat(domain, net->net_value, sizeof(domain));
			break;
		default:
			break;
		}
	}

	if (has_domain)
		fileout(domain, "a", "/etc/resolv.conf");

	return (0);
}

void
agent_unconfigure(void)
{
	/* Disable root pasword */
	(void)shell("chpass", "-a",
	    "root:*:0:0:daemon:0:0:Charlie &:/root:/bin/ksh", NULL);

	/* Delete keys */
	(void)shell("sh", "-c",
	    "rm -rf /etc/{iked,isakmpd}/{local.pub,private/local.key}"
	    " /etc/ssh/ssh_host_*"
	    " /etc/dhclient.conf /var/db/dhclient.leases.*"
	    " /tmp/{.[!.],}*", NULL);

	/* Delete old seed files */
	(void)fileout(NULL, "w", "/etc/random.seed");
	(void)fileout(NULL, "w", "/var/db/host.random");

	/* Clear logfiles */
	(void)shell("sh", "-c",
	    "for _l in $(find /var/log -type f ! -name '*.gz' -size +0); do"
	    " >${_l}; "
	    "done", NULL);

	(void)fileout("permit keepenv persist :wheel as root\n"
	    "permit keepenv nopass root\n", "w", "/etc/doas.conf");

	/* Remove cloud-instance file */
	(void)unlink("/var/db/cloud-instance");
}

static char *
metadata_parse(char *s, size_t sz, enum strtype type)
{
	char	*str;

	switch (type) {
	case TEXT:
		/* multi-line string, always printable */
		str = get_string(s, sz);
		break;
	case LINE:
		str = get_line(s, sz);
		break;
	case WORD:
		str = get_word(s, sz);
		break;
	}

	return (str);
}

char *
metadata(struct system_config *sc, const char *path, enum strtype type)
{
	struct httpget	*g = NULL;
	char		*str = NULL;

	g = http_get(&sc->sc_addr, 1,
	    sc->sc_endpoint, 80, path, NULL, 0, NULL);
	if (g != NULL)
		log_debug("%s: HTTP %d http://%s%s", __func__, g->code,
		    sc->sc_endpoint, path);

	if (g != NULL && g->code == 200 && g->bodypartsz > 0)
		str = metadata_parse(g->bodypart, g->bodypartsz, type);
	http_get_free(g);

	return (str);
}

char *
metadata_file(struct system_config *sc, const char *name, enum strtype type)
{
	FILE		*fp, *mfp;
	char		 buf[BUFSIZ], *mbuf, *str;
	size_t		 sz, msz;

	if ((fp = fopen(name, "r")) == NULL) {
		log_warn("%s: could not open %s", __func__, name);
		return (NULL);
	}

	if ((mfp = open_memstream(&mbuf, &msz)) == NULL) {
		log_warn("%s: open_memstream", __func__);
		fclose(fp);
		return (NULL);
	}

	do {
		if ((sz = fread(buf, 1, sizeof(buf), fp)) < 1)
			break;
		if (fwrite(buf, sz, 1, mfp) != 1)
			break;
	} while (sz == sizeof(buf));

	fclose(mfp);
	fclose(fp);

	str = metadata_parse(mbuf, msz, type);
	free(mbuf);

	return (str);
}

int
connect_wait(int s, const struct sockaddr *name, socklen_t namelen)
{
	struct pollfd	 pfd[1];
	int		 error = 0, flag;
	socklen_t	 errlen = sizeof(error);

	if ((flag = fcntl(s, F_GETFL, 0)) == -1 ||
	    (fcntl(s, F_SETFL, flag | O_NONBLOCK)) == -1)
		return (-1);

	error = connect(s, name, namelen);
	do {
		pfd[0].fd = s;
		pfd[0].events = POLLOUT;

		if ((error = poll(pfd, 1, agent_timeout)) == -1)
			continue;
		if (error == 0) {
			error = ETIMEDOUT;
			goto done;
		}
		if (getsockopt(s, SOL_SOCKET, SO_ERROR, &error, &errlen) == -1)
			continue;
	} while (error != 0 && error == EINTR);

 done:
	if (fcntl(s, F_SETFL, flag & ~O_NONBLOCK) == -1)
		return (-1);

	if (error != 0) {
		errno = error;
		return (-1);
	}

	return (0);
}

int
dhcp_getendpoint(struct system_config *sc)
{
	char	 path[PATH_MAX], buf[BUFSIZ], *ep = NULL;
	int	 a[4], has245 = 0;
	size_t	 sz;
	FILE	*fp;

	if ((size_t)snprintf(path, sizeof(path), "/var/db/dhclient.leases.%s",
	    sc->sc_interface) >= sizeof(path)) {
		log_debug("%s: invalid path", __func__);
		return (-1);
	}

	if ((fp = fopen(path, "r")) == NULL) {
		log_debug("%s: failed to open %s", __func__, path);
		return (-1);
	}

	while (fgets(buf, sizeof(buf), fp) != NULL) {
		buf[strcspn(buf, ";\n")] = '\0';

		/* Find last occurence of dhcp-server-identifier */
		sz = strlen("  option dhcp-server-identifier ");
		if (!has245 &&
		    strncmp(buf, "  option dhcp-server-identifier ", sz) == 0) {
			free(ep);
			if ((ep = strdup(buf + sz)) == NULL) {
				log_debug("%s: strdup", __func__);
				fclose(fp);
				return (-1);
			}
		}

		/* Find last occurence of option-245 (only on Azure) */
		if (sscanf(buf, "  option option-245 %x:%x:%x:%x",
		    &a[0], &a[1], &a[2], &a[3]) == 4) {
			has245 = 1;
			free(ep);
			if (asprintf(&ep, "%d.%d.%d.%d",
			    a[0], a[1], a[2], a[3]) == -1) {
				log_debug("%s: asprintf", __func__);
				fclose(fp);
				return (-1);
			}
		}
	}

	fclose(fp);

	if (ep == NULL)
		return (-1);

	sc->sc_endpoint = ep;
	sc->sc_addr.ip = sc->sc_endpoint;
	sc->sc_addr.family = 4;

	if (log_getverbose() > 2)
		log_debug("%s: %s", __func__, ep);

	return (0);
}

__dead void
usage(void)
{
	extern char	*__progname;

	fprintf(stderr, "usage: %s [-nuv] "
	    "[-c cloud[,...]] [-p length] [-r rootdisk]\n\t"
	    "[-t 3] [-U puffy] interface\n", __progname);
	exit(1);
}

static char *
get_args(int argc, char *const *argv)
{
	char		*args, path[PATH_MAX];
	size_t		 argslen = 0;
	int		 i;

	/* Store args in a string */
	for (i = 0; i < argc; i++) {
		if (i == 0) {
			realpath(argv[0], path);
			argslen += strlen(path) + 1;
		} else {
			argslen += strlen(argv[i]) + 1;
		}
	}
	if ((args = calloc(1, argslen + 1)) == NULL)
		return (NULL);
	for (i = 0; i < argc; i++) {
		if (i == 0)
			strlcat(args, path, argslen);
		else {
			strlcat(args, " ", argslen);
			strlcat(args, argv[i], argslen);
		}
	}

	return (args);
}

static char *
pwgen(size_t len, char **hash)
{
	char		*password;
	size_t		 i, alphabet_len;
	const char	*alphabet =
	    "0123456789_"
	    "abcdefghijklmnopqrstuvwxyz"
	    "ABCDEFGHIJKLMNOPQRSTUVWXYZ";

	alphabet_len = strlen(alphabet);
	*hash = NULL;

	/* XXX use calloc_conceal() post-6.6 */
	if ((password = calloc(1, len + 1)) == NULL)
		return (NULL);

	/* Simple password generator */
	for (i = 0; i < len; i++)
		password[i] = alphabet[arc4random_uniform(alphabet_len)];

	if ((*hash = calloc(1, _PASSWORD_LEN)) == NULL) {
		freezero(password, len + 1);
		return (NULL);
	}

	if (crypt_newhash(password, "bcrypt,a",
	    *hash, _PASSWORD_LEN) != 0) {
		freezero(password, len + 1);
		freezero(*hash, _PASSWORD_LEN);
		password = NULL;
		*hash = NULL;
	}

	return (password);
}

static void
cloud_add(enum cloudname name, struct clouds *clouds)
{
	struct cloud	*cloud;

	TAILQ_FOREACH(cloud, clouds, cloud_entry) {
		if (cloud->cloud_name == name)
			fatalx("cloud %s defined twice",
			    cloudnames[name]);
	}

	if ((cloud = calloc(1, sizeof(*cloud))) == NULL)
		fatal("%s: calloc", __func__);

	cloud->cloud_name = name;
	switch (name) {
	case AZURE:
		cloud->fetch = azure;
		break;
	case CLOUDINIT:
		cloud->fetch = cloudinit;
		break;
	case EC2:
		cloud->fetch = ec2;
		break;
	case OPENNEBULA:
		cloud->fetch = opennebula;
		break;
	case OPENSTACK:
		cloud->fetch = openstack;
		break;
	}

	TAILQ_INSERT_TAIL(clouds, cloud, cloud_entry);
}

static int
trycloud(struct system_config *sc, struct cloud *cloud)
{
	int		 errfd = -1, ret = -1;

	free(sc->sc_endpoint);
	sc->sc_endpoint = NULL;

	switch (sc->sc_state) {
	case STATE_INIT:
		if ((cloud = TAILQ_FIRST(sc->sc_clouds)) == NULL)
			return (0);
		sc->sc_stack = cloudnames[cloud->cloud_name];
		log_debug("%s: %s", __func__, sc->sc_stack);
		TAILQ_REMOVE(sc->sc_clouds, cloud, cloud_entry);
		break;
	case STATE_DHCP:
		sc->sc_state = STATE_169;
		if (dhcp_getendpoint(sc) == -1)
			return trycloud(sc, cloud);
		break;
	case STATE_169:
		sc->sc_state = STATE_DONE;
		if ((sc->sc_endpoint = strdup(DEFAULT_ENDPOINT)) == NULL) {
			log_warnx("failed to set defaults");
			goto done;
		}
		break;
	case STATE_DONE:
		sc->sc_state = STATE_INIT;
		ret = trycloud(sc, NULL);
		goto done;
	}

	errfd = disable_output(sc, STDERR_FILENO);
	ret = (*cloud->fetch)(sc);
	enable_output(sc, STDERR_FILENO, errfd);
	if (ret != 0)
		return trycloud(sc, cloud);

 done:
	free(cloud);
	return (0);
}

int
main(int argc, char *const *argv)
{
	struct system_config	*sc;
	int			 verbose = 0, dryrun = 0, unconfigure = 0, sub;
	int			 genpw = 0, ch, ret, timeout = CONNECT_TIMEOUT;
	const char		*error = NULL, *rootdisk = NULL;
	char			*args, *username = NULL, *options, *value;
	struct clouds		 clouds;

	/* log to stderr */
	log_init(1, LOG_DAEMON);

	if ((args = get_args(argc, argv)) == NULL)
		fatalx("failed to save args");

	TAILQ_INIT(&clouds);

	while ((ch = getopt(argc, argv, "c:np:r:t:U:uv")) != -1) {
		switch (ch) {
		case 'c':
			options = optarg;
			while (*options) {
				if ((sub = getsubopt(&options,
				    cloudnames, &value)) == -1)
					fatalx("invalid cloud stack");
				else if (value != NULL)
					fatalx("unexpected value");
				cloud_add(sub, &clouds);
			}
			break;
		case 'n':
			dryrun = 1;
			break;
		case 'p':
			genpw = strtonum(optarg, 8, 8192, &error);
			if (error != NULL)
				fatalx("invalid password length: %s", error);
			break;
		case 'r':
			rootdisk = optarg;
			break;
		case 't':
			timeout = strtonum(optarg, -1, 86400, &error);
			if (error != NULL)
				fatalx("invalid timeout: %s", error);
			break;
		case 'U':
			if ((username = strdup(optarg)) == NULL)
				fatal("username");
			break;
		case 'u':
			unconfigure = 1;
			break;
		case 'v':
			verbose += 2;
			break;
		default:
			usage();
		}
	}

	argv += optind;
	argc -= optind;

	log_setverbose(verbose);

	if (unconfigure) {
		agent_unconfigure();
		exit(0);
	}

	if (argc != 1)
		usage();

	if (pledge(rootdisk == NULL ?
	    "stdio cpath rpath wpath exec proc dns inet" :
	    "stdio cpath rpath wpath exec proc dns inet disklabel",
	    NULL) == -1)
		fatal("pledge");

	if ((sc = agent_init(argv[0], rootdisk, dryrun, timeout)) == NULL)
		fatalx("agent");

	if (rootdisk != NULL && growdisk(sc) == -1)
		fatalx("failed to grow %s", rootdisk);

	sc->sc_clouds = &clouds;
	sc->sc_args = args;
	if (username != NULL) {
		free(sc->sc_username);
		sc->sc_username = username;
	}

	if (TAILQ_EMPTY(&clouds)) {
		/*
		 * XXX Auto-detect cloud with help from hostctl and
		 * XXX sysctl in addition to the interface name.
		 */
		if (strcmp("hvn0", sc->sc_interface) == 0) {
			cloud_add(AZURE, &clouds);
		} else if (strcmp("xnf0", sc->sc_interface) == 0) {
			cloud_add(OPENNEBULA, &clouds);
			cloud_add(EC2, &clouds);
		} else {
			cloud_add(OPENNEBULA, &clouds);
			cloud_add(OPENSTACK, &clouds);
			cloud_add(CLOUDINIT, &clouds);
		}
	}
	ret = trycloud(sc, NULL);

	if (genpw) {
		if (sc->sc_password_hash != NULL)
			log_debug("%s: user password hash: %s", __func__,
			    sc->sc_password_hash);
		else if ((sc->sc_password_plain = pwgen(genpw,
		    &sc->sc_password_hash)) != NULL) {
			if (log_getverbose() > 2)
				log_debug("%s: generated password: %s",
				    __func__, sc->sc_password_plain);
		} else
			log_warnx("failed to generate password");
	}

	/* Debug userdata */
	if (sc->sc_dryrun && sc->sc_userdata) {
		if (agent_userdata(sc, sc->sc_userdata,
		    strlen(sc->sc_userdata)) != 0)
			log_warnx("user-data failed");
	}

	if (sc->sc_stack != NULL)
		log_debug("%s: %s", __func__, sc->sc_stack);

	if (sc->sc_dryrun) {
		agent_free(sc);
		return (0);
	}

	if (agent_pf(sc, 0) != 0)
		fatalx("pf");

	if (pledge("stdio cpath rpath wpath exec proc", NULL) == -1)
		fatal("pledge");

	if (ret == 0 && agent_configure(sc) != 0)
		fatal("provisioning failed");

	agent_free(sc);

	return (0);
}