服务器:
- #include
-
- //do-while只是为了不让花括号单独存在,并不循环
- #define ERR_MSG(msg) do{\
- fprintf(stderr,"%d:",__LINE__);\
- perror(msg);\
- }while(0);
-
- #define PORT 8888//端口号1024-49151
- #define IP "192.168.2.54"//本机IP,终端输入ifconfig可得
-
- int main(int argc, const char *argv[])
- {
- //创建流式套接字 socket
- int sfd = socket(AF_INET,SOCK_STREAM,0);
- if(sfd<0){
- //printf("%d:",__LINE__);
- //perror("socket error");
- ERR_MSG("socket");
- return -1;
- }
- printf("socket create success sfd=%d\n",sfd);
-
- //允许端口快速的被复用
- int reuse = 1;
- if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
- {
- ERR_MSG("setsockopt");
- return -1;
- }
- printf("允许端口快速的被复用成功\n");
-
- //填充地址信息结构体给bind函数绑定
- //真是的地址信息结构体根据地址族制定 AF_INET:man 7 ip
- struct sockaddr_in sin;
- sin.sin_family = AF_INET;//必须填AF_INET
- sin.sin_port = htons(PORT);//端口号的网络字节序
- sin.sin_addr.s_addr = inet_addr(IP);//本机IP
- //绑定服务器的地址信息 -->必须绑定 bind
- if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
- ERR_MSG("bind");
- return -1;
- }//指针类型的强转不会有数据丢失
- printf("bind success\n");
-
- //将套接字设置为被动监听状态 listen
- if(listen(sfd,128)<0){
- ERR_MSG("listen");
- return -1;
- }
- printf("listen success\n");
-
- //定义文件描述符集合
- fd_set readfds,tmpfds;
-
- //将文件描述符集合清空
- FD_ZERO(&readfds);
- //将文件描述符放入集合之中
- FD_SET(sfd,&readfds);
- FD_SET(0, &readfds);
-
- int maxfd = sfd;
- int res;
- char buf[128] = "";
- struct sockaddr_in cin; //存储客户端的地址信息
- socklen_t addrlen = sizeof(cin); //真实的地址信息结构体的大小
- //获取一个已经完成的客户端信息,生成一个新的文件描述符 accept
- int newfd;
- struct sockaddr_in saveCin[1024]; //备份连接成功的客户端的地址信息,用下标来对应地址信息
- while(1){
- tmpfds = readfds;
-
- //调用select函数
- res = select(maxfd+1,&tmpfds, NULL, NULL, NULL);
- if(res < 0){
- ERR_MSG("select error");
- break;
- }else if(0 == res){
- printf("time out\n");
- break;
- }else{
- for(int i=0; i<=maxfd; i++){
- if(0==FD_ISSET(i, &tmpfds)){
- continue;
- }else if(-1==FD_ISSET(i, &tmpfds)){
- ERR_MSG("FD_ISSET error");
- break;
- }
-
- if(0==i){
- //终端输入
- printf("从终端读取成功::");
- fgets(buf,sizeof(buf),stdin);
-
- buf[strlen(buf)-1] = 0;
-
- printf("%s\n",buf);
- }
- else if(sfd == i){
- if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen))<0){
- ERR_MSG("accept");
- return -1;
-
-
- }
- printf("[%s:%d]客户端链接成功 newfd = %d\n",\
- inet_ntoa(cin.sin_addr),\
- ntohs(cin.sin_port),newfd);
-
- saveCin[newfd] = cin; //将cin另存到下标为文件描述符的位置
-
- //将newfd添加到集合中
- FD_SET(newfd, &readfds);
-
- //更新maxfd
- maxfd = maxfd>newfd?maxfd:newfd;
-
-
- }else{
- if(FD_ISSET(i, &readfds)){
- //清空字符串
- bzero(buf,sizeof(buf)); //memset
-
- //接收
- res = recv(i, buf, sizeof(buf) ,0);
- if(res<0){
- ERR_MSG("recv");
- return -1;
- }else if(0==res){
- printf("[%s:%d]客户端下线 newfd=%d\n",\
- inet_ntoa(saveCin[i].sin_addr),\
- ntohs(saveCin[i].sin_port),i);
- //将文件描述符从集合中踢出
- FD_CLR(i, &readfds);
- //由于踢出的文件描述符可能是最大文件描述符,更新maxfd
- /* for(; maxfd>=0; maxfd--){
- if(FD_ISSET(maxfd, &readfds)){
- break;
- }
- } */
-
- while(FD_ISSET(maxfd, &readfds)==0 && maxfd-->=0);
-
- //关闭文件描述符
- close(i);
- continue;
- }
- printf("[%s:%d]客户端消息 newfd=%d:%s\n",\
- inet_ntoa(saveCin[i].sin_addr),\
- ntohs(saveCin[i].sin_port),i,buf);
-
-
- //发送
- strcat(buf,">_<");
- if(send(i,buf,strlen(buf),0)<0){
- ERR_MSG("send");
- return -1;
- }
- printf("send success\n");
- }
-
-
- }
- }
- /*
- //如果是终端输入就绪
- if(FD_ISSET(0,&tmpfds)){
- printf("从终端读取成功::");
- fgets(buf,sizeof(buf),stdin);
- buf[strlen(buf)-1] = 0;
- printf("%s\n",buf);
- }
- //如果是客户端连接就绪
- if(FD_ISSET(sfd,&tmpfds)){
- if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen))<0){
- ERR_MSG("accept");
- return -1;
- }
- printf("[%s:%d]客户端链接成功 newfd = %d\n",\
- inet_ntoa(cin.sin_addr),\
- ntohs(cin.sin_port),newfd);
- saveCin[newfd] = cin; //将cin另存到下标为文件描述符的位置
- //将newfd添加到集合中
- FD_SET(newfd, &readfds);
- //更新maxfd
- maxfd = maxfd>newfd?maxfd:newfd;
- }
- for(int i=4; i<=maxfd; i++){
- if(FD_ISSET(i, &readfds)){
- //清空字符串
- bzero(buf,sizeof(buf)); //memset
- //接收
- res = recv(i, buf, sizeof(buf) ,0);
- if(res<0){
- ERR_MSG("recv");
- return -1;
- }else if(0==res){
- printf("[%s:%d]客户端下线 newfd=%d\n",\
- inet_ntoa(saveCin[i].sin_addr),\
- ntohs(saveCin[i].sin_port),i);
- //将文件描述符从集合中踢出
- FD_CLR(i, &readfds);
- //由于踢出的文件描述符可能是最大文件描述符,更新maxfd
- for(; maxfd>=0; maxfd--){
- if(FD_ISSET(maxfd, &readfds)){
- break;
- }
- }
- while(FD_ISSET(maxfd, &readfds)==0 && maxfd-->=0);
- //关闭文件描述符
- close(i);
- continue;
- }
- printf("[%s:%d]客户端消息 newfd=%d:%s\n",\
- inet_ntoa(saveCin[i].sin_addr),\
- ntohs(saveCin[i].sin_port),i,buf);
- //发送
- strcat(buf,">_<");
- if(send(i,buf,strlen(buf),0)<0){
- ERR_MSG("send");
- return -1;
- }
- printf("send success\n");
- }
- }*/
- }
- }
-
-
- //关闭文件描述符
- close(newfd);
- if(close(sfd)<0){
- ERR_MSG("close");
- return -1;
- }
-
-
- return 0;
- }
服务器:
- #include
-
- //do-while只是为了不让花括号单独存在,并不循环
- #define ERR_MSG(msg) do{\
- fprintf(stderr,"%d:",__LINE__);\
- perror(msg);\
- }while(0);
-
- #define PORT 8888//端口号1024-49151
- #define IP "192.168.2.54"//本机IP,终端输入ifconfig可得
-
- int main(int argc, const char *argv[])
- {
- //创建流式套接字 socket
- int cfd = socket(AF_INET,SOCK_STREAM,0);
- if(cfd<0){
- //printf("%d:",__LINE__);
- //perror("socket error");
- ERR_MSG("socket");
- return -1;
- }
- printf("socket create success sfd=%d\n",cfd);
-
- //填充服务器的地址信息结构体给connect函数连接
- //相连接哪个服务器就输入哪个服务器的地址信息
- //真是的地址信息结构体根据地址族制定 AF_INET:man 7 ip
- struct sockaddr_in sin;
- sin.sin_family = AF_INET;//必须填AF_INET
- sin.sin_port = htons(PORT);//服务器绑定的端口号的网络字节序
- sin.sin_addr.s_addr = inet_addr(IP);//服务器绑定的IP
- socklen_t addrlen = sizeof(sin);
- //连接指定服务器 connect
- if(connect(cfd,(struct sockaddr*)&sin,addrlen)<0){
- ERR_MSG("connect");
- return -1;
- }
-
- char buf[128] = "";
- ssize_t res = 0;
-
- //定义文件描述符集合
- fd_set readfds,tmpfds;
- //初始化集合
- FD_ZERO(&readfds);
- //把cfd加入集合
- FD_SET(cfd, &readfds);
- //把标准输入加入集合
- FD_SET(0, &readfds);
- int ret = -1;
- while(1){
- tmpfds = readfds;
- ret = select(cfd+1, &tmpfds, NULL, NULL, NULL);
-
- if(ret<0){
- ERR_MSG("select error");
- break;
- }else if(0 == ret){
- printf("time out\n");
- break;
- }else{
- if(FD_ISSET(0, &tmpfds)){
- //说明有来自终端的输入,需要发送
- //清空字符串
- bzero(buf,sizeof(buf)); //memset
- printf("请输入>> ");
- fgets(buf,sizeof(buf),stdin);
- buf[strlen(buf)-1] = 0;
-
- //发送
- strcat(buf,">_<");
- if(send(cfd,buf,strlen(buf),0)<0){
- ERR_MSG("send");
- return -1;
- }
- printf("send success\n");
- }
-
- if(FD_ISSET(cfd, &tmpfds)){
- //套接字文件的接收缓冲区有数据,需要接收
- //说明有来自终端的输入,需要发送
- //清空字符串
- bzero(buf,sizeof(buf)); //memset
-
- //接收
- res = recv(cfd, buf, sizeof(buf) ,0);
- if(res<0){
- ERR_MSG("recv");
- return -1;
- }else if(0==res){
- printf("[%s:%d]服务器下线 cfd=%d\n",\
- inet_ntoa(sin.sin_addr),\
- ntohs(sin.sin_port),cfd);
- break;
- }
- printf("[%s:%d]服务器消息 cfd=%d:%s\n",\
- inet_ntoa(sin.sin_addr),\
- ntohs(sin.sin_port),cfd,buf);
- }
-
- }
-
-
- }
-
-
- /*
- while(1){
- //清空字符串
- bzero(buf,sizeof(buf)); //memset
- printf("请输入>> ");
- fgets(buf,sizeof(buf),stdin);
- buf[strlen(buf)-1] = 0;
- //发送
- strcat(buf,">_<");
- if(send(cfd,buf,strlen(buf),0)<0){
- ERR_MSG("send");
- return -1;
- }
- printf("send success\n");
- //接收
- res = recv(cfd, buf, sizeof(buf) ,0);
- if(res<0){
- ERR_MSG("recv");
- return -1;
- }else if(0==res){
- printf("[%s:%d]服务器下线 cfd=%d\n",\
- inet_ntoa(sin.sin_addr),\
- ntohs(sin.sin_port),cfd);
- break;
- }
- printf("[%s:%d]服务器消息 cfd=%d:%s\n",\
- inet_ntoa(sin.sin_addr),\
- ntohs(sin.sin_port),cfd,buf);
- }
- */
- //关闭文件描述符
- //close(newfd);
- if(close(cfd)<0){
- ERR_MSG("close");
- return -1;
- }
-
-
- return 0;
- }
效果:
