C++ SOCKET多线程实现聊天小程序

#include "stdafx.h"
#include<WinSock2.h>
#include<string.h>
#include<iOStream>
#pragma comment (lib, "ws2_32.lib")
using namespace std;
const int PORT = 8000;
#define IP "127.0.0.1"
#define MaxClient 10//最多能接受同时在线的客户端数量，可以随意修改
#define MaxBufSize 1024
int num =0;//客户端数量计数器
#define _CRT_SECURE_NO_WARINGS

//服务线程
DWord WINapi SeverThread(LPVOID lpParameter)
{
    //新建一个SOCKET用于通信
 SOCKET *ClientSocket = (SOCKET*)lpParameter;
 int receByt = 0;
 char RecvBuf[MaxBufSize];
 char SendBuf[MaxBufSize];
 char exitBuf[5];
 //开始接收
 while (1)
 {
  receByt = recv(*ClientSocket, RecvBuf, sizeof(RecvBuf), 0);
  if (receByt > 0)
  {
      //当客户端发来的消息是“exit”，就关闭连接
   if (strlen(RecvBuf)==4)
   {
    for (int i = 0; i < 5; i++)
    {
     exitBuf[i] = RecvBuf[i];
    }
    int flag = strcmp(exitBuf, "exit");
    if (flag==0)//接收到exit消息
    {
     cout << "client " << *ClientSocket << " exit!" << endl;
     num--;
     send(*ClientSocket, "Your server has been closed", sizeof(SendBuf), 0);
     closesocket(*ClientSocket);
     return 0;
    }
   }
    cout << "receive message :" << RecvBuf << " from client:" << *ClientSocket << endl;
   
  }
  else
  {
      //下面说到的客户端关闭连接是指客户端掉线了
   if (WSAGetLastError() == 10054)//检测到客户端关闭连接
   {
    cout << "client " << *ClientSocket << " exit!" << endl;
    closesocket(*ClientSocket);
    num--;
    return 0;
   }
   else//接收失败显示错误信息
   {
    cout << "failed to receive,Error:" << WSAGetLastError() << endl;
    break;
   }
   
  }
  memset(RecvBuf, 0, 1024);
  cout << "input your message to client:" << endl;
  scanf_s("%s",SendBuf,MaxBufSize);
  int k = 0;
  k = send(*ClientSocket, SendBuf, sizeof(SendBuf), 0);
  if (k < 0)
  {
   if (WSAGetLastError()==10054)//检测到客户端主动关闭连接
   {
    cout << "client " << *ClientSocket << " exit!" << endl;
    closesocket(*ClientSocket);
    num--;
    return 0;
   }
   else//发送失败显示错误信息
   cout << "failed to send, Error:" << WSAGetLastError()<<endl;
  }
  memset(SendBuf, 0, 1024);
 }
 if (*ClientSocket != INVALID_SOCKET)
 {
  closesocket(*ClientSocket);
 }
 return 0;
}

int _tmain(int arGC, _TCHAR* argv[])
{
 WSAData wsd;
 WSAStartup(MAKEWORD(2, 2), &wsd);
 SOCKET ListenSocket = socket(AF_INET, SOCK_STREAM, 0);
 SOCKADDR_IN ListenAddr;
 ListenAddr.sin_family = AF_INET;
 ListenAddr.sin_addr.S_un.S_addr = INADDR_ANY;//本机ip
 ListenAddr.sin_port = htons(PORT);
 //绑定监听端口
 int n;
 n = bind(ListenSocket, (LPSOCKADDR)&ListenAddr, sizeof(ListenAddr));
 if (n == SOCKET_ERROR)
 {
  cout << "failed to bind!" << endl;
  return -1;
 }
 else
 {
  cout << "bind success to:" << PORT << endl;
 }
 //开始监听
 int l = listen(ListenSocket, MaxClient);
 if (l == 0)
 {
  cout << "server ready, wait to requirement..." << endl;
 }
 else
 {
  cout << "Error:" << GetLastError() << "listen return" << l << endl;
 }
 while (1)
 {
  //循环接收客户端连接请求并创建服务线程
  if(num < MaxClient)
  {
   SOCKET *ClientSocket=new SOCKET;
   HANDLE hThread;
   int SockAddrlen = sizeof(sockaddr);
   *ClientSocket = accept(ListenSocket, 0, 0);
   cout << "client " << *ClientSocket << " has connect to server" << endl;
   num++;
   hThread = CreateThread(NULL, NULL, &SeverThread, (LPVOID)ClientSocket, 0, NULL);
   CloseHandle(hThread);
  }
  else
  {
   cout << "Max Client!Please wait for accept..." << endl;
  }
 }
 closesocket(ListenSocket);
 WSACleanup();
 return 0;
}

#include "stdafx.h"
#include<iostream>
#include<cstdio>
#include<string>
#include<Winsock2.h>
#pragma comment(lib,"ws2_32.lib")
using namespace std;
const int PORT = 8000;
#define MaxBufSize 1024
#define _CRT_SECURE_NO_WARINGS

int _tmain(int argc, _TCHAR* argv[])
{
 WSADATA wsd;
 WSAStartup(MAKEWORD(2, 2), &wsd);
 SOCKET SocketClient = socket(AF_INET, SOCK_STREAM, 0);
 SOCKADDR_IN ClientAddr;
 ClientAddr.sin_family = AF_INET;
 ClientAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
 ClientAddr.sin_port = htons(PORT);
 int n = 0;
 n = connect(SocketClient, (struct sockaddr*)&ClientAddr, sizeof(ClientAddr));
 if (n == SOCKET_ERROR)
 {
  cout << "failed to connect" << endl;
  return -1;
 }
 cout << "success to connect to Server" << endl;
 
 char info[1024];//数据输入缓冲区
 char SendBuff[MaxBufSize];//发送数据缓冲区
 char RecvBuff[MaxBufSize];//接收数据缓冲区
 while (1)
 {
  cout << "input your message:" << endl;
  scanf_s("%s",&info,MaxBufSize);
  
  if (info[0] == '\0')
   break;
  strcpy(SendBuff, info);
  memset(info, 0, sizeof(info));
  int k = 0;
  k = send(SocketClient, SendBuff, sizeof(SendBuff), 0);
  memset(SendBuff, 0, sizeof(SendBuff));
  if (k < 0)
  {
   cout << WSAGetLastError() << endl;
   cout << "failed to send" << endl;
  }
  int n = 0;
  n = recv(SocketClient, RecvBuff, sizeof(RecvBuff), 0);
  if (n>0)
  {
   cout << "receive message from Server:" << RecvBuff << endl;
   memset(RecvBuff, 0, sizeof(RecvBuff));
  }
 }
 closesocket(SocketClient);
 WSACleanup();
 return 0;
}