src/plaintcp.c

   1 /*
   2     ashd - A Sane HTTP Daemon
   3     Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
   4
   5     This program is free software: you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation, either version 3 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include <stdlib.h>
  20 #include <unistd.h>
  21 #include <stdio.h>
  22 #include <sys/socket.h>
  23 #include <netinet/in.h>
  24 #include <arpa/inet.h>
  25 #include <errno.h>
  26 #include <string.h>
  27
  28 #ifdef HAVE_CONFIG_H
  29 #include <config.h>
  30 #endif
  31 #include <utils.h>
  32 #include <req.h>
  33 #include <mt.h>
  34 #include <mtio.h>
  35 #include <log.h>
  36
  37 #include "htparser.h"
  38
  39 struct tcpport {
  40     int fd;
  41     int sport;
  42 };
  43
  44 struct tcpconn {
  45     struct sockaddr_storage name;
  46     struct tcpport *port;
  47     int fd;
  48 };
  49
  50 int listensock4(int port)
  51 {
  52     struct sockaddr_in name;
  53     int fd;
  54     int valbuf;
  55
  56     memset(&name, 0, sizeof(name));
  57     name.sin_family = AF_INET;
  58     name.sin_port = htons(port);
  59     if((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
  60         return(-1);
  61     valbuf = 1;
  62     setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &valbuf, sizeof(valbuf));
  63     if(bind(fd, (struct sockaddr *)&name, sizeof(name))) {
  64         close(fd);
  65         return(-1);
  66     }
  67     if(listen(fd, 16) < 0) {
  68         close(fd);
  69         return(-1);
  70     }
  71     return(fd);
  72 }
  73
  74 int listensock6(int port)
  75 {
  76     struct sockaddr_in6 name;
  77     int fd;
  78     int valbuf;
  79
  80     memset(&name, 0, sizeof(name));
  81     name.sin6_family = AF_INET6;
  82     name.sin6_port = htons(port);
  83     if((fd = socket(PF_INET6, SOCK_STREAM, 0)) < 0)
  84         return(-1);
  85     valbuf = 1;
  86     setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &valbuf, sizeof(valbuf));
  87     if(bind(fd, (struct sockaddr *)&name, sizeof(name))) {
  88         close(fd);
  89         return(-1);
  90     }
  91     if(listen(fd, 16) < 0) {
  92         close(fd);
  93         return(-1);
  94     }
  95     return(fd);
  96 }
  97
  98 char *formathaddress(struct sockaddr *name, socklen_t namelen)
  99 {
 100     static char buf[128];
 101     struct sockaddr_in *v4;
 102     struct sockaddr_in6 *v6;
 103
 104     switch(name->sa_family) {
 105     case AF_INET:
 106         v4 = (struct sockaddr_in *)name;
 107         if(!inet_ntop(AF_INET, &v4->sin_addr, buf, sizeof(buf)))
 108             return(NULL);
 109         return(buf);
 110     case AF_INET6:
 111         v6 = (struct sockaddr_in6 *)name;
 112         if(IN6_IS_ADDR_V4MAPPED(&v6->sin6_addr)) {
 113             if(!inet_ntop(AF_INET, ((char *)&v6->sin6_addr) + 12, buf, sizeof(buf)))
 114                 return(NULL);
 115         } else {
 116             if(!inet_ntop(AF_INET6, &v6->sin6_addr, buf, sizeof(buf)))
 117                 return(NULL);
 118         }
 119         return(buf);
 120     default:
 121         errno = EPFNOSUPPORT;
 122         return(NULL);
 123     }
 124 }
 125
 126 static int initreq(struct conn *conn, struct hthead *req)
 127 {
 128     struct tcpconn *tcp = conn->pdata;
 129     struct sockaddr_storage sa;
 130     socklen_t salen;
 131
 132     headappheader(req, "X-Ash-Address", formathaddress((struct sockaddr *)&tcp->name, sizeof(sa)));
 133     if(tcp->name.ss_family == AF_INET)
 134         headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in *)&tcp->name)->sin_port)));
 135     else if(tcp->name.ss_family == AF_INET6)
 136         headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in6 *)&tcp->name)->sin6_port)));
 137     salen = sizeof(sa);
 138     if(!getsockname(tcp->fd, (struct sockaddr *)&sa, &salen))
 139         headappheader(req, "X-Ash-Server-Address", formathaddress((struct sockaddr *)&sa, sizeof(sa)));
 140     headappheader(req, "X-Ash-Server-Port", sprintf3("%i", tcp->port->sport));
 141     headappheader(req, "X-Ash-Protocol", "http");
 142     return(0);
 143 }
 144
 145 void servetcp(struct muth *muth, va_list args)
 146 {
 147     vavar(int, fd);
 148     vavar(struct sockaddr_storage, name);
 149     vavar(struct tcpport *, stcp);
 150     FILE *in;
 151     struct conn conn;
 152     struct tcpconn tcp;
 153
 154     memset(&conn, 0, sizeof(conn));
 155     memset(&tcp, 0, sizeof(tcp));
 156     in = mtstdopen(fd, 1, 60, "r+");
 157     conn.pdata = &tcp;
 158     conn.initreq = initreq;
 159     tcp.fd = fd;
 160     tcp.name = name;
 161     tcp.port = stcp;
 162     serve(in, &conn);
 163 }
 164
 165 static void listenloop(struct muth *muth, va_list args)
 166 {
 167     vavar(struct tcpport *, tcp);
 168     int ns;
 169     struct sockaddr_storage name;
 170     socklen_t namelen;
 171
 172     while(1) {
 173         namelen = sizeof(name);
 174         block(tcp->fd, EV_READ, 0);
 175         ns = accept(tcp->fd, (struct sockaddr *)&name, &namelen);
 176         if(ns < 0) {
 177             flog(LOG_ERR, "accept: %s", strerror(errno));
 178             goto out;
 179         }
 180         mustart(servetcp, ns, name, tcp);
 181     }
 182
 183 out:
 184     close(tcp->fd);
 185     free(tcp);
 186 }
 187
 188 void handleplain(int argc, char **argp, char **argv)
 189 {
 190     int port, fd;
 191     int i;
 192     struct tcpport *tcp;
 193
 194     port = 80;
 195     for(i = 0; i < argc; i++) {
 196         if(!strcmp(argp[i], "help")) {
 197             printf("plain handler parameters:\n");
 198             printf("\tport=TCP-PORT   [80]\n");
 199             printf("\t\tThe TCP port to listen on.\n");
 200             exit(0);
 201         } else if(!strcmp(argp[i], "port")) {
 202             port = atoi(argv[i]);
 203         } else {
 204             flog(LOG_ERR, "unknown parameter `%s' to plain handler", argp[i]);
 205             exit(1);
 206         }
 207     }
 208     if((fd = listensock6(port)) < 0) {
 209         flog(LOG_ERR, "could not listen on IPv6 (port %i): %s", port, strerror(errno));
 210         exit(1);
 211     }
 212     omalloc(tcp);
 213     tcp->fd = fd;
 214     tcp->sport = port;
 215     mustart(listenloop, tcp);
 216     if((fd = listensock4(port)) < 0) {
 217         if(errno != EADDRINUSE) {
 218             flog(LOG_ERR, "could not listen on IPv4 (port %i): %s", port, strerror(errno));
 219             exit(1);
 220         }
 221     } else {
 222         omalloc(tcp);
 223         tcp->fd = fd;
 224         tcp->sport = port;
 225         mustart(listenloop, tcp);
 226     }
 227 }