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, )) == -){
        perror("socket");
        exit(errno);
    }else
        printf("socket create success!\n");
    memset(&s_addr,,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))) == -){
        perror("bind");
        exit(errno);
    }else
        printf("bind success!\n");
    /*侦听本地端口*/
    if(listen(sockfd,listnum) == -){
        perror("listen");
        exit(errno);
    }else
        printf("the server is listening!\n");
    while(){
        printf("*****************聊天开始***************\n");
        len = sizeof(struct sockaddr);
        if((newfd = accept(sockfd,(struct sockaddr*) &c_addr, &len)) == -){
            perror("accept");
            exit(errno);
        }else
            printf("正在与您聊天的客户端是：%s: %d\n",inet_ntoa(c_addr.sin_addr),ntohs(c_addr.sin_port));
        while(){
            FD_ZERO(&rfds);
            FD_SET(, &rfds);
            maxfd = ;
            FD_SET(newfd, &rfds);
            /*找出文件描述符集合中最大的文件描述符*/
            if(maxfd < newfd)
                maxfd = newfd;
            /*设置超时时间*/
            tv.tv_sec = ;
            tv.tv_usec = ;
            /*等待聊天*/
            retval = select(maxfd+, &rfds, NULL, NULL, &tv);
            if(retval == -){
                printf("select出错，与该客户端连接的程序将退出\n");
                break;
            }else if(retval == ){
                printf("waiting...\n");
                continue;
            }else{
                /*用户输入信息了*/
                if(FD_ISSET(, &rfds)){ 

                    /******发送消息*******/
                    memset(buf,,sizeof(buf));
                    /*fgets函数：从流中读取BUFLEN-1个字符*/
                    fgets(buf,BUFLEN,stdin);
                    /*打印发送的消息*/
                    //fputs(buf,stdout);
                    if(!strncasecmp(buf,"quit",)){
                        printf("server 请求终止聊天!\n");
                        break;
                    }
                        len = send(newfd,buf,strlen(buf),);
                    if(len > )
                        printf("\t消息发送成功：%s\n",buf);
                    else{
                        printf("消息发送失败!\n");
                        break;
                    }
                }
                /*客户端发来了消息*/
                if(FD_ISSET(newfd, &rfds)){
                    /******接收消息*******/
                    memset(buf,,sizeof(buf));
                    /*fgets函数：从流中读取BUFLEN-1个字符*/
                    len = recv(newfd,buf,BUFLEN,);
                    if(len > )
                        printf("客户端发来的信息是：%s\n",buf);
                    else{
                        if(len <  )
                            printf("接受消息失败！\n");
                        else
                            printf("客户端退出了，聊天终止！\n");
                        break;
                    }
                }
            }
        }
        /*关闭聊天的套接字*/
        close(newfd);
        /*是否退出服务器*/
        printf("服务器是否退出程序：y->是；n->否? ");
        bzero(buf, BUFLEN);
        fgets(buf,BUFLEN, stdin);
        if(!strncasecmp(buf,"y",)){
            printf("server 退出!\n");
            break;
        }
    }
    /*关闭服务器的套接字*/
    close(sockfd);
    return ;
}

客户端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, )) == -){
        perror("socket");
        exit(errno);
    }else
        printf("socket create success!\n");

    /*设置服务器ip*/
    memset(&s_addr,,sizeof(s_addr));
    s_addr.sin_family = AF_INET;
     s_addr.sin_port = htons(PORT);
    if (inet_aton(argv[], (struct in_addr *)&s_addr.sin_addr.s_addr) == ) {
        perror(argv[]);
        exit(errno);
    }
    /*开始连接服务器*/
    if(connect(sockfd,(struct sockaddr*)&s_addr,sizeof(struct sockaddr)) == -){
        perror("connect");
        exit(errno);
    }else
        printf("conncet success!\n");

    while(){
        FD_ZERO(&rfds);
        FD_SET(, &rfds);
        maxfd = ;
        FD_SET(sockfd, &rfds);
        if(maxfd < sockfd)
            maxfd = sockfd;
        tv.tv_sec = ;
        tv.tv_usec = ;
        retval = select(maxfd+, &rfds, NULL, NULL, &tv);
        if(retval == -){
            printf("select出错，客户端程序退出\n");
            break;
        }else if(retval == ){
            printf("waiting...\n");
            continue;
        }else{
            /*服务器发来了消息*/
            if(FD_ISSET(sockfd,&rfds)){
                /******接收消息*******/
                bzero(buf,BUFLEN);
                len = recv(sockfd,buf,BUFLEN,);
                if(len > )
                    printf("服务器发来的消息是：%s\n",buf);
                else{
                    if(len <  )
                        printf("接受消息失败！\n");
                    else
                        printf("服务器退出了，聊天终止！\n");
                break;
                }
            }
            /*用户输入信息了,开始处理信息并发送*/
            if(FD_ISSET(, &rfds)){
                /******发送消息*******/
                bzero(buf,BUFLEN);
                fgets(buf,BUFLEN,stdin);

                if(!strncasecmp(buf,"quit",)){
                    printf("client 请求终止聊天!\n");
                    break;
                }
                    len = send(sockfd,buf,strlen(buf),);
                if(len > )
                    printf("\t消息发送成功：%s\n",buf);
                else{
                    printf("消息发送失败!\n");
                    break;
                }
            }
        }

    }
    /*关闭连接*/
    close(sockfd);

    return ;
}

在程序的服务端输入

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