linux+多路复用socket,linux socket多路复用服务器

/************client**********/

#include

#include

#include

#include

#include

#include

#include

#include

#include

#define PORT 1234

#define SERVER_IP "127.0.0.1"

main()

{

int s;

struct sockaddr_in addr;

char buffer[256];

static int flag = 1;

if((s = socket(AF_INET,SOCK_STREAM,0))<0)

{

perror("socket");

exit(1);

}

/*

填写sockaddr_in结构*/

bzero(&addr,sizeof(addr));

addr.sin_family = AF_INET;

addr.sin_port=htons(PORT);

addr.sin_addr.s_addr = inet_addr(SERVER_IP);

/* 尝试连线*/

if(connect(s,&addr,sizeof(addr))<0)

{

perror("connect");

exit(1);

}

/* 接收由server端传来的信息*/

recv(s,buffer,sizeof(buffer),0);

printf("%s\n",buffer);

while(1)

{

bzero(buffer,sizeof(buffer));

/* 从标准输入设备取得字符串*/

if (flag == 1)

read(STDIN_FILENO,buffer,sizeof(buffer));

printf("Insert data:%s\n", buffer);

flag = 0;}

else

{

read(s,buffer,sizeof(buffer));

printf("From sever: %d\n\n", buffer[0]);

}

/*

将字符串传给server端*/

sleep(2);

buffer[0] += 1;

if(write(s,buffer,sizeof(buffer))<0)

{

perror("send");

exit(1);

}

printf("To sever: %d\n", buffer[0]);

}

}

/***************sever*************/

#include

#include

#include

#include

#include

#include

#include

#include

#include

#define PORT 1234

#define MAXSOCKFD 10

void main()

{

int sockfd,newsockfd,is_connected[MAXSOCKFD],fd;

struct sockaddr_in addr;

int addr_len = sizeof(struct sockaddr_in);

fd_set readfds;

char buffer[256];

char msg[ ] ="Welcome to server!";

if ((sockfd = socket(AF_INET,SOCK_STREAM,0))<0)

{

perror("socket");

exit(1);

}

bzero(&addr,sizeof(addr));

addr.sin_family =AF_INET;

addr.sin_port = htons(PORT);

addr.sin_addr.s_addr = htonl(INADDR_ANY);

//addr.sin_addr.s_addr = inet_addr(SERVER_IP);

if(bind(sockfd,&addr,sizeof(addr))<0)

{

perror("connect");

exit(1);

}

if(listen(sockfd,3)<0)

{

perror("listen");

exit(1);

}

for(fd=0;fd

is_connected[fd]=0;

while(1)

{

FD_ZERO(&readfds);

FD_SET(sockfd,&readfds);

for(fd=0;fd

{

if(is_connected[fd])

FD_SET(fd,&readfds);

}

/*

阻塞，如果文件描述符没有改变

*/

if(!select(MAXSOCKFD,&readfds,NULL,NULL,NULL))

{

continue;

}

for(fd=0;fd

{

if(FD_ISSET(fd,&readfds))

{

printf("sockfd = %d, fd = %d\n", sockfd, fd);

/*

两个描述符相等时表示有新的客户端连接到达

*/if(sockfd ==fd)

{

if((newsockfd = accept (sockfd,&addr,&addr_len))<0)

perror("accept");

write(newsockfd,msg,sizeof(msg));

printf("newsockfd = %d\n", newsockfd);

is_connected[newsockfd] =1;

printf("cnnect from %s\n",inet_ntoa(addr.sin_addr));

}

else

{

bzero(buffer,sizeof(buffer));

if(read(fd,buffer,sizeof(buffer))<=0)

{

printf("connect closed.\n");

is_connected[fd]=0;

close(fd);

}

else

{

printf("From client: %d\n",buffer[0]);

if (write(fd, buffer, strlen(buffer)) < 0)

perror("write");printf("To client: %d\n\n",buffer[0]);

}

}}

}

}

}