2. 程式人生 > 其它 >【TCP/IP】TCP伺服器併發處理&原始碼

【TCP/IP】TCP伺服器併發處理&原始碼

阿新 • • 發佈：2021-06-28

前言

本筆記記錄的是 單個服務端併發式處理多個客戶端。
下次有空在發個 單執行緒多個服務端併發式處理多種客戶端。其實就是本筆記的一個改良版，用到select() / poll() / epoll()。

原文：https://www.cnblogs.com/lizhuming/p/14943969.html

實現原理

實現原理很簡單，寫出一個簡單的TCP伺服器後，其客戶端處理方式採用執行緒化處理即可。
其中要注意的是多執行緒併發問題。

多執行緒處理客戶端是把 connect_sockfd 傳到執行緒，然後讓執行緒處理。

TCP 服務端

簡要步驟

  1. 建立 socket 。
  2. 配置地址資料。
  3. socket 繫結地址。
  4. 監聽。（ listen() ）
  5. 進入迴圈處理。
    1. accept() 獲取一個連線進來客戶端的 connect_socket。
    2. 建立一條執行緒專門處理該連結，並把該 connect_socket 傳給該執行緒。

注意

注意：

  1. 傳遞給客戶端處理執行緒的是 connect_socket 的值。注意傳遞的技巧及執行裝置的系統 bit 位數。
  2. 本程式採用一個全域性原子整數來維護客戶端的連線數。
  3. 其中 #include "atomic_lzm.h" 是個人實現的 linux 應用層的原子庫，就只是簡單的封裝一下而已，利用的是 posix 互斥鎖 API。有需要的直接去我 gitee 上拿。

TCP 客戶端

簡要步驟

  1. 建立 socket 。
  2. 配置地址資料。
  3. 建立 TCP 連線。
  4. 連線成功後進入資料互動。

注意

注意：

  1. 注意服務端的 IP 和 埠號。

實驗結果

需要測試客戶端限制的可以多開幾個客戶端即可。有興趣的同學可以自己驗證哈哈。

參考原始碼

服務端參考原始碼

/** @file         server.c
 *  @brief        簡要說明
 *  @details      詳細說明
 *  @author       lzm
 *  @date         2021-06-10 16:49:46
 *  @version      v1.0
 *  @copyright    Copyright By lizhuming, All Rights Reserved
 *  @blog         https://www.cnblogs.com/lizhuming/
 **********************************************************
 *  @LOG 修改日誌:
 **********************************************************
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <semaphore.h>
#include "atomic_lzm.h"

// 注意: 接收緩衝區對應對方的傳送緩衝區小於等於就不會出現對方接收多次。

#define CLIENT_BUFF_MAX 1024 // 客戶端接收緩衝區 size
#define MY_SERVER_PORT 8080 // 服務端監聽埠
#define CLIENT_PENDING_NUM 10 // 核心未完成佇列的客戶端緩衝數
#define CLIENT_MAX_NUM 2 // 客戶端連線數限制

// 定義一個無名訊號量
//sem_t sem_client;

atomic_lzm_t g_client_num;

/**
 * @name   client_fun
 * @brief  執行緒函式，處理 client
 * @param  arg：資源
 * @retval 
 * @author lzm
 */
void *client_fun(void *arg)
{
    int recv_len = 0; // 接收長度
    char recv_buff[CLIENT_BUFF_MAX] = "";
    long connfd = (long)arg; // 已連線socket。 值傳遞（地址傳遞時注意多執行緒併發問題）

    // [1] 接收資料
    while((recv_len = recv(connfd, recv_buff, sizeof(recv_buff), 0)) > 0)
    {
        printf("[%ld]recv_buff:%s\r\n", connfd, recv_buff); // 列印資料
        send(connfd, "OK", 2, 0); // 返回 client
    }

    // [2] 關閉 socket
    printf("client closed!\r\n");
    close(connfd); // 關閉 socket
    //sem_post(&sem_client);
    atomic_lzm_sub(&g_client_num);
    printf("can link num is [%d]\n", CLIENT_MAX_NUM-atomic_lzm_get(&g_client_num));
    return NULL;
}

/**
 * @name   main
 * @brief  main函式，服務端處理
 * @param  
 * @retval 
 * @author lzm
 */
int main(int argc, char *argv[])
{
    int sockfd = 0; // 監聽 socket
    long connfd = 0; // 已連線 socket 。long 避免使用 64bit 機器而出錯
    int ret = 0; // 返回緩衝
    struct sockaddr_in local_server_addr; // 本地伺服器地址
    unsigned short local_server_port = MY_SERVER_PORT; // 本地伺服器監聽埠
    pthread_t thread_client_id; // client 執行緒 id

    atomic_lzm_init(&g_client_num, 0);

    printf("TCP server started at port [%d]!\n", local_server_port);

    // [1] 建立套接字
    printf("create server socket!\n");
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd < 0)
    {
        printf("%s-%s-%d:sockfd create faild!", __FILE__, __FUNCTION__, __LINE__);
        perror("socket create error");
        exit(-1);
    }

    // [2] 初始化地址資料
    printf("init server address!\n");
    bzero(&local_server_addr, sizeof(local_server_addr)); // 初始化伺服器地址
    local_server_addr.sin_family = AF_INET;
    local_server_addr.sin_port = htons(local_server_port);
    local_server_addr.sin_addr.s_addr = htonl(INADDR_ANY);

    // [3] 繫結
    printf("bing server socket and addr!\n");
    ret = bind(sockfd, (struct sockaddr*)&local_server_addr, sizeof(local_server_addr));
    if(ret != 0)
    {
        printf("%s-%s-%d:sockfd bind faild!", __FILE__, __FUNCTION__, __LINE__);
        perror("socket bind error");
        close(sockfd);
        exit(-1);
    }

    // [4] 監聽
    printf("listen server socket!\n");
    ret = listen(sockfd, CLIENT_PENDING_NUM);
    if(ret != 0)
    {
        printf("%s-%s-%d:sockfd listen faild!", __FILE__, __FUNCTION__, __LINE__);
        perror("socket listen error");
        close(sockfd);
        exit(-1);
    }

    //sem_init(&sem_client, 0, CLIENT_MAX_NUM);

    printf("accept!\n");

    // [5] 處理 client
    while(1)
    {
        char client_ip[INET_ADDRSTRLEN] = ""; // use for save client ip
        struct sockaddr_in client_addr; // use for save client address
        socklen_t client_len = sizeof(client_addr); // 必須初始化

        //sem_wait(&sem_client);

        // [5][1] 獲取一個已建立的連線
        connfd = accept(sockfd, (struct sockaddr *)&client_addr, &client_len);
        if(connfd < 0)
        {
            printf("%s-%s-%d:sockfd accept faild!", __FILE__, __FUNCTION__, __LINE__);
            perror("accept error");
            continue;
        }

        if(atomic_lzm_get(&g_client_num) >= CLIENT_MAX_NUM)
        {
            close(connfd);
        }
        else
        {
            // [5][2] 處理客戶端資料
            inet_ntop(AF_INET, &client_addr.sin_addr, client_ip, INET_ADDRSTRLEN);
            printf("clien ip = [%s], port = [%d]\n", client_ip, ntohs(client_addr.sin_port));

            if(connfd > 0)
            {
                // 執行緒處理客戶端資料
                pthread_create(&thread_client_id, NULL, (void *)client_fun, (void *)connfd); // creat thread。注意 64bit 機器地址是8byte的
                pthread_detach(thread_client_id); // thread 分離。即時，執行緒回撥函式結束時自動回收該執行緒資源
                atomic_lzm_add(&g_client_num);
            }
        }
        printf("can link num is [%d]\n", CLIENT_MAX_NUM-atomic_lzm_get(&g_client_num));
    }

    // [6] 關閉 server socket
    close(sockfd);

    return 0;
}

TCP 客戶端原始碼

/** @file         client.c
 *  @brief        簡要說明
 *  @details      詳細說明
 *  @author       lzm
 *  @date         2021-06-10 09:49:46
 *  @version      v1.0
 *  @copyright    Copyright By lizhuming, All Rights Reserved
 *  @blog         https://www.cnblogs.com/lizhuming/
 **********************************************************
 *  @LOG 修改日誌:
 **********************************************************
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <pthread.h>

#define sHOST "192.168.112.128" // 伺服器端IP
#define sPORT 8080                  // 伺服器程序埠號

#define BUFFER_SIZE (1*1024)   // 資料域大小

// 注意: 接收緩衝區對應對方的傳送緩衝區小於等於就不會出現對方接收多次。


/**
  * @brief  讀 執行緒
  * @param 
  * @retval 
  * @author
  */
void *client_read_fun(void *arg) 
{
    int recv_len = 0; // 接收長度
    char recv_buff[BUFFER_SIZE] = "";
    int sockfd = *(int *)arg; // 已連線的 socket

    while((recv_len = recv(sockfd, recv_buff, sizeof(recv_buff), 0)) > 0)
    {
        printf("[%d]recv_buff:%s\r\n", sockfd, recv_buff);
    }

    printf("exit read thread!");
    return NULL;
}

/**
  * @brief  主函式
  * @param 
  * @retval 
  * @author
  */
int main(void)
{
    int sockfd; // 套接字描述符
    int ret;    // 返回結果
    struct sockaddr_in server; // 套接字地址
    char data_buffer[BUFFER_SIZE];  // app 資料域
    pthread_t thread_read_id; // 讀 執行緒 id

    /* [1] 建立套接字 */
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd == -1)
    {
        printf("creat socket failed!\n");
        exit(1);
    }

    /* [2] 初始化地址 */
    bzero(&server, sizeof(server));
    server.sin_family = AF_INET;
    server.sin_port = htons(sPORT);
    server.sin_addr.s_addr = inet_addr(sHOST);

    /* [3] 建立TCP連線 */
    ret = connect(sockfd, (struct sockaddr *)&server, sizeof(struct sockaddr));
    if(ret == -1)
    {
        printf("connect server failed!\n");
        close(sockfd);
        exit(1);
    }

    printf("connect server success!");

    /* [4][1] 建立 讀 執行緒 */
    //pthread_create(&thread_read_id, NULL, (void *)client_read_fun, (void *)&(sockfd));
    //pthread_detach(thread_read_id); // 執行緒分離

    /* [4][2] 進入迴圈 寫 */
    while(1)
    {
        printf("please enter some text:");
        fgets(data_buffer, BUFFER_SIZE, stdin);

        // 輸入end，退出迴圈
        if(strncmp(data_buffer, "end", 3) == 0)
            break;

        send(sockfd, data_buffer, sizeof(data_buffer), 0);

        bzero(data_buffer, sizeof(data_buffer));

        recv(sockfd, data_buffer, sizeof(data_buffer), 0);

        printf("data_buff:%s\r\n", data_buffer);

    }

    /* [5] 退出 */
    close(sockfd);
    exit(0);
}

相關推薦

【TCP/IP】TCP伺服器併發處理&amp;原始碼

前言 本筆記記錄的是 單個服務端併發式處理多個客戶端。 下次有空在發個 單執行緒多個服務端併發式處理多種客戶端。其實就是本筆記的一個改良版，用到select() / poll() / epoll()。

【TCP/IP】TCP協議的流程圖解

一、TCP協議起步 1. 什麼是TCP協議 TCP是面向連線的協議，這是因為在一個應用程序可以開始向另一個應用程序傳送資料之前，這兩個程序必須先相互“握手”，即它們必須相互發送某些預備報文段，以建立確保資料傳輸的引

【網路基礎】TCP/IP協議

TCP和UDP TCP是面向連線的，UDP是無連線的； TCP是可靠傳輸，UDP不可靠； TCP是面向位元組流的，UDP是面向報文的；

【115期】TCP協議面試10連問

能不能說一說 TCP 和 UDP 的區別？ TCP是一個面向連線的、可靠的、基於位元組流的傳輸層協議。而UDP是一個面向無連線的傳輸層協議。和 UDP 相比，TCP 有三大核心特性：

【計算機網路】TCP的三次握手與四次揮手

今天講一下TCP的三次握手與四次揮手 1.首先說TCP 協議的特點：TCP是在不可靠的IP層之上實現的可靠的資料傳輸協議，它主要解決傳輸的可靠、有序、無丟失和不重複問題。TCP 是TCP/IP 體系中非常複雜的一個協議，主要特

【網路排查】TCP連線都是用TCP協議溝通的嗎？

TCP 連線都是用 TCP 協議溝通的嗎？ 一般來說 TCP 連線是標準的 TCP 三次握手完成的：

【改進版】 Tcp協議服務端和客戶端傳輸資料

前面一個版本的tcp傳輸 有點問題 1.讀取內容  發現如果兩端不是同一種編碼就會讀取亂碼，下面改進下可以自定義編碼

【web自動化】測試的登入處理

一.複用已有瀏覽器 應用場景：登入頁面需要使用者進行掃碼登入的 1.推出當前所有的谷歌瀏覽器

TCP/IP協議 TCP淺析

技術標籤：Socketsocket 文章目錄 TCP特點三次握手wireshark抓到的包分析包 四次揮手總結參考資料

【Python技巧】將jfif檔案處理為jpg檔案

技術標籤：python影象處理 方法如下： from PIL import Image jfif_path = ""# 在這裡填上jfif檔案的路徑

【基礎10】【自學筆記】python錯誤異常處理

一、如何丟擲異常？ 異常的產生有兩種來源： 一種是程式自動丟擲，比如1/0會自動丟擲 ZeroDivisionError

linux下一個網絡卡配置多個ip【虛擬ip】

來自：https://www.cnblogs.com/wanghuaijun/p/6155832.html Linux下配置網絡卡ip別名何謂ip別名？用windows的話說，就是為一個網絡卡配置多個ip。什麼場合增加ip別名能派上用場？佈網需要、多ip訪問測試、特定軟體

【專案實戰】多維度特徵處理

視訊連結：https://www.bilibili.com/video/BV1Y7411d7Ys?p=7 資料集的下載 通過百度網盤下載至程式碼的資料夾

【工具相關】ffmpeg音視訊處理

一、前言 ffprobe 是一個多媒體流分析工具。它從多媒體流中收集資訊，並且以人類和機器可讀的形式打印出來。

【十三】【vlc-anroid】EGL OpenGL模組元件載入原始碼實現分析

技術標籤：【vlc-android】【音視訊】vlcopengl 由第十二章節分析，可知，vlc使用OpenGL載入時，android端會載入【“opengl”】名的元件模組。 如下找到該元件宣告：

【React Router】React Router API錦囊以及原始碼（持續更新）

目錄Router 的 型別Router HooksuseParamsuseRouteMatchuseLoactionuseSearchParamsuseHistorywithRouter

【免費領取】雲開發工具箱小程式原始碼/遊戲取名助手小程式原始碼

雲開發工具箱小程式原始碼，遊戲取名助手小程式原始碼，搭建就不多說了，直接用微信開發者工具匯入，修改名字，釋出即可。

【Python】【爬蟲系列】【爬狼】002_自定義獲取網頁原始碼的函式

自定義獲取網頁原始碼的函式，基於上一篇爬狼系列筆記 在上一篇筆記（【Python】【爬蟲系列】【爬狼】001_urllib_get_獲取響應結果頁面程式碼 - 萌狼藍天 - 部落格園 (cnblogs.com/mllt)）我們將一個網頁，解析為

【刷題】牛客模擬面試 &amp;gt; 模擬面試報告

https://www.nowcoder.com/interview/ai/index 1-TCP協議的流量控制和擁塞控制  TCP的流量控制是基於視窗機制實現的：

node.js使用net模組建立伺服器和客戶端示例【基於TCP協議】

本文例項講述了node.js使用net模組建立伺服器和客戶端。分享給大家供大家參考，具體如下：