mirror of
				https://github.com/ElvishArtisan/rivendell.git
				synced 2025-11-03 23:53:59 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			145 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			145 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
// rdservice.cpp
 | 
						|
//
 | 
						|
// Rivendell Services Manager
 | 
						|
//
 | 
						|
//   (C) Copyright 2018 Fred Gleason <fredg@paravelsystems.com>
 | 
						|
//
 | 
						|
//   This program is free software; you can redistribute it and/or modify
 | 
						|
//   it under the terms of the GNU General Public License version 2 as
 | 
						|
//   published by the Free Software Foundation.
 | 
						|
//
 | 
						|
//   This program is distributed in the hope that it will be useful,
 | 
						|
//   but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
//   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
//   GNU General Public License for more details.
 | 
						|
//
 | 
						|
//   You should have received a copy of the GNU General Public
 | 
						|
//   License along with this program; if not, write to the Free Software
 | 
						|
//   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 | 
						|
//
 | 
						|
 | 
						|
#include <signal.h>
 | 
						|
#include <stdio.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <unistd.h>
 | 
						|
#include <sys/types.h>
 | 
						|
 | 
						|
#include <qcoreapplication.h>
 | 
						|
 | 
						|
#include <rd.h>
 | 
						|
#include <rdapplication.h>
 | 
						|
#include <rdcmd_switch.h>
 | 
						|
#include <rdconf.h>
 | 
						|
 | 
						|
#include "rdservice.h"
 | 
						|
 | 
						|
bool global_exiting=false;
 | 
						|
bool global_reload_dropboxes=false;
 | 
						|
 | 
						|
void SigHandler(int signo)
 | 
						|
{
 | 
						|
  switch(signo) {
 | 
						|
  case SIGTERM:
 | 
						|
  case SIGINT:
 | 
						|
    global_exiting=true;
 | 
						|
    break;
 | 
						|
 | 
						|
  case SIGUSR1:
 | 
						|
    global_reload_dropboxes=true;
 | 
						|
    break;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
MainObject::MainObject(QObject *parent)
 | 
						|
  :QObject(parent)
 | 
						|
{
 | 
						|
  QString err_msg;
 | 
						|
 | 
						|
  //
 | 
						|
  // Check for prior instance
 | 
						|
  //
 | 
						|
  if(RDGetPids("rdservice").size()>1) {
 | 
						|
    fprintf(stderr,"rdservice: prior instance found\n");
 | 
						|
    exit(1);
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Open the Database
 | 
						|
  //
 | 
						|
  rda=new RDApplication("rdservice","rdservice","\n\n",this);
 | 
						|
  if(!rda->open(&err_msg)) {
 | 
						|
    fprintf(stderr,"rdservice: %s\n",(const char *)err_msg.utf8());
 | 
						|
    exit(1);
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Exit Timer
 | 
						|
  //
 | 
						|
  ::signal(SIGINT,SigHandler);
 | 
						|
  ::signal(SIGTERM,SigHandler);
 | 
						|
  ::signal(SIGUSR1,SigHandler);
 | 
						|
  svc_exit_timer=new QTimer(this);
 | 
						|
  connect(svc_exit_timer,SIGNAL(timeout()),this,SLOT(exitData()));
 | 
						|
  svc_exit_timer->start(100);
 | 
						|
 | 
						|
  if(!Startup(&err_msg)) {
 | 
						|
    Shutdown();
 | 
						|
    fprintf(stderr,"rdservice: %s\n",(const char *)err_msg.toUtf8());
 | 
						|
    exit(1);
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Maintenance Routine Timer
 | 
						|
  //
 | 
						|
  srandom(QTime::currentTime().msec());
 | 
						|
  svc_maint_timer=new QTimer(this);
 | 
						|
  svc_maint_timer->setSingleShot(true);
 | 
						|
  connect(svc_maint_timer,SIGNAL(timeout()),this,SLOT(checkMaintData()));
 | 
						|
  int interval=GetMaintInterval();
 | 
						|
  if(!rda->config()->disableMaintChecks()) {
 | 
						|
    svc_maint_timer->start(interval);
 | 
						|
  }
 | 
						|
  else {
 | 
						|
    rda->log(RDConfig::LogInfo,"maintenance checks disabled on this host!");
 | 
						|
  }
 | 
						|
 | 
						|
  if(!RDWritePid(RD_PID_DIR,"rdservice.pid",getuid())) {
 | 
						|
    fprintf(stderr,"rdservice: can't write pid file\n");
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void MainObject::processFinishedData(int id)
 | 
						|
{
 | 
						|
  svc_processes[id]->deleteLater();
 | 
						|
  svc_processes.remove(id);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void MainObject::exitData()
 | 
						|
{
 | 
						|
  QString err_msg;
 | 
						|
 | 
						|
  if(global_exiting) {
 | 
						|
    Shutdown();
 | 
						|
    RDDeletePid(RD_PID_DIR,"rdservice.pid");
 | 
						|
    exit(0);
 | 
						|
  }
 | 
						|
 | 
						|
  if(global_reload_dropboxes) {
 | 
						|
    ShutdownDropboxes();
 | 
						|
    StartDropboxes(&err_msg);
 | 
						|
    ::signal(SIGUSR1,SigHandler);
 | 
						|
    global_reload_dropboxes=false;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
int main(int argc,char *argv[])
 | 
						|
{
 | 
						|
  QCoreApplication a(argc,argv);
 | 
						|
  new MainObject();
 | 
						|
  return a.exec();
 | 
						|
}
 |