Merge branch 'for-linus' of ssh://master.kernel.org/pub/scm/linux/kernel/git/hskinnem...
[linux-2.6] / fs / signalfd.c
1 /*
2  *  fs/signalfd.c
3  *
4  *  Copyright (C) 2003  Linus Torvalds
5  *
6  *  Mon Mar 5, 2007: Davide Libenzi <davidel@xmailserver.org>
7  *      Changed ->read() to return a siginfo strcture instead of signal number.
8  *      Fixed locking in ->poll().
9  *      Added sighand-detach notification.
10  *      Added fd re-use in sys_signalfd() syscall.
11  *      Now using anonymous inode source.
12  *      Thanks to Oleg Nesterov for useful code review and suggestions.
13  *      More comments and suggestions from Arnd Bergmann.
14  *  Sat May 19, 2007: Davi E. M. Arnaut <davi@haxent.com.br>
15  *      Retrieve multiple signals with one read() call
16  *  Sun Jul 15, 2007: Davide Libenzi <davidel@xmailserver.org>
17  *      Attach to the sighand only during read() and poll().
18  */
19
20 #include <linux/file.h>
21 #include <linux/poll.h>
22 #include <linux/init.h>
23 #include <linux/fs.h>
24 #include <linux/sched.h>
25 #include <linux/kernel.h>
26 #include <linux/signal.h>
27 #include <linux/list.h>
28 #include <linux/anon_inodes.h>
29 #include <linux/signalfd.h>
30
31 struct signalfd_ctx {
32         sigset_t sigmask;
33 };
34
35 static int signalfd_release(struct inode *inode, struct file *file)
36 {
37         kfree(file->private_data);
38         return 0;
39 }
40
41 static unsigned int signalfd_poll(struct file *file, poll_table *wait)
42 {
43         struct signalfd_ctx *ctx = file->private_data;
44         unsigned int events = 0;
45
46         poll_wait(file, &current->sighand->signalfd_wqh, wait);
47
48         spin_lock_irq(&current->sighand->siglock);
49         if (next_signal(&current->pending, &ctx->sigmask) ||
50             next_signal(&current->signal->shared_pending,
51                         &ctx->sigmask))
52                 events |= POLLIN;
53         spin_unlock_irq(&current->sighand->siglock);
54
55         return events;
56 }
57
58 /*
59  * Copied from copy_siginfo_to_user() in kernel/signal.c
60  */
61 static int signalfd_copyinfo(struct signalfd_siginfo __user *uinfo,
62                              siginfo_t const *kinfo)
63 {
64         long err;
65
66         BUILD_BUG_ON(sizeof(struct signalfd_siginfo) != 128);
67
68         /*
69          * Unused memebers should be zero ...
70          */
71         err = __clear_user(uinfo, sizeof(*uinfo));
72
73         /*
74          * If you change siginfo_t structure, please be sure
75          * this code is fixed accordingly.
76          */
77         err |= __put_user(kinfo->si_signo, &uinfo->ssi_signo);
78         err |= __put_user(kinfo->si_errno, &uinfo->ssi_errno);
79         err |= __put_user((short) kinfo->si_code, &uinfo->ssi_code);
80         switch (kinfo->si_code & __SI_MASK) {
81         case __SI_KILL:
82                 err |= __put_user(kinfo->si_pid, &uinfo->ssi_pid);
83                 err |= __put_user(kinfo->si_uid, &uinfo->ssi_uid);
84                 break;
85         case __SI_TIMER:
86                  err |= __put_user(kinfo->si_tid, &uinfo->ssi_tid);
87                  err |= __put_user(kinfo->si_overrun, &uinfo->ssi_overrun);
88                  err |= __put_user((long) kinfo->si_ptr, &uinfo->ssi_ptr);
89                 break;
90         case __SI_POLL:
91                 err |= __put_user(kinfo->si_band, &uinfo->ssi_band);
92                 err |= __put_user(kinfo->si_fd, &uinfo->ssi_fd);
93                 break;
94         case __SI_FAULT:
95                 err |= __put_user((long) kinfo->si_addr, &uinfo->ssi_addr);
96 #ifdef __ARCH_SI_TRAPNO
97                 err |= __put_user(kinfo->si_trapno, &uinfo->ssi_trapno);
98 #endif
99                 break;
100         case __SI_CHLD:
101                 err |= __put_user(kinfo->si_pid, &uinfo->ssi_pid);
102                 err |= __put_user(kinfo->si_uid, &uinfo->ssi_uid);
103                 err |= __put_user(kinfo->si_status, &uinfo->ssi_status);
104                 err |= __put_user(kinfo->si_utime, &uinfo->ssi_utime);
105                 err |= __put_user(kinfo->si_stime, &uinfo->ssi_stime);
106                 break;
107         case __SI_RT: /* This is not generated by the kernel as of now. */
108         case __SI_MESGQ: /* But this is */
109                 err |= __put_user(kinfo->si_pid, &uinfo->ssi_pid);
110                 err |= __put_user(kinfo->si_uid, &uinfo->ssi_uid);
111                 err |= __put_user((long) kinfo->si_ptr, &uinfo->ssi_ptr);
112                 break;
113         default: /* this is just in case for now ... */
114                 err |= __put_user(kinfo->si_pid, &uinfo->ssi_pid);
115                 err |= __put_user(kinfo->si_uid, &uinfo->ssi_uid);
116                 break;
117         }
118
119         return err ? -EFAULT: sizeof(*uinfo);
120 }
121
122 static ssize_t signalfd_dequeue(struct signalfd_ctx *ctx, siginfo_t *info,
123                                 int nonblock)
124 {
125         ssize_t ret;
126         DECLARE_WAITQUEUE(wait, current);
127
128         spin_lock_irq(&current->sighand->siglock);
129         ret = dequeue_signal(current, &ctx->sigmask, info);
130         switch (ret) {
131         case 0:
132                 if (!nonblock)
133                         break;
134                 ret = -EAGAIN;
135         default:
136                 spin_unlock_irq(&current->sighand->siglock);
137                 return ret;
138         }
139
140         add_wait_queue(&current->sighand->signalfd_wqh, &wait);
141         for (;;) {
142                 set_current_state(TASK_INTERRUPTIBLE);
143                 ret = dequeue_signal(current, &ctx->sigmask, info);
144                 if (ret != 0)
145                         break;
146                 if (signal_pending(current)) {
147                         ret = -ERESTARTSYS;
148                         break;
149                 }
150                 spin_unlock_irq(&current->sighand->siglock);
151                 schedule();
152                 spin_lock_irq(&current->sighand->siglock);
153         }
154         spin_unlock_irq(&current->sighand->siglock);
155
156         remove_wait_queue(&current->sighand->signalfd_wqh, &wait);
157         __set_current_state(TASK_RUNNING);
158
159         return ret;
160 }
161
162 /*
163  * Returns a multiple of the size of a "struct signalfd_siginfo", or a negative
164  * error code. The "count" parameter must be at least the size of a
165  * "struct signalfd_siginfo".
166  */
167 static ssize_t signalfd_read(struct file *file, char __user *buf, size_t count,
168                              loff_t *ppos)
169 {
170         struct signalfd_ctx *ctx = file->private_data;
171         struct signalfd_siginfo __user *siginfo;
172         int nonblock = file->f_flags & O_NONBLOCK;
173         ssize_t ret, total = 0;
174         siginfo_t info;
175
176         count /= sizeof(struct signalfd_siginfo);
177         if (!count)
178                 return -EINVAL;
179
180         siginfo = (struct signalfd_siginfo __user *) buf;
181         do {
182                 ret = signalfd_dequeue(ctx, &info, nonblock);
183                 if (unlikely(ret <= 0))
184                         break;
185                 ret = signalfd_copyinfo(siginfo, &info);
186                 if (ret < 0)
187                         break;
188                 siginfo++;
189                 total += ret;
190                 nonblock = 1;
191         } while (--count);
192
193         return total ? total: ret;
194 }
195
196 static const struct file_operations signalfd_fops = {
197         .release        = signalfd_release,
198         .poll           = signalfd_poll,
199         .read           = signalfd_read,
200 };
201
202 asmlinkage long sys_signalfd(int ufd, sigset_t __user *user_mask, size_t sizemask)
203 {
204         int error;
205         sigset_t sigmask;
206         struct signalfd_ctx *ctx;
207         struct file *file;
208         struct inode *inode;
209
210         if (sizemask != sizeof(sigset_t) ||
211             copy_from_user(&sigmask, user_mask, sizeof(sigmask)))
212                 return -EINVAL;
213         sigdelsetmask(&sigmask, sigmask(SIGKILL) | sigmask(SIGSTOP));
214         signotset(&sigmask);
215
216         if (ufd == -1) {
217                 ctx = kmalloc(sizeof(*ctx), GFP_KERNEL);
218                 if (!ctx)
219                         return -ENOMEM;
220
221                 ctx->sigmask = sigmask;
222
223                 /*
224                  * When we call this, the initialization must be complete, since
225                  * anon_inode_getfd() will install the fd.
226                  */
227                 error = anon_inode_getfd(&ufd, &inode, &file, "[signalfd]",
228                                          &signalfd_fops, ctx);
229                 if (error)
230                         goto err_fdalloc;
231         } else {
232                 file = fget(ufd);
233                 if (!file)
234                         return -EBADF;
235                 ctx = file->private_data;
236                 if (file->f_op != &signalfd_fops) {
237                         fput(file);
238                         return -EINVAL;
239                 }
240                 spin_lock_irq(&current->sighand->siglock);
241                 ctx->sigmask = sigmask;
242                 spin_unlock_irq(&current->sighand->siglock);
243
244                 wake_up(&current->sighand->signalfd_wqh);
245                 fput(file);
246         }
247
248         return ufd;
249
250 err_fdalloc:
251         kfree(ctx);
252         return error;
253 }
254