代码来源是《TCP/IP》网络编程
服务器代码
#include<stdio.h>
#include<stdlib.h>
#include<winsock.h>
#pragma comment(lib, "ws2_32.lib")
void ErrorHandling(char* message);
int main(int argc,char* argv[])
{
WSADATA wsaData;
SOCKET hServSock, hClnSock;
SOCKADDR_IN servAddr, cIntAddr;
int szClntAddr;
char message[] = "Hello World";
if (argc != 2)
{
printf("Usage : %s <port>\n", argv[0]);
exit(1);
}
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
ErrorHandling("WSAStartup() error!");
hServSock = socket(PF_INET, SOCK_STREAM, 0);
if (hServSock == INVALID_SOCKET)
ErrorHandling("socket() error!");
memset(&servAddr, 0, sizeof(servAddr));
servAddr.sin_family = AF_INET;
servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
servAddr.sin_port = htons(atoi(argv[1]));
if (bind(hServSock, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
ErrorHandling("bind() error");
if (listen(hServSock, 5) == SOCKET_ERROR)
ErrorHandling("listen() error");
szClntAddr = sizeof(szClntAddr);
hClnSock = accept(hServSock, (SOCKADDR*)&szClntAddr, &szClntAddr);
if (hClnSock == INVALID_SOCKET)
ErrorHandling("accept() error");
send(hClnSock, message, sizeof(message), 0);
closesocket(hClnSock);
closesocket(hServSock);
WSACleanup();
return 0;
}
void ErrorHandling(char* message)
{
fputs(message, stderr);
fputs("\n", stderr);
exit(1);
}
客户端代码
int main(int argc, char* argv[])
{
WSADATA wsaData;
SOCKET hSocket;
SOCKADDR_IN servAddr;
char message[30];
int strLen = 0;
int idx = 0, readLen = 0;
if (argc != 3)
{
printf("Usage : % s <IP><port>\n", argv[0]);
exit(1);
}
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
ErrorHandling("WSAStartup() error!");
hSocket = socket(PF_INET, SOCK_STREAM, 0);
if (hSocket == INVALID_SOCKET)
ErrorHandling("socket() error!");
memset(&servAddr, 0, sizeof(servAddr));
servAddr.sin_family = AF_INET;
servAddr.sin_addr.s_addr = inet_addr(argv[1]);
servAddr.sin_port = htons(atoi(argv[2]));
if (connect(hSocket, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
ErrorHandling("connect() error");
while (readLen = recv(hSocket, &message[idx++], 1, 0))
{
if (readLen == 1)
{
ErrorHandling("read() error");
}
strLen += readLen;
if (message[idx - 1] == '\0')
break;
}
printf("Message form server; %s\n", message);
printf("function read call count:%d", strLen);
closesocket(hSocket);
WSACleanup();
return 0;
}
void ErrorHandling(char* message)
{
fputs(message, stderr);
fputs("\n", stderr);
exit(1);
}
代码经过调试已无问题但实现结果与书本不同