From ccdeb0c556968bafb9853cdb430c1b87ac22fc67 Mon Sep 17 00:00:00 2001 From: Curtis McEnroe Date: Sat, 4 Aug 2018 13:35:29 -0400 Subject: Split source into several files Can't believe I have to do this. --- client.c | 129 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 129 insertions(+) create mode 100644 client.c (limited to 'client.c') diff --git a/client.c b/client.c new file mode 100644 index 0000000..64f14cf --- /dev/null +++ b/client.c @@ -0,0 +1,129 @@ +/* Copyright (C) 2018 Curtis McEnroe + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "chat.h" + +static struct tls *client; + +static void webirc(const char *pass) { + const char *ssh = getenv("SSH_CLIENT"); + if (!ssh) return; + int len = strlen(ssh); + const char *sp = strchr(ssh, ' '); + if (sp) len = sp - ssh; + clientFmt( + "WEBIRC %s %s %.*s %.*s\r\n", + pass, chat.user, len, ssh, len, ssh + ); +} + +int clientConnect(const char *host, const char *port, const char *webPass) { + int error; + + struct tls_config *config = tls_config_new(); + error = tls_config_set_ciphers(config, "compat"); + if (error) errx(EX_SOFTWARE, "tls_config: %s", tls_config_error(config)); + + client = tls_client(); + if (!client) errx(EX_SOFTWARE, "tls_client"); + + error = tls_configure(client, config); + if (error) errx(EX_SOFTWARE, "tls_configure"); + tls_config_free(config); + + struct addrinfo *ai; + struct addrinfo hints = { + .ai_family = AF_UNSPEC, + .ai_socktype = SOCK_STREAM, + .ai_protocol = IPPROTO_TCP, + }; + error = getaddrinfo(host, port, &hints, &ai); + if (error) errx(EX_NOHOST, "getaddrinfo: %s", gai_strerror(error)); + + int sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); + if (sock < 0) err(EX_OSERR, "socket"); + + error = connect(sock, ai->ai_addr, ai->ai_addrlen); + if (error) err(EX_UNAVAILABLE, "connect"); + freeaddrinfo(ai); + + error = tls_connect_socket(client, sock, host); + if (error) err(EX_PROTOCOL, "tls_connect"); + + if (webPass) webirc(webPass); + clientFmt("NICK %s\r\n", chat.nick); + clientFmt("USER %s 0 * :%s\r\n", chat.user, chat.nick); + + return sock; +} + +void clientWrite(const char *ptr, size_t len) { + while (len) { + ssize_t ret = tls_write(client, ptr, len); + if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) continue; + if (ret < 0) errx(EX_IOERR, "tls_write: %s", tls_error(client)); + ptr += ret; + len -= ret; + } +} + +void clientFmt(const char *format, ...) { + char *buf; + va_list ap; + va_start(ap, format); + int len = vasprintf(&buf, format, ap); + va_end(ap); + if (!buf) err(EX_OSERR, "vasprintf"); + if (chat.verbose) uiFmt("<<< %.*s", len - 2, buf); + clientWrite(buf, len); + free(buf); +} + +void clientRead(void) { + static char buf[4096]; + static size_t len; + + ssize_t read = tls_read(client, &buf[len], sizeof(buf) - len); + if (read < 0) errx(EX_IOERR, "tls_read: %s", tls_error(client)); + if (!read) { + uiHide(); + exit(EX_OK); + } + len += read; + + char *crlf, *line = buf; + while ((crlf = strnstr(line, "\r\n", &buf[len] - line))) { + crlf[0] = '\0'; + if (chat.verbose) uiFmt(">>> %s", line); + handle(line); + line = &crlf[2]; + } + + len -= line - buf; + memmove(buf, line, len); +} -- cgit 1.4.1-2-gfad0