src/header.c

   1 /*======================================================================*
   2  * Copyright (C) 2008 Light Weight Event System                         *
   3  * All rights reserved.                                                 *
   4  *                                                                      *
   5  * This program is free software; you can redistribute it and/or modify *
   6  * it under the terms of the GNU General Public License as published by *
   7  * the Free Software Foundation; either version 2 of the License, or    *
   8  * (at your option) any later version.                                  *
   9  *                                                                      *
  10  * This program is distributed in the hope that it will be useful,      *
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
  13  * GNU General Public License for more details.                         *
  14  *                                                                      *
  15  * You should have received a copy of the GNU General Public License    *
  16  * along with this program; if not, write to the Free Software          *
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor,                   *
  18  * Boston, MA 02110-1301 USA.                                           *
  19  *======================================================================*/
  20 #include "config.h"
  21
  22 #include "header.h"
  23
  24 #include "log.h"
  25 #include "opt.h"
  26 #include "perror.h"
  27 #include "stats.h"
  28
  29 #include "lwes.h"
  30 #include "marshal.h"
  31
  32 #include <netinet/in.h>
  33 #include <sys/time.h>
  34
  35 void header_add(void* buf, int count, unsigned long addr, unsigned short port)
  36 {
  37   unsigned char* cp = (unsigned char*)buf;
  38
  39   struct timeval t;
  40   unsigned long long tm;
  41
  42   if ( -1 == gettimeofday(&t, 0) ) /* This is where we need to */
  43     PERROR("gettimeofday"); /*  timestamp the packet. */
  44
  45   tm = ((((long long)t.tv_sec) * 1000LL) + (long long)(t.tv_usec/1000));
  46
  47   marshal_short(cp, count);    /* Size of message body. */
  48   marshal_ulong_long(cp, tm);  /* Now in msec. */
  49   marshal_long(cp, addr);      /* Sender IP address. */
  50   marshal_short(cp, port);     /* Sender port number. */
  51   marshal_short(cp, arg_site); /* Site ID number */
  52   /* TODO: should be perfectly alright, but un-QA'd  marshal_long(cp, 0); */
  53   /* reserved */
  54 }
  55
  56 int header_is_rotate (void* buf, time_t* when)
  57 {
  58   if ( toknam_eq((unsigned char *)buf + HEADER_LENGTH,
  59                  (unsigned char *)ROTATE_COMMAND) )
  60     {
  61       unsigned long long tm;
  62       unsigned char* cp = (unsigned char*)buf + 2 ;
  63       LOG_PROG("Command::Rotate message received.\n");
  64       unmarshal_ulong_long(cp, tm);
  65       *when = (time_t)(tm / 1000LL);
  66       return 1;
  67     }
  68
  69   // Ping-Pong ?
  70   if ( toknam_eq((unsigned char *)buf + HEADER_LENGTH,
  71                  (unsigned char *)JOURNALLER_PING_EVENT_TYPE) )
  72     {
  73       LOG_PROG("System::Ping message received.\n");
  74       return 2;
  75     }
  76
  77   return 0;
  78 }
  79
  80 void header_fingerprint(void* buf, struct packet_check* pc)
  81 {
  82   unsigned char* cp = (unsigned char*)buf;
  83   long long zero = 0LL;
  84   unsigned long long tm;
  85   short length;
  86
  87   /* Extract the timestamp. */
  88   unmarshal_short(cp, length);
  89   unmarshal_ulong_long(cp, tm);
  90
  91   /* Add phony timestamp of zero. */
  92   cp -= 8;
  93   marshal_long_long(cp, zero);
  94
  95   pc->received = tm;
  96   /* pc->md5 = md5(); */
  97 }
  98
  99 /////////////////////////////////////////////////////////////////////////////
 100 // It's a small thing, but toknam_eq() is quicker than memcmp().
 101 //   * memcmp does a tremendous amount of setup before comparing the first
 102 //     character (it's designed to compare large strings efficently)
 103 //   * we don't care if greater-than or less-than, just equal or not
 104 //   * syntactically simpler (taking advantage of token's length byte)
 105 int toknam_eq(const unsigned char* toknam, const unsigned char* nam)
 106 {
 107   unsigned char len = *nam+1 ;
 108   while ( *toknam++ == *nam++ )
 109     if ( --len == 0 )
 110       return 1 ;
 111   return 0 ;
 112 }
 113
 114 /////////////////////////////////////////////////////////////////////////////
 115 int non_revenue_bearing (const unsigned char* buf)
 116 {
 117   if ( toknam_eq(buf + HEADER_LENGTH,(unsigned char *)JOURNALLER_CM_SERVE) )
 118     {
 119       st.hurryup_discards[0] += 1 ;
 120       return 1 ;
 121     }
 122   if ( toknam_eq(buf + HEADER_LENGTH,(unsigned char *)JOURNALLER_DM_SERVE) )
 123     {
 124       st.hurryup_discards[1] += 1 ;
 125       return 1 ;
 126     }
 127   if ( toknam_eq(buf + HEADER_LENGTH,(unsigned char *)JOURNALLER_SS_SERVE) )
 128     {
 129       st.hurryup_discards[2] += 1 ;
 130       return 1 ;
 131     }
 132   return 0 ;
 133 }
 134
 135 /* globals */
 136 static struct lwes_emitter *emitter = NULL ;
 137 static struct lwes_event_deserialize_tmp *dtmp = NULL;
 138 #define JOURNALLER_PING_SENDER_IP_FIELD "SenderIP"
 139 #define JOURNALLER_PING_RETURN_IP_FIELD "ReturnIP"
 140 #define JOURNALLER_PING_RETURN_PORT_FIELD "ReturnPort"
 141 #define JOURNALLER_PING_DEFAULT_PORT 64646
 142
 143 int
 144 journaller_ping_transport_send_pong (char *address, int port)
 145 {
 146   LOG_PROG("Sending System::Pong to %s:%i\n", address, port);
 147
 148   struct lwes_event *pong_event =
 149     lwes_event_create( (struct lwes_event_type_db *) NULL,
 150         (LWES_SHORT_STRING) JOURNALLER_PONG_EVENT_TYPE);
 151   if ( pong_event == NULL )
 152     {
 153       return 1 ;
 154     }
 155
 156   lwes_emitter_emitto(address, NULL/*arg_interface*/, port, emitter, pong_event);
 157   lwes_event_destroy(pong_event);
 158
 159   LOG_PROG("Sending System::Pong finished\n");
 160   return 0;
 161 }
 162
 163 int
 164 ping (void* buf, size_t bufsiz)
 165 {
 166   unsigned short int return_port ;
 167   struct in_addr return_ip ; memset(&return_ip, 0, sizeof(return_ip)) ;
 168   struct lwes_event *ping_event ;
 169   char* evt = (char*)buf ;
 170
 171   if ( arg_nopong )
 172     {
 173       return 1 ;
 174     }
 175
 176   /* setup System::Pong's transport */
 177   if ( emitter == NULL )
 178     {
 179       emitter = lwes_emitter_create( (LWES_CONST_SHORT_STRING) arg_ip,
 180           (LWES_CONST_SHORT_STRING) NULL, //arg_interface,
 181           (LWES_U_INT_32) arg_port, 0, 60 );
 182     }
 183   if ( emitter == NULL )
 184     {
 185       return -1 ;
 186     }
 187
 188   if ( dtmp == NULL )
 189     {
 190       dtmp = (struct lwes_event_deserialize_tmp *)
 191         malloc(sizeof(struct lwes_event_deserialize_tmp));
 192       if(dtmp == NULL)
 193         {
 194           return -1;
 195         }
 196     }
 197
 198   ping_event = lwes_event_create_no_name(NULL);
 199
 200   if ( ping_event != NULL )
 201     {
 202
 203       lwes_event_from_bytes (ping_event,
 204                              (LWES_BYTE_P)&evt[HEADER_LENGTH],
 205                              bufsiz-HEADER_LENGTH, 0, dtmp);
 206
 207       if( lwes_event_get_IP_ADDR (ping_event,
 208                                   JOURNALLER_PING_RETURN_IP_FIELD,
 209                                   &return_ip) != 0 )
 210         {
 211           if ( lwes_event_get_IP_ADDR(ping_event,
 212                                       JOURNALLER_PING_SENDER_IP_FIELD,
 213                                       &return_ip) != 0 )
 214             { // none-of-the-above, so use header's sender-ip
 215               char* xxx = (char*)&return_ip ;
 216               xxx[0] = evt[13] ;
 217               xxx[1] = evt[12] ;
 218               xxx[2] = evt[11] ;
 219               xxx[3] = evt[10] ;
 220             }
 221         }
 222
 223       if( lwes_event_get_U_INT_16(ping_event,
 224                                   JOURNALLER_PING_RETURN_PORT_FIELD,
 225                                   &return_port) != 0 )
 226         {
 227           return_port = JOURNALLER_PING_DEFAULT_PORT;
 228         }
 229
 230       lwes_event_get_U_INT_16(ping_event, JOURNALLER_PING_RETURN_PORT_FIELD,
 231                               &return_port);
 232
 233       journaller_ping_transport_send_pong(inet_ntoa(return_ip), return_port);
 234       /* cleanup */
 235       lwes_event_destroy(ping_event);
 236     }
 237
 238   /* we're done */
 239   return 0;
 240 }
 241
 242 /* end-of-file */