Linux udp poll函数

一、UDP服务器与客户端的区别

对于UDP服务器与客户端,两者都可以通过sendto和recvfrom函数收发数据,它们的主要区别是:

1.服务器一般是等待并响应来自客户端的请求,客户端则是主动发送请求并且等待服务器的响应。

2.服务器端要将地址和端口号绑定,如果不绑定就无法使用recvfrom函数接受数据(也就是说服务器需要调用bind函数将一个套接字与一个地址绑定,而客户端不需要)。所以对于UDP,其服务器与其说是服务端不如说是后发端。

所以如果阅读一份源码,要快速判断其是UDP服务器还是客户端,一个简单的方法是查看代码中是否调用了bind函数就可以了。

二、UDP是否可以使用select/poll/epoll

UDP是一种无连接的传输协议,因此通常情况下没有必要使用多路复用。UDP只是逐个接收数据段(数据包),并按照其规则进行处理,其并不关心单个数据包所属的任何特定数据流或数据包。但是UDP是可以使用select/poll/epoll的,某些开源软件,比如FFmpeg内部使用了这些多路复用来处理UDP。

三、使用poll函数编写UDP客户端、服务器程序

代码摘自:《IPV4 UDP server client program with Poll system call》

UDP客户端代码udpClient.c:

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#include <arpa/inet.h>

#include <signal.h>

#include <poll.h>

#define BUFFER_SIZE 1024

int client_socket = -1;

static void sigint_handler(int signo)

{

(void)close(client_socket);

sleep(2);

(void)printf("Caught sigINT!\n");

exit(EXIT_SUCCESS);

}

void validate_convert_port(

char *port_str,

struct sockaddr_in *sock_addr)

{

int port;

if (port_str == NULL) {

perror("Invalid port_str\n");

exit(EXIT_FAILURE);

}

if (sock_addr == NULL) {

perror("Invalid sock_addr\n");

exit(EXIT_FAILURE);

}

port = atoi(port_str);

if (port == 0) {

perror("Invalid port\n");

exit(EXIT_FAILURE);

}

sock_addr->sin_port = htons(

(uint16_t)port);

printf("Port: %d\n",

ntohs(sock_addr->sin_port));

}

void validate_convert_addr(

char *ip_str,

struct sockaddr_in *sock_addr)

{

if (ip_str == NULL) {

perror("Invalid ip_str\n");

exit(EXIT_FAILURE);

}

if (sock_addr == NULL) {

perror("Invalid sock_addr\n");

exit(EXIT_FAILURE);

}

printf("IP Address: %s\n", ip_str);

if (inet_pton(AF_INET, ip_str,

&(sock_addr->sin_addr)) <= 0) {

perror("Invalid address\n");

exit(EXIT_FAILURE);

}

}

void recv_data(char *buffer)

{

int ret, len;

len = recvfrom(client_socket,

buffer, BUFFER_SIZE, 0, NULL, NULL);

if (len > 0) {

buffer[len] = '\0';

(void)printf("Received: %s\n",

buffer);

} else if (len == 0) {

printf("Connection closed\n");

exit(EXIT_FAILURE);

}

}

void register_signal_handler(

int signum,

void (*handler)(int))

{

if (signal(signum, handler) == SIG_ERR)

{

printf("Cannot handle signal\n");

exit(EXIT_FAILURE);

}

}

int main(int argc, char *argv[])

{

int ret, len;

struct sockaddr_in

server_addr;

char buffer[BUFFER_SIZE];

struct pollfd fds[1];

char *str = "HI";

register_signal_handler(SIGINT,

sigint_handler);

if (argc != 3) {

printf("%s\n",

argv[0]);

exit(EXIT_FAILURE);

}

memset(&server_addr, 0,

sizeof(server_addr));

server_addr.sin_family = AF_INET;

validate_convert_port(argv[1],

&server_addr);

validate_convert_addr(argv[2],

&server_addr);

client_socket = socket(AF_INET,

SOCK_DGRAM,

IPPROTO_UDP);

if (client_socket < 0) {

perror("socket");

return -1;

}

while (1) {

ret = sendto(client_socket, str,

strlen(str), 0,

(struct sockaddr*)&server_addr,

sizeof(server_addr));

printf("sendbuffer = %s\n", str);

if (ret < 0) {
   perror("send error\n");
   (void)close(client_socket);
   break;
}

fds[0].fd = client_socket;  
fds[0].events = POLLIN;

ret = poll(fds, 2, 1000);

if (ret < 0) {
   perror("poll");
   (void)close(client_socket);
   break;
}

if (fds[0].revents & POLLIN) {
   recv_data(buffer);
}

}

(void)close(client_socket);

return 0;

}

UDP服务器代码udpServer.c:

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#include <arpa/inet.h>

#include <signal.h>

#include <poll.h>

#define BUFFER_SIZE 1024

int server_socket = -1;

static void sigint_handler(int signo)

{

(void)close(server_socket);

sleep(2);

(void)printf("Caught sigINT!\n");

exit(EXIT_SUCCESS);

}

void register_signal_handler(

int signum,

void (*handler)(int))

{

if (signal(signum, handler) == SIG_ERR) {

printf("Cannot handle signal\n");

exit(EXIT_FAILURE);

}

}

void validate_convert_port(

char *port_str,

struct sockaddr_in *sock_addr)

{

int port;

if (port_str == NULL) {

perror("Invalid port_str\n");

exit(EXIT_FAILURE);

}

if (sock_addr == NULL) {

perror("Invalid sock_addr\n");

exit(EXIT_FAILURE);

}

port = atoi(port_str);

if (port == 0) {

perror("Invalid port\n");

exit(EXIT_FAILURE);

}

sock_addr->sin_port = htons(

(uint16_t)port);

printf("Port: %d\n",

ntohs(sock_addr->sin_port));

}

void recv_send(

char *buffer,

struct sockaddr_in *client_addr)

{

int len, ret;

socklen_t client_addr_len = sizeof(

client_addr);

len = recvfrom(server_socket,

buffer, BUFFER_SIZE, 0,

(struct sockaddr*)&client_addr,

&client_addr_len);

if (len > 0) {

buffer[len] = '\0';

printf("Received: %s\n",

buffer);

memset(buffer, 0,
sizeof(buffer));
strncpy(buffer, "HELLO",
strlen("HELLO") + 1);
buffer[strlen(buffer) + 1] = '\0';

ret = sendto(server_socket,
buffer,
strlen(buffer), 0,
(struct sockaddr*)&client_addr,
client_addr_len);

if (ret < 0) {
   perror("sendto");
   exit(EXIT_FAILURE);
  }
} else if (len < 0) {
    perror("recvfrom");
    exit(EXIT_FAILURE);
}
printf("Sentbuffer = %s\n",
buffer);

}

int main(int argc, char *argv[])

{

int ret;

struct sockaddr_in

server_addr,

client_addr;

char buffer[BUFFER_SIZE];

struct pollfd fds[1];

register_signal_handler(SIGINT,

sigint_handler);

if (argc != 2) {

printf("%s ",

argv[0]);

exit(EXIT_FAILURE);

}

memset(&server_addr, 0,

sizeof(server_addr));

server_addr.sin_family = AF_INET;

server_addr.sin_addr.s_addr =

INADDR_ANY;

validate_convert_port(argv[1],

&server_addr);

server_socket = socket(AF_INET,

SOCK_DGRAM,

IPPROTO_UDP);

if (server_socket < 0) {

perror("socket");

return -1;

}

ret = bind(server_socket,

(struct sockaddr*)&server_addr,

sizeof(server_addr));

if (ret < 0) {

perror("bind");

(void)close(server_socket);

return -2;

}

printf("UDP listining\n");

memset(fds, 0, sizeof(fds));

fds[0].fd = server_socket;

fds[0].events = POLLIN;

while (1) {

ret = poll(fds, 1, 1000);

if (ret < 0) {
perror("poll");
	break; 
}

if (fds[0].revents & POLLIN) {
  recv_send(buffer, 
  &client_addr);
}

}

(void)close(server_socket);

return 0;

}

编译:

gcc udpClient.c -o udpClient -g

gcc udpServer.c -o udpServer -g

运行:

客户端执行命令:

./udpClient 1234 127.0.0.1

服务器执行命令:

./udpServer 1234

运行后两者即可通讯。

从上面的代码可以看出来,上述UDP客户端和服务器都使用了poll函数监视文件描述符是否可读。区别在于服务器调用了bind函数将套接字与地址绑定,而客户端没有调用bind函数而已。

相关推荐
啥都想学的又啥都不会的研究生18 分钟前
Redis设计与实现-服务器中的数据库
运维·服务器·数据库·redis·笔记·缓存·性能优化
djykkkkkk30 分钟前
ubuntu 和 RV1126 交叉编译Mosqutiio-1.6.9
linux·运维·ubuntu
Vacant Seat1 小时前
图论-实现Trie(前缀树)
java·开发语言·数据结构·图论
大小科圣1 小时前
Tomcat介绍及部署
运维·服务器
LiDAR点云1 小时前
Matlab中快速查找元素索引号
数据结构·算法·matlab
好多知识都想学1 小时前
第二章Linux 命令概述
linux·运维·服务器
wo3258661451 小时前
浪潮英政服务器CS5420H2配置阵列时报错The reguested command has inualid arguments.解决方法
运维·服务器
熊峰峰2 小时前
Linux第0节:Linux环境的搭建
linux·运维·服务器
JKHaaa2 小时前
数据结构之线性表
数据结构
鸭梨山大。2 小时前
linux命令-iptables与firewalld 命令详解
linux·运维·网络