source: orxonox.OLD/trunk/src/subprojects/network/network_unit_test.cc @ 10345

#include "stdlibincl.h"
#include <stdarg.h>
#include <stdio.h>
#include "shell_buffer.h"
#include "class_list.h"

#include "network_manager.h"
#include "network_socket.h"
#include "server_socket.h"
#include "udp_server_socket.h"
#include "udp_socket.h"
#include "network_stream.h"
#include "synchronizeable.h"
#include "converter.h"
#include "state.h"
#include "shared_network_data.h"
#include "message_manager.h"
#include "network_log.h"

#include "simple_sync.h"
#include "read_sync.h"

int verbose = 4;


/* outputs the help */
int startHelp(int argc, char** argv)
{
  printf("Network is a network unit test\n");
  printf(" --help                     this output\n");
  printf(" -st,  --sockettest         test network_socket\n");
  printf(" -ft,  --frameworktest      test the network module\n");
  printf(" --server [port number]     creates a test server\n");
  printf(" --client [address] [port]  connects to a server\n");
  printf(" --listen [address] [port]  just listens to this connection");
  printf("\n");
}


/* test SDL network socket */
int testSocket(int argc, char** argv)
{
  UdpServerSocket server(9999);

  NetworkSocket* client1 = new UdpSocket("localhost", 9999);

  NetworkSocket* server1 = NULL;
  while ( server1 == NULL )
  {
    server.update();
    server1 = server.getNewSocket();
  }

  assert( server1->isOk() );

  NetworkSocket* client2 = new UdpSocket("localhost", 9999);

  NetworkSocket* server2 = NULL;
  while ( server2 == NULL )
  {
    server.update();
    server2 = server.getNewSocket();
  }

  char buf[1024];

  printf("read from client1 before sending data\n");
  printf("result: %d bytes\n", client1->readPacket((byte*)buf, 1024));

  printf("read from client2 before sending data\n");
  printf("result: %d bytes\n", client2->readPacket((byte*)buf, 1024));

  int n;
  char * str1 = "client1 to server";
  char * str2 = "server1 to client";
  char * str3 = "client2 to server";
  char * str4 = "server2 to client";
  n = client1->writePacket((byte*)str1, strlen(str1)+1);
  printf("%d bytes send from client1\n", n);
  n = server1->writePacket((byte*)str2, strlen(str2)+1);
  printf("%d bytes send from server1\n", n);
  n = client2->writePacket((byte*)str3, strlen(str3)+1);
  printf("%d bytes send from client2\n", n);
  n = server2->writePacket((byte*)str4, strlen(str4)+1);
  printf("%d bytes send from server2\n", n);
  SDL_Delay(10);

  server.update();

  printf("read from server1\n");
  n = server1->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  printf("read from server2\n");
  n = server2->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  printf("read from client1\n");
  n = client1->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  printf("read from client2\n");
  n = client2->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);


  //22222222222222222222222222222222222222222
  n = client1->writePacket((byte*)str1, strlen(str1)+1);
  printf("%d bytes send from client1\n", n);
  n = server1->writePacket((byte*)str2, strlen(str2)+1);
  printf("%d bytes send from server1\n", n);
  n = client2->writePacket((byte*)str3, strlen(str3)+1);
  printf("%d bytes send from client2\n", n);
  n = server2->writePacket((byte*)str4, strlen(str4)+1);
  printf("%d bytes send from server2\n", n);
  SDL_Delay(10);

  server.update();

  printf("read from server1\n");
  n = server1->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  printf("read from server2\n");
  n = server2->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  printf("read from client1\n");
  n = client1->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  printf("read from client2\n");
  n = client2->readPacket((byte*)buf, 1024);
  printf("read %d bytes\n", n);
  if (n<0)
    return -1;

  printf("data: '%s'\n", buf);

  //sending bigger packets than 255 is not supported
#if 0
  printf("try to send more than 255 bytes\n");
  printf("result: %d\n", client1->writePacket((byte*)buf, 1000));

  server1->writePacket((byte*)str1, strlen(str1)+1);
  SDL_Delay(500);
  printf("try to read with a too small buffer\n");
  printf("result: %d\n", client1->readPacket((byte*)buf, strlen(str1)));
#endif

  delete client1;
  delete client2;
  delete server1;
  delete server2;

  return 0;
}


int testFramework(int argc, char** argv)
{
  printf("=================\n");
  printf("TestFramework\n");
  printf("=================\n");

  Synchronizeable* clientSync = new SimpleSync( std::string("Client") );
  Synchronizeable* serverSync = new SimpleSync( std::string("Server") );

  unsigned int port = 9999;

  /* create the network manager */
  NetworkManager* nm = NetworkManager::getInstance();

  /* initialize the network manager */
  nm->initialize();

  /* create a server stream */
  nm->createMasterServer(port);

  /* esatblish a connection */
  IP ip;
  int error = SDLNet_ResolveHost(&ip, "127.0.0.1", port);
  //SDLNet_ResolveHost(&ip, "localhost", port);
  if(error == -1)
    printf("\n\nerror on address resolution, program inconsistancy\n\n");
  nm->createClient("localhost", port);
  nm->connectSynchronizeable( *clientSync );
  /* adding some break for connection setup */
  SDL_Delay(20);

  /* synchronize the data 1 time (increment for longer tests) */
  for( int i = 0; i < 3; i++)
  {
    nm->synchronize( 1000 );
    /* simulate the network delay */
    SDL_Delay(50);
  }

  printf("Test finished\n");


  /* delete the network manager again */
  delete nm;

  delete clientSync;
  delete serverSync;

  return 0;
}

bool testCB( MessageType messageType, byte * data, int dataLength, void * someData, int userId )
{
  printf("GOT MESSAGE: %s\n", data);
  return true;
}

/**
 *
 * @param argc
 * @param argv
 * @return
 */
int startServer(int argc, char** argv)
{


  State::setOnline(true);
  SharedNetworkData::getInstance()->setGameServer( true );
  printf("Starting Server on port %i\n", 9999);

  NetworkManager* netMan = NetworkManager::getInstance();

  netMan->initialize();

  netMan->createMasterServer(/**ss, */ 9999);

  SimpleSync* ss = new SimpleSync("Server");
  ss->setSynchronized( true );

  NetworkLog::getInstance()->listen( 8888 );

  //MessageManager::getInstance()->initUser( 1 );
  MessageManager::getInstance()->registerMessageHandler( TESTMESSAGEID, testCB, NULL );

  SDL_Delay(20);

  for(;;)
  {
    MessageManager::getInstance()->sendMessage( TESTMESSAGEID, (byte*)"server to client", 18, RT_ALL_BUT_ME, NET_UNASSIGNED, MP_HIGHBANDWIDTH );
    netMan->synchronize( 100 );
    SDL_Delay(100);
    ss->debug();
  }

  delete netMan;
  delete ss;


  return 0;
}


int startClient(int argc, char** argv)
{


  State::setOnline(true);
  NetworkManager* netMan = NetworkManager::getInstance();


  netMan->initialize();

  std::string host;

  if ( argc > 2 )
    host = argv[2];
  else
    host = "localhost";

  netMan->createClient(host, 9999);

//  SimpleSync* ss = new SimpleSync("SimpleSync");
//  ss->setSynchronized( true );
//  netMan->connectSynchronizeable( *ss );

  NetworkLog::getInstance()->listen( 7777 );

  SimpleSync * ss = NULL;

  //MessageManager::getInstance()->initUser( 0 );
  MessageManager::getInstance()->registerMessageHandler( TESTMESSAGEID, testCB, NULL );

  for(;;)
  {
    netMan->synchronize( 100 );
    MessageManager::getInstance()->sendMessage( TESTMESSAGEID, (byte*)"client to server", 18, RT_ALL_BUT_ME, NET_UNASSIGNED, MP_HIGHBANDWIDTH );
    ss = dynamic_cast<SimpleSync*>(ClassList::getObject( "Server", CL_SIMPLE_SYNC ) );
    SDL_Delay(100);
    if (ss)
      ss->debug();
  }


  delete netMan;
  delete ss;

  return 0;
}



int startListen(int argc, char** argv)
{
  if( argc < 3)
  {
    printf(" Wrong arguments try following notations:\n");
    printf("   --listen [server ip] [port number]\n");
    printf("   --listen [server name] [port number]\n");
    return 0;
  }

  char* name = argv[2];
  int port = atoi(argv[3]);
  printf("Connecting to %s, on port %i\n", name, port);

  IP ip;
  int error = SDLNet_ResolveHost(&ip, name, port);
  if(error == -1)
    printf("\n\nerror on address resolution, program inconsistancy\n\n");

  NetworkManager* netMan = NetworkManager::getInstance();
  Synchronizeable* ss = new SimpleSync("SimpleSync\0");

  netMan->createClient( name, port );
  netMan->connectSynchronizeable( *ss );

  for(;;)
  {
    netMan->synchronize( 1000 );
    SDL_Delay(10);
  }


  delete netMan;
  delete ss;

  return 0;
}

void testFloatConverter(float f)
{
#if 0
  char* s = Converter::floatToBinString(f);
  printf("%f = ", f);
  printf(s); printf("\n");

  byte* res = Converter::floatToByteArray(f);
  printf("Byte Array: ");
  for (int i = 0; i < 4; i++)
//    printf("%i  ", res[i]);
  printf("\n");

  float b = Converter::byteArrayToFloat(res);
  printf("ReConvert: %f \n", b);
#endif
}

void testFloatConverter2(float f)
{
#if 0
  char* s = Converter::floatToBinString(f);
  printf("### %f = ", f);
  printf(s); printf("\n");

  byte* res = Converter::_floatToByteArray(f);
  printf("Byte Array: ");
  for (int i = 0; i < 4; i++)
    printf("%i  ", res[i]);
  printf("\n");

  float b = Converter::_byteArrayToFloat(res);
  printf("ReConvert: %f \n", b);
#endif
}
int converter(int argc, char** argv)
{
  /*
  int x = 200564786;
  printf("To convert: %i\n", x);
  byte* res = Converter::intToByteArray(x);
  for (int i = 0; i < 4; i++)
    printf("%i  ", res[i]);
  printf("\n");

  int z = Converter::byteArrayToInt(res);

  printf("ReConvert: %i\n", z);


  //float a = 5.4f;
  //float b = 2.0f;
  //printf("%f mod %f = %f", a, b, a % b);

  printf("\n");

  */
  /*
  float y;
  char* s;

  y = 12.0f;
  s = Converter::floatToBinString(y);
  printf("%f = ", y);
  printf(s); printf("\n");

  y = 24549026.0f;
  s = Converter::floatToBinString(y);
  printf("%f = ", y);
  printf(s); printf("\n");

  y = 12.4e20f;
  s = Converter::floatToBinString(y);
  printf("%f = ", y);
  printf(s); printf("\n");

  y = 4.7824f;
  s = Converter::floatToBinString(y);
  printf("%f = ", y);
  printf(s); printf("\n");

  y = -4.7824f;
  s = Converter::floatToBinString(y);
  printf("%f = ", y);
  printf(s); printf("\n");

  y = -14.35e14f;
  s = Converter::floatToBinString(y);
  printf("%f = ", y);
  printf(s); printf("\n");
                                                            */


  /*
  float a = 12.3f;

  char* s = Converter::floatToBinString(a);
  printf("%f = ", a);
  printf(s); printf("\n");

  byte* res = Converter::floatToByteArray(a);
  printf("Byte Array: \n");
  for (int i = 0; i < 4; i++)
    printf("%i  ", res[i]);
  printf("\n");

  float b = Converter::byteArrayToFloat(res);
  printf("ReConvert: %f \n", b);
  */
//  testFloatConverter(12.3e-53f); printf("\n");
//  testFloatConverter(134.26455646546548741661675165f); printf("\n");
 // testFloatConverter(35.67e14f); printf("\n");

  testFloatConverter(12.3e-7f); printf("\n");
  testFloatConverter(134.26455646546548741661675165f); printf("\n");
  testFloatConverter(35.67e14f); printf("\n");

  return 0;
}



/**
 *
 *  main function
 *
 * here the journey begins
 */
int main(int argc, char** argv)
{
  int i;
  // here the pre-arguments are loaded, these are needed to go either to orxonx itself, Help, or Benchmark.
  for(i = 1; i < argc; ++i)
  {
    //else if(!strcmp( "--gui", argv[i]) || !strcmp("-g", argv[i])) showGui = true;
    if (! strcmp( "--sockettest", argv[i]) || !strcmp("-st", argv[i]))
      return testSocket(argc, argv);
    else if (! strcmp( "--frameworktest", argv[i]) || !strcmp("-ft", argv[i]))
      return testFramework(argc, argv);
    else if (! strcmp( "--server", argv[i]) || !strcmp("-s", argv[i]))
      return startServer(argc, argv);
    else if (! strcmp( "--client", argv[i]) || !strcmp("-c", argv[i]))
      return startClient(argc, argv);
    else if (! strcmp( "--listen", argv[i]) || !strcmp("-l", argv[i]))
      return startListen(argc, argv);
    else if (! strcmp( "--converter", argv[i]) || !strcmp("-o", argv[i]))
      return converter(argc, argv);
  }

  startHelp(argc, argv);

  return 0;
}


//bool ShellBuffer::addBufferLineStatic(const char* line, ...)
//{
  //va_list arguments;
  //vprintf(line, arguments);
  //  printf("%s", line);
//}