Linux下后台服务程序开发

 #include <sys/types.h>
    #include <sys/stat.h>
    #include <signal.h>
 
static volatile int  gIsExit = 1;                        /*程序退出标识*/
 
/*******************************************************************/
static void UsageInfo(const char * argv);   /*用法提示*/
static void CatchSignal(int signo);     /*捕捉信号*/
 
int main(int argc, char * argv[])
{
    //公用变量
    char sPidFileName[256];
    //文件控制变量
    struct stat fstat;
    pid_t Pid;
    FILE * fpPid;
    int ret = 0;
    int flag = 0;
    char buf[1024] = {0};
 
    if ( argc != 2 ) {
        UsageInfo(argv[0]);
        exit(0);
    }
 
    sprintf(sPidFileName, "/tmp/WDKMS.PID");
 
    if ( strcasecmp(argv[1],"stop") == 0 ) {
        if ( stat(sPidFileName, &fstat) < 0 ) {
            printf("[%s] isn't runing!/n", argv[0]);
            exit(0);
        }
        if ( fstat.st_uid != geteuid() ) {
            printf("stop failed!/n", argv[0]);
            exit(0);
        }
 
        if ( (fpPid = fopen(sPidFileName, "r")) == NULL ) {
            printf("open [%s] error!/n", argv[0]);
            exit(0);
        }
 
        fscanf(fpPid,"%d",&Pid);
        fclose(fpPid);
 
        printf("[%s] stoping....../n", argv[0]);
 
        if ( kill(Pid,SIGUSR1) != 0 ) {
            printf("[%s] send signal failed!/n", argv[0]);
            exit(0);
        }
 
        while ( stat(sPidFileName, &fstat)==0 ){
            printf("waiting.../n");
            apr_sleep(apr_time_from_sec(1));
        }
        printf("DONE!/n", argv[0]);
        exit(0);
    } else if ( strcasecmp(argv[1],"start") != 0 ) {
        UsageInfo(argv[0]);
        exit(0);
    }
 
    if ( stat(sPidFileName, &fstat)==0 ) {
        printf("[%s] is exist!/n", argv[0]);
        exit(0);
    } else {
        /* 将程序转入后台运行，成为守护程序 */
        if ( fork() > 0 ) exit(0);
        setsid();
        if ( fork() > 0 ) exit(0);
 
        /* 把本程序的pid写入 /tmp/HSMServer_pid 中*/
        if ( (fpPid = fopen(sPidFileName, "w")) == NULL ) {
            printf("create [%s] failed!", sPidFileName);
            exit(0);
        }
        Pid = getpid();
        fprintf(fpPid,"%d",Pid);
        fclose(fpPid);
    }
 
    if ( server_start("HSM.ini") != 0 ) {
        goto quit;
    }
 
    signal(SIGUSR1, CatchSignal);
   
    while ( gIsExit ) {
        apr_sleep(apr_time_from_sec(SLEEP_TIME)); 
    }
 
quit:
    server_close();
 
    KMS_LogEntry("main", "exit DONE", 0);
    if ( remove(sPidFileName) != 0 ) {
        printf("app exit! rm pid failed!/n");
    }
 
    return 0;
}
 
static void CatchSignal( int signo)
{
    gIsExit = 0;
    return;
}
 
static void UsageInfo(const char * argv)
{
    printf("Usage:/n/t %s start|stop/n",argv);
}