src/tests/test_proto_inet.c

   1 /**************************************************************************/
   2 /* ---------------------------------------------------------------------- */
   3 /* Copyright (C) 2006 - 2008 FRESCOR consortium partners:                 */
   4 /*                                                                        */
   5 /*   Universidad de Cantabria,              SPAIN                         */
   6 /*   University of York,                    UK                            */
   7 /*   Scuola Superiore Sant'Anna,            ITALY                         */
   8 /*   Kaiserslautern University,             GERMANY                       */
   9 /*   Univ. Politécnica  Valencia,           SPAIN                        */
  10 /*   Czech Technical University in Prague,  CZECH REPUBLIC                */
  11 /*   ENEA                                   SWEDEN                        */
  12 /*   Thales Communication S.A.              FRANCE                        */
  13 /*   Visual Tools S.A.                      SPAIN                         */
  14 /*   Rapita Systems Ltd                     UK                            */
  15 /*   Evidence                               ITALY                         */
  16 /*                                                                        */
  17 /*   See http://www.frescor.org for a link to partners' websites          */
  18 /*                                                                        */
  19 /*          FRESCOR project (FP6/2005/IST/5-034026) is funded             */
  20 /*       in part by the European Union Sixth Framework Programme          */
  21 /*       The European Union is not liable of any use that may be          */
  22 /*       made of this code.                                               */
  23 /*                                                                        */
  24 /*                                                                        */
  25 /*  This file is part of FORB (Frescor Object Request Broker)             */
  26 /*                                                                        */
  27 /* FORB is free software; you can redistribute it and/or modify it        */
  28 /* under terms of the GNU General Public License as published by the      */
  29 /* Free Software Foundation; either version 2, or (at your option) any    */
  30 /* later version.  FORB is distributed in the hope that it will be        */
  31 /* useful, but WITHOUT ANY WARRANTY; without even the implied warranty    */
  32 /* of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU    */
  33 /* General Public License for more details. You should have received a    */
  34 /* copy of the GNU General Public License along with FORB; see file       */
  35 /* COPYING. If not, write to the Free Software Foundation, 675 Mass Ave,  */
  36 /* Cambridge, MA 02139, USA.                                              */
  37 /*                                                                        */
  38 /* As a special exception, including FORB header files in a file,         */
  39 /* instantiating FORB generics or templates, or linking other files       */
  40 /* with FORB objects to produce an executable application, does not       */
  41 /* by itself cause the resulting executable application to be covered     */
  42 /* by the GNU General Public License. This exception does not             */
  43 /* however invalidate any other reasons why the executable file might be  */
  44 /* covered by the GNU Public License.                                     */
  45 /**************************************************************************/
  46
  47 /**
  48  * @file   test_proto_inet.c
  49  * @author Michal Sojka <sojkam1@fel.cvut.cz>
  50  * @date   Thu Aug 28 15:20:37 2008
  51  *
  52  * @brief Unit test for INET socket based FORB transport protocol
  53  */
  54
  55 #include <error.h>
  56 #include "../proto_inet.c"
  57 #include <wvtest.h>
  58
  59 const char *msg = "Hello world!";
  60 const char *msg2 = "Blablabla!";
  61 #define NUM_PORTS 10
  62
  63 char *timeout_msg = "timeout\n";
  64
  65 void timeout(int signal)
  66 {
  67         //error(1, 0, timeout_msg);
  68 }
  69
  70 WVTEST_MAIN("AF_INET inter-orb protocol")
  71 {
  72         int len;
  73         struct inet_addr addr[NUM_PORTS];
  74         forb_port_t port[NUM_PORTS];
  75         int i, ret;
  76
  77         signal(SIGALRM, timeout);
  78
  79         memset(port, 0, sizeof(port));
  80
  81         for (i=0; i<NUM_PORTS; i++) {
  82                 addr[i].addr.s_addr = htonl(0x7f000001+i);
  83                 if (!WVPASSEQ(forb_inet_port_init(&port[i].desc, addr[i].addr, 0), 0))
  84                         error(1, errno, "forb_inet_port_init(%d)", i);
  85         }
  86
  87
  88         /* Send from zeroth node to all others (starting from 1) */
  89         for (i=1; i<NUM_PORTS; i++) {
  90                 char buf[100];
  91                 char tmsg[100];
  92                 forb_peer_t *peer;
  93
  94                 WVPASS(peer = forb_peer_new());
  95                 peer->addr = port[i].desc.addr;
  96                 peer->port = &port[0];
  97
  98                 len = strlen(msg)+1;
  99                 WVPASSEQ(ret = inet_send(peer, msg, len), len);
 100
 101                 peer->addr = NULL;
 102                 forb_peer_put(peer);
 103                 if (ret != len)
 104                         error(1, errno, "send(to %d) = %d", i, ret);
 105
 106                 sprintf(tmsg, "Timeout when sending from 0 to %d\n", i);
 107                 timeout_msg = tmsg;
 108                 alarm(1);       /* timeout 1s */
 109                 WVPASSEQ(ret = inet_recv(&port[i], buf, sizeof(buf)), len);
 110                 alarm(0);
 111                 if (ret != len)
 112                         error(1, errno, "recv(port[%d]) = %d", i, ret);
 113
 114                 if (!WVPASS(strcmp(msg, buf) == 0))
 115                         error(1, errno, "port %d: received wrong data", i);
 116         }
 117
 118 #ifndef CONFIG_FORB_PROTO_INET_DEFAULT
 119         len = strlen(msg2)+1;
 120         WVPASSEQ(ret = inet_broadcast(&port[0], msg2, len), len);
 121         if (ret != len)
 122                 error(1, errno, "broadcast(port[0]) = %d should be %d", ret, len);
 123
 124         for (i=0; i<NUM_PORTS; i++) {
 125                 char buf[100];
 126                 char tmsg[100];
 127
 128                 sprintf(tmsg, "Timeout when receiving broadcast for port %d\n", i);
 129                 timeout_msg = tmsg;
 130                 alarm(1);
 131                 WVPASSEQ(ret = inet_recv(&port[i], buf, sizeof(buf)), len);
 132                 alarm(0);
 133                 if (ret != len)
 134                         error(1, errno, "broadcast recv(port[%d]) = %d should be %d (errno=%d)", i, ret, len, errno);
 135                 //printf("broadcast received: %s\n", buf);
 136
 137                 if (!WVPASS(strcmp(msg2, buf) == 0))
 138                         error(1, errno, "port %d: broadcast received wrong data", i);
 139
 140         }
 141 #endif
 142
 143         for (i=0; i<NUM_PORTS; i++) {
 144                 inet_port_destroy(&port[i]);
 145                 if (port[i].new_peer)
 146                         forb_peer_put(port[i].new_peer);
 147         }
 148 }