4.socket套接字及TCP的实现框架
Socket套接字
Socket套接字是网络编程中用于实现不同计算机之间通信的一个基本构建块。
在现代计算机网络中,Socket套接字扮演着至关重要的角色。它们为应用程序提供了一种方式,通过这种方式,程序能够通过网络发送和接收数据包。以下是对socket的相关介绍:
- 定义功能:
- 核心作用:
- 类型应用:
- 流式套接字(SOCK_STREAM)基于TCP协议,提供面向连接的可靠数据传输服务,适用于需要确保数据完整性的应用,如网页浏览。
- 数据报套接字(SOCK_DGRAM)基于UDP协议,提供无连接的服务,适用于对实时性要求高但可以容忍少量数据丢失的应用,如视频通话。
- 工作过程:
- 服务器端创建一个Socket并绑定到一个特定的IP地址和端口上,然后开始监听来自客户端的连接请求。
- 客户端创建一个Socket,并尝试连接到服务器的IP地址和端口。一旦连接建立,双方就可以通过这个Socket进行数据的发送和接收。
- 高级功能:
- 使用Socket,可以实现多种网络应用程序,包括聊天应用、文件传输、远程控制等。在多用户游戏、实时数据处理等方面也广泛应用Socket技术。
- 在多线程或多进程的环境中,Socket也能有效地处理并发连接,提高系统的响应速度和处理能力。
TCP实现框架
以下是C语言中常见的Socket API及其功能:
以下是C语言中常见的Socket API及其功能:
-
socket函数:该函数用于创建一个套接字,并返回一个套接字描述符。
#include <sys/types.h> #include <sys/socket.h> int socket(int domain, int type, int protocol); /* 参数:domain:指定协议族,如AF_INET表示IPv4地址。type:指定套接字类型,如SOCK_STREAM表示提供有序、可靠、双向字节流的连接。protocol:通常设置为0,表示系统将选择与指定类型相匹配的默认协议。 返回值:成功时返回一个非负整数,表示新创建的套接字描述符。失败时返回-1,并设置errno为相应的错误码。 */
-
bind函数:此函数用于将套接字绑定到一个特定的地址和端口号上。这对于服务器来说尤其重要,因为它们需要在已知的地址上监听来自客户端的连接请求。
#include <sys/types.h> #include <sys/socket.h> int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen); /* 参数:sockfd:表示套接字的描述符。addr:指向特定于域的套接字地址结构的指针,该结构包含了要绑定的地址信息。addrlen:指定地址结构的大小。 返回值:成功时返回0。失败时返回-1,并设置errno为相应的错误码。 */
-
listen函数:服务器使用此函数将套接字设置为监听状态,等待客户端的连接请求。可以指定一个队列限制,以控制同时可处理的连接数。
#include <sys/types.h> #include <sys/socket.h> int listen(int sockfd, int backlog); /* 参数:sockfd:表示套接字的描述符。backlog:指定在未完成连接队列中允许的最大连接数。 返回值:成功时返回0。失败时返回-1,并设置errno为相应的错误码。 */
-
accept函数:当有客户端连接请求时,服务器使用此函数接受连接,并返回一个新的套接字,用于与客户端通信。
#include <sys/types.h> #include <sys/socket.h> int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen); /* 参数:sockfd:表示套接字的描述符。addr:指向特定于域的套接字地址结构的指针,该结构用于接收客户端的地址信息。addrlen:指向一个变量的指针,该变量用于接收地址结构的大小。 返回值:成功时返回一个新的套接字描述符,用于与客户端通信。失败时返回-1,并设置errno为相应的错误码。 */
-
connect函数:该函数用于建立一个到服务器的连接。对于TCP套接字,这将触发TCP的三次握手过程。
#include <sys/types.h> #include <sys/socket.h> int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen); /* 参数:sockfd:表示套接字的描述符。addr:指向特定于域的套接字地址结构的指针,该结构包含了服务器的地址信息。addrlen:指定地址结构的大小。 返回值:成功时返回0。失败时返回-1,并设置errno为相应的错误码。 */
-
send和recv函数:这两个函数分别用于发送和接收数据。它们是TCP通信中常用的数据传输函数。
#include <sys/types.h> #include <sys/socket.h> ssize_t send(int sockfd, const void *buf, size_t len, int flags); ssize_t recv(int sockfd, void *buf, size_t len, int flags); /* 参数:sockfd:表示套接字的描述符。buf:指向要发送或接收数据的缓冲区的指针。len:指定要发送或接收的数据的长度。flags:可选的标志位,用于控制发送或接收的行为。 返回值:成功时返回实际发送或接收的字节数。失败时返回-1,并设置errno为相应的错误码。 */
-
sendto和recvfrom函数:对于UDP通信,不需要建立连接,直接使用sendto和recvfrom函数进行数据报文的发送和接收。
#include <sys/types.h> #include <sys/socket.h> ssize_t sendto(int sockfd, const void *buf, size_t len, int flags, const struct sockaddr *dest_addr, socklen_t addrlen); ssize_t recvfrom(int sockfd, void *buf, size_t len, int flags, struct sockaddr *src_addr, socklen_t *addrlen); /* 参数:sockfd:表示套接字的描述符。buf:指向要发送或接收数据的缓冲区的指针。len:指定要发送或接收的数据的长度。flags:可选的标志位,用于控制发送或接收的行为。dest_addr:指向目标地址结构的指针,用于UDP发送操作。src_addr:指向源地址结构的指针,用于UDP接收操作。addrlen:指向一个变量的指针,用于UDP接收操作,该变量用于接收源地址结构的大小。 返回值:成功时返回实际发送或接收的字节数。失败时返回-1,并设置errno为相应的错误码。 */
示例-服务端
TCP_socket.c
#include <stdio.h> // 引入标准输入输出库
#include <sys/socket.h> // 引入套接字库
#include <sys/types.h> // 引入类型定义库
#include <stdlib.h> // 引入标准库
#include <arpa/inet.h> // 引入互联网协议库
#include <unistd.h> // 引入Unix标准库
#include <string.h> #define PORT 5001 // 定义端口号为5001
#define BACKLOG 5 // 定义最大连接数为5int main(int argc, char *argv[]) // 主函数,参数为命令行参数个数和参数列表
{int fd,newfd,ret; // 定义文件描述符变量char buf[BUFSIZ] = {}; //BUFSIZ 8142struct sockaddr_in addr; // 定义IPv4地址结构体if(argc < 3){fprintf(stderr,"%s<addr><PORT>",argv[0]);}// 创建套接字fd = socket(AF_INET, SOCK_STREAM, 0);if(fd < 0){perror("socket"); exit(0); }// 设置地址结构体addr.sin_family = AF_INET; // 使用IPv4协议addr.sin_port = htons(atoi(argv[2])); // 设置端口号,atoi()将字符串转换为整数if(inet_aton(argv[1],&addr.sin_addr)==0){//inet_aton()函数将字符串形式的IP地址转换为网络字节序的二进制形式fprintf(stderr,"Invalid address\n");exit(EXIT_FAILURE);}// 绑定通信结构体if(bind(fd, (struct sockaddr *)&addr, sizeof(addr) ) == -1){perror("bind"); exit(0); }// 设置套接字为监听模式if(listen(fd, BACKLOG) == -1){perror("listen"); exit(0); }// 接受客户端的连接请求,生成新的用于和客户端通信的套接字newfd = accept(fd, NULL, NULL);if(newfd < 0){perror("accept"); exit(0);}/*循环接收*/while(1){memset(buf,0,BUFSIZ);ret = read(newfd,buf,BUFSIZ);if(ret < 0){perror("read");exit(0);}else if(ret == 0)break;elseprintf("buf = %s\n",buf);}close(newfd);close(fd); // 关闭套接字return 0;
}
示例-客户端
TCP_socket2.c
#include <stdio.h> // 引入标准输入输出库 #include <sys/socket.h> // 引入套接字库 #include <sys/types.h> // 引入类型定义库 #include <stdlib.h> // 引入标准库 #include <arpa/inet.h> // 引入互联网协议库 #include <unistd.h> // 引入Unix标准库 #include <string.h>#define PORT 5001 // 定义端口号为5001 #define BACKLOG 5 // 定义最大连接数为5 #define STR "Hello World!" // 定义要发送的字符串为"Hello World!"int main(int argc, char *argv[]) // 主函数,参数为命令行参数个数和参数列表 {int fd; // 定义文件描述符变量struct sockaddr_in addr; // 定义IPv4地址结构体char buf[BUFSIZ] = {}; //BUFSIZ 8142if(argc < 3){fprintf(stderr,"%s<addr><port>\n",argv[0]);exit(0);}// 创建套接字fd = socket(AF_INET, SOCK_STREAM, 0);if(fd < 0){perror("socket"); // 如果创建失败,打印错误信息exit(0); // 退出程序}// 设置地址结构体addr.sin_family = AF_INET; // 使用IPv4协议addr.sin_port = htons(atoi(argv[2])); // 设置端口号,atoi()将字符串转换为整数if(inet_aton(argv[1],&addr.sin_addr)==0){//inet_aton()函数将字符串形式的IP地址转换为网络字节序的二进制形式fprintf(stderr,"Invalid address\n");exit(EXIT_FAILURE);} // 向服务端发起连接请求if(connect(fd, (struct sockaddr *)&addr, sizeof(addr) ) == -1){perror("connect"); exit(0); }/*循环发送*/while(1){printf(">");fgets(buf,BUFSIZ,stdin);write(fd,buf,strlen(buf));}// 关闭套接字close(fd);return 0; // 返回0表示程序正常结束 }
通信成功
如果结束程序,再次运行服务端程序,会出现报错如下(地址已经被使用)
解决办法:
在程序中加入如下程序段:
/*地址快速重用*/ int flag=1,len= sizeof (int); if ( setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flag, len) == -1) { perror("setsockopt"); exit(1); }
5.TCP多进程并发
示例-服务端
my_sever.c
#include <stdio.h> // 引入标准输入输出库
#include <sys/socket.h> // 引入套接字库
#include <sys/types.h> // 引入类型定义库
#include <stdlib.h> // 引入标准库
#include <arpa/inet.h> // 引入网络地址库
#include <unistd.h> // 引入Unix系统调用库
#include <string.h> // 引入字符串处理库
#include <strings.h> // 引入字符串操作库
#include <signal.h> // 引入信号处理库
#include <sys/wait.h> // 引入进程等待库#define BACKLOG 5 // 定义最大连接数为5void ClinetHandle(int newfd); // 客户端处理函数/*信号处理函数,防止出现僵尸进程*/
void SigHandle(int sig){ if(sig == SIGCHLD){ // 如果接收到SIGCHLD信号printf("client exited\n"); // 打印客户端退出信息wait(NULL); // 等待子进程结束}
}/*主函数*/
int main(int argc, char *argv[])
{int fd, newfd; // 定义文件描述符变量struct sockaddr_in addr, clint_addr; // 定义套接字地址结构体socklen_t addrlen = sizeof(clint_addr); // 定义地址长度变量#if 0struct sigaction act; // 定义信号处理结构体act.sa_handler = SigHandle; // 设置信号处理函数act.sa_flags = SA_RESTART; // 设置信号处理标志sigemptyset(&act.sa_mask); // 清空信号屏蔽集sigaction(SIGCHLD, &act, NULL); // 注册信号处理函数
#elsesignal(SIGCHLD, SigHandle); // 注册信号处理函数
#endifpid_t pid; // 定义进程ID变量if(argc < 3){ // 如果参数个数小于3fprintf(stderr, "%s<addr><port>\n", argv[0]);exit(0); }/*创建套接字*/fd = socket(AF_INET, SOCK_STREAM, 0); // 创建TCP套接字if(fd < 0){ perror("socket"); }addr.sin_family = AF_INET; // 设置地址族为IPv4addr.sin_port = htons( atoi(argv[2]) ); // 设置端口号if ( inet_aton(argv[1], &addr.sin_addr) == 0) { // 如果转换IP地址失败fprintf(stderr, "Invalid address\n"); exit(EXIT_FAILURE); }/*地址快速重用*/int flag=1,len= sizeof (int); if ( setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flag, len) == -1) { // 如果设置失败perror("setsockopt"); exit(1); } /*绑定通信结构体*/if(bind(fd, (struct sockaddr *)&addr, sizeof(addr) ) == -1){ // 如果绑定失败perror("bind"); exit(0); }/*设置套接字为监听模式*/if(listen(fd, BACKLOG) == -1){ perror("listen"); exit(0); }while(1){ // 循环监听连接请求/*接受客户端的连接请求,生成新的用于和客户端通信的套接字*/newfd = accept(fd, (struct sockaddr *)&clint_addr, &addrlen); // 接受连接请求if(newfd < 0){ perror("accept"); exit(0); }/*打印客户端地址和端口号*/ // inet_ntoa()将网络字节序转换为点分十进制IP地址格式的字符串;// ntohs()将一个16位数从网络字节顺序转换为主机字节顺序printf("addr:%s port:%d\n", inet_ntoa(clint_addr.sin_addr), ntohs(clint_addr.sin_port) ); if( (pid = fork() ) < 0){ perror("fork"); exit(0); }else if(pid == 0){ // 如果当前进程是子进程close(fd); // 关闭父进程的文件描述符ClinetHandle(newfd); // 处理客户端请求exit(0); }elseclose(newfd); // 关闭子进程的文件描述符}close(fd); // 关闭服务器套接字return 0;
}/*客户端处理函数*/
void ClinetHandle(int newfd){int ret; // 定义返回值变量char buf[BUFSIZ] = {}; // 定义缓冲区数组并初始化为0while(1){ // 循环读取客户端数据//memset(buf, 0, BUFSIZ); // 清空缓冲区数组bzero(buf, BUFSIZ); // 使用bzero函数清空缓冲区数组ret = read(newfd, buf, BUFSIZ); // 从客户端读取数据if(ret < 0) {perror("read"); exit(0); }else if(ret == 0) // 如果读取到EOF标志break; // 跳出循环elseprintf("buf = %s\n", buf); // 打印读取到的数据}close(newfd); // 关闭与客户端的连接
}
示例-客户端
my_client.c
#include <stdio.h> // 引入标准输入输出库
#include <sys/socket.h> // 引入套接字库
#include <sys/types.h> // 引入类型定义库
#include <stdlib.h> // 引入标准库
#include <arpa/inet.h> // 引入互联网协议族库
#include <unistd.h> // 引入Unix标准库
#include <string.h> // 引入字符串处理库#define BACKLOG 5 // 定义最大连接数为5int main(int argc, char *argv[]) // 主函数,参数为命令行参数个数和参数列表
{int fd; // 定义文件描述符变量struct sockaddr_in addr; // 定义套接字地址结构体char buf[BUFSIZ] = {}; // 定义缓冲区数组并初始化为0if(argc < 3){ // 如果参数个数小于3fprintf(stderr, "%s<addr><port>\n", argv[0]); // 打印错误信息exit(0); // 退出程序}/*创建套接字*/fd = socket(AF_INET, SOCK_STREAM, 0); // 创建TCP套接字if(fd < 0){ perror("socket"); exit(0); }addr.sin_family = AF_INET; // 设置地址族为IPv4addr.sin_port = htons( atoi(argv[2]) ); // 设置端口号if ( inet_aton(argv[1], &addr.sin_addr) == 0) { fprintf(stderr, "Invalid address\n"); exit(EXIT_FAILURE); }/*向服务端发起连接请求*/if(connect(fd, (struct sockaddr *)&addr, sizeof(addr) ) == -1){ // 如果连接失败perror("connect"); exit(0); }while(1){ // 循环读取用户输入的数据printf("Input->"); // 提示用户输入fgets(buf, BUFSIZ, stdin); // 从标准输入读取数据到缓冲区write(fd, buf, strlen(buf) ); // 将缓冲区的数据发送给服务器}close(fd); // 关闭套接字return 0;
}