Linux网络编程(四) 同时处理一个端口的UDP与TCP连接

2024-05-09 19:28
文章标签 linux 连接 编程 处理 udp 网络 端口 tcp

本文主要是介绍Linux网络编程(四) 同时处理一个端口的UDP与TCP连接，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

从bind系统调用的参数来看，一个socket只能与一个socket地址绑定，即一个socket只能用来监听一个端口。因此，服务器如果要同时监听多个端口，就必须创建多个socket，并将它们分别绑定到各个端口上。这样一来，服务器程序就需要同时管理多个监听socket，I/O复用技术就有了用武之地。

即使是同一个端口，如果服务器要同时处理该端口上的TCP和UDP请求，则也需要创建两个不同的socket：一个是流socket，另一个是数据报socket，并将它们都绑定到该端口上。

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <assert.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <fcntl.h>
#include <stdlib.h>#define MAX_EVENT_NUMBER 1024
#define TCP_BUFFER_SIZE 512
#define UDP_BUFFER_SIZE 1024/* 设置文件为非阻塞 */
int setnonblocking(int fd)
{int old_option = fcntl(fd, F_GETFL);int new_option = old_option | O_NONBLOCK;fcntl(fd, F_SETFL, new_option);return old_option;
}/* 添加epoll事件，ET边缘触发模式 */
void addfd(int epollfd, int fd)
{struct epoll_event event;event.data.fd = fd;event.events = EPOLLIN | EPOLLET;epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &event);setnonblocking(fd);
}int main()
{const char *ip = "127.0.0.1";int port = 8080;int ret = 0;struct sockaddr_in address;bzero(&address, sizeof(address));address.sin_family = AF_INET;inet_pton(AF_INET, ip, &address.sin_addr);address.sin_port = htons(port);/*创建TCP socket，并将其绑定到端口port上*/int tcpfd = socket(PF_INET, SOCK_STREAM, 0);ret = bind(tcpfd, (struct sockaddr *)&address, sizeof(address));assert(ret != -1);ret = listen(tcpfd, 5);assert(ret != -1);/*创建UDP socket，并将其绑定到端口port上*/bzero(&address, sizeof(address));address.sin_family = AF_INET;inet_pton(AF_INET, ip, &address.sin_addr);address.sin_port = htons(port);int udpfd = socket(PF_INET, SOCK_DGRAM, 0);ret = bind(udpfd, (struct sockaddr *)&address, sizeof(address));assert(ret != -1);/* 创建epoll事件表 */struct epoll_event events[MAX_EVENT_NUMBER];int epollfd = epoll_create(256);assert(epollfd != -1);/* 注册TCP socket和UDP socket上的可读事件 */addfd(epollfd, tcpfd);addfd(epollfd, udpfd);while (1){int number = epoll_wait(epollfd, events, MAX_EVENT_NUMBER, -1);if (number < 0){printf("epoll failure\n");break;}// 处理事件for (int i = 0; i < number; i++){int sockfd = events[i].data.fd;// 如果是tcp连接事件，那么将其注册到epoll事件表中if (sockfd == tcpfd){struct sockaddr_in client_address;socklen_t client_addrlength = sizeof(client_address);int connfd = accept(tcpfd, (struct sockaddr *)&client_address, &client_addrlength);addfd(epollfd, connfd);}// 如果是udp连接事件，那么else if (sockfd == udpfd){char buf[UDP_BUFFER_SIZE];memset(buf, '\0', UDP_BUFFER_SIZE);struct sockaddr_in client_address;socklen_t client_addrlength = sizeof(client_address);// 将udp中的数据读取到buf中ret = recvfrom(udpfd, buf, UDP_BUFFER_SIZE - 1, 0,(struct sockaddr *)&client_address, &client_addrlength);// 如果收到了数据,打印if (ret > 0){printf("%s\n",buf);}}// 如果是一个socket有数据输入else if (events[i].events & EPOLLIN){char buf[TCP_BUFFER_SIZE];while (1){memset(buf, '\0', TCP_BUFFER_SIZE);ret = recv(sockfd, buf, TCP_BUFFER_SIZE - 1, 0);if (ret < 0){   // 如果当前操作非阻塞，操作无法立即完成，那么先不做处理if ((errno == EAGAIN) || (errno == EWOULDBLOCK)) break;// 如果是出现了其他错误，那么关闭socketclose(sockfd);break;}// 如果数据已经读完else if (ret == 0){close(sockfd);}// 如果是其他情况，那么返回收到的数据else{printf("%s\n",buf);}}}else{printf("something else happened\n");}}}close(tcpfd);return 0;
}

主要的精髓在这里

int tcpfd = socket(PF_INET, SOCK_STREAM, 0);
int udpfd = socket(PF_INET, SOCK_DGRAM, 0);

看一下客户端：

#include <stdio.h>  
#include <stdlib.h>  
#include <string.h>  
#include <unistd.h>  
#include <arpa/inet.h>  
#include <sys/types.h>  
#include <sys/socket.h>  
#include <netinet/in.h>  #define SERVER_IP "127.0.0.1"  
#define SERVER_PORT 8080  
#define BUFFER_SIZE 1024  int main() {  int tcp_socket, udp_socket;  struct sockaddr_in server_addr;  char buffer[BUFFER_SIZE];  ssize_t bytes_sent;  // 创建TCP socket  tcp_socket = socket(AF_INET, SOCK_STREAM, 0);  if (tcp_socket == -1) {  perror("TCP socket creation failed");  exit(EXIT_FAILURE);  }  // 创建UDP socket  udp_socket = socket(AF_INET, SOCK_DGRAM, 0); if (udp_socket == -1) {  perror("UDP socket creation failed");  exit(EXIT_FAILURE);  }  // 设置服务器地址信息  memset(&server_addr, 0, sizeof(server_addr));  server_addr.sin_family = AF_INET;  server_addr.sin_port = htons(SERVER_PORT);  if (inet_pton(AF_INET, SERVER_IP, &server_addr.sin_addr) <= 0) {  perror("Invalid server address");  close(tcp_socket);  close(udp_socket);  exit(EXIT_FAILURE);  }   // 连接到TCP服务器  if (connect(tcp_socket, (struct sockaddr *) &server_addr, sizeof(server_addr)) == -1) {  perror("TCP connection failed");  close(tcp_socket);  close(udp_socket);  exit(EXIT_FAILURE);  }  printf("Connected to server\n");  // 发送TCP消息给服务器  const char *tcp_message = "Hello, this is TCP message!\n";bytes_sent = send(tcp_socket, tcp_message, strlen(tcp_message), 0);  if (bytes_sent == -1) {perror("TCP send failed");close(tcp_socket);close(udp_socket);exit(EXIT_FAILURE);}// 发送UDP消息给服务器  const char *udp_message = "Hello, this is UDP message!\n";bytes_sent = sendto(udp_socket, udp_message, strlen(udp_message), 0, (struct sockaddr *) &server_addr, sizeof(server_addr));if (bytes_sent == -1) {perror("UDP sendto failed");close(tcp_socket);close(udp_socket);exit(EXIT_FAILURE);}// 关闭连接close(tcp_socket);close(udp_socket);return 0;
}

仿真

image-20240507214628171

这篇关于Linux网络编程(四) 同时处理一个端口的UDP与TCP连接的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/974283。

相关文章

linux hostname设置全过程

linux hostname设置全过程

《linuxhostname设置全过程》：本文主要介绍linuxhostname设置全过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录查询hostname设置步骤其它相关点hostid/etc/hostsEDChina编程A工具license破解注意事项总结以RHE
阅读更多...
Spring Boot @RestControllerAdvice全局异常处理最佳实践

Spring Boot @RestControllerAdvice全局异常处理最佳实践

《SpringBoot@RestControllerAdvice全局异常处理最佳实践》本文详解SpringBoot中通过@RestControllerAdvice实现全局异常处理,强调代码复用、统... 目录前言一、为什么要使用全局异常处理？二、核心注解解析1. @RestControllerAdvice2
阅读更多...
Linux中压缩、网络传输与系统监控工具的使用完整指南

Linux中压缩、网络传输与系统监控工具的使用完整指南

《Linux中压缩、网络传输与系统监控工具的使用完整指南》在Linux系统管理中,压缩与传输工具是数据备份和远程协作的桥梁,而系统监控工具则是保障服务器稳定运行的眼睛,下面小编就来和大家详细介绍一下它... 目录引言一、压缩与解压：数据存储与传输的优化核心1. zip/unzip：通用压缩格式的便捷操作2.
阅读更多...
Linux中SSH服务配置的全面指南

Linux中SSH服务配置的全面指南

《Linux中SSH服务配置的全面指南》作为网络安全工程师,SSH（SecureShell）服务的安全配置是我们日常工作中不可忽视的重要环节,本文将从基础配置到高级安全加固,全面解析SSH服务的各项参... 目录概述基础配置详解端口与监听设置主机密钥配置认证机制强化禁用密码认证禁止root直接登录实现双因素
阅读更多...
Go语言数据库编程GORM 的基本使用详解

Go语言数据库编程GORM 的基本使用详解

《Go语言数据库编程GORM的基本使用详解》GORM是Go语言流行的ORM框架,封装database/sql,支持自动迁移、关联、事务等,提供CRUD、条件查询、钩子函数、日志等功能,简化数据库操作... 目录一、安装与初始化1. 安装 GORM 及数据库驱动2. 建立数据库连接二、定义模型结构体三、自动迁
阅读更多...
MySQL中的表连接原理分析

MySQL中的表连接原理分析

《MySQL中的表连接原理分析》：本文主要介绍MySQL中的表连接原理分析,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录1、背景2、环境3、表连接原理【1】驱动表和被驱动表【2】内连接【3】外连接【4编程】嵌套循环连接【5】join buffer4、总结1、背景
阅读更多...
在Linux终端中统计非二进制文件行数的实现方法

在Linux终端中统计非二进制文件行数的实现方法

《在Linux终端中统计非二进制文件行数的实现方法》在Linux系统中,有时需要统计非二进制文件（如CSV、TXT文件）的行数,而不希望手动打开文件进行查看,例如,在处理大型日志文件、数据文件时,了解... 目录在linux终端中统计非二进制文件的行数技术背景实现步骤1. 使用wc命令2. 使用grep命令
阅读更多...
Linux如何快速检查服务器的硬件配置和性能指标

Linux如何快速检查服务器的硬件配置和性能指标

《Linux如何快速检查服务器的硬件配置和性能指标》在运维和开发工作中,我们经常需要快速检查Linux服务器的硬件配置和性能指标,本文将以CentOS为例,介绍如何通过命令行快速获取这些关键信息,... 目录引言一、查询CPU核心数编程（几C？）1. 使用 nproc（最简单）2. 使用 lscpu（详细信
阅读更多...
前端如何通过nginx访问本地端口

前端如何通过nginx访问本地端口

《前端如何通过nginx访问本地端口》：本文主要介绍前端如何通过nginx访问本地端口的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、nginx安装1、下载（1）下载地址（2）系统选择（3）版本选择2、安装部署（1）解压（2）配置文件修改（3）启动（4）
阅读更多...
linux重启命令有哪些? 7个实用的Linux系统重启命令汇总

linux重启命令有哪些? 7个实用的Linux系统重启命令汇总

《linux重启命令有哪些?7个实用的Linux系统重启命令汇总》Linux系统提供了多种重启命令，常用的包括shutdown-r、reboot、init6等，不同命令适用于不同场景，本文将详细... 在管理和维护 linux 服务器时，完成系统更新、故障排查或日常维护后，重启系统往往是必不可少的步骤。本文
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2