C++實現多人聊天室

// Test_Console.cpp : 定義控制臺應用程序的入口點。
//

#include "stdafx.h"
#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <Windows.h>
#include <thread>
#include <cstdio>

using namespace std;

#pragma region 全局變量

SOCKET server;     // 服務端套接字
sockaddr_in sai_server;   // 服務端信息（ip、端口）

// 消息格式
struct umsg {
 int type;    // 協議(1：加入 2：退出 3：發消息)
 char name[64];   // 用戶名字
 char text[512];   // 文本信息
};

// 客戶端鏈表
typedef struct ucnode {
 sockaddr_in addr;  // 客戶端的地址和端口號
 umsg msg;    // 客戶端傳來的消息
 ucnode* next;
} *ucnode_t;

#pragma endregion


#pragma region 依賴函數

// 鏈表插入數據
ucnode* insertNode(ucnode* head, sockaddr_in addr,umsg msg) {
 ucnode* newNode = new ucnode();
 newNode->addr = addr;
 newNode->msg = msg;
 ucnode* p = head;
 if (p == nullptr) {
  head = newNode;
 }
 else {
  while (p->next != nullptr) {
   p = p->next;
  }
  p->next = newNode;
 }
 return head;
}

// 鏈表刪除數據
ucnode* deleteNode(ucnode* head, umsg msg) {
 ucnode* p = head;
 if (p == nullptr) {
  return head;
 }
 if (strcmp(p->msg.name, msg.name) == 0){
  head = p->next;
  delete p;
  return head;
 }
 while (p->next != nullptr && strcmp(p->next->msg.name, msg.name) != 0) {
  p = p->next;
 }
 if (p->next == nullptr) {
  return head;
 }
 ucnode* deleteNode = p->next;
 p->next = deleteNode->next;
 delete deleteNode;
 return head;
}

#pragma endregion

int main()
{
 cout << "我是服務端" << endl;

 // 初始化 WSA ，激活 socket
 WSADATA wsaData;
 if (WSAStartup(
  MAKEWORD(2, 2),   // 規定 socket 版本為 2.2
  &wsaData    // 接收關於套接字的更多信息
  )) {
  cout << "WSAStartup failed : " << GetLastError() << endl;
 }

 // 初始化 socket、服務器信息
 server = socket(
  AF_INET,   // IPV4
  SOCK_DGRAM,  // UDP
  0    // 不指定協議
  );
 sai_server.sin_addr.S_un.S_addr = 0; // IP地址
 sai_server.sin_family = AF_INET;  // IPV4
 sai_server.sin_port = htons(8090);  // 傳輸協議端口

 // 本地地址關聯套接字
 if (bind(
  server,      // 要與本地地址綁定的套接字
  (sockaddr*)&sai_server,  // 用來接收客戶端消息的 sockaddr_in 結構體指針
  sizeof(sai_server)   
  )) {
  cout << "bind failed : " << GetLastError() << endl;
  WSACleanup();
 }

 // 初始化客戶端鏈表
 ucnode* listHead = new ucnode();
 listHead->next = nullptr;
 ucnode* lp = listHead;

 // 監聽消息
 while (1) {
  // 接收來自客戶端的消息
  umsg msg;
  int len_client = sizeof(sockaddr);
  recvfrom(
   server,      // 本地套接字
   (char*)&msg,     // 存放接收到的消息
   sizeof(msg),     
   0,        // 不修改函數調用行為
   (sockaddr*)&sai_server,  // 接收客戶端的IP、端口
   &len_client     // 接收消息的長度，必須初始化，否則默認為0 收不到消息    
  );
  
  // sin_addr 轉 char[]（char[] 轉 sin_addr 使用 inet_top）
  char arr_ip[20];
  inet_ntop(AF_INET, &sai_server.sin_addr, arr_ip, 16);
  
  // 處理消息（1：用戶登錄，2：用戶退出，3：普通會話）
  switch (msg.type) {
  case 1: 
   insertNode(listHead, sai_server, msg); 
   cout << "[" << arr_ip << ":" << ntohs(sai_server.sin_port) << "] " << msg.name << ":" << "---登錄---" << endl;
   break;
  case 2:
   deleteNode(listHead, msg);
   cout << "[" << arr_ip << ":" << ntohs(sai_server.sin_port) << "] " << msg.name << ":" << "---退出---" << endl;
   break;
  case 3:
   cout << "[" << arr_ip << ":" << ntohs(sai_server.sin_port) << "] " << msg.name << ":" << msg.text << endl;
   // 更新 msg.text
   lp = listHead;
   while (lp) {
    if (strcmp(lp->msg.name, msg.name) == 0) {
     strncpy(lp->msg.text, msg.text, sizeof(msg.text));
     lp->msg.type = msg.type;
     break;
    }
    lp = lp->next;
   }
   // 向其他客戶端廣播（除自己之外）
   lp = listHead;
   while (lp) {
    if (strcmp(lp->msg.name,"") != 0 && strcmp(lp->msg.name, msg.name) != 0) {
     sendto(
      server,      // 本地套接字
      (char*)&msg,     // 消息結構體
      sizeof(msg),     
      0,        // 不修改函數調用行為
      (sockaddr*) & lp->addr,  // 目標客戶端地址
      sizeof(lp->addr)
     );
    }
    lp = lp->next;
   }
   break;
  }
 }

 // 禁用 socket
 WSACleanup();

 getchar();
    return 0;
}

// Test_Console_2.cpp : 此文件包含 "main" 函數。程序執行將在此處開始並結束。
//

#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <Windows.h>
#include <thread>
#include <cstdio>
#include <string>

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

using namespace std;

#pragma region 全局變量

SOCKET client;     // 客戶端套接字  
sockaddr_in sai_client;   // 存放客戶端地址、端口
sockaddr_in sai_server;   // 存放服務端發送的消息

// 發送和接收的信息體
struct umsg {
 int type;     // 協議（1：登錄，2：退出，3：發消息）
 char name[64];    // 用戶名字
 char text[512];    // 文本
};

#pragma endregion

#pragma region 依賴函數

// 監聽服務器消息
void recvMessage()
{
 while (1){
  umsg msg;
  int len_server = sizeof(sockaddr);
  int len = recvfrom(client, (char*)&msg,sizeof(msg),0,(sockaddr*)&sai_server,&len_server);
  
  cout << msg.name << ": " << msg.text << endl;
 }
}

#pragma endregion

int main()
{
 cout << "我是客戶端" << endl;

 // 初始化 WSA ，激活 socket
 WSADATA wsaData;
 if (WSAStartup(
  MAKEWORD(2, 2),  // 規定 socket 版本
  &wsaData   // 接收 socket 的更多信息
  )) {
  cout << "WSAStartup failed : " << GetLastError() << endl;
 }

 // 初始化 socket、客戶端信息
 client = socket(
  AF_INET,  // IPV4
  SOCK_DGRAM,  // UDP
  0    // 不指定協議
  );
 sai_client.sin_family = AF_INET;         // IPV4
 inet_pton(AF_INET, "192.168.1.105", &sai_client.sin_addr);   // 服務器 IP地址
 sai_client.sin_port = htons(8090);         // 端口

 // 輸入用戶名
 string name;
 getline(cin, name);

 // 發送登錄消息
 umsg msg;
 msg.type = 1;
 strncpy_s(msg.name, sizeof(msg.name), name.c_str(), 64);
 strncpy_s(msg.text, sizeof(msg.text), "", 512);
 sendto(
  client,       // 本地套接字
  (char*)&msg,      // 發送的消息
  sizeof(msg), 
  0,         // 不修改函數調用行為
  (sockaddr*) & sai_client,  // 消息目標
  sizeof(sai_client)
 );

 // 接收服務器消息
 HANDLE h_recvMes = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)recvMessage, 0, 0, 0);
 if (!h_recvMes) { cout << "CreateThread failed :" << GetLastError() << endl; }
    
 // 發送消息
 while (1) {
  string content;
  getline(cin, content);
  
  // 如果是退出消息
  if (content == "quit") {
   msg.type = 2;
   sendto(client, (char*)&msg, sizeof msg, 0, (struct sockaddr*) & sai_client, sizeof(sai_client));
   closesocket(client);
   WSACleanup();
   return 0;
  }

  // 如果是會話消息
  msg.type = 3;
  strncpy_s(msg.text, sizeof(msg.text), content.c_str(), 512);
  sendto(
   client,       // 本地套接字
   (char*)&msg,      // 要發送的消息
   sizeof(msg), 
   0,         // 不修改函數調用行為
   (sockaddr*) & sai_client,   // 發送目標
   sizeof(sai_client)
  );
 }


    getchar();
    return 0;
}

// Test_Console.cpp : 定義控制臺應用程序的入口點。
//

#include "stdafx.h"
#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <Windows.h>
#include <thread>
#include <cstdio>

using namespace std;

#pragma region 全局變量

SOCKET server;    // 本地套接字
sockaddr_in sai_server;  // 存放服務器IP、端口

// 消息格式
struct umsg {
 int type;    // 協議（1：登錄，2：退出，3：發消息）
 char name[64];   // 用戶名字
 char text[512];   // 文本信息
};

// 客戶端信息
struct clientInfo {
 SOCKET client;
 sockaddr_in saddr;
 umsg msg;
};

// 客戶端鏈表
typedef struct ucnode {
 clientInfo cInfo;
 ucnode* next;
} *ucnode_t;

ucnode* listHead;  // 客戶端鏈表頭
ucnode* lp;    // 客戶端鏈表指針

#pragma endregion

#pragma region 依賴函數

// 鏈表插入數據
ucnode* insertNode(ucnode* head,SOCKET client, sockaddr_in addr, umsg msg) {
 ucnode* newNode = new ucnode();
 newNode->cInfo.client = client;
 newNode->cInfo.saddr = addr; 
 newNode->cInfo.msg = msg;
 ucnode* p = head;
 if (p == nullptr) {
  head = newNode;
 }
 else {
  while (p->next != nullptr) {
   p = p->next;
  }
  p->next = newNode;
 }
 return head;
}

// 鏈表刪除數據
ucnode* deleteNode(ucnode* head, SOCKET client) {
 ucnode* p = head;
 if (p == nullptr) {
  return head;
 }
 if (p->cInfo.client == client) {
  head = p->next;
  delete p;
  return head;
 }
 while (p->next != nullptr && p->next->cInfo.client != client) {
  p = p->next;
 }
 if (p->next == nullptr) {
  return head;
 }
 ucnode* deleteNode = p->next;
 p->next = deleteNode->next;
 delete deleteNode;
 return head;
}

// 接收客戶端消息（某個）
void recvMessage(PVOID pParam) {
 clientInfo* cInfo = (clientInfo*)pParam;

 while (1) {
  // 接收來自客戶端的消息
  umsg msg;
  int len_client = sizeof(sockaddr);
  int ret_recv = recv(
   cInfo->client, // 本地套接字 
   (char*)&msg, // 存放接收的消息
   sizeof(msg), // 消息大小
   0    // 不修改函數調用行為
  ); 
  if (ret_recv <= 0) { cout << msg.name << "斷開連接: " << GetLastError() << endl; break; }
  cInfo->msg = msg;

  // sin_addr 轉 char[]（char[] 轉 sin_addr 使用 inet_top）
  char arr_ip[20];
  inet_ntop(AF_INET, &cInfo->saddr.sin_addr, arr_ip, 16);

  // 處理消息（1：登錄，2：退出，3：會話）
  switch (cInfo->msg.type) {
  case 1:
   // 插入數據到鏈表
   insertNode(listHead,cInfo->client, cInfo->saddr,cInfo->msg);
   // 打印消息
   cout << "[" << arr_ip << ":" << ntohs(cInfo->saddr.sin_port) << "] " << msg.name << ":" << "---登錄---" << endl;
   break;
  case 2:
   // 從鏈表刪除數據
   deleteNode(listHead, /*cInfo->msg*/cInfo->client);
   // 打印消息
   cout << "[" << arr_ip << ":" << ntohs(cInfo->saddr.sin_port) << "] " << msg.name << ":" << "---退出---" << endl;
   break;
  case 3:
   // 打印消息
   cout << "[" << arr_ip << ":" << ntohs(cInfo->saddr.sin_port) << "] " << msg.name << ":" << cInfo->msg.text << endl;
   // 向其他客戶端廣播（除自己之外）
   lp = listHead;
   while (lp) {
    if (strcmp(lp->cInfo.msg.name, "") != 0 && strcmp(lp->cInfo.msg.name, cInfo->msg.name) != 0) {
     send(
      lp->cInfo.client,  // 本地套接字
      (char*)&cInfo->msg,  // 發送的消息
      sizeof(cInfo->msg),  // 消息大小
      0      // 不指定調用方式
     );
     int error_send = GetLastError();
     if (error_send != 0) { cout << "send failed:" << error_send << endl; }
    }
    lp = lp->next;
   }
   break;
  }
 }
}

#pragma endregion

int main()
{
 cout << "我是服務端" << endl;

 // 初始化 WSA ，激活 socket
 WSADATA wsaData;
 if (WSAStartup(
  MAKEWORD(2, 2),   // 規定 socket 版本為 2.2
  &wsaData    // 接收關於套接字的更多信息
 )) {
  cout << "WSAStartup failed : " << GetLastError() << endl;
 }

 // 初始化 socket、服務器信息
 server = socket(
  AF_INET,   // IPV4
  SOCK_STREAM, // TCP
  0    // 不指定協議
 );
 sai_server.sin_addr.S_un.S_addr = 0; // IP地址
 sai_server.sin_family = AF_INET;  // IPV4
 sai_server.sin_port = htons(8090);  // 傳輸協議端口

 // 本地地址關聯套接字
 if (bind(
  server,      // 要與本地地址綁定的套接字
  (sockaddr*)&sai_server,  // 用來接收客戶端消息的 sockaddr_in 結構體指針
  sizeof(sai_server)
 )) {
  cout << "bind failed : " << GetLastError() << endl;
  WSACleanup();
 }

 // 套接字進入監聽狀態
 listen(
  server,  // 本地套接字
  SOMAXCONN // 掛起連接隊列的最大長度，SOMAXCONN：最大合理值
 );

 // 初始化客戶端鏈表
 listHead = new ucnode();
 listHead->next = nullptr;
 lp = listHead;
  
 // 接收消息
 while (1) {
  // 接收登錄消息（首次連接是觸發，之後發送消息不觸發）
  clientInfo* cInfo = new clientInfo();
  int len_client = sizeof(sockaddr);
  cInfo->client = accept(server, (sockaddr*) &cInfo->saddr, &len_client); 
  if (GetLastError() != 0) { continue; }
  
  // 接收登錄者的消息（每個客戶端對應一個線程）
  HANDLE h_recvMes = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)recvMessage, cInfo, 0, 0);
  if (!h_recvMes) { cout << "CreateThread failed :" << GetLastError() << endl; }
 }

 // 禁用 socket
 WSACleanup();

 getchar();
    return 0;
}

// Test_Console_2.cpp : 此文件包含 "main" 函數。程序執行將在此處開始並結束。
//

#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <Windows.h>
#include <thread>
#include <cstdio>
#include <string>

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

using namespace std;

#pragma region 全局變量

SOCKET client;     // 本地套接字
sockaddr_in sai_client;   // 存放客戶端IP地址、端口

// 消息格式
struct umsg {
 int type;     // 協議（1：登錄，2：退出，3：發消息）
 char name[64];    // 用戶名字
 char text[512];    // 文本
};

#pragma endregion

#pragma region 依賴函數

// 監聽服務器消息
void recvMessage()
{
 while (1){
  umsg msg;
  int ret_recv = recv(
   client,   // 本地套接字
   (char*)&msg, // 存放接收的消息
   sizeof(msg), // 消息大小
   0    // 不指定調用方式
  );
  if (ret_recv <= 0) { cout << "recv failed: " << GetLastError() << endl; break; }
  
  // 打印消息
  cout << msg.name << ": " << msg.text << endl;
 }
}

#pragma endregion

int main()
{
 cout << "我是客戶端" << endl;

 // 初始化 WSA ，激活 socket
 WSADATA wsaData;
 if (WSAStartup(
  MAKEWORD(2, 2),  // 規定 socket 版本
  &wsaData   // 接收 socket 的更多信息
 )) {
  cout << "WSAStartup failed : " << GetLastError() << endl;
 }

 // 初始化 socket、客戶端信息
 client = socket(
  AF_INET,  // IPV4
  SOCK_STREAM, // TCP
  0    // 不指定協議
 );
 sai_client.sin_family = AF_INET;         // IPV4
 inet_pton(AF_INET, "192.168.1.100", &sai_client.sin_addr);   // 服務器 IP地址
 sai_client.sin_port = htons(8090);         // 端口

 // 連接服務器
 int ret_connect = connect(
  client,      // 本地套接字
  (sockaddr*) &sai_client,  // 目標
  sizeof(sai_client)
 );if (ret_connect != 0) { cout << "connect failed:" << GetLastError() << endl; }

 // 輸入用戶名
 umsg msg;
 msg.type = 1;
 string name;
 getline(cin, name);
 strncpy_s(msg.name, sizeof(msg.name), name.c_str(), 64);
 strncpy_s(msg.text, sizeof(msg.text), "", 512);

 // 發送登錄消息
 send(
  client,   // 本地套接字
  (char*)&msg, // 發送的消息
  sizeof(msg), // 消息大小
  0    // 不指定調用方式
 ); 
 int error_send = GetLastError();
 if (error_send != 0) { cout << "send failed:" << error_send << endl; }
 
 // 接收服務器消息
 HANDLE h_recvMes = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)recvMessage, 0, 0, 0);
 if (!h_recvMes) { cout << "CreateThread failed :" << GetLastError() << endl; }

 // 發送消息
 while (1) {
  string content;
  getline(cin, content);

  // 退出消息
  if (content == "quit") {
   msg.type = 2;
   send(
    client,   // 本地套接字
    (char*)&msg, // 發送的消息
    sizeof(msg), // 消息大小
    0    // 不指定調用方式
   );
   error_send = GetLastError();
   if (error_send != 0) { cout << "send failed:" << error_send << endl; }
   closesocket(client);
   WSACleanup();
   return 0;
  }
  
  // 會話消息
  msg.type = 3;
  strncpy_s(msg.text, sizeof(msg.text), content.c_str(), 512);
  send(
   client,   // 本體套接字
   (char*)&msg, // 發送的消息
   sizeof(msg), // 消息大小
   0    // 不指定調用方式
  );
  error_send = GetLastError();
  if (error_send != 0) { cout << "send failed:" << error_send << endl; }

 }

    getchar();
    return 0;
}