src/recv-announce.cc

   1 /*
   2  * DEBUG: section 00    Announcement Server
   3  * AUTHOR: Harvest Derived
   4  *
   5  * SQUID Web Proxy Cache          http://www.squid-cache.org/
   6  * ----------------------------------------------------------
   7  *
   8  *  Squid is the result of efforts by numerous individuals from
   9  *  the Internet community; see the CONTRIBUTORS file for full
  10  *  details.   Many organizations have provided support for Squid's
  11  *  development; see the SPONSORS file for full details.  Squid is
  12  *  Copyrighted (C) 2001 by the Regents of the University of
  13  *  California; see the COPYRIGHT file for full details.  Squid
  14  *  incorporates software developed and/or copyrighted by other
  15  *  sources; see the CREDITS file for full details.
  16  *
  17  *  This program is free software; you can redistribute it and/or modify
  18  *  it under the terms of the GNU General Public License as published by
  19  *  the Free Software Foundation; either version 2 of the License, or
  20  *  (at your option) any later version.
  21  *
  22  *  This program is distributed in the hope that it will be useful,
  23  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  24  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  25  *  GNU General Public License for more details.
  26  *
  27  *  You should have received a copy of the GNU General Public License
  28  *  along with this program; if not, write to the Free Software
  29  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  30  *
  31  */
  32 #include "squid.h"
  33
  34 #include <csignal>
  35 #include <cstring>
  36 #include <fcntl.h>
  37 #include <sys/socket.h>
  38 #include <netinet/in.h>
  39 #include <arpa/inet.h>
  40 #include <netdb.h>
  41 #include <unistd.h>
  42
  43 #define RECV_BUF_SIZE 8192
  44
  45 /*
  46  * This program must be run from inetd.  First add something like this
  47  * to /etc/services:
  48  *
  49  * cached_announce 3131/udp             # cache announcements
  50  *
  51  * And then add something like this to /etc/inetd/conf:
  52  *
  53  * cached_announce dgram udp       wait cached /tmp/recv-announce recv-announce /tmp/recv-announce.log
  54  *
  55  *
  56  * A single instance of this process will continue to handle incoming
  57  * requests.  If it dies, or is killed, inetd should restart it when the
  58  * next message arrives.
  59  *
  60  */
  61
  62 /*
  63  * usage: recv-announce logfile
  64  */
  65
  66 static void
  67 sig_handle(int)
  68 {
  69     fflush(stdout);
  70     close(2);
  71     close(1);
  72     close(0);
  73     exit(0);
  74 }
  75
  76 int
  77 main(int argc, char *argv[])
  78 {
  79     char buf[RECV_BUF_SIZE];
  80
  81     struct sockaddr_in R;
  82     socklen_t len;
  83
  84     struct hostent *hp = NULL;
  85     const char *logfile;
  86     char ip[4];
  87
  88     for (len = 0; len < 32; ++len) {
  89         signal(len, sig_handle);
  90     }
  91
  92     if (argc > 1)
  93         logfile = argv[1];
  94     else
  95         logfile = "/tmp/recv-announce.log";
  96
  97     close(1);
  98
  99     if (open(logfile, O_WRONLY | O_CREAT | O_APPEND, 0660) < 0) {
 100         perror(logfile);
 101         exit(1);
 102     }
 103
 104     close(2);
 105     dup(1);
 106
 107     IPAddress ipa;
 108     chat tmp[MAX_HOSTNAMELEN];
 109     for (;;) {
 110         memset(buf, '\0', RECV_BUF_SIZE);
 111         memset(&R, '\0', len = sizeof(R));
 112
 113         if (recvfrom(0, buf, RECV_BUF_SIZE, 0, (sockaddr *)&R, &len) < 0) {
 114             perror("recv");
 115             exit(2);
 116         }
 117
 118         memcpy(ip, &R.sin_addr.s_addr, 4);
 119         hp = gethostbyaddr(ip, 4, AF_INET);
 120         ipa = R.sin_addr;
 121         printf("==============================================================================\n");
 122         printf("Received from %s [%s]\n",
 123                ipa.toStr(tmp,MAX_HOSTNAMELEN),
 124                (hp && hp->h_name) ? hp->h_name : "Unknown");
 125         fputs(buf, stdout);
 126         fflush(stdout);
 127     }
 128
 129     return 0;
 130 }