-rw-r--r-- | COPYING | 2 | ||||
-rw-r--r-- | NEWS.xml | 3 | ||||
-rw-r--r-- | configure.ac | 2 | ||||
-rw-r--r-- | src/process.cc | 14 |
4 files changed, 13 insertions, 8 deletions
@@ -1,19 +1,19 @@ | |||
1 | Copyright (c) 2004 Klever Group (http://www.klever.net/) | 1 | Copyright (c) 2004-2006 Klever Group (http://www.klever.net/) |
2 | 2 | ||
3 | Permission is hereby granted, free of charge, to any person obtaining a copy of | 3 | Permission is hereby granted, free of charge, to any person obtaining a copy of |
4 | this software and associated documentation files (the "Software"), to deal in | 4 | this software and associated documentation files (the "Software"), to deal in |
5 | the Software without restriction, including without limitation the rights to | 5 | the Software without restriction, including without limitation the rights to |
6 | use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies | 6 | use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies |
7 | of the Software, and to permit persons to whom the Software is furnished to do | 7 | of the Software, and to permit persons to whom the Software is furnished to do |
8 | so, subject to the following conditions: | 8 | so, subject to the following conditions: |
9 | 9 | ||
10 | The above copyright notice and this permission notice shall be included in all | 10 | The above copyright notice and this permission notice shall be included in all |
11 | copies or substantial portions of the Software. | 11 | copies or substantial portions of the Software. |
12 | 12 | ||
13 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | 13 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
14 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | 14 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
15 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | 15 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
16 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | 16 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
17 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | 17 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
18 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | 18 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
19 | SOFTWARE. | 19 | SOFTWARE. |
@@ -1,19 +1,22 @@ | |||
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.2" date="November 8th, 2006"> | ||
4 | <ni>Do not try to set unspecified group</ni> | ||
5 | </version> | ||
3 | <version version="0.2.1" date="August 1st, 2004"> | 6 | <version version="0.2.1" date="August 1st, 2004"> |
4 | <ni>do not act -e when doing -r</ni> | 7 | <ni>do not act -e when doing -r</ni> |
5 | <ni>proper handling of <kbd>ProcessName</kbd> on <kbd>FreeBSD</kbd></ni> | 8 | <ni>proper handling of <kbd>ProcessName</kbd> on <kbd>FreeBSD</kbd></ni> |
6 | </version> | 9 | </version> |
7 | <version version="0.2" date="July 24th, 2004"> | 10 | <version version="0.2" date="July 24th, 2004"> |
8 | <ni>now dudki sends arbitrary signals to the processes being monitored from the command line</ni> | 11 | <ni>now dudki sends arbitrary signals to the processes being monitored from the command line</ni> |
9 | <ni>detection of running processes which do not keep pidfiles, using process name</ni> | 12 | <ni>detection of running processes which do not keep pidfiles, using process name</ni> |
10 | </version> | 13 | </version> |
11 | <version version="0.1" date="July 21st, 2004"> | 14 | <version version="0.1" date="July 21st, 2004"> |
12 | <ni><kbd>initgroups()</kbd> called before executing <kbd>RestartCommand</kbd></ni> | 15 | <ni><kbd>initgroups()</kbd> called before executing <kbd>RestartCommand</kbd></ni> |
13 | <ni>more civilized way of restarting on <kbd>SIGHUP</kbd></ni> | 16 | <ni>more civilized way of restarting on <kbd>SIGHUP</kbd></ni> |
14 | <ni>minor changes to build process</ni> | 17 | <ni>minor changes to build process</ni> |
15 | </version> | 18 | </version> |
16 | <version version="0.0" date="July 11th, 2004"> | 19 | <version version="0.0" date="July 11th, 2004"> |
17 | <ni>Initial release</ni> | 20 | <ni>Initial release</ni> |
18 | </version> | 21 | </version> |
19 | </news> | 22 | </news> |
diff --git a/configure.ac b/configure.ac index eed8e97..a0e01e4 100644 --- a/configure.ac +++ b/configure.ac | |||
@@ -1,40 +1,40 @@ | |||
1 | AC_INIT([dudki], [0.2.1], [dudki-bugs@klever.net]) | 1 | AC_INIT([dudki], [0.2.2], [dudki-bugs@klever.net]) |
2 | AC_CONFIG_SRCDIR([src/dudki.cc]) | 2 | AC_CONFIG_SRCDIR([src/dudki.cc]) |
3 | AC_CONFIG_HEADER([config.h]) | 3 | AC_CONFIG_HEADER([config.h]) |
4 | AM_INIT_AUTOMAKE([dist-bzip2]) | 4 | AM_INIT_AUTOMAKE([dist-bzip2]) |
5 | 5 | ||
6 | AC_PROG_CXX | 6 | AC_PROG_CXX |
7 | AC_PROG_CC | 7 | AC_PROG_CC |
8 | 8 | ||
9 | AC_HEADER_SYS_WAIT | 9 | AC_HEADER_SYS_WAIT |
10 | AC_CHECK_HEADERS([syslog.h unistd.h getopt.h]) | 10 | AC_CHECK_HEADERS([syslog.h unistd.h getopt.h]) |
11 | 11 | ||
12 | AC_HEADER_STDBOOL | 12 | AC_HEADER_STDBOOL |
13 | AC_C_CONST | 13 | AC_C_CONST |
14 | AC_TYPE_UID_T | 14 | AC_TYPE_UID_T |
15 | AC_TYPE_PID_T | 15 | AC_TYPE_PID_T |
16 | 16 | ||
17 | AC_FUNC_FORK | 17 | AC_FUNC_FORK |
18 | AC_HEADER_STDC | 18 | AC_HEADER_STDC |
19 | AC_TYPE_SIGNAL | 19 | AC_TYPE_SIGNAL |
20 | AC_CHECK_FUNCS([dup2 memmove strtol]) | 20 | AC_CHECK_FUNCS([dup2 memmove strtol]) |
21 | AC_CHECK_FUNC([getopt_long],[ | 21 | AC_CHECK_FUNC([getopt_long],[ |
22 | AC_DEFINE([HAVE_GETOPT_LONG],[1],[Define to make use of getopt_long]) | 22 | AC_DEFINE([HAVE_GETOPT_LONG],[1],[Define to make use of getopt_long]) |
23 | AC_SUBST([HAVE_GETOPT_LONG],1) | 23 | AC_SUBST([HAVE_GETOPT_LONG],1) |
24 | ],[ | 24 | ],[ |
25 | AC_SUBST([HAVE_GETOPT_LONG],0) | 25 | AC_SUBST([HAVE_GETOPT_LONG],0) |
26 | ]) | 26 | ]) |
27 | 27 | ||
28 | AC_PATH_PROG([XSLTPROC],[xsltproc],[true]) | 28 | AC_PATH_PROG([XSLTPROC],[xsltproc],[true]) |
29 | 29 | ||
30 | PKG_CHECK_MODULES([DOTCONF],[dotconf],,[ | 30 | PKG_CHECK_MODULES([DOTCONF],[dotconf],,[ |
31 | AC_MSG_ERROR([no dotconf library found]) | 31 | AC_MSG_ERROR([no dotconf library found]) |
32 | ]) | 32 | ]) |
33 | 33 | ||
34 | AC_CONFIG_FILES([ | 34 | AC_CONFIG_FILES([ |
35 | Makefile | 35 | Makefile |
36 | src/Makefile | 36 | src/Makefile |
37 | man/Makefile | 37 | man/Makefile |
38 | man/dudki.8 man/dudki.conf.5 | 38 | man/dudki.8 man/dudki.conf.5 |
39 | ]) | 39 | ]) |
40 | AC_OUTPUT | 40 | AC_OUTPUT |
diff --git a/src/process.cc b/src/process.cc index 96c874f..6d3b2a2 100644 --- a/src/process.cc +++ b/src/process.cc | |||
@@ -1,295 +1,297 @@ | |||
1 | #include <stdio.h> | 1 | #include <stdio.h> |
2 | #include <sys/types.h> | 2 | #include <sys/types.h> |
3 | #include <unistd.h> | 3 | #include <unistd.h> |
4 | #include <signal.h> | 4 | #include <signal.h> |
5 | #include <pwd.h> | 5 | #include <pwd.h> |
6 | #include <grp.h> | 6 | #include <grp.h> |
7 | #include <dirent.h> | 7 | #include <dirent.h> |
8 | #include <sys/wait.h> | 8 | #include <sys/wait.h> |
9 | #include <syslog.h> | 9 | #include <syslog.h> |
10 | #include <errno.h> | 10 | #include <errno.h> |
11 | #include <iostream> | 11 | #include <iostream> |
12 | #include <fstream> | 12 | #include <fstream> |
13 | #include <sstream> | 13 | #include <sstream> |
14 | #include <stdexcept> | 14 | #include <stdexcept> |
15 | #include <string> | 15 | #include <string> |
16 | #include <vector> | 16 | #include <vector> |
17 | using namespace std; | 17 | using namespace std; |
18 | #include "process.h" | 18 | #include "process.h" |
19 | #include "configuration.h" | 19 | #include "configuration.h" |
20 | 20 | ||
21 | static multimap<string,pid_t> procpids; | 21 | static multimap<string,pid_t> procpids; |
22 | 22 | ||
23 | void process::check() const { | 23 | void process::check() const { |
24 | if(!pidfile.empty()) { | 24 | if(!pidfile.empty()) { |
25 | signal(0); | 25 | signal(0); |
26 | }else if(!process_name.empty()) { | 26 | }else if(!process_name.empty()) { |
27 | if(procpids.empty()) | 27 | if(procpids.empty()) |
28 | gather_proc_info(); | 28 | gather_proc_info(); |
29 | if(procpids.find(process_name)==procpids.end()) | 29 | if(procpids.find(process_name)==procpids.end()) |
30 | throw runtime_error("no such process"); | 30 | throw runtime_error("no such process"); |
31 | } // XXX: or else? | 31 | } // XXX: or else? |
32 | } | 32 | } |
33 | void process::check(const string& id,configuration& config) { | 33 | void process::check(const string& id,configuration& config) { |
34 | try { | 34 | try { |
35 | check(); | 35 | check(); |
36 | patience = 0; | 36 | patience = 0; |
37 | }catch(exception& e) { | 37 | }catch(exception& e) { |
38 | if(patience>60) { // TODO: configurable | 38 | if(patience>60) { // TODO: configurable |
39 | patience = 0; | 39 | patience = 0; |
40 | }else{ | 40 | }else{ |
41 | if(patience<10) { // TODO: configurable | 41 | if(patience<10) { // TODO: configurable |
42 | syslog(LOG_NOTICE,"The process '%s' is down, trying to launch.",id.c_str()); | 42 | syslog(LOG_NOTICE,"The process '%s' is down, trying to launch.",id.c_str()); |
43 | do_notify(id,"Starting up", | 43 | do_notify(id,"Starting up", |
44 | "The named process seems to be down. Dudki will try\n" | 44 | "The named process seems to be down. Dudki will try\n" |
45 | "to revive it by running the specified command.\n", | 45 | "to revive it by running the specified command.\n", |
46 | config); | 46 | config); |
47 | try { | 47 | try { |
48 | launch(id,config); | 48 | launch(id,config); |
49 | }catch(exception& e) { | 49 | }catch(exception& e) { |
50 | syslog(LOG_ERR,"Error trying to launch process '%s': %s",id.c_str(),e.what()); | 50 | syslog(LOG_ERR,"Error trying to launch process '%s': %s",id.c_str(),e.what()); |
51 | } | 51 | } |
52 | }else if(patience==10){ // TODO: configurable like the above | 52 | }else if(patience==10){ // TODO: configurable like the above |
53 | syslog(LOG_NOTICE,"Giving up on process '%s' for a while",id.c_str()); | 53 | syslog(LOG_NOTICE,"Giving up on process '%s' for a while",id.c_str()); |
54 | do_notify(id,"Giving up", | 54 | do_notify(id,"Giving up", |
55 | "After a number of attempts to relaunch the named process\n" | 55 | "After a number of attempts to relaunch the named process\n" |
56 | "It still seems to be down. Dudki is giving up attempts\n" | 56 | "It still seems to be down. Dudki is giving up attempts\n" |
57 | "to revive the process for a while.\n", | 57 | "to revive the process for a while.\n", |
58 | config); | 58 | config); |
59 | } | 59 | } |
60 | patience++; | 60 | patience++; |
61 | } | 61 | } |
62 | } | 62 | } |
63 | } | 63 | } |
64 | 64 | ||
65 | void process::launch(const string& id,configuration& config) { | 65 | void process::launch(const string& id,configuration& config) { |
66 | uid_t uid = (uid_t)-1; | 66 | uid_t uid = (uid_t)-1; |
67 | gid_t gid = (gid_t)-1; | 67 | gid_t gid = (gid_t)-1; |
68 | if(!user.empty()) { | 68 | if(!user.empty()) { |
69 | struct passwd *ptmp = getpwnam(user.c_str()); | 69 | struct passwd *ptmp = getpwnam(user.c_str()); |
70 | if(ptmp) { | 70 | if(ptmp) { |
71 | uid = ptmp->pw_uid; | 71 | uid = ptmp->pw_uid; |
72 | gid = ptmp->pw_gid; | 72 | gid = ptmp->pw_gid; |
73 | }else{ | 73 | }else{ |
74 | errno=0; | 74 | errno=0; |
75 | uid = strtol(user.c_str(),NULL,0); | 75 | uid = strtol(user.c_str(),NULL,0); |
76 | if(errno) | 76 | if(errno) |
77 | throw runtime_error("Failed to resolve User value to uid"); | 77 | throw runtime_error("Failed to resolve User value to uid"); |
78 | } | 78 | } |
79 | } | 79 | } |
80 | if(!group.empty()) { | 80 | if(!group.empty()) { |
81 | struct group *gtmp = getgrnam(group.c_str()); | 81 | struct group *gtmp = getgrnam(group.c_str()); |
82 | if(gtmp) { | 82 | if(gtmp) { |
83 | gid = gtmp->gr_gid; | 83 | gid = gtmp->gr_gid; |
84 | }else{ | 84 | }else{ |
85 | errno = 0; | 85 | errno = 0; |
86 | gid = strtol(group.c_str(),NULL,0); | 86 | gid = strtol(group.c_str(),NULL,0); |
87 | if(errno) | 87 | if(errno) |
88 | throw runtime_error("Failed to reslove Group value to gid"); | 88 | throw runtime_error("Failed to reslove Group value to gid"); |
89 | } | 89 | } |
90 | } | 90 | } |
91 | pid_t p = fork(); | 91 | pid_t p = fork(); |
92 | if(p<0) | 92 | if(p<0) |
93 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to fork()"); | 93 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to fork()"); |
94 | if(!p) { | 94 | if(!p) { |
95 | // child | 95 | // child |
96 | try { | 96 | try { |
97 | setsid(); | 97 | setsid(); |
98 | if(user.empty()) { | 98 | if(!group.empty()) { |
99 | if((getgid()!=gid) && setgid(gid)) | 99 | if(user.empty()) { |
100 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to setgid()"); | 100 | if((getgid()!=gid) && setgid(gid)) |
101 | }else{ | 101 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to setgid()"); |
102 | if(initgroups(user.c_str(),gid)) | 102 | }else{ |
103 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to initgroups()"); | 103 | if(initgroups(user.c_str(),gid)) |
104 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to initgroups()"); | ||
105 | } | ||
104 | } | 106 | } |
105 | if(!chroot.empty()) { | 107 | if(!chroot.empty()) { |
106 | if(::chroot(chroot.c_str())) | 108 | if(::chroot(chroot.c_str())) |
107 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to chroot()"); | 109 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to chroot()"); |
108 | } | 110 | } |
109 | if(!user.empty()) { | 111 | if(!user.empty()) { |
110 | if((getuid()!=uid) && setuid(uid)) | 112 | if((getuid()!=uid) && setuid(uid)) |
111 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to setuid()"); | 113 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to setuid()"); |
112 | } | 114 | } |
113 | char *argv[] = { "/bin/sh", "-c", (char*)restart_cmd.c_str(), NULL }; | 115 | char *argv[] = { "/bin/sh", "-c", (char*)restart_cmd.c_str(), NULL }; |
114 | close(0); close(1); close(2); | 116 | close(0); close(1); close(2); |
115 | execv("/bin/sh",argv); | 117 | execv("/bin/sh",argv); |
116 | }catch(exception& e) { | 118 | }catch(exception& e) { |
117 | syslog(LOG_ERR,"Error trying to launch process '%s': %s",id.c_str(),e.what()); | 119 | syslog(LOG_ERR,"Error trying to launch process '%s': %s",id.c_str(),e.what()); |
118 | } | 120 | } |
119 | _exit(-1); | 121 | _exit(-1); |
120 | } | 122 | } |
121 | // parent | 123 | // parent |
122 | int rv; | 124 | int rv; |
123 | if(waitpid(p,&rv,0)<0) | 125 | if(waitpid(p,&rv,0)<0) |
124 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to waitpid()"); | 126 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to waitpid()"); |
125 | } | 127 | } |
126 | 128 | ||
127 | void process::do_notify(const string& id,const string& event,const string& description,configuration& config) { | 129 | void process::do_notify(const string& id,const string& event,const string& description,configuration& config) { |
128 | string the_notify; | 130 | string the_notify; |
129 | if(!notify.empty()) | 131 | if(!notify.empty()) |
130 | the_notify=notify; | 132 | the_notify=notify; |
131 | else if(!config.notify.empty()) | 133 | else if(!config.notify.empty()) |
132 | the_notify=config.notify; | 134 | the_notify=config.notify; |
133 | else | 135 | else |
134 | return; | 136 | return; |
135 | try { | 137 | try { |
136 | string::size_type colon = the_notify.find(':'); | 138 | string::size_type colon = the_notify.find(':'); |
137 | if(colon==string::npos) | 139 | if(colon==string::npos) |
138 | throw runtime_error("invalid notify action specification"); | 140 | throw runtime_error("invalid notify action specification"); |
139 | string nschema = the_notify.substr(0,colon); | 141 | string nschema = the_notify.substr(0,colon); |
140 | string ntarget = the_notify.substr(colon+1); | 142 | string ntarget = the_notify.substr(colon+1); |
141 | if(nschema=="mailto") { | 143 | if(nschema=="mailto") { |
142 | notify_mailto(ntarget,id,event,description,config); | 144 | notify_mailto(ntarget,id,event,description,config); |
143 | }else | 145 | }else |
144 | throw runtime_error("unrecognized notification schema"); | 146 | throw runtime_error("unrecognized notification schema"); |
145 | }catch(exception& e) { | 147 | }catch(exception& e) { |
146 | syslog(LOG_ERR,"Notification error: %s",e.what()); | 148 | syslog(LOG_ERR,"Notification error: %s",e.what()); |
147 | } | 149 | } |
148 | } | 150 | } |
149 | 151 | ||
150 | void process::notify_mailto(const string& email,const string& id,const string& event,const string& description,configuration& config) { | 152 | void process::notify_mailto(const string& email,const string& id,const string& event,const string& description,configuration& config) { |
151 | int files[2]; | 153 | int files[2]; |
152 | if(pipe(files)) | 154 | if(pipe(files)) |
153 | throw runtime_error("Failed to pipe()"); | 155 | throw runtime_error("Failed to pipe()"); |
154 | pid_t pid = vfork(); | 156 | pid_t pid = vfork(); |
155 | if(pid==-1) { | 157 | if(pid==-1) { |
156 | close(files[0]); | 158 | close(files[0]); |
157 | close(files[1]); | 159 | close(files[1]); |
158 | throw runtime_error("Failed to vfork()"); | 160 | throw runtime_error("Failed to vfork()"); |
159 | } | 161 | } |
160 | if(!pid) { | 162 | if(!pid) { |
161 | // child | 163 | // child |
162 | if(dup2(files[0],0)!=0) | 164 | if(dup2(files[0],0)!=0) |
163 | _exit(-1); | 165 | _exit(-1); |
164 | close(1); | 166 | close(1); |
165 | close(files[0]); | 167 | close(files[0]); |
166 | close(files[1]); | 168 | close(files[1]); |
167 | execl("/usr/sbin/sendmail","usr/sbin/sendmail","-i",email.c_str(),NULL); | 169 | execl("/usr/sbin/sendmail","usr/sbin/sendmail","-i",email.c_str(),NULL); |
168 | _exit(-1); | 170 | _exit(-1); |
169 | } | 171 | } |
170 | // parent | 172 | // parent |
171 | close(files[0]); | 173 | close(files[0]); |
172 | FILE *mta = fdopen(files[1],"w"); | 174 | FILE *mta = fdopen(files[1],"w"); |
173 | for(headers_t::const_iterator i=mailto_headers.begin();i!=mailto_headers.end();++i) { | 175 | 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()); | 176 | fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str()); |
175 | } | 177 | } |
176 | for(headers_t::const_iterator i=config.mailto_headers.begin();i!=config.mailto_headers.end();++i) { | 178 | 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()) | 179 | if(mailto_headers.find(i->first)!=mailto_headers.end()) |
178 | continue; | 180 | continue; |
179 | fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str()); | 181 | fprintf(mta,"%s: %s\n",i->first.c_str(),i->second.c_str()); |
180 | } | 182 | } |
181 | fprintf(mta, | 183 | fprintf(mta, |
182 | "Subject: [%s] %s\n\n" | 184 | "Subject: [%s] %s\n\n" |
183 | "%s\n" | 185 | "%s\n" |
184 | "---\n" | 186 | "---\n" |
185 | "This message was sent automatically by the 'dudki' daemon\n", | 187 | "This message was sent automatically by the 'dudki' daemon\n", |
186 | id.c_str(), event.c_str(), | 188 | id.c_str(), event.c_str(), |
187 | description.c_str() ); | 189 | description.c_str() ); |
188 | fclose(mta); | 190 | fclose(mta); |
189 | int status; | 191 | int status; |
190 | waitpid(pid,&status,0); | 192 | waitpid(pid,&status,0); |
191 | // TODO: check the return code | 193 | // TODO: check the return code |
192 | } | 194 | } |
193 | 195 | ||
194 | void process::signal(int signum) const { | 196 | void process::signal(int signum) const { |
195 | if(!pidfile.empty()) { | 197 | if(!pidfile.empty()) { |
196 | ifstream pids(pidfile.c_str(),ios::in); | 198 | ifstream pids(pidfile.c_str(),ios::in); |
197 | if(!pids) | 199 | if(!pids) |
198 | throw runtime_error("no pidfile found"); | 200 | throw runtime_error("no pidfile found"); |
199 | pid_t pid = 0; | 201 | pid_t pid = 0; |
200 | pids >> pid; | 202 | pids >> pid; |
201 | pids.close(); | 203 | pids.close(); |
202 | if(!pid) | 204 | if(!pid) |
203 | throw runtime_error("no pid in pidfile"); | 205 | throw runtime_error("no pid in pidfile"); |
204 | if(kill(pid,signum)) | 206 | if(kill(pid,signum)) |
205 | throw runtime_error("failed to signal process"); | 207 | throw runtime_error("failed to signal process"); |
206 | }else if(!process_name.empty()) { | 208 | }else if(!process_name.empty()) { |
207 | if(procpids.empty()) | 209 | if(procpids.empty()) |
208 | gather_proc_info(); | 210 | gather_proc_info(); |
209 | pair<multimap<string,pid_t>::const_iterator,multimap<string,pid_t>::const_iterator> range = procpids.equal_range(process_name); | 211 | pair<multimap<string,pid_t>::const_iterator,multimap<string,pid_t>::const_iterator> range = procpids.equal_range(process_name); |
210 | int count = 0; | 212 | int count = 0; |
211 | for(multimap<string,pid_t>::const_iterator i=range.first;i!=range.second;++i) { | 213 | for(multimap<string,pid_t>::const_iterator i=range.first;i!=range.second;++i) { |
212 | pid_t pid = i->second; | 214 | pid_t pid = i->second; |
213 | if(kill(i->second,signum)) | 215 | if(kill(i->second,signum)) |
214 | throw runtime_error("failed to signal process"); | 216 | throw runtime_error("failed to signal process"); |
215 | count++; | 217 | count++; |
216 | } | 218 | } |
217 | if(!count) | 219 | if(!count) |
218 | throw runtime_error("no running instance detected"); | 220 | throw runtime_error("no running instance detected"); |
219 | }else | 221 | }else |
220 | throw runtime_error("nothing is known about the process"); | 222 | throw runtime_error("nothing is known about the process"); |
221 | } | 223 | } |
222 | 224 | ||
223 | void process::prepare_herd() { | 225 | void process::prepare_herd() { |
224 | procpids.clear(); | 226 | procpids.clear(); |
225 | } | 227 | } |
226 | void process::unprepare_herd() { | 228 | void process::unprepare_herd() { |
227 | procpids.clear(); | 229 | procpids.clear(); |
228 | } | 230 | } |
229 | void process::gather_proc_info() { | 231 | void process::gather_proc_info() { |
230 | vector<pid_t> allpids; | 232 | vector<pid_t> allpids; |
231 | DIR *pd = opendir("/proc"); | 233 | DIR *pd = opendir("/proc"); |
232 | if(!pd) | 234 | if(!pd) |
233 | throw runtime_error("failed to open /proc"); | 235 | throw runtime_error("failed to open /proc"); |
234 | struct dirent *pde; | 236 | struct dirent *pde; |
235 | pid_t selfpid = getpid(); | 237 | pid_t selfpid = getpid(); |
236 | while(pde=readdir(pd)) { | 238 | while(pde=readdir(pd)) { |
237 | errno=0; | 239 | errno=0; |
238 | pid_t pid = atoi(pde->d_name); | 240 | pid_t pid = atoi(pde->d_name); |
239 | if((!pid) || pid==selfpid) | 241 | if((!pid) || pid==selfpid) |
240 | continue; | 242 | continue; |
241 | allpids.push_back(pid); | 243 | allpids.push_back(pid); |
242 | } | 244 | } |
243 | closedir(pd); | 245 | closedir(pd); |
244 | char s[256]; | 246 | char s[256]; |
245 | procpids.clear(); | 247 | procpids.clear(); |
246 | for(vector<pid_t>::const_iterator i=allpids.begin();i!=allpids.end();++i) { | 248 | for(vector<pid_t>::const_iterator i=allpids.begin();i!=allpids.end();++i) { |
247 | int r = snprintf(s,sizeof(s),"/proc/%d/stat",*i); | 249 | int r = snprintf(s,sizeof(s),"/proc/%d/stat",*i); |
248 | if(r>=sizeof(s) || r<1) | 250 | if(r>=sizeof(s) || r<1) |
249 | continue; | 251 | continue; |
250 | string cmd; | 252 | string cmd; |
251 | ifstream ss(s,ios::in); | 253 | ifstream ss(s,ios::in); |
252 | if(ss) { | 254 | if(ss) { |
253 | getline(ss,cmd); | 255 | getline(ss,cmd); |
254 | string::size_type op = cmd.find('('); | 256 | string::size_type op = cmd.find('('); |
255 | if(op==string::npos) | 257 | if(op==string::npos) |
256 | continue; | 258 | continue; |
257 | cmd.erase(0,op+1); | 259 | cmd.erase(0,op+1); |
258 | string::size_type cp = cmd.find(')'); | 260 | string::size_type cp = cmd.find(')'); |
259 | if(cp==string::npos) | 261 | if(cp==string::npos) |
260 | continue; | 262 | continue; |
261 | cmd.erase(cp); | 263 | cmd.erase(cp); |
262 | }else{ | 264 | }else{ |
263 | r = snprintf(s,sizeof(s),"/proc/%d/status",*i); | 265 | r = snprintf(s,sizeof(s),"/proc/%d/status",*i); |
264 | if(r>=sizeof(s) || r<1) | 266 | if(r>=sizeof(s) || r<1) |
265 | continue; | 267 | continue; |
266 | ifstream ss(s,ios::in); | 268 | ifstream ss(s,ios::in); |
267 | if(!ss) | 269 | if(!ss) |
268 | continue; | 270 | continue; |
269 | ss >> cmd; | 271 | ss >> cmd; |
270 | if(cmd.empty()) | 272 | if(cmd.empty()) |
271 | continue; | 273 | continue; |
272 | } | 274 | } |
273 | r = snprintf(s,sizeof(s),"/proc/%d/cmdline",*i); | 275 | r = snprintf(s,sizeof(s),"/proc/%d/cmdline",*i); |
274 | if(r>=sizeof(s) || r<1) | 276 | if(r>=sizeof(s) || r<1) |
275 | continue; | 277 | continue; |
276 | ifstream cs(s,ios::binary); | 278 | ifstream cs(s,ios::binary); |
277 | if(!cs) | 279 | if(!cs) |
278 | continue; | 280 | continue; |
279 | string command; | 281 | string command; |
280 | while(cs) { | 282 | while(cs) { |
281 | string cl; | 283 | string cl; |
282 | getline(cs,cl,(char)0); | 284 | getline(cs,cl,(char)0); |
283 | string::size_type lsl = cl.rfind('/'); | 285 | string::size_type lsl = cl.rfind('/'); |
284 | if(lsl!=string::npos) | 286 | if(lsl!=string::npos) |
285 | cl.erase(0,lsl+1); | 287 | cl.erase(0,lsl+1); |
286 | if(cl.substr(0,cmd.length())==cmd) { | 288 | if(cl.substr(0,cmd.length())==cmd) { |
287 | command = cl; | 289 | command = cl; |
288 | break; | 290 | break; |
289 | } | 291 | } |
290 | } | 292 | } |
291 | procpids.insert(pair<string,pid_t>(cmd,*i)); | 293 | procpids.insert(pair<string,pid_t>(cmd,*i)); |
292 | if((!command.empty()) && cmd!=command) | 294 | if((!command.empty()) && cmd!=command) |
293 | procpids.insert(pair<string,pid_t>(command,*i)); | 295 | procpids.insert(pair<string,pid_t>(command,*i)); |
294 | } | 296 | } |
295 | } | 297 | } |