一、设计思路
采用多进程的方式,实现聊天室功能。
客户端:开启一个子线程,接受用户输入的信息。
主线程负责将用户的信息发送给服务器,并且接受并显示服务器发送过来的消息
服务器端: 每当有一个客户端发送了连接请求,就创建一个子进程处理。
程序启动时,会开启一个共享内存,存放客户端发送过来的消息。
并且每个客户端发送过来的消息,都会发送给每个连接的客户端,实现聊天室的功能。
二、 代码展示
1. 客户端代码:
参见 http://blog.csdn.net/realizelizijun2013/article/details/24671999#t2
2. 服务器端代码
/*
* 功能: 每当有一个客户端发送了连接请求,就创建一个子进程处理
* 程序启动时,会开启一个共享内存,存放客户端发送过来的消息
* 并且每个客户端发送过来的消息,都会发送给每个连接的客户端,实现聊天室的功能
* 作者: lizijun
* 日期: 2014.05.18
* 版本: 1.0
*/
#include <stdio.h>
#include <string.h>
#include <sys/select.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <unistd.h>
#include <errno.h>
#include <netinet/in.h> //for INADDR_ANY
#include <semaphore.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <sys/msg.h> //for IPC_CREAT
#include <signal.h> //for SIGCLD
#include "charrrom_server_02.h"
sem_t *sem = NULL; //POSIX有名信号量
int shm_id; //系统V共享内存标识
static int iCurrentClientIndex = -1; //当前客户端的索引号
int main(void)
{
int listenSocket = -1;
InitSign();
InitSocket(&listenSocket);
CreateSem();
CreateShareMemory();
HandleClientConnect(listenSocket);
return 0;
}
/*
* 功能:初始化信号
*/
void InitSign(void)
{
//父进程可将SIGCHLD的处理函数设为SIG_IGN
//以使内核可以自动回收已终止的子进程的资源,防止僵尸进程
signal(SIGCLD, SIG_IGN);
}
/*
* 功能: 创建POSIX有名信号量
*/
void CreateSem(void)
{
int flag = 0;
unsigned int value = 0;
flag = O_CREAT | O_EXCL;
value = 1;
sem = sem_open(SEM_NAME_PATH, flag, 0644, value);
if (SEM_FAILED == sem) //信号量可能存在
{
//如果信号量存在,删除,重新创建,目的是保证信号量初始化值为1
//这样做的目的: 当进程异常断开,此时信号量又没有释放,再加上有名信号量是随内核存在的
//当下次进程重启的时候,信号量就会处于使用状态,初始化是为0
if (errno == EEXIST)
{
printf("sem exist, create again.\n");
sem_unlink(SEM_NAME_PATH);
sem = sem_open(SEM_NAME_PATH, flag, 0644, value);
}
if (SEM_FAILED == sem)
{
printf("sem_open error\n");
return;
}
}
}
/*
* 功能: 创建系统V共享内存
*/
void CreateShareMemory(void)
{
key_t key;
key = ftok(SHARE_MEMORY_PATH,0);