Contents of /trunk/mkinitrd-magellan/klibc/usr/dash/redir.c
Parent Directory | Revision Log
Revision 532 -
(show annotations)
(download)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 10319 byte(s)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 10319 byte(s)
-import if magellan mkinitrd; it is a fork of redhats mkinitrd-5.0.8 with all magellan patches and features; deprecates magellan-src/mkinitrd
1 | /*- |
2 | * Copyright (c) 1991, 1993 |
3 | * The Regents of the University of California. All rights reserved. |
4 | * Copyright (c) 1997-2005 |
5 | * Herbert Xu <herbert@gondor.apana.org.au>. All rights reserved. |
6 | * |
7 | * This code is derived from software contributed to Berkeley by |
8 | * Kenneth Almquist. |
9 | * |
10 | * Redistribution and use in source and binary forms, with or without |
11 | * modification, are permitted provided that the following conditions |
12 | * are met: |
13 | * 1. Redistributions of source code must retain the above copyright |
14 | * notice, this list of conditions and the following disclaimer. |
15 | * 2. Redistributions in binary form must reproduce the above copyright |
16 | * notice, this list of conditions and the following disclaimer in the |
17 | * documentation and/or other materials provided with the distribution. |
18 | * 3. Neither the name of the University nor the names of its contributors |
19 | * may be used to endorse or promote products derived from this software |
20 | * without specific prior written permission. |
21 | * |
22 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND |
23 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
24 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
25 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
26 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
27 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
28 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
29 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
30 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
31 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
32 | * SUCH DAMAGE. |
33 | */ |
34 | |
35 | #include <sys/stat.h> |
36 | #include <sys/types.h> |
37 | #include <sys/param.h> /* PIPE_BUF */ |
38 | #include <signal.h> |
39 | #include <string.h> |
40 | #include <fcntl.h> |
41 | #include <unistd.h> |
42 | #include <stdlib.h> |
43 | |
44 | /* |
45 | * Code for dealing with input/output redirection. |
46 | */ |
47 | |
48 | #include "main.h" |
49 | #include "shell.h" |
50 | #include "nodes.h" |
51 | #include "jobs.h" |
52 | #include "options.h" |
53 | #include "expand.h" |
54 | #include "redir.h" |
55 | #include "output.h" |
56 | #include "memalloc.h" |
57 | #include "error.h" |
58 | |
59 | |
60 | #define EMPTY -2 /* marks an unused slot in redirtab */ |
61 | #ifndef PIPE_BUF |
62 | # define PIPESIZE 4096 /* amount of buffering in a pipe */ |
63 | #else |
64 | # define PIPESIZE PIPE_BUF |
65 | #endif |
66 | |
67 | |
68 | MKINIT |
69 | struct redirtab { |
70 | struct redirtab *next; |
71 | int renamed[10]; |
72 | int nullredirs; |
73 | }; |
74 | |
75 | |
76 | MKINIT struct redirtab *redirlist; |
77 | MKINIT int nullredirs; |
78 | |
79 | STATIC int openredirect(union node *); |
80 | #ifdef notyet |
81 | STATIC void dupredirect(union node *, int, char[10]); |
82 | #else |
83 | STATIC void dupredirect(union node *, int); |
84 | #endif |
85 | STATIC int openhere(union node *); |
86 | STATIC int noclobberopen(const char *); |
87 | |
88 | |
89 | /* |
90 | * Process a list of redirection commands. If the REDIR_PUSH flag is set, |
91 | * old file descriptors are stashed away so that the redirection can be |
92 | * undone by calling popredir. If the REDIR_BACKQ flag is set, then the |
93 | * standard output, and the standard error if it becomes a duplicate of |
94 | * stdout, is saved in memory. |
95 | */ |
96 | |
97 | void |
98 | redirect(union node *redir, int flags) |
99 | { |
100 | union node *n; |
101 | struct redirtab *sv; |
102 | int i; |
103 | int fd; |
104 | int newfd; |
105 | int *p; |
106 | #if notyet |
107 | char memory[10]; /* file descriptors to write to memory */ |
108 | |
109 | for (i = 10 ; --i >= 0 ; ) |
110 | memory[i] = 0; |
111 | memory[1] = flags & REDIR_BACKQ; |
112 | #endif |
113 | nullredirs++; |
114 | if (!redir) { |
115 | return; |
116 | } |
117 | sv = NULL; |
118 | INTOFF; |
119 | if (flags & REDIR_PUSH) { |
120 | struct redirtab *q; |
121 | q = ckmalloc(sizeof (struct redirtab)); |
122 | q->next = redirlist; |
123 | redirlist = q; |
124 | q->nullredirs = nullredirs - 1; |
125 | for (i = 0 ; i < 10 ; i++) |
126 | q->renamed[i] = EMPTY; |
127 | nullredirs = 0; |
128 | sv = q; |
129 | } |
130 | n = redir; |
131 | do { |
132 | fd = n->nfile.fd; |
133 | if ((n->nfile.type == NTOFD || n->nfile.type == NFROMFD) && |
134 | n->ndup.dupfd == fd) |
135 | continue; /* redirect from/to same file descriptor */ |
136 | |
137 | newfd = openredirect(n); |
138 | if (fd == newfd) |
139 | continue; |
140 | if (sv && *(p = &sv->renamed[fd]) == EMPTY) { |
141 | int i = fcntl(fd, F_DUPFD, 10); |
142 | if (i == -1) { |
143 | i = errno; |
144 | if (i != EBADF) { |
145 | const char *m = strerror(i); |
146 | close(newfd); |
147 | sh_error("%d: %s", fd, m); |
148 | /* NOTREACHED */ |
149 | } |
150 | } else { |
151 | *p = i; |
152 | close(fd); |
153 | } |
154 | } else { |
155 | close(fd); |
156 | } |
157 | #ifdef notyet |
158 | dupredirect(n, newfd, memory); |
159 | #else |
160 | dupredirect(n, newfd); |
161 | #endif |
162 | } while ((n = n->nfile.next)); |
163 | INTON; |
164 | #ifdef notyet |
165 | if (memory[1]) |
166 | out1 = &memout; |
167 | if (memory[2]) |
168 | out2 = &memout; |
169 | #endif |
170 | if (flags & REDIR_SAVEFD2 && sv && sv->renamed[2] >= 0) |
171 | preverrout.fd = sv->renamed[2]; |
172 | } |
173 | |
174 | |
175 | STATIC int |
176 | openredirect(union node *redir) |
177 | { |
178 | char *fname; |
179 | int f; |
180 | |
181 | switch (redir->nfile.type) { |
182 | case NFROM: |
183 | fname = redir->nfile.expfname; |
184 | if ((f = open64(fname, O_RDONLY)) < 0) |
185 | goto eopen; |
186 | break; |
187 | case NFROMTO: |
188 | fname = redir->nfile.expfname; |
189 | if ((f = open64(fname, O_RDWR|O_CREAT|O_TRUNC, 0666)) < 0) |
190 | goto ecreate; |
191 | break; |
192 | case NTO: |
193 | /* Take care of noclobber mode. */ |
194 | if (Cflag) { |
195 | fname = redir->nfile.expfname; |
196 | if ((f = noclobberopen(fname)) < 0) |
197 | goto ecreate; |
198 | break; |
199 | } |
200 | /* FALLTHROUGH */ |
201 | case NCLOBBER: |
202 | fname = redir->nfile.expfname; |
203 | if ((f = open64(fname, O_WRONLY|O_CREAT|O_TRUNC, 0666)) < 0) |
204 | goto ecreate; |
205 | break; |
206 | case NAPPEND: |
207 | fname = redir->nfile.expfname; |
208 | if ((f = open64(fname, O_WRONLY|O_CREAT|O_APPEND, 0666)) < 0) |
209 | goto ecreate; |
210 | break; |
211 | default: |
212 | #ifdef DEBUG |
213 | abort(); |
214 | #endif |
215 | /* Fall through to eliminate warning. */ |
216 | case NTOFD: |
217 | case NFROMFD: |
218 | f = -1; |
219 | break; |
220 | case NHERE: |
221 | case NXHERE: |
222 | f = openhere(redir); |
223 | break; |
224 | } |
225 | |
226 | return f; |
227 | ecreate: |
228 | sh_error("cannot create %s: %s", fname, errmsg(errno, E_CREAT)); |
229 | eopen: |
230 | sh_error("cannot open %s: %s", fname, errmsg(errno, E_OPEN)); |
231 | } |
232 | |
233 | |
234 | STATIC void |
235 | #ifdef notyet |
236 | dupredirect(redir, f, memory) |
237 | #else |
238 | dupredirect(redir, f) |
239 | #endif |
240 | union node *redir; |
241 | int f; |
242 | #ifdef notyet |
243 | char memory[10]; |
244 | #endif |
245 | { |
246 | int fd = redir->nfile.fd; |
247 | |
248 | #ifdef notyet |
249 | memory[fd] = 0; |
250 | #endif |
251 | if (redir->nfile.type == NTOFD || redir->nfile.type == NFROMFD) { |
252 | if (redir->ndup.dupfd >= 0) { /* if not ">&-" */ |
253 | #ifdef notyet |
254 | if (memory[redir->ndup.dupfd]) |
255 | memory[fd] = 1; |
256 | else |
257 | #endif |
258 | copyfd(redir->ndup.dupfd, fd); |
259 | } |
260 | return; |
261 | } |
262 | |
263 | if (f != fd) { |
264 | copyfd(f, fd); |
265 | close(f); |
266 | } |
267 | return; |
268 | } |
269 | |
270 | |
271 | /* |
272 | * Handle here documents. Normally we fork off a process to write the |
273 | * data to a pipe. If the document is short, we can stuff the data in |
274 | * the pipe without forking. |
275 | */ |
276 | |
277 | STATIC int |
278 | openhere(union node *redir) |
279 | { |
280 | int pip[2]; |
281 | size_t len = 0; |
282 | |
283 | if (pipe(pip) < 0) |
284 | sh_error("Pipe call failed"); |
285 | if (redir->type == NHERE) { |
286 | len = strlen(redir->nhere.doc->narg.text); |
287 | if (len <= PIPESIZE) { |
288 | xwrite(pip[1], redir->nhere.doc->narg.text, len); |
289 | goto out; |
290 | } |
291 | } |
292 | if (forkshell((struct job *)NULL, (union node *)NULL, FORK_NOJOB) == 0) { |
293 | close(pip[0]); |
294 | signal(SIGINT, SIG_IGN); |
295 | signal(SIGQUIT, SIG_IGN); |
296 | signal(SIGHUP, SIG_IGN); |
297 | #ifdef SIGTSTP |
298 | signal(SIGTSTP, SIG_IGN); |
299 | #endif |
300 | signal(SIGPIPE, SIG_DFL); |
301 | if (redir->type == NHERE) |
302 | xwrite(pip[1], redir->nhere.doc->narg.text, len); |
303 | else |
304 | expandhere(redir->nhere.doc, pip[1]); |
305 | _exit(0); |
306 | } |
307 | out: |
308 | close(pip[1]); |
309 | return pip[0]; |
310 | } |
311 | |
312 | |
313 | |
314 | /* |
315 | * Undo the effects of the last redirection. |
316 | */ |
317 | |
318 | void |
319 | popredir(int drop) |
320 | { |
321 | struct redirtab *rp; |
322 | int i; |
323 | |
324 | if (--nullredirs >= 0) |
325 | return; |
326 | INTOFF; |
327 | rp = redirlist; |
328 | for (i = 0 ; i < 10 ; i++) { |
329 | if (rp->renamed[i] != EMPTY) { |
330 | if (!drop) { |
331 | close(i); |
332 | copyfd(rp->renamed[i], i); |
333 | } |
334 | close(rp->renamed[i]); |
335 | } |
336 | } |
337 | redirlist = rp->next; |
338 | nullredirs = rp->nullredirs; |
339 | ckfree(rp); |
340 | INTON; |
341 | } |
342 | |
343 | /* |
344 | * Undo all redirections. Called on error or interrupt. |
345 | */ |
346 | |
347 | #ifdef mkinit |
348 | |
349 | INCLUDE "redir.h" |
350 | |
351 | RESET { |
352 | clearredir(0); |
353 | } |
354 | |
355 | #endif |
356 | |
357 | /* |
358 | * Discard all saved file descriptors. |
359 | */ |
360 | |
361 | void |
362 | clearredir(int drop) |
363 | { |
364 | for (;;) { |
365 | nullredirs = 0; |
366 | if (!redirlist) |
367 | break; |
368 | popredir(drop); |
369 | } |
370 | } |
371 | |
372 | |
373 | |
374 | /* |
375 | * Copy a file descriptor to be >= to. Returns -1 |
376 | * if the source file descriptor is closed, EMPTY if there are no unused |
377 | * file descriptors left. |
378 | */ |
379 | |
380 | int |
381 | copyfd(int from, int to) |
382 | { |
383 | int newfd; |
384 | |
385 | newfd = fcntl(from, F_DUPFD, to); |
386 | if (newfd < 0) { |
387 | int errno2 = errno; |
388 | if (errno2 == EMFILE) |
389 | return EMPTY; |
390 | else |
391 | sh_error("%d: %s", from, strerror(errno2)); |
392 | } |
393 | return newfd; |
394 | } |
395 | |
396 | |
397 | /* |
398 | * Open a file in noclobber mode. |
399 | * The code was copied from bash. |
400 | */ |
401 | int |
402 | noclobberopen(fname) |
403 | const char *fname; |
404 | { |
405 | int r, fd; |
406 | struct stat64 finfo, finfo2; |
407 | |
408 | /* |
409 | * If the file exists and is a regular file, return an error |
410 | * immediately. |
411 | */ |
412 | r = stat64(fname, &finfo); |
413 | if (r == 0 && S_ISREG(finfo.st_mode)) { |
414 | errno = EEXIST; |
415 | return -1; |
416 | } |
417 | |
418 | /* |
419 | * If the file was not present (r != 0), make sure we open it |
420 | * exclusively so that if it is created before we open it, our open |
421 | * will fail. Make sure that we do not truncate an existing file. |
422 | * Note that we don't turn on O_EXCL unless the stat failed -- if the |
423 | * file was not a regular file, we leave O_EXCL off. |
424 | */ |
425 | if (r != 0) |
426 | return open64(fname, O_WRONLY|O_CREAT|O_EXCL, 0666); |
427 | fd = open64(fname, O_WRONLY|O_CREAT, 0666); |
428 | |
429 | /* If the open failed, return the file descriptor right away. */ |
430 | if (fd < 0) |
431 | return fd; |
432 | |
433 | /* |
434 | * OK, the open succeeded, but the file may have been changed from a |
435 | * non-regular file to a regular file between the stat and the open. |
436 | * We are assuming that the O_EXCL open handles the case where FILENAME |
437 | * did not exist and is symlinked to an existing file between the stat |
438 | * and open. |
439 | */ |
440 | |
441 | /* |
442 | * If we can open it and fstat the file descriptor, and neither check |
443 | * revealed that it was a regular file, and the file has not been |
444 | * replaced, return the file descriptor. |
445 | */ |
446 | if (fstat64(fd, &finfo2) == 0 && !S_ISREG(finfo2.st_mode) && |
447 | finfo.st_dev == finfo2.st_dev && finfo.st_ino == finfo2.st_ino) |
448 | return fd; |
449 | |
450 | /* The file has been replaced. badness. */ |
451 | close(fd); |
452 | errno = EEXIST; |
453 | return -1; |
454 | } |
455 | |
456 | |
457 | int |
458 | redirectsafe(union node *redir, int flags) |
459 | { |
460 | int err; |
461 | volatile int saveint; |
462 | struct jmploc *volatile savehandler = handler; |
463 | struct jmploc jmploc; |
464 | |
465 | SAVEINT(saveint); |
466 | if (!(err = setjmp(jmploc.loc) * 2)) { |
467 | handler = &jmploc; |
468 | redirect(redir, flags); |
469 | } |
470 | handler = savehandler; |
471 | if (err && exception != EXERROR) |
472 | longjmp(handler->loc, 1); |
473 | RESTOREINT(saveint); |
474 | return err; |
475 | } |