linux下对socket的简单封装

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include"fcntl.h"
#include"sys/types.h"
#include"sys/stat.h"
#include <sys/types.h> /* basic system data types */
#include <sys/socket.h> /* basic socket definitions */
#include <sys/time.h> /* timeval{} for select() */
#include <netinet/in.h> /* sockaddr_in{} and other Internet defns */
#include <arpa/inet.h> /* inet(3) functions */
#include <errno.h>
#include <netdb.h>
#include <sys/time.h>
#include <limits.h>
#include <net/if_arp.h>
#include <net/if.h>
#include <netinet/tcp.h>
#include <limits.h>
#define BACKLOG 10
int server_start(int port)
{
    struct hostent *host;
    struct sockaddr_in serv_addr;
    int sockfd;
    /*
        获取ip地址信息
    */
/*    if((host=gethostbyname(server))==NULL){
        errorlogs("gethostbyname");
        return -1;
    }*/
    
    serv_addr.sin_family=AF_INET;//地址族

    serv_addr.sin_port=htons(port);//端口号

    serv_addr.sin_addr.s_addr=INADDR_ANY;//表示机器上的所用网络地址

    bzero(&(serv_addr.sin_zero),8);//填充0,保证与struct sockaddr同样大小

    /*
        建立socket连接
    */
    
    if((sockfd = socket(AF_INET,SOCK_STREAM,0))==-1){
        errorlogs("socket server start");
        return -1;
    }
    logs("socket create success");
    
    //设置sockets发送/接受缓存大小

    /*
     must setting it before listen or accept
    */
    /*
    int buffsize = 81960; //default value is 8196
    setsockopt(sockfd ,SOL_SOCKET,SO_SNDBUF,(char *)&buffsize,sizeof(buffsize));
    setsockopt(sockfd ,SOL_SOCKET,SO_RCVBUF,(char *)&buffsize,sizeof(buffsize));
    */
    
    /*
        与网卡地址绑定
    */
    if(bind(sockfd,(struct sockaddr *)&serv_addr,sizeof(struct sockaddr))==-1){
        errorlogs("bind");
        return -1;
    }
    logs("bind success");
    /*
        侦听连接
    */
    if(listen(sockfd,BACKLOG)==-1){
        errorlogs("listen");
        return -1;
    }
    logs("listening . . .");
    return sockfd;
}
int get_connect(int sockfd)
{
    fd_set readfd;
    struct sockaddr_in client_sockaddr;
    /*
        初始化文件描述符集
    */
    //FD_ZERO(&readfd);

    //FD_SET(sockfd,&readfd);

    int sin_size;
    int client_fd;
    sin_size=sizeof(struct sockaddr_in);
    while(1){
        /*
            调用select函数,一直等待,直到有连接到达
        */
//        if(select(sockfd+1,&readfd,NULL,NULL,(struct timeval *)0)>0){

            /*
                描述符集有变化,有新的连接
            */
//            if(FD_ISSET(sockfd,&readfd)>0){

                /*
                    接受连接
                */
                if((client_fd=accept(sockfd,(struct sockaddr *)&client_sockaddr,&sin_size))==-1){
                    errorlogs("accept");
                    return -1;
                }
                return client_fd;
    }/*while*/
}
int server_connect(char *server,int port)
{
    struct hostent *host;
    struct sockaddr_in serv_addr;
    int sockfd;
    /*
        获取ip地址信息
    */
    printf("%s:%d\n",server,port);
    if((host=gethostbyname(server))==NULL){
        errorlogs("gethostbyname");
        return -1;
    }
    serv_addr.sin_family=AF_INET;//地址族

    serv_addr.sin_port=htons(port);//端口号

    serv_addr.sin_addr=*((struct in_addr *)host->h_addr);//IP地址,host->h_addr返回h_addr_list中的第一地址

    bzero(&(serv_addr.sin_zero),8);//填充0,保证与struct sockaddr同样大小

    /*
        建立socket连接
    */
    
    if((sockfd = socket(AF_INET,SOCK_STREAM,0))==-1){
        errorlogs("socket create");
        return -1;
    }
    logs("socket create success");
    
    /*
    int buffsize = 81960; //default value is 8196
    setsockopt(sockfd ,SOL_SOCKET,SO_SNDBUF,(char *)&buffsize,sizeof(buffsize));
    setsockopt(sockfd ,SOL_SOCKET,SO_RCVBUF,(char *)&buffsize,sizeof(buffsize));
    */
    /*
        尝试连接远程服务器
    */
    if(connect(sockfd,(struct sockaddr *)&serv_addr,sizeof(struct sockaddr))==-1){
        errorlogs("connect");
        close(sockfd);
        return -1;
    }
    return sockfd;
}
int data_recv_length_timeout(int sockfd,char *data,int length,int sec)
{
    int recvbytes,wait_length;
    int total=0,ret;    
    fd_set rdfds;
    struct timeval tv;
    FD_ZERO(&rdfds);
    FD_SET(sockfd, &rdfds);
    tv.tv_sec = sec;
    tv.tv_usec = 0;
    if(length==0)
        wait_length=USHRT_MAX;
    else
        wait_length=length;
    while(total<wait_length)
    {
        ret = select(sockfd + 1, &rdfds, NULL, NULL, &tv);
        if(ret < 0)
        {
            errorlogs("data_recv_length select error:");
            return -1;
        }else if(ret==0)
        {
            logs("data_recv_length select timeout");
            return 0;
        }else{
            if(FD_ISSET(sockfd, &rdfds))
            {
            
                if((recvbytes=recv(sockfd,data+total,wait_length-total,MSG_NOSIGNAL))==-1)
                {
                    errorlogs("ENGAIN recv");
                    return -1;
                    
                }else if(recvbytes==0){
                    logs("remote close the connect");
                    return -1;
                }
                if(length==0)
                    return recvbytes;
                total+=recvbytes;
                printf("total:%d\n",total);
            }
        }
    }
/*    if(DEBUG){
        char s[30];
        logs("***********************************\n");
        sprintf(s,"Recv %d BYTE data\n",recvbytes);
        logs(s);
        logs("***********************************\n");
    }*/
    return total;
}
//int data_send_timeout(int sockfd,char *data,int length,int sec)

int data_send_timeout(int sockfd,char *data,int length,int sec)
{
    int sendbytes;
    int total=0;
    fd_set rdfds;
    struct timeval tv;
    FD_ZERO(&rdfds);
    FD_SET(sockfd, &rdfds);
    tv.tv_sec = sec;
    tv.tv_usec = 0;
    int ret;
    /*
        发送数据直至完成
    */
    while(total<length){
    /*
        若操作系统一次没有发完,根据实际发送的字节数,发送剩余的
    */
        logs("wait select ...............");
        ret = select(sockfd + 1, NULL, &rdfds, NULL, &tv);
        if(ret < 0)
        {
            errorlogs("data_send select error:");
            return -1;
        }else if(ret==0)
        {
            logs("data_send select timeout");
            return -1;
        }else{
            if(FD_ISSET(sockfd, &rdfds)){
                if((sendbytes=send(sockfd,data+total,length-total,MSG_NOSIGNAL))==-1){
                    errorlogs("send");
                    return -1;
                }
                total+=sendbytes;                
                logs("size %d data already send!",sendbytes);                
            }else
            {
                logs("send function not run");
            }
        }
    }
    return total;
}
int main()
{
    
}

你可能感兴趣的:(linux下对socket的简单封装)