C语言socket客户端ipv6非阻塞模式收发
时间: 2023-07-25 10:20:01 浏览: 258
以下是一个简单的 C 语言 IPv6 Socket 客户端示例代码,使用非阻塞模式收发数据:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netdb.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <errno.h>
#define PORT "8080" // 连接的端口号
#define MAXDATASIZE 100 // 接收缓冲区的大小
int main(int argc, char *argv[]) {
int sockfd, numbytes, flags;
struct addrinfo hints, *servinfo, *p;
int rv;
char s[INET6_ADDRSTRLEN];
char message[100]; // 发送的消息
char buf[MAXDATASIZE]; // 接收缓冲区
fd_set readfds, writefds;
struct timeval tv;
int retval;
// 如果没有指定服务器的 IP 地址
if (argc != 2) {
fprintf(stderr,"usage: client hostname\n");
exit(1);
}
memset(&hints, 0, sizeof hints);
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
if ((rv = getaddrinfo(argv[1], PORT, &hints, &servinfo)) != 0) {
fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(rv));
return 1;
}
// 遍历所有结果,直到成功连接为止
for (p = servinfo; p != NULL; p = p->ai_next) {
if ((sockfd = socket(p->ai_family, p->ai_socktype, p->ai_protocol)) == -1) {
perror("client: socket");
continue;
}
if (connect(sockfd, p->ai_addr, p->ai_addrlen) == -1) {
close(sockfd);
perror("client: connect");
continue;
}
break;
}
if (p == NULL) {
fprintf(stderr, "client: failed to connect\n");
return 2;
}
inet_ntop(p->ai_family, get_in_addr((struct sockaddr *)p->ai_addr), s, sizeof s);
printf("client: connecting to %s\n", s);
freeaddrinfo(servinfo); // 全部都用这个结构体
// 设置套接字为非阻塞模式
flags = fcntl(sockfd, F_GETFL, 0);
fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
// 发送消息
printf("Enter a message to send: ");
fgets(message, 100, stdin);
// 开始循环等待读写事件
while(1) {
FD_ZERO(&readfds);
FD_ZERO(&writefds);
FD_SET(sockfd, &readfds);
FD_SET(sockfd, &writefds);
tv.tv_sec = 1; // 超时时间为 1 秒
tv.tv_usec = 0;
retval = select(sockfd+1, &readfds, &writefds, NULL, &tv);
if (retval == -1) {
perror("select");
exit(1);
} else if (retval == 0) {
printf("Timeout occurred! No data after 1 seconds.\n");
continue;
} else {
// 有读写事件发生
if (FD_ISSET(sockfd, &writefds)) {
int len = strlen(message);
int bytes_sent = send(sockfd, message, len, 0);
if (bytes_sent == -1) {
perror("send");
exit(1);
}
printf("Sent %d bytes.\n", bytes_sent);
}
if (FD_ISSET(sockfd, &readfds)) {
numbytes = recv(sockfd, buf, MAXDATASIZE-1, 0);
if (numbytes == -1) {
if (errno != EAGAIN && errno != EWOULDBLOCK) {
perror("recv");
exit(1);
}
} else if (numbytes == 0) {
printf("Connection closed by server.\n");
break;
} else {
buf[numbytes] = '\0';
printf("Received: %s\n", buf);
}
}
}
}
close(sockfd);
return 0;
}
```
需要注意的是,在使用非阻塞模式时,我们需要使用 `select` 函数来循环等待读写事件的发生。在上述代码中,我们设置超时时间为 1 秒,即如果在 1 秒内没有读写事件发生,就继续等待。
在 `select` 中,我们使用 `FD_SET` 函数将套接字加入到读写事件的集合中,并使用 `FD_ISSET` 函数来检查套接字是否就绪。如果套接字可写,我们就调用 `send` 函数发送消息;如果套接字可读,我们就调用 `recv` 函数接收数据。需要注意的是,`recv` 函数在非阻塞模式下可能会返回 -1,此时我们需要检查 `errno` 是否为 `EAGAIN` 或 `EWOULDBLOCK`,如果是,则说明当前没有数据可读,需要继续等待。如果 `recv` 函数返回 0,则说明服务器关闭了连接,此时需要退出循环。
希望这份代码能够对你有所帮助!
阅读全文