socket實例C語言：一個簡單的聊天程序

我們老師讓寫一個簡單的聊天軟件，並且實現不同機子之間的通信，我用的是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 = 6; 
            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 = 6;
        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;
}

 

在程序的服務端輸入

lwp@lwp-linux:~/Desktop/1234$ ./server 

在程序的客戶端輸入

lwp@lwp-linux:~/Desktop/1234$ ./client 192.168.100.80

當然這都是我在我的機子上截的圖，客戶端我們可以放到任何一台機子上，只要保證都能在同一個局域網下，連到我的IP地址就行。開啟服務端后，在開啟客戶端，我們就可以聊天了：

server端：

waiting...
waiting...
waiting...
nihal
    消息發送成功：nihal

客戶端發來的信息是：zhuzhu
hehe
    消息發送成功：hehe

客戶端發來的信息是：haha 
客戶端退出了，聊天終止！
服務器是否退出程序：y->是；n->否? y
server端退出。

client端：

waiting...
waiting...
waiting...
服務器發來的消息是：nihal
zhuzhu
消息發送成功：zhuzhu
服務器發來的消息是：hehe
haha
消息發送成功：haha