linux网络编程练习---聊天室（TCP/IP实现）

为了更深刻的锻炼认识TCP/IP协议，加强自己对Linux系统的网络编程部分的编写代码能力，编写基于控制台的聊天窗口，用本机既当服务器又当客户端，先打开一个shell，运行服务器程序，然后再打开一个shell窗口，运行客户端程序，显示连接成功，开始聊天吧。不知道为什么上传效果图不成功，把图放在相册了。

/*客户端*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#define BUFFER_SIZE 256
#define PORT 4568
char recv_buff[BUFFER_SIZE];
char send_buff[BUFFER_SIZE];
int main(void)
{
    char write_flag=0;
    char read_flag=1;
    int sockfd;// 套结字
    struct sockaddr_in cli_addr;//服务器的IPv4的套结字结构
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))<0)//创建套结字
    {
        perror("socket");
        exit(1);
    }
    else
    {
        printf("creat client socket success id:%d\n",sockfd);
    }
    bzero(&cli_addr,sizeof(struct sockaddr_in));
    cli_addr.sin_family=AF_INET;
    cli_addr.sin_port=htons(PORT);
    cli_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    if(connect(sockfd,(struct sockaddr*)(&cli_addr),sizeof(struct sockaddr))<0)
    {
        perror("connect");
        exit(2);
    }
    else
    {
        printf("connect server success.\n");
    }
    memset(send_buff,0x00,sizeof(send_buff));
    memset(recv_buff,0x00,sizeof(send_buff));
   while(1)
    {
        if(strlen(send_buff)==0&&write_flag==1)//判断发送缓冲区是否为空
        {
            gets(send_buff);
            if(write(sockfd,send_buff,strlen(send_buff))<0)
            {
                perror("write");
                exit(3);
            }
            else
            {
                printf("Client:\n");
                printf("       %s\n",send_buff);
            }
            read_flag=1;
            write_flag=0;
            memset(recv_buff,0x00,strlen(recv_buff));
        }
       else if(strlen(recv_buff)==0&&read_flag==1)
        {

            if(read(sockfd,recv_buff,sizeof(recv_buff))<0)
            {
                perror("read");
                exit(4);
            }
            else if(strlen(recv_buff)>0)
            {
                printf("server:\n");
                printf("      : %s\n",recv_buff);
                memset(send_buff,0x00,strlen(send_buff));//接受到数据后清空发送缓冲区
                write_flag=1;
                read_flag=0;
            }
        }

    }
    printf("end.\n");
    close(sockfd);
    return 0;
}

/*服务器*/
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#define BUFFER_SIZE 256
#define PORT 4568
char recv_buff[BUFFER_SIZE];
char send_buff[BUFFER_SIZE];
int main(void)
{
    char write_flag=0;
    char read_flag=1;
    int sockfd,newsockfd;// 套结字
    socklen_t addr_len;
    struct sockaddr_in ser_addr;//服务器的IPv4的套结字结构
    addr_len=sizeof(struct sockaddr_in);
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))<0)//创建套结字
    {
        perror("socket");
        exit(1);
    }
    else
    {
        printf("creat socket success id:%d\n",sockfd);
    }
    bzero(&ser_addr,sizeof(struct sockaddr_in));
    ser_addr.sin_family=AF_INET;
    ser_addr.sin_port=htons(PORT);
    ser_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    if(bind(sockfd,(struct sockaddr *)(&ser_addr),sizeof(struct sockaddr))<0)//绑定
    {
        perror("bind");
        exit(2);
    }
    else
    {
        printf("bind success.\n");
    }
    if(listen(sockfd,10)<0)//最多允许10个客户端进程请求,监听
    {
        perror("listen");
        exit(3);
    }
    else
    {
        printf("listening.....\n");
    }
    memset(send_buff,0x00,sizeof(send_buff));
    memset(recv_buff,0x00,sizeof(send_buff));
    if((newsockfd=accept(sockfd,(struct sockaddr*)(&ser_addr),&addr_len))<0)
    {
        //accept函数使用错误注意复习
        perror("accept");
        exit(4);
    }
    else
    {
        printf("connect success.\n");
    }
    write(newsockfd,"have connect!",strlen("have connect"));
    while(1)
    {
       if(strlen(send_buff)==0&&write_flag==1)//判断发送缓冲区是否为空
        {

            gets(send_buff);
            if(write(newsockfd,send_buff,strlen(send_buff))<0)
            {
                perror("write");
                exit(3);
            }
            else
            {
                printf("Client:\n");
                printf("       %s\n",send_buff);
            }
            read_flag=1;
            write_flag=0;
            memset(recv_buff,0x00,strlen(recv_buff));
        }
       else if(strlen(recv_buff)==0&&read_flag==1)
        {

            if(read(newsockfd,recv_buff,sizeof(recv_buff))<0)
            {
                perror("read");
                exit(4);
            }
            else if(strlen(recv_buff)>0)
            {
                printf("Server:\n");
                printf("      : %s\n",recv_buff);
                memset(send_buff,0x00,strlen(send_buff));//接受到数据后清空发送缓冲区
                write_flag=1;
                read_flag=0;
            }
        }

    }
    close(sockfd);
    return 0;
}

效果图：