- #include <stdlib.h>
- #include <stdio.h>
- #include <errno.h>
- #include <string.h>
- #include <netdb.h>
- #include <sys/types.h>
- #include <netinet/in.h>
- #include <sys/socket.h>
- #include<unistd.h>
- #include<arpa/inet.h>
- #include<ctype.h>
- /* 宏定义端口号 */
- #define portnumber 8000
- #define MAX_LINE 80
- int main(void)
- {
- int lfd;
- int cfd;
- int sfd;
- int rdy;
- struct sockaddr_in sin;
- struct sockaddr_in cin;
- int client[FD_SETSIZE]; /* 客户端连接的套接字描述符数组 */
- int maxi;
- int maxfd; /* 最大连接数 */
- fd_set rset;
- fd_set allset;
- socklen_t addr_len; /* 地址结构长度 */
- char buffer[MAX_LINE];
- int i;
- int n;
- int len;
- int opt = 1; /* 套接字选项 */
- char addr_p[20];
- /* 对server_addr_in 结构进行赋值 */
- bzero(&sin,sizeof(struct sockaddr_in)); /* 先清零 */
- sin.sin_family=AF_INET; //
- sin.sin_addr.s_addr=htonl(INADDR_ANY); //表示接受任何ip地址 将ip地址转换成网络字节序
- sin.sin_port=htons(portnumber); //将端口号转换成网络字节序
- /* 调用socket函数创建一个TCP协议套接口 */
- if((lfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:IPV4;SOCK_STREAM:TCP
- {
- fprintf(stderr,"Socket error:%s\n\a",strerror(errno));
- exit(1);
- }
- printf("socket!\n");
- /*设置套接字选项 使用默认选项*/
- setsockopt(lfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
- /* 调用bind函数 将serer_addr结构绑定到sockfd上 */
- if(bind(lfd,(struct sockaddr *)(&sin),sizeof(struct sockaddr))==-1)
- {
- fprintf(stderr,"Bind error:%s\n\a",strerror(errno));
- exit(1);
- }
- printf("bind!\n");
- /* 开始监听端口 等待客户的请求 */
- if(listen(lfd,20)==-1)
- {
- fprintf(stderr,"Listen error:%s\n\a",strerror(errno));
- exit(1);
- }
- printf("listen!\n");
- printf("Accepting connections .......\n");
- maxfd = lfd; /*对最大文件描述符进行初始化*/
- maxi = -1;
- /*初始化客户端连接描述符集合*/
- for(i = 0;i < FD_SETSIZE;i++)
- {
- client[i] = -1;
- }
- FD_ZERO(&allset); /* 清空文件描述符集合 */
- FD_SET(lfd,&allset); /* 将监听字设置在集合内 */
- /* 开始服务程序的死循环 */
- while(1)
- {
- rset = allset;
- printf("selecting!\n");
- /*得到当前可以读的文件描述符数*/
- rdy = select(maxfd + 1, &rset, NULL, NULL, NULL);
- printf("selected!\n");
- if(FD_ISSET(lfd, &rset))
- {
- addr_len = sizeof(sin);
- printf("accepting!\n");
- /* 接受客户端的请求 */
- if((cfd=accept(lfd,(struct sockaddr *)(&cin),&addr_len))==-1)
- {
- fprintf(stderr,"Accept error:%s\n\a",strerror(errno));
- exit(1);
- }
- printf("accepted!\n");
- /*查找一个空闲位置*/
- for(i = 0; i<FD_SETSIZE; i++)
- { //printf("%d\t",client[i]);
- if(client[i] <= 0)
- {
- client[i] = cfd; /* 将处理该客户端的连接套接字设置到该位置 */
- break;
- }
- }
- /* 太多的客户端连接 服务器拒绝俄请求 跳出循环 */
- if(i == FD_SETSIZE)
- {
- printf("too many clients");
- exit(1);
- }
- FD_SET(cfd, &allset); /* 设置连接集合 */
- if(cfd > maxfd) /* 新的连接描述符 */
- {
- maxfd = cfd;
- }
- if(i > maxi)
- {
- maxi = i;
- }
- if(--rdy <= 0) /* 减少一个连接描述符 */
- {
- continue;
- }
- }
- /* 对每一个连接描述符做处理 */
- for(i = 0;i< FD_SETSIZE;i++)
- {
- if((sfd = client[i]) < 0)
- {
- continue;
- }
- if(FD_ISSET(sfd, &rset))
- {
- printf("reading!\n");
- n = read(sfd,buffer,MAX_LINE);
- printf("%s\n",buffer);
- printf("read!\n");
- if(n == 0)
- {
- printf("the other side has been closed. \n");
- fflush(stdout); /* 刷新 输出终端 */
- close(sfd);
- FD_CLR(sfd, &allset); /*清空连接描述符数组*/
- client[i] = -1;
- }
- else
- {
- /* 将客户端地址转换成字符串 */
- inet_ntop(AF_INET, &cin.sin_addr, addr_p, sizeof(addr_p));
- addr_p[strlen(addr_p)] = '\0';
- /*打印客户端地址 和 端口号*/
- printf("Client Ip is %s, port is %d\n",addr_p,ntohs(cin.sin_port));
- /* 谐函数出错 */
- if(n == 1)
- {
- exit(1);
- }
- }
- /*如果没有可以读的套接字 退出循环*/
- if(--rdy <= 0)
- {
- break;
- }
- }
- }
- close(lfd); /* 关闭链接套接字 */
- return 0;
- }
客户端
- #include <stdlib.h>
- #include <stdio.h>
- #include <errno.h>
- #include <string.h>
- #include <netdb.h>
- #include <sys/types.h>
- #include <netinet/in.h>
- #include <sys/socket.h>
- #define portnumber 8000
- int main(int argc, charchar *argv[])
- {
- int nbytes;
- int sockfd;
- char buffer[80];
- char buffer_2[80];
- struct sockaddr_in server_addr;
- struct hostent *host;
- if(argc!=2)
- {
- fprintf(stderr,"Usage:%s hostname \a\n",argv[0]);
- exit(1);
- }
- if((host=gethostbyname(argv[1]))==NULL)
- {
- fprintf(stderr,"Gethostname error\n");
- exit(1);
- }
- /* 调用socket函数创建一个TCP协议套接口 */
- if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:Internet;SOCK_STREAM:TCP
- {
- fprintf(stderr,"Socket Error:%s\a\n",strerror(errno));
- exit(1);
- }
- bzero(&server_addr,sizeof(server_addr)); //
- server_addr.sin_family=AF_INET;
- server_addr.sin_port=htons(portnumber);
- server_addr.sin_addr = *((struct in_addr *)host->h_addr);//?
- if(connect(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)
- {
- fprintf(stderr,"Connect Error:%s\a\n",strerror(errno));
- exit(1);
- }
- while(1){
- printf("Please input char:\n");
- fgets(buffer,1024,stdin);
- write(sockfd,buffer,strlen(buffer));
- #if 1
- if((nbytes=read(sockfd,buffer_2,81))==-1)
- {
- fprintf(stderr,"Read Error:%s\n",strerror(errno));
- exit(1);
- }
- buffer_2[nbytes]='\0';
- printf("Client received from Server %s\n",buffer_2);
- #endif
- }
- close(sockfd);
- exit(0);
- }