【linux】多进程实现聊天室功能

一、设计思路

采用多进程的方式，实现聊天室功能。

客户端：开启一个子线程，接受用户输入的信息。
主线程负责将用户的信息发送给服务器，并且接受并显示服务器发送过来的消息

 服务器端： 每当有一个客户端发送了连接请求，就创建一个子进程处理。
 程序启动时，会开启一个共享内存，存放客户端发送过来的消息。
 并且每个客户端发送过来的消息，都会发送给每个连接的客户端，实现聊天室的功能。

二、 代码展示

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);