Merge git://git.kernel.org/pub/scm/linux/kernel/git/herbert/crypto-2.6
[linux-2.6] / arch / um / os-Linux / helper.c
1 /*
2  * Copyright (C) 2002 Jeff Dike (jdike@karaya.com)
3  * Licensed under the GPL
4  */
5
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <unistd.h>
9 #include <errno.h>
10 #include <sched.h>
11 #include <limits.h>
12 #include <sys/signal.h>
13 #include <sys/wait.h>
14 #include <sys/socket.h>
15 #include "user.h"
16 #include "kern_util.h"
17 #include "os.h"
18 #include "um_malloc.h"
19 #include "kern_constants.h"
20
21 struct helper_data {
22         void (*pre_exec)(void*);
23         void *pre_data;
24         char **argv;
25         int fd;
26         char *buf;
27 };
28
29 static int helper_child(void *arg)
30 {
31         struct helper_data *data = arg;
32         char **argv = data->argv;
33         int errval;
34
35         if (data->pre_exec != NULL)
36                 (*data->pre_exec)(data->pre_data);
37         errval = execvp_noalloc(data->buf, argv[0], argv);
38         printk("helper_child - execvp of '%s' failed - errno = %d\n", argv[0],
39                -errval);
40         write(data->fd, &errval, sizeof(errval));
41         kill(os_getpid(), SIGKILL);
42         return 0;
43 }
44
45 /* Returns either the pid of the child process we run or -E* on failure.
46  * XXX The alloc_stack here breaks if this is called in the tracing thread, so
47  * we need to receive a preallocated stack (a local buffer is ok). */
48 int run_helper(void (*pre_exec)(void *), void *pre_data, char **argv)
49 {
50         struct helper_data data;
51         unsigned long stack, sp;
52         int pid, fds[2], ret, n;
53
54         stack = alloc_stack(0, __cant_sleep());
55         if (stack == 0)
56                 return -ENOMEM;
57
58         ret = socketpair(AF_UNIX, SOCK_STREAM, 0, fds);
59         if (ret < 0) {
60                 ret = -errno;
61                 printk("run_helper : pipe failed, errno = %d\n", errno);
62                 goto out_free;
63         }
64
65         ret = os_set_exec_close(fds[1]);
66         if (ret < 0) {
67                 printk("run_helper : setting FD_CLOEXEC failed, ret = %d\n",
68                        -ret);
69                 goto out_close;
70         }
71
72         sp = stack + UM_KERN_PAGE_SIZE - sizeof(void *);
73         data.pre_exec = pre_exec;
74         data.pre_data = pre_data;
75         data.argv = argv;
76         data.fd = fds[1];
77         data.buf = __cant_sleep() ? kmalloc(PATH_MAX, UM_GFP_ATOMIC) :
78                                         kmalloc(PATH_MAX, UM_GFP_KERNEL);
79         pid = clone(helper_child, (void *) sp, CLONE_VM, &data);
80         if (pid < 0) {
81                 ret = -errno;
82                 printk("run_helper : clone failed, errno = %d\n", errno);
83                 goto out_free2;
84         }
85
86         close(fds[1]);
87         fds[1] = -1;
88
89         /*
90          * Read the errno value from the child, if the exec failed, or get 0 if
91          * the exec succeeded because the pipe fd was set as close-on-exec.
92          */
93         n = read(fds[0], &ret, sizeof(ret));
94         if (n == 0) {
95                 ret = pid;
96         } else {
97                 if (n < 0) {
98                         n = -errno;
99                         printk("run_helper : read on pipe failed, ret = %d\n",
100                                -n);
101                         ret = n;
102                         kill(pid, SIGKILL);
103                 }
104                 CATCH_EINTR(waitpid(pid, NULL, __WCLONE));
105         }
106
107 out_free2:
108         kfree(data.buf);
109 out_close:
110         if (fds[1] != -1)
111                 close(fds[1]);
112         close(fds[0]);
113 out_free:
114         free_stack(stack, 0);
115         return ret;
116 }
117
118 int run_helper_thread(int (*proc)(void *), void *arg, unsigned int flags,
119                       unsigned long *stack_out)
120 {
121         unsigned long stack, sp;
122         int pid, status, err;
123
124         stack = alloc_stack(0, __cant_sleep());
125         if (stack == 0)
126                 return -ENOMEM;
127
128         sp = stack + UM_KERN_PAGE_SIZE - sizeof(void *);
129         pid = clone(proc, (void *) sp, flags, arg);
130         if (pid < 0) {
131                 err = -errno;
132                 printk("run_helper_thread : clone failed, errno = %d\n",
133                        errno);
134                 return err;
135         }
136         if (stack_out == NULL) {
137                 CATCH_EINTR(pid = waitpid(pid, &status, __WCLONE));
138                 if (pid < 0) {
139                         err = -errno;
140                         printk("run_helper_thread - wait failed, errno = %d\n",
141                                errno);
142                         pid = err;
143                 }
144                 if (!WIFEXITED(status) || (WEXITSTATUS(status) != 0))
145                         printk("run_helper_thread - thread returned status "
146                                "0x%x\n", status);
147                 free_stack(stack, 0);
148         } else
149                 *stack_out = stack;
150         return pid;
151 }
152
153 int helper_wait(int pid, int nohang, char *pname)
154 {
155         int ret, status;
156         int wflags = __WCLONE;
157
158         if (nohang)
159                 wflags |= WNOHANG;
160
161         if (!pname)
162                 pname = "helper_wait";
163
164         CATCH_EINTR(ret = waitpid(pid, &status, wflags));
165         if (ret < 0) {
166                 printk(UM_KERN_ERR "%s : waitpid process %d failed, "
167                        "errno = %d\n", pname, pid, errno);
168                 return -errno;
169         } else if (nohang && ret == 0) {
170                 printk(UM_KERN_ERR "%s : process %d has not exited\n",
171                        pname, pid);
172                 return -ECHILD;
173         } else if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
174                 printk(UM_KERN_ERR "%s : process %d didn't exit with "
175                        "status 0\n", pname, pid);
176                 return -ECHILD;
177         } else
178                 return 0;
179 }