Net/include/net.h

//
//  net.h
//  Net
//
//  Created by 胡一兵 on 2019/1/13.
//  Copyright © 2019年 Bakantu. All rights reserved.
//

#ifndef net_h
#define net_h

#include "type.h"

//IP地址管理类
class Addr{
//    IP地址管理结构
    struct sockaddr_in address;
//    IP地址管理结构的大小
    socklen_t addr_size = 0;
public:
    Addr(string ip_addr, int port = 0, bool ipv4 = true);
    Addr(struct sockaddr_in saddri);
    Addr();
    Addr(const Addr &t_addr);
//    获得记录IP地址管理结构的大小的变量本身
    socklen_t *SizeP(void);
//    获得IP地址管理结构大小
    socklen_t Size(void);
//    重新设置IP地址管理结构所对应的端口
    void SetPort(int port);
//    重新设置IP地址管理结构所对应的IP地址
    void SetIP(string ip_addr);
//    IP地址管理结构的大小变量
    void SetSize(void);
    void SetSockAddr(struct sockaddr_in);
//    获得指向IP地址管理结构的指针
    struct sockaddr_in *Obj(void);
//    获得指向IP地址管理结构的指针
    struct sockaddr *RawObj(void);
    void SetIpv4(void);
    void SetIpv6(void);
    static bool checkValidIP(string ipaddr);

};

//服务器套接字类
class SocketServer{
public :
//    套接字操作柄
    int server_sfd;
//    服务器IP及端口管理类
    Addr server_addr;
//    传输协议参数
    int ipptl;
//    临时缓冲区
    char buff[BUFSIZ];
    SocketServer(int port,bool ipv4){
		server_addr.SetIP("127.0.0.1");
        server_addr.SetPort(port);
        if(ipv4){
            ipptl = AF_INET;
            server_addr.SetIpv4();
        }
        else{
            ipptl = AF_INET6;
            server_addr.SetIpv6();
        }
    }
    SocketServer(void){
        server_sfd = -1;
    }
    ~SocketServer(){
        //close(server_sfd);
    }
//    接受储存简单字符串
    virtual ssize_t Recv(string &str) = 0;
//    接受储存二进制串
    virtual ssize_t RecvRAW(char **p_rdt, Addr &taddr) = 0;
};

//客户端套接字类
class SocketClient{
public :
//    目标服务器IP地址及端口管理类
    Addr send_addr;
//    套接字操作柄
    int client_sfd;
//    传输协议参数
    int ipptl;
//    临时缓冲区
    char buff[BUFSIZ];
    
    SocketClient(string ip,int port,bool ipv4){
        send_addr.SetIP(ip);
        send_addr.SetPort(port);
        if(ipv4){
            ipptl = PF_INET;
            send_addr.SetIpv4();
        }
        else{
            ipptl = PF_INET6;
            send_addr.SetIpv6();
        }
    }
    SocketClient(sockaddr_in &taddr,bool ipv4){
        send_addr.SetSockAddr(taddr);
        if(ipv4){
            ipptl = PF_INET;
            send_addr.SetIpv4();
        }
        else{
            ipptl = PF_INET6;
            send_addr.SetIpv6();
        }
    }
    ~SocketClient(){
        //close(client_sfd);
    }
//    接受储存简单字符串
    virtual void Send(string buff) = 0;
//    接受储存二进制串
    virtual ssize_t SendRAW(char *buff, unsigned long size) = 0;
//    重新设置发送目的地的端口
    void SetSendPort(int port);
//    重新设置发送目的地的IP地址
    void SetSendIP(string ip);
//    共享设置发送地址相关信息管理结构
    void SetSendSockAddr(struct sockaddr_in);
};








//TCP服务器套接字类
class SocketTCPCServer:public SocketServer{
//    连接操作柄
    int data_sfd;
    void (*func)(class Socket &,int ,Addr);
    Addr client_addr;
    
public:
    SocketTCPCServer(int port):SocketServer(port,true){
//        获得套接字操作柄
        server_sfd = socket(ipptl,SOCK_STREAM, 0);
        if(!~server_sfd) throw "fail to get server sfd";
//        绑定IP地址与端口
        if(!~bind(server_sfd, server_addr.RawObj(), server_addr.Size())) throw "fail to bind";

		//设置超时
		struct timeval timeout = { 3,0 };
		setsockopt(server_sfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(struct timeval));

    }
    SocketTCPCServer(void):SocketServer(){
        data_sfd = -1;
    }
//    监听端口
    int Listen(void);
//    接受连接
    void Accept(void);
    Addr &GetAddr(void);
    Addr &GetClientAddr(void);
    int GetDataSFD(void);
    void SetDataSFD(int tdata_sfd);
    void SetClientAddr(Addr &caddr);
    void CloseConnection(void);
	void Close(void);
//    接收简单字符串数据
    ssize_t Recv(string &str);
//    接受储存二进制串
    ssize_t RecvRAW(char **p_rdt, Addr &taddr);
    ssize_t RecvRAW_SM(char **p_rdt, Addr &taddr);
    void SendRespond(string &str);

};

//TCP客户端套接字类
class SocketTCPClient:public SocketClient{
public:
//    连接操作柄
    int ctn_sfd;
    SocketTCPClient(string ip,int port):SocketClient(ip,port,true){
//        获得套接字操作柄
        client_sfd = socket(ipptl,SOCK_STREAM,0);
        if(!~client_sfd) throw "fail to get client sfd";
//        建立TCP连接
        if(!~connect(client_sfd,send_addr.RawObj(),send_addr.Size())) throw "fail to connect";
    }
    SocketTCPClient(sockaddr_in &taddr):SocketClient(taddr,true){
//        获得套接字操作柄
        client_sfd = socket(ipptl,SOCK_STREAM,0);
        if(!~client_sfd) throw "fail to get client sfd";
        //send_addr.SetIP("127.0.0.1");
        //send_addr.SetPort(9053);
//        建立TCP连接
        if(connect(client_sfd,send_addr.RawObj(),send_addr.Size()) < 0) throw "fail to connect";
		struct timeval timeout = { 3,0 };
		//设置超时
		setsockopt(client_sfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(struct timeval));
    }
//    发送简单字符串数据
    void Send(string str);
    void Reconnect(void);
    Addr &GetAddr(void);
    void SetAddr(Addr &);
    ssize_t SendRAW(char *buff, unsigned long size);
    ssize_t RecvRAW(char **p_rdt, Addr &taddr);
    
    void GetRespond(string &str);
    void Close(void);
};





//UDP服务端套接字类
class SocketUDPServer:public SocketServer{
//    是否设置非阻塞
    bool set_fcntl = false;
public:
    SocketUDPServer(int port):SocketServer(port,true){
//        获得套接字操作柄
        server_sfd = socket(ipptl,SOCK_DGRAM,0);
        if(!~server_sfd) throw "fail to get server sfd";
//        绑定IP地址与端口
        if(!~bind(server_sfd, server_addr.RawObj(), server_addr.Size())) throw "fail to bind";
        
//        设置非阻塞
        //int flags = fcntl(server_sfd, F_GETFL, 0);
        //fcntl(server_sfd, F_SETFL, flags | O_NONBLOCK);
    }
//    接受储存简单字符串信息的数据包
    ssize_t Recv(string &str);
//    接受储存二进制信息的数据包
    ssize_t RecvRAW(char **p_rdt, Addr &taddr);
//    设置非阻塞模式
    void UDPSetFCNTL(void);
};

//UDP客户端套接字类
class SocketUDPClient:public SocketClient{
public:
    SocketUDPClient(string ip,int port):SocketClient(ip,port,true){
//        获得套接字操作柄
        client_sfd = socket(ipptl,SOCK_DGRAM,0);
        if(!~client_sfd) throw "fail to get client sfd";
    }
//    发送简单字符串数据
    void Send(string buff);
//    发送一个带有二进制原始信息的数据包
    ssize_t SendRAW(char *buff, unsigned long size);
};

#endif /* net_h */