int main(int argc, char* argv[])
{
pid_t pid;
int bufSize; // Bounded buffer size
int itemCnt; // Number of items to be produced
int randSeed; // Seed for the random number generator
if(argc != 4){
printf("Invalid number of command-line arguments\n");
exit(1);
}
bufSize = atoi(argv[1]);
itemCnt = atoi(argv[2]);
randSeed = atoi(argv[3]);
// Function that creates a shared memory segment and initializes its header
InitShm(bufSize, itemCnt);
/* fork a child process */
pid = fork();
if (pid < 0) { /* error occurred */
fprintf(stderr, "Fork Failed\n");
exit(1);
}
else if (pid == 0) { /* child process */
printf("Launching Consumer \n");
execlp("./bin/consumer","consumer",NULL);
}
else { /* parent process */
/* parent will wait for the child to complete */
printf("Starting Producer\n");
// The function that actually implements the production
Producer(bufSize, itemCnt, randSeed);
printf("Producer done and waiting for consumer\n");
wait(NULL);
printf("Consumer Completed\n");
}
return 0;
}
int main(int argc, char *argv[])
{
char buffer[512], prog_name[512], sTemp[256];
int nret, i, share_size, count;
int shm_id;
int shm_need_init;
int bRet;
/*检测程序启动参数是否正确*/
if (CheckParam(argc, argv) == -1)
{
return 0;
}
strcpy(buffer, argv[0]);
split_str(buffer, prog_name, "/", 1, 1, 1, 1);
/*检测程序是否已经在后台运行*/
if (strcmp(argv[1], "start") != 0)
{
if ( !if_program_running(prog_name) )
{
WriteLog(0, 0, OUT_SCREEN, "系统尚未启动");
return 0;
}
}
WriteLog(0, 0, OUT_SCREEN, "读取配置文件 ... ");
if (GetInitInfo(argv[0]) == -1)
{
return 0;
}
/*WriteLog(0, 0, OUT_SCREEN, "OK !");*/
if (strcmp(G_ini.cur_version, CURRENT_SERVER_VERSION) != 0)
{
WriteLog(0, 0, OUT_SCREEN, "程序版本错误，禁止运行!");
return 0;
}
G_shm = NULL;
G_ctrl_data = NULL;
G_trans_buffer = NULL;
WriteLog(0, 0, OUT_SCREEN, "读取共享内存 ... ");
shm_need_init = 0;
if (strcmp(argv[1], "start") == 0 && argc == 2)
{
if (atoi(sTemp) > 1)
{
WriteLog(0, 0, OUT_SCREEN, "系统已经执行完成该命令");
return 0;
}
share_size = G_ini.proc_num * ((int)sizeof(struct SHM_CONF)) + (int)sizeof(struct SHM_HEAD);
shm_id = shmget((key_t)G_ini.shm_key, (size_t)share_size, IPC_CREAT | IPC_EXCL | 0777);
shm_need_init = 1;
bRet = sem_requ(G_ini.sem_key, 1, 1);
}
else
{
shm_id = shmget((key_t)G_ini.shm_key, 0, 0777);
bRet = sem_get(G_ini.sem_key, 1);
}
if (shm_id < 0)
{
/*EEXIST 17 File exists*/
WriteLog(0, 0, OUT_SCREEN, " shmget error(%d):%s", errno, strerror(errno));
FreeShm(shm_id);
return 0;
}
if (bRet == SEM_ERROR)
{
WriteLog(0, 0, OUT_SCREEN, " semget error:%s", gs_sem_err);
FreeShm(shm_id);
return 0;
}
G_shm = (char *)shmat(shm_id, 0, 0);
if (G_shm == (void *)-1)
{
WriteLog(0, 0, OUT_SCREEN, " shmat error(%d):%s", errno, strerror(errno));
FreeShm(shm_id);
return 0;
}
if (shm_need_init == 1)
{
/*共享内存初试化*/
InitShm();
}
else
{
if (ShmConfCheck() == -1)
{
WriteLog(0, 0, OUT_SCREEN, "INI文件在启动之后被修改过,禁止执行当前指令!");
if (shm_need_init == 1)
{
FreeShm(shm_id);
}
return 0;
}
}
/*WriteLog(0, 0, OUT_SCREEN, "OK !");*/
if (strcmp(argv[1], "stop") == 0 || strcmp(argv[1], "query") == 0 || strcmp(argv[1], "refresh") == 0)
{
strcpy(buffer, argv[0]);
split_str(buffer, prog_name, "/", 1, 1, 1, 1);
if (strcmp(argv[1], "query") == 0)
{
QueryClient(prog_name);
}
if (strcmp(argv[1], "refresh") == 0)
{
if (argc != 3)
{
RefreshParam(prog_name, 0);
}
else
{
RefreshParam(prog_name, atoi(argv[2]));
}
}
if (strcmp(argv[1], "stop") == 0)
{
WriteLog(0, 0, OUT_SCREEN, "server stop ...");
if (argc != 3)
{
StopClient(prog_name, 0);
}
else
{
StopClient(prog_name, atoi(argv[2]));
}
}
if (strcmp(argv[1], "refresh") != 0)
{
if (GetRunCount() == 0)
{
/*WriteLog(0, 0, OUT_SCREEN, "释放资源 ...");*/
FreeShm(shm_id);
}
}
return 0;
}
/****************************************************************************************/
count = 0;
for (i = 0 ; i < G_ini.port_num ; i++)
{
if (argc != 2)
{
if (atoi(argv[2]) != G_ini.port_list[i].lsnr_port)
{
continue;
}
}
WriteLog(0, 0, OUT_SCREEN, "\n启动:%s接口,port=[%d]", G_ini.port_list[i].name, G_ini.port_list[i].lsnr_port);
nret = StartPort(G_ini.port_list[i].lsnr_port, G_ini.port_list[i].lsnr_num, G_ini.port_list[i].lsnr_len, G_ini.port_list[i].is_reuse_addr);
FreeResource();
if (nret < 0)
{
continue;
}
count++;
}
if (count == 0)
{
WriteLog(0, 0, OUT_SCREEN, "系统启动失败,释放系统资源!");
FreeShm(shm_id);
}
else
{
WriteLog(0, 0, OUT_SCREEN, "系统启动指令执行完成!");
sleep(2);
if (GetRunCount() == 0)
{
WriteLog(0, 0, OUT_SCREEN, "进程启动失败");
FreeShm(shm_id);
}
}
return 0;
}