客户端 tcp
#include <myhand.h>
#define SERIP "192.168.114.118"
#define SERPORT 8888
#define CLIIP "192.168.246.135"
#define CLIPORT 7777
int main(int argc, const char *argv[])
{
//1.建立套接字
int cfd =socket(AF_INET,SOCK_STREAM,0);
if(cfd ==-1)
{
perror("socket error");
return -1;
}
printf("socket success cfd=%d\n",cfd);
//设置端口快速重用
int reuse =1;
if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
{
perror("setsockopt error");
return -1;
}
printf("设置端口快速重用成功_%d_%s_%s\n",__LINE__,__FILE__,__func__);
//2.绑定
//定义地址结构体
struct sockaddr_in cin;
cin.sin_family =AF_INET;
cin.sin_port =htons(CLIPORT);
cin.sin_addr.s_addr =inet_addr(CLIIP);
//bind
if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
{
perror("bind error");
return -1;
}
//3.连接服务器
//填充服务器地址结构体
struct sockaddr_in sin;
sin.sin_family =AF_INET;
sin.sin_port =htons(SERPORT);
sin.sin_addr.s_addr =inet_addr(SERIP);
//connect
if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
{
perror("connect error ");
return -1;
}
printf("connect success \n");
//4.收发数据(send recv read write)
char buf[128]="";
char rbuf[128]="";
定义一个用于检测文件描述符的结合
fd_set readfds,tempfds; //在栈区定义的
清空容器
FD_ZERO(&readfds);
将要检测的文件描述符放进去
FD_SET(cfd,&readfds);
FD_SET(0,&readfds);
//定义一个res接收select的返回值
int res=0;
while(1)
{
tempfds =readfds; //将集合内容复制一份
使用select阻塞等待集合的文件描述符有事件产生
res =select(cfd+1,&tempfds,NULL,NULL,NULL);
if(res==-1)
{
perror("select error");
return -1;
}
else if(res==0) //判断是否超时
{
printf("time out\n");
return -1 ;
}
判断cfd是否还在集合中
if(FD_ISSET(cfd,&tempfds))
{
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1];
//将消息发给服务端
send(cfd,buf,sizeof(buf),0);
//如果输入quit则退出
if(strcmp(buf,"quit")==0)
{
break;
}
}
if(FD_ISSET(0,&tempfds))
{
//接收服务端发送过来的消息
int res=recv(cfd,rbuf,sizeof(rbuf),0);
if(res ==0)
{
printf("服务器已经关闭\n");
break;
}
printf("rbuf=%s\n",rbuf);
}
}
//5.关闭套接字
close(cfd);
return 0;
}
服务器端
#include <myhand.h>
#define PORT 8888
#define IP "192.168.246.135"
int main(int argc, const char *argv[])
{
//创建套接字
int sfd =socket(AF_INET,SOCK_STREAM,0);
if(sfd==-1)
{
perror("socket error");
return -1;
}
printf("socket success sfd=%d\n",sfd);
//设置端口快速重用
int reuse =1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
{
perror("setsockopt error");
return -1;
}
printf("设置端口快速重用成功_%d_%s_%s\n",__LINE__,__FILE__,__func__);
//绑定IP地址和端口
//填充绑定信息地址结构体
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 _%d_%s_%s_\n",__LINE__,__FILE__,__func__);
//阻塞客户端连接请求,如果有新的客户端连接,则创建一个新的用于通信的套接字
//定义客户端地址信息结构体
struct sockaddr_in cin;
cin.sin_family =AF_INET;
socklen_t socklen=sizeof(cin); //客户端地址信息大小
定义一个用于检测文件描述符的集合
fd_set readfds,tempfds; //在栈区定义
清空容器内容
FD_ZERO(&readfds);
将要检测的文件描述符放入集合中
FD_SET(sfd,&readfds); //将sfd文件描述符放入
FD_SET(0,&readfds); //将0号文件描述符放入
//定义一个容器
char buf[128]="";
int res =0; //接收select的返回值
int newfd =-1; //存放用于最新连接客户端的套接字
int maxfd =sfd; //定义控制select函数中的最大文件描述符
struct sockaddr_in saveCin[1024]; //用于存放客户端地址信息结构体
while(1)
{
将集合内容复制一份
tempfds =readfds;
使用select阻塞等待集合中的文件描述符有事件产生
res =select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(res ==-1)
{
perror("select error");
return -1;
}
else if(res==0)
{
printf("time out\n");
return -1;
}
//遍历所以集合中文件描述符
for(int i=0; i<=maxfd ;i++ )
{
//判断当前i是否在集合中,如果不在,直接判断下一个
if(!FD_ISSET(i,&tempfds))
{
continue;
}
判断sfd是否在集合中
if(i==sfd)
{
//阻塞接收客户端的连接请求,并获取客户端地址信息
newfd=accept(sfd,(struct sockaddr*)&cin,&socklen);
if(newfd==-1)
{
perror("accept error");
return -1;
}
printf("accept success _%d_%s_%s\n",__LINE__,__FILE__,__func__);
将newfd放入readfds中
FD_SET(newfd ,&readfds);
//更新maxfd
if(newfd>maxfd)
{
maxfd=newfd;
}
//将最新的客户端套接字放入数组的下标为new的位置
saveCin[newfd] =cin;
printf("newfd=%d\n",newfd);
}
else if(i==0) //判断是否是终端输入
{
char buf1[128]="";
bzero(buf,sizeof(buf));
//从终端获取数据
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]='\0';
printf("触发终端输入事件:%s\n",buf);
sprintf(buf1,"%s%s","系统消息:",buf);
//将数据发送给所有用客户端
for(int j=4;j<=maxfd;j++)
{
send(j,buf1,sizeof(buf1),0);
}
}
else
{
//收发数据使用newfd完成通信
char buf[128]="";
//清空字符串
bzero(buf,sizeof(buf));
int ret =recv(i,buf,sizeof(buf),0);
if(ret==0)
{
printf("客户端已经下线\n");
将当前的文件描述符从集合中删除
FD_CLR(i,&readfds);
更新maxfd
for(int j=maxfd;j>=0;j--)
{
//判断当前的j是否在集合中,如果在,则为maxfd
if(FD_ISSET(j,&readfds))
{
maxfd=j;
break;
}
}
continue; //继续判断下一个
}
else if(ret<0)
{
perror("recv error");
return -1;
}
printf("[%s:%d]:%s\n",inet_ntoa(saveCin[i].sin_addr),ntohs(saveCin[i].sin_port),buf);
//将读取的信息,加上一下字符发送回去
strcat(buf,"\(*_*)/");
send(i,buf,sizeof(buf),0);
}
}
}
close(sfd);
return 0;
}