./server Port 1*/
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <time.h>
#define MAXBUF 1024
int main(int argc, char **argv)
{
int sockfd, new_fd, fd;
socklen_t len;
struct sockaddr_in my_addr, their_addr;
unsigned int myport, lisnum;
char buf[MAXBUF + 1],s1[]="wma said:/n",s2[]="Client said:/n";
fd_set rfds;
struct timeval tv;
int retval, maxfd = -1;
char *s=NULL;
time_t lt;
if((fd=open("./chat_server_log", O_WRONLY|O_CREAT|O_APPEND))<0){
perror("open log file error/n");
exit(1);
}
if (argv[1])
myport = atoi(argv[1]);
else
myport = 4000;
if (argv[2])
lisnum = atoi(argv[2]);
else
lisnum = 2;
if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
perror("socket");
exit(1);
}
bzero(&my_addr, sizeof(my_addr));
my_addr.sin_family = PF_INET;
my_addr.sin_port = htons(myport);
if (argv[3])
my_addr.sin_addr.s_addr = inet_addr(argv[3]);
else
my_addr.sin_addr.s_addr = INADDR_ANY;
if (bind(sockfd, (struct sockaddr *) &my_addr, sizeof(struct sockaddr))
== -1) {
perror("bind");
exit(1);
}
if (listen(sockfd, lisnum) == -1) {
perror("listen");
exit(1);
}
while (1) {
printf
("/n^_^ Chat with client ^_^/n");
len = sizeof(struct sockaddr);
if ((new_fd = accept(sockfd, (struct sockaddr *) &their_addr, //返回新的socket,用來和客戶端通信。
&len)) == -1) {
perror("accept");
exit(errno);
} else
printf("server: got connection from %s, port %d, socket %d/n",
inet_ntoa(their_addr.sin_addr),
ntohs(their_addr.sin_port), new_fd);
/* 開始處理每個新連接上的數據收發 */
while (1) {
/* 把集合清空 */
FD_ZERO(&rfds);
/* 把標準輸入句柄0加入到集合中 */
FD_SET(0, &rfds);
maxfd = 0;
/* 把當前連接句柄new_fd加入到集合中 */
FD_SET(new_fd, &rfds);
if (new_fd > maxfd)
maxfd = new_fd;
/* 設置最大等待時間 */
tv.tv_sec = 1;
tv.tv_usec = 0;
/* 開始等待 */
retval = select(maxfd + 1, &rfds, NULL, NULL, &tv);
if (retval == -1) {
printf("select error! %s", strerror(errno));
break;
} else if (retval == 0) {
/* printf
("沒有任何消息到來,用戶也沒有按鍵,繼續等待……/n"); */
continue;
} else {
if (FD_ISSET(0, &rfds)) {
/* 用戶按鍵了,則讀取用戶輸入的內容發送出去 */
bzero(buf, MAXBUF + 1);
fgets(buf, MAXBUF, stdin);
if (!strncasecmp(buf, "quit", 4)) {
printf("I want to quit!/n");
break;
}
len = send(new_fd, buf, strlen(buf) - 1, 0);
lt =time(NULL);
s=ctime(<);
write(fd,s,strlen(s));
write(fd,s1,strlen(s1));
write(fd,buf,strlen(buf));
write(fd,"/n",1);
if (len > 0)
printf("/n");
else {
printf
("Send fail!error code %d,error:'%s'/n",
buf, errno, strerror(errno));
break;
}
}
if (FD_ISSET(new_fd, &rfds)) {
/* 當前連接的socket上有消息到來則接收對方發過來的消息並顯示 */
bzero(buf, MAXBUF + 1);
/* 接收客戶端的消息 */
len = recv(new_fd, buf, MAXBUF, 0);
lt =time(NULL);
s=ctime(<);
write(fd,s,strlen(s));
write(fd,s2,strlen(s2));
write(fd,buf,strlen(buf));
write(fd,"/n",1);
if (len > 0){
printf("Client said: %s/n",buf);
}
else {
if (len < 0)
printf
("Recv fail!error code %d,error:'%s'/n",
errno, strerror(errno));
else
printf("Peer quit,stop chatting/n");
break;
}
}
}
}
close(new_fd);
/* 處理每個新連接上的數據收發結束 */
printf("Chat with others?(no->quit)");
fflush(stdout);
bzero(buf, MAXBUF + 1);
fgets(buf, MAXBUF, stdin);
if (!strncasecmp(buf, "no", 2)) {
printf("Stop chatting!/n");
break;
}
}
close(fd);
close(sockfd);
return 0;
}
=================================
/*client.c
./client IP Port*/
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <sys/socket.h>
#include <resolv.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/types.h>
#include <netdb.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <time.h>
#define MAXBUF 1024
int main(int argc, char **argv)
{
int sockfd, len, fd;
struct sockaddr_in dest;
char buffer[MAXBUF + 1], s1[]="wma said:/n",s2[]="I said:/n";
fd_set rfds;
struct timeval tv;
int retval, maxfd = -1;
struct hostent *host;
char *s=NULL;
time_t lt;
if((fd=open("./chat_client_log", O_WRONLY|O_CREAT|O_APPEND))<0){
perror("open log file error/n");
exit(1);
}
if (argc != 3) {
printf
("Para usage error!Like this:/n/t/t%s servername port/n/tSuch as:/t%s 10.67.7.106 4000/n",
argv[0], argv[0]);
exit(0);
}
printf("You will chat with wma, enter your message:/n");
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
perror("Socket");
exit(errno);
}
/* 初始化服務器端(對方)的地址和端口信息 */
bzero(&dest, sizeof(dest));
dest.sin_family = AF_INET;
dest.sin_port = htons(atoi(argv[2]));
/* if((host=gethostbyname(argv[1]))==NULL) {
perror("gethostbyname error!");
exit(1);
}
dest.sin_addr=*((struct in_addr *)host->h_addr); */
if (inet_aton(argv[1], (struct in_addr *) &dest.sin_addr.s_addr) == 0) {
perror(argv[1]);
exit(errno);
}
/* 連接服務器 */
if (connect(sockfd, (struct sockaddr *) &dest, sizeof(dest)) != 0) {
perror("Connect ");
exit(errno);
}
printf
("/n");
while (1) {
/* 把集合清空 */
FD_ZERO(&rfds);
/* 把標準輸入句柄0加入到集合中 */
FD_SET(0, &rfds);
maxfd = 0;
/* 把當前連接句柄sockfd加入到集合中 */
FD_SET(sockfd, &rfds);
if (sockfd > maxfd)
maxfd = sockfd;
/* 設置最大等待時間 */
tv.tv_sec = 1;
tv.tv_usec = 0;
/* 開始等待 */
retval = select(maxfd + 1, &rfds, NULL, NULL, &tv);
if (retval == -1) {
printf("select error! %s", strerror(errno));
break;
} else if (retval == 0) {
/* printf
("沒有任何消息到來,用戶也沒有按鍵,繼續等待……/n"); */
continue;
} else {
if (FD_ISSET(sockfd, &rfds)) {
/* 連接的socket上有消息到來則接收對方發過來的消息並顯示 */
bzero(buffer, MAXBUF + 1);
/* 接收對方發過來的消息,最多接收 MAXBUF 個字節 */
len = recv(sockfd, buffer, MAXBUF, 0);
lt =time(NULL);
s=ctime(<);
write(fd,s,strlen(s));
write(fd,s1,strlen(s1));
write(fd,buffer,strlen(buffer));
write(fd,"/n",1);
if (len > 0)
printf("wma said: %s/n",buffer);
else {
if (len < 0)
printf
("Recv fail!error code %d,error:'%s'/n",
errno, strerror(errno));
else
printf("Peer quit,stop chatting/n");
break;
}
}
if (FD_ISSET(0, &rfds)) {
/* 用戶按鍵了,則讀取用戶輸入的內容發送出去 */
bzero(buffer, MAXBUF + 1);
fgets(buffer, MAXBUF, stdin);
if (!strncasecmp(buffer, "quit", 4)) {
printf("I want to quit!/n");
break;
}
/* 發消息給服務器 */
len = send(sockfd, buffer, strlen(buffer) - 1, 0);
lt =time(NULL);
s=ctime(<);
write(fd,s,strlen(s));
write(fd,s2,strlen(s2));
write(fd,buffer,strlen(buffer));
write(fd,"/n",1);
if (len < 0) {
printf
("Send fail!error code %d,error:'%s'/n",
buffer, errno, strerror(errno));
break;
} else
printf
("/n");
}
}
}
/* 關閉連接 */
close(sockfd);
return 0;
}
**************************************
int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval *timeout);
先說明兩個結構體:
第一,struct fd_set可以理解爲一個集合,這個集合中存放的是文件描述符(file descriptor),即文件句柄,這可以是我們所說的普通意義的文件,當然Unix下任何設備、管道、FIFO等都是文件形式,全部包括在內,所以毫 無疑問一個socket就是一個文件,socket句柄就是一個文件描述符。fd_set集合可以通過一些宏由人爲來操作,比如清空集合FD_ZERO (fd_set *),將一個給定的文件描述符加入集合之中FD_SET(int ,fd_set *),將一個給定的文件描述符從集合中刪除FD_CLR(int ,fd_set*),檢查集合中指定的文件描述符是否可以讀寫FD_ISSET(int ,fd_set* )。一會兒舉例說明。
第二,struct timeval是一個大家常用的結構,用來代表時間值,有兩個成員,一個是秒數,另一個是毫秒數。
具體解釋select的參數:
int maxfdp是一個整數值,是指集合中所有文件描述符的範圍,即所有文件描述符的最大值加1,不能錯!在Windows中這個參數的值無所謂,可以設置不正確。
fd_set *readfds是指向fd_set結構的指針,這個集合中應該包括文件描述符,我們是要監視這些文件描述符的讀變化的,即我們關心是否可以從這些文件中 讀取數據了,如果這個集合中有一個文件可讀,select就會返回一個大於0的值,表示有文件可讀,如果沒有可讀的文件,則根據timeout參數再判斷 是否超時,若超出timeout的時間,select返回0,若發生錯誤返回負值。可以傳入NULL值,表示不關心任何文件的讀變化。
fd_set *writefds是指向fd_set結構的指針,這個集合中應該包括文件描述符,我們是要監視這些文件描述符的寫變化的,即我們關心是否可以向這些文件 中寫入數據了,如果這個集合中有一個文件可寫,select就會返回一個大於0的值,表示有文件可寫,如果沒有可寫的文件,則根據timeout參數再判 斷是否超時,若超出timeout的時間,select返回0,若發生錯誤返回負值。可以傳入NULL值,表示不關心任何文件的寫變化。
fd_set *errorfds同上面兩個參數的意圖,用來監視文件錯誤異常。
struct timeval* timeout是select的超時時間,這個參數至關重要,它可以使select處於三種狀態,第一,若將NULL以形參傳入,即不傳入時間結構,就是 將select置於阻塞狀態,一定等到監視文件描述符集合中某個文件描述符發生變化爲止;第二,若將時間值設爲0秒0毫秒,就變成一個純粹的非阻塞函數, 不管文件描述符是否有變化,都立刻返回繼續執行,文件無變化返回0,有變化返回一個正值;第三,timeout的值大於0,這就是等待的超時時間,即 select在timeout時間內阻塞,超時時間之內有事件到來就返回了,否則在超時後不管怎樣一定返回,返回值同上述。
返回值:
負值:select錯誤 正值:某些文件可讀寫或出錯 0:等待超時,沒有可讀寫或錯誤的文件
在有了select後可以寫出像樣的網絡程序來!舉個簡單的例子,就是從網絡上接受數據寫入一個文件中。
例子:
main()
{
int sock;
FILE *fp;
struct fd_set fds;
struct timeval timeout={3,0}; //select等待3秒,3秒輪詢,要非阻塞就置0
char buffer[256]={0}; //256字節的接收緩衝區
/* 假定已經建立UDP連接,具體過程不寫,簡單,當然TCP也同理,主機ip和port都已經給定,要寫的文件已經打開
sock=socket(...);
bind(...);
fp=fopen(...); */
while(1)
{
FD_ZERO(&fds); //每次循環都要清空集合,否則不能檢測描述符變化
FD_SET(sock,&fds); //添加描述符
FD_SET(fp,&fds); //同上
maxfdp=sock>fp?sock+1:fp+1; //描述符最大值加1
switch(select(maxfdp,&fds,&fds,NULL,&timeout)) //select使用
{
case -1: exit(-1);break; //select錯誤,退出程序
case 0:break; //再次輪詢
default:
if(FD_ISSET(sock,&fds)) //測試sock是否可讀,即是否網絡上有數據
{
recvfrom(sock,buffer,256,.....);//接受網絡數據
if(FD_ISSET(fp,&fds)) //測試文件是否可寫
fwrite(fp,buffer...);//寫入文件
buffer清空;
}// end if break;
}// end switch
}//end while
}//end main