recvmsg demo 非阻塞式
时间: 2024-10-08 20:05:23 浏览: 34
`recvmsg`函数是Unix/Linux系统编程中的一个标准库函数,用于从套接字接收消息。非阻塞模式下,它允许程序在数据未就绪时立即返回,而不是阻塞等待。下面是一个简单的`recvmsg`在非阻塞模式下的C语言示例:
```c
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
int main() {
struct sockaddr_un addr;
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, "/tmp/test.sock", sizeof(addr.sun_path) - 1);
int sock = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
if (sock == -1) {
perror("socket");
return 1;
}
// 将套接字设置为非阻塞
int flags = fcntl(sock, F_GETFL, 0);
if (flags == -1) {
perror("fcntl");
close(sock);
return 1;
}
fcntl(sock, F_SETFL, flags | O_NONBLOCK); // 添加O_NONBLOCK标志
if (connect(sock, (struct sockaddr*)&addr, sizeof(addr)) == -1 && errno != EINPROGRESS) {
perror("connect");
close(sock);
return 1;
}
struct msghdr msg;
memset(&msg, 0, sizeof(msg));
msg.msg_name = &addr;
msg.msg_namelen = sizeof(addr);
char buf[1024];
msg.msg_control = NULL;
msg.msg_controllen = 0;
ssize_t bytes_received = recvmsg(sock, &msg, MSG_DONTWAIT); // MSG_DONTWAIT用于非阻塞模式
if (bytes_received > 0) {
printf("Received %zd bytes: ", bytes_received);
print_data(buf, bytes_received);
} else if (bytes_received == 0) { // 客户端关闭连接
printf("Connection closed\n");
} else if (errno != EAGAIN || errno != EWOULDBLOCK) { // 如果不是预期的错误
perror("recvmsg");
close(sock);
return 1;
}
close(sock);
return 0;
}
// 打印数据函数
void print_data(char* data, size_t len) {
for (size_t i = 0; i < len; ++i) {
printf("%c", data[i]);
}
}
阅读全文