-rw-r--r-- | NEWS.xml | 1 | ||||
-rw-r--r-- | src/process.cc | 32 |
2 files changed, 22 insertions, 11 deletions
@@ -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 | ||
194 | void process::signal(int signum) const { | 194 | void 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 | ||
223 | void process::prepare_herd() { | 223 | void process::prepare_herd() { |
224 | procpids.clear(); | 224 | procpids.clear(); |
225 | } | 225 | } |
226 | void process::unprepare_herd() { | 226 | void process::unprepare_herd() { |
227 | procpids.clear(); | 227 | procpids.clear(); |
228 | } | 228 | } |
229 | void process::gather_proc_info() { | 229 | void 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 | } |