td_lmsg.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 <stdlib.h>
  30 #include <string.h>
  31
  32 #include "tlog.h"
  33 #include "connection.h"
  34 #include "top.h"
  35 #include "tmalloc.h"
  36
  37 static void
  38 getTopData(TC_Connection *conn) {
  39         TCMsg *pmsg;
  40         conn->len = TCMSGHDRSZ + MUHDRSZ + cfg.mu->number * sizeof(UnitInfo);
  41         conn->buf = trealloc( conn->buf, conn->len );
  42         pmsg = (TCMsg*) conn->buf;
  43
  44         pmsg->len = conn->len;
  45         pmsg->type = TOPGETTYPE;
  46         memcpy( pmsg->data, cfg.mu, pmsg->len - TCMSGHDRSZ );
  47         ((MassiveUnit*)pmsg->data)->maxnumber = cfg.mu->number;
  48         TC_Send(conn);
  49 }
  50
  51 static void
  52 gotLongMessage( TC_Connection *conn ) {
  53         TCMsg *pmsg = (TCMsg*) conn->buf;
  54
  55         if ( conn->ptr - conn->buf < TCMSGHDRSZ ) {
  56                 tlog(TL_ALARM,"Wrong size of long message (%d bytes)", conn->ptr - conn->buf);
  57                 conn->state = CS_ERROR;
  58                 return;
  59         }
  60
  61         switch(pmsg->type) {
  62                 case TOPGETTYPE:
  63                         getTopData(conn);
  64                         break;
  65                 default:
  66                         tlog(TL_ALARM,"Got unknown packet type: %d", pmsg->type);
  67                         conn->state = CS_ERROR;
  68         }
  69 }
  70
  71 static void
  72 sendedLongMessage( TC_Connection *conn ) {
  73         TCMsg *pmsg = (TCMsg*) conn->buf;
  74         switch(pmsg->type) {
  75                 case TOPGETTYPE:
  76                         conn->state = CS_FINISH;
  77                         break;
  78                 default:
  79                         tlog(TL_ALARM,"Send unknown packet type: %d", pmsg->type);
  80                         conn->state = CS_ERROR;
  81         }
  82 }
  83
  84 void
  85 ConnectionWorks() {
  86         int i;
  87         TC_Connection *conn;
  88         for(i=TC_SERVER_CONN_START; i<cfg.pool.number; i++) {
  89                 conn = cfg.pool.conn[i];
  90                 switch( conn->state ) {
  91                         case CS_INPROCESS:
  92                                 TC_ServerConnect( conn, 0 );
  93                                 break;
  94                         case CS_CONNECTED: /* we have connected to remote host */
  95                                 TC_Send( conn );
  96                                 break;
  97                         case CS_FINISHSEND:
  98                                 sendedLongMessage( conn );
  99                                 break;
 100                         case CS_FINISHREAD:
 101                                 gotLongMessage( conn );
 102                                 break;
 103                         case CS_READ:
 104                                 if ( conn->readyio ) {
 105                                         TC_Read( conn, TCMSGHDRSZ );
 106                                         if ( conn->state != CS_READ ) i--; /* check it one more time */
 107                                 }
 108                                 break;
 109                         case CS_SEND:
 110                                 if ( conn->readyio ) {
 111                                         TC_Send( conn );
 112                                         if ( conn->state != CS_SEND ) i--; /* check it one more time */
 113                                 }
 114                                 break;
 115                         /* ignore */
 116                         case CS_OK:
 117                         case CS_AGAIN:
 118                         case CS_WAIT: /* ??? */
 119                                 break;
 120                         /* out */
 121                         case CS_TIMEOUT:
 122                         case CS_CLOSED:
 123                         case CS_NOTINITED:
 124                         case CS_FINISH:
 125                         case CS_ERROR:
 126                         default:
 127                                 tlog(TL_DEBUG, "Clear N%d connection", i);
 128                                 TC_deleteConnectionByN(&cfg.pool,i);
 129                                 i--;
 130                 }
 131         }
 132 }