Linux TCP套接字编程

套接字概念

在Linux环境下，socket用于表示进程间网络通信的特殊文件类型。本质为内核借助缓冲区形成的伪文件。既然是文件，那么理所当然的，我们可以使用文件描述符引用套接字。与管道类似的，Linux系统将其封装成文件的目的是为了统一接口，使得读写套接字和读写文件的操作一致。区别是管道主要应用于本地进程间通信，而套接字多应用于网络进程间数据的传递。
在网络通信中，套接字一定是成对出现的。一端的发送缓冲区对应对端的接收缓冲区。我们使用同一个文件描述符索发送缓冲区和接收缓冲区。

---

socket创建流程图

---

socket函数

#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
int socket(int domain, int type, int protocol);

domain:
    AF_INET 这是大多数用来产生socket的协议，使用TCP或UDP来传输，用IPv4的地址
    AF_INET6 与上面类似，不过是来用IPv6的地址
    AF_UNIX 本地协议，使用在Unix和Linux系统上，一般都是当客户端和服务器在同一台及其上的时候使用

type:
    SOCK_STREAM 这个协议是按照顺序的、可靠的、数据完整的基于字节流的连接。这是一个使用最多的socket类型，这个socket是使用TCP来进行传输。
    SOCK_DGRAM 这个协议是无连接的、固定长度的传输调用。该协议是不可靠的，使用UDP来进行它的连接。
    SOCK_SEQPACKET该协议是双线路的、可靠的连接，发送固定长度的数据包进行传输。必须把这个包完整的接受才能进行读取。
    SOCK_RAW socket类型提供单一的网络访问，这个socket类型使用ICMP公共协议。（ping、traceroute使用该协议）
    SOCK_RDM 这个类型是很少使用的，在大部分的操作系统上没有实现，它是提供给数据链路层使用，不保证数据包的顺序

protocol:
    传0 表示使用默认协议。

返回值：
    成功：返回指向新创建的socket的文件描述符，失败：返回-1，设置errno

socket函数在成功时返回一个小的非负整数值，我们称他为套接字描述符（socketfd）。为了得到这个套接字描述符，我们制订了协议族（domain）和套接字类型（type）。

bind函数

#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

sockfd：       socket文件描述符

addr:          构造出IP地址加端口号

addrlen:       sizeof(addr)长度

返回值：        成功返回0，失败返回-1, 设置errno

服务器程序所监听的网络地址和端口号通常是固定不变的，客户端程序得知服务器程序的地址和端口号后就可以向服务器发起连接，因此服务器需要调用bind绑定一个固定的网络地址和端口号。
bind()的作用是将参数sockfd和addr绑定在一起，使sockfd这个用于网络通讯的文件描述符监听addr所描述的地址和端口号。
struct sockaddr

struct sockaddr_in servaddr;
bzero(&servaddr, sizeof(servaddr));  // 将结构体清零
servaddr.sin_family = AF_INET;  // 设置地址类型为AF_INET
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);  // 网络地址为INADDR_ANY，这个宏表示本地的任意IP地址
servaddr.sin_port = htons(5678);  // 设置端口号为5678

listen函数

#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
int listen(int sockfd, int backlog);

sockfd:     socket文件描述符
backlog:    排队建立3次握手队列和刚刚建立3次握手队列的链接数和

典型的服务器程序可以同时服务于多个客户端，当有客户端发起连接时，服务器调用的accept()返回并接受这个连接，如果有大量的客户端发起连接而服务器来不及处理，尚未accept的客户端就处于连接等待状态，listen()声明sockfd处于监听状态，并且最多允许有backlog个客户端处于连接待状态，如果接收到更多的连接请求就忽略。根据TCP状态转换与，调用listen导致套接字从CLOSED状态转换到LISTEN状态。

accept函数

#include <sys/types.h>      /* See NOTES */
#include <sys/socket.h>
int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

sockdf:     socket文件描述符

addr:       传出参数，返回链接客户端地址信息，含IP地址和端口号

addrlen:    传入传出参数（值-结果）,传入sizeof(addr)大小，函数返回时返回真正接收到地址结构体的大小

返回值：      成功返回一个新的socket文件描述符，用于和客户端通信，失败返回-1，设置errno

accept函数由TCP服务器调用，用于从已完成链接队列队头返回下一个已完成连接。如果完成连接队列为空，就阻塞等待直到有客户端连接上来。addr是一个传出参数，accept()返回时传出客户端的地址和端口号。addrlen参数是一个传入传出参数，传入的是调用者提供的缓冲区addr的长度以避免缓冲区溢出问题，传出的是客户端地址结构体的实际长度（有可能没有占满调用者提供的缓冲区）。如果给addr参数传NULL，表示不关心客户端的地址。

connect函数

#include <sys/types.h>                  /* See NOTES */
#include <sys/socket.h>
int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

sockdf:     socket文件描述符
addr:       传入参数，指定服务器端地址信息，含IP地址和端口号
addrlen:    传入参数,传入sizeof(addr)大小
返回值：      成功返回0，失败返回-1，设置errno

客户端需要调用connect()连接服务器，connect和bind的参数形式一致，区别在于bind的参数是自己的地址，而connect的参数是对方的地址。并且调用connect将会激发TCP三次握手过程，并在连接建立成功或出错时才返回。

---

socket服务器示例

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

int main()
{
    int sfd, cfd;
    int len, i;
    char buf[1024];
    memset(buf, 0 ,sizeof(buf));
    struct sockaddr_in saddr, caddr;
    sfd = socket(AF_INET, SOCK_STREAM, 0);

    bzero(&saddr, sizeof(saddr));
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = htonl(INADDR_ANY);
    saddr.sin_port = htons(6666);

    bind(sfd, (struct sockaddr*)&saddr, sizeof(saddr));

    listen(sfd,64);

    printf("wait connect\n");
    int clen = sizeof(caddr);
    cfd = accept(sfd, (struct sockaddr*)&caddr, &clen);

    while(1)
    {
        len = read(cfd, buf, sizeof(buf));
        write(STDOUT_FILENO, buf, len);
        for(i = 0;i<len;++i)
        {
            buf[i] = toupper(buf[i]);
        }

        write(cfd, buf, len);
    }

    close(sfd);
    close(cfd);

    return 0;
}

socket客户端示例

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

int main()
{
    int sfd, len;
    char buf[1024];
    memset(buf, 0, sizeof(buf));
    sfd = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in saddr;
    bzero(&saddr, sizeof(saddr));
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = htonl(INADDR_ANY);
    saddr.sin_port = htons(6666);

    connect(sfd, (struct sockaddr*)&saddr, sizeof(saddr));

    while(1)
    {
        fgets(buf,sizeof(buf),stdin);
        write(sfd,buf,strlen(buf));

        len = read(sfd, buf, sizeof(buf));
        write(STDOUT_FILENO, buf, len);
    }

    close(sfd);

    return 0;
}

---

附件

TCP状态转换图