就以一个简单的基于TCP协议的服务器程序记录一下今天的学习结果吧。此服务器的功能是接受客户机的连接后,将来自客户机的消息显示并echo给客户机。整个程序很简单,主要是记录一下程序的流程。
#include <WinSock2.h>
#include <stdio.h>
#include <Windows.h>
/* 要使用winsock API,就要包含ws2_32.lib这个库 */
#pragma comment(lib, "ws2_32.lib")
int main(int argc, char* argv[])
{
//判断输入参数是否正确
if(argc != 2)
{
printf("Usage:%s PortNumber\n", argv[0]);
exit(-1);
}
//为端口号赋值
short port;
if((port = atoi(argv[1])) == 0)
{
printf("portnumber wrong\n");
exit(-1);
}
/* 第一步:初始化(注册)Winsock.dll */
WSADATA wsaData;
//init Winsock.dll
if (WSAStartup(MAKEWORD(1,1), &wsaData) != 0)
{
printf("fail to init winsock.dll!\n");
exit(-1);
}
/* 第二步:创建监听套接字 */
//creat socket
SOCKET servSoc;
if ((servSoc = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)
{
printf("create socket error!\n");
exit(-1);
}
/* 填写一些服务器的地址信息(三元组) */
struct sockaddr_in servAddr;
memset(&servAddr, 0, sizeof(sockaddr_in));
servAddr.sin_family = AF_INET;
servAddr.sin_port = htons(port);
servAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
/* 第三步:绑定地址到监听套接字 */
//bind
if (bind(servSoc, (sockaddr*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
{
printf("bind error! port:%d\n",port);
exit(-1);
}
/* 第四步:开始监听(最多允许2个客户机连接) */
//listen
if (listen(servSoc, 2) == SOCKET_ERROR)
{
printf("listen error!\n");
exit(-1);
}
printf("Server %d is listening...\n", port);
/* 创建连接套接字,负责和客户端通信
* 初始化客户端地址信息
*/
SOCKET clientSoc;
struct sockaddr_in clientAddr;
memset(&clientAddr, 0, sizeof(clientAddr));
int addrlen = sizeof(clientAddr);
/* 第五步:接受客户端的连接 */
//accept
if ((clientSoc = accept(servSoc, (sockaddr*)&clientAddr, &addrlen)) == INVALID_SOCKET)
{
printf("accept error!\n");
exit(-1);
}
printf("Accept connection from %s\n", inet_ntoa(clientAddr.sin_addr));
/* 第六步:数据交互 */
char buf[4096];
while (1)
{
int bytes;
if ((bytes = recv(clientSoc, buf, sizeof(buf), 0)) == SOCKET_ERROR)
{
printf("recv error!\n");
exit(-1);
}
buf[bytes] = '\0';
printf("Message from %s:%s\n", inet_ntoa(clientAddr.sin_addr), buf);
if (send(clientSoc, buf, bytes, 0) == SOCKET_ERROR)
{
printf("send error!\n");
exit(-1);
}
}
/* 第七步:清除套接字,释放该条连接所占用的全部资源 */
//clean socket
WSACleanup();
return 0;
}
转载于:https://blog.51cto.com/nyc1991/1250071