src/Sockets/tests/retry.cpp

   1 #include <TcpSocket.h>
   2 #include <SocketHandler.h>
   3 #include <ListenSocket.h>
   4
   5
   6         bool quit = false;
   7
   8 /*
   9         virtual bool OnConnectRetry();
  10 -       void SetRetryClientConnect(bool x = true);
  11 -       bool RetryClientConnect();
  12         void SetConnectionRetry(int n);
  13         int GetConnectionRetry();
  14         void IncreaseConnectionRetries();
  15         int GetConnectionRetries();
  16         void ResetConnectionRetries();
  17 */
  18 class RetrySocket : public TcpSocket
  19 {
  20 public:
  21         RetrySocket(ISocketHandler& h) : TcpSocket(h) {
  22                 SetConnectTimeout(2);
  23                 SetConnectionRetry(-1);
  24         }
  25
  26         bool OnConnectRetry() {
  27                 printf("Connection attempt#%d\n", GetConnectionRetries());
  28                 if (GetConnectionRetries() == 3)
  29                 {
  30                         ListenSocket<RetrySocket> *l = new ListenSocket<RetrySocket>(Handler());
  31                         if (l -> Bind(12345))
  32                         {
  33                                 printf("Bind port 12345 failed\n");
  34                         }
  35                         l -> SetDeleteByHandler();
  36                         Handler().Add(l);
  37                 }
  38                 return true;
  39         }
  40
  41         void OnConnect() {
  42                 printf("Connected\n");
  43                 printf("GetRemoteAddress(): %s\n", GetRemoteAddress().c_str());
  44                 printf("Remote address: %s\n", GetRemoteSocketAddress() -> Convert(false).c_str());
  45                 printf("Remote address: %s\n", GetRemoteSocketAddress() -> Convert(true).c_str());
  46                 SetCloseAndDelete();
  47         }
  48
  49         void OnDelete() {
  50                 quit = true;
  51         }
  52 };
  53
  54
  55 int main(int argc, char *argv[])
  56 {
  57         SocketHandler h;
  58         RetrySocket sock(h);
  59         sock.Open("localhost", 12345);
  60         h.Add(&sock);
  61         while (!quit)
  62         {
  63                 h.Select(0, 200000);
  64         }
  65 }
  66
  67