-rw-r--r-- | libopie/oprocctrl.cpp | 268 | ||||
-rw-r--r-- | libopie/oprocess.cpp | 922 |
2 files changed, 1190 insertions, 0 deletions
diff --git a/libopie/oprocctrl.cpp b/libopie/oprocctrl.cpp new file mode 100644 index 0000000..e7db622 --- a/dev/null +++ b/libopie/oprocctrl.cpp | |||
@@ -0,0 +1,268 @@ | |||
1 | /* This file is part of the KDE libraries | ||
2 | Copyright (C) 1997 Christian Czezakte (e9025461@student.tuwien.ac.at) | ||
3 | |||
4 | This library is free software; you can redistribute it and/or | ||
5 | modify it under the terms of the GNU Library General Public | ||
6 | License as published by the Free Software Foundation; either | ||
7 | version 2 of the License, or (at your option) any later version. | ||
8 | |||
9 | This library is distributed in the hope that it will be useful, | ||
10 | but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
12 | Library General Public License for more details. | ||
13 | |||
14 | You should have received a copy of the GNU Library General Public License | ||
15 | along with this library; see the file COPYING.LIB. If not, write to | ||
16 | the Free Software Foundation, Inc., 59 Temple Place - Suite 330, | ||
17 | Boston, MA 02111-1307, USA. | ||
18 | */ | ||
19 | // | ||
20 | // KPROCESSCONTROLLER -- A helper class for KProcess | ||
21 | // | ||
22 | // version 0.3.1, Jan, 8th 1997 | ||
23 | // | ||
24 | // (C) Christian Czezatke | ||
25 | // e9025461@student.tuwien.ac.at | ||
26 | // Ported by Holger Freyther | ||
27 | // | ||
28 | |||
29 | //#include <config.h> | ||
30 | |||
31 | #include <sys/types.h> | ||
32 | #include <sys/socket.h> | ||
33 | |||
34 | #include <errno.h> | ||
35 | #include <fcntl.h> | ||
36 | #include <stdio.h> | ||
37 | #include <string.h> | ||
38 | #include <unistd.h> | ||
39 | #include <assert.h> | ||
40 | |||
41 | #include <qsocketnotifier.h> | ||
42 | #include "oprocess.h" | ||
43 | #include "oprocctrl.h" | ||
44 | |||
45 | OProcessController *OProcessController::theOProcessController = 0; | ||
46 | |||
47 | struct sigaction OProcessController::oldChildHandlerData; | ||
48 | bool OProcessController::handlerSet = false; | ||
49 | |||
50 | OProcessController::OProcessController() | ||
51 | { | ||
52 | assert( theOProcessController == 0 ); | ||
53 | |||
54 | if (0 > pipe(fd)) | ||
55 | printf(strerror(errno)); | ||
56 | |||
57 | notifier = new QSocketNotifier(fd[0], QSocketNotifier::Read); | ||
58 | notifier->setEnabled(true); | ||
59 | QObject::connect(notifier, SIGNAL(activated(int)), | ||
60 | this, SLOT(slotDoHousekeeping(int))); | ||
61 | connect( &delayedChildrenCleanupTimer, SIGNAL( timeout()), | ||
62 | SLOT( delayedChildrenCleanup())); | ||
63 | |||
64 | theOProcessController = this; | ||
65 | |||
66 | setupHandlers(); | ||
67 | } | ||
68 | |||
69 | |||
70 | void OProcessController::setupHandlers() | ||
71 | { | ||
72 | if( handlerSet ) | ||
73 | return; | ||
74 | struct sigaction act; | ||
75 | act.sa_handler=theSigCHLDHandler; | ||
76 | sigemptyset(&(act.sa_mask)); | ||
77 | sigaddset(&(act.sa_mask), SIGCHLD); | ||
78 | // Make sure we don't block this signal. gdb tends to do that :-( | ||
79 | sigprocmask(SIG_UNBLOCK, &(act.sa_mask), 0); | ||
80 | |||
81 | act.sa_flags = SA_NOCLDSTOP; | ||
82 | |||
83 | // CC: take care of SunOS which automatically restarts interrupted system | ||
84 | // calls (and thus does not have SA_RESTART) | ||
85 | |||
86 | #ifdef SA_RESTART | ||
87 | act.sa_flags |= SA_RESTART; | ||
88 | #endif | ||
89 | |||
90 | sigaction( SIGCHLD, &act, &oldChildHandlerData ); | ||
91 | |||
92 | act.sa_handler=SIG_IGN; | ||
93 | sigemptyset(&(act.sa_mask)); | ||
94 | sigaddset(&(act.sa_mask), SIGPIPE); | ||
95 | act.sa_flags = 0; | ||
96 | sigaction( SIGPIPE, &act, 0L); | ||
97 | handlerSet = true; | ||
98 | } | ||
99 | |||
100 | void OProcessController::resetHandlers() | ||
101 | { | ||
102 | if( !handlerSet ) | ||
103 | return; | ||
104 | sigaction( SIGCHLD, &oldChildHandlerData, 0 ); | ||
105 | // there should be no problem with SIGPIPE staying SIG_IGN | ||
106 | handlerSet = false; | ||
107 | } | ||
108 | |||
109 | // block SIGCHLD handler, because it accesses processList | ||
110 | void OProcessController::addOProcess( OProcess* p ) | ||
111 | { | ||
112 | sigset_t newset, oldset; | ||
113 | sigemptyset( &newset ); | ||
114 | sigaddset( &newset, SIGCHLD ); | ||
115 | sigprocmask( SIG_BLOCK, &newset, &oldset ); | ||
116 | processList.append( p ); | ||
117 | sigprocmask( SIG_SETMASK, &oldset, 0 ); | ||
118 | } | ||
119 | |||
120 | void OProcessController::removeOProcess( OProcess* p ) | ||
121 | { | ||
122 | sigset_t newset, oldset; | ||
123 | sigemptyset( &newset ); | ||
124 | sigaddset( &newset, SIGCHLD ); | ||
125 | sigprocmask( SIG_BLOCK, &newset, &oldset ); | ||
126 | processList.remove( p ); | ||
127 | sigprocmask( SIG_SETMASK, &oldset, 0 ); | ||
128 | } | ||
129 | |||
130 | //using a struct which contains both the pid and the status makes it easier to write | ||
131 | //and read the data into the pipe | ||
132 | //especially this solves a problem which appeared on my box where slotDoHouseKeeping() received | ||
133 | //only 4 bytes (with some debug output around the write()'s it received all 8 bytes) | ||
134 | //don't know why this happened, but when writing all 8 bytes at once it works here, aleXXX | ||
135 | struct waitdata | ||
136 | { | ||
137 | pid_t pid; | ||
138 | int status; | ||
139 | }; | ||
140 | |||
141 | void OProcessController::theSigCHLDHandler(int arg) | ||
142 | { | ||
143 | struct waitdata wd; | ||
144 | // int status; | ||
145 | // pid_t this_pid; | ||
146 | int saved_errno; | ||
147 | |||
148 | saved_errno = errno; | ||
149 | // since waitpid and write change errno, we have to save it and restore it | ||
150 | // (Richard Stevens, Advanced programming in the Unix Environment) | ||
151 | |||
152 | bool found = false; | ||
153 | if( theOProcessController != 0 ) { | ||
154 | // iterating the list doesn't perform any system call | ||
155 | for( QValueList<OProcess*>::ConstIterator it = theOProcessController->processList.begin(); | ||
156 | it != theOProcessController->processList.end(); | ||
157 | ++it ) | ||
158 | { | ||
159 | if( !(*it)->isRunning()) | ||
160 | continue; | ||
161 | wd.pid = waitpid( (*it)->pid(), &wd.status, WNOHANG ); | ||
162 | if ( wd.pid > 0 ) { | ||
163 | ::write(theOProcessController->fd[1], &wd, sizeof(wd)); | ||
164 | found = true; | ||
165 | } | ||
166 | } | ||
167 | } | ||
168 | if( !found && oldChildHandlerData.sa_handler != SIG_IGN | ||
169 | && oldChildHandlerData.sa_handler != SIG_DFL ) | ||
170 | oldChildHandlerData.sa_handler( arg ); // call the old handler | ||
171 | // handle the rest | ||
172 | if( theOProcessController != 0 ) { | ||
173 | static const struct waitdata dwd = { 0, 0 }; // delayed waitpid() | ||
174 | ::write(theOProcessController->fd[1], &dwd, sizeof(dwd)); | ||
175 | } else { | ||
176 | int dummy; | ||
177 | while( waitpid( -1, &dummy, WNOHANG ) > 0 ) | ||
178 | ; | ||
179 | } | ||
180 | |||
181 | errno = saved_errno; | ||
182 | } | ||
183 | |||
184 | |||
185 | |||
186 | void OProcessController::slotDoHousekeeping(int ) | ||
187 | { | ||
188 | unsigned int bytes_read = 0; | ||
189 | unsigned int errcnt=0; | ||
190 | // read pid and status from the pipe. | ||
191 | struct waitdata wd; | ||
192 | while ((bytes_read < sizeof(wd)) && (errcnt < 50)) { | ||
193 | int r = ::read(fd[0], ((char *)&wd) + bytes_read, sizeof(wd) - bytes_read); | ||
194 | if (r > 0) bytes_read += r; | ||
195 | else if (r < 0) errcnt++; | ||
196 | } | ||
197 | if (errcnt >= 50) { | ||
198 | fprintf(stderr, | ||
199 | "Error: Max. error count for pipe read " | ||
200 | "exceeded in OProcessController::slotDoHousekeeping\n"); | ||
201 | return; // it makes no sense to continue here! | ||
202 | } | ||
203 | if (bytes_read != sizeof(wd)) { | ||
204 | fprintf(stderr, | ||
205 | "Error: Could not read info from signal handler %d <> %d!\n", | ||
206 | bytes_read, sizeof(wd)); | ||
207 | return; // it makes no sense to continue here! | ||
208 | } | ||
209 | if (wd.pid==0) { // special case, see delayedChildrenCleanup() | ||
210 | delayedChildrenCleanupTimer.start( 1000, true ); | ||
211 | return; | ||
212 | } | ||
213 | |||
214 | for( QValueList<OProcess*>::ConstIterator it = processList.begin(); | ||
215 | it != processList.end(); | ||
216 | ++it ) { | ||
217 | OProcess* proc = *it; | ||
218 | if (proc->pid() == wd.pid) { | ||
219 | // process has exited, so do emit the respective events | ||
220 | if (proc->run_mode == OProcess::Block) { | ||
221 | // If the reads are done blocking then set the status in proc | ||
222 | // but do nothing else because OProcess will perform the other | ||
223 | // actions of processHasExited. | ||
224 | proc->status = wd.status; | ||
225 | proc->runs = false; | ||
226 | } else { | ||
227 | proc->processHasExited(wd.status); | ||
228 | } | ||
229 | return; | ||
230 | } | ||
231 | } | ||
232 | } | ||
233 | |||
234 | // this is needed e.g. for popen(), which calls waitpid() checking | ||
235 | // for its forked child, if we did waitpid() directly in the SIGCHLD | ||
236 | // handler, popen()'s waitpid() call would fail | ||
237 | void OProcessController::delayedChildrenCleanup() | ||
238 | { | ||
239 | struct waitdata wd; | ||
240 | while(( wd.pid = waitpid( -1, &wd.status, WNOHANG ) ) > 0 ) { | ||
241 | for( QValueList<OProcess*>::ConstIterator it = processList.begin(); | ||
242 | it != processList.end(); | ||
243 | ++it ) | ||
244 | { | ||
245 | if( !(*it)->isRunning() || (*it)->pid() != wd.pid ) | ||
246 | continue; | ||
247 | // it's OProcess, handle it | ||
248 | ::write(fd[1], &wd, sizeof(wd)); | ||
249 | break; | ||
250 | } | ||
251 | } | ||
252 | } | ||
253 | |||
254 | OProcessController::~OProcessController() | ||
255 | { | ||
256 | assert( theOProcessController == this ); | ||
257 | resetHandlers(); | ||
258 | |||
259 | notifier->setEnabled(false); | ||
260 | |||
261 | close(fd[0]); | ||
262 | close(fd[1]); | ||
263 | |||
264 | delete notifier; | ||
265 | theOProcessController = 0; | ||
266 | } | ||
267 | |||
268 | //#include "kprocctrl.moc" | ||
diff --git a/libopie/oprocess.cpp b/libopie/oprocess.cpp new file mode 100644 index 0000000..f3e52bd --- a/dev/null +++ b/libopie/oprocess.cpp | |||
@@ -0,0 +1,922 @@ | |||
1 | /* | ||
2 | |||
3 | $Id$ | ||
4 | |||
5 | This file is part of the KDE libraries | ||
6 | Copyright (C) 1997 Christian Czezatke (e9025461@student.tuwien.ac.at) | ||
7 | |||
8 | This library is free software; you can redistribute it and/or | ||
9 | modify it under the terms of the GNU Library General Public | ||
10 | License as published by the Free Software Foundation; either | ||
11 | version 2 of the License, or (at your option) any later version. | ||
12 | |||
13 | This library is distributed in the hope that it will be useful, | ||
14 | but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
15 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
16 | Library General Public License for more details. | ||
17 | |||
18 | You should have received a copy of the GNU Library General Public License | ||
19 | along with this library; see the file COPYING.LIB. If not, write to | ||
20 | the Free Software Foundation, Inc., 59 Temple Place - Suite 330, | ||
21 | Boston, MA 02111-1307, USA. | ||
22 | |||
23 | */ | ||
24 | |||
25 | |||
26 | // | ||
27 | // KPROCESS -- A class for handling child processes in KDE without | ||
28 | // having to take care of Un*x specific implementation details | ||
29 | // | ||
30 | // version 0.3.1, Jan 8th 1998 | ||
31 | // | ||
32 | // (C) Christian Czezatke | ||
33 | // e9025461@student.tuwien.ac.at | ||
34 | // | ||
35 | // Changes: | ||
36 | // | ||
37 | // March 2nd, 1998: Changed parameter list for KShellProcess: | ||
38 | // Arguments are now placed in a single string so that | ||
39 | // <shell> -c <commandstring> is passed to the shell | ||
40 | // to make the use of "operator<<" consistent with KProcess | ||
41 | // | ||
42 | // | ||
43 | // Ported by Holger Freyther | ||
44 | // <zekce> Harlekin: oprocess and say it was ported to Qt by the Opie developers an Qt 2 | ||
45 | |||
46 | |||
47 | |||
48 | #include "oprocess.h" | ||
49 | #define _MAY_INCLUDE_KPROCESSCONTROLLER_ | ||
50 | #include "oprocctrl.h" | ||
51 | |||
52 | //#include <config.h> | ||
53 | |||
54 | #include <qfile.h> | ||
55 | #include <qsocketnotifier.h> | ||
56 | #include <qregexp.h> | ||
57 | |||
58 | #include <sys/time.h> | ||
59 | #include <sys/types.h> | ||
60 | #include <sys/stat.h> | ||
61 | #include <sys/socket.h> | ||
62 | |||
63 | #include <errno.h> | ||
64 | #include <fcntl.h> | ||
65 | #include <stdlib.h> | ||
66 | #include <signal.h> | ||
67 | #include <stdio.h> | ||
68 | #include <string.h> | ||
69 | #include <unistd.h> | ||
70 | #ifdef HAVE_SYS_SELECT_H | ||
71 | #include <sys/select.h> | ||
72 | #endif | ||
73 | #ifdef HAVE_INITGROUPS | ||
74 | #include <grp.h> | ||
75 | #endif | ||
76 | #include <pwd.h> | ||
77 | |||
78 | #include <qapplication.h> | ||
79 | #include <qmap.h> | ||
80 | //#include <kdebug.h> | ||
81 | |||
82 | ///////////////////////////// | ||
83 | // public member functions // | ||
84 | ///////////////////////////// | ||
85 | |||
86 | class OProcessPrivate { | ||
87 | public: | ||
88 | OProcessPrivate() : useShell(false) { } | ||
89 | |||
90 | bool useShell; | ||
91 | QMap<QString,QString> env; | ||
92 | QString wd; | ||
93 | QCString shell; | ||
94 | }; | ||
95 | |||
96 | |||
97 | OProcess::OProcess() | ||
98 | : QObject(), | ||
99 | run_mode(NotifyOnExit), | ||
100 | runs(false), | ||
101 | pid_(0), | ||
102 | status(0), | ||
103 | keepPrivs(false), | ||
104 | innot(0), | ||
105 | outnot(0), | ||
106 | errnot(0), | ||
107 | communication(NoCommunication), | ||
108 | input_data(0), | ||
109 | input_sent(0), | ||
110 | input_total(0), | ||
111 | d(0) | ||
112 | { | ||
113 | if (0 == OProcessController::theOProcessController) { | ||
114 | (void) new OProcessController(); | ||
115 | CHECK_PTR(OProcessController::theOProcessController); | ||
116 | } | ||
117 | |||
118 | OProcessController::theOProcessController->addOProcess(this); | ||
119 | out[0] = out[1] = -1; | ||
120 | in[0] = in[1] = -1; | ||
121 | err[0] = err[1] = -1; | ||
122 | } | ||
123 | |||
124 | void | ||
125 | OProcess::setEnvironment(const QString &name, const QString &value) | ||
126 | { | ||
127 | if (!d) | ||
128 | d = new OProcessPrivate; | ||
129 | d->env.insert(name, value); | ||
130 | } | ||
131 | |||
132 | void | ||
133 | OProcess::setWorkingDirectory(const QString &dir) | ||
134 | { | ||
135 | if (!d) | ||
136 | d = new OProcessPrivate; | ||
137 | d->wd = dir; | ||
138 | } | ||
139 | |||
140 | void | ||
141 | OProcess::setupEnvironment() | ||
142 | { | ||
143 | if (d) | ||
144 | { | ||
145 | QMap<QString,QString>::Iterator it; | ||
146 | for(it = d->env.begin(); it != d->env.end(); ++it) | ||
147 | setenv(QFile::encodeName(it.key()).data(), | ||
148 | QFile::encodeName(it.data()).data(), 1); | ||
149 | if (!d->wd.isEmpty()) | ||
150 | chdir(QFile::encodeName(d->wd).data()); | ||
151 | } | ||
152 | } | ||
153 | |||
154 | void | ||
155 | OProcess::setRunPrivileged(bool keepPrivileges) | ||
156 | { | ||
157 | keepPrivs = keepPrivileges; | ||
158 | } | ||
159 | |||
160 | bool | ||
161 | OProcess::runPrivileged() const | ||
162 | { | ||
163 | return keepPrivs; | ||
164 | } | ||
165 | |||
166 | |||
167 | OProcess::~OProcess() | ||
168 | { | ||
169 | // destroying the OProcess instance sends a SIGKILL to the | ||
170 | // child process (if it is running) after removing it from the | ||
171 | // list of valid processes (if the process is not started as | ||
172 | // "DontCare") | ||
173 | |||
174 | OProcessController::theOProcessController->removeOProcess(this); | ||
175 | // this must happen before we kill the child | ||
176 | // TODO: block the signal while removing the current process from the process list | ||
177 | |||
178 | if (runs && (run_mode != DontCare)) | ||
179 | kill(SIGKILL); | ||
180 | |||
181 | // Clean up open fd's and socket notifiers. | ||
182 | closeStdin(); | ||
183 | closeStdout(); | ||
184 | closeStderr(); | ||
185 | |||
186 | // TODO: restore SIGCHLD and SIGPIPE handler if this is the last OProcess | ||
187 | delete d; | ||
188 | } | ||
189 | |||
190 | void OProcess::detach() | ||
191 | { | ||
192 | OProcessController::theOProcessController->removeOProcess(this); | ||
193 | |||
194 | runs = false; | ||
195 | pid_ = 0; | ||
196 | |||
197 | // Clean up open fd's and socket notifiers. | ||
198 | closeStdin(); | ||
199 | closeStdout(); | ||
200 | closeStderr(); | ||
201 | } | ||
202 | |||
203 | bool OProcess::setExecutable(const QString& proc) | ||
204 | { | ||
205 | if (runs) return false; | ||
206 | |||
207 | if (proc.isEmpty()) return false; | ||
208 | |||
209 | if (!arguments.isEmpty()) | ||
210 | arguments.remove(arguments.begin()); | ||
211 | arguments.prepend(QFile::encodeName(proc)); | ||
212 | |||
213 | return true; | ||
214 | } | ||
215 | |||
216 | OProcess &OProcess::operator<<(const QStringList& args) | ||
217 | { | ||
218 | QStringList::ConstIterator it = args.begin(); | ||
219 | for ( ; it != args.end() ; ++it ) | ||
220 | arguments.append(QFile::encodeName(*it)); | ||
221 | return *this; | ||
222 | } | ||
223 | |||
224 | OProcess &OProcess::operator<<(const QCString& arg) | ||
225 | { | ||
226 | return operator<< (arg.data()); | ||
227 | } | ||
228 | |||
229 | OProcess &OProcess::operator<<(const char* arg) | ||
230 | { | ||
231 | arguments.append(arg); | ||
232 | return *this; | ||
233 | } | ||
234 | |||
235 | OProcess &OProcess::operator<<(const QString& arg) | ||
236 | { | ||
237 | arguments.append(QFile::encodeName(arg)); | ||
238 | return *this; | ||
239 | } | ||
240 | |||
241 | void OProcess::clearArguments() | ||
242 | { | ||
243 | arguments.clear(); | ||
244 | } | ||
245 | |||
246 | bool OProcess::start(RunMode runmode, Communication comm) | ||
247 | { | ||
248 | uint i; | ||
249 | uint n = arguments.count(); | ||
250 | char **arglist; | ||
251 | |||
252 | if (runs || (0 == n)) { | ||
253 | return false; // cannot start a process that is already running | ||
254 | // or if no executable has been assigned | ||
255 | } | ||
256 | run_mode = runmode; | ||
257 | status = 0; | ||
258 | |||
259 | QCString shellCmd; | ||
260 | if (d && d->useShell) | ||
261 | { | ||
262 | if (d->shell.isEmpty()) | ||
263 | { | ||
264 | qWarning( "Could not find a valid shell" ); | ||
265 | return false; | ||
266 | } | ||
267 | |||
268 | arglist = static_cast<char **>(malloc( (4)*sizeof(char *))); | ||
269 | for (i=0; i < n; i++) { | ||
270 | shellCmd += arguments[i]; | ||
271 | shellCmd += " "; // CC: to separate the arguments | ||
272 | } | ||
273 | |||
274 | arglist[0] = d->shell.data(); | ||
275 | arglist[1] = (char *) "-c"; | ||
276 | arglist[2] = shellCmd.data(); | ||
277 | arglist[3] = 0; | ||
278 | } | ||
279 | else | ||
280 | { | ||
281 | arglist = static_cast<char **>(malloc( (n+1)*sizeof(char *))); | ||
282 | for (i=0; i < n; i++) | ||
283 | arglist[i] = arguments[i].data(); | ||
284 | arglist[n]= 0; | ||
285 | } | ||
286 | |||
287 | if (!setupCommunication(comm)) | ||
288 | qWarning( "Could not setup Communication!"); | ||
289 | |||
290 | // We do this in the parent because if we do it in the child process | ||
291 | // gdb gets confused when the application runs from gdb. | ||
292 | uid_t uid = getuid(); | ||
293 | gid_t gid = getgid(); | ||
294 | #ifdef HAVE_INITGROUPS | ||
295 | struct passwd *pw = getpwuid(uid); | ||
296 | #endif | ||
297 | |||
298 | int fd[2]; | ||
299 | if (0 > pipe(fd)) | ||
300 | { | ||
301 | fd[0] = fd[1] = 0; // Pipe failed.. continue | ||
302 | } | ||
303 | |||
304 | runs = true; | ||
305 | |||
306 | QApplication::flushX(); | ||
307 | |||
308 | // WABA: Note that we use fork() and not vfork() because | ||
309 | // vfork() has unclear semantics and is not standardized. | ||
310 | pid_ = fork(); | ||
311 | |||
312 | if (0 == pid_) { | ||
313 | if (fd[0]) | ||
314 | close(fd[0]); | ||
315 | if (!runPrivileged()) | ||
316 | { | ||
317 | setgid(gid); | ||
318 | #if defined( HAVE_INITGROUPS) | ||
319 | if(pw) | ||
320 | initgroups(pw->pw_name, pw->pw_gid); | ||
321 | #endif | ||
322 | setuid(uid); | ||
323 | } | ||
324 | // The child process | ||
325 | if(!commSetupDoneC()) | ||
326 | qWarning( "Could not finish comm setup in child!" ); | ||
327 | |||
328 | setupEnvironment(); | ||
329 | |||
330 | // Matthias | ||
331 | if (run_mode == DontCare) | ||
332 | setpgid(0,0); | ||
333 | // restore default SIGPIPE handler (Harri) | ||
334 | struct sigaction act; | ||
335 | sigemptyset(&(act.sa_mask)); | ||
336 | sigaddset(&(act.sa_mask), SIGPIPE); | ||
337 | act.sa_handler = SIG_DFL; | ||
338 | act.sa_flags = 0; | ||
339 | sigaction(SIGPIPE, &act, 0L); | ||
340 | |||
341 | // We set the close on exec flag. | ||
342 | // Closing of fd[1] indicates that the execvp succeeded! | ||
343 | if (fd[1]) | ||
344 | fcntl(fd[1], F_SETFD, FD_CLOEXEC); | ||
345 | execvp(arglist[0], arglist); | ||
346 | char resultByte = 1; | ||
347 | if (fd[1]) | ||
348 | write(fd[1], &resultByte, 1); | ||
349 | _exit(-1); | ||
350 | } else if (-1 == pid_) { | ||
351 | // forking failed | ||
352 | |||
353 | runs = false; | ||
354 | free(arglist); | ||
355 | return false; | ||
356 | } else { | ||
357 | if (fd[1]) | ||
358 | close(fd[1]); | ||
359 | // the parent continues here | ||
360 | |||
361 | // Discard any data for stdin that might still be there | ||
362 | input_data = 0; | ||
363 | |||
364 | // Check whether client could be started. | ||
365 | if (fd[0]) for(;;) | ||
366 | { | ||
367 | char resultByte; | ||
368 | int n = ::read(fd[0], &resultByte, 1); | ||
369 | if (n == 1) | ||
370 | { | ||
371 | // Error | ||
372 | runs = false; | ||
373 | close(fd[0]); | ||
374 | free(arglist); | ||
375 | pid_ = 0; | ||
376 | return false; | ||
377 | } | ||
378 | if (n == -1) | ||
379 | { | ||
380 | if ((errno == ECHILD) || (errno == EINTR)) | ||
381 | continue; // Ignore | ||
382 | } | ||
383 | break; // success | ||
384 | } | ||
385 | if (fd[0]) | ||
386 | close(fd[0]); | ||
387 | |||
388 | if (!commSetupDoneP()) // finish communication socket setup for the parent | ||
389 | qWarning( "Could not finish comm setup in parent!" ); | ||
390 | |||
391 | if (run_mode == Block) { | ||
392 | commClose(); | ||
393 | |||
394 | // The SIGCHLD handler of the process controller will catch | ||
395 | // the exit and set the status | ||
396 | while(runs) | ||
397 | { | ||
398 | OProcessController::theOProcessController-> | ||
399 | slotDoHousekeeping(0); | ||
400 | } | ||
401 | runs = FALSE; | ||
402 | emit processExited(this); | ||
403 | } | ||
404 | } | ||
405 | free(arglist); | ||
406 | return true; | ||
407 | } | ||
408 | |||
409 | |||
410 | |||
411 | bool OProcess::kill(int signo) | ||
412 | { | ||
413 | bool rv=false; | ||
414 | |||
415 | if (0 != pid_) | ||
416 | rv= (-1 != ::kill(pid_, signo)); | ||
417 | // probably store errno somewhere... | ||
418 | return rv; | ||
419 | } | ||
420 | |||
421 | |||
422 | |||
423 | bool OProcess::isRunning() const | ||
424 | { | ||
425 | return runs; | ||
426 | } | ||
427 | |||
428 | |||
429 | |||
430 | pid_t OProcess::pid() const | ||
431 | { | ||
432 | return pid_; | ||
433 | } | ||
434 | |||
435 | |||
436 | |||
437 | bool OProcess::normalExit() const | ||
438 | { | ||
439 | int _status = status; | ||
440 | return (pid_ != 0) && (!runs) && (WIFEXITED((_status))); | ||
441 | } | ||
442 | |||
443 | |||
444 | |||
445 | int OProcess::exitStatus() const | ||
446 | { | ||
447 | int _status = status; | ||
448 | return WEXITSTATUS((_status)); | ||
449 | } | ||
450 | |||
451 | |||
452 | |||
453 | bool OProcess::writeStdin(const char *buffer, int buflen) | ||
454 | { | ||
455 | bool rv; | ||
456 | |||
457 | // if there is still data pending, writing new data | ||
458 | // to stdout is not allowed (since it could also confuse | ||
459 | // kprocess... | ||
460 | if (0 != input_data) | ||
461 | return false; | ||
462 | |||
463 | if (runs && (communication & Stdin)) { | ||
464 | input_data = buffer; | ||
465 | input_sent = 0; | ||
466 | input_total = buflen; | ||
467 | slotSendData(0); | ||
468 | innot->setEnabled(true); | ||
469 | rv = true; | ||
470 | } else | ||
471 | rv = false; | ||
472 | return rv; | ||
473 | } | ||
474 | |||
475 | void OProcess::suspend() | ||
476 | { | ||
477 | if ((communication & Stdout) && outnot) | ||
478 | outnot->setEnabled(false); | ||
479 | } | ||
480 | |||
481 | void OProcess::resume() | ||
482 | { | ||
483 | if ((communication & Stdout) && outnot) | ||
484 | outnot->setEnabled(true); | ||
485 | } | ||
486 | |||
487 | bool OProcess::closeStdin() | ||
488 | { | ||
489 | bool rv; | ||
490 | |||
491 | if (communication & Stdin) { | ||
492 | communication = (Communication) (communication & ~Stdin); | ||
493 | delete innot; | ||
494 | innot = 0; | ||
495 | close(in[1]); | ||
496 | rv = true; | ||
497 | } else | ||
498 | rv = false; | ||
499 | return rv; | ||
500 | } | ||
501 | |||
502 | bool OProcess::closeStdout() | ||
503 | { | ||
504 | bool rv; | ||
505 | |||
506 | if (communication & Stdout) { | ||
507 | communication = (Communication) (communication & ~Stdout); | ||
508 | delete outnot; | ||
509 | outnot = 0; | ||
510 | close(out[0]); | ||
511 | rv = true; | ||
512 | } else | ||
513 | rv = false; | ||
514 | return rv; | ||
515 | } | ||
516 | |||
517 | bool OProcess::closeStderr() | ||
518 | { | ||
519 | bool rv; | ||
520 | |||
521 | if (communication & Stderr) { | ||
522 | communication = static_cast<Communication>(communication & ~Stderr); | ||
523 | delete errnot; | ||
524 | errnot = 0; | ||
525 | close(err[0]); | ||
526 | rv = true; | ||
527 | } else | ||
528 | rv = false; | ||
529 | return rv; | ||
530 | } | ||
531 | |||
532 | |||
533 | ///////////////////////////// | ||
534 | // protected slots // | ||
535 | ///////////////////////////// | ||
536 | |||
537 | |||
538 | |||
539 | void OProcess::slotChildOutput(int fdno) | ||
540 | { | ||
541 | if (!childOutput(fdno)) | ||
542 | closeStdout(); | ||
543 | } | ||
544 | |||
545 | |||
546 | void OProcess::slotChildError(int fdno) | ||
547 | { | ||
548 | if (!childError(fdno)) | ||
549 | closeStderr(); | ||
550 | } | ||
551 | |||
552 | |||
553 | void OProcess::slotSendData(int) | ||
554 | { | ||
555 | if (input_sent == input_total) { | ||
556 | innot->setEnabled(false); | ||
557 | input_data = 0; | ||
558 | emit wroteStdin(this); | ||
559 | } else | ||
560 | input_sent += ::write(in[1], input_data+input_sent, input_total-input_sent); | ||
561 | } | ||
562 | |||
563 | |||
564 | |||
565 | ////////////////////////////// | ||
566 | // private member functions // | ||
567 | ////////////////////////////// | ||
568 | |||
569 | |||
570 | |||
571 | void OProcess::processHasExited(int state) | ||
572 | { | ||
573 | if (runs) | ||
574 | { | ||
575 | runs = false; | ||
576 | status = state; | ||
577 | |||
578 | commClose(); // cleanup communication sockets | ||
579 | |||
580 | // also emit a signal if the process was run Blocking | ||
581 | if (DontCare != run_mode) | ||
582 | { | ||
583 | emit processExited(this); | ||
584 | } | ||
585 | } | ||
586 | } | ||
587 | |||
588 | |||
589 | |||
590 | int OProcess::childOutput(int fdno) | ||
591 | { | ||
592 | if (communication & NoRead) { | ||
593 | int len = -1; | ||
594 | emit receivedStdout(fdno, len); | ||
595 | errno = 0; // Make sure errno doesn't read "EAGAIN" | ||
596 | return len; | ||
597 | } | ||
598 | else | ||
599 | { | ||
600 | char buffer[1024]; | ||
601 | int len; | ||
602 | |||
603 | len = ::read(fdno, buffer, 1024); | ||
604 | |||
605 | if ( 0 < len) { | ||
606 | emit receivedStdout(this, buffer, len); | ||
607 | } | ||
608 | return len; | ||
609 | } | ||
610 | } | ||
611 | |||
612 | |||
613 | |||
614 | int OProcess::childError(int fdno) | ||
615 | { | ||
616 | char buffer[1024]; | ||
617 | int len; | ||
618 | |||
619 | len = ::read(fdno, buffer, 1024); | ||
620 | |||
621 | if ( 0 < len) | ||
622 | emit receivedStderr(this, buffer, len); | ||
623 | return len; | ||
624 | } | ||
625 | |||
626 | |||
627 | |||
628 | int OProcess::setupCommunication(Communication comm) | ||
629 | { | ||
630 | int ok; | ||
631 | |||
632 | communication = comm; | ||
633 | |||
634 | ok = 1; | ||
635 | if (comm & Stdin) | ||
636 | ok &= socketpair(AF_UNIX, SOCK_STREAM, 0, in) >= 0; | ||
637 | |||
638 | if (comm & Stdout) | ||
639 | ok &= socketpair(AF_UNIX, SOCK_STREAM, 0, out) >= 0; | ||
640 | |||
641 | if (comm & Stderr) | ||
642 | ok &= socketpair(AF_UNIX, SOCK_STREAM, 0, err) >= 0; | ||
643 | |||
644 | return ok; | ||
645 | } | ||
646 | |||
647 | |||
648 | |||
649 | int OProcess::commSetupDoneP() | ||
650 | { | ||
651 | int ok = 1; | ||
652 | |||
653 | if (communication != NoCommunication) { | ||
654 | if (communication & Stdin) | ||
655 | close(in[0]); | ||
656 | if (communication & Stdout) | ||
657 | close(out[1]); | ||
658 | if (communication & Stderr) | ||
659 | close(err[1]); | ||
660 | |||
661 | // Don't create socket notifiers and set the sockets non-blocking if | ||
662 | // blocking is requested. | ||
663 | if (run_mode == Block) return ok; | ||
664 | |||
665 | if (communication & Stdin) { | ||
666 | // ok &= (-1 != fcntl(in[1], F_SETFL, O_NONBLOCK)); | ||
667 | innot = new QSocketNotifier(in[1], QSocketNotifier::Write, this); | ||
668 | CHECK_PTR(innot); | ||
669 | innot->setEnabled(false); // will be enabled when data has to be sent | ||
670 | QObject::connect(innot, SIGNAL(activated(int)), | ||
671 | this, SLOT(slotSendData(int))); | ||
672 | } | ||
673 | |||
674 | if (communication & Stdout) { | ||
675 | // ok &= (-1 != fcntl(out[0], F_SETFL, O_NONBLOCK)); | ||
676 | outnot = new QSocketNotifier(out[0], QSocketNotifier::Read, this); | ||
677 | CHECK_PTR(outnot); | ||
678 | QObject::connect(outnot, SIGNAL(activated(int)), | ||
679 | this, SLOT(slotChildOutput(int))); | ||
680 | if (communication & NoRead) | ||
681 | suspend(); | ||
682 | } | ||
683 | |||
684 | if (communication & Stderr) { | ||
685 | // ok &= (-1 != fcntl(err[0], F_SETFL, O_NONBLOCK)); | ||
686 | errnot = new QSocketNotifier(err[0], QSocketNotifier::Read, this ); | ||
687 | CHECK_PTR(errnot); | ||
688 | QObject::connect(errnot, SIGNAL(activated(int)), | ||
689 | this, SLOT(slotChildError(int))); | ||
690 | } | ||
691 | } | ||
692 | return ok; | ||
693 | } | ||
694 | |||
695 | |||
696 | |||
697 | int OProcess::commSetupDoneC() | ||
698 | { | ||
699 | int ok = 1; | ||
700 | struct linger so; | ||
701 | memset(&so, 0, sizeof(so)); | ||
702 | |||
703 | if (communication & Stdin) | ||
704 | close(in[1]); | ||
705 | if (communication & Stdout) | ||
706 | close(out[0]); | ||
707 | if (communication & Stderr) | ||
708 | close(err[0]); | ||
709 | |||
710 | if (communication & Stdin) | ||
711 | ok &= dup2(in[0], STDIN_FILENO) != -1; | ||
712 | else { | ||
713 | int null_fd = open( "/dev/null", O_RDONLY ); | ||
714 | ok &= dup2( null_fd, STDIN_FILENO ) != -1; | ||
715 | close( null_fd ); | ||
716 | } | ||
717 | if (communication & Stdout) { | ||
718 | ok &= dup2(out[1], STDOUT_FILENO) != -1; | ||
719 | ok &= !setsockopt(out[1], SOL_SOCKET, SO_LINGER, (char*)&so, sizeof(so)); | ||
720 | } | ||
721 | else { | ||
722 | int null_fd = open( "/dev/null", O_WRONLY ); | ||
723 | ok &= dup2( null_fd, STDOUT_FILENO ) != -1; | ||
724 | close( null_fd ); | ||
725 | } | ||
726 | if (communication & Stderr) { | ||
727 | ok &= dup2(err[1], STDERR_FILENO) != -1; | ||
728 | ok &= !setsockopt(err[1], SOL_SOCKET, SO_LINGER, reinterpret_cast<char *>(&so), sizeof(so)); | ||
729 | } | ||
730 | else { | ||
731 | int null_fd = open( "/dev/null", O_WRONLY ); | ||
732 | ok &= dup2( null_fd, STDERR_FILENO ) != -1; | ||
733 | close( null_fd ); | ||
734 | } | ||
735 | return ok; | ||
736 | } | ||
737 | |||
738 | |||
739 | |||
740 | void OProcess::commClose() | ||
741 | { | ||
742 | if (NoCommunication != communication) { | ||
743 | bool b_in = (communication & Stdin); | ||
744 | bool b_out = (communication & Stdout); | ||
745 | bool b_err = (communication & Stderr); | ||
746 | if (b_in) | ||
747 | delete innot; | ||
748 | |||
749 | if (b_out || b_err) { | ||
750 | // If both channels are being read we need to make sure that one socket buffer | ||
751 | // doesn't fill up whilst we are waiting for data on the other (causing a deadlock). | ||
752 | // Hence we need to use select. | ||
753 | |||
754 | // Once one or other of the channels has reached EOF (or given an error) go back | ||
755 | // to the usual mechanism. | ||
756 | |||
757 | int fds_ready = 1; | ||
758 | fd_set rfds; | ||
759 | |||
760 | int max_fd = 0; | ||
761 | if (b_out) { | ||
762 | fcntl(out[0], F_SETFL, O_NONBLOCK); | ||
763 | if (out[0] > max_fd) | ||
764 | max_fd = out[0]; | ||
765 | delete outnot; | ||
766 | outnot = 0; | ||
767 | } | ||
768 | if (b_err) { | ||
769 | fcntl(err[0], F_SETFL, O_NONBLOCK); | ||
770 | if (err[0] > max_fd) | ||
771 | max_fd = err[0]; | ||
772 | delete errnot; | ||
773 | errnot = 0; | ||
774 | } | ||
775 | |||
776 | |||
777 | while (b_out || b_err) { | ||
778 | // * If the process is still running we block until we | ||
779 | // receive data. (p_timeout = 0, no timeout) | ||
780 | // * If the process has already exited, we only check | ||
781 | // the available data, we don't wait for more. | ||
782 | // (p_timeout = &timeout, timeout immediately) | ||
783 | struct timeval timeout; | ||
784 | timeout.tv_sec = 0; | ||
785 | timeout.tv_usec = 0; | ||
786 | struct timeval *p_timeout = runs ? 0 : &timeout; | ||
787 | |||
788 | FD_ZERO(&rfds); | ||
789 | if (b_out) | ||
790 | FD_SET(out[0], &rfds); | ||
791 | |||
792 | if (b_err) | ||
793 | FD_SET(err[0], &rfds); | ||
794 | |||
795 | fds_ready = select(max_fd+1, &rfds, 0, 0, p_timeout); | ||
796 | if (fds_ready <= 0) break; | ||
797 | |||
798 | if (b_out && FD_ISSET(out[0], &rfds)) { | ||
799 | int ret = 1; | ||
800 | while (ret > 0) ret = childOutput(out[0]); | ||
801 | if ((ret == -1 && errno != EAGAIN) || ret == 0) | ||
802 | b_out = false; | ||
803 | } | ||
804 | |||
805 | if (b_err && FD_ISSET(err[0], &rfds)) { | ||
806 | int ret = 1; | ||
807 | while (ret > 0) ret = childError(err[0]); | ||
808 | if ((ret == -1 && errno != EAGAIN) || ret == 0) | ||
809 | b_err = false; | ||
810 | } | ||
811 | } | ||
812 | } | ||
813 | |||
814 | if (b_in) { | ||
815 | communication = (Communication) (communication & ~Stdin); | ||
816 | close(in[1]); | ||
817 | } | ||
818 | if (b_out) { | ||
819 | communication = (Communication) (communication & ~Stdout); | ||
820 | close(out[0]); | ||
821 | } | ||
822 | if (b_err) { | ||
823 | communication = (Communication) (communication & ~Stderr); | ||
824 | close(err[0]); | ||
825 | } | ||
826 | } | ||
827 | } | ||
828 | |||
829 | void OProcess::setUseShell(bool useShell, const char *shell) | ||
830 | { | ||
831 | if (!d) | ||
832 | d = new OProcessPrivate; | ||
833 | d->useShell = useShell; | ||
834 | d->shell = shell; | ||
835 | if (d->shell.isEmpty()) | ||
836 | d->shell = searchShell(); | ||
837 | } | ||
838 | |||
839 | QString OProcess::quote(const QString &arg) | ||
840 | { | ||
841 | QString res = arg; | ||
842 | res.replace(QRegExp(QString::fromLatin1("\'")), | ||
843 | QString::fromLatin1("'\"'\"'")); | ||
844 | res.prepend('\''); | ||
845 | res.append('\''); | ||
846 | return res; | ||
847 | } | ||
848 | |||
849 | QCString OProcess::searchShell() | ||
850 | { | ||
851 | QCString tmpShell = QCString(getenv("SHELL")).stripWhiteSpace(); | ||
852 | if (!isExecutable(tmpShell)) | ||
853 | { | ||
854 | tmpShell = "/bin/sh"; | ||
855 | } | ||
856 | |||
857 | return tmpShell; | ||
858 | } | ||
859 | |||
860 | bool OProcess::isExecutable(const QCString &filename) | ||
861 | { | ||
862 | struct stat fileinfo; | ||
863 | |||
864 | if (filename.isEmpty()) return false; | ||
865 | |||
866 | // CC: we've got a valid filename, now let's see whether we can execute that file | ||
867 | |||
868 | if (-1 == stat(filename.data(), &fileinfo)) return false; | ||
869 | // CC: return false if the file does not exist | ||
870 | |||
871 | // CC: anyway, we cannot execute directories, block/character devices, fifos or sockets | ||
872 | if ( (S_ISDIR(fileinfo.st_mode)) || | ||
873 | (S_ISCHR(fileinfo.st_mode)) || | ||
874 | (S_ISBLK(fileinfo.st_mode)) || | ||
875 | #ifdef S_ISSOCK | ||
876 | // CC: SYSVR4 systems don't have that macro | ||
877 | (S_ISSOCK(fileinfo.st_mode)) || | ||
878 | #endif | ||
879 | (S_ISFIFO(fileinfo.st_mode)) || | ||
880 | (S_ISDIR(fileinfo.st_mode)) ) { | ||
881 | return false; | ||
882 | } | ||
883 | |||
884 | // CC: now check for permission to execute the file | ||
885 | if (access(filename.data(), X_OK) != 0) return false; | ||
886 | |||
887 | // CC: we've passed all the tests... | ||
888 | return true; | ||
889 | } | ||
890 | |||
891 | void OProcess::virtual_hook( int, void* ) | ||
892 | { /*BASE::virtual_hook( id, data );*/ } | ||
893 | |||
894 | |||
895 | /////////////////////////// | ||
896 | // CC: Class KShellProcess | ||
897 | /////////////////////////// | ||
898 | |||
899 | KShellProcess::KShellProcess(const char *shellname): | ||
900 | OProcess() | ||
901 | { | ||
902 | setUseShell(true, shellname); | ||
903 | } | ||
904 | |||
905 | |||
906 | KShellProcess::~KShellProcess() { | ||
907 | } | ||
908 | |||
909 | QString KShellProcess::quote(const QString &arg) | ||
910 | { | ||
911 | return OProcess::quote(arg); | ||
912 | } | ||
913 | |||
914 | bool KShellProcess::start(RunMode runmode, Communication comm) | ||
915 | { | ||
916 | return OProcess::start(runmode, comm); | ||
917 | } | ||
918 | |||
919 | void KShellProcess::virtual_hook( int id, void* data ) | ||
920 | { OProcess::virtual_hook( id, data ); } | ||
921 | |||
922 | //#include "kprocess.moc" | ||