apps/astreamer/socket.h

Go to the documentation of this file.

#ifndef _SOCKETS_H_
#define _SOCKETS_H_

#include <string>
#include <list>

using namespace std;

class TCP_Client {
public:
        TCP_Client(bool enable_exception);
        ~TCP_Client();
        void connect(string ip, string port);
        void disconnect(void);
        void send(const string *data, int opt = 0);
        void send(const string data, int opt = 0);
        void send(void *data, int len, int opt = 0);
        void recv(string &r, int opt = 0);
protected:
        bool ex;
        int fd;
        char *buf;
};

class Socket {
public:
        enum stype {TYPE_TCP, TYPE_UDP};
        enum state {STATE_EMPTY, STATE_IN, STATE_DISCONNECT};

        Socket(string ip, int port, stype _t = TYPE_TCP);
        ~Socket();

        bool recv(string &rez);
        bool send(void *data, int len);
        bool send(const string *data);
        bool send(const string data);
        bool send2v(void **v_ptr, int *v_len);
        bool send3v(void **v_ptr, int *v_len);

        static int poll(list<Socket *> &s);
        void listen(int in);
        Socket *accept(void);
//      bool connect(void);

        inline state state_refresh(void) {
                state st = _state;
                _state = STATE_EMPTY;
                return st;
        }
        int get_fd(void) {      return fd;};
        int fd;
        string source_ip(void) {
                return ip;
        }
protected:
        Socket(int _fd, string _ip, int _port, stype _t);

        string ip;
        int port;
        stype type;
//      int fd;
        state _state;

        int ttl;
        unsigned short ip_id;
        bool _is_multicast(string ip);
        unsigned long ip_to_number(string ip);
        bool is_multicast;
/*
        struct sockaddr_ll s_ll;
        struct msghdr msg;
        struct udphdr udp_hdr;
        struct iphdr ip_hdr;
*/
};

#endif

---