1. 程式人生 > 其它 >Socket 聊天小程式

Socket 聊天小程式

伺服器端程式碼 server.c

#include <stdio.h> 
#include <stdlib.h> 
#include <string.h> 
#include <errno.h> 
#include <sys/socket.h> 
#include <arpa/inet.h> 
#include <netinet/in.h> 
#include <sys/types.h> 
#include <unistd.h> 
#include <sys/time.h> 
  
#define
BUFLEN 1024 #define PORT 6666 #define LISTNUM 20 int main() { int sockfd, newfd; struct sockaddr_in s_addr, c_addr; char buf[BUFLEN]; socklen_t len; unsigned int port, listnum; fd_set rfds; struct timeval tv; int retval,maxfd; /*建立socket*/ if
((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1){ perror("socket"); exit(errno); }else printf("socket create success!\n"); memset(&s_addr,0,sizeof(s_addr)); s_addr.sin_family = AF_INET; s_addr.sin_port = htons(PORT); s_addr.sin_addr.s_addr = htons(INADDR_ANY);
/*把地址和埠幫定到套接字上*/ if((bind(sockfd, (struct sockaddr*) &s_addr,sizeof(struct sockaddr))) == -1){ perror("bind"); exit(errno); }else printf("bind success!\n"); /*偵聽本地埠*/ if(listen(sockfd,listnum) == -1){ perror("listen"); exit(errno); }else printf("the server is listening!\n"); while(1){ printf("*****************聊天開始***************\n"); len = sizeof(struct sockaddr); if((newfd = accept(sockfd,(struct sockaddr*) &c_addr, &len)) == -1){ perror("accept"); exit(errno); }else printf("正在與您聊天的客戶端是:%s: %d\n",inet_ntoa(c_addr.sin_addr),ntohs(c_addr.sin_port)); while(1){ FD_ZERO(&rfds); FD_SET(0, &rfds); maxfd = 0; FD_SET(newfd, &rfds); /*找出檔案描述符集合中最大的檔案描述符*/ if(maxfd < newfd) maxfd = newfd; /*設定超時時間*/ tv.tv_sec = 30; tv.tv_usec = 0; /*等待聊天*/ retval = select(maxfd+1, &rfds, NULL, NULL, &tv); if(retval == -1){ printf("select出錯,與該客戶端連線的程式將退出\n"); break; }else if(retval == 0){ printf("waiting...\n"); continue; }else{ /*使用者輸入資訊了*/ if(FD_ISSET(0, &rfds)){ /******傳送訊息*******/ memset(buf,0,sizeof(buf)); /*fgets函式:從流中讀取BUFLEN-1個字元*/ fgets(buf,BUFLEN,stdin); /*打印發送的訊息*/ //fputs(buf,stdout); if(!strncasecmp(buf,"quit",4)){ printf("server 請求終止聊天!\n"); break; } len = send(newfd,buf,strlen(buf),0); if(len > 0) printf("\t訊息傳送成功:%s\n",buf); else{ printf("訊息傳送失敗!\n"); break; } } /*客戶端發來了訊息*/ if(FD_ISSET(newfd, &rfds)){ /******接收訊息*******/ memset(buf,0,sizeof(buf)); /*fgets函式:從流中讀取BUFLEN-1個字元*/ len = recv(newfd,buf,BUFLEN,0); if(len > 0) printf("客戶端發來的資訊是:%s\n",buf); else{ if(len < 0 ) printf("接受訊息失敗!\n"); else printf("客戶端退出了,聊天終止!\n"); break; } } } } /*關閉聊天的套接字*/ close(newfd); /*是否退出伺服器*/ printf("伺服器是否退出程式:y->是;n->否? "); bzero(buf, BUFLEN); fgets(buf,BUFLEN, stdin); if(!strncasecmp(buf,"y",1)){ printf("server 退出!\n"); break; } } /*關閉伺服器的套接字*/ close(sockfd); return 0; }

客戶端程式碼 client.c :

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/time.h>

#define BUFLEN 1024
#define PORT 6666

int main(int argc, char **argv)
{
    int sockfd;
    struct sockaddr_in s_addr;
    socklen_t len;
    unsigned int port;
    char buf[BUFLEN];
    fd_set rfds;
    struct timeval tv;
    int retval, maxfd; 
    
    /*建立socket*/
    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
        perror("socket");
        exit(errno);
    }else
        printf("socket create success!\n");

    
    /*設定伺服器ip*/
    memset(&s_addr,0,sizeof(s_addr));
    s_addr.sin_family = AF_INET;
     s_addr.sin_port = htons(PORT);
    if (inet_aton(argv[1], (struct in_addr *)&s_addr.sin_addr.s_addr) == 0) {
        perror(argv[1]);
        exit(errno);
    }
    /*開始連線伺服器*/ 
    if(connect(sockfd,(struct sockaddr*)&s_addr,sizeof(struct sockaddr)) == -1){
        perror("connect");
        exit(errno);
    }else
        printf("conncet success!\n");
    
    while(1){
        FD_ZERO(&rfds);
        FD_SET(0, &rfds);
        maxfd = 0;
        FD_SET(sockfd, &rfds);
        if(maxfd < sockfd)
            maxfd = sockfd;
        tv.tv_sec = 30;
        tv.tv_usec = 0;
        retval = select(maxfd+1, &rfds, NULL, NULL, &tv);
        if(retval == -1){
            printf("select出錯,客戶端程式退出\n");
            break;
        }else if(retval == 0){
            printf("waiting...\n");
            continue;
        }else{
            /*伺服器發來了訊息*/
            if(FD_ISSET(sockfd,&rfds)){
                /******接收訊息*******/
                bzero(buf,BUFLEN);
                len = recv(sockfd,buf,BUFLEN,0);
                if(len > 0)
                    printf("伺服器發來的訊息是:%s\n",buf);
                else{
                    if(len < 0 )
                        printf("接受訊息失敗!\n");
                    else
                        printf("伺服器退出了,聊天終止!\n");
                break; 
                }
            }
            /*使用者輸入資訊了,開始處理資訊併發送*/
            if(FD_ISSET(0, &rfds)){ 
                /******傳送訊息*******/ 
                bzero(buf,BUFLEN);
                fgets(buf,BUFLEN,stdin);
               
                if(!strncasecmp(buf,"quit",4)){
                    printf("client 請求終止聊天!\n");
                    break;
                }
                    len = send(sockfd,buf,strlen(buf),0);
                if(len > 0)
                    printf("\t訊息傳送成功:%s\n",buf); 
                else{
                    printf("訊息傳送失敗!\n");
                    break; 
                } 
            }
        }
    
    }
    /*關閉連線*/
    close(sockfd);

    return 0;
}

伺服器端截圖:

客戶端截圖: