From e6fdfd4c4c753fe3a06edc4ae3b767c57c10d3f7 Mon Sep 17 00:00:00 2001 From: David Härdeman Date: Sat, 11 Jul 2020 18:07:27 +0200 Subject: Flesh out the new command a bit more --- minecctl/misc-commands.c | 352 ++++++++++++++++++++++++++++++++++++----------- 1 file changed, 269 insertions(+), 83 deletions(-) (limited to 'minecctl/misc-commands.c') diff --git a/minecctl/misc-commands.c b/minecctl/misc-commands.c index 34cbedf..f3ddff2 100644 --- a/minecctl/misc-commands.c +++ b/minecctl/misc-commands.c @@ -7,6 +7,8 @@ #include #include #include +#include +#include #include "shared/utils.h" #include "minecctl.h" @@ -35,10 +37,11 @@ static bool create_link(int dfd, const char *source, const char *target) return true; } -static bool write_cfg_file(int dfd, const char *name, unsigned char *content, +static bool write_cfg_file(int dfd, const char *name, + const unsigned char *content, size_t len) { - int fd; + _cleanup_close_ int fd; ssize_t done = 0, r; bool rv = true; @@ -61,13 +64,12 @@ static bool write_cfg_file(int dfd, const char *name, unsigned char *content, done += r; } - close(fd); return rv; } static bool find_user_service(int xfd, const char *service) { - int dfd; + _cleanup_close_ int dfd = -1; /* FIXME: Make this a macro, make paths #defines */ char sub_path[STRLEN("systemd/user/") + strlen(service) + 1]; char etc_path[STRLEN("/etc/systemd/user/") + strlen(service) + 1]; @@ -102,10 +104,8 @@ static bool find_user_service(int xfd, const char *service) if (faccessat(dfd, sub_path, R_OK, 0) == 0) { info("User service %s already installed in " "$XDG_DATA_HOME/systemd/user/", service); - close(dfd); return true; } - close(dfd); } if (access(usr_path, R_OK) == 0) { @@ -120,118 +120,98 @@ static bool find_user_service(int xfd, const char *service) static bool create_user_service(int xfd, const char *service, unsigned char *content, size_t len) { - int sfd, ufd = -1; - bool rv = false; + _cleanup_close_ int sfd = -1; + _cleanup_close_ int ufd = -1; if (find_user_service(xfd, service)) return true; sfd = open_subdir(xfd, "systemd", true); if (sfd < 0) - goto out; + return false; ufd = open_subdir(sfd, "user", true); if (ufd < 0) - goto out; + return false; if (!write_cfg_file(ufd, service, content, len)) - goto out; + return false; info("Created user service file $XDG_CONFIG_HOME/systemd/user/%s", service); - rv = true; - -out: - if (ufd >= 0) - close(ufd); - if (sfd >= 0) - close(sfd); - return rv; + return true; } static bool create_server_cfg(int sfd, int cfd, const char *name, - const char *filename) + const char *filename, + const unsigned char *properties, + size_t properties_len) { - int efd = -1; - bool rv = false; + _cleanup_close_ int efd = -1; efd = open_subdir(sfd, name, true); if (efd < 0) { error("Failed to create configuration directory \"%s\"", name); - goto out; + return false; } if (!write_cfg_file(efd, "eula.txt", ___examples_eula_txt, ___examples_eula_txt_len)) - goto out; + return false; if (!write_cfg_file(efd, "server.properties", - ___examples_server_properties, - ___examples_server_properties_len)) - goto out; + properties, properties_len)) + return false; if (!create_link(efd, "server.jar", "../server.jar")) - goto out; + return false; if (!write_cfg_file(cfd, filename, ___examples_example_mcserver, ___examples_example_mcserver_len)) - goto out; - - rv = true; - -out: - if (efd >= 0) - close(efd); + return false; - return rv; + return true; } static bool create_config_tree(int xfd) { - int mfd = -1, sfd = -1, cfd = -1; - bool rv = false; + _cleanup_close_ int mfd = -1; + _cleanup_close_ int sfd = -1; + _cleanup_close_ int cfd = -1; mfd = open_subdir(xfd, "minecproxy", true); if (mfd < 0) - goto out; + return false; sfd = open_subdir(mfd, "servers", true); if (sfd < 0) - goto out; + return false; if (!write_cfg_file(sfd, "README.TXT", ___examples_README_TXT, ___examples_README_TXT_len)) - goto out; + return false; cfd = open_subdir(mfd, "config", true); if (cfd < 0) - goto out; + return false; if (!write_cfg_file(cfd, "minecproxy.conf", ___examples_minecproxy_conf, ___examples_minecproxy_conf_len)) - goto out; + return false; - if (!create_server_cfg(sfd, cfd, "example", "example.mcserver")) - goto out; + if (!create_server_cfg(sfd, cfd, "example", "example.mcserver", + ___examples_server_properties, + ___examples_server_properties_len)) + return false; info("Created configuration in $XDG_CONFIG_HOME/minecproxy"); - rv = true; - -out: - if (mfd >= 0) - close(mfd); - if (sfd >= 0) - close(sfd); - if (cfd >= 0) - close(cfd); - return rv; + return true; } bool do_init(struct cfg *cfg) { - int xfd; - bool rv = false; + _cleanup_close_ int xfd = -1; if (cfg->dir_path) { xfd = open(cfg->dir_path, O_PATH | O_CLOEXEC | O_DIRECTORY); @@ -241,32 +221,27 @@ bool do_init(struct cfg *cfg) xfd = open_xdg_cfg_dir(true); if (xfd < 0) - goto out; + return false; if (!create_config_tree(xfd)) - goto out; + return false; if (cfg->dir_path) { info("Option -c used, not creating systemd services"); - rv = true; - goto out; + return true; } if (!create_user_service(xfd, "minecserver@.service", ___examples_minecserver__service, ___examples_minecserver__service_len)) - goto out; + return false; if (!create_user_service(xfd, "minecproxy.service", ___examples_minecproxy_service, ___examples_minecproxy_service_len)) - goto out; + return false; - rv = true; -out: - if (xfd >= 0) - close(xfd); - return rv; + return true; } static bool valid_name(const char *name) @@ -300,12 +275,204 @@ static bool valid_name(const char *name) return true; } +/* FIXME: Share this */ +static inline char tohex(uint8_t val) +{ + static const char hex[] = "0123456789abcdef"; + + return hex[val & 0x0f]; +} + +static bool generate_random_password(char *buf, size_t len) +{ + size_t written; + ssize_t r; + const char *s; + char *d; + + if (len < 3) + return false; + + char rnd[(len - 1) / 2]; + + for (written = 0; written < sizeof(rnd); written += r) { + r = getrandom(rnd + written, sizeof(rnd) - written, 0); + if (r < 0) { + if (errno == EAGAIN || errno == EINTR) + continue; + + error("getrandom failure: %m"); + return false; + } + } + + for (s = rnd, d = buf; s < (rnd + sizeof(rnd)); s++) { + *(d++) = tohex(*s >> 4); + *(d++) = tohex(*s); + } + *d = '\0'; + + info("Hex = %s", buf); + return true; +} + +static unsigned char *create_mc_properties(const char *name, uint16_t mc_port, + uint16_t rcon_port, + const char *rcon_password, + size_t *len) +{ + char hexrnd[16 + 1]; + char *properties; + int r; + + if (!rcon_password) { + if (!generate_random_password(hexrnd, sizeof(hexrnd))) { + error("Failed to generate random password"); + return NULL; + } + } + + /* FIXME: server-ip= */ + r = asprintf(&properties, + "#Minecraft server properties\n" + "# This is a partial file, it will be replaced with a\n" + "# fleshed out version the first time the Minecraft\n" + "# server is executed.\n" + "motd=%s\n" + "server-port=%" PRIu16 "\n" + "enable-rcon=true\n" + "rcon.port=%" PRIu16 "\n" + "rcon.password=%s", + name, mc_port, rcon_port, + rcon_password ? rcon_password : hexrnd); + if (r < 0) { + error("asprintf failed: %m"); + return NULL; + } + + if (len) + *len = r; + return (unsigned char *)properties; +} + +static uint16_t get_port(struct list_head *list) +{ + struct saddr *saddr; + + if (!list || list_empty(list)) + return 0; + + list_for_each_entry(saddr, list, list) { + switch (saddr->st.ss_family) { + case AF_INET: + if (saddr->in4.sin_addr.s_addr == htonl(INADDR_LOOPBACK) || + saddr->in4.sin_addr.s_addr == htonl(INADDR_ANY) || + saddr->in4.sin_addr.s_addr == htonl(INADDR_BROADCAST)) + return htons(saddr->in4.sin_port); + break; + case AF_INET6: + if (!memcmp(&saddr->in6.sin6_addr, &in6addr_any, + sizeof(saddr->in6.sin6_addr)) || + !memcmp(&saddr->in6.sin6_addr, &in6addr_loopback, + sizeof(saddr->in6.sin6_addr))) + return htons(saddr->in6.sin6_port); + break; + } + } + + return 0; +} + +static bool saddr_port_match(struct list_head *list, uint16_t port) +{ + struct saddr *a; + + list_for_each_entry(a, list, list) { + switch (a->st.ss_family) { + case AF_INET: + if (a->in4.sin_port == port) + return true; + break; + + case AF_INET6: + if (a->in6.sin6_port == port) + return true; + break; + + default: + break; + } + } + + return false; +} + +static bool select_free_ports(struct cfg *cfg, uint16_t *listen_port, + uint16_t *mc_port, uint16_t *rcon_port) +{ + uint16_t lport = cfg->listen_port_min; + uint16_t mport = cfg->mc_port_min; + uint16_t rport = cfg->rcon_port_min; + struct server *server; + bool used; + + for (lport = cfg->listen_port_min, + mport = cfg->mc_port_min, + rport = cfg->rcon_port_min; + lport <= cfg->listen_port_max && + mport <= cfg->mc_port_max && + rport <= cfg->rcon_port_max; + lport++, mport++, rport++) { + + used = false; + + list_for_each_entry(server, &cfg->servers, list) { + if (!server->scfg.filename) + continue; + + if (saddr_port_match(&server->scfg.locals, lport)) { + used = true; + break; + } + + if (saddr_port_match(&server->scfg.remotes, mport)) { + used = true; + break; + } + + if (saddr_port_match(&server->scfg.rcons, rport)) { + used = true; + break; + } + } + + if (!used ) { + if (listen_port && *listen_port == 0) + *listen_port = lport; + + if (mc_port && *mc_port == 0) + *mc_port = mport; + + if (rcon_port && *rcon_port == 0) + *rcon_port = rport; + + return true; + } + } + + return false; +} + bool do_new(struct cfg *cfg) { const char *name = cfg->commands[0]; struct server *server; - int sfd = -1; - bool rv = false; + struct server *defserver = NULL; + _cleanup_free_ unsigned char *properties = NULL; + size_t properties_len; + _cleanup_close_ int sfd = -1; + uint16_t rcon_port = 0, mc_port = 0; + const char *rcon_password = NULL; if (!valid_name(name)) return false; @@ -316,33 +483,52 @@ bool do_new(struct cfg *cfg) server_load_all_known(cfg); if (!cfg->dir) - goto out; + return false; list_for_each_entry(server, &cfg->servers, list) { - if (streq(server->name, name)) { + if (!server->scfg.filename) { + defserver = server; + continue; + } + + if (streq(server->scfg.filename, filename)) { error("Server \"%s\" already exists", name); - goto out; + return false; } } + if (defserver) { + /* FIXME: check list empty and tailor error msg */ + rcon_port = get_port(&defserver->scfg.rcons); + mc_port = get_port(&defserver->scfg.remotes); + rcon_password = defserver->scfg.rcon_password; + } else if (cfg->rcon_password) + rcon_password = cfg->rcon_password; + + if (!select_free_ports(cfg, NULL, &mc_port, &rcon_port)) { + error("Failed to find a free port"); + return false; + } + sfd = open_subdir(dirfd(cfg->dir), "../servers", false); if (sfd < 0) { error("Failed to open configuration directory \"servers\""); - goto out; + return false; } - if (!create_server_cfg(sfd, dirfd(cfg->dir), name, filename)) - goto out; - - info("Created server configuration %s", name); + properties = create_mc_properties(name, mc_port, rcon_port, + rcon_password, &properties_len); + if (!properties) + return false; - rv = true; + error("Created config file (%zu):\n%s", properties_len, properties); -out: - if (sfd >= 0) - close(sfd); + if (!create_server_cfg(sfd, dirfd(cfg->dir), name, filename, properties, + properties_len)) + return false; - return rv; + info("Created server configuration %s", name); + return true; } bool do_list(struct cfg *cfg) -- cgit v1.2.3