服务器server.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>
int main(int argc, char const *argv[])
{
if (argc != 2)
{
printf("please input %s <port>\n", argv[0]);
return -1;
}
//1.创建套接字 socket
int sockfd, acceptfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0)
{
perror("socket err.");
return -1;
}
printf("socket ok %d\n", sockfd);
//填充ipv4的通信结构体
struct sockaddr_in serveraddr, caddr;
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(atoi(argv[1]));
//serveraddr.sin_addr.s_addr = inet_addr("0.0.0.0"); //0.0.0.0 自动获取主机ip
// serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
serveraddr.sin_addr.s_addr = INADDR_ANY;
socklen_t len = sizeof(caddr);
//&serveraddr -->struct sockaddr_in *
//2.绑定套接字 bind (绑定自己的ip和端口,便于别人找到自己)
if (bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
{
perror("bind err.");
return -1;
}
printf("bind ok.\n");
//3.listen 监听 将主动套接子变为被动等待
if (listen(sockfd, 5) < 0)
{
perror("listen err.");
return -1;
}
printf("listen ok.\n");
//功能:select 服务器响应多个客户端请求且进行通信
//1.创建表
fd_set readfds, tempfds;
FD_ZERO(&readfds); //清空表
//2.添加关心的文件描述符
FD_SET(0, &readfds);
FD_SET(sockfd, &readfds);
int maxfd = sockfd, ret;
char buf[128];
int recvbyte;
//3.循环调用select检测 ,select阻塞等待有事件产生返回
while (1)
{
tempfds = readfds;
ret = select(maxfd + 1, &tempfds, NULL, NULL, NULL);
if (ret < 0)
{
perror("select err.");
return -1;
}
for (int i = 0; i <= maxfd; i++)
{
if (FD_ISSET(i, &tempfds))
{
if (i == 0)
{
fgets(buf, sizeof(buf), stdin);
printf("key:%s\n", buf);
//将服务器段输入的数据作为通知的消息发送给所有链接的客户端
for(int j=4;j<=maxfd;j++)
{
if(FD_ISSET(j,&readfds))
{
send(j,buf,sizeof(buf),0);
}
}
}
else if (i == sockfd)
{
//4.阻塞等待客户端链接,链接成功返回一个通信文件描述符 accept
acceptfd = accept(sockfd, (struct sockaddr *)&caddr, &len);
if (acceptfd < 0)
{
perror("accept err.");
return -1;
}
printf("accept ok.\n");
//输出查看链接的客户端的ip和端口
printf("ip:%s ,port:%d\n",
inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port));
//需要将产生通信文件描述符添加到表中进行检测
FD_SET(acceptfd, &readfds);
if (maxfd < acceptfd)
maxfd = acceptfd;
}
else
{
//接收消息
recvbyte = recv(i, buf, sizeof(buf), 0);
if (recvbyte < 0)
{
perror("recv err.");
// return -1;
}
else if (recvbyte == 0)
{
printf("%d client exit.\n",i);
FD_CLR(i, &readfds);
close(i);
break;
}
else
{
printf("%d buf:%s\n",i, buf);
}
}
}
}
}
close(sockfd);
return 0;
}
客户端client.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>
int main(int argc, char const *argv[])
{
if (argc != 3)
{
printf("please input %s <ip> <port>\n", argv[0]);
return -1;
}
//1.创建套接字 socket
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0)
{
perror("socket err.");
return -1;
}
printf("socket ok %d\n", sockfd);
//填充ipv4的通信结构体 服务器端ip和端口
struct sockaddr_in serveraddr;
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(atoi(argv[2]));
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
//&serveraddr -->struct sockaddr_in *
//2.请求链接 connect
if (connect(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
{
perror("connect err.");
return -1;
}
printf("connect ok.\n");
//3.循环收发消息
char buf[128];
int recvbyte;
pid_t pid = fork();
if (pid < 0)
{
perror("fork err.");
return -1;
}
else if (pid == 0)
{
while (1)
{
fgets(buf, sizeof(buf), stdin);
if (buf[strlen(buf) - 1] == '\n')
buf[strlen(buf) - 1] = '\0';
//发送消息
send(sockfd, buf, sizeof(buf), 0);
}
}
else
{
while (1)
{
//接收消息
recvbyte = recv(sockfd, buf, sizeof(buf), 0);
if (recvbyte < 0)
{
perror("recv err.");
// return -1;
}
printf("buf:%s\n", buf);
}
}
close(sockfd);
return 0;
}