tools/Bounce/Bounce.h

   1 #include <sys/types.h>
   2 #include <sys/time.h>
   3 #include <sys/wait.h>
   4 #include <sys/socket.h>
   5 #include <sys/ioctl.h>
   6 #include <arpa/inet.h>
   7 #include <netinet/in.h>
   8 #include <netdb.h>
   9 #include <ctype.h>
  10 #include <time.h>
  11 #include <stdio.h>
  12 #include <unistd.h>
  13 #include <stdarg.h>
  14 #include <signal.h>
  15 #include <fcntl.h>
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18 #include <errno.h>
  19 #include <string.h>
  20 #include <netdb.h>
  21 #include <ctype.h>
  22 #include <time.h>
  23 #include <list>
  24 using std::list;
  25
  26
  27 /*
  28  *  "Bounce" Class.
  29  */
  30
  31 class Listener;
  32 class Connection;
  33 class Bounce
  34 {
  35 public:
  36   list<Listener*> listenerList; // List of Listeners.
  37   list<Connection*> connectionsList; // List of 'Connections'.
  38
  39   void bindListeners(); //Binds Listening Ports.
  40   void checkSockets(); // Polls all sockets.
  41   void recieveNewConnection(Listener*);
  42 };
  43
  44 /*
  45  *  "Socket" Class.
  46  */
  47
  48 class Socket
  49 {
  50 public:
  51   int fd;
  52   int lastReadSize;
  53   struct sockaddr_in address;
  54   int connectTo(char*, unsigned short);
  55   int write(char*, int);
  56   int write(char*);
  57   char* read();
  58   Socket();
  59 };
  60
  61 /*
  62  *  "Listener" Class.
  63  */
  64
  65 class Bounce;
  66 class Listener
  67 {
  68 public:
  69   int fd;
  70   int remotePort;
  71   int localPort;
  72   char myVhost[15];
  73   char remoteServer[15];
  74
  75   void beginListening();
  76   Socket* handleAccept();
  77 };
  78
  79 /*
  80  *  "Connection" Class.
  81  */
  82
  83 class Connection
  84 {
  85 public:
  86   Socket* localSocket;
  87   Socket* remoteSocket;
  88 };
  89