Linux 嵌入式 C 实现简单的网络聊天室
版权声明： https://blog.csdn.net/qq_39603089/article/details/85080802
使用C/S 模式，参考下图：
原代码可从直接下载，与所示代码稍有不同，有所改进与完善。
https://blog.csdn.net/qq_39603089/article/details/85327599
服务端无用户界面；
客户端情看: https://blog.csdn.net/qq_39603089/article/details/85080521
服务端代码：
/*-------------------------服务器端chat_server.c-------------------------------
time:2018/12/18
author:zyb-dy
*/
#include <signal.h>
#include <stdio.h>
#include <syslog.h>
#include <stdlib.h>
#include <sys/types.h> //数据类型定义
#include <sys/stat.h> //文件属性
#include <netinet/in.h> //定义数据结构sockaddr_in
#include <sys/socket.h> //提供socket函数和数据结构
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <sys/ipc.h>
#include <arpa/inet.h>
#include <errno.h>
#include<sys/sem.h>
#include <time.h>
#include <pthread.h> //线程
#define THREAD_NUMBER 30 //最大链接数
#define MYPORT 8787
#define BUFFER_SIZE 1024
#define WELCOME "Welcome"
struct Users{ //表示用户与线程之间的关联
char name[50];
pthread_t thread;
char buf[BUFFER_SIZE];
int client_fd;
char address[20];
int login; //是否在线
}users[THREAD_NUMBER];
int sem_id;
//返回系统时间
void get_now_time(char *nt){
time_t tmpcal_ptr;
struct tm *tmp_ptr = NULL;
time(&tmpcal_ptr);
tmp_ptr = localtime(&tmpcal_ptr);
sprintf(nt,"%d:%d:%d", tmp_ptr->tm_hour, tmp_ptr->tm_min, tmp_ptr->tm_sec);
}
int init_sem(int sem_id, int init_value){
union semun{
int val;
struct semid_ds *buf;
unsigned short *array;
};
union semun sem_union;
sem_union.val = init_value;
if(semctl(sem_id, 0, SETVAL, sem_union) == -1){
syslog(LOG_ERR, "Initialize semaphore");
perror("Initialize semaphore");
return -1;
}
return 0;
}
int del_sem(int sem_id){
union semun{
int val;
struct semid_ds *buf;
unsigned short *array;
};
union semun sem_union;
if(semctl(sem_id, 1, IPC_RMID, sem_union)==-1){
syslog(LOG_ERR, "Delete semaphore");
perror("Delete semaphore");
return -1;
}
}
//P 操作函数
int sem_p(int sem_id){
struct sembuf sem_b;
sem_b.sem_num = 0;
sem_b.sem_op = -1;
sem_b.sem_flg = SEM_UNDO;
if(semop(sem_id, &sem_b, 1)==-1){
syslog(LOG_ERR, "P operation");
perror("P operation");
return -1;
}
return 0;
}
//V 操作函数
int sem_v(int sem_id){
struct sembuf sem_b;
sem_b.sem_num = 0;
sem_b.sem_op = 1;
sem_b.sem_flg = SEM_UNDO;
if(semop(sem_id, &sem_b, 1) == -1){
syslog(LOG_ERR, "V operation");
perror("V operation");
return -1;
}
return 0;
}
//发送给个人
void send_all(char loadsend[BUFFER_SIZE]){
int j;
for(j=0;j<THREAD_NUMBER;j++){
if(users[j].login==1){
send(users[j].client_fd, loadsend, strlen(loadsend), 0);
}
}
}
//发送给在线的所有人
void send_only(char name[50],char loadsend[BUFFER_SIZE]){
int j;
for(j=0;j<THREAD_NUMBER;j++){
if((users[j].login==1)&&(strcmp(users[j].name,name)==0)){
send(users[j].client_fd, loadsend, strlen(loadsend), 0);
}
}
}
//处理接受到的字符并调用相应的函数转发；
void strdeal(char s[],char se_name[]){
char sign[8];
char name[56];
char buf[BUFFER_SIZE];
char send_buf[BUFFER_SIZE];
memset(sign, 0, strlen(sign));
memset(name, 0, strlen(name));
memset(buf, 0, strlen(buf));
memset(send_buf, 0, strlen(send_buf));
char nt[10];
int i=0;
int n=0;
int j=0;
for(i;i<strlen(s);i++){
if(n>1){
buf[j]=s[i];
j++;
}else if(n==1){
if(s[i]==':'){
n++;
name[j]='\0';
j=0;
continue;
}
name[j]=s[i];
j++;
}else{
if(s[i]==':'){
n++;
sign[j]='\0';
j=0;
continue;
}
sign[j]=s[i];
j++;
}
}
if(strcmp(sign,"All")==0){
get_now_time(nt);
sprintf(send_buf,"%s用户< %s >群发消息->\t\t%s:\n\t%s","User:",se_name,nt,buf);
send_all(send_buf);
}else{
get_now_time(nt);
sprintf(send_buf,"%s用户< %s >------>\t\t%s:\n\t%s","User:",se_name,nt,buf);
send_only(name,send_buf);
}
}
//线程函数
void *thrd_func(void *arg)
{
long i = (long)arg;
int recvbytes;
char nt[10];
while(1){
memset(users[i].buf , 0, sizeof(users[i].buf));
if ((recvbytes = recv(users[i].client_fd, users[i].buf, BUFFER_SIZE, 0)) <= 0)//结束当前线程
{
char end[100];
memset(end, 0, 100);
get_now_time(nt);
sprintf(end,"%s%s%s\n用户：%s%s\n","Inform:",nt,"-通知：",users[i].name,"退出聊天室"); // 添加时间：
send_all(end);
users[i].login = 0;
sem_v(sem_id);
close(users[i].client_fd);
int n=0;
int j=0;
for(j;j<THREAD_NUMBER;j++){
if(users[j].login==0)
n++;
}
printf("%s用户退出，还可以上线%d个\n",users[i].name, n);
pthread_exit(0); //结束线程
}
strdeal(users[i].buf,users[i].name);
}
}
//端口绑定函数。创建套接字，并绑定到指定端口
int bindPort(unsigned short int port)
{
int sockfd, sendbytes;
struct sockaddr_in my_addr;
if ((sockfd = socket(AF_INET,SOCK_STREAM,0))== -1)//创建基于流套接字
{
syslog(LOG_ERR, "socket");
perror("socket");
}
/*设置sockaddr_in 结构体中相关参数*/
bzero(&my_addr, sizeof(my_addr));
my_addr.sin_family = AF_INET; //IPV4协议族
my_addr.sin_port = htons(port); //转换端口为网络字节序
my_addr.sin_addr.s_addr = INADDR_ANY;
memset(&(my_addr.sin_zero),0,8);
/* 允许重复使用本地地址与套接字进行绑定 */
int i = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &i, sizeof(i));
if (bind(sockfd, (struct sockaddr*)&my_addr, sizeof(struct sockaddr)) == -1)
{
syslog(LOG_ERR, "fail to bind");
perror("fail to bind");
}
printf("success!\n");
return sockfd;
}
/*自定义信号处理函数*/
void my_func(int sign_no)
{
char loadsend[100];
sprintf(loadsend,"%s%s","Inform:","over"); //发送通知 over 客户端收到后自动退出
send_all(loadsend);
printf("即将退出服务器\n");
fflush(stdout);
sleep(1);
exit(0);
}
int main(int argc, char *argv[])
{
int sockfd,recvbytes;
char loadsend[100];
long i=0; //循环来做连接
int sendbytes, res;
struct sockaddr_in client_sockaddr;
int sin_size;
int repetition,j,n;
sockfd = bindPort(MYPORT);
for(i=0;i<THREAD_NUMBER;i++)
users[i].login=0;
i=0;
if (listen(sockfd, THREAD_NUMBER) == -1)
{
syslog(LOG_ERR, "listen");
perror("listen");
}
printf("Listening....\n");
sem_id = semget(ftok("/", 1), 1, 0666|IPC_CREAT);
init_sem(sem_id, THREAD_NUMBER);
char nt[10];
openlog("daemon_syslog", LOG_PID, LOG_DAEMON); //打开系统日志服务
//捕捉信号量，并跳转到信号处理函数处
signal(SIGINT, my_func);
signal(SIGQUIT, my_func);
signal(SIGTSTP,my_func);
while(1){
sem_p(sem_id);
while(i<THREAD_NUMBER){ //遍历列表
if(users[i].login==0) //找到可用资源 0 表示可用 1 不可用
break;
i++;
if(i==THREAD_NUMBER){ //循环查找
i=0;
}
}
n=0;
j=0;
for(j;j<THREAD_NUMBER;j++){
if(users[j].login==1)
n++;
}
printf("已经上线%d个用户，还可以上线%d个\n",n,THREAD_NUMBER-n);
if(users[i].login==0){
printf("等待第下个连接\n");
/*调用accept()函数，等待客户端的连接*/
if ((users[i].client_fd = accept(sockfd,(struct sockaddr*)&client_sockaddr, &sin_size)) == -1){
syslog(LOG_ERR, "accept");
perror("accept");
}
//获取客户端的IP
inet_ntop(AF_INET, &client_sockaddr.sin_addr, users[i].address, sizeof(users[i].address));
if ((recvbytes=recv(users[i].client_fd, users[i].name, BUFFER_SIZE, 0)) <= 0){
sem_v(sem_id);
continue;
};
printf("本次连接的是用户：%s\n",users[i].name);
j=0;
repetition=0;//判断是否重名
for(j=0;j<THREAD_NUMBER;j++){
if(users[j].login==1) //从连接的用户寻找
if(strcmp(users[i].name,users[j].name)==0){
repetition=1;
}
}
if(repetition==1){
send(users[i].client_fd, "g", strlen("g"), 0);
sem_v(sem_id);
continue;
}else{
users[i].login = 1;
send(users[i].client_fd, WELCOME, strlen(WELCOME), 0);
}
//广播上线消息。
memset(loadsend, 0, 100);
get_now_time(nt);
sprintf(loadsend,"%s%s%s\n用户：%s%s\n","Inform:",nt,"-通知：",users[i].name,"-上线了！大家欢迎！");
send_all(loadsend);
res = pthread_create(&users[i].thread, NULL, thrd_func, (void*)i); //开启线程
if (res != 0)
{
syslog(LOG_ERR, "Create thread failed");
perror("Create thread failed");
users[i].login = 0;
sem_v(sem_id);
}
}
}
close(sockfd);
return 0;
}
编译：
gcc chat_server.c -lpthread
效果如下：
扫一扫
1330
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
