util.c

   1 /* Misc util functions */
   2
   3 char const *cvs_id_util = "$Id: util.c,v 1.7 2004-11-29 02:17:18 bodea Exp $";
   4
   5 #include <unistd.h>
   6 #include <errno.h>
   7 #include <sched.h>
   8 #include <sys/socket.h>
   9 #include <netinet/in.h>
  10 #include <arpa/inet.h>
  11 #include <string.h>
  12 #include <sys/mman.h>
  13
  14 #include "l2tpns.h"
  15 #ifdef BGP
  16 #include "bgp.h"
  17 #endif
  18
  19 // format ipv4 addr as a dotted-quad; n chooses one of 4 static buffers
  20 // to use
  21 char *fmtaddr(ipt addr, int n)
  22 {
  23         static char addrs[4][16];
  24         struct in_addr in;
  25
  26         if (n < 0 || n >= 4) return "";
  27         in.s_addr = addr;
  28         return strcpy(addrs[n], inet_ntoa(in));
  29 }
  30
  31 void *shared_malloc(unsigned int size)
  32 {
  33         void * p;
  34         p = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANONYMOUS, 0, 0);
  35
  36         if (p == MAP_FAILED)
  37                 p = NULL;
  38
  39         return p;
  40 }
  41
  42 extern int udpfd, controlfd, tunfd, snoopfd, ifrfd, cluster_sockfd;
  43 extern int *radfds;
  44
  45 pid_t fork_and_close()
  46 {
  47         pid_t pid = fork();
  48         int i;
  49
  50         if (pid)
  51                 return pid;
  52
  53         if (config->scheduler_fifo)
  54         {
  55                 struct sched_param params = {0};
  56                 params.sched_priority = 0;
  57                 if (sched_setscheduler(0, SCHED_OTHER, &params))
  58                 {
  59                         LOG(0, 0, 0, "Error setting scheduler to OTHER after fork: %s\n", strerror(errno));
  60                         LOG(0, 0, 0, "This is probably really really bad.\n");
  61                 }
  62         }
  63
  64         signal(SIGPIPE, SIG_DFL);
  65         signal(SIGCHLD, SIG_DFL);
  66         signal(SIGHUP, SIG_DFL);
  67         signal(SIGUSR1, SIG_DFL);
  68         signal(SIGQUIT, SIG_DFL);
  69         signal(SIGKILL, SIG_DFL);
  70         signal(SIGALRM, SIG_DFL);
  71         signal(SIGTERM, SIG_DFL);
  72
  73         // Close sockets
  74         if (tunfd != -1)          close(tunfd);
  75         if (udpfd != -1)          close(udpfd);
  76         if (controlfd != -1)      close(controlfd);
  77         if (snoopfd != -1)        close(snoopfd);
  78         if (ifrfd != -1)          close(ifrfd);
  79         if (cluster_sockfd != -1) close(cluster_sockfd);
  80         if (clifd != -1)          close(clifd);
  81
  82         for (i = 0; radfds && i < config->num_radfds; i++)
  83                 close(radfds[i]);
  84 #ifdef BGP
  85         for (i = 0; i < BGP_NUM_PEERS; i++)
  86                 if (bgp_peers[i].sock != -1)
  87                         close(bgp_peers[i].sock);
  88 #endif /* BGP */
  89
  90         return pid;
  91 }