udp.c

   1 /*
   2  * Copyright (C) 2006-2008 Stig Venaas <venaas@uninett.no>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  */
   8
   9 #include <signal.h>
  10 #include <sys/socket.h>
  11 #include <netinet/in.h>
  12 #include <netdb.h>
  13 #include <string.h>
  14 #include <unistd.h>
  15 #include <limits.h>
  16 #ifdef SYS_SOLARIS9
  17 #include <fcntl.h>
  18 #endif
  19 #include <sys/time.h>
  20 #include <sys/types.h>
  21 #include <sys/select.h>
  22 #include <ctype.h>
  23 #include <sys/wait.h>
  24 #include <arpa/inet.h>
  25 #include <regex.h>
  26 #include <pthread.h>
  27 #include "list.h"
  28 #include "radsecproxy.h"
  29
  30 #ifdef RADPROT_UDP
  31 #include "debug.h"
  32 #include "util.h"
  33
  34 static void setprotoopts(struct commonprotoopts *opts);
  35 static char **getlistenerargs();
  36 void *udpserverrd(void *arg);
  37 int clientradputudp(struct server *server, unsigned char *rad);
  38 void addclientudp(struct client *client);
  39 void addserverextraudp(struct clsrvconf *conf);
  40 void udpsetsrcres();
  41 void initextraudp();
  42
  43 static const struct protodefs protodefs = {
  44     "udp",
  45     NULL, /* secretdefault */
  46     SOCK_DGRAM, /* socktype */
  47     "1812", /* portdefault */
  48     REQUEST_RETRY_COUNT, /* retrycountdefault */
  49     10, /* retrycountmax */
  50     REQUEST_RETRY_INTERVAL, /* retryintervaldefault */
  51     60, /* retryintervalmax */
  52     DUPLICATE_INTERVAL, /* duplicateintervaldefault */
  53     setprotoopts, /* setprotoopts */
  54     getlistenerargs, /* getlistenerargs */
  55     udpserverrd, /* listener */
  56     NULL, /* connecter */
  57     NULL, /* clientconnreader */
  58     clientradputudp, /* clientradput */
  59     addclientudp, /* addclient */
  60     addserverextraudp, /* addserverextra */
  61     udpsetsrcres, /* setsrcres */
  62     initextraudp /* initextra */
  63 };
  64
  65 static int client4_sock = -1;
  66 static int client6_sock = -1;
  67 static struct gqueue *server_replyq = NULL;
  68
  69 static struct addrinfo *srcres = NULL;
  70 static uint8_t handle;
  71 static struct commonprotoopts *protoopts = NULL;
  72
  73 const struct protodefs *udpinit(uint8_t h) {
  74     handle = h;
  75     return &protodefs;
  76 }
  77
  78 static void setprotoopts(struct commonprotoopts *opts) {
  79     protoopts = opts;
  80 }
  81
  82 static char **getlistenerargs() {
  83     return protoopts ? protoopts->listenargs : NULL;
  84 }
  85
  86 void udpsetsrcres() {
  87     if (!srcres)
  88         srcres = resolve_hostport_addrinfo(handle, protoopts ? protoopts->sourcearg : NULL);
  89 }
  90
  91 void removeudpclientfromreplyq(struct client *c) {
  92     struct list_node *n;
  93     struct request *r;
  94
  95     /* lock the common queue and remove replies for this client */
  96     pthread_mutex_lock(&c->replyq->mutex);
  97     for (n = list_first(c->replyq->entries); n; n = list_next(n)) {
  98         r = (struct request *)n->data;
  99         if (r->from == c)
 100             r->from = NULL;
 101     }
 102     pthread_mutex_unlock(&c->replyq->mutex);
 103 }
 104
 105 static int addr_equal(struct sockaddr *a, struct sockaddr *b) {
 106     switch (a->sa_family) {
 107     case AF_INET:
 108         return !memcmp(&((struct sockaddr_in*)a)->sin_addr,
 109                        &((struct sockaddr_in*)b)->sin_addr,
 110                        sizeof(struct in_addr));
 111     case AF_INET6:
 112         return IN6_ARE_ADDR_EQUAL(&((struct sockaddr_in6*)a)->sin6_addr,
 113                                   &((struct sockaddr_in6*)b)->sin6_addr);
 114     default:
 115         /* Must not reach */
 116         return 0;
 117     }
 118 }
 119
 120 uint16_t port_get(struct sockaddr *sa) {
 121     switch (sa->sa_family) {
 122     case AF_INET:
 123         return ntohs(((struct sockaddr_in *)sa)->sin_port);
 124     case AF_INET6:
 125         return ntohs(((struct sockaddr_in6 *)sa)->sin6_port);
 126     }
 127     return 0;
 128 }
 129
 130 /* exactly one of client and server must be non-NULL */
 131 /* return who we received from in *client or *server */
 132 /* return from in sa if not NULL */
 133 unsigned char *radudpget(int s, struct client **client, struct server **server, uint16_t *port) {
 134     int cnt, len;
 135     unsigned char buf[4], *rad = NULL;
 136     struct sockaddr_storage from;
 137     struct sockaddr *fromcopy;
 138     socklen_t fromlen = sizeof(from);
 139     struct clsrvconf *p;
 140     struct list_node *node;
 141     fd_set readfds;
 142     struct client *c = NULL;
 143     struct timeval now;
 144
 145     for (;;) {
 146         if (rad) {
 147             free(rad);
 148             rad = NULL;
 149         }
 150         FD_ZERO(&readfds);
 151         FD_SET(s, &readfds);
 152         if (select(s + 1, &readfds, NULL, NULL, NULL) < 1)
 153             continue;
 154         cnt = recvfrom(s, buf, 4, MSG_PEEK | MSG_TRUNC, (struct sockaddr *)&from, &fromlen);
 155         if (cnt == -1) {
 156             debug(DBG_WARN, "radudpget: recv failed");
 157             continue;
 158         }
 159
 160         p = client
 161             ? find_clconf(handle, (struct sockaddr *)&from, NULL)
 162             : find_srvconf(handle, (struct sockaddr *)&from, NULL);
 163         if (!p) {
 164             debug(DBG_WARN, "radudpget: got packet from wrong or unknown UDP peer %s, ignoring", addr2string((struct sockaddr *)&from));
 165             recv(s, buf, 4, 0);
 166             continue;
 167         }
 168
 169         len = RADLEN(buf);
 170         if (len < 20) {
 171             debug(DBG_WARN, "radudpget: length too small");
 172             recv(s, buf, 4, 0);
 173             continue;
 174         }
 175
 176         rad = malloc(len);
 177         if (!rad) {
 178             debug(DBG_ERR, "radudpget: malloc failed");
 179             recv(s, buf, 4, 0);
 180             continue;
 181         }
 182
 183         cnt = recv(s, rad, len, MSG_TRUNC);
 184         debug(DBG_DBG, "radudpget: got %d bytes from %s", cnt, addr2string((struct sockaddr *)&from));
 185
 186         if (cnt < len) {
 187             debug(DBG_WARN, "radudpget: packet smaller than length field in radius header");
 188             continue;
 189         }
 190         if (cnt > len)
 191             debug(DBG_DBG, "radudpget: packet was padded with %d bytes", cnt - len);
 192
 193         if (client) {
 194             *client = NULL;
 195             pthread_mutex_lock(p->lock);
 196             for (node = list_first(p->clients); node;) {
 197                 c = (struct client *)node->data;
 198                 node = list_next(node);
 199                 if (s != c->sock)
 200                     continue;
 201                 gettimeofday(&now, NULL);
 202                 if (!*client && addr_equal((struct sockaddr *)&from, c->addr)) {
 203                     c->expiry = now.tv_sec + 60;
 204                     *client = c;
 205                 }
 206                 if (c->expiry >= now.tv_sec)
 207                     continue;
 208
 209                 debug(DBG_DBG, "radudpget: removing expired client (%s)", addr2string(c->addr));
 210                 removeudpclientfromreplyq(c);
 211                 c->replyq = NULL; /* stop removeclient() from removing common udp replyq */
 212                 removelockedclient(c);
 213                 break;
 214             }
 215             if (!*client) {
 216                 fromcopy = addr_copy((struct sockaddr *)&from);
 217                 if (!fromcopy) {
 218                     pthread_mutex_unlock(p->lock);
 219                     continue;
 220                 }
 221                 c = addclient(p, 0);
 222                 if (!c) {
 223                     free(fromcopy);
 224                     pthread_mutex_unlock(p->lock);
 225                     continue;
 226                 }
 227                 c->sock = s;
 228                 c->addr = fromcopy;
 229                 gettimeofday(&now, NULL);
 230                 c->expiry = now.tv_sec + 60;
 231                 *client = c;
 232             }
 233             pthread_mutex_unlock(p->lock);
 234         } else if (server)
 235             *server = p->servers;
 236         break;
 237     }
 238     if (port)
 239         *port = port_get((struct sockaddr *)&from);
 240     return rad;
 241 }
 242
 243 int clientradputudp(struct server *server, unsigned char *rad) {
 244     size_t len;
 245     struct clsrvconf *conf = server->conf;
 246
 247     len = RADLEN(rad);
 248     if (sendto(server->sock, rad, len, 0, conf->addrinfo->ai_addr, conf->addrinfo->ai_addrlen) >= 0) {
 249         debug(DBG_DBG, "clienradputudp: sent UDP of length %d to %s port %d", len, conf->host, port_get(conf->addrinfo->ai_addr));
 250         return 1;
 251     }
 252
 253     debug(DBG_WARN, "clientradputudp: send failed");
 254     return 0;
 255 }
 256
 257 void *udpclientrd(void *arg) {
 258     struct server *server;
 259     unsigned char *buf;
 260     int *s = (int *)arg;
 261
 262     for (;;) {
 263         server = NULL;
 264         buf = radudpget(*s, NULL, &server, NULL);
 265         replyh(server, buf);
 266     }
 267 }
 268
 269 void *udpserverrd(void *arg) {
 270     struct request *rq;
 271     int *sp = (int *)arg;
 272
 273     for (;;) {
 274         rq = newrequest();
 275         if (!rq) {
 276             sleep(5); /* malloc failed */
 277             continue;
 278         }
 279         rq->buf = radudpget(*sp, &rq->from, NULL, &rq->udpport);
 280         rq->udpsock = *sp;
 281         radsrv(rq);
 282     }
 283     free(sp);
 284     return NULL;
 285 }
 286
 287 void *udpserverwr(void *arg) {
 288     struct gqueue *replyq = (struct gqueue *)arg;
 289     struct request *reply;
 290     struct sockaddr_storage to;
 291
 292     for (;;) {
 293         pthread_mutex_lock(&replyq->mutex);
 294         while (!(reply = (struct request *)list_shift(replyq->entries))) {
 295             debug(DBG_DBG, "udp server writer, waiting for signal");
 296             pthread_cond_wait(&replyq->cond, &replyq->mutex);
 297             debug(DBG_DBG, "udp server writer, got signal");
 298         }
 299         /* do this with lock, udpserverrd may set from = NULL if from expires */
 300         if (reply->from)
 301             memcpy(&to, reply->from->addr, SOCKADDRP_SIZE(reply->from->addr));
 302         pthread_mutex_unlock(&replyq->mutex);
 303         if (reply->from) {
 304             port_set((struct sockaddr *)&to, reply->udpport);
 305             if (sendto(reply->udpsock, reply->replybuf, RADLEN(reply->replybuf), 0, (struct sockaddr *)&to, SOCKADDR_SIZE(to)) < 0)
 306                 debug(DBG_WARN, "udpserverwr: send failed");
 307         }
 308         debug(DBG_DBG, "udpserverwr: refcount %d", reply->refcount);
 309         freerq(reply);
 310     }
 311 }
 312
 313 void addclientudp(struct client *client) {
 314     client->replyq = server_replyq;
 315 }
 316
 317 void addserverextraudp(struct clsrvconf *conf) {
 318     switch (conf->addrinfo->ai_family) {
 319     case AF_INET:
 320         if (client4_sock < 0) {
 321             client4_sock = bindtoaddr(srcres, AF_INET, 0, 1);
 322             if (client4_sock < 0)
 323                 debugx(1, DBG_ERR, "addserver: failed to create client socket for server %s", conf->host);
 324         }
 325         conf->servers->sock = client4_sock;
 326         break;
 327     case AF_INET6:
 328         if (client6_sock < 0) {
 329             client6_sock = bindtoaddr(srcres, AF_INET6, 0, 1);
 330             if (client6_sock < 0)
 331                 debugx(1, DBG_ERR, "addserver: failed to create client socket for server %s", conf->host);
 332         }
 333         conf->servers->sock = client6_sock;
 334         break;
 335     default:
 336         debugx(1, DBG_ERR, "addserver: unsupported address family");
 337     }
 338 }
 339
 340 void initextraudp() {
 341     pthread_t cl4th, cl6th, srvth;
 342
 343     if (srcres) {
 344         freeaddrinfo(srcres);
 345         srcres = NULL;
 346     }
 347
 348     if (client4_sock >= 0)
 349         if (pthread_create(&cl4th, NULL, udpclientrd, (void *)&client4_sock))
 350             debugx(1, DBG_ERR, "pthread_create failed");
 351     if (client6_sock >= 0)
 352         if (pthread_create(&cl6th, NULL, udpclientrd, (void *)&client6_sock))
 353             debugx(1, DBG_ERR, "pthread_create failed");
 354
 355     if (find_clconf_type(handle, NULL)) {
 356         server_replyq = newqueue();
 357         if (pthread_create(&srvth, NULL, udpserverwr, (void *)server_replyq))
 358             debugx(1, DBG_ERR, "pthread_create failed");
 359     }
 360 }
 361 #else
 362 const struct protodefs *udpinit(uint8_t h) {
 363     return NULL;
 364 }
 365 #endif