技術文章 » 14.1 Socket 通訊端程式設計入門

14.1 Socket 通訊端程式設計入門

2023-10-11 12:02:41

Winsock是Windows作業系統上的通訊端API，用於在網路上進行資料通訊。通訊端通訊是一種允許應用程式在計算機網路上進行實時資料交換的技術。通過使用Windows提供的API，應用程式可以建立一個通訊端來進行資料通訊。這個通訊端可以繫結到一個埠，以允許其他應用程式連線它。另外，Winsock可以使用TCP/IP、UDP等協定來完成不同型別的資料傳輸任務。在網路應用程式開發中，通訊端通訊可以幫助應用程式開發者實現使用者端/伺服器端模型，並實現資料的可靠傳輸。

一般通訊端通訊需要經歷，建立通訊端(Socket)，繫結(Bind)，監聽(Listen)，接受(Accept)，連線(Connect)，傳送資料(Send)，接收資料(Receive)，關閉(Close)等幾個關鍵步驟，當讀者需要使用網路通訊時需引入winsock2.h標頭檔案，並通過#pragma comment(lib,"ws2_32.lib")包含對應庫，需要注意的是該標頭檔案與windows.h頭衝突，如果兩者同時存在則會出現編譯不通過的情況；

14.1.1 伺服器端通訊

（1）WSAStartup(MAKEWORD(2, 0), &WSAData)

當讀者需要使用通訊端程式設計時，不論是伺服器端還是使用者端都需要呼叫WSAStartup初始化通訊端庫，該函數接受兩個引數傳遞，第一個引數一般預設會傳遞MAKEWORD(2, 0) 它是一個宏，用於將兩個8位元的位元組合併成一個16位元的字，在MAKEWORD(2, 0)中，括號內的數位分別代表高位位元組（2）和低位位元組（0），宏會將它們合併成一個16位元的無符號short整型資料，即0000001000000000（二進位制），表示Winsock的版本號為2.0。第二個引數WSADATA結構體，用於Winsock初始化時儲存相關的資訊，一般會在全域性WSADATA WSAData;直接定義得到。

#include <iostream>
#include <winsock2.h>
#include <WS2tcpip.h>

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

// 定義結構體
WSADATA WSAData;

// 啟動winsock中的WSAStartup()函數對Winsock DLL進行初始化
if (WSAStartup(MAKEWORD(2, 0), &WSAData) == SOCKET_ERROR)
{
    std::cout << "WSA動態庫初始化失敗" << std::endl;
    return 0;
}

（2）socket(AF_INET, SOCK_STREAM, 0)

通訊的第二步則是呼叫Socket()函數，該函數是用於建立一個通訊端的系統呼叫。在該函數中，給定三個引數，分別為地址族(Address Family)、通訊端型別(Socket Type)和協定(Protocol)，通訊端在初始化並完成時會返回一個SOCKET型別的檔案描述符控制程式碼，此處我們將該控制程式碼儲存至server_socket變數內。AF_INET用於指定通訊端地址族為IPv4型別，SOCK_STREAM則用於指定該通訊端的型別為流式通訊端，用於面向連線的可靠資料傳輸(TCP協定)。

// 服務程序建立通訊端控制程式碼（用於監聽）
SOCKET server_socket;

// 呼叫socket()函數建立一個流通訊端，引數（網路地址型別，通訊端型別，網路協定）
if ((server_socket = socket(AF_INET, SOCK_STREAM, 0)) == ERROR)
{
  std::cout << "Socket 建立失敗" << std::endl;
  WSACleanup();
  return 0;
}

（3）bind(server_socket, (LPSOCKADDR)&ServerAddr, sizeof(ServerAddr))

通訊端程式設計的第三步則是繫結，通訊端的繫結需要呼叫bind()函數實現，該函數接受三個引數傳遞，第一個引數是socket()中建立的通訊端檔案描述符控制程式碼，該引數用於指定針對哪一個通訊端進行操作，第二個引數則是sockaddr_in型別的結構體，該結構體內用於指定需要繫結通訊端的具體型別引數等資訊，在如下程式碼中我們通過ServerAddr.sin_family = AF_INET;將通訊端型別設定為了網際網路域模式，通過ServerAddr.sin_port = htons(9999);指定了需要繫結的埠號，而ServerAddr.sin_addr.s_addr = inet_addr("0.0.0.0");則用於指定了要繫結本機的那個網口，一般而言如果讀者需要在本機使用此處可填入127.0.0.1而如果偵聽任意一個網口則可使用0.0.0.0，第三個引數則是傳入結構體的長度，此處通過sizeof(ServerAddr)方法得到，最終將結構體ServerAddr直接填入繫結函數即可實現對網路通訊端的繫結。

// 結構sockaddr_in用來標識TCP/IP協定下的地址，可強制轉換為sockaddr結構
struct sockaddr_in ServerAddr;

// 欄位sin_family必須設為AF_INET，表示該Socket處於Internet域
ServerAddr.sin_family = AF_INET;

// 欄位sin_port用於指定伺服器埠，注意避免衝突
ServerAddr.sin_port = htons(9999);

// 欄位sin_addr用於把一個IP地址儲存為一個4位元組，無符號長整型，根據不同用法還可表示本地或遠端IP地址
// 該欄位可以直接使用INADDR_ANY代表偵聽所有地址，也可指定地址
ServerAddr.sin_addr.s_addr = inet_addr("0.0.0.0");

// 呼叫bind()函數將本地地址繫結到所建立的通訊端上，以在網路上標識該通訊端
if (bind(server_socket, (LPSOCKADDR)&ServerAddr, sizeof(ServerAddr)) == SOCKET_ERROR)
{
  std::cout << "繫結通訊端失敗" << std::endl;
  closesocket(server_socket);
  WSACleanup();
  return 0;
}

（4）listen(server_socket, 10)

當通訊端被繫結後，接下來則是偵聽通訊端，通過呼叫listen()函數將通訊端置入監聽模式並準備接受連線請求，該函數需要傳入兩個引數，引數1為通訊端通訊端控制程式碼，引數二為偵聽通訊端最大連線數，如果進入偵聽狀態則說明該通訊端是等待連線狀態，一旦伺服器接受了連線，它可以使用返回的通訊端物件與發起連線的使用者端進行通訊。

// 將 ServerAddr.sin_addr 網路位元組序,轉為本機偵聽IP地址
char local_address[20];
inet_ntop(AF_INET, &ServerAddr.sin_addr, local_address, 16);
std::cout << "偵聽本地地址: " << local_address << " 偵聽本地埠: " << ntohs(ServerAddr.sin_port) << std::endl;

// 引數（已捆綁未連線的通訊端描述字，正在等待連線的最大佇列長度）
if (listen(server_socket, 10) == SOCKET_ERROR)
{
  std::cout << "偵聽通訊端失敗" << std::endl;
  closesocket(server_socket);
  WSACleanup();
  return 0;
}

（5）accept(server_socket, (LPSOCKADDR)0, (int*)0)

當一個通訊端進入偵聽狀態後則下一步是需要等待有使用者端連線到本端，當伺服器通過呼叫listen()函數開始監聽連線請求時，使用者端可以通過使用connect()函數嘗試與伺服器建立連線。一旦使用者端傳送連線請求，伺服器將收到通知。然後伺服器可以使用accept()函數接受連線請求並建立一個新的通訊端物件，該物件可以用於與使用者端進行通訊。

accept() 函數通常在一個迴圈中使用，以便伺服器可以在等待新連線時繼續處理已連線的使用者端。每次呼叫accept()函數時，如果有連線請求，則函數將阻塞直到一個連線請求被接受。一旦連線請求被接受，函數將返回一個新的通訊端物件和使用者端的地址資訊。

在接受連線請求並建立新的通訊端物件之後，伺服器可以使用該物件與使用者端進行通訊。同時，伺服器可以使用原始的server_socket通訊端物件來等待更多的連線請求，以便能夠接受更多的使用者端連線。

如下的程式碼中當accept()接收到等待訊息時，則會將該控制程式碼儲存至message_socket變數內，此時使用者只需要向該指標中傳送recv()或接收send()資料即可，此時通訊端通訊即可正式被建立起來。

// 資料接收緩衝區
SOCKET message_socket;
char buf[8192] = {0};
while (1)
{
  // 進入監聽狀態後，呼叫accept()函數接收使用者端的連線請求，並把連線傳給msgsock通訊端
  // 原sock通訊端繼續監聽其他客戶機連線請求
  if ((message_socket = accept(server_socket, (LPSOCKADDR)0, (int*)0)) == INVALID_SOCKET)
  {
    continue;
  }

  // 初始化資料接收緩衝區
  memset(buf, 0, sizeof(buf));
   
  // 接收使用者端傳送過來的資料
  bool ref = recv(message_socket, buf, 8192, 0);
  if (ref != 0)
  {
    std::cout << "接收資料: " << buf << std::endl;
  }
  
  // 關閉子通訊端
  closesocket(message_socket);
}

至此我們的伺服器端將被執行起來，需要注意的是伺服器端程式如果需要結束本次對談則需要手動呼叫closesocket(server_socket);關閉一個通訊端控制程式碼，當整個程序執行結束後讀者還需要呼叫WSACleanup()終止對Winsock DLL的使用，並釋放資源。

14.1.2 使用者端通訊

對於使用者端通訊而言其流程與伺服器端通訊基本保持一致，該流程分別是，建立通訊端，連線到伺服器，建立連線，傳送資料，關閉連線，對於初始化部分使用者端通訊與伺服器端沒有任何區別，唯一的區別在於對於伺服器端而言一般是使用listen()函數偵聽通訊端，而對於使用者端而言則是使用connect()函數連線到伺服器端，一旦連線建立成功，使用者端可以通過向伺服器傳送資料來與伺服器進行通訊。

在呼叫connect(socket_addr)時，需要傳遞一個引數sockaddr。sockaddr 是一個結構體，包含了使用者端與伺服器的地址資訊，包括其IP地址和埠號。在C/C++中，sockaddr 結構體通常被定義為sockaddr_in結構體，包含了IP地址和埠號等資訊。如果連線建立成功，connect() 函數將返回 0。如果連線失敗，則會返回一個錯誤程式碼，其中最常見的錯誤是連線超時或目標主機拒絕連線。

一旦連線建立成功，使用者端可以使用新建立的通訊端物件向伺服器傳送資料，並使用recv()函數從伺服器接收資料。一般來說，在與伺服器進行通訊之前，使用者端通訊端需要使用bind()函數指定一個本地地址和埠，以確保資料可以正確地傳輸。

int main(int argc, char* argv[])
{
  char buf[8192] = { 0 };

  while (1)
  {
    std::cout << "傳送資料: ";
    int inputLen = 0;
    memset(buf, 0, sizeof(buf));

    // 輸入以確認鍵為結束標識
    while ((buf[inputLen++] = getchar()) != '\n'){ ; }

    // 初始化
    WSADATA WSAData;
    if (WSAStartup(MAKEWORD(2, 0), &WSAData) == SOCKET_ERROR)
    {
      continue;
    }

    // 建立通訊端
    SOCKET client_socket;
    if ((client_socket = socket(AF_INET, SOCK_STREAM, 0)) == SOCKET_ERROR)
    {
      WSACleanup();
      continue;
    }

    // 填充通訊結構體
    struct sockaddr_in ClientAddr;
    ClientAddr.sin_family = AF_INET;
    ClientAddr.sin_port = htons(9999);
    ClientAddr.sin_addr.s_addr = inet_addr("127.0.0.1");

    // 連線到伺服器端
    if (connect(client_socket, (LPSOCKADDR)&ClientAddr, sizeof(ClientAddr)) == SOCKET_ERROR)
    {
      closesocket(client_socket);
      WSACleanup();
      continue;
    }

    // 向伺服器端傳送資料
    send(client_socket, buf, 8192, 0);

    // 關閉通訊端
    closesocket(client_socket);
    WSACleanup();
  }

  return 0;
}

讀者可自行執行上述程式，啟動伺服器端與使用者端，並行送測試資料觀察變化，當傳送資料後讀者應該能看到如下圖所示的提示資訊；

本文作者： 王瑞
本文連結： https://www.lyshark.com/post/83cf7258.html
版權宣告： 本部落格所有文章除特別宣告外，均採用 BY-NC-SA 許可協定。轉載請註明出處！