写在前面
这篇文章由博主的复习笔记整理而成,希望对读者有所帮助。
TCP/IP协议栈简介
现代网络通信建立在TCP/IP协议栈基础上,这个协议栈采用分层设计,每层都有明确的职责(从下到上):
-
链路层(物理层)
-
负责物理连接的标准定义
-
定义了各种网络类型:
- LAN(局域网)
- WAN(广域网)
- MAN(城域网)
-
处理硬件寻址(如MAC地址)
-
-
IP层(网络层)
-
解决数据包在网络中的路由选择问题
-
特点:
- 面向消息(无连接)
- 不可靠(不保证送达)
- 路径可能不一致(不同数据包可能走不同路径)
-
-
传输层(TCP/UDP层)
-
TCP:
- 面向连接
- 可靠传输(通过确认和重传机制)
- 保证数据顺序
-
UDP:
- 无连接
- 不可靠但高效
- 不保证顺序
-
-
应用层
- 程序员直接操作的层级
- 使用套接字API进行网络编程
- 包含各种应用协议(HTTP、FTP等)
作为程序员,我们是在应用层上编写代码的。
TCP类型的服务器和客户端的实现
TCP服务器默认的函数调用顺序
- socket创建套接字
- bind分配套接字地址
- listen等待连接请求状态,此时客户端才可以调用connect函数
- accept允许连接
- read/write进行数据交换
- close断开连接
listen函数第二个参数指定了连接请求等待队列的大小。listen函数第一个参数是套接字描述符,用于接收客户端连接请求的网络数据,此时的套接字控制着谁能连接服务器。
**accept函数会自动创建套接字以处理客户端发送的数据。**它处理连接请求等待队列中的客户端连接请求,并返回用于数据的I/O套接字的文件描述符,这个套接字会自动与发起连接请求的客户端建立连接并交换数据。实际上,在服务器代码中我们手动创建的套接字只负责处理客户端的连接请求,而不负责与客户端进行数据交换。与客户端进行数据交换的套接字是accept函数自动创建的,具体的值是该函数的返回值。
接下来给出示例代码:
#include <stdio.h>
#include <stdlib.h>
#include <WinSock2.h>
void ErrorHandling(char* message)
{
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}
int main(int argc, char* argv[])
{
WSADATA wsaData;
SOCKET hServSock, hClntSock;
SOCKADDR_IN servAddr, clntAddr;
int szClntAddr;
char message[] = "Hello World!";
if (argc != 2)
{
printf("Usage : %s <port>\n", argv[0]);
exit(1);
}
//调用WSAStartup函数来初始化Winsock库
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
{
ErrorHandling(const_cast<char*>("WSAStartup() error"));
}
//调用socket函数创建一个套接字
hServSock = socket(PF_INET, SOCK_STREAM, 0);
if (hServSock == INVALID_SOCKET)
{
ErrorHandling(const_cast<char*>("socket() error"));
}
memset(&servAddr, 0, sizeof(servAddr));
servAddr.sin_family = AF_INET;
//htonl将主机字节序转换为网络字节序
servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
//htons将主机字节序转换为网络字节序
servAddr.sin_port = htons(atoi(argv[1]));
//调用bind函数将套接字与本地地址和端口绑定
if (bind(hServSock, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
{
ErrorHandling(const_cast<char*>("bind() error"));
}
//调用listen函数将套接字设置为监听状态
if (listen(hServSock, 5) == SOCKET_ERROR)
{
ErrorHandling(const_cast<char*>("listen() error"));
}
szClntAddr = sizeof(clntAddr);
//调用accept函数接受客户端的连接请求
hClntSock = accept(hServSock, (SOCKADDR*)&clntAddr, &szClntAddr);
if (hClntSock == INVALID_SOCKET)
{
ErrorHandling(const_cast<char*>("accept() error"));
}
//发送消息给客户端
send(hClntSock, message, sizeof(message), 0);
closesocket(hClntSock);
closesocket(hServSock);
WSACleanup();
return 0;
}
TCP客户端默认函数调用顺序
- socket
- connect
- read/write
- close
客户端调用connect函数后,发生以下情况之一才会返回:
- 服务器端收到连接请求,这并不意味着服务器端调用了accept函数,而是服务器把客户端的连接请求信息放到等待队列。
- 发生断网等异常情况而中断连接请求。
connect函数返回后并不会立即进行数据交换。客户端的套接字地址信息在调用connect函数后,由操作系统内核自动分配。
客户端只有在服务端调用了listen函数之后才可以调用connect函数。
下面是示例代码:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <winsock2.h>
#include <ws2tcpip.h>
void error(const char* msg) {
perror(msg);
exit(1);
}
int main(int argc, char* argv[])
{
if (argc != 3)
{
fprintf(stderr, "usage: %s <hostname> <port>\n", argv[0]);
exit(1);
}
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
error("WSAStartup failed");
}
SOCKET hSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (hSocket == INVALID_SOCKET)
{
error("socket failed");
}
SOCKADDR_IN serverAddr;
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(atoi(argv[2]));
inet_pton(AF_INET, argv[1], &serverAddr.sin_addr.s_addr);
if (connect(hSocket, (SOCKADDR*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR)
{
error("connect failed");
}
char message[30];
int strLen = 0;
int idx = 0, readLen = 0;
while (readLen = recv(hSocket, message, sizeof(message) - 1, 0))
{
if (readLen == -1)
{
error("recv failed");
}
strLen += readLen;
}
printf("message from server: %s\n", message);
printf("function recv(0 called %d times)\n", strLen);
closesocket(hSocket);
WSACleanup();
return 0;
}
使服务器循环接收客户端请求
迭代服务器端,调用listen函数之后,在循环语句里调用accept函数、close函数。
while(1) {
// 接受新连接
int clnt_sock = accept(serv_sock, ...);
// 处理客户端请求
handle_client(clnt_sock);
// 关闭连接
close(clnt_sock);
}
注意事项总结
连接建立时机:
- 客户端connect成功仅表示请求进入服务器队列
- 实际连接在服务器accept后才完全建立
套接字分工:
- 监听套接字:专门接收新连接
- 数据套接字:处理具体通信
资源管理:
- 及时关闭不再使用的套接字
- 注意端口复用问题
错误处理:
- 所有网络操作都应检查返回值
- 考虑超时和重试机制