src/recv-announce.cc

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