02.07 TCP服务器与客户端的搭建

news/2025/2/9 1:32:34 标签: 服务器, 运维

一.思维导图

二.使用动态协议包实现服务器与客户端

1. 协议包的结构定义

首先,是协议包的结构定义。在两段代码中,pack_t结构体都被用来表示协议包:

typedef struct Pack {
    int size;        // 记录整个协议包的实际大小
    enum Type type;  // 协议包的类型
    char buf[2048];  // 存储实际数据
    int count;       // 记录buf中已使用的字节数
} pack_t;

enum Type定义了协议包的类型,例如TYPE_REGISTTYPE_LOGIN,分别表示注册和登录操作。

2. 服务器端的协议包解析

服务器端代码中,read_pack函数负责解析从客户端接收到的协议包。该函数的主要步骤如下:

  1. 读取数据大小:从buf中读取前2个字节,表示接下来要读取的数据大小。

  2. 读取数据:根据读取到的数据大小,从buf中读取相应长度的数据。

  3. 处理数据:将读取到的数据打印出来。

    void read_pack(pack_t* pack) {
        char *buf = pack->buf;
        int readed_size = 0;
        while(1) {
            short data_size = *(short*)(buf + readed_size);
            if(data_size == 0) {
                printf("数据解析完毕\n");
                break;
            }
            readed_size += 2;
            char temp[data_size + 1];
            memset(temp, 0, data_size + 1);
            memcpy(temp, buf + readed_size, data_size);
            readed_size += data_size;
            printf("temp = %s\n", temp);
        }
    }

    3. 客户端的协议包构建

    在客户端代码中,append函数负责将数据按照协议格式添加到pack_t结构体中。该函数的主要步骤如下:

  4. 记录数据长度:将数据的长度存储在buf的前2个字节中。

  5. 存储数据:将数据本身存储在buf中。

  6. 更新协议包大小:更新pack_t结构体中的sizecount字段。

    void append(pack_t* pack, const char* data) {
        char* buf = pack->buf;
        int len = strlen(data);
        *(short*)(buf + pack->count) = len;
        pack->count += 2;
        memcpy(buf + pack->count, data, len);
        pack->count += len;
        pack->size = pack->count + 8;
    }

    4. 客户端与服务器的交互

    在客户端代码中,用户输入账号和密码后,append函数将数据添加到协议包中,然后通过write函数将协议包发送给服务器

    while(1) {
        pack_t pack = {0};
        pack.type = TYPE_LOGIN;
        char name[20] = "";
        char pswd[20] = "";
        printf("请输入账号:");
        scanf("%19s", name);
        while(getchar() != 10);
        printf("请输入密码:");
        scanf("%19s", pswd);
        while(getchar() != 10);
        append(&pack, name);
        append(&pack, pswd);
        write(client, &pack, pack.size);
    }

    服务器端,read函数接收客户端发送的协议包,并调用read_pack函数解析数据。

    while(1) {
        int pack_size = 0;
        read(client, &pack_size, 4);
        pack_t pack = {0};
        read(client, (char*)&pack + 4, pack_size - 4);
        pack.size = pack_size;
        read_pack(&pack);
    }

    5. 总结

    通过这两段代码,我们可以看到如何在C语言中实现一个简单的网络协议包的构建与解析。服务器端负责接收和解析协议包,而客户端则负责构建和发送协议包。这种设计模式在网络编程中非常常见,理解其原理对于开发网络应用程序至关重要。

      6.完整代码

1>服务器

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>

typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;
typedef struct sockaddr_un addr_un_t;

enum Type {
    TYPE_REGIST,
    TYPE_LOGIN
};

typedef struct Pack {
    int size;
    enum Type type;
    char buf[2048];
    int count;
} pack_t;

void read_pack(pack_t *pack) {
    char *buf = pack->buf;
    int readed_size = 0;

    while (1) {
        short data_size = *(short *)(buf + readed_size);
        if (data_size == 0) {
            printf("数据解析完毕\n");
            break;
        }
        readed_size += 2;
        char temp[data_size + 1];
        memset(temp, 0, data_size + 1);
        memcpy(temp, buf + readed_size, data_size);
        readed_size += data_size;
        printf("temp = %s\n", temp);
    }
}

int main(int argc, const char *argv[]) {
    if (argc != 2) {
        printf("请输入端口号\n");
        return 1;
    }
    int port = atoi(argv[1]);

    int server = socket(AF_INET, SOCK_STREAM, 0);
    addr_in_t addr = {0};
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = inet_addr("0.0.0.0");

    if (bind(server, (addr_t *)&addr, sizeof(addr)) == -1) {
        perror("bind");
        return 1;
    }

    listen(server, 10);

    addr_in_t client_addr = {0};
    int client_addr_len = sizeof(client_addr);
    int client = accept(server, (addr_t *)&client_addr, &client_addr_len);
    if (client != -1) {
        printf("有客户端连接\n");
    }

    while (1) {
        int pack_size = 0;
        read(client, &pack_size, 4);
        pack_t pack = {0};
        read(client, (char *)&pack + 4, pack_size - 4);
        pack.size = pack_size;
        read_pack(&pack);
    }

    return 0;
}

2>客户端

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>

typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;
typedef struct sockaddr_un addr_un_t;

enum Type {
    TYPE_REGIST,
    TYPE_LOGIN
};

typedef struct Pack {
    int size;
    enum Type type;
    char buf[2048];
    int count;
} pack_t;

void append(pack_t *pack, const char *data) {
    char *buf = pack->buf;
    int len = strlen(data);
    *(short *)(buf + pack->count) = len;
    memcpy(buf + pack->count + 2, data, len);
    pack->count += 2;
    pack->count += len;
    pack->size = pack->count + 8;
}

int main(int argc, const char *argv[]) {
    if (argc != 2) {
        printf("请输入端口号\n");
        return 1;
    }
    int port = atoi(argv[1]);

    int client = socket(AF_INET, SOCK_STREAM, 0);
    addr_in_t addr = {0};
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = inet_addr("192.168.60.77");

    if (connect(client, (addr_t *)&addr, sizeof(addr)) == -1) {
        perror("connect");
        return 1;
    }

    while (1) {
        pack_t pack = {0};
        pack.type = TYPE_LOGIN;
        char name[20] = "";
        char pswd[20] = "";
        printf("请输入账号:");
        scanf("%19s", name);
        while (getchar() != '\n');
        printf("请输入密码:");
        scanf("%19s", pswd);
        while (getchar() != '\n');

        append(&pack, name);
        append(&pack, pswd);

        write(client, &pack, pack.size);
    }

    return 0;
}

三、基于以上代码,将读取到的一条条代码保存到链表中

1.服务器代码

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <semaphore.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>

// 定义网络地址结构体类型
typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;
typedef struct sockaddr_un addr_un_t;

// 定义协议包类型枚举
enum Type {
	TYPE_REGIST,  // 注册类型
	TYPE_LOGIN    // 登录类型
};

// 定义协议包结构体
typedef struct Pack {
	int size;         // 协议包的总大小
	enum Type type;   // 协议包的类型
	char buf[2048];   // 数据缓冲区
	int count;        // 记录缓冲区中已使用的字节数
} pack_t;

// 定义链表节点结构体
typedef struct Node {
	char* data;       
	struct Node* next; 
} Node;

// 创建新节点
Node* create_node(const char* data) {
	Node* node = (Node*)malloc(sizeof(Node)); 
    if (data != NULL) {
    size_t len = strlen(data) + 1;
    node->data = (char*)malloc(len * sizeof(char));
    if (node->data != NULL) {
       strcpy(node->data, data);
    } else {
        fprintf(stderr, "内存分配失败!\n");
        exit(EXIT_FAILURE);
    }
} else {
    // 处理输入为 NULL 的情况
    node->data = NULL;
} 
	node->next = NULL; 
	return node;
}

// 将数据添加到链表
void append_to_list(Node** head, const char* data) {
	Node* new_node = create_node(data);
	if (*head == NULL) {
		*head = new_node; 
	} else {
		Node* current = *head;
		while (current->next != NULL) {
			current = current->next;
		}
		current->next = new_node; 
	}
}

// 释放链表内存
void free_list(Node* head) {
	Node* current = head;
	while (current != NULL) {
		Node* next = current->next; 
		free(current->data); 
		free(current); 
		current = next; 
	}
}

// 解析协议包并将数据保存到链表
void read_pack(pack_t* pack, Node** head) {
	char *buf = pack->buf; 
	int readed_size = 0; // 记录已读取的字节数
	while(1) {
		short data_size = *(short*)(buf + readed_size); // 读取数据大小
		if(data_size == 0) {
			printf("数据解析完毕\n"); 
			break;
		}
		readed_size += 2; 
		char temp[data_size + 1]; 
		memset(temp, 0, data_size + 1); 
		memcpy(temp, buf + readed_size, data_size); 
		readed_size += data_size; // 更新已读取的字节数
		printf("temp = %s\n", temp); 
		append_to_list(head, temp); // 将数据添加到链表
	}
}

// 主函数
int main(int argc, const char *argv[]) 
{
	if(argc != 2) 
	{
		printf("请输入端口号\n"); 
		return 1;
	}
	int port = atoi(argv[1]); // 将端口号字符串转换为整数

	// 创建服务器套接字
	int server = socket(AF_INET, SOCK_STREAM, 0);
	addr_in_t addr = {0};
	addr.sin_family = AF_INET; 
	addr.sin_port = htons(port); 
	addr.sin_addr.s_addr = inet_addr("0.0.0.0"); 

	// 绑定套接字到地址
	if(bind(server, (struct sockaddr*)&addr, sizeof(addr)) == -1) 
	{
		perror("bind"); // 绑定失败
		return 1;
	}

	// 监听连接
	listen(server, 10);

	// 接受客户端连接
	addr_in_t client_addr = {0};
	int client_addr_len = sizeof(client_addr);
	int client = accept(server, (struct sockaddr*)&client_addr, &client_addr_len);
	if(client != -1) 
	{
		printf("有客户端连接\n"); 
	}

	Node* head = NULL; // 初始化链表头

	while(1) {
		int pack_size = 0;
		int ret = read(client, &pack_size, 4); // 读取协议包大小
		if (ret == -1) 
		{
			printf("客户端断开连接\n"); 
			break; // 退出循环
		}
		pack_t pack = {0};

		ret = read(client, (char*)&pack + 4, pack_size - 4); // 读取协议包数据
		if (ret == -1) {
			printf("客户端断开连接\n"); // 客户端断开连接
			break; // 退出循环
		}
		pack.size = pack_size; // 设置协议包大小

		read_pack(&pack, &head); // 解析协议包并保存数据到链表

		// 打印链表中的数据
		Node* current = head;
		printf("--------保存的数据如下--------\n");
		while (current != NULL) {
			printf("链表数据: %s\n", current->data);
			current = current->next;
		}
		printf("\n");
	}

	free_list(head); // 释放链表内存
	return 0;
}

2.客户端代码

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <semaphore.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>

typedef struct sockaddr_in addr_in_t;  
typedef struct sockaddr addr_t;        
typedef struct sockaddr_un addr_un_t;  

// 定义消息类型枚举
enum Type{
    TYPE_REGIST,    // 注册类型
    TYPE_LOGIN      // 登录类型
};

// 自定义数据包结构体
typedef struct Pack
{
    int size;           
    enum Type type;     
    char buf[2048];    
    int count;         
}pack_t;


void append(pack_t* pack, const char* data)
{
    char *buf = pack->buf;
    int len = strlen(data);
    
    *(short*)(buf + pack->count) = len;  
    pack->count += 2;                   
    
    memcpy(buf + pack->count, data, len); 
    pack->count += len;                   
    // 更新数据包总大小(头部8字节 + 数据长度)
    pack->size = pack->count + 8;         
}

int main(int argc, const char *argv[])
{
    if(argc != 2)
    {
        printf("Usage: %s <port>\n", argv[0]);
        return 1;
    }

    int port = atoi(argv[1]);  // 将端口参数转为整数

    // 创建TCP套接字
    int client = socket(AF_INET, SOCK_STREAM, 0);
    if(client == -1){
        perror("socket creation failed");
        return 1;
    }

    // 配置服务器地址信息
    addr_in_t addr = {0};
    addr.sin_family = AF_INET;                   
    addr.sin_port = htons(port);                
    addr.sin_addr.s_addr = inet_addr("192.168.126.233");

    // 连接服务器
    if(connect(client, (addr_t*)&addr, sizeof(addr)) == -1)
    {
        perror("connect failed");
        return 1;
    }

    while(1)
    {
        pack_t pack = {0};      
        pack.type = TYPE_LOGIN; 
        
        char name[20] = "";
        char pswd[20] = "";
        
        // 获取用户名输入
        printf("请输入账号:");
        scanf("%19s", name);
        while(getchar() != '\n'); // 清空输入缓冲区
        
        // 获取密码输入
        printf("请输入密码:");
        scanf("%19s", pswd);
        while(getchar() != '\n'); // 清空输入缓冲区
        
        // 将用户名和密码打包到数据包
        append(&pack, name);
        append(&pack, pswd);
        
        // 发送整个数据包(发送大小为计算后的总大小)
        write(client, &pack, pack.size);
    }
    
    return 0;
}


http://www.niftyadmin.cn/n/5845435.html

相关文章

STM32G474--Whetstone程序移植(单精度)笔记

1 准备基本工程代码 参考这篇笔记从我的仓库中选择合适的基本工程&#xff0c;进行程序移植。这里我用的是stm32g474的基本工程。 使用git clone一个指定文件或者目录 2 移植程序 2.1 修改Whetstone.c 主要修改原本变量定义的类型&#xff0c;以及函数接口全部更换为单精度…

Android修行手册-五种比较图片相似或相同

Unity3D特效百例案例项目实战源码Android-Unity实战问题汇总游戏脚本-辅助自动化Android控件全解手册再战Android系列Scratch编程案例软考全系列Unity3D学习专栏蓝桥系列ChatGPT和AIGC👉关于作者 专注于Android/Unity和各种游戏开发技巧,以及各种资源分享(网站、工具、素材…

【Leetcode 热题 100】136. 只出现一次的数字

问题背景 给你一个 非空 整数数组 n u m s nums nums&#xff0c;除了某个元素只出现一次以外&#xff0c;其余每个元素均出现两次。找出那个只出现了一次的元素。 你必须设计并实现线性时间复杂度的算法来解决此问题&#xff0c;且该算法只使用常量额外空间。 数据约束 1 ≤…

MCU应用踩坑笔记(ADC 中断 / 查询法)

问题描述 IC&#xff1a;SC92F7596,在使用过程中&#xff0c;发现一个问题&#xff0c;就是我们使用到了ADC功能&#xff0c;程序的代码如下&#xff1a; ADC采样周期200ms &#xff0c;采样个数&#xff1a;4 在使用过程中&#xff0c;因配置了ADC中断使能&#xff0c;在中断…

洛谷P8742 [蓝桥杯 2021 省 AB] 砝码称重(dp初始)

归纳蓝桥杯的这道题总结了一定对于dp的看法&#xff0c;虽然还没看到y总的动态规划&#xff0c;自己搜了搜上学期算法中学到的01背包问题。 首先动态规划问题最重要的是状态转移方程&#xff0c;将问题抽象成数学问题&#xff0c;列出方程就可以得解。 #include<cstdio> …

数据结构在 Web 开发中的重要性与应用

数据结构是 Web 开发的基石&#xff0c;直接关系到应用程序的效率、可扩展性和可维护性。 根据实际需求选择合适的数据结构&#xff0c;能够有效优化性能、简化代码&#xff0c;并提升用户体验。 本文将深入探讨 PHP 和 Laravel 中的常用数据结构&#xff0c;并结合实际案例&am…

【C++】 STL -- 算法(一)

【C】 STL – 算法&#xff08;一&#xff09; 文章目录 【C】 STL -- 算法&#xff08;一&#xff09;前言一、函数对象二、谓词三、内建函数对象四、适配器总结 前言 本篇文章将讲到函数对象&#xff0c;谓词&#xff0c;内建函数对象&#xff0c;适配器。 一、函数对象 本质…

用AVFrame + AVPacket 完成accede编码和直接用ffmpeg命令行实现acc编码的对比

在使用 FFmpeg 进行 AAC 音频编码时,可以选择两种方式:通过编程接口(如 AVFrame 和 AVPacket)实现 AAC 编码,或者直接使用 FFmpeg 命令行工具。这两种方式各有特点,适用于不同的场景。以下是对两种方法的详细分析,包括它们的区别、优缺点以及适用场景。 一、通过 AVFram…