windows socket聊天程序 VC环境 用线程实现 适用初学者

//一对一聊天，用线程实现，一个线程负责发送消息，一个线程负责接收消息

//服务器端

#include <winsock2.h>

#include <windows.h>

#include <iostream.h>

#define PORT_SERVER 6666

#define NUM_CLIENTS 10

#pragma comment(lib, "ws2_32.lib")

void SendMsg(LPVOID lpParameter);

void RecvMsg(LPVOID lpParameter);

SOCKET sock_client;

int main(int argc, char *argv[])

{

cout << "Start server..." << endl;

WSADATA wsaData;

WORD sockVersion = MAKEWORD(2, 0);



if(WSAStartup(sockVersion, &wsaData) != 0)

{

cout << "Failed to retrive version." << endl;

return 0;

}



SOCKET sock_sev;

sock_sev = socket(AF_INET,SOCK_STREAM, IPPROTO_TCP);

if(sock_sev == INVALID_SOCKET)

{

cout << "Invalid socket." << endl;

WSACleanup();

return 0;

}

sockaddr_in addr_sev;

addr_sev.sin_family = AF_INET;

addr_sev.sin_port = htons(PORT_SERVER);

addr_sev.sin_addr.s_addr = INADDR_ANY;



if (SOCKET_ERROR == bind(sock_sev, (sockaddr *)&addr_sev, sizeof(addr_sev)))

{

cout << "Failed to bind." << endl;

WSACleanup();

return 0;

}

if(SOCKET_ERROR == listen(sock_sev, NUM_CLIENTS))

{

cout << "Failed to listen." << endl;

WSACleanup();

return 0;

}



//SOCKET sock_client;

sockaddr_in addr_client;

int nAddrLen = sizeof(addr_client);

sock_client = accept(sock_sev, (sockaddr *)&addr_client, &nAddrLen);



if (INVALID_SOCKET == sock_client)

{

cout << "Failed to accept." << endl;



}

HANDLE hThread1, hThread2;

hThread2=CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)RecvMsg, NULL, 0, NULL);

hThread1=CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)SendMsg, NULL, 0, NULL);





CloseHandle(SendMsg);

CloseHandle(RecvMsg);



Sleep(200000); //主线程沉睡200秒

//cout << "Connection from " << inet_ntoa(addr_client.sin_addr) << endl;

closesocket(sock_client);

return 1;

}

void SendMsg(LPVOID lpParameter)

{

//cout << "进入发送消息线程" << endl;

char tmp[50];



while(1)

{

cin >> tmp;

send(sock_client, tmp, strlen(tmp), 0);

}

}

void RecvMsg(LPVOID lpParameter)

{

//cout << "进入接收消息线程" << endl;

char buf[50];

int recv_size;

while (1)

{

recv_size = recv(sock_client, buf, 512, 0);



if (512 > recv_size && 0 < recv_size)

{

buf[recv_size] = '/0';



if(strcmp(buf, "quit") == 0)

{

//ExitThread(NULL);退出该线程

cout << "客户端退出" << endl;

}else

{

cout << "receive data is :" << buf << endl;

}

memset(buf, 0, sizeof(buf));

}



}

}



//客户端



#include <winsock2.h>

#include <windows.h>

#include <iostream.h>

#define PORT_SERVER 6666

#pragma comment(lib,"ws2_32.lib")

SOCKET sock_client;

void SendMsg(LPVOID lpParameter);

void RecvMsg(LPVOID lpParameter);

int main(int argc, char* argv[])

{

cout << "Start up tcp client." << endl;



WSADATA wsaData;

WORD sockVersion = MAKEWORD(2, 0);

if (0 != WSAStartup(sockVersion, &wsaData))

{

cout << "Failed to retrive socket version."

<< endl;

return 0;

}





sock_client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if (INVALID_SOCKET == sock_client)

{

cout << "Invalid socket." << endl;

WSACleanup();

return 0;

}

sockaddr_in addr_sev;



addr_sev.sin_family = AF_INET;

addr_sev.sin_port = htons(PORT_SERVER);

addr_sev.sin_addr.s_addr = inet_addr("127.0.0.1");



if (SOCKET_ERROR == connect(sock_client, (sockaddr *)&addr_sev, sizeof(addr_sev)))

{

cout << "Failed to connect." << endl;

WSACleanup();

return 0;

}



cout << "连接成功" << endl;



HANDLE hThread1, hThread2;

hThread1=CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)SendMsg, NULL, 0, NULL);

hThread2=CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)RecvMsg, NULL, 0, NULL);



CloseHandle(SendMsg);

CloseHandle(RecvMsg);



Sleep(200000); //主线程沉睡200秒



closesocket(sock_client);

WSACleanup();

//system("PAUSE");

return 0;

}

void SendMsg(LPVOID lpParameter)

{

//cout << "进入发送消息线程" << endl;

char tmp[50];



while(1)

{

cin >> tmp;

send(sock_client, tmp, strlen(tmp), 0);



if(strcmp(tmp, "quit") == 0)

{

cout << "退出" << endl;

ExitProcess(NULL);

}

}

}

void RecvMsg(LPVOID lpParameter)

{

//cout << "进入接收消息线程" << endl;



char buf[50];

int recv_size;



while (1)

{

recv_size = recv(sock_client, buf, 512, 0);



if (512 > recv_size && 0 < recv_size)

{

buf[recv_size] = '/0';

cout << "receive data is :" << buf << endl;

memset(buf, 0, sizeof(buf));

}



}

}