resources/net/tcp_client_socket.h

   1 #ifndef _TCP_CLIENT_SOCKET_H_
   2 #define _TCP_CLIENT_SOCKET_H_
   3
   4 #include "tcp.h"
   5 #include "client_socket.h"
   6 #include "libs/fifo.h"
   7
   8 namespace net {
   9         class tcp_client_socket : public client_socket {
  10         private:
  11                 p<tcp> down_tcp;
  12
  13                 int client_port;
  14
  15                 int server_port;
  16                 ipv4_addr server;
  17
  18                 fifo<buffer> incoming_data;
  19
  20                 int sequence_number;
  21                 int ack_number;
  22
  23                 enum connection_state {
  24                         unknown,
  25                         //listen,
  26                         syn_sent,
  27                         //syn_recv,
  28                         established,
  29                         fin_wait1,
  30                         fin_wait2,
  31                         //time_wait,
  32                         //close_wait,
  33                         last_ack,
  34                         closed
  35                 };
  36
  37                 volatile connection_state state;
  38
  39                 void receive(const ipv4_addr &addr, u16 sender_port, const buffer &data, tcp::tcp_flags, int, int);
  40                 void clean();
  41         public:
  42                 tcp_client_socket(p<transport_layer> tl);
  43                 ~tcp_client_socket();
  44
  45                 void connect(const ipv4_addr &addr, int port);
  46
  47                 void write(const buffer &data);
  48                 void read(buffer &data);
  49
  50                 void close();
  51         };
  52 }
  53
  54 #endif