基于TCP的多人聊天工具

客户端（两个线程）
1、一个用于发消息
2、一个用于接消息

服务端：（多线程）
主线程用于接受连接请求，并依次分配给子线程。
子线程：1、接受用户登录信息并足保存给文件
2、接受用户聊天信息并转发给所有其他用户
细节：send函数发送失败处理。

不足：第一个连接服务器的会出现putty
用户下线时没有提示，send和receive消息不及时

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <pthread.h>
#include <signal.h>
void* socket_handle(void* arg);
void send_msg(int send_conn,char* name,char* msg,int flag);
void save_msg(int conn);
pthread_mutex_t m;
pthread_mutex_t m1;
int main()
{
    FILE* fp=fopen("msg.txt","w");//每次服务器开机自动清空用户
    fclose(fp);

    signal(SIGPIPE,SIG_IGN);//send发送，而对方不在线时

    pthread_mutex_init (&m,NULL);
    pthread_mutex_init (&m1,NULL);

//买手机
    int socket_listen=socket(AF_INET,SOCK_STREAM,0);

    int opt_val=1;//设置端口复用
    setsockopt(socket_listen,SOL_SOCKET,SO_REUSEADDR,&opt_val,sizeof(opt_val));
//绑卡
    struct sockaddr_in myaddr;
    myaddr.sin_family=AF_INET;
    myaddr.sin_addr.s_addr=INADDR_ANY;
    myaddr.sin_port=htons(6666);

    if(-1==bind(socket_listen,(struct sockaddr*)&myaddr,sizeof(myaddr)))
    {
        perror("bind");
        exit(-1);
    }
//待机
    listen(socket_listen,5);
//接电话
    struct sockaddr_in client_addr;
    socklen_t len;
    pthread_t tid;
    while(1)
    {
        len=sizeof(client_addr);
        int socket_conn=accept(socket_listen,(struct sockaddr*)&client_addr,&len);
        printf("%s:%d已连接!\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
        if(socket_conn==-1)
        {
            perror("accept");
            continue;
        }
        if(0!=pthread_create(&tid,NULL,socket_handle,(void*)socket_conn))
        {
            perror("pthread_create");
            close(socket_conn);
        }
    }
//关手机
    pthread_mutex_destroy(&m);
    pthread_mutex_destroy(&m1);

    close(socket_listen);
    return 0;
}
void* socket_handle(void* arg)
{
    pthread_detach(pthread_self());
    int socket_conn=(int)arg;//电话线
    int flag=0;
    char buf[100];
    int ret;
    char name[20];

//通话
    while(1)//不停收东西//收到名字保存//收到信息转发
    {
            ret=recv(socket_conn,buf,sizeof(buf),0);//收
        if(ret<0)
        {
            printf("连接已断开\n");
        }
        if(ret>0 && flag==0)
        {
            buf[ret]='\0';
            printf("客户昵称:%s\n",buf);
            strcpy(name,buf);//保存用户登录信息
            save_msg(socket_conn);
            send_msg(socket_conn,name,buf,0);
            flag=1; 
            continue;
        }
        if(ret>0 && flag==1)
        {
            buf[ret]='\0';
            send_msg(socket_conn,name,buf,1);//发送线路，姓名，消息
        }
    }
//关电话
    close(socket_conn);
    return NULL;
}
void save_msg(int conn)
{
    pthread_mutex_lock(&m);
    FILE* fp=fopen("msg.txt","a");
    fprintf(fp,"%d\n",conn);
    fclose(fp);
    pthread_mutex_unlock(&m);
}
void send_msg(int send_conn,char* name,char* msg,int flag)//发送线路，姓名，话语,是登陆信息（0）
{
    int conn;
    char buf[100];
    pthread_mutex_lock(&m1);
    FILE* fp=fopen("msg.txt","r");
    while(!feof(fp))
    {
        fscanf(fp,"%d\n",&conn);

        if(send_conn!=conn && flag==0)
            sprintf(buf,"用户(%s)进入聊天室",name);
        if(send_conn==conn && flag==1)
                sprintf(buf,"我：%s",msg);
        if(send_conn!=conn && flag==1)
                sprintf(buf,"%s：%s",name,msg);  
        send(conn,buf,sizeof(buf),0);
    }
    fclose(fp);
    pthread_mutex_unlock(&m1);
}

客户端

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <pthread.h>
void* receive_msg(void* arg);
int main(int argc,char** argv)
{
    int flag=0;
    if(3!=argc)
    {
        printf("参数有误\n");
        exit(-1);
    }
//买手机
    int socket_conn=socket(AF_INET,SOCK_STREAM,0);

//打电话
    struct sockaddr_in ser_addr;
    ser_addr.sin_family=AF_INET;
    ser_addr.sin_addr.s_addr=inet_addr(argv[1]);
    ser_addr.sin_port=htons(atoi(argv[2]));

    if(-1==connect(socket_conn,(struct sockaddr*)&ser_addr,sizeof(ser_addr)))
    {
        perror("connect");
        exit(-1);
    }

    pthread_t tid;
    pthread_create(&tid,NULL,receive_msg,(void*)socket_conn);
    char buf[100];
    int ret;
    while(1)
    {
        if(flag==0)
        {
            printf("请输入您的昵称：");//第一次发的是昵称，后面发的是聊天内容
            flag=1;
        }
        while(getchar()!='\n');
        scanf("%s",buf);
    //  if(strcmp("PuTTY",buf)!=0)
        ret=send(socket_conn,buf,strlen(buf),0);//发
        if(ret<0)
        {
            printf("连接已断开****\n");
            exit(-1);
        }
    }
//挂电话
    close(socket_conn);
    return 0;
}

void* receive_msg(void* arg)
{
    pthread_detach(pthread_self());
    int socket_conn=(int)arg;

    char msg[100];
    int ret;
    while(1)
    {
        ret=recv(socket_conn,msg,sizeof(msg),0);
        if(ret>0)
        {
            msg[ret]='\0';
            printf("%s\n",msg);
        }
    }
    return NULL;
}