summaryrefslogtreecommitdiffabout
Unidiff
Diffstat (more/less context) (ignore whitespace changes)
-rw-r--r--NEWS.xml1
-rw-r--r--src/process.cc32
2 files changed, 22 insertions, 11 deletions
diff --git a/NEWS.xml b/NEWS.xml
index bc0af31..2e35c11 100644
--- a/NEWS.xml
+++ b/NEWS.xml
@@ -1,18 +1,19 @@
1<?xml version="1.0" encoding="us-ascii"?> 1<?xml version="1.0" encoding="us-ascii"?>
2<news> 2<news>
3 <version version="0.2.1"> 3 <version version="0.2.1">
4 <ni>do not act -e when doing -r</ni> 4 <ni>do not act -e when doing -r</ni>
5 <ni>proper handling of <kbd>ProcessName</kbd> on <kbd>FreeBSD</kbd></ni>
5 </version> 6 </version>
6 <version version="0.2" date="July 24th, 2004"> 7 <version version="0.2" date="July 24th, 2004">
7 <ni>now dudki sends arbitrary signals to the processes being monitored from the command line</ni> 8 <ni>now dudki sends arbitrary signals to the processes being monitored from the command line</ni>
8 <ni>detection of running processes which do not keep pidfiles, using process name</ni> 9 <ni>detection of running processes which do not keep pidfiles, using process name</ni>
9 </version> 10 </version>
10 <version version="0.1" date="July 21st, 2004"> 11 <version version="0.1" date="July 21st, 2004">
11 <ni><kbd>initgroups()</kbd> called before executing <kbd>RestartCommand</kbd></ni> 12 <ni><kbd>initgroups()</kbd> called before executing <kbd>RestartCommand</kbd></ni>
12 <ni>more civilized way of restarting on <kbd>SIGHUP</kbd></ni> 13 <ni>more civilized way of restarting on <kbd>SIGHUP</kbd></ni>
13 <ni>minor changes to build process</ni> 14 <ni>minor changes to build process</ni>
14 </version> 15 </version>
15 <version version="0.0" date="July 11th, 2004"> 16 <version version="0.0" date="July 11th, 2004">
16 <ni>Initial release</ni> 17 <ni>Initial release</ni>
17 </version> 18 </version>
18</news> 19</news>
diff --git a/src/process.cc b/src/process.cc
index 8a5b5d2..96c874f 100644
--- a/src/process.cc
+++ b/src/process.cc
@@ -156,130 +156,140 @@ void process::notify_mailto(const string& email,const string& id,const string& e
156 close(files[0]); 156 close(files[0]);
157 close(files[1]); 157 close(files[1]);
158 throw runtime_error("Failed to vfork()"); 158 throw runtime_error("Failed to vfork()");
159 } 159 }
160 if(!pid) { 160 if(!pid) {
161 // child 161 // child
162 if(dup2(files[0],0)!=0) 162 if(dup2(files[0],0)!=0)
163 _exit(-1); 163 _exit(-1);
164 close(1); 164 close(1);
165 close(files[0]); 165 close(files[0]);
166 close(files[1]); 166 close(files[1]);
167 execl("/usr/sbin/sendmail","usr/sbin/sendmail","-i",email.c_str(),NULL); 167 execl("/usr/sbin/sendmail","usr/sbin/sendmail","-i",email.c_str(),NULL);
168 _exit(-1); 168 _exit(-1);
169 } 169 }
170 // parent 170 // parent
171 close(files[0]); 171 close(files[0]);
172 FILE *mta = fdopen(files[1],"w"); 172 FILE *mta = fdopen(files[1],"w");
173 for(headers_t::const_iterator i=mailto_headers.begin();i!=mailto_headers.end();++i) { 173 for(headers_t::const_iterator i=mailto_headers.begin();i!=mailto_headers.end();++i) {
174 fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str()); 174 fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str());
175 } 175 }
176 for(headers_t::const_iterator i=config.mailto_headers.begin();i!=config.mailto_headers.end();++i) { 176 for(headers_t::const_iterator i=config.mailto_headers.begin();i!=config.mailto_headers.end();++i) {
177 if(mailto_headers.find(i->first)!=mailto_headers.end()) 177 if(mailto_headers.find(i->first)!=mailto_headers.end())
178 continue; 178 continue;
179 fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str()); 179 fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str());
180 } 180 }
181 fprintf(mta, 181 fprintf(mta,
182 "Subject: [%s] %s\n\n" 182 "Subject: [%s] %s\n\n"
183 "%s\n" 183 "%s\n"
184 "---\n" 184 "---\n"
185 "This message was sent automatically by the 'dudki' daemon\n", 185 "This message was sent automatically by the 'dudki' daemon\n",
186 id.c_str(), event.c_str(), 186 id.c_str(), event.c_str(),
187 description.c_str() ); 187 description.c_str() );
188 fclose(mta); 188 fclose(mta);
189 int status; 189 int status;
190 waitpid(pid,&status,0); 190 waitpid(pid,&status,0);
191 // TODO: check the return code 191 // TODO: check the return code
192} 192}
193 193
194void process::signal(int signum) const { 194void process::signal(int signum) const {
195 if(!pidfile.empty()) { 195 if(!pidfile.empty()) {
196 ifstream pids(pidfile.c_str(),ios::in); 196 ifstream pids(pidfile.c_str(),ios::in);
197 if(!pids) 197 if(!pids)
198 throw runtime_error("no pidfile found"); 198 throw runtime_error("no pidfile found");
199 pid_t pid = 0; 199 pid_t pid = 0;
200 pids >> pid; 200 pids >> pid;
201 pids.close(); 201 pids.close();
202 if(!pid) 202 if(!pid)
203 throw runtime_error("no pid in pidfile"); 203 throw runtime_error("no pid in pidfile");
204 if(kill(pid,signum)) 204 if(kill(pid,signum))
205 throw runtime_error("failed to signal process"); 205 throw runtime_error("failed to signal process");
206 }else if(!process_name.empty()) { 206 }else if(!process_name.empty()) {
207 if(procpids.empty()) 207 if(procpids.empty())
208 gather_proc_info(); 208 gather_proc_info();
209 pair<multimap<string,pid_t>::const_iterator,multimap<string,pid_t>::const_iterator> range = procpids.equal_range(process_name); 209 pair<multimap<string,pid_t>::const_iterator,multimap<string,pid_t>::const_iterator> range = procpids.equal_range(process_name);
210 int count = 0; 210 int count = 0;
211 for(multimap<string,pid_t>::const_iterator i=range.first;i!=range.second;++i) { 211 for(multimap<string,pid_t>::const_iterator i=range.first;i!=range.second;++i) {
212 pid_t pid = i->second; 212 pid_t pid = i->second;
213 if(kill(i->second,signum)) 213 if(kill(i->second,signum))
214 throw runtime_error("failed to signal process"); 214 throw runtime_error("failed to signal process");
215 count++; 215 count++;
216 } 216 }
217 if(!count) 217 if(!count)
218 throw runtime_error("no running instance detected"); 218 throw runtime_error("no running instance detected");
219 }else 219 }else
220 throw runtime_error("nothing is known about the process"); 220 throw runtime_error("nothing is known about the process");
221} 221}
222 222
223void process::prepare_herd() { 223void process::prepare_herd() {
224 procpids.clear(); 224 procpids.clear();
225} 225}
226void process::unprepare_herd() { 226void process::unprepare_herd() {
227 procpids.clear(); 227 procpids.clear();
228} 228}
229void process::gather_proc_info() { 229void process::gather_proc_info() {
230 vector<pid_t> allpids; 230 vector<pid_t> allpids;
231 DIR *pd = opendir("/proc"); 231 DIR *pd = opendir("/proc");
232 if(!pd) 232 if(!pd)
233 throw runtime_error("failed to open /proc"); 233 throw runtime_error("failed to open /proc");
234 struct dirent *pde; 234 struct dirent *pde;
235 pid_t selfpid = getpid(); 235 pid_t selfpid = getpid();
236 while(pde=readdir(pd)) { 236 while(pde=readdir(pd)) {
237 errno=0; 237 errno=0;
238 pid_t pid = atoi(pde->d_name); 238 pid_t pid = atoi(pde->d_name);
239 if((!pid) || pid==selfpid) 239 if((!pid) || pid==selfpid)
240 continue; 240 continue;
241 allpids.push_back(pid); 241 allpids.push_back(pid);
242 } 242 }
243 closedir(pd); 243 closedir(pd);
244 char s[256]; 244 char s[256];
245 procpids.clear(); 245 procpids.clear();
246 for(vector<pid_t>::const_iterator i=allpids.begin();i!=allpids.end();++i) { 246 for(vector<pid_t>::const_iterator i=allpids.begin();i!=allpids.end();++i) {
247 int r = snprintf(s,sizeof(s),"/proc/%d/stat",*i); 247 int r = snprintf(s,sizeof(s),"/proc/%d/stat",*i);
248 if(r>=sizeof(s) || r<1) 248 if(r>=sizeof(s) || r<1)
249 continue; 249 continue;
250 string cmd; 250 string cmd;
251 ifstream ss(s,ios::in); 251 ifstream ss(s,ios::in);
252 if(!ss) 252 if(ss) {
253 continue; 253 getline(ss,cmd);
254 getline(ss,cmd); 254 string::size_type op = cmd.find('(');
255 string::size_type op = cmd.find('('); 255 if(op==string::npos)
256 if(op==string::npos) 256 continue;
257 continue; 257 cmd.erase(0,op+1);
258 cmd.erase(0,op+1); 258 string::size_type cp = cmd.find(')');
259 string::size_type cp = cmd.find(')'); 259 if(cp==string::npos)
260 if(cp==string::npos) 260 continue;
261 continue; 261 cmd.erase(cp);
262 cmd.erase(cp); 262 }else{
263 r = snprintf(s,sizeof(s),"/proc/%d/status",*i);
264 if(r>=sizeof(s) || r<1)
265 continue;
266 ifstream ss(s,ios::in);
267 if(!ss)
268 continue;
269 ss >> cmd;
270 if(cmd.empty())
271 continue;
272 }
263 r = snprintf(s,sizeof(s),"/proc/%d/cmdline",*i); 273 r = snprintf(s,sizeof(s),"/proc/%d/cmdline",*i);
264 if(r>=sizeof(s) || r<1) 274 if(r>=sizeof(s) || r<1)
265 continue; 275 continue;
266 ifstream cs(s,ios::binary); 276 ifstream cs(s,ios::binary);
267 if(!cs) 277 if(!cs)
268 continue; 278 continue;
269 string command; 279 string command;
270 while(cs) { 280 while(cs) {
271 string cl; 281 string cl;
272 getline(cs,cl,(char)0); 282 getline(cs,cl,(char)0);
273 string::size_type lsl = cl.rfind('/'); 283 string::size_type lsl = cl.rfind('/');
274 if(lsl!=string::npos) 284 if(lsl!=string::npos)
275 cl.erase(0,lsl+1); 285 cl.erase(0,lsl+1);
276 if(cl.substr(0,cmd.length())==cmd) { 286 if(cl.substr(0,cmd.length())==cmd) {
277 command = cl; 287 command = cl;
278 break; 288 break;
279 } 289 }
280 } 290 }
281 procpids.insert(pair<string,pid_t>(cmd,*i)); 291 procpids.insert(pair<string,pid_t>(cmd,*i));
282 if((!command.empty()) && cmd!=command) 292 if((!command.empty()) && cmd!=command)
283 procpids.insert(pair<string,pid_t>(command,*i)); 293 procpids.insert(pair<string,pid_t>(command,*i));
284 } 294 }
285} 295}