td_smsg.c

   1 /*
   2  * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *      notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *      notice, this list of conditions and the following disclaimer in the
  12  *      documentation and/or other materials provided with the distribution.
  13  * 3. Neither the name of the author nor the names of any co-contributors
  14  *      may be used to endorse or promote products derived from this software
  15  *      without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
  18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
  21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  23  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  26  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
  27  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29 #include <string.h>
  30 #include <time.h>
  31
  32 #include "tlog.h"
  33 #include "connection.h"
  34 #include "top.h"
  35 #include "tmalloc.h"
  36
  37
  38 static void
  39 addMsgTop( MassiveUnit *mu, TCMsgTop *tmsg, struct in_addr *client ) {
  40         UnitInfo unit;
  41
  42         memcpy( &(unit.top), tmsg, sizeof( TCMsgTop ));
  43         unit.stamp=time(NULL);
  44         memcpy( &(unit.ip), client, sizeof(struct in_addr) );
  45
  46         fillUnit(mu, &unit);
  47 }
  48
  49
  50 void
  51 gotMsg(TC_Connection *msgconn) {
  52         Msg gmsg;
  53         TCMsg *pmsg;
  54
  55         if ( TC_getMsg( msgconn->fd, &gmsg ) )
  56                         return;
  57
  58         pmsg = gmsg.msg;
  59
  60         if ( cfg.debug )
  61                 tlog(TL_DEBUG, "Got %d bytes %08x type from %s:%d",
  62                         pmsg->len,
  63                         pmsg->type,
  64                         inet_ntoa(gmsg.host_addr.sin_addr),
  65                         ntohs(gmsg.host_addr.sin_port)
  66                 );
  67
  68         switch( pmsg->type ) {
  69                 case TOPMSGTYPE:
  70                         addMsgTop( cfg.mu, (TCMsgTop*)(pmsg->data), &(gmsg.host_addr.sin_addr) );
  71                         break;
  72                 default:
  73                         tlog(TL_ALARM, "Unknown message type: %d from %s",
  74                                 pmsg->type, inet_ntoa(gmsg.host_addr.sin_addr));
  75         }
  76 }