作业:用select完成tcp客户端
#include <head.h>
#define PORT 8888
#define IP "192.168.125.164"
int main(int argc, const char *argv[])
{
int sfd = -1;
if((sfd=socket(AF_INET, SOCK_STREAM, 0)) == -1)
{
perror("socket error\n");
return -1;
}
//重置端口
int reuse = -1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
{
perror("setsockopt error\n");
return -1;
}
//定义结构体绑定地址和端口号
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
if(connect(sfd, (struct sockaddr*)&sin, sizeof(sin)) == -1)
{
perror("connect error\n");
return -1;
}
printf("已连接服务器\n");
//定义检测文件描述符的集合
fd_set readfds;
//清空集合
FD_ZERO(&readfds);
//将sfd和0文件描述符放入检测集合中
FD_SET(sfd, &readfds);
FD_SET(0, &readfds);
fd_set tempfds; //定义一个临时集合
char buf[128] = "";
int s=0; //接收select的返回值
while(1)
{
tempfds = readfds;
//使用select检测集合中释放有事件产生
s = select(sfd+1, &tempfds, NULL, NULL, NULL);
if(s == -1)
{
perror("select error");
return -1;
}
else if(s==0)
{
printf("time out\n");
return -1;
}
if(FD_ISSET(0, &tempfds))
{
bzero(buf, sizeof(buf));
printf("请输入: ");
fgets(buf,sizeof(buf), stdin);
buf[strlen(buf) - 1] = 0;
send(sfd, buf,sizeof(buf), 0);
}
else
{
int res = recv(sfd, buf, sizeof(buf), 0);
if(res == 0)
{
printf("客户端已下线\n");
break;
}
printf("[%s:%d]: %s\n", IP, PORT, buf);
}
}
close(sfd);
return 0;
}
作业:用poll完成tcp服务器
#include <head.h>
#define PORT 8888
#define IP "192.168.125.164"
int main(int argc, const char *argv[])
{
int sfd = -1;
if((sfd=socket(AF_INET, SOCK_STREAM, 0))==-1)
{
perror("socket error\n");
return -1;
}
//设置端口快速重用
int reuse = -1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1)
{
perror("setsockopt error\n");
return -1;
}
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin))==-1)
{
perror("bind error");
return -1;
}
printf("bind success\n");
if(listen(sfd, 128) == -1)
{
perror("listen error\n");
return -1;
}
printf("listen success\n");
//定义接收客户端地址结构体
struct sockaddr_in cin;
socklen_t socklen = sizeof(cin);
int newfd=-1;
char buf[128]="";
int len=4;
//定义检测文件描述符的集合
struct pollfd fds[len];
//将文件描述符放入集合
fds[0].fd = 0;
fds[0].events=POLLIN; //表示等待读事件
//将文件描述符sfd放入集合
fds[3].fd=sfd;
fds[3].events=POLLIN; //表示客户端中读取数据
int res=0; //接收select的返回值
int maxfd=sfd; //存放集合中的最大文件描述符
char rbuf[128]=""; //读消息的容器
struct sockaddr_in arr_cin[1024];
while(1)
{
int res = poll(fds, len, -1);
if(res < 0)
{
perror("poll error");
return -1;
}
else if(res==0)
{
printf("time out\n");
return -1;
}
for(int index=0;index<=maxfd&&fds[index].fd!=-1;index++)
{
if(fds[sfd].revents==POLLIN&&index==sfd)
{
if((newfd=accept(sfd, (struct sockaddr*)&cin, &socklen)) == -1)
{
perror("accept error");
return -1;
}
arr_cin[newfd]=cin;
printf("[%s:%d] 连接成功, newfd = %d\n",inet_ntoa(arr_cin[newfd].sin_addr), ntohs(arr_cin[newfd].sin_port), newfd);
//将newfd放入待检测集合中
fds[newfd].fd=newfd;
fds[newfd].events=POLLIN;
len++;
//更新maxfd
if(newfd>maxfd)
{
maxfd = newfd;
}
}
else if(fds[0].revents==POLLIN&&index==0) //判断0号文件描述符是否出发了事件
{
bzero(rbuf, sizeof(rbuf));
fgets(rbuf, sizeof(rbuf), stdin);
rbuf[strlen(rbuf)-1]=0;
//将数据发送给所有客户端
for(int j=4;j<=maxfd&&fds[j].fd!=-1; j++)
{
send(j,rbuf, sizeof(rbuf), 0);
}
}
else if(fds[index].revents==POLLIN)
{
//判断是否为客户端发来消息,newfd触发的事件
//由于有多个客户端要处理,所以需要进行遍历
bzero(rbuf, sizeof(rbuf));
int res=recv(index, rbuf, sizeof(rbuf), 0);
if(res == 0)
{
printf("客户端下线\n");
close(index);
//移除newfd
fds[index].fd=-1;
}
printf("[%s:%d] : %s\n", inet_ntoa(arr_cin[index].sin_addr), ntohs(arr_cin[index].sin_port), rbuf);
}
}
}
close(sfd);
return 0;
}