bool choose()
{
print();
bool res = true;
printf("please choose (1 or 2)\n");
int flag =0;
scanf("%d",&flag);
getchar();//清掉回车
switch(flag)
{
case 1:res=registe();break;
case 2:res=login();break;
default:printf("no choose\n"),res=false;break;
}
return res;
}
int main(int argc, char**argv){
int server_specified=0;
check_args(argc, argv, &server_specified); /*checks if arguments in command line are correct*/
int port, fd, ret;
struct sockaddr_in addr;
/*struct sockaddr_in addr_server;*/
struct hostent *h;
struct in_addr *a;
char *buffer=malloc(128*sizeof(char));
char *buff=malloc(128*sizeof(char));
fd_set rfds;
int counter;
int nread;
char buf[15];
socklen_t addrlen;
char *name=malloc(35*sizeof(char));
char *surname=malloc(35*sizeof(char));
char*ip=malloc(30*sizeof(char));
int scport;
user*root=NULL; /*pointer to binary tree to store users*/
if(server_specified==1){/*use ip and port specified by user*/
a=(struct in_addr*)argv[8];
port=atoi(argv[10]);
}else{
/*get host IP*/
if((h=gethostbyname("tejo.tecnico.ulisboa.pt"))==NULL)exit(1);//error
a=(struct in_addr*)h->h_addr_list[0];
port=58000;
}
/*register in server*/
fd=socket(AF_INET, SOCK_DGRAM, 0); /*UDP socket*/
if(fd==-1) exit(-1);/*error*/
memset((void*)&addr, (int)'\0', sizeof(addr));
addr.sin_family=AF_INET;
addr.sin_addr=*a;
addr.sin_port=htons(port);
registe(&buffer, argv, fd, addr, "register"); /*register surname in server*/
close(fd);
/* ******************************************** */
/*listen to clients*/
fd=socket(AF_INET, SOCK_DGRAM, 0); /*UDP socket*/
if(fd==-1) exit(-1);/*error*/
memset((void*)&addr, (int)'\0', sizeof(addr));
addr.sin_family=AF_INET;
addr.sin_addr.s_addr=htonl(INADDR_ANY);
addr.sin_port=htons(atoi(argv[6]));
ret=bind(fd, (struct sockaddr*)&addr, sizeof(addr));
if(ret==-1){
printf("Error in bind\n");
exit(1);//error
}
while(1){
FD_ZERO(&rfds);
FD_SET(fd,&rfds);
FD_SET(fileno(stdin), &rfds);
counter=select(fd + fileno(stdin) + 1,&rfds,(fd_set*)NULL,(fd_set*)NULL,(struct timeval *)NULL);
//-------------------------------------------------------------------------------------------------------------------AQUI
if(counter<=0){
printf("Error in select\n");
exit(1);//errror
}
if(FD_ISSET(fileno(stdin), &rfds)){ /* user interface */
if(fgets(buf, 15, stdin)){
if(strcmp(buf, "exit\n")==0){
close(fd);
fd=socket(AF_INET, SOCK_DGRAM, 0); /*UDP socket*/
if(fd==-1) exit(-1);/*error*/
memset((void*)&addr, (int)'\0', sizeof(addr));
addr.sin_family=AF_INET;
addr.sin_addr=*a;
addr.sin_port=htons(port);
registe(&buffer, argv, fd, addr, "exit"); /*tells the server that the surname wont be available any more*/
close(fd);
free(name);
free(surname);
free(ip);
PosFixed_delete_tree(&root);
exit(0);
}else{
if(strcmp(buf, "list\n")==0){
printf("name\t\tsurname\t\tip\t\tscport\n");
list(root, argv[2]);/*displays names database*/
}
}
}
}
if(FD_ISSET(fd,&rfds)){
addrlen=sizeof(addr);
nread=recvfrom(fd, buff,128,0,(struct sockaddr*)&addr, &addrlen); /*receives requests*/
if(nread==-1)exit(1);//error
if(nread<128) buff[nread]='\0';
write(1, "received: ",10);//stdout
write(1, buff, strlen(buff));
printf("\n");
validate_user_command(&buff, &name, &surname, &ip, &scport, argv[2], &root); /*checks for valid requests and generates answer*/
ret=sendto(fd, buff, 128,0,(struct sockaddr*)&addr, addrlen);/*sends generated answer*/
if(ret==-1)exit(1);
}
}
/* ********************************************************** */
}