12.22

2

服务器

#include "myhead.h"
#define SERPORT 9999
#define SERIP "192.168.125.253"
#define CLIPORT 6666
#define CLIIP "192.168.125.253"

int main(int argc, const char *argv[])
{
	int cfd=-1;
	if((cfd=socket(AF_INET,SOCK_DGRAM,0))==-1){
		perror("socket error");
		return -1;
	}
	printf("cfd=%d\n",cfd);
	int reuse=1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1){
		perror("setsockopt error");
		return -1;	
	}
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;
	cin.sin_port=htons(CLIPORT);
	cin.sin_addr.s_addr=inet_addr(CLIIP);
/*	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1){
		perror("bind error");
		return -1;
	}	
*/	struct sockaddr_in sin;
	socklen_t socklen =sizeof(sin);
	sin.sin_family=AF_INET;
	sin.sin_port=htons(SERPORT);
	sin.sin_addr.s_addr=inet_addr(SERIP);
	
	struct pollfd fds[2];
	fds[0].fd=0;
	fds[0].events=POLLIN;
	fds[1].fd=cfd;
	fds[1].events=POLLIN;
	int res=-1;
	char wbuf[128]="";
			char type;
			char name[20];
			char text[100];
	
	while(1){
		res=poll(fds,2,-1);
		if(res<0){
			perror("pollerror");
			return -1;
		}else if(res==0){
			printf("time out\n");
			return -1;
		}
		if(fds[0].revents==POLLIN){
			printf("please enter type(L/C/Q)\n");
			scanf("%c",&type);
			getchar();
			if(type=='L'||type=='Q'){
			printf("please enter name :\n");
			scanf("%s",name);
			getchar();
			sprintf(wbuf,"%c%s\n",type,name);
			sendto(cfd,wbuf,sizeof(wbuf),0,(struct sockaddr*)&sin,sizeof(sin));
			}
			if(type=='C'){
			printf("please enter name :\n");
			scanf("%s",name);
			getchar();
			printf("please enter text:\n");
			scanf("%s",text);
			sprintf(wbuf,"%c%s:%s\n",type,name,text);
			printf("wbuf=%s",wbuf);
			sendto(cfd,wbuf,sizeof(wbuf),0,(struct sockaddr*)&sin,sizeof(sin));	
			}
		}
		if(fds[1].revents==POLLIN){
			bzero(wbuf,sizeof(wbuf));
			int res=recvfrom(cfd,wbuf,sizeof(wbuf),0,NULL,NULL);
			if(res==0){
				printf("客户端已经下线\n");
				break;
			}
			printf("[%s:%d]:%s\n",SERIP,SERPORT,wbuf);

		}
	}
	close(cfd);
	return 0;
}

#include "myhead.h"
#define PORT 9999
#define IP "192.168.125.253"
#define MAXSIZE 1024
typedef struct Node{
	struct sockaddr_in cin;
	struct Node *next;
}*Linklist;
Linklist create_node(){
	Linklist s=(Linklist)malloc(sizeof(struct Node));
	if(NULL==s){
		return NULL;
	}
	s->cin.sin_family=AF_INET;
	s->cin.sin_port=htons(PORT);
	s->cin.sin_addr.s_addr=inet_addr(IP);
	s->next=NULL;
	return s;
}
Linklist insert_rear(struct sockaddr_in cin1,Linklist head){
	Linklist s=create_node();
	if(NULL==s){
		return head;
	}
	s->cin=cin1;
	if(NULL==head){
		head=s;
	}else{
		Linklist p=head;
		while(p->next!=NULL){
			p=p->next;
		}
		p->next=s;
	}
	return head;
}
Linklist delete_head(Linklist head){
	if(NULL==head){
		return head;
	}else{
		Linklist del=head;
		head=head->next;
		free(del);
		del=NULL;
	}
	return head;
}
Linklist delete_pos(int pos,Linklist head){
	if(NULL==head||pos<1){
		return head;
	}
	if(head->next==NULL||pos==1){
		head=delete_head(head);
	}else{
		Linklist p=head;
		for(int i=1;i<pos-1;i++){
			p=p->next;
		}
		Linklist q=p->next;
		p->next=q->next;
		free(q);
		q=NULL;
	}
	return head;
}
int main(int argc, const char *argv[])
{
	Linklist head=NULL;
	int sfd=-1;
	if((sfd=socket(AF_INET,SOCK_DGRAM,0))==-1){
		perror("socket error");
		return -1;
	}
	printf("sfd=%d\n",sfd);
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1){
		perror("setsockopt error");
				int num=0;
		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 _%d_%s_%s\n",__LINE__,__FILE__,__func__);
	struct pollfd fds[1024];
	for(int i=0;i<1024;i++){
		fds[i].fd=-1;
	}
	fds[0].fd=0;
	fds[0].events=POLLIN;
	fds[1].fd=sfd;
	fds[1].events=POLLIN;
	struct sockaddr_in cin1;
	socklen_t socklen=sizeof(cin1);
	int newfd=-1;
	char buf[128]="";
	int res=-1;
				int num=0;
	char rbuf[128]="";
	int count=0;
	while(1){
		res=poll(fds,MAXSIZE,-1);
		if(res<0){
			perror("poll error");
			return -1;
		}else if(res==0){
			printf("time out\n");
			return -1;
		}
		if(fds[0].revents==POLLIN){
			bzero(buf,sizeof(buf));
			scanf("%s",buf);
			printf("触发终端输入:%s\n",buf);
			Linklist z=head;
			while(z){
			sendto(sfd,buf,sizeof(buf),0,(struct sockaddr*)&z->cin,sizeof(z->cin));
			z=z->next;
			}
		}
		if(fds[1].revents==POLLIN){
			bzero(buf,sizeof(buf));
			int res=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&cin1,&socklen);
			buf[strlen(buf)-1]=0;
		//	printf("buf=%s\n",buf);

			if(buf[0]=='L'){
				bzero(rbuf,sizeof(rbuf));	
				head=insert_rear(cin1,head);		
				sprintf(rbuf,"有新用户%s连接\n",buf+1);
				printf("rbuf=%s\n",rbuf);
				Linklist p=head;
				while(p){
				sendto(sfd,rbuf,sizeof(rbuf),0,(struct sockaddr*)&p->cin,sizeof(p->cin));
				p=p->next;
				}
			}
			if(buf[0]=='C'){
				int num=0;
				bzero(rbuf,sizeof(rbuf));
				strncpy(rbuf,buf+1,sizeof(buf)-1);
				Linklist q=head;
				while(q){
				sendto(sfd,rbuf,sizeof(rbuf),0,(struct sockaddr*)&q->cin,sizeof(q->cin));
				q=q->next;
				}
			}
		if(buf[0]=='Q'){
				bzero(rbuf,sizeof(rbuf));
				Linklist k=head;
				int pos=1;
				while(k){
				if(cin1.sin_port==k->cin.sin_port&&\
						cin1.sin_addr.s_addr==k->cin.sin_addr.s_addr){
					head=delete_pos(pos,head);
					printf("pos=%d\n",pos);
				int num=0;
					}else{
						k=k->next;
						pos++;
					}
				}
				sprintf(rbuf,"用户%s下线\n",buf+1);
				printf("%s\n",rbuf);
				Linklist w=head;
				while(w){
					sendto(sfd,rbuf,sizeof(rbuf),0,(struct sockaddr*)&w->cin,sizeof(w->cin));
				   w=w->next;	
				}
			}
		}
	}
	
	return 0;
}