C语言多人聊天室 ---chat(客户端聊天)

head.h

#ifndef __HEAD_H
#define __HEAD_H// 常用头文件
#include <stdio.h>
#include <stdlib.h>
#include <string.h>// 网络编程涉及的头文件
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>#include <sys/types.h>// 本机字节序和网络字节序转换相关函数的头文件
#include <arpa/inet.h>// 关闭套接字用close函数需要的头文件
#include <unistd.h>//线程相关的函数头, mutex相关的函数
#include <pthread.h>// 类型重命名：地址结构体的规范
typedef struct sockaddr SockAddr;// 地址结构体的规范的实现结构体
typedef struct sockaddr_in SockAddrIn;// if_nametoindex
#include <net/if.h>#include <sys/select.h>
#include <sys/time.h>// JSON字符串的封装
#include <stdio.h>
#include <stdlib.h>
#include <string.h>// 文件操作需要的头文件
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>#include <fcntl.h>
#include <mysql/mysql.h>// selet
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/types.h>// 定义外部变量
//int sockfdNUM[10] = {0};#endif

p_char.h

#ifndef _P_CHAT_H
#define _P_CHAT_H#define BUF_SIZE 1024#include "head.h"// 错误处理函数，用于输出错误信息并终止程序
void error_handling(const char *message);#endif

p_chat.c

#include "p_chat.h"// 私聊接收任务函数
void *s_private_chat(void *arg)
{// 1.获取客户端套接字int sock = *(int *)arg;char buf[1024] = {0};while (1){scanf("%s", buf);send(sock, buf, strlen(buf), 0);if (strcmp(buf, "quit") == 0){break;}memset(buf, 0, sizeof(buf));}// 关闭服务端套接字close(sock);exit(0); // 结束整个线程
}void *r_private_chat(void *arg)
{// 1.获取客户端套接字int sock = *(int *)arg;char buf[1024] = {0};while (1){recv(sock, buf, sizeof(buf), 0);printf("%s\n", buf);memset(buf, 0, sizeof(buf));}// 关闭服务端套接字close(sock);
}// 客户端
int main(int argc, char const *argv[])
{// 创建线程idpthread_t tid1;pthread_t tid2;// 1.创建socketint s_sockfd = socket(AF_INET, SOCK_STREAM, 0);if (-1 == s_sockfd){perror("socket failed");return -1;}// 2.绑定地址SockAddrIn serverAddr;// ipv4协议serverAddr.sin_family = AF_INET;// 端口serverAddr.sin_port = htons(8888);// ip地址serverAddr.sin_addr.s_addr = inet_addr("192.168.118.129");// 创建一个用于存放的json字符的数组char *jsonString;// 3.连接服务器int ret = connect(s_sockfd, (SockAddr *)&serverAddr, sizeof(serverAddr));if (-1 == ret){perror("connect failed");return -1;}// 4.1 接收服务器消息char buf[2048] = {0};ret = recv(s_sockfd, buf, sizeof(buf), 0);if (-1 == ret){perror("recv failed");return -1;}int uid, fid;uid = atoi(argv[1]);fid = atoi(argv[2]);printf("uid = %d\n", uid);printf("fid = %d\n", fid);printf("s_sockfd = %d\n", s_sockfd);// 判断上一个进程传来的数据if (strcmp(argv[3], "私聊") == 0){// 向服务端发送自己的状态int stat = 1;send(s_sockfd, &stat, sizeof(stat), 0);// 私聊// 发送uidsend(s_sockfd, &uid, sizeof(uid), 0);// 发送fidsend(s_sockfd, &fid, sizeof(fid), 0);//puts("待接收历史消息");// 清空bufmemset(buf, 0, sizeof(buf));// 接收聊天记录recv(s_sockfd, buf, sizeof(buf), 0);//puts("历史消息接收完毕:");puts(buf);puts("以上是历史消息");// 创建线程pthread_create(&tid1, NULL, s_private_chat, &s_sockfd);pthread_create(&tid2, NULL, r_private_chat, &s_sockfd);// 等待线程结束pthread_join(tid1, NULL);}else if (strcmp(argv[3], "群聊") == 0){// 向服务端发送自己的状态int stat = 2;send(s_sockfd, &stat, sizeof(stat), 0);// 公屏聊天// 发送uidsend(s_sockfd, &uid, sizeof(uid), 0);puts("开始聊天");// 创建线程pthread_create(&tid1, NULL, s_private_chat, &s_sockfd);pthread_create(&tid2, NULL, r_private_chat, &s_sockfd);// 等待线程结束pthread_join(tid1, NULL);}// 5.关闭socketclose(s_sockfd);puts("关闭socket");return 0;
}

makefile

SRCS = $(wildcard *.c)
OBJS = $(patsubst *.c,*.o,$(SRCS))
CC = gcc
TARGET = p_chat
LDFLAGS = -lpthread -lmysqlclient.PHONY : clean$(TARGET) : $(OBJS)$(CC)  $^ -o $@ $(LDFLAGS)clean :rm *.orun :./$(TARGET)show :echo $(SRCS) / $(OBJS)