客户端:
#include <stdio.h>
#include <winsock2.h>
#include <iostream>
using namespace std;
#pragma comment(lib, "ws2_32.lib")
const char SERVER[] = "172.18.9.71";
const int PORT = 5555;
const int MSGSIZE = 1024;
int main()
{
WSADATA wsaData;
SOCKET sClient;
SOCKADDR_IN server;
char szMessage[MSGSIZE];
int ret;
WSAStartup(0x0202, &wsaData);
sClient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
memset(&server, 0, sizeof(SOCKADDR_IN));
server.sin_family = AF_INET;
server.sin_addr.S_un.S_addr = inet_addr(SERVER);
server.sin_port = htons(PORT);
cout << "Waiting for connect!" << endl;
if (connect(sClient, (struct sockaddr*)(&server), sizeof(SOCKADDR_IN)) == 0)
{
cout << "connect!!~" << endl;
while (TRUE)
{
cout << "send:" << endl;
cin >> szMessage;
send(sClient, szMessage, strlen(szMessage), 0);
ret = recv(sClient, szMessage, MSGSIZE, 0);
szMessage[ret] = '\0';
cout << "receive: " << szMessage << endl;
}
}
closesocket(sClient);
WSACleanup();
return 0;
}
服务端:
#include <stdio.h>
#include <process.h>
#include <winsock.h>
#include <iostream>
using namespace std;
#pragma comment(lib, "ws2_32.lib")
const char SERVER[] = "172.18.9.71";
const int PORT = 5555;
const int MSGSIZE = 1024;
int cnt = 0;
SOCKET sockets[FD_SETSIZE];
DWORD WINAPI WorkerThread(LPVOID lpParam)
{
fd_set fdread;
int ret;
struct timeval tv = { 1, 0 };
char szMessage[MSGSIZE];
while (TRUE)
{
FD_ZERO(&fdread);
for (int i = 0; i < cnt; i++)
{
FD_SET(sockets[i], &fdread);//将套接口加入集合中
}
//检查可读性的套接口
ret = select(0, &fdread, NULL, NULL, &tv);
if (ret == 0)
{
cout << "ret == 0?" << endl;
continue;
}
for (int i = 0; i < cnt; i++)
{
if (FD_ISSET(sockets[i], &fdread)) //如果可读, 奶蛋的,写错了。
{
ret = recv(sockets[i], szMessage, MSGSIZE, NULL);
if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
{
printf("Client socket %d closed.\n", sockets[i]);
closesocket(sockets[i]);
if (i < cnt - 1)
{
sockets[i--] = sockets[--cnt];
}
}
else
{
szMessage[ret] = '\0';
cout << "The message that client send is :" << szMessage << endl;
char szBuffer[1001];
sprintf(szBuffer, "I have bean received your messages, %s", szMessage);
send(sockets[i], szBuffer, strlen(szBuffer), NULL);
}
}
}
}
return 0;
}
int main()
{
WSADATA wsaData; //WSADATA framework 用于存储 Windows 套接字调用返回的初始化信息
SOCKET sListen, sClient;
SOCKADDR_IN local, client;
char szMessage[MSGSIZE];
int ret;
DWORD dwThreadId;
int iaddrSize = sizeof(SOCKADDR_IN);
WSAStartup(0x0202, &wsaData); //初始化ws2_32.lib
//create listening socket
sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
//bind
local.sin_family = AF_INET;
local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
local.sin_port = htons(PORT);
if (bind(sListen, reinterpret_cast<sockaddr*>(&local), sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
{
cout << "bind failed!" << endl;
}
//listen
if (listen(sListen, 3) == SOCKET_ERROR)
{
cout << "listen failed!" << endl;
}
//create WorkerThread
CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);
cout << "PID = " << dwThreadId << endl;
while (TRUE)
{
//accept a connection
sClient = accept(sListen, reinterpret_cast<sockaddr*>(&client), &iaddrSize);
if (sClient == INVALID_SOCKET)
{
cout << "accept error!" << endl;
break;
}
else
{
printf("Accept client: %s: %d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
//加入到socket的数组中
sockets[cnt++] = sClient;
}
}
closesocket(sListen);
WSACleanup();
return 0;
}