net_socket.h

//---------------------------------------------------------------------------
// net_socket
//---------------------------------------------------------------------------
#ifndef __NET_SOCKET_H
#define __NET_SOCKET_H

#include "net_stdafx.h"
#include "net_host.h"
#include "system.h"
#include <string>

namespace Net{
    using namespace std;

        class ConnectionException: public System::Exception{
        public:
                        ConnectionException(const char* msg, ...);
                        void show();
        };

    #define NS_SOCKET_BUFFER_LEN 1024

    class Socket{
            protected:
                    int sockId;

            public:
                    Socket(){};

                    void get(char* buffer, int size);

                    void set(char* buffer, int size);

                    int getOnce(char* buffer, int size);
                    void destroy();
                    //bool waitingData();
                    string peerName();
                    void operator=(Socket s){ sockId = s.sockId; };
    };

    class ClientSocket: public Socket{
            public:
                    ClientSocket(){};
                    ClientSocket(Host host);        // connect to host
                    void operator=(ClientSocket s){ sockId = s.sockId; };

    };

    class AcceptedSocket: public Socket{
            protected:
                    Host host;
            public:
                    AcceptedSocket() {};
                    AcceptedSocket(Host h,int fd): host(h) {sockId = fd;};
                    void operator=(AcceptedSocket s){ sockId = s.sockId; };

    };

    class ServerSocket: public Socket{
            public:
                    ServerSocket(){};
                    ServerSocket(unsigned int port);
                    AcceptedSocket wait();
                    AcceptedSocket wait(Host &host);
                    void operator=(ServerSocket s){ sockId = s.sockId; };

    };

} // end of namespace

#endif // __NET_SOCKET_H

---