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

2024-05-09 19:28

本文主要是介绍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

相关文章

Python中Tkinter GUI编程详细教程

《Python中TkinterGUI编程详细教程》Tkinter作为Python编程语言中构建GUI的一个重要组件,其教程对于任何希望将Python应用到实际编程中的开发者来说都是宝贵的资源,这篇文... 目录前言1. Tkinter 简介2. 第一个 Tkinter 程序3. 窗口和基础组件3.1 创建窗

基于C++的UDP网络通信系统设计与实现详解

《基于C++的UDP网络通信系统设计与实现详解》在网络编程领域,UDP作为一种无连接的传输层协议,以其高效、低延迟的特性在实时性要求高的应用场景中占据重要地位,下面我们就来看看如何从零开始构建一个完整... 目录前言一、UDP服务器UdpServer.hpp1.1 基本框架设计1.2 初始化函数Init详解

Python字符串处理方法超全攻略

《Python字符串处理方法超全攻略》字符串可以看作多个字符的按照先后顺序组合,相当于就是序列结构,意味着可以对它进行遍历、切片,:本文主要介绍Python字符串处理方法的相关资料,文中通过代码介... 目录一、基础知识:字符串的“不可变”特性与创建方式二、常用操作:80%场景的“万能工具箱”三、格式化方法

Spring Boot 处理带文件表单的方式汇总

《SpringBoot处理带文件表单的方式汇总》本文详细介绍了六种处理文件上传的方式,包括@RequestParam、@RequestPart、@ModelAttribute、@ModelAttr... 目录方式 1:@RequestParam接收文件后端代码前端代码特点方式 2:@RequestPart接

Linux内核定时器使用及说明

《Linux内核定时器使用及说明》文章详细介绍了Linux内核定时器的特性、核心数据结构、时间相关转换函数以及操作API,通过示例展示了如何编写和使用定时器,包括按键消抖的应用... 目录1.linux内核定时器特征2.Linux内核定时器核心数据结构3.Linux内核时间相关转换函数4.Linux内核定时

Python+FFmpeg实现视频自动化处理的完整指南

《Python+FFmpeg实现视频自动化处理的完整指南》本文总结了一套在Python中使用subprocess.run调用FFmpeg进行视频自动化处理的解决方案,涵盖了跨平台硬件加速、中间素材处理... 目录一、 跨平台硬件加速:统一接口设计1. 核心映射逻辑2. python 实现代码二、 中间素材处

Linux镜像文件制作方式

《Linux镜像文件制作方式》本文介绍了Linux镜像文件制作的过程,包括确定磁盘空间布局、制作空白镜像文件、分区与格式化、复制引导分区和其他分区... 目录1.确定磁盘空间布局2.制作空白镜像文件3.分区与格式化1) 分区2) 格式化4.复制引导分区5.复制其它分区1) 挂载2) 复制bootfs分区3)

Python实现快速扫描目标主机的开放端口和服务

《Python实现快速扫描目标主机的开放端口和服务》这篇文章主要为大家详细介绍了如何使用Python编写一个功能强大的端口扫描器脚本,实现快速扫描目标主机的开放端口和服务,感兴趣的小伙伴可以了解下... 目录功能介绍场景应用1. 网络安全审计2. 系统管理维护3. 网络故障排查4. 合规性检查报错处理1.

Go异常处理、泛型和文件操作实例代码

《Go异常处理、泛型和文件操作实例代码》Go语言的异常处理机制与传统的面向对象语言(如Java、C#)所使用的try-catch结构有所不同,它采用了自己独特的设计理念和方法,:本文主要介绍Go异... 目录一:异常处理常见的异常处理向上抛中断程序恢复程序二:泛型泛型函数泛型结构体泛型切片泛型 map三:文

JavaWeb项目创建、部署、连接数据库保姆级教程(tomcat)

《JavaWeb项目创建、部署、连接数据库保姆级教程(tomcat)》:本文主要介绍如何在IntelliJIDEA2020.1中创建和部署一个JavaWeb项目,包括创建项目、配置Tomcat服务... 目录简介:一、创建项目二、tomcat部署1、将tomcat解压在一个自己找得到路径2、在idea中添加