www.pudn.com > UDP-based_Reliable_Data_Transfer_Library.zip > appserver.cpp

#ifndef WIN32
#include 
#include 
#include 
#include 
#else
#include 
#include 
#endif
#include 
#include 
//#include "cc.h"

using namespace std;

#ifndef WIN32
void* recvdata(void*);
#else
DWORD WINAPI recvdata(LPVOID);
#endif

int main(int argc, char* argv[])
{
   if ((1 != argc) && ((2 != argc) || (0 == atoi(argv[1]))))
   {
      cout << "usage: appserver [server_port]" << endl;
      return 0;
   }

   addrinfo hints;
   addrinfo* res;

   memset(&hints, 0, sizeof(struct addrinfo));

   hints.ai_flags = AI_PASSIVE;
   hints.ai_family = AF_INET;
   hints.ai_socktype = SOCK_STREAM;
   //hints.ai_socktype = SOCK_DGRAM;

   char* service = "9000";
   if (2 == argc)
      service = argv[1];

   if (0 != getaddrinfo(NULL, service, &hints, &res))
   {
      cout << "illegal port number or port is busy.\n" << endl;
      return 0;
   }

   UDTSOCKET serv = UDT::socket(res->ai_family, res->ai_socktype, res->ai_protocol);


   // UDT Options
   //UDT::setsockopt(serv, 0, UDT_CC, new CCCFactory, sizeof(CCCFactory));
   //UDT::setsockopt(serv, 0, UDT_MSS, new int(7500), sizeof(int));
   //UDT::setsockopt(serv, 0, UDT_RCVBUF, new int(100000000), sizeof(int));
   //UDT::setsockopt(serv, 0, UDP_RCVBUF, new int(100000000), sizeof(int));


   if (UDT::ERROR == UDT::bind(serv, res->ai_addr, res->ai_addrlen))
   {
      cout << "bind: " << UDT::getlasterror().getErrorMessage() << endl;
      return 0;
   }

   freeaddrinfo(res);


   cout << "server is ready at port: " << service << endl;

   if (UDT::ERROR == UDT::listen(serv, 10))
   {
      cout << "listen: " << UDT::getlasterror().getErrorMessage() << endl;
      return 0;
   }


   sockaddr_storage clientaddr;
   int addrlen = sizeof(clientaddr);

   UDTSOCKET recver;

   while (true)
   {
      if (UDT::INVALID_SOCK == (recver = UDT::accept(serv, (sockaddr*)&clientaddr, &addrlen)))
      {
         cout << "accept: " << UDT::getlasterror().getErrorMessage() << endl;
         return 0;
      }

      char clienthost[NI_MAXHOST];
      char clientservice[NI_MAXSERV];
      getnameinfo((sockaddr *)&clientaddr, addrlen, clienthost, sizeof(clienthost), clientservice, sizeof(clientservice), NI_NUMERICHOST|NI_NUMERICSERV);
      cout << "new connection: " << clienthost << ":" << clientservice << endl;

#ifndef WIN32
      pthread_t rcvthread;
      pthread_create(&rcvthread, NULL, recvdata, new UDTSOCKET(recver));
      pthread_detach(rcvthread);
#else
      CreateThread(NULL, 0, recvdata, new UDTSOCKET(recver), 0, NULL);
#endif
   }

   UDT::close(serv);

   return 1;
}

#ifndef WIN32
void* recvdata(void* usocket)
#else
DWORD WINAPI recvdata(LPVOID usocket)
#endif
{
   UDTSOCKET recver = *(UDTSOCKET*)usocket;
   delete (UDTSOCKET*)usocket;

   char* data;
   int size = 10000000;
   data = new char[size];

   int handle;

   while (true)
   {
      if (UDT::ERROR == UDT::recv(recver, data, size, 0, &handle, NULL))
      //if (UDT::ERROR == UDT::recvmsg(recver, data, size))
      {
         cout << "recv:" << UDT::getlasterror().getErrorMessage() << endl;
         break;
      }
   }

   delete [] data;

   UDT::close(recver);

#ifndef WIN32
   return NULL;
#else
   return 0;
#endif
}