1
2
3
4
5
6
7
8
9
10
11
12
13
14#include <linux/init.h>
15#include <linux/kernel.h>
16#include <linux/sched.h>
17#include <linux/fs.h>
18#include <linux/file.h>
19#include <linux/signal.h>
20#include <linux/errno.h>
21#include <linux/mm.h>
22#include <linux/slab.h>
23#include <linux/poll.h>
24#include <linux/string.h>
25#include <linux/list.h>
26#include <linux/hash.h>
27#include <linux/spinlock.h>
28#include <linux/syscalls.h>
29#include <linux/rbtree.h>
30#include <linux/wait.h>
31#include <linux/eventpoll.h>
32#include <linux/mount.h>
33#include <linux/bitops.h>
34#include <linux/mutex.h>
35#include <linux/anon_inodes.h>
36#include <linux/device.h>
37#include <asm/uaccess.h>
38#include <asm/io.h>
39#include <asm/mman.h>
40#include <linux/atomic.h>
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91#define EP_PRIVATE_BITS (EPOLLWAKEUP | EPOLLONESHOT | EPOLLET)
92
93
94#define EP_MAX_NESTS 4
95
96#define EP_MAX_EVENTS (INT_MAX / sizeof(struct epoll_event))
97
98#define EP_UNACTIVE_PTR ((void *) -1L)
99
100#define EP_ITEM_COST (sizeof(struct epitem) + sizeof(struct eppoll_entry))
101
102struct epoll_filefd {
103 struct file *file;
104 int fd;
105};
106
107
108
109
110
111struct nested_call_node {
112 struct list_head llink;
113 void *cookie;
114 void *ctx;
115};
116
117
118
119
120
121struct nested_calls {
122 struct list_head tasks_call_list;
123 spinlock_t lock;
124};
125
126
127
128
129
130struct epitem {
131
132 struct rb_node rbn;
133
134
135 struct list_head rdllink;
136
137
138
139
140
141 struct epitem *next;
142
143
144 struct epoll_filefd ffd;
145
146
147 int nwait;
148
149
150 struct list_head pwqlist;
151
152
153 struct eventpoll *ep;
154
155
156 struct list_head fllink;
157
158
159 struct wakeup_source *ws;
160
161
162 struct epoll_event event;
163};
164
165
166
167
168
169
170struct eventpoll {
171
172 spinlock_t lock;
173
174
175
176
177
178
179
180 struct mutex mtx;
181
182
183 wait_queue_head_t wq;
184
185
186 wait_queue_head_t poll_wait;
187
188
189 struct list_head rdllist;
190
191
192 struct rb_root rbr;
193
194
195
196
197
198
199 struct epitem *ovflist;
200
201
202 struct wakeup_source *ws;
203
204
205 struct user_struct *user;
206
207 struct file *file;
208
209
210 int visited;
211 struct list_head visited_list_link;
212};
213
214
215struct eppoll_entry {
216
217 struct list_head llink;
218
219
220 struct epitem *base;
221
222
223
224
225
226 wait_queue_t wait;
227
228
229 wait_queue_head_t *whead;
230};
231
232
233struct ep_pqueue {
234 poll_table pt;
235 struct epitem *epi;
236};
237
238
239struct ep_send_events_data {
240 int maxevents;
241 struct epoll_event __user *events;
242};
243
244
245
246
247
248static long max_user_watches __read_mostly;
249
250
251
252
253static DEFINE_MUTEX(epmutex);
254
255
256static struct nested_calls poll_loop_ncalls;
257
258
259static struct nested_calls poll_safewake_ncalls;
260
261
262static struct nested_calls poll_readywalk_ncalls;
263
264
265static struct kmem_cache *epi_cache __read_mostly;
266
267
268static struct kmem_cache *pwq_cache __read_mostly;
269
270
271static LIST_HEAD(visited_list);
272
273
274
275
276
277static LIST_HEAD(tfile_check_list);
278
279#ifdef CONFIG_SYSCTL
280
281#include <linux/sysctl.h>
282
283static long zero;
284static long long_max = LONG_MAX;
285
286ctl_table epoll_table[] = {
287 {
288 .procname = "max_user_watches",
289 .data = &max_user_watches,
290 .maxlen = sizeof(max_user_watches),
291 .mode = 0644,
292 .proc_handler = proc_doulongvec_minmax,
293 .extra1 = &zero,
294 .extra2 = &long_max,
295 },
296 { }
297};
298#endif
299
300static const struct file_operations eventpoll_fops;
301
302static inline int is_file_epoll(struct file *f)
303{
304 return f->f_op == &eventpoll_fops;
305}
306
307
308static inline void ep_set_ffd(struct epoll_filefd *ffd,
309 struct file *file, int fd)
310{
311 ffd->file = file;
312 ffd->fd = fd;
313}
314
315
316static inline int ep_cmp_ffd(struct epoll_filefd *p1,
317 struct epoll_filefd *p2)
318{
319 return (p1->file > p2->file ? +1:
320 (p1->file < p2->file ? -1 : p1->fd - p2->fd));
321}
322
323
324static inline int ep_is_linked(struct list_head *p)
325{
326 return !list_empty(p);
327}
328
329static inline struct eppoll_entry *ep_pwq_from_wait(wait_queue_t *p)
330{
331 return container_of(p, struct eppoll_entry, wait);
332}
333
334
335static inline struct epitem *ep_item_from_wait(wait_queue_t *p)
336{
337 return container_of(p, struct eppoll_entry, wait)->base;
338}
339
340
341static inline struct epitem *ep_item_from_epqueue(poll_table *p)
342{
343 return container_of(p, struct ep_pqueue, pt)->epi;
344}
345
346
347static inline int ep_op_has_event(int op)
348{
349 return op != EPOLL_CTL_DEL;
350}
351
352
353static void ep_nested_calls_init(struct nested_calls *ncalls)
354{
355 INIT_LIST_HEAD(&ncalls->tasks_call_list);
356 spin_lock_init(&ncalls->lock);
357}
358
359
360
361
362
363
364
365
366
367static inline int ep_events_available(struct eventpoll *ep)
368{
369 return !list_empty(&ep->rdllist) || ep->ovflist != EP_UNACTIVE_PTR;
370}
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388static int ep_call_nested(struct nested_calls *ncalls, int max_nests,
389 int (*nproc)(void *, void *, int), void *priv,
390 void *cookie, void *ctx)
391{
392 int error, call_nests = 0;
393 unsigned long flags;
394 struct list_head *lsthead = &ncalls->tasks_call_list;
395 struct nested_call_node *tncur;
396 struct nested_call_node tnode;
397
398 spin_lock_irqsave(&ncalls->lock, flags);
399
400
401
402
403
404
405 list_for_each_entry(tncur, lsthead, llink) {
406 if (tncur->ctx == ctx &&
407 (tncur->cookie == cookie || ++call_nests > max_nests)) {
408
409
410
411
412 error = -1;
413 goto out_unlock;
414 }
415 }
416
417
418 tnode.ctx = ctx;
419 tnode.cookie = cookie;
420 list_add(&tnode.llink, lsthead);
421
422 spin_unlock_irqrestore(&ncalls->lock, flags);
423
424
425 error = (*nproc)(priv, cookie, call_nests);
426
427
428 spin_lock_irqsave(&ncalls->lock, flags);
429 list_del(&tnode.llink);
430out_unlock:
431 spin_unlock_irqrestore(&ncalls->lock, flags);
432
433 return error;
434}
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461#ifdef CONFIG_DEBUG_LOCK_ALLOC
462static inline void ep_wake_up_nested(wait_queue_head_t *wqueue,
463 unsigned long events, int subclass)
464{
465 unsigned long flags;
466
467 spin_lock_irqsave_nested(&wqueue->lock, flags, subclass);
468 wake_up_locked_poll(wqueue, events);
469 spin_unlock_irqrestore(&wqueue->lock, flags);
470}
471#else
472static inline void ep_wake_up_nested(wait_queue_head_t *wqueue,
473 unsigned long events, int subclass)
474{
475 wake_up_poll(wqueue, events);
476}
477#endif
478
479static int ep_poll_wakeup_proc(void *priv, void *cookie, int call_nests)
480{
481 ep_wake_up_nested((wait_queue_head_t *) cookie, POLLIN,
482 1 + call_nests);
483 return 0;
484}
485
486
487
488
489
490
491
492
493
494
495
496static void ep_poll_safewake(wait_queue_head_t *wq)
497{
498 int this_cpu = get_cpu();
499
500 ep_call_nested(&poll_safewake_ncalls, EP_MAX_NESTS,
501 ep_poll_wakeup_proc, NULL, wq, (void *) (long) this_cpu);
502
503 put_cpu();
504}
505
506static void ep_remove_wait_queue(struct eppoll_entry *pwq)
507{
508 wait_queue_head_t *whead;
509
510 rcu_read_lock();
511
512 whead = rcu_dereference(pwq->whead);
513 if (whead)
514 remove_wait_queue(whead, &pwq->wait);
515 rcu_read_unlock();
516}
517
518
519
520
521
522
523static void ep_unregister_pollwait(struct eventpoll *ep, struct epitem *epi)
524{
525 struct list_head *lsthead = &epi->pwqlist;
526 struct eppoll_entry *pwq;
527
528 while (!list_empty(lsthead)) {
529 pwq = list_first_entry(lsthead, struct eppoll_entry, llink);
530
531 list_del(&pwq->llink);
532 ep_remove_wait_queue(pwq);
533 kmem_cache_free(pwq_cache, pwq);
534 }
535}
536
537
538
539
540
541
542
543
544
545
546
547
548
549static int ep_scan_ready_list(struct eventpoll *ep,
550 int (*sproc)(struct eventpoll *,
551 struct list_head *, void *),
552 void *priv,
553 int depth)
554{
555 int error, pwake = 0;
556 unsigned long flags;
557 struct epitem *epi, *nepi;
558 LIST_HEAD(txlist);
559
560
561
562
563
564 mutex_lock_nested(&ep->mtx, depth);
565
566
567
568
569
570
571
572
573
574 spin_lock_irqsave(&ep->lock, flags);
575 list_splice_init(&ep->rdllist, &txlist);
576 ep->ovflist = NULL;
577 spin_unlock_irqrestore(&ep->lock, flags);
578
579
580
581
582 error = (*sproc)(ep, &txlist, priv);
583
584 spin_lock_irqsave(&ep->lock, flags);
585
586
587
588
589
590 for (nepi = ep->ovflist; (epi = nepi) != NULL;
591 nepi = epi->next, epi->next = EP_UNACTIVE_PTR) {
592
593
594
595
596
597
598 if (!ep_is_linked(&epi->rdllink)) {
599 list_add_tail(&epi->rdllink, &ep->rdllist);
600 __pm_stay_awake(epi->ws);
601 }
602 }
603
604
605
606
607
608 ep->ovflist = EP_UNACTIVE_PTR;
609
610
611
612
613 list_splice(&txlist, &ep->rdllist);
614 __pm_relax(ep->ws);
615
616 if (!list_empty(&ep->rdllist)) {
617
618
619
620
621 if (waitqueue_active(&ep->wq))
622 wake_up_locked(&ep->wq);
623 if (waitqueue_active(&ep->poll_wait))
624 pwake++;
625 }
626 spin_unlock_irqrestore(&ep->lock, flags);
627
628 mutex_unlock(&ep->mtx);
629
630
631 if (pwake)
632 ep_poll_safewake(&ep->poll_wait);
633
634 return error;
635}
636
637
638
639
640
641static int ep_remove(struct eventpoll *ep, struct epitem *epi)
642{
643 unsigned long flags;
644 struct file *file = epi->ffd.file;
645
646
647
648
649
650
651
652
653
654 ep_unregister_pollwait(ep, epi);
655
656
657 spin_lock(&file->f_lock);
658 if (ep_is_linked(&epi->fllink))
659 list_del_init(&epi->fllink);
660 spin_unlock(&file->f_lock);
661
662 rb_erase(&epi->rbn, &ep->rbr);
663
664 spin_lock_irqsave(&ep->lock, flags);
665 if (ep_is_linked(&epi->rdllink))
666 list_del_init(&epi->rdllink);
667 spin_unlock_irqrestore(&ep->lock, flags);
668
669 wakeup_source_unregister(epi->ws);
670
671
672 kmem_cache_free(epi_cache, epi);
673
674 atomic_long_dec(&ep->user->epoll_watches);
675
676 return 0;
677}
678
679static void ep_free(struct eventpoll *ep)
680{
681 struct rb_node *rbp;
682 struct epitem *epi;
683
684
685 if (waitqueue_active(&ep->poll_wait))
686 ep_poll_safewake(&ep->poll_wait);
687
688
689
690
691
692
693
694
695
696 mutex_lock(&epmutex);
697
698
699
700
701 for (rbp = rb_first(&ep->rbr); rbp; rbp = rb_next(rbp)) {
702 epi = rb_entry(rbp, struct epitem, rbn);
703
704 ep_unregister_pollwait(ep, epi);
705 }
706
707
708
709
710
711
712
713 while ((rbp = rb_first(&ep->rbr)) != NULL) {
714 epi = rb_entry(rbp, struct epitem, rbn);
715 ep_remove(ep, epi);
716 }
717
718 mutex_unlock(&epmutex);
719 mutex_destroy(&ep->mtx);
720 free_uid(ep->user);
721 wakeup_source_unregister(ep->ws);
722 kfree(ep);
723}
724
725static int ep_eventpoll_release(struct inode *inode, struct file *file)
726{
727 struct eventpoll *ep = file->private_data;
728
729 if (ep)
730 ep_free(ep);
731
732 return 0;
733}
734
735static int ep_read_events_proc(struct eventpoll *ep, struct list_head *head,
736 void *priv)
737{
738 struct epitem *epi, *tmp;
739 poll_table pt;
740
741 init_poll_funcptr(&pt, NULL);
742 list_for_each_entry_safe(epi, tmp, head, rdllink) {
743 pt._key = epi->event.events;
744 if (epi->ffd.file->f_op->poll(epi->ffd.file, &pt) &
745 epi->event.events)
746 return POLLIN | POLLRDNORM;
747 else {
748
749
750
751
752
753 __pm_relax(epi->ws);
754 list_del_init(&epi->rdllink);
755 }
756 }
757
758 return 0;
759}
760
761static int ep_poll_readyevents_proc(void *priv, void *cookie, int call_nests)
762{
763 return ep_scan_ready_list(priv, ep_read_events_proc, NULL, call_nests + 1);
764}
765
766static unsigned int ep_eventpoll_poll(struct file *file, poll_table *wait)
767{
768 int pollflags;
769 struct eventpoll *ep = file->private_data;
770
771
772 poll_wait(file, &ep->poll_wait, wait);
773
774
775
776
777
778
779
780 pollflags = ep_call_nested(&poll_readywalk_ncalls, EP_MAX_NESTS,
781 ep_poll_readyevents_proc, ep, ep, current);
782
783 return pollflags != -1 ? pollflags : 0;
784}
785
786
787static const struct file_operations eventpoll_fops = {
788 .release = ep_eventpoll_release,
789 .poll = ep_eventpoll_poll,
790 .llseek = noop_llseek,
791};
792
793
794
795
796
797
798void eventpoll_release_file(struct file *file)
799{
800 struct list_head *lsthead = &file->f_ep_links;
801 struct eventpoll *ep;
802 struct epitem *epi;
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817 mutex_lock(&epmutex);
818
819 while (!list_empty(lsthead)) {
820 epi = list_first_entry(lsthead, struct epitem, fllink);
821
822 ep = epi->ep;
823 list_del_init(&epi->fllink);
824 mutex_lock_nested(&ep->mtx, 0);
825 ep_remove(ep, epi);
826 mutex_unlock(&ep->mtx);
827 }
828
829 mutex_unlock(&epmutex);
830}
831
832static int ep_alloc(struct eventpoll **pep)
833{
834 int error;
835 struct user_struct *user;
836 struct eventpoll *ep;
837
838 user = get_current_user();
839 error = -ENOMEM;
840 ep = kzalloc(sizeof(*ep), GFP_KERNEL);
841 if (unlikely(!ep))
842 goto free_uid;
843
844 spin_lock_init(&ep->lock);
845 mutex_init(&ep->mtx);
846 init_waitqueue_head(&ep->wq);
847 init_waitqueue_head(&ep->poll_wait);
848 INIT_LIST_HEAD(&ep->rdllist);
849 ep->rbr = RB_ROOT;
850 ep->ovflist = EP_UNACTIVE_PTR;
851 ep->user = user;
852
853 *pep = ep;
854
855 return 0;
856
857free_uid:
858 free_uid(user);
859 return error;
860}
861
862
863
864
865
866
867static struct epitem *ep_find(struct eventpoll *ep, struct file *file, int fd)
868{
869 int kcmp;
870 struct rb_node *rbp;
871 struct epitem *epi, *epir = NULL;
872 struct epoll_filefd ffd;
873
874 ep_set_ffd(&ffd, file, fd);
875 for (rbp = ep->rbr.rb_node; rbp; ) {
876 epi = rb_entry(rbp, struct epitem, rbn);
877 kcmp = ep_cmp_ffd(&ffd, &epi->ffd);
878 if (kcmp > 0)
879 rbp = rbp->rb_right;
880 else if (kcmp < 0)
881 rbp = rbp->rb_left;
882 else {
883 epir = epi;
884 break;
885 }
886 }
887
888 return epir;
889}
890
891
892
893
894
895
896static int ep_poll_callback(wait_queue_t *wait, unsigned mode, int sync, void *key)
897{
898 int pwake = 0;
899 unsigned long flags;
900 struct epitem *epi = ep_item_from_wait(wait);
901 struct eventpoll *ep = epi->ep;
902
903 if ((unsigned long)key & POLLFREE) {
904 ep_pwq_from_wait(wait)->whead = NULL;
905
906
907
908
909
910
911 list_del_init(&wait->task_list);
912 }
913
914 spin_lock_irqsave(&ep->lock, flags);
915
916
917
918
919
920
921
922 if (!(epi->event.events & ~EP_PRIVATE_BITS))
923 goto out_unlock;
924
925
926
927
928
929
930
931 if (key && !((unsigned long) key & epi->event.events))
932 goto out_unlock;
933
934
935
936
937
938
939
940 if (unlikely(ep->ovflist != EP_UNACTIVE_PTR)) {
941 if (epi->next == EP_UNACTIVE_PTR) {
942 epi->next = ep->ovflist;
943 ep->ovflist = epi;
944 if (epi->ws) {
945
946
947
948
949 __pm_stay_awake(ep->ws);
950 }
951
952 }
953 goto out_unlock;
954 }
955
956
957 if (!ep_is_linked(&epi->rdllink)) {
958 list_add_tail(&epi->rdllink, &ep->rdllist);
959 __pm_stay_awake(epi->ws);
960 }
961
962
963
964
965
966 if (waitqueue_active(&ep->wq))
967 wake_up_locked(&ep->wq);
968 if (waitqueue_active(&ep->poll_wait))
969 pwake++;
970
971out_unlock:
972 spin_unlock_irqrestore(&ep->lock, flags);
973
974
975 if (pwake)
976 ep_poll_safewake(&ep->poll_wait);
977
978 return 1;
979}
980
981
982
983
984
985static void ep_ptable_queue_proc(struct file *file, wait_queue_head_t *whead,
986 poll_table *pt)
987{
988 struct epitem *epi = ep_item_from_epqueue(pt);
989 struct eppoll_entry *pwq;
990
991 if (epi->nwait >= 0 && (pwq = kmem_cache_alloc(pwq_cache, GFP_KERNEL))) {
992 init_waitqueue_func_entry(&pwq->wait, ep_poll_callback);
993 pwq->whead = whead;
994 pwq->base = epi;
995 add_wait_queue(whead, &pwq->wait);
996 list_add_tail(&pwq->llink, &epi->pwqlist);
997 epi->nwait++;
998 } else {
999
1000 epi->nwait = -1;
1001 }
1002}
1003
1004static void ep_rbtree_insert(struct eventpoll *ep, struct epitem *epi)
1005{
1006 int kcmp;
1007 struct rb_node **p = &ep->rbr.rb_node, *parent = NULL;
1008 struct epitem *epic;
1009
1010 while (*p) {
1011 parent = *p;
1012 epic = rb_entry(parent, struct epitem, rbn);
1013 kcmp = ep_cmp_ffd(&epi->ffd, &epic->ffd);
1014 if (kcmp > 0)
1015 p = &parent->rb_right;
1016 else
1017 p = &parent->rb_left;
1018 }
1019 rb_link_node(&epi->rbn, parent, p);
1020 rb_insert_color(&epi->rbn, &ep->rbr);
1021}
1022
1023
1024
1025#define PATH_ARR_SIZE 5
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037static const int path_limits[PATH_ARR_SIZE] = { 1000, 500, 100, 50, 10 };
1038static int path_count[PATH_ARR_SIZE];
1039
1040static int path_count_inc(int nests)
1041{
1042
1043 if (nests == 0)
1044 return 0;
1045
1046 if (++path_count[nests] > path_limits[nests])
1047 return -1;
1048 return 0;
1049}
1050
1051static void path_count_init(void)
1052{
1053 int i;
1054
1055 for (i = 0; i < PATH_ARR_SIZE; i++)
1056 path_count[i] = 0;
1057}
1058
1059static int reverse_path_check_proc(void *priv, void *cookie, int call_nests)
1060{
1061 int error = 0;
1062 struct file *file = priv;
1063 struct file *child_file;
1064 struct epitem *epi;
1065
1066 list_for_each_entry(epi, &file->f_ep_links, fllink) {
1067 child_file = epi->ep->file;
1068 if (is_file_epoll(child_file)) {
1069 if (list_empty(&child_file->f_ep_links)) {
1070 if (path_count_inc(call_nests)) {
1071 error = -1;
1072 break;
1073 }
1074 } else {
1075 error = ep_call_nested(&poll_loop_ncalls,
1076 EP_MAX_NESTS,
1077 reverse_path_check_proc,
1078 child_file, child_file,
1079 current);
1080 }
1081 if (error != 0)
1082 break;
1083 } else {
1084 printk(KERN_ERR "reverse_path_check_proc: "
1085 "file is not an ep!\n");
1086 }
1087 }
1088 return error;
1089}
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101static int reverse_path_check(void)
1102{
1103 int error = 0;
1104 struct file *current_file;
1105
1106
1107 list_for_each_entry(current_file, &tfile_check_list, f_tfile_llink) {
1108 path_count_init();
1109 error = ep_call_nested(&poll_loop_ncalls, EP_MAX_NESTS,
1110 reverse_path_check_proc, current_file,
1111 current_file, current);
1112 if (error)
1113 break;
1114 }
1115 return error;
1116}
1117
1118static int ep_create_wakeup_source(struct epitem *epi)
1119{
1120 const char *name;
1121
1122 if (!epi->ep->ws) {
1123 epi->ep->ws = wakeup_source_register("eventpoll");
1124 if (!epi->ep->ws)
1125 return -ENOMEM;
1126 }
1127
1128 name = epi->ffd.file->f_path.dentry->d_name.name;
1129 epi->ws = wakeup_source_register(name);
1130 if (!epi->ws)
1131 return -ENOMEM;
1132
1133 return 0;
1134}
1135
1136static void ep_destroy_wakeup_source(struct epitem *epi)
1137{
1138 wakeup_source_unregister(epi->ws);
1139 epi->ws = NULL;
1140}
1141
1142
1143
1144
1145static int ep_insert(struct eventpoll *ep, struct epoll_event *event,
1146 struct file *tfile, int fd)
1147{
1148 int error, revents, pwake = 0;
1149 unsigned long flags;
1150 long user_watches;
1151 struct epitem *epi;
1152 struct ep_pqueue epq;
1153
1154 user_watches = atomic_long_read(&ep->user->epoll_watches);
1155 if (unlikely(user_watches >= max_user_watches))
1156 return -ENOSPC;
1157 if (!(epi = kmem_cache_alloc(epi_cache, GFP_KERNEL)))
1158 return -ENOMEM;
1159
1160
1161 INIT_LIST_HEAD(&epi->rdllink);
1162 INIT_LIST_HEAD(&epi->fllink);
1163 INIT_LIST_HEAD(&epi->pwqlist);
1164 epi->ep = ep;
1165 ep_set_ffd(&epi->ffd, tfile, fd);
1166 epi->event = *event;
1167 epi->nwait = 0;
1168 epi->next = EP_UNACTIVE_PTR;
1169 if (epi->event.events & EPOLLWAKEUP) {
1170 error = ep_create_wakeup_source(epi);
1171 if (error)
1172 goto error_create_wakeup_source;
1173 } else {
1174 epi->ws = NULL;
1175 }
1176
1177
1178 epq.epi = epi;
1179 init_poll_funcptr(&epq.pt, ep_ptable_queue_proc);
1180 epq.pt._key = event->events;
1181
1182
1183
1184
1185
1186
1187
1188
1189 revents = tfile->f_op->poll(tfile, &epq.pt);
1190
1191
1192
1193
1194
1195
1196 error = -ENOMEM;
1197 if (epi->nwait < 0)
1198 goto error_unregister;
1199
1200
1201 spin_lock(&tfile->f_lock);
1202 list_add_tail(&epi->fllink, &tfile->f_ep_links);
1203 spin_unlock(&tfile->f_lock);
1204
1205
1206
1207
1208
1209 ep_rbtree_insert(ep, epi);
1210
1211
1212 error = -EINVAL;
1213 if (reverse_path_check())
1214 goto error_remove_epi;
1215
1216
1217 spin_lock_irqsave(&ep->lock, flags);
1218
1219
1220 if ((revents & event->events) && !ep_is_linked(&epi->rdllink)) {
1221 list_add_tail(&epi->rdllink, &ep->rdllist);
1222 __pm_stay_awake(epi->ws);
1223
1224
1225 if (waitqueue_active(&ep->wq))
1226 wake_up_locked(&ep->wq);
1227 if (waitqueue_active(&ep->poll_wait))
1228 pwake++;
1229 }
1230
1231 spin_unlock_irqrestore(&ep->lock, flags);
1232
1233 atomic_long_inc(&ep->user->epoll_watches);
1234
1235
1236 if (pwake)
1237 ep_poll_safewake(&ep->poll_wait);
1238
1239 return 0;
1240
1241error_remove_epi:
1242 spin_lock(&tfile->f_lock);
1243 if (ep_is_linked(&epi->fllink))
1244 list_del_init(&epi->fllink);
1245 spin_unlock(&tfile->f_lock);
1246
1247 rb_erase(&epi->rbn, &ep->rbr);
1248
1249error_unregister:
1250 ep_unregister_pollwait(ep, epi);
1251
1252
1253
1254
1255
1256
1257
1258 spin_lock_irqsave(&ep->lock, flags);
1259 if (ep_is_linked(&epi->rdllink))
1260 list_del_init(&epi->rdllink);
1261 spin_unlock_irqrestore(&ep->lock, flags);
1262
1263 wakeup_source_unregister(epi->ws);
1264
1265error_create_wakeup_source:
1266 kmem_cache_free(epi_cache, epi);
1267
1268 return error;
1269}
1270
1271
1272
1273
1274
1275static int ep_modify(struct eventpoll *ep, struct epitem *epi, struct epoll_event *event)
1276{
1277 int pwake = 0;
1278 unsigned int revents;
1279 poll_table pt;
1280
1281 init_poll_funcptr(&pt, NULL);
1282
1283
1284
1285
1286
1287
1288 epi->event.events = event->events;
1289 pt._key = event->events;
1290 epi->event.data = event->data;
1291 if (epi->event.events & EPOLLWAKEUP) {
1292 if (!epi->ws)
1293 ep_create_wakeup_source(epi);
1294 } else if (epi->ws) {
1295 ep_destroy_wakeup_source(epi);
1296 }
1297
1298
1299
1300
1301
1302 revents = epi->ffd.file->f_op->poll(epi->ffd.file, &pt);
1303
1304
1305
1306
1307
1308 if (revents & event->events) {
1309 spin_lock_irq(&ep->lock);
1310 if (!ep_is_linked(&epi->rdllink)) {
1311 list_add_tail(&epi->rdllink, &ep->rdllist);
1312 __pm_stay_awake(epi->ws);
1313
1314
1315 if (waitqueue_active(&ep->wq))
1316 wake_up_locked(&ep->wq);
1317 if (waitqueue_active(&ep->poll_wait))
1318 pwake++;
1319 }
1320 spin_unlock_irq(&ep->lock);
1321 }
1322
1323
1324 if (pwake)
1325 ep_poll_safewake(&ep->poll_wait);
1326
1327 return 0;
1328}
1329
1330static int ep_send_events_proc(struct eventpoll *ep, struct list_head *head,
1331 void *priv)
1332{
1333 struct ep_send_events_data *esed = priv;
1334 int eventcnt;
1335 unsigned int revents;
1336 struct epitem *epi;
1337 struct epoll_event __user *uevent;
1338 poll_table pt;
1339
1340 init_poll_funcptr(&pt, NULL);
1341
1342
1343
1344
1345
1346
1347 for (eventcnt = 0, uevent = esed->events;
1348 !list_empty(head) && eventcnt < esed->maxevents;) {
1349 epi = list_first_entry(head, struct epitem, rdllink);
1350
1351
1352
1353
1354
1355
1356
1357
1358
1359
1360 if (epi->ws && epi->ws->active)
1361 __pm_stay_awake(ep->ws);
1362 __pm_relax(epi->ws);
1363 list_del_init(&epi->rdllink);
1364
1365 pt._key = epi->event.events;
1366 revents = epi->ffd.file->f_op->poll(epi->ffd.file, &pt) &
1367 epi->event.events;
1368
1369
1370
1371
1372
1373
1374
1375 if (revents) {
1376 if (__put_user(revents, &uevent->events) ||
1377 __put_user(epi->event.data, &uevent->data)) {
1378 list_add(&epi->rdllink, head);
1379 __pm_stay_awake(epi->ws);
1380 return eventcnt ? eventcnt : -EFAULT;
1381 }
1382 eventcnt++;
1383 uevent++;
1384 if (epi->event.events & EPOLLONESHOT)
1385 epi->event.events &= EP_PRIVATE_BITS;
1386 else if (!(epi->event.events & EPOLLET)) {
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398 list_add_tail(&epi->rdllink, &ep->rdllist);
1399 __pm_stay_awake(epi->ws);
1400 }
1401 }
1402 }
1403
1404 return eventcnt;
1405}
1406
1407static int ep_send_events(struct eventpoll *ep,
1408 struct epoll_event __user *events, int maxevents)
1409{
1410 struct ep_send_events_data esed;
1411
1412 esed.maxevents = maxevents;
1413 esed.events = events;
1414
1415 return ep_scan_ready_list(ep, ep_send_events_proc, &esed, 0);
1416}
1417
1418static inline struct timespec ep_set_mstimeout(long ms)
1419{
1420 struct timespec now, ts = {
1421 .tv_sec = ms / MSEC_PER_SEC,
1422 .tv_nsec = NSEC_PER_MSEC * (ms % MSEC_PER_SEC),
1423 };
1424
1425 ktime_get_ts(&now);
1426 return timespec_add_safe(now, ts);
1427}
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
1444
1445
1446static int ep_poll(struct eventpoll *ep, struct epoll_event __user *events,
1447 int maxevents, long timeout)
1448{
1449 int res = 0, eavail, timed_out = 0;
1450 unsigned long flags;
1451 long slack = 0;
1452 wait_queue_t wait;
1453 ktime_t expires, *to = NULL;
1454
1455 if (timeout > 0) {
1456 struct timespec end_time = ep_set_mstimeout(timeout);
1457
1458 slack = select_estimate_accuracy(&end_time);
1459 to = &expires;
1460 *to = timespec_to_ktime(end_time);
1461 } else if (timeout == 0) {
1462
1463
1464
1465
1466 timed_out = 1;
1467 spin_lock_irqsave(&ep->lock, flags);
1468 goto check_events;
1469 }
1470
1471fetch_events:
1472 spin_lock_irqsave(&ep->lock, flags);
1473
1474 if (!ep_events_available(ep)) {
1475
1476
1477
1478
1479
1480 init_waitqueue_entry(&wait, current);
1481 __add_wait_queue_exclusive(&ep->wq, &wait);
1482
1483 for (;;) {
1484
1485
1486
1487
1488
1489 set_current_state(TASK_INTERRUPTIBLE);
1490 if (ep_events_available(ep) || timed_out)
1491 break;
1492 if (signal_pending(current)) {
1493 res = -EINTR;
1494 break;
1495 }
1496
1497 spin_unlock_irqrestore(&ep->lock, flags);
1498 if (!schedule_hrtimeout_range(to, slack, HRTIMER_MODE_ABS))
1499 timed_out = 1;
1500
1501 spin_lock_irqsave(&ep->lock, flags);
1502 }
1503 __remove_wait_queue(&ep->wq, &wait);
1504
1505 set_current_state(TASK_RUNNING);
1506 }
1507check_events:
1508
1509 eavail = ep_events_available(ep);
1510
1511 spin_unlock_irqrestore(&ep->lock, flags);
1512
1513
1514
1515
1516
1517
1518 if (!res && eavail &&
1519 !(res = ep_send_events(ep, events, maxevents)) && !timed_out)
1520 goto fetch_events;
1521
1522 return res;
1523}
1524
1525
1526
1527
1528
1529
1530
1531
1532
1533
1534
1535
1536
1537
1538
1539
1540static int ep_loop_check_proc(void *priv, void *cookie, int call_nests)
1541{
1542 int error = 0;
1543 struct file *file = priv;
1544 struct eventpoll *ep = file->private_data;
1545 struct eventpoll *ep_tovisit;
1546 struct rb_node *rbp;
1547 struct epitem *epi;
1548
1549 mutex_lock_nested(&ep->mtx, call_nests + 1);
1550 ep->visited = 1;
1551 list_add(&ep->visited_list_link, &visited_list);
1552 for (rbp = rb_first(&ep->rbr); rbp; rbp = rb_next(rbp)) {
1553 epi = rb_entry(rbp, struct epitem, rbn);
1554 if (unlikely(is_file_epoll(epi->ffd.file))) {
1555 ep_tovisit = epi->ffd.file->private_data;
1556 if (ep_tovisit->visited)
1557 continue;
1558 error = ep_call_nested(&poll_loop_ncalls, EP_MAX_NESTS,
1559 ep_loop_check_proc, epi->ffd.file,
1560 ep_tovisit, current);
1561 if (error != 0)
1562 break;
1563 } else {
1564
1565
1566
1567
1568
1569
1570
1571
1572 if (list_empty(&epi->ffd.file->f_tfile_llink))
1573 list_add(&epi->ffd.file->f_tfile_llink,
1574 &tfile_check_list);
1575 }
1576 }
1577 mutex_unlock(&ep->mtx);
1578
1579 return error;
1580}
1581
1582
1583
1584
1585
1586
1587
1588
1589
1590
1591
1592
1593static int ep_loop_check(struct eventpoll *ep, struct file *file)
1594{
1595 int ret;
1596 struct eventpoll *ep_cur, *ep_next;
1597
1598 ret = ep_call_nested(&poll_loop_ncalls, EP_MAX_NESTS,
1599 ep_loop_check_proc, file, ep, current);
1600
1601 list_for_each_entry_safe(ep_cur, ep_next, &visited_list,
1602 visited_list_link) {
1603 ep_cur->visited = 0;
1604 list_del(&ep_cur->visited_list_link);
1605 }
1606 return ret;
1607}
1608
1609static void clear_tfile_check_list(void)
1610{
1611 struct file *file;
1612
1613
1614 while (!list_empty(&tfile_check_list)) {
1615 file = list_first_entry(&tfile_check_list, struct file,
1616 f_tfile_llink);
1617 list_del_init(&file->f_tfile_llink);
1618 }
1619 INIT_LIST_HEAD(&tfile_check_list);
1620}
1621
1622
1623
1624
1625SYSCALL_DEFINE1(epoll_create1, int, flags)
1626{
1627 int error, fd;
1628 struct eventpoll *ep = NULL;
1629 struct file *file;
1630
1631
1632 BUILD_BUG_ON(EPOLL_CLOEXEC != O_CLOEXEC);
1633
1634 if (flags & ~EPOLL_CLOEXEC)
1635 return -EINVAL;
1636
1637
1638
1639 error = ep_alloc(&ep);
1640 if (error < 0)
1641 return error;
1642
1643
1644
1645
1646 fd = get_unused_fd_flags(O_RDWR | (flags & O_CLOEXEC));
1647 if (fd < 0) {
1648 error = fd;
1649 goto out_free_ep;
1650 }
1651 file = anon_inode_getfile("[eventpoll]", &eventpoll_fops, ep,
1652 O_RDWR | (flags & O_CLOEXEC));
1653 if (IS_ERR(file)) {
1654 error = PTR_ERR(file);
1655 goto out_free_fd;
1656 }
1657 ep->file = file;
1658 fd_install(fd, file);
1659 return fd;
1660
1661out_free_fd:
1662 put_unused_fd(fd);
1663out_free_ep:
1664 ep_free(ep);
1665 return error;
1666}
1667
1668SYSCALL_DEFINE1(epoll_create, int, size)
1669{
1670 if (size <= 0)
1671 return -EINVAL;
1672
1673 return sys_epoll_create1(0);
1674}
1675
1676
1677
1678
1679
1680
1681SYSCALL_DEFINE4(epoll_ctl, int, epfd, int, op, int, fd,
1682 struct epoll_event __user *, event)
1683{
1684 int error;
1685 int did_lock_epmutex = 0;
1686 struct file *file, *tfile;
1687 struct eventpoll *ep;
1688 struct epitem *epi;
1689 struct epoll_event epds;
1690
1691 error = -EFAULT;
1692 if (ep_op_has_event(op) &&
1693 copy_from_user(&epds, event, sizeof(struct epoll_event)))
1694 goto error_return;
1695
1696
1697 error = -EBADF;
1698 file = fget(epfd);
1699 if (!file)
1700 goto error_return;
1701
1702
1703 tfile = fget(fd);
1704 if (!tfile)
1705 goto error_fput;
1706
1707
1708 error = -EPERM;
1709 if (!tfile->f_op || !tfile->f_op->poll)
1710 goto error_tgt_fput;
1711
1712
1713 if ((epds.events & EPOLLWAKEUP) && !capable(CAP_BLOCK_SUSPEND))
1714 epds.events &= ~EPOLLWAKEUP;
1715
1716
1717
1718
1719
1720
1721 error = -EINVAL;
1722 if (file == tfile || !is_file_epoll(file))
1723 goto error_tgt_fput;
1724
1725
1726
1727
1728
1729 ep = file->private_data;
1730
1731
1732
1733
1734
1735
1736
1737
1738
1739
1740
1741
1742
1743 if (op == EPOLL_CTL_ADD || op == EPOLL_CTL_DEL) {
1744 mutex_lock(&epmutex);
1745 did_lock_epmutex = 1;
1746 }
1747 if (op == EPOLL_CTL_ADD) {
1748 if (is_file_epoll(tfile)) {
1749 error = -ELOOP;
1750 if (ep_loop_check(ep, tfile) != 0) {
1751 clear_tfile_check_list();
1752 goto error_tgt_fput;
1753 }
1754 } else
1755 list_add(&tfile->f_tfile_llink, &tfile_check_list);
1756 }
1757
1758 mutex_lock_nested(&ep->mtx, 0);
1759
1760
1761
1762
1763
1764
1765 epi = ep_find(ep, tfile, fd);
1766
1767 error = -EINVAL;
1768 switch (op) {
1769 case EPOLL_CTL_ADD:
1770 if (!epi) {
1771 epds.events |= POLLERR | POLLHUP;
1772 error = ep_insert(ep, &epds, tfile, fd);
1773 } else
1774 error = -EEXIST;
1775 clear_tfile_check_list();
1776 break;
1777 case EPOLL_CTL_DEL:
1778 if (epi)
1779 error = ep_remove(ep, epi);
1780 else
1781 error = -ENOENT;
1782 break;
1783 case EPOLL_CTL_MOD:
1784 if (epi) {
1785 epds.events |= POLLERR | POLLHUP;
1786 error = ep_modify(ep, epi, &epds);
1787 } else
1788 error = -ENOENT;
1789 break;
1790 }
1791 mutex_unlock(&ep->mtx);
1792
1793error_tgt_fput:
1794 if (did_lock_epmutex)
1795 mutex_unlock(&epmutex);
1796
1797 fput(tfile);
1798error_fput:
1799 fput(file);
1800error_return:
1801
1802 return error;
1803}
1804
1805
1806
1807
1808
1809SYSCALL_DEFINE4(epoll_wait, int, epfd, struct epoll_event __user *, events,
1810 int, maxevents, int, timeout)
1811{
1812 int error;
1813 struct file *file;
1814 struct eventpoll *ep;
1815
1816
1817 if (maxevents <= 0 || maxevents > EP_MAX_EVENTS)
1818 return -EINVAL;
1819
1820
1821 if (!access_ok(VERIFY_WRITE, events, maxevents * sizeof(struct epoll_event))) {
1822 error = -EFAULT;
1823 goto error_return;
1824 }
1825
1826
1827 error = -EBADF;
1828 file = fget(epfd);
1829 if (!file)
1830 goto error_return;
1831
1832
1833
1834
1835
1836 error = -EINVAL;
1837 if (!is_file_epoll(file))
1838 goto error_fput;
1839
1840
1841
1842
1843
1844 ep = file->private_data;
1845
1846
1847 error = ep_poll(ep, events, maxevents, timeout);
1848
1849error_fput:
1850 fput(file);
1851error_return:
1852
1853 return error;
1854}
1855
1856
1857
1858
1859
1860SYSCALL_DEFINE6(epoll_pwait, int, epfd, struct epoll_event __user *, events,
1861 int, maxevents, int, timeout, const sigset_t __user *, sigmask,
1862 size_t, sigsetsize)
1863{
1864 int error;
1865 sigset_t ksigmask, sigsaved;
1866
1867
1868
1869
1870
1871 if (sigmask) {
1872 if (sigsetsize != sizeof(sigset_t))
1873 return -EINVAL;
1874 if (copy_from_user(&ksigmask, sigmask, sizeof(ksigmask)))
1875 return -EFAULT;
1876 sigdelsetmask(&ksigmask, sigmask(SIGKILL) | sigmask(SIGSTOP));
1877 sigprocmask(SIG_SETMASK, &ksigmask, &sigsaved);
1878 }
1879
1880 error = sys_epoll_wait(epfd, events, maxevents, timeout);
1881
1882
1883
1884
1885
1886
1887
1888 if (sigmask) {
1889 if (error == -EINTR) {
1890 memcpy(¤t->saved_sigmask, &sigsaved,
1891 sizeof(sigsaved));
1892 set_restore_sigmask();
1893 } else
1894 sigprocmask(SIG_SETMASK, &sigsaved, NULL);
1895 }
1896
1897 return error;
1898}
1899
1900static int __init eventpoll_init(void)
1901{
1902 struct sysinfo si;
1903
1904 si_meminfo(&si);
1905
1906
1907
1908 max_user_watches = (((si.totalram - si.totalhigh) / 25) << PAGE_SHIFT) /
1909 EP_ITEM_COST;
1910 BUG_ON(max_user_watches < 0);
1911
1912
1913
1914
1915
1916 ep_nested_calls_init(&poll_loop_ncalls);
1917
1918
1919 ep_nested_calls_init(&poll_safewake_ncalls);
1920
1921
1922 ep_nested_calls_init(&poll_readywalk_ncalls);
1923
1924
1925 epi_cache = kmem_cache_create("eventpoll_epi", sizeof(struct epitem),
1926 0, SLAB_HWCACHE_ALIGN | SLAB_PANIC, NULL);
1927
1928
1929 pwq_cache = kmem_cache_create("eventpoll_pwq",
1930 sizeof(struct eppoll_entry), 0, SLAB_PANIC, NULL);
1931
1932 return 0;
1933}
1934fs_initcall(eventpoll_init);
1935