src/rxpd_socket.c

   1 /*
   2     rxpd_socket.c - regex policy daemon
   3
   4   Copyright (C)
   5     2007,               Christian Thaeter <ct@pipapo.org>
   6
   7   This program is free software; you can redistribute it and/or
   8   modify it under the terms of the GNU General Public License as
   9   published by the Free Software Foundation; either version 2 of the
  10   License, or (at your option) any later version.
  11
  12   This program is distributed in the hope that it will be useful,
  13   but WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15   GNU General Public License for more details.
  16
  17   You should have received a copy of the GNU General Public License
  18   along with this program; if not, write to the Free Software
  19   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "rxpd.h"
  23
  24 struct rxpd_socket*
  25 rxpd_socket_new_tcp4 (struct rxpd_base* base, const char* addr, unsigned short port)
  26 {
  27   struct rxpd_socket* self;
  28   self = rxpd_malloc (sizeof (struct rxpd_socket));
  29
  30   self->base = base;
  31
  32   llist_init (&self->node);
  33
  34   self->fd = socket (PF_INET, SOCK_STREAM, 0);
  35   if (self->fd == -1)
  36     goto esocket;
  37
  38   struct sockaddr_in listen_addr;
  39   memset (&listen_addr, 0, sizeof (listen_addr));
  40
  41   listen_addr.sin_family = AF_INET;
  42   if (addr)
  43     {
  44       if (inet_aton (addr, &listen_addr.sin_addr) == 0)
  45         goto esocket;
  46     }
  47   else
  48     listen_addr.sin_addr.s_addr = INADDR_ANY;
  49
  50   listen_addr.sin_port = htons(port);
  51
  52   if (bind (self->fd, (struct sockaddr*)&listen_addr, sizeof (listen_addr)) == -1)
  53     goto esocket;
  54
  55   static int yes = 1;
  56   if (setsockopt (self->fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1)
  57     goto esocket;
  58
  59   if (listen (self->fd, 20) == -1)
  60     goto esocket;
  61
  62   self->rxpd_socket_addr = rxpd_socket_tcp4addr;
  63
  64   self->accepter = NULL;
  65   llist_insert_tail (&base->sockets, &self->node);
  66
  67   rxpd_log (base, LOG_INFO, "Listening on tcp4:%d\n", port);
  68   return self;
  69
  70  esocket:
  71   rxpd_die ("failed creating socket\n");
  72   /* free (self); rxpd_die never returns */
  73   return NULL;
  74 }
  75
  76 int
  77 rxpd_socket_tcp4addr (struct rxpd_connection* conn, char* dst, const char* pfx, size_t size)
  78 {
  79   struct sockaddr_in peer;
  80   socklen_t len = sizeof (peer);
  81   getpeername (conn->fd, (struct sockaddr*)&peer, &len);
  82
  83   char* addr;
  84   addr = inet_ntoa (peer.sin_addr);
  85   if (sizeof (":tcp4:") + strlen (pfx) + strlen (addr) > size)
  86     return 0;
  87
  88   strcat (dst, pfx);
  89   strcat (dst, ":tcp4:");
  90   strcat (dst, addr);
  91   return 1;
  92 }
  93
  94 void
  95 rxpd_socket_delete (struct rxpd_socket* self)
  96 {
  97   if (self)
  98     {
  99       llist_unlink (&self->node);
 100       close (self->fd);
 101     }
 102   free (self);
 103 }
 104
 105 struct rxpd_socket*
 106 rxpd_socket_join (struct rxpd_socket* self)
 107 {
 108   pth_join (self->accepter, NULL);
 109   self->accepter = NULL;
 110   return self;
 111 }
 112
 113 struct rxpd_socket*
 114 rxpd_socket_spawn (struct rxpd_socket* self)
 115 {
 116   if (self)
 117     {
 118       if (self->accepter)
 119         rxpd_die ("socket thread already spawned\n");
 120
 121       self->accepter = pth_spawn (PTH_ATTR_DEFAULT, rxpd_socket_accept, self);
 122
 123       if (!self->accepter)
 124         rxpd_die ("failed spawning thread\n");
 125     }
 126   return self;
 127 }
 128
 129 struct rxpd_socket*
 130 rxpd_socket_cancel (struct rxpd_socket* self)
 131 {
 132   pth_cancel (self->accepter);
 133   return self;
 134 }
 135
 136 void *
 137 rxpd_socket_accept (void* ptr)
 138 {
 139   struct rxpd_socket* self = ptr;
 140
 141   pth_event_t ev = pth_event (PTH_EVENT_FD|PTH_UNTIL_FD_READABLE, self->fd);
 142
 143   while (pth_wait (ev))
 144     {
 145       int fd = pth_accept (self->fd, NULL, 0);
 146       if (fd == -1)
 147         rxpd_die ("error accepting connection\n");
 148
 149       struct rxpd_connection* conn = rxpd_connection_new (self, fd);
 150
 151       char buf[512];
 152
 153       if (rxpd_socket_tcp4addr (conn, buf, "", 511))
 154         {
 155           rxpd_log (self->base, LOG_INFO, "incoming connection '%s'\n", buf);
 156           rxpd_connection_spawn (conn);
 157         }
 158       else
 159         rxpd_log (self->base, LOG_ERR, "illegal hostname\n", buf);
 160     }
 161
 162   rxpd_log (NULL, LOG_NOTICE, "closed\n");
 163   return NULL;
 164 }
 165