利用C语言完成一对一的 Socket 客户端与服务器程序,解决 recv 和 send 函数的阻塞问题,可以让程序同时进行 收发通信。
时间: 2024-03-17 11:40:08 浏览: 131
好的,以下是一个简单的一对一 Socket 客户端和服务器程序,它能够解决 recv 和 send 函数的阻塞问题,让程序同时进行收发通信:
服务器端代码:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE 1024
void error_handling(char *message);
int main(int argc, char *argv[])
{
int serv_sock, clnt_sock;
struct sockaddr_in serv_addr, clnt_addr;
socklen_t clnt_addr_size;
char message[BUF_SIZE];
if(argc!=2) {
printf("Usage: %s <port>\n", argv[0]);
exit(1);
}
serv_sock=socket(PF_INET, SOCK_STREAM, 0);
if(serv_sock == -1)
error_handling("socket() error");
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family=AF_INET;
serv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
serv_addr.sin_port=htons(atoi(argv[1]));
if(bind(serv_sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr))==-1)
error_handling("bind() error");
if(listen(serv_sock, 5)==-1)
error_handling("listen() error");
clnt_addr_size=sizeof(clnt_addr);
clnt_sock=accept(serv_sock, (struct sockaddr*)&clnt_addr,&clnt_addr_size);
if(clnt_sock==-1)
error_handling("accept() error");
while(1)
{
memset(message, 0, BUF_SIZE);
/* 接收数据 */
if(recv(clnt_sock, message, BUF_SIZE, MSG_DONTWAIT) > 0)
printf("Client: %s", message);
/* 发送数据 */
fgets(message, BUF_SIZE, stdin);
if(send(clnt_sock, message, strlen(message), 0) == -1)
error_handling("send() error");
}
close(clnt_sock);
close(serv_sock);
return 0;
}
void error_handling(char *message)
{
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}
```
客户端代码:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE 1024
void error_handling(char *message);
int main(int argc, char *argv[])
{
int sock;
struct sockaddr_in serv_addr;
char message[BUF_SIZE];
if(argc!=3) {
printf("Usage: %s <IP> <port>\n", argv[0]);
exit(1);
}
sock=socket(PF_INET, SOCK_STREAM, 0);
if(sock == -1)
error_handling("socket() error");
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family=AF_INET;
serv_addr.sin_addr.s_addr=inet_addr(argv[1]);
serv_addr.sin_port=htons(atoi(argv[2]));
if(connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr))==-1)
error_handling("connect() error");
while(1)
{
memset(message, 0, BUF_SIZE);
/* 发送数据 */
fgets(message, BUF_SIZE, stdin);
if(send(sock, message, strlen(message), 0) == -1)
error_handling("send() error");
/* 接收数据 */
if(recv(sock, message, BUF_SIZE, MSG_DONTWAIT) > 0)
printf("Server: %s", message);
}
close(sock);
return 0;
}
void error_handling(char *message)
{
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}
```
在上述代码中,我们使用了 MSG_DONTWAIT 标志来设置 recv 和 send 函数为非阻塞模式,这样就可以让程序同时进行收发通信了。
阅读全文