#include "sem_util.h"
|
#include "logger_factory.h"
|
#include <sys/sem.h>
|
#include <sys/shm.h>
|
#include <sys/types.h>
|
#include <sys/ipc.h>
|
#include <sys/msg.h>
|
#include <errno.h>
|
#include "msg_mgr.h"
|
|
static Logger *logger = LoggerFactory::getLogger();
|
|
static int sem_msg_id;
|
static Msg_info gMsg_buf[MAX_LOCK];
|
|
static int rsv_msg_id;
|
|
int msg_init(void)
|
{
|
key_t key = ftok(MSG_PATH, 0x01);
|
|
memset(gMsg_buf, 0x00, sizeof(gMsg_buf));
|
sem_msg_id = msgget((key_t)key, 0666 | IPC_CREAT);
|
if(sem_msg_id < 0) {
|
logger->error("msgget failed with error: %d\n", sem_msg_id);
|
|
return -1;
|
}
|
|
key_t key_rsv = ftok(MSG_RSV_PATH, 0x02);
|
rsv_msg_id = msgget((key_t)key_rsv, 0666 | IPC_CREAT);
|
if(rsv_msg_id == -1) {
|
logger->error("msgget failed with error: %d\n", rsv_msg_id);
|
|
return -1;
|
}
|
|
return 0;
|
|
}
|
|
void msg_distrib(int msg_id, Msg_info *message)
|
{
|
int ret;
|
Msg_info* msg_obj = message;
|
|
switch(msg_id)
|
{
|
case SEM_TYPE_ID:
|
msg_obj->mtype = MSG_TYPE_SEM;
|
ret = msgsnd(sem_msg_id, msg_obj, sizeof(Msg_info) - sizeof(long), IPC_NOWAIT);
|
if(ret < 0) {
|
logger->error("msg send fail: %d\n", ret);
|
|
return;
|
|
}
|
|
break;
|
|
case RSV_TYPE_ID:
|
msg_obj->mtype = MSG_TYPE_RSV;
|
ret = msgsnd(rsv_msg_id, msg_obj, sizeof(Msg_info), 0);
|
if(ret < 0) {
|
logger->error("msg send fail: %d\n", ret);
|
|
return;
|
|
}
|
|
break;
|
|
default:
|
logger->error("unknown msg type!\n");
|
|
}
|
|
}
|
|
int get_msg_info(int Msgid, Msg_info *message)
|
{
|
int ret = 0;
|
Msg_info msg_obj;
|
int msg_id;
|
|
if (Msgid == SEM_TYPE_ID) {
|
msg_id = sem_msg_id;
|
} else if (Msgid == RSV_TYPE_ID) {
|
msg_id = rsv_msg_id;
|
}
|
|
ret = msgrcv(msg_id, (void*)&msg_obj, sizeof(Msg_info) - sizeof(long), MSG_TYPE_SEM, 0);
|
if(ret < 0) {
|
|
logger->error("msg recv fail: %d\n", ret);
|
|
return -1;
|
|
}
|
|
switch(Msgid) {
|
case SEM_TYPE_ID:
|
msg_info_set(SEM_TYPE_ID, msg_obj);
|
|
break;
|
|
case RSV_TYPE_ID:
|
msg_info_set(SEM_TYPE_ID, msg_obj);
|
|
break;
|
|
default:
|
logger->error("unknown msg type!\n");
|
|
}
|
|
return 0;
|
|
}
|
|
void msg_info_set(int index, Msg_info msg_obj)
|
{
|
gMsg_buf[index].key = msg_obj.key;
|
gMsg_buf[index].id = msg_obj.id;
|
|
if (msg_obj.act == SEM_POST) {
|
gMsg_buf[index].count++;
|
} else if (msg_obj.act == SEM_GET) {
|
gMsg_buf[index].count--;
|
} else if (msg_obj.act == SEM_RESET) {
|
gMsg_buf[index].count = 0;
|
}
|
|
}
|
|
void *sem_msg_handler(void *skptr)
|
{
|
int ret;
|
Msg_info msg_obj;
|
|
msg_init();
|
while(true) {
|
ret = get_msg_info(SEM_TYPE_ID, &msg_obj);
|
if (ret == 0) {
|
//process_msg(msg_obj);
|
}
|
}
|
}
|
