7b941abbcde09401404a73c5298b73611eea23da
[linux-2.6.git] / 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  */
17
18 #include <linux/file.h>
19 #include <linux/poll.h>
20 #include <linux/init.h>
21 #include <linux/fs.h>
22 #include <linux/sched.h>
23 #include <linux/kernel.h>
24 #include <linux/signal.h>
25 #include <linux/list.h>
26 #include <linux/anon_inodes.h>
27 #include <linux/signalfd.h>
28
29 struct signalfd_ctx {
30         struct list_head lnk;
31         wait_queue_head_t wqh;
32         sigset_t sigmask;
33         struct task_struct *tsk;
34 };
35
36 struct signalfd_lockctx {
37         struct task_struct *tsk;
38         unsigned long flags;
39 };
40
41 /*
42  * Tries to acquire the sighand lock. We do not increment the sighand
43  * use count, and we do not even pin the task struct, so we need to
44  * do it inside an RCU read lock, and we must be prepared for the
45  * ctx->tsk going to NULL (in signalfd_deliver()), and for the sighand
46  * being detached. We return 0 if the sighand has been detached, or
47  * 1 if we were able to pin the sighand lock.
48  */
49 static int signalfd_lock(struct signalfd_ctx *ctx, struct signalfd_lockctx *lk)
50 {
51         struct sighand_struct *sighand = NULL;
52
53         rcu_read_lock();
54         lk->tsk = rcu_dereference(ctx->tsk);
55         if (likely(lk->tsk != NULL))
56                 sighand = lock_task_sighand(lk->tsk, &lk->flags);
57         rcu_read_unlock();
58
59         if (sighand && !ctx->tsk) {
60                 unlock_task_sighand(lk->tsk, &lk->flags);
61                 sighand = NULL;
62         }
63
64         return sighand != NULL;
65 }
66
67 static void signalfd_unlock(struct signalfd_lockctx *lk)
68 {
69         unlock_task_sighand(lk->tsk, &lk->flags);
70 }
71
72 /*
73  * This must be called with the sighand lock held.
74  */
75 void signalfd_deliver(struct task_struct *tsk, int sig)
76 {
77         struct sighand_struct *sighand = tsk->sighand;
78         struct signalfd_ctx *ctx, *tmp;
79
80         BUG_ON(!sig);
81         list_for_each_entry_safe(ctx, tmp, &sighand->signalfd_list, lnk) {
82                 /*
83                  * We use a negative signal value as a way to broadcast that the
84                  * sighand has been orphaned, so that we can notify all the
85                  * listeners about this. Remember the ctx->sigmask is inverted,
86                  * so if the user is interested in a signal, that corresponding
87                  * bit will be zero.
88                  */
89                 if (sig < 0) {
90                         if (ctx->tsk == tsk) {
91                                 ctx->tsk = NULL;
92                                 list_del_init(&ctx->lnk);
93                                 wake_up(&ctx->wqh);
94                         }
95                 } else {
96                         if (!sigismember(&ctx->sigmask, sig))
97                                 wake_up(&ctx->wqh);
98                 }
99         }
100 }
101
102 static void signalfd_cleanup(struct signalfd_ctx *ctx)
103 {
104         struct signalfd_lockctx lk;
105
106         /*
107          * This is tricky. If the sighand is gone, we do not need to remove
108          * context from the list, the list itself won't be there anymore.
109          */
110         if (signalfd_lock(ctx, &lk)) {
111                 list_del(&ctx->lnk);
112                 signalfd_unlock(&lk);
113         }
114         kfree(ctx);
115 }
116
117 static int signalfd_release(struct inode *inode, struct file *file)
118 {
119         signalfd_cleanup(file->private_data);
120         return 0;
121 }
122
123 static unsigned int signalfd_poll(struct file *file, poll_table *wait)
124 {
125         struct signalfd_ctx *ctx = file->private_data;
126         unsigned int events = 0;
127         struct signalfd_lockctx lk;
128
129         poll_wait(file, &ctx->wqh, wait);
130
131         /*
132          * Let the caller get a POLLIN in this case, ala socket recv() when
133          * the peer disconnects.
134          */
135         if (signalfd_lock(ctx, &lk)) {
136                 if ((lk.tsk == current &&
137                      next_signal(&lk.tsk->pending, &ctx->sigmask) > 0) ||
138                     next_signal(&lk.tsk->signal->shared_pending,
139                                 &ctx->sigmask) > 0)
140                         events |= POLLIN;
141                 signalfd_unlock(&lk);
142         } else
143                 events |= POLLIN;
144
145         return events;
146 }
147
148 /*
149  * Copied from copy_siginfo_to_user() in kernel/signal.c
150  */
151 static int signalfd_copyinfo(struct signalfd_siginfo __user *uinfo,
152                              siginfo_t const *kinfo)
153 {
154         long err;
155
156         BUILD_BUG_ON(sizeof(struct signalfd_siginfo) != 128);
157
158         /*
159          * Unused memebers should be zero ...
160          */
161         err = __clear_user(uinfo, sizeof(*uinfo));
162
163         /*
164          * If you change siginfo_t structure, please be sure
165          * this code is fixed accordingly.
166          */
167         err |= __put_user(kinfo->si_signo, &uinfo->signo);
168         err |= __put_user(kinfo->si_errno, &uinfo->err);
169         err |= __put_user((short)kinfo->si_code, &uinfo->code);
170         switch (kinfo->si_code & __SI_MASK) {
171         case __SI_KILL:
172                 err |= __put_user(kinfo->si_pid, &uinfo->pid);
173                 err |= __put_user(kinfo->si_uid, &uinfo->uid);
174                 break;
175         case __SI_TIMER:
176                  err |= __put_user(kinfo->si_tid, &uinfo->tid);
177                  err |= __put_user(kinfo->si_overrun, &uinfo->overrun);
178                  err |= __put_user((long)kinfo->si_ptr, &uinfo->svptr);
179                 break;
180         case __SI_POLL:
181                 err |= __put_user(kinfo->si_band, &uinfo->band);
182                 err |= __put_user(kinfo->si_fd, &uinfo->fd);
183                 break;
184         case __SI_FAULT:
185                 err |= __put_user((long)kinfo->si_addr, &uinfo->addr);
186 #ifdef __ARCH_SI_TRAPNO
187                 err |= __put_user(kinfo->si_trapno, &uinfo->trapno);
188 #endif
189                 break;
190         case __SI_CHLD:
191                 err |= __put_user(kinfo->si_pid, &uinfo->pid);
192                 err |= __put_user(kinfo->si_uid, &uinfo->uid);
193                 err |= __put_user(kinfo->si_status, &uinfo->status);
194                 err |= __put_user(kinfo->si_utime, &uinfo->utime);
195                 err |= __put_user(kinfo->si_stime, &uinfo->stime);
196                 break;
197         case __SI_RT: /* This is not generated by the kernel as of now. */
198         case __SI_MESGQ: /* But this is */
199                 err |= __put_user(kinfo->si_pid, &uinfo->pid);
200                 err |= __put_user(kinfo->si_uid, &uinfo->uid);
201                 err |= __put_user((long)kinfo->si_ptr, &uinfo->svptr);
202                 break;
203         default: /* this is just in case for now ... */
204                 err |= __put_user(kinfo->si_pid, &uinfo->pid);
205                 err |= __put_user(kinfo->si_uid, &uinfo->uid);
206                 break;
207         }
208
209         return err ? -EFAULT: sizeof(*uinfo);
210 }
211
212 static ssize_t signalfd_dequeue(struct signalfd_ctx *ctx, siginfo_t *info,
213                                 int nonblock)
214 {
215         ssize_t ret;
216         struct signalfd_lockctx lk;
217         DECLARE_WAITQUEUE(wait, current);
218
219         if (!signalfd_lock(ctx, &lk))
220                 return 0;
221
222         ret = dequeue_signal(lk.tsk, &ctx->sigmask, info);
223         switch (ret) {
224         case 0:
225                 if (!nonblock)
226                         break;
227                 ret = -EAGAIN;
228         default:
229                 signalfd_unlock(&lk);
230                 return ret;
231         }
232
233         add_wait_queue(&ctx->wqh, &wait);
234         for (;;) {
235                 set_current_state(TASK_INTERRUPTIBLE);
236                 ret = dequeue_signal(lk.tsk, &ctx->sigmask, info);
237                 signalfd_unlock(&lk);
238                 if (ret != 0)
239                         break;
240                 if (signal_pending(current)) {
241                         ret = -ERESTARTSYS;
242                         break;
243                 }
244                 schedule();
245                 ret = signalfd_lock(ctx, &lk);
246                 if (unlikely(!ret)) {
247                         /*
248                          * Let the caller read zero byte, ala socket
249                          * recv() when the peer disconnect. This test
250                          * must be done before doing a dequeue_signal(),
251                          * because if the sighand has been orphaned,
252                          * the dequeue_signal() call is going to crash
253                          * because ->sighand will be long gone.
254                          */
255                          break;
256                 }
257         }
258
259         remove_wait_queue(&ctx->wqh, &wait);
260         __set_current_state(TASK_RUNNING);
261
262         return ret;
263 }
264
265 /*
266  * Returns either the size of a "struct signalfd_siginfo", or zero if the
267  * sighand we are attached to, has been orphaned. The "count" parameter
268  * must be at least the size of a "struct signalfd_siginfo".
269  */
270 static ssize_t signalfd_read(struct file *file, char __user *buf, size_t count,
271                              loff_t *ppos)
272 {
273         struct signalfd_ctx *ctx = file->private_data;
274         struct signalfd_siginfo __user *siginfo;
275         int nonblock = file->f_flags & O_NONBLOCK;
276         ssize_t ret, total = 0;
277         siginfo_t info;
278
279         count /= sizeof(struct signalfd_siginfo);
280         if (!count)
281                 return -EINVAL;
282
283         siginfo = (struct signalfd_siginfo __user *) buf;
284
285         do {
286                 ret = signalfd_dequeue(ctx, &info, nonblock);
287                 if (unlikely(ret <= 0))
288                         break;
289                 ret = signalfd_copyinfo(siginfo, &info);
290                 if (ret < 0)
291                         break;
292                 siginfo++;
293                 total += ret;
294                 nonblock = 1;
295         } while (--count);
296
297         return total ? total : ret;
298 }
299
300 static const struct file_operations signalfd_fops = {
301         .release        = signalfd_release,
302         .poll           = signalfd_poll,
303         .read           = signalfd_read,
304 };
305
306 /*
307  * Create a file descriptor that is associated with our signal
308  * state. We can pass it around to others if we want to, but
309  * it will always be _our_ signal state.
310  */
311 asmlinkage long sys_signalfd(int ufd, sigset_t __user *user_mask, size_t sizemask)
312 {
313         int error;
314         sigset_t sigmask;
315         struct signalfd_ctx *ctx;
316         struct sighand_struct *sighand;
317         struct file *file;
318         struct inode *inode;
319         struct signalfd_lockctx lk;
320
321         if (sizemask != sizeof(sigset_t) ||
322             copy_from_user(&sigmask, user_mask, sizeof(sigmask)))
323                 return -EINVAL;
324         sigdelsetmask(&sigmask, sigmask(SIGKILL) | sigmask(SIGSTOP));
325         signotset(&sigmask);
326
327         if (ufd == -1) {
328                 ctx = kmalloc(sizeof(*ctx), GFP_KERNEL);
329                 if (!ctx)
330                         return -ENOMEM;
331
332                 init_waitqueue_head(&ctx->wqh);
333                 ctx->sigmask = sigmask;
334                 ctx->tsk = current;
335
336                 sighand = current->sighand;
337                 /*
338                  * Add this fd to the list of signal listeners.
339                  */
340                 spin_lock_irq(&sighand->siglock);
341                 list_add_tail(&ctx->lnk, &sighand->signalfd_list);
342                 spin_unlock_irq(&sighand->siglock);
343
344                 /*
345                  * When we call this, the initialization must be complete, since
346                  * anon_inode_getfd() will install the fd.
347                  */
348                 error = anon_inode_getfd(&ufd, &inode, &file, "[signalfd]",
349                                          &signalfd_fops, ctx);
350                 if (error)
351                         goto err_fdalloc;
352         } else {
353                 file = fget(ufd);
354                 if (!file)
355                         return -EBADF;
356                 ctx = file->private_data;
357                 if (file->f_op != &signalfd_fops) {
358                         fput(file);
359                         return -EINVAL;
360                 }
361                 /*
362                  * We need to be prepared of the fact that the sighand this fd
363                  * is attached to, has been detched. In that case signalfd_lock()
364                  * will return 0, and we'll just skip setting the new mask.
365                  */
366                 if (signalfd_lock(ctx, &lk)) {
367                         ctx->sigmask = sigmask;
368                         signalfd_unlock(&lk);
369                 }
370                 wake_up(&ctx->wqh);
371                 fput(file);
372         }
373
374         return ufd;
375
376 err_fdalloc:
377         signalfd_cleanup(ctx);
378         return error;
379 }
380