socket网络编程（Linux）

一、socket TCP网络编程总结

1、Linux中socket网络编程涉及到的函数有socket()函数、bind()函数、listen()函数、accept()函数、read()函数、recv()函数、write()函数、send()函数、connect()函数、close()函数。头文件用man手册查对应的函数就可以查。
2、Linux中socket网络编程涉及到的结构体有，struct sockaddr_in(用于IPV4)或者struct sockaddr_in6(用于IPV6)，并且在使用bind()函数、accept()函数和connect()函数时，要将sockaddr_in结构体类型的变量强制类型转换为(struct sockaddr*)&+变量名。另外结构体成员变量赋值时，需要用到inet_addr()函数和htons()函数，如下：

    struct sockaddr_in serv_addr;
    memset(&serv_addr, 0, sizeof(serv_addr));  //每个字节都用0填充
    serv_addr.sin_family = AF_INET;  //使用IPv4地址
    serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");  //具体的IP地址
    serv_addr.sin_port = htons(1234);  //端口
    bind(serv_sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr));

3、socket通信服务器端搭建：

• 创建套接字（socket()）
• 绑定IP和端口（bind()）
• 进入监听状态（listen()）
• 等待客户端发起连接（accept()）
• 收发数据（write()、send()\read()）
• 关闭套接字（close()）

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>

int main(int argc,char *argv[])
{
    if(argc!=2)
    {
        printf("Using:./server port\nExample:./server 5005\n\n");return -1;   
    }
    //first,create socket
    int listenfd;
    if((listenfd = socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("socket");
        return -1;
    }

    //second,bind ip and port
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;//ipv4
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);//任意ip，也可以用inet_addr()来指定具体的ip
    servaddr.sin_port = htons(atoi(argv[1]));
    if(bind(listenfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) != 0)
    {
        perror("bind");
        close(listenfd);
        return -1;
    }

    //third,listen
    if(listen(listenfd,5) != 0)//5表示请求队列中最多容纳5个客户端
    {
        perror("listen");
        close(listenfd);
        return -1;
    }

    //fourth,accept
    int clientfd;
    int socklen=sizeof(struct sockaddr_in);
    struct sockaddr_in clientaddr;
    clientfd=accept(listenfd,(struct sockaddr *)&clientaddr,(socklen_t *)&socklen);
    printf("client:(%s) have linked\n",inet_ntoa(clientaddr.sin_addr));

    //fifth,rev and send
    char buffer[1024];
    while(1)
    {
        int iret;
        memset(buffer,0,sizeof(buffer));
        if((iret = recv(clientfd,buffer,sizeof(buffer),0)) <= 0)
        {
            printf("iret = %d\n",iret);
            break;
        }
        printf("received:%s\n",buffer);
        strcpy(buffer,"i miss you!");
        if((iret = send(clientfd,buffer,strlen(buffer),0)) <= 0)
        {
            perror("send");
            break;
        }
        printf("send:%s\n",buffer);
    }

    //sixth,close
    close(listenfd);
    close(clientfd);
}

4、socket通信客户端搭建

• 创建套接字（socket()）
• 向服务器特定IP和端口发起连接请求（connect()）
• 收发数据（write()、read()）
• 关闭套接字（close()）

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>

int main(int argc,char *argv[])
{
    if(argc!=3)
    {
        printf("Using:./client ip port\nExample:./client 127.0.0.1 5005\n\n");
         return -1;
    }
    //1.socket
    int sockfd;
    if((sockfd = socket(AF_INET,SOCK_STREAM,0))== -1)//调用socket()函数返回套接字描述符
    {
        perror("client");
        return -1;
    }

    //2.connect
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;//使用ipv4协议族
    servaddr.sin_port =htons(atoi(argv[2]));//端口号
    servaddr.sin_addr.s_addr = inet_addr(argv[1]);//IP地址
    if(connect(sockfd,(struct sockaddr*)&servaddr,sizeof(servaddr))!= 0)
    {
        perror("connect");
        close(sockfd);
        return -1;
    }
    char buffer[1024];

    //3.write and read
    int i=0;
    while(1)
    {
        int iret;
        memset(buffer,0,sizeof(buffer));
        sprintf(buffer,"hello,you can trust me!%3d times\n",i);
        if((iret=send(sockfd,buffer,strlen(buffer),0))<=0)
        {
            perror("send");
            break;
        }
        printf("send: %s\n",buffer);
        memset(buffer,0,sizeof(buffer));
        if((iret=recv(sockfd,buffer,sizeof(buffer),0))<=0)
        {
            printf("iret = %d\n",iret);
            break;
        }
        printf("receive: %s\n",buffer);
        i++;
        sleep(2);
    }

    //4.colse
    close(sockfd);
    return 0;
}

ps：服务器运行到accept()时会阻塞，等待客户端发起连接。运行recv()时会一直等待接收报文，直到收到数据或者连接断开。需要特别注意的是，send()和recv()这两个函数并不能保证一定会根据你所指定的发送和接收的数据大小去拿去数据，特别明显的是当你调用recv()这个函数时，你想让它拿到1000个字节后再返回，可是实际上它可能就拿了700个字节就返回了，剩下的数据还在接收缓冲区中，需要再次调用recv()去获取剩下的数据。可以如我这样写一个函数，在函数中循环接收，直到接收到你指定大小的数据后在返回。

/*
*@description               :循环接收，直到接收完毕
*@param -fd                 :与对端通信的文件描述符
*@param -buf                :要接收数据的buf的起始位置
*@param -buf_size           :要接收数据的buf的大小
*@param -data_len           :要接收数据的长度
*@return                    :成功：ture; 失败：false
*/
bool recvall(int fd, char *buf, int buf_size, int data_len)
{
     if(fd < 0 || buf == NULL || data_len < 0) return false;
     if(buf_size < data_len) return false;

    int ret = 0;
    int reserve_len = data_len;
    while(reserve_len)
    {
        ret = recv(fd, buf+(data_len-reserve_len), reserve_len, 0);
        if(recv <= 0)
        {
            return false;
        }
        reserve_len = reserve_len - ret;
    }
    return true;
}

发送也是一样，在循环中发送，直到发送完你指定大小的数据，代码如下：

/*
*@description               :循环发送，直到发送完毕
*@param -fd                 :与对端通信的文件描述符
*@param -buf                :要发送数据的起始位置
*@param -data_len           :要发送数据的长度
*@return                    :成功：ture; 失败：false
*/
bool sendall(int fd, char *buf, int data_len)
{
    if(fd < 0 || buf == NULL || data_len < 0) return false;
    int ret = 0;
    int reserve_len = data_len;
    while(reserve_len)
    {
        ret = send(fd, buf+(data_len-reserve_len), reserve_len, 0);
        if(ret <= 0)
        {
            return false;
        }
        reserve_len = reserve_len - ret;
    }
    return true;
}

最后再说一点，当调用close()关闭服务器端的套接字时，会通知客户端此连接已关闭，随后客户端也会释放相应的资源，客户端也就无法再使用改套接字和服务器进行数据收发了。

二、参考文档和资料

1、socke网络编程入门
2、socke编程实现回声客户端
3、socket让服务器持续监听客户端的连接请求
4、socket编程实现文件传输功能