From fbc32792b8d8266ff90aa60403f5da78739236f4 Mon Sep 17 00:00:00 2001 From: Michael Krelin Date: Sat, 24 Jul 2004 00:24:07 +0000 Subject: processes specified by process names (pidof-like). --- diff --git a/NEWS.xml b/NEWS.xml index cd12d05..cc6085d 100644 --- a/NEWS.xml +++ b/NEWS.xml @@ -1,7 +1,8 @@ - + now dudki sends arbitrary signals to the processes being monitored from the command line + detection of running processes which do not keep pidfiles, using process name initgroups() called before executing RestartCommand diff --git a/man/dudki.conf.5.in b/man/dudki.conf.5.in index 23f636d..7d365d3 100644 --- a/man/dudki.conf.5.in +++ b/man/dudki.conf.5.in @@ -1,4 +1,4 @@ -.TH dudki.conf 5 "July 9th, 2004" "dudki.conf(5)" "Klever Group (http://www.klever.net/)" +.TH dudki.conf 5 "July 24th, 2004" "dudki.conf(5)" "Klever Group (http://www.klever.net/)" .hla en .SH NAME @@ -60,6 +60,10 @@ monitored from. The absence of file, as well as the absence of process specified by the pid stored in the file signifies the process death and triggers restart. .TP +\fBProcessName\fR \fIprocess name\fR +Specifies the name of the process. The alternative way to find process if it +doesn't keep pid in the file. Similar to \fBpidof\fR(1). +.TP \fBRestartCommand\fR \fIcommand\fR Specifies the command to run in order to restart the process. .TP diff --git a/src/configuration.cc b/src/configuration.cc index eb010c1..edc8c04 100644 --- a/src/configuration.cc +++ b/src/configuration.cc @@ -92,6 +92,10 @@ static DOTCONF_CB(dco__process) { dc_context *dcc = (dc_context*)ctx; return NULL; } +static DOTCONF_CB(dco_process_name) { dc_context *dcc = (dc_context*)ctx; + dcc->ps->process_name = cmd->data.str; + return NULL; +} static DOTCONF_CB(dco_restart_command) { dc_context *dcc = (dc_context*)ctx; dcc->ps->restart_cmd = cmd->data.str; return NULL; @@ -116,6 +120,7 @@ static const configoption_t dc_options[] = { { "MailtoHeader", ARG_STR, dco_mailto_header, NULL, DCC_ROOT|DCC_PROCESS }, { "Notify", ARG_STR, dco_notify, NULL, DCC_ROOT|DCC_PROCESS }, { "second.check(i->first,config); + process::unprepare_herd(); } void signal_self(const configuration& config,int signum) { @@ -220,7 +222,10 @@ int main(int argc,char **argv) { processes_t::const_iterator i = config.processes.find(argv[narg]); if(i==config.processes.end()) throw runtime_error("no such process configured"); - i->second.signal(op_signum); + if(op_signum) + i->second.signal(op_signum); + else + i->second.check(); }catch(exception& e) { cerr << "dudki(" << argv[narg] << "): " << e.what() << endl; failures++; diff --git a/src/process.cc b/src/process.cc index 1ffac9f..8a5b5d2 100644 --- a/src/process.cc +++ b/src/process.cc @@ -4,19 +4,35 @@ #include #include #include +#include #include #include #include #include #include +#include #include +#include +#include using namespace std; #include "process.h" #include "configuration.h" +static multimap procpids; + +void process::check() const { + if(!pidfile.empty()) { + signal(0); + }else if(!process_name.empty()) { + if(procpids.empty()) + gather_proc_info(); + if(procpids.find(process_name)==procpids.end()) + throw runtime_error("no such process"); + } // XXX: or else? +} void process::check(const string& id,configuration& config) { try { - signal(0); + check(); patience = 0; }catch(exception& e) { if(patience>60) { // TODO: configurable @@ -176,14 +192,94 @@ void process::notify_mailto(const string& email,const string& id,const string& e } void process::signal(int signum) const { - ifstream pids(pidfile.c_str(),ios::in); - if(!pids) - throw runtime_error("no pidfile found"); - pid_t pid = 0; - pids >> pid; - pids.close(); - if(!pid) - throw runtime_error("no pid in pidfile"); - if(kill(pid,signum)) - throw runtime_error("failed to signal process"); + if(!pidfile.empty()) { + ifstream pids(pidfile.c_str(),ios::in); + if(!pids) + throw runtime_error("no pidfile found"); + pid_t pid = 0; + pids >> pid; + pids.close(); + if(!pid) + throw runtime_error("no pid in pidfile"); + if(kill(pid,signum)) + throw runtime_error("failed to signal process"); + }else if(!process_name.empty()) { + if(procpids.empty()) + gather_proc_info(); + pair::const_iterator,multimap::const_iterator> range = procpids.equal_range(process_name); + int count = 0; + for(multimap::const_iterator i=range.first;i!=range.second;++i) { + pid_t pid = i->second; + if(kill(i->second,signum)) + throw runtime_error("failed to signal process"); + count++; + } + if(!count) + throw runtime_error("no running instance detected"); + }else + throw runtime_error("nothing is known about the process"); +} + +void process::prepare_herd() { + procpids.clear(); +} +void process::unprepare_herd() { + procpids.clear(); +} +void process::gather_proc_info() { + vector allpids; + DIR *pd = opendir("/proc"); + if(!pd) + throw runtime_error("failed to open /proc"); + struct dirent *pde; + pid_t selfpid = getpid(); + while(pde=readdir(pd)) { + errno=0; + pid_t pid = atoi(pde->d_name); + if((!pid) || pid==selfpid) + continue; + allpids.push_back(pid); + } + closedir(pd); + char s[256]; + procpids.clear(); + for(vector::const_iterator i=allpids.begin();i!=allpids.end();++i) { + int r = snprintf(s,sizeof(s),"/proc/%d/stat",*i); + if(r>=sizeof(s) || r<1) + continue; + string cmd; + ifstream ss(s,ios::in); + if(!ss) + continue; + getline(ss,cmd); + string::size_type op = cmd.find('('); + if(op==string::npos) + continue; + cmd.erase(0,op+1); + string::size_type cp = cmd.find(')'); + if(cp==string::npos) + continue; + cmd.erase(cp); + r = snprintf(s,sizeof(s),"/proc/%d/cmdline",*i); + if(r>=sizeof(s) || r<1) + continue; + ifstream cs(s,ios::binary); + if(!cs) + continue; + string command; + while(cs) { + string cl; + getline(cs,cl,(char)0); + string::size_type lsl = cl.rfind('/'); + if(lsl!=string::npos) + cl.erase(0,lsl+1); + if(cl.substr(0,cmd.length())==cmd) { + command = cl; + break; + } + } + procpids.insert(pair(cmd,*i)); + if((!command.empty()) && cmd!=command) + procpids.insert(pair(command,*i)); + } } diff --git a/src/process.h b/src/process.h index 27ee049..90b12d9 100644 --- a/src/process.h +++ b/src/process.h @@ -12,6 +12,7 @@ typedef map headers_t; class process { public: string pidfile; + string process_name; string restart_cmd; string notify; string user; @@ -31,6 +32,11 @@ class process { const string& description,configuration& config); void signal(int signum) const; + void check() const; + + static void prepare_herd(); + static void gather_proc_info(); + static void unprepare_herd(); }; typedef map processes_t; -- cgit v0.9.0.2