|
- /* See the LICENSE file for licensing information */
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <sys/socket.h>
-
- #include <openssl/ssl.h>
- #include <openssl/err.h>
-
- #include <errno.h>
- #include <fcntl.h>
- #include <netdb.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <time.h>
- #include <unistd.h>
-
- #define PING_TIMEOUT 240
-
- static SSL_CTX *ctx;
- static SSL *ssl;
- static char *host = "chat.freenode.net";
- static int port = 6667;
- static FILE *log;
- static short use_ssl = 0;
- static unsigned int last_response;
-
- static FILE *
- slog(char *file)
- {
- const char *home = getenv("HOME");
- char path[100];
- FILE *fp;
-
- snprintf(path, 100, "%s/%s", home, file);
- mknod(path, 0 | 0666, 0);
- fp = fopen(path, "w");
-
- return fp;
- }
-
- static int
- dial(char *host, int port)
- {
- int sockfd, err;
- struct addrinfo hints, *serv;
-
- char tmp[8];
- snprintf(tmp, 8, "%d", port);
-
- memset(&hints, 0, sizeof(hints));
- hints.ai_family = AF_INET;
- hints.ai_socktype = SOCK_STREAM;
- if ((err = getaddrinfo(host, tmp, &hints, &serv)) != 0) {
- fprintf(log, "getaddrinfo: %s\n", gai_strerror(err));
- return -1;
- } else if ((sockfd = socket(serv->ai_family, serv->ai_socktype,
- serv->ai_protocol)) == -1) {
- fprintf(log, "tubes: error on socket()\n");
- return -1;
- } else if (connect(sockfd, serv->ai_addr, serv->ai_addrlen) == -1) {
- fprintf(log, "tubes: error on connect().\n");
- close(sockfd);
- return -1;
- }
- freeaddrinfo(serv);
-
- return sockfd;
- }
-
- static int
- sslify(int *sockfd)
- {
- int r;
-
- SSL_library_init();
- SSL_load_error_strings();
-
- ctx = SSL_CTX_new(SSLv23_client_method());
- SSL_CTX_set_options(ctx, SSL_OP_NO_SSLv2 | SSL_OP_NO_SSLv3
- | SSL_OP_SINGLE_DH_USE);
- ssl = SSL_new(ctx);
- SSL_set_fd(ssl, *sockfd);
- SSL_set_connect_state(ssl);
-
- if ((r = SSL_connect(ssl)) < 1) {
- fprintf(log, "sslify: %s\n", strerror(SSL_get_error(ssl, r)));
- SSL_CTX_free(ctx);
- return -1;
- }
-
- return 0;
- }
-
- int
- main(int argc, char **argv)
- {
- int sockfd, in, out;
- int maxfd;
- fd_set rd;
- struct timeval tv;
- char buf[512];
- int i, r, status;
-
- for (i = 1; i < argc; i++) {
- r = argv[i][1];
- if (argv[i][0] != '-' || argv[i][2])
- r = -1;
- switch (r) {
- case 'S':
- use_ssl = 1;
- port = 6697;
- break;
- case 'h':
- if (++i < argc)
- host = argv[i];
- break;
- case 'p':
- if (++i < argc)
- port = atoi(argv[i]);
- break;
- default:
- fprintf(stderr,
- "usage: tubes [-S] [-h host] [-p port]\n");
- exit(0);
- }
- }
-
- if ((log = slog(".tubes.err")) == NULL) {
- fprintf(stderr, "error on slog()");
- exit(-1);
- }
- if (daemon(0, 0) == -1) {
- fprintf(log, "error on daemon()\n");
- exit(-1);
- }
- if ((sockfd = dial(host, port)) == -1)
- exit(-1);
- if (use_ssl && sslify(&sockfd) == -1)
- exit(-1);
-
- snprintf(buf, 512, "/tmp/%s.in", host);
- unlink(buf);
- mkfifo(buf, 0660);
- if ((in = open(buf, O_RDWR | O_NONBLOCK)) < 0) {
- fprintf(log, "in: error on open()\n");
- exit(-1);
- }
- snprintf(buf, 512, "/tmp/%s.out", host);
- unlink(buf);
- mkfifo(buf, 0660);
- if ((out = open(buf, O_RDWR)) < 0) {
- fprintf(log, "out: error on open()\n");
- exit(-1);
- }
- for (status = 0;;) {
- FD_ZERO(&rd);
- maxfd = (out >= sockfd) ? out : sockfd;
- FD_SET(out, &rd);
- FD_SET(sockfd, &rd);
-
- tv.tv_sec = 10;
- tv.tv_usec = 0;
- r = select(maxfd+1, &rd, 0, 0, &tv);
- if (r < 0) {
- if (errno == EINTR)
- continue;
- fprintf(log, "error on select()\n");
- status = -1;
- break;
- } else if (r == 0 && last_response-time(NULL) >= PING_TIMEOUT) {
- fprintf(log, "ping timeout\n");
- status = -1;
- break;
- }
- if (FD_ISSET(out, &rd))
- if ((i = read(out, buf, sizeof(buf))) > 0) {
- buf[i] = 0;
- if (use_ssl)
- SSL_write(ssl, buf, strlen(buf));
- else
- send(sockfd, buf, strlen(buf), 0);
- }
- if (FD_ISSET(sockfd, &rd)) {
- if (use_ssl) {
- do {
- r = 0;
- i = SSL_read(ssl, buf, sizeof(buf));
- if (SSL_get_error(ssl, i)
- == SSL_ERROR_WANT_READ)
- r = 1;
- } while (SSL_pending(ssl) && !r);
- } else
- i = recv(sockfd, buf, sizeof(buf), 0);
- if (i != -1) {
- if (i == 0) {
- fprintf(log, "connection closed\n");
- break;
- }
- buf[i] = 0;
- if (write(in, buf, strlen(buf)) < 0) {
- if (errno == EINTR)
- continue;
- fprintf(log, "error on write()\n");
- status = -1;
- break;
- }
- last_response = time(NULL);
- }
- }
- }
-
- close(sockfd);
- close(in);
- close(out);
- fclose(log);
-
- ERR_free_strings();
- EVP_cleanup();
- SSL_shutdown(ssl);
- SSL_free(ssl);
- SSL_CTX_free(ctx);
-
- snprintf(buf, 512, "/tmp/%s.in", host);
- unlink(buf);
- snprintf(buf, 512, "/tmp/%s.out", host);
- unlink(buf);
- exit(status);
- }
|