blob: 1b8ff05e8311e8f7294f7da7e4d18afb9cfb2817 [file] [log] [blame]
Greg Kroah-Hartmanb2441312017-11-01 15:07:57 +01001// SPDX-License-Identifier: GPL-2.0
Linus Torvalds1da177e2005-04-16 15:20:36 -07002/*
3 * linux/fs/file.c
4 *
5 * Copyright (C) 1998-1999, Stephen Tweedie and Bill Hawes
6 *
7 * Manage the dynamic fd arrays in the process files_struct.
8 */
9
Al Virofe17f222012-08-21 11:48:11 -040010#include <linux/syscalls.h>
Paul Gortmaker630d9c42011-11-16 23:57:37 -050011#include <linux/export.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070012#include <linux/fs.h>
Christian Brauner278a5fb2019-05-24 11:30:34 +020013#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070014#include <linux/mm.h>
Ingo Molnar3f07c012017-02-08 18:51:30 +010015#include <linux/sched/signal.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070016#include <linux/slab.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070017#include <linux/file.h>
Al Viro9f3acc32008-04-24 07:44:08 -040018#include <linux/fdtable.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070019#include <linux/bitops.h>
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070020#include <linux/spinlock.h>
21#include <linux/rcupdate.h>
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070022
Alexey Dobriyan9b80a182016-09-02 00:38:52 +030023unsigned int sysctl_nr_open __read_mostly = 1024*1024;
24unsigned int sysctl_nr_open_min = BITS_PER_LONG;
Rasmus Villemoes752343b2015-10-29 12:01:41 +010025/* our min() is unusable in constant expressions ;-/ */
26#define __const_min(x, y) ((x) < (y) ? (x) : (y))
Alexey Dobriyan9b80a182016-09-02 00:38:52 +030027unsigned int sysctl_nr_open_max =
28 __const_min(INT_MAX, ~(size_t)0/sizeof(void *)) & -BITS_PER_LONG;
Eric Dumazet9cfe0152008-02-06 01:37:16 -080029
Changli Gaoa892e2d2010-08-10 18:01:35 -070030static void __free_fdtable(struct fdtable *fdt)
Vadim Lobanov5466b452006-12-10 02:21:22 -080031{
Al Virof6c0a192014-04-23 10:18:46 -040032 kvfree(fdt->fd);
33 kvfree(fdt->open_fds);
Changli Gaoa892e2d2010-08-10 18:01:35 -070034 kfree(fdt);
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070035}
36
Al Viro7cf4dc32012-08-15 19:56:12 -040037static void free_fdtable_rcu(struct rcu_head *rcu)
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070038{
Al Viroac3e3c52013-04-28 21:42:33 -040039 __free_fdtable(container_of(rcu, struct fdtable, rcu));
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070040}
41
Linus Torvaldsf3f86e32015-10-30 16:53:57 -070042#define BITBIT_NR(nr) BITS_TO_LONGS(BITS_TO_LONGS(nr))
43#define BITBIT_SIZE(nr) (BITBIT_NR(nr) * sizeof(long))
44
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070045/*
Eric Biggersea5c58e2015-11-06 00:32:04 -060046 * Copy 'count' fd bits from the old table to the new table and clear the extra
47 * space if any. This does not copy the file pointers. Called with the files
48 * spinlock held for write.
49 */
50static void copy_fd_bitmaps(struct fdtable *nfdt, struct fdtable *ofdt,
51 unsigned int count)
52{
53 unsigned int cpy, set;
54
55 cpy = count / BITS_PER_BYTE;
56 set = (nfdt->max_fds - count) / BITS_PER_BYTE;
57 memcpy(nfdt->open_fds, ofdt->open_fds, cpy);
58 memset((char *)nfdt->open_fds + cpy, 0, set);
59 memcpy(nfdt->close_on_exec, ofdt->close_on_exec, cpy);
60 memset((char *)nfdt->close_on_exec + cpy, 0, set);
61
62 cpy = BITBIT_SIZE(count);
63 set = BITBIT_SIZE(nfdt->max_fds) - cpy;
64 memcpy(nfdt->full_fds_bits, ofdt->full_fds_bits, cpy);
65 memset((char *)nfdt->full_fds_bits + cpy, 0, set);
66}
67
68/*
69 * Copy all file descriptors from the old table to the new, expanded table and
70 * clear the extra space. Called with the files spinlock held for write.
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070071 */
Vadim Lobanov5466b452006-12-10 02:21:22 -080072static void copy_fdtable(struct fdtable *nfdt, struct fdtable *ofdt)
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070073{
Al Viro4e89b722020-05-19 17:48:52 -040074 size_t cpy, set;
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070075
Vadim Lobanov5466b452006-12-10 02:21:22 -080076 BUG_ON(nfdt->max_fds < ofdt->max_fds);
Vadim Lobanov5466b452006-12-10 02:21:22 -080077
78 cpy = ofdt->max_fds * sizeof(struct file *);
79 set = (nfdt->max_fds - ofdt->max_fds) * sizeof(struct file *);
80 memcpy(nfdt->fd, ofdt->fd, cpy);
Eric Biggersea5c58e2015-11-06 00:32:04 -060081 memset((char *)nfdt->fd + cpy, 0, set);
Vadim Lobanov5466b452006-12-10 02:21:22 -080082
Eric Biggersea5c58e2015-11-06 00:32:04 -060083 copy_fd_bitmaps(nfdt, ofdt, ofdt->max_fds);
Linus Torvalds1da177e2005-04-16 15:20:36 -070084}
85
Vadim Lobanov5466b452006-12-10 02:21:22 -080086static struct fdtable * alloc_fdtable(unsigned int nr)
Linus Torvalds1da177e2005-04-16 15:20:36 -070087{
Vadim Lobanov5466b452006-12-10 02:21:22 -080088 struct fdtable *fdt;
David Howells1fd36ad2012-02-16 17:49:54 +000089 void *data;
Linus Torvalds1da177e2005-04-16 15:20:36 -070090
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070091 /*
Vadim Lobanov5466b452006-12-10 02:21:22 -080092 * Figure out how many fds we actually want to support in this fdtable.
93 * Allocation steps are keyed to the size of the fdarray, since it
94 * grows far faster than any of the other dynamic data. We try to fit
95 * the fdarray into comfortable page-tuned chunks: starting at 1024B
96 * and growing in powers of two from there on.
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -070097 */
Vadim Lobanov5466b452006-12-10 02:21:22 -080098 nr /= (1024 / sizeof(struct file *));
99 nr = roundup_pow_of_two(nr + 1);
100 nr *= (1024 / sizeof(struct file *));
Al Viro5c598b32008-04-27 20:04:15 -0400101 /*
102 * Note that this can drive nr *below* what we had passed if sysctl_nr_open
103 * had been set lower between the check in expand_files() and here. Deal
104 * with that in caller, it's cheaper that way.
105 *
106 * We make sure that nr remains a multiple of BITS_PER_LONG - otherwise
107 * bitmaps handling below becomes unpleasant, to put it mildly...
108 */
109 if (unlikely(nr > sysctl_nr_open))
110 nr = ((sysctl_nr_open - 1) | (BITS_PER_LONG - 1)) + 1;
Vadim Lobanovbbea9f62006-12-10 02:21:12 -0800111
Vladimir Davydov5d097052016-01-14 15:18:21 -0800112 fdt = kmalloc(sizeof(struct fdtable), GFP_KERNEL_ACCOUNT);
Vadim Lobanov5466b452006-12-10 02:21:22 -0800113 if (!fdt)
Vadim Lobanovbbea9f62006-12-10 02:21:12 -0800114 goto out;
Vadim Lobanov5466b452006-12-10 02:21:22 -0800115 fdt->max_fds = nr;
Michal Hockoc823bd92017-07-06 15:36:19 -0700116 data = kvmalloc_array(nr, sizeof(struct file *), GFP_KERNEL_ACCOUNT);
Vadim Lobanov5466b452006-12-10 02:21:22 -0800117 if (!data)
118 goto out_fdt;
David Howells1fd36ad2012-02-16 17:49:54 +0000119 fdt->fd = data;
120
Michal Hockoc823bd92017-07-06 15:36:19 -0700121 data = kvmalloc(max_t(size_t,
122 2 * nr / BITS_PER_BYTE + BITBIT_SIZE(nr), L1_CACHE_BYTES),
123 GFP_KERNEL_ACCOUNT);
Vadim Lobanov5466b452006-12-10 02:21:22 -0800124 if (!data)
125 goto out_arr;
David Howells1fd36ad2012-02-16 17:49:54 +0000126 fdt->open_fds = data;
Vadim Lobanov5466b452006-12-10 02:21:22 -0800127 data += nr / BITS_PER_BYTE;
David Howells1fd36ad2012-02-16 17:49:54 +0000128 fdt->close_on_exec = data;
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700129 data += nr / BITS_PER_BYTE;
130 fdt->full_fds_bits = data;
Vadim Lobanov5466b452006-12-10 02:21:22 -0800131
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700132 return fdt;
Vadim Lobanov5466b452006-12-10 02:21:22 -0800133
134out_arr:
Al Virof6c0a192014-04-23 10:18:46 -0400135 kvfree(fdt->fd);
Vadim Lobanov5466b452006-12-10 02:21:22 -0800136out_fdt:
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700137 kfree(fdt);
Vadim Lobanov5466b452006-12-10 02:21:22 -0800138out:
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700139 return NULL;
140}
141
142/*
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700143 * Expand the file descriptor table.
144 * This function will allocate a new fdtable and both fd array and fdset, of
145 * the given size.
146 * Return <0 error code on error; 1 on successful completion.
147 * The files->file_lock should be held on entry, and will be held on exit.
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700148 */
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300149static int expand_fdtable(struct files_struct *files, unsigned int nr)
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700150 __releases(files->file_lock)
151 __acquires(files->file_lock)
152{
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700153 struct fdtable *new_fdt, *cur_fdt;
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700154
Linus Torvalds1da177e2005-04-16 15:20:36 -0700155 spin_unlock(&files->file_lock);
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700156 new_fdt = alloc_fdtable(nr);
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200157
158 /* make sure all __fd_install() have seen resize_in_progress
159 * or have finished their rcu_read_lock_sched() section.
160 */
161 if (atomic_read(&files->count) > 1)
Paul E. McKenneyc93ffc12018-11-05 17:31:31 -0800162 synchronize_rcu();
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200163
Linus Torvalds1da177e2005-04-16 15:20:36 -0700164 spin_lock(&files->file_lock);
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700165 if (!new_fdt)
166 return -ENOMEM;
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700167 /*
Al Viro5c598b32008-04-27 20:04:15 -0400168 * extremely unlikely race - sysctl_nr_open decreased between the check in
169 * caller and alloc_fdtable(). Cheaper to catch it here...
170 */
171 if (unlikely(new_fdt->max_fds <= nr)) {
Changli Gaoa892e2d2010-08-10 18:01:35 -0700172 __free_fdtable(new_fdt);
Al Viro5c598b32008-04-27 20:04:15 -0400173 return -EMFILE;
174 }
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700175 cur_fdt = files_fdtable(files);
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200176 BUG_ON(nr < cur_fdt->max_fds);
177 copy_fdtable(new_fdt, cur_fdt);
178 rcu_assign_pointer(files->fdt, new_fdt);
179 if (cur_fdt != &files->fdtab)
180 call_rcu(&cur_fdt->rcu, free_fdtable_rcu);
181 /* coupled with smp_rmb() in __fd_install() */
182 smp_wmb();
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700183 return 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700184}
185
186/*
187 * Expand files.
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700188 * This function will expand the file structures, if the requested size exceeds
189 * the current capacity and there is room for expansion.
190 * Return <0 error code on error; 0 when nothing done; 1 when files were
191 * expanded and execution may have blocked.
192 * The files->file_lock should be held on entry, and will be held on exit.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700193 */
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300194static int expand_files(struct files_struct *files, unsigned int nr)
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200195 __releases(files->file_lock)
196 __acquires(files->file_lock)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700197{
Dipankar Sarmabadf1662005-09-09 13:04:10 -0700198 struct fdtable *fdt;
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200199 int expanded = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700200
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200201repeat:
Dipankar Sarmabadf1662005-09-09 13:04:10 -0700202 fdt = files_fdtable(files);
Al Viro4e1e0182008-07-26 16:01:20 -0400203
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700204 /* Do we need to expand? */
Vadim Lobanovbbea9f62006-12-10 02:21:12 -0800205 if (nr < fdt->max_fds)
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200206 return expanded;
Al Viro4e1e0182008-07-26 16:01:20 -0400207
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700208 /* Can we expand? */
Eric Dumazet9cfe0152008-02-06 01:37:16 -0800209 if (nr >= sysctl_nr_open)
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700210 return -EMFILE;
211
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200212 if (unlikely(files->resize_in_progress)) {
213 spin_unlock(&files->file_lock);
214 expanded = 1;
215 wait_event(files->resize_wait, !files->resize_in_progress);
216 spin_lock(&files->file_lock);
217 goto repeat;
218 }
219
Vadim Lobanov74d392a2006-09-29 02:01:43 -0700220 /* All good, so we try */
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200221 files->resize_in_progress = true;
222 expanded = expand_fdtable(files, nr);
223 files->resize_in_progress = false;
224
225 wake_up_all(&files->resize_wait);
226 return expanded;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700227}
Dipankar Sarmaab2af1f2005-09-09 13:04:13 -0700228
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300229static inline void __set_close_on_exec(unsigned int fd, struct fdtable *fdt)
Al Virob8318b02012-08-21 20:09:42 -0400230{
231 __set_bit(fd, fdt->close_on_exec);
232}
233
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300234static inline void __clear_close_on_exec(unsigned int fd, struct fdtable *fdt)
Al Virob8318b02012-08-21 20:09:42 -0400235{
Linus Torvaldsfc908882015-10-31 16:06:40 -0700236 if (test_bit(fd, fdt->close_on_exec))
237 __clear_bit(fd, fdt->close_on_exec);
Al Virob8318b02012-08-21 20:09:42 -0400238}
239
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700240static inline void __set_open_fd(unsigned int fd, struct fdtable *fdt)
Al Virob8318b02012-08-21 20:09:42 -0400241{
242 __set_bit(fd, fdt->open_fds);
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700243 fd /= BITS_PER_LONG;
244 if (!~fdt->open_fds[fd])
245 __set_bit(fd, fdt->full_fds_bits);
Al Virob8318b02012-08-21 20:09:42 -0400246}
247
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700248static inline void __clear_open_fd(unsigned int fd, struct fdtable *fdt)
Al Virob8318b02012-08-21 20:09:42 -0400249{
250 __clear_bit(fd, fdt->open_fds);
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700251 __clear_bit(fd / BITS_PER_LONG, fdt->full_fds_bits);
Al Virob8318b02012-08-21 20:09:42 -0400252}
253
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300254static unsigned int count_open_files(struct fdtable *fdt)
Al Viro02afc6262008-05-08 19:42:56 -0400255{
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300256 unsigned int size = fdt->max_fds;
257 unsigned int i;
Al Viro02afc6262008-05-08 19:42:56 -0400258
259 /* Find the last open fd */
David Howells1fd36ad2012-02-16 17:49:54 +0000260 for (i = size / BITS_PER_LONG; i > 0; ) {
261 if (fdt->open_fds[--i])
Al Viro02afc6262008-05-08 19:42:56 -0400262 break;
263 }
David Howells1fd36ad2012-02-16 17:49:54 +0000264 i = (i + 1) * BITS_PER_LONG;
Al Viro02afc6262008-05-08 19:42:56 -0400265 return i;
266}
267
Al Viro02afc6262008-05-08 19:42:56 -0400268/*
269 * Allocate a new files structure and copy contents from the
270 * passed in files structure.
271 * errorp will be valid only when the returned files_struct is NULL.
272 */
273struct files_struct *dup_fd(struct files_struct *oldf, int *errorp)
274{
275 struct files_struct *newf;
276 struct file **old_fds, **new_fds;
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300277 unsigned int open_files, i;
Al Viro02afc6262008-05-08 19:42:56 -0400278 struct fdtable *old_fdt, *new_fdt;
279
280 *errorp = -ENOMEM;
Al Viroafbec7f2008-05-08 21:11:17 -0400281 newf = kmem_cache_alloc(files_cachep, GFP_KERNEL);
Al Viro02afc6262008-05-08 19:42:56 -0400282 if (!newf)
283 goto out;
284
Al Viroafbec7f2008-05-08 21:11:17 -0400285 atomic_set(&newf->count, 1);
286
287 spin_lock_init(&newf->file_lock);
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200288 newf->resize_in_progress = false;
289 init_waitqueue_head(&newf->resize_wait);
Al Viroafbec7f2008-05-08 21:11:17 -0400290 newf->next_fd = 0;
291 new_fdt = &newf->fdtab;
292 new_fdt->max_fds = NR_OPEN_DEFAULT;
David Howells1fd36ad2012-02-16 17:49:54 +0000293 new_fdt->close_on_exec = newf->close_on_exec_init;
294 new_fdt->open_fds = newf->open_fds_init;
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700295 new_fdt->full_fds_bits = newf->full_fds_bits_init;
Al Viroafbec7f2008-05-08 21:11:17 -0400296 new_fdt->fd = &newf->fd_array[0];
Al Viroafbec7f2008-05-08 21:11:17 -0400297
Al Viro02afc6262008-05-08 19:42:56 -0400298 spin_lock(&oldf->file_lock);
299 old_fdt = files_fdtable(oldf);
Al Viro02afc6262008-05-08 19:42:56 -0400300 open_files = count_open_files(old_fdt);
301
302 /*
303 * Check whether we need to allocate a larger fd array and fd set.
Al Viro02afc6262008-05-08 19:42:56 -0400304 */
Al Viroadbecb12008-05-08 21:19:42 -0400305 while (unlikely(open_files > new_fdt->max_fds)) {
Al Viro02afc6262008-05-08 19:42:56 -0400306 spin_unlock(&oldf->file_lock);
Al Viro9dec3c42008-05-08 21:02:45 -0400307
Changli Gaoa892e2d2010-08-10 18:01:35 -0700308 if (new_fdt != &newf->fdtab)
309 __free_fdtable(new_fdt);
Al Viroadbecb12008-05-08 21:19:42 -0400310
Al Viro9dec3c42008-05-08 21:02:45 -0400311 new_fdt = alloc_fdtable(open_files - 1);
312 if (!new_fdt) {
313 *errorp = -ENOMEM;
Al Viro02afc6262008-05-08 19:42:56 -0400314 goto out_release;
Al Viro9dec3c42008-05-08 21:02:45 -0400315 }
316
317 /* beyond sysctl_nr_open; nothing to do */
318 if (unlikely(new_fdt->max_fds < open_files)) {
Changli Gaoa892e2d2010-08-10 18:01:35 -0700319 __free_fdtable(new_fdt);
Al Viro9dec3c42008-05-08 21:02:45 -0400320 *errorp = -EMFILE;
321 goto out_release;
322 }
Al Viro9dec3c42008-05-08 21:02:45 -0400323
Al Viro02afc6262008-05-08 19:42:56 -0400324 /*
325 * Reacquire the oldf lock and a pointer to its fd table
326 * who knows it may have a new bigger fd table. We need
327 * the latest pointer.
328 */
329 spin_lock(&oldf->file_lock);
330 old_fdt = files_fdtable(oldf);
Al Viroadbecb12008-05-08 21:19:42 -0400331 open_files = count_open_files(old_fdt);
Al Viro02afc6262008-05-08 19:42:56 -0400332 }
333
Eric Biggersea5c58e2015-11-06 00:32:04 -0600334 copy_fd_bitmaps(new_fdt, old_fdt, open_files);
335
Al Viro02afc6262008-05-08 19:42:56 -0400336 old_fds = old_fdt->fd;
337 new_fds = new_fdt->fd;
338
Al Viro02afc6262008-05-08 19:42:56 -0400339 for (i = open_files; i != 0; i--) {
340 struct file *f = *old_fds++;
341 if (f) {
342 get_file(f);
343 } else {
344 /*
345 * The fd may be claimed in the fd bitmap but not yet
346 * instantiated in the files array if a sibling thread
347 * is partway through open(). So make sure that this
348 * fd is available to the new process.
349 */
David Howells1dce27c2012-02-16 17:49:42 +0000350 __clear_open_fd(open_files - i, new_fdt);
Al Viro02afc6262008-05-08 19:42:56 -0400351 }
352 rcu_assign_pointer(*new_fds++, f);
353 }
354 spin_unlock(&oldf->file_lock);
355
Eric Biggersea5c58e2015-11-06 00:32:04 -0600356 /* clear the remainder */
357 memset(new_fds, 0, (new_fdt->max_fds - open_files) * sizeof(struct file *));
Al Viro02afc6262008-05-08 19:42:56 -0400358
Al Viroafbec7f2008-05-08 21:11:17 -0400359 rcu_assign_pointer(newf->fdt, new_fdt);
360
Al Viro02afc6262008-05-08 19:42:56 -0400361 return newf;
362
363out_release:
364 kmem_cache_free(files_cachep, newf);
365out:
366 return NULL;
367}
368
Oleg Nesterovce08b622014-01-11 19:19:53 +0100369static struct fdtable *close_files(struct files_struct * files)
Al Viro7cf4dc32012-08-15 19:56:12 -0400370{
Al Viro7cf4dc32012-08-15 19:56:12 -0400371 /*
372 * It is safe to dereference the fd table without RCU or
373 * ->file_lock because this is the last reference to the
Oleg Nesterovce08b622014-01-11 19:19:53 +0100374 * files structure.
Al Viro7cf4dc32012-08-15 19:56:12 -0400375 */
Oleg Nesterovce08b622014-01-11 19:19:53 +0100376 struct fdtable *fdt = rcu_dereference_raw(files->fdt);
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300377 unsigned int i, j = 0;
Oleg Nesterovce08b622014-01-11 19:19:53 +0100378
Al Viro7cf4dc32012-08-15 19:56:12 -0400379 for (;;) {
380 unsigned long set;
381 i = j * BITS_PER_LONG;
382 if (i >= fdt->max_fds)
383 break;
384 set = fdt->open_fds[j++];
385 while (set) {
386 if (set & 1) {
387 struct file * file = xchg(&fdt->fd[i], NULL);
388 if (file) {
389 filp_close(file, files);
Paul E. McKenney388a4c82017-10-24 08:39:34 -0700390 cond_resched();
Al Viro7cf4dc32012-08-15 19:56:12 -0400391 }
392 }
393 i++;
394 set >>= 1;
395 }
396 }
Oleg Nesterovce08b622014-01-11 19:19:53 +0100397
398 return fdt;
Al Viro7cf4dc32012-08-15 19:56:12 -0400399}
400
401struct files_struct *get_files_struct(struct task_struct *task)
402{
403 struct files_struct *files;
404
405 task_lock(task);
406 files = task->files;
407 if (files)
408 atomic_inc(&files->count);
409 task_unlock(task);
410
411 return files;
412}
413
414void put_files_struct(struct files_struct *files)
415{
Al Viro7cf4dc32012-08-15 19:56:12 -0400416 if (atomic_dec_and_test(&files->count)) {
Oleg Nesterovce08b622014-01-11 19:19:53 +0100417 struct fdtable *fdt = close_files(files);
418
Al Virob9e02af2012-08-15 20:00:58 -0400419 /* free the arrays if they are not embedded */
420 if (fdt != &files->fdtab)
421 __free_fdtable(fdt);
422 kmem_cache_free(files_cachep, files);
Al Viro7cf4dc32012-08-15 19:56:12 -0400423 }
424}
425
426void reset_files_struct(struct files_struct *files)
427{
428 struct task_struct *tsk = current;
429 struct files_struct *old;
430
431 old = tsk->files;
432 task_lock(tsk);
433 tsk->files = files;
434 task_unlock(tsk);
435 put_files_struct(old);
436}
437
438void exit_files(struct task_struct *tsk)
439{
440 struct files_struct * files = tsk->files;
441
442 if (files) {
443 task_lock(tsk);
444 tsk->files = NULL;
445 task_unlock(tsk);
446 put_files_struct(files);
447 }
448}
449
Al Virof52111b2008-05-08 18:19:16 -0400450struct files_struct init_files = {
451 .count = ATOMIC_INIT(1),
452 .fdt = &init_files.fdtab,
453 .fdtab = {
454 .max_fds = NR_OPEN_DEFAULT,
455 .fd = &init_files.fd_array[0],
David Howells1fd36ad2012-02-16 17:49:54 +0000456 .close_on_exec = init_files.close_on_exec_init,
457 .open_fds = init_files.open_fds_init,
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700458 .full_fds_bits = init_files.full_fds_bits_init,
Al Virof52111b2008-05-08 18:19:16 -0400459 },
Thomas Gleixnereece09e2011-07-17 21:25:03 +0200460 .file_lock = __SPIN_LOCK_UNLOCKED(init_files.file_lock),
Shuriyc Chu5704a062019-03-05 15:41:56 -0800461 .resize_wait = __WAIT_QUEUE_HEAD_INITIALIZER(init_files.resize_wait),
Al Virof52111b2008-05-08 18:19:16 -0400462};
Al Viro1027abe2008-07-30 04:13:04 -0400463
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300464static unsigned int find_next_fd(struct fdtable *fdt, unsigned int start)
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700465{
Alexey Dobriyan9b80a182016-09-02 00:38:52 +0300466 unsigned int maxfd = fdt->max_fds;
467 unsigned int maxbit = maxfd / BITS_PER_LONG;
468 unsigned int bitbit = start / BITS_PER_LONG;
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700469
470 bitbit = find_next_zero_bit(fdt->full_fds_bits, maxbit, bitbit) * BITS_PER_LONG;
471 if (bitbit > maxfd)
472 return maxfd;
473 if (bitbit > start)
474 start = bitbit;
475 return find_next_zero_bit(fdt->open_fds, maxfd, start);
476}
477
Al Viro1027abe2008-07-30 04:13:04 -0400478/*
479 * allocate a file descriptor, mark it busy.
480 */
Al Virodcfadfa2012-08-12 17:27:30 -0400481int __alloc_fd(struct files_struct *files,
482 unsigned start, unsigned end, unsigned flags)
Al Viro1027abe2008-07-30 04:13:04 -0400483{
Al Viro1027abe2008-07-30 04:13:04 -0400484 unsigned int fd;
485 int error;
486 struct fdtable *fdt;
487
488 spin_lock(&files->file_lock);
489repeat:
490 fdt = files_fdtable(files);
491 fd = start;
492 if (fd < files->next_fd)
493 fd = files->next_fd;
494
495 if (fd < fdt->max_fds)
Linus Torvaldsf3f86e32015-10-30 16:53:57 -0700496 fd = find_next_fd(fdt, fd);
Al Viro1027abe2008-07-30 04:13:04 -0400497
Al Virof33ff992012-08-12 16:17:59 -0400498 /*
499 * N.B. For clone tasks sharing a files structure, this test
500 * will limit the total number of files that can be opened.
501 */
502 error = -EMFILE;
503 if (fd >= end)
504 goto out;
505
Al Viro1027abe2008-07-30 04:13:04 -0400506 error = expand_files(files, fd);
507 if (error < 0)
508 goto out;
509
510 /*
511 * If we needed to expand the fs array we
512 * might have blocked - try again.
513 */
514 if (error)
515 goto repeat;
516
517 if (start <= files->next_fd)
518 files->next_fd = fd + 1;
519
David Howells1dce27c2012-02-16 17:49:42 +0000520 __set_open_fd(fd, fdt);
Al Viro1027abe2008-07-30 04:13:04 -0400521 if (flags & O_CLOEXEC)
David Howells1dce27c2012-02-16 17:49:42 +0000522 __set_close_on_exec(fd, fdt);
Al Viro1027abe2008-07-30 04:13:04 -0400523 else
David Howells1dce27c2012-02-16 17:49:42 +0000524 __clear_close_on_exec(fd, fdt);
Al Viro1027abe2008-07-30 04:13:04 -0400525 error = fd;
526#if 1
527 /* Sanity check */
Paul E. McKenneyadd1f092014-02-12 12:51:09 -0800528 if (rcu_access_pointer(fdt->fd[fd]) != NULL) {
Al Viro1027abe2008-07-30 04:13:04 -0400529 printk(KERN_WARNING "alloc_fd: slot %d not NULL!\n", fd);
530 rcu_assign_pointer(fdt->fd[fd], NULL);
531 }
532#endif
533
534out:
535 spin_unlock(&files->file_lock);
536 return error;
537}
538
Al Viroad47bd72012-08-21 20:11:34 -0400539static int alloc_fd(unsigned start, unsigned flags)
Al Virodcfadfa2012-08-12 17:27:30 -0400540{
541 return __alloc_fd(current->files, start, rlimit(RLIMIT_NOFILE), flags);
542}
543
Jens Axboe4022e7a2020-03-19 19:23:18 -0600544int __get_unused_fd_flags(unsigned flags, unsigned long nofile)
545{
546 return __alloc_fd(current->files, 0, nofile, flags);
547}
548
Al Viro1a7bd222012-08-12 17:18:05 -0400549int get_unused_fd_flags(unsigned flags)
Al Viro1027abe2008-07-30 04:13:04 -0400550{
Jens Axboe4022e7a2020-03-19 19:23:18 -0600551 return __get_unused_fd_flags(flags, rlimit(RLIMIT_NOFILE));
Al Viro1027abe2008-07-30 04:13:04 -0400552}
Al Viro1a7bd222012-08-12 17:18:05 -0400553EXPORT_SYMBOL(get_unused_fd_flags);
Al Viro56007ca2012-08-15 21:03:26 -0400554
555static void __put_unused_fd(struct files_struct *files, unsigned int fd)
556{
557 struct fdtable *fdt = files_fdtable(files);
558 __clear_open_fd(fd, fdt);
559 if (fd < files->next_fd)
560 files->next_fd = fd;
561}
562
563void put_unused_fd(unsigned int fd)
564{
565 struct files_struct *files = current->files;
566 spin_lock(&files->file_lock);
567 __put_unused_fd(files, fd);
568 spin_unlock(&files->file_lock);
569}
570
571EXPORT_SYMBOL(put_unused_fd);
572
573/*
574 * Install a file pointer in the fd array.
575 *
576 * The VFS is full of places where we drop the files lock between
577 * setting the open_fds bitmap and installing the file in the file
578 * array. At any such point, we are vulnerable to a dup2() race
579 * installing a file in the array before us. We need to detect this and
580 * fput() the struct file we are about to overwrite in this case.
581 *
582 * It should never happen - if we allow dup2() do it, _really_ bad things
583 * will follow.
Al Virof869e8a2012-08-15 21:06:33 -0400584 *
585 * NOTE: __fd_install() variant is really, really low-level; don't
586 * use it unless you are forced to by truly lousy API shoved down
587 * your throat. 'files' *MUST* be either current->files or obtained
588 * by get_files_struct(current) done by whoever had given it to you,
589 * or really bad things will happen. Normally you want to use
590 * fd_install() instead.
Al Viro56007ca2012-08-15 21:03:26 -0400591 */
592
Al Virof869e8a2012-08-15 21:06:33 -0400593void __fd_install(struct files_struct *files, unsigned int fd,
594 struct file *file)
Al Viro56007ca2012-08-15 21:03:26 -0400595{
Al Viro56007ca2012-08-15 21:03:26 -0400596 struct fdtable *fdt;
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200597
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200598 rcu_read_lock_sched();
599
Mateusz Guzikc02b1a92017-10-03 12:58:15 +0200600 if (unlikely(files->resize_in_progress)) {
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200601 rcu_read_unlock_sched();
Mateusz Guzikc02b1a92017-10-03 12:58:15 +0200602 spin_lock(&files->file_lock);
603 fdt = files_fdtable(files);
604 BUG_ON(fdt->fd[fd] != NULL);
605 rcu_assign_pointer(fdt->fd[fd], file);
606 spin_unlock(&files->file_lock);
607 return;
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200608 }
609 /* coupled with smp_wmb() in expand_fdtable() */
610 smp_rmb();
611 fdt = rcu_dereference_sched(files->fdt);
Al Viro56007ca2012-08-15 21:03:26 -0400612 BUG_ON(fdt->fd[fd] != NULL);
613 rcu_assign_pointer(fdt->fd[fd], file);
Eric Dumazet8a81252b2015-06-30 15:54:08 +0200614 rcu_read_unlock_sched();
Al Viro56007ca2012-08-15 21:03:26 -0400615}
616
Al Virof869e8a2012-08-15 21:06:33 -0400617void fd_install(unsigned int fd, struct file *file)
618{
619 __fd_install(current->files, fd, file);
620}
621
Al Viro56007ca2012-08-15 21:03:26 -0400622EXPORT_SYMBOL(fd_install);
Al Viro0ee8cdf2012-08-15 21:12:10 -0400623
Christian Brauner278a5fb2019-05-24 11:30:34 +0200624static struct file *pick_file(struct files_struct *files, unsigned fd)
Al Viro483ce1d2012-08-19 12:04:24 -0400625{
Christian Brauner278a5fb2019-05-24 11:30:34 +0200626 struct file *file = NULL;
Al Viro483ce1d2012-08-19 12:04:24 -0400627 struct fdtable *fdt;
628
629 spin_lock(&files->file_lock);
630 fdt = files_fdtable(files);
631 if (fd >= fdt->max_fds)
632 goto out_unlock;
633 file = fdt->fd[fd];
634 if (!file)
635 goto out_unlock;
636 rcu_assign_pointer(fdt->fd[fd], NULL);
Al Viro483ce1d2012-08-19 12:04:24 -0400637 __put_unused_fd(files, fd);
Al Viro483ce1d2012-08-19 12:04:24 -0400638
639out_unlock:
640 spin_unlock(&files->file_lock);
Christian Brauner278a5fb2019-05-24 11:30:34 +0200641 return file;
642}
643
644/*
645 * The same warnings as for __alloc_fd()/__fd_install() apply here...
646 */
647int __close_fd(struct files_struct *files, unsigned fd)
648{
649 struct file *file;
650
651 file = pick_file(files, fd);
652 if (!file)
653 return -EBADF;
654
655 return filp_close(file, files);
Al Viro483ce1d2012-08-19 12:04:24 -0400656}
Dominik Brodowski2ca2a09d62018-03-11 11:34:55 +0100657EXPORT_SYMBOL(__close_fd); /* for ksys_close() */
Al Viro483ce1d2012-08-19 12:04:24 -0400658
Christian Brauner278a5fb2019-05-24 11:30:34 +0200659/**
660 * __close_range() - Close all file descriptors in a given range.
661 *
662 * @fd: starting file descriptor to close
663 * @max_fd: last file descriptor to close
664 *
665 * This closes a range of file descriptors. All file descriptors
666 * from @fd up to and including @max_fd are closed.
667 */
668int __close_range(struct files_struct *files, unsigned fd, unsigned max_fd)
669{
670 unsigned int cur_max;
671
672 if (fd > max_fd)
673 return -EINVAL;
674
675 rcu_read_lock();
676 cur_max = files_fdtable(files)->max_fds;
677 rcu_read_unlock();
678
679 /* cap to last valid index into fdtable */
680 cur_max--;
681
682 max_fd = min(max_fd, cur_max);
683 while (fd <= max_fd) {
684 struct file *file;
685
686 file = pick_file(files, fd++);
687 if (!file)
688 continue;
689
690 filp_close(file, files);
691 cond_resched();
692 }
693
694 return 0;
695}
696
Todd Kjos80cd7952018-12-14 15:58:21 -0800697/*
Jens Axboe6e802a42019-12-11 14:10:35 -0700698 * variant of __close_fd that gets a ref on the file for later fput.
699 * The caller must ensure that filp_close() called on the file, and then
700 * an fput().
Todd Kjos80cd7952018-12-14 15:58:21 -0800701 */
702int __close_fd_get_file(unsigned int fd, struct file **res)
703{
704 struct files_struct *files = current->files;
705 struct file *file;
706 struct fdtable *fdt;
707
708 spin_lock(&files->file_lock);
709 fdt = files_fdtable(files);
710 if (fd >= fdt->max_fds)
711 goto out_unlock;
712 file = fdt->fd[fd];
713 if (!file)
714 goto out_unlock;
715 rcu_assign_pointer(fdt->fd[fd], NULL);
716 __put_unused_fd(files, fd);
717 spin_unlock(&files->file_lock);
718 get_file(file);
719 *res = file;
Jens Axboe6e802a42019-12-11 14:10:35 -0700720 return 0;
Todd Kjos80cd7952018-12-14 15:58:21 -0800721
722out_unlock:
723 spin_unlock(&files->file_lock);
724 *res = NULL;
725 return -ENOENT;
726}
727
Al Viro6a6d27d2012-08-21 09:56:33 -0400728void do_close_on_exec(struct files_struct *files)
729{
730 unsigned i;
731 struct fdtable *fdt;
732
733 /* exec unshares first */
Al Viro6a6d27d2012-08-21 09:56:33 -0400734 spin_lock(&files->file_lock);
735 for (i = 0; ; i++) {
736 unsigned long set;
737 unsigned fd = i * BITS_PER_LONG;
738 fdt = files_fdtable(files);
739 if (fd >= fdt->max_fds)
740 break;
741 set = fdt->close_on_exec[i];
742 if (!set)
743 continue;
744 fdt->close_on_exec[i] = 0;
745 for ( ; set ; fd++, set >>= 1) {
746 struct file *file;
747 if (!(set & 1))
748 continue;
749 file = fdt->fd[fd];
750 if (!file)
751 continue;
752 rcu_assign_pointer(fdt->fd[fd], NULL);
753 __put_unused_fd(files, fd);
754 spin_unlock(&files->file_lock);
755 filp_close(file, files);
756 cond_resched();
757 spin_lock(&files->file_lock);
758 }
759
760 }
761 spin_unlock(&files->file_lock);
762}
763
Sargun Dhillon5e876fb2020-01-07 09:59:24 -0800764static struct file *__fget_files(struct files_struct *files, unsigned int fd,
765 fmode_t mask, unsigned int refs)
Al Viro0ee8cdf2012-08-15 21:12:10 -0400766{
Oleg Nesterov1deb46e2014-01-13 16:48:19 +0100767 struct file *file;
Al Viro0ee8cdf2012-08-15 21:12:10 -0400768
769 rcu_read_lock();
Eric Dumazet5ba97d22015-06-29 17:10:30 +0200770loop:
Al Viro0ee8cdf2012-08-15 21:12:10 -0400771 file = fcheck_files(files, fd);
772 if (file) {
Eric Dumazet5ba97d22015-06-29 17:10:30 +0200773 /* File object ref couldn't be taken.
774 * dup2() atomicity guarantee is the reason
775 * we loop to catch the new file (or NULL pointer)
776 */
777 if (file->f_mode & mask)
Al Viro0ee8cdf2012-08-15 21:12:10 -0400778 file = NULL;
Jens Axboe091141a2018-11-21 10:32:39 -0700779 else if (!get_file_rcu_many(file, refs))
Eric Dumazet5ba97d22015-06-29 17:10:30 +0200780 goto loop;
Al Viro0ee8cdf2012-08-15 21:12:10 -0400781 }
782 rcu_read_unlock();
783
784 return file;
785}
786
Sargun Dhillon5e876fb2020-01-07 09:59:24 -0800787static inline struct file *__fget(unsigned int fd, fmode_t mask,
788 unsigned int refs)
789{
790 return __fget_files(current->files, fd, mask, refs);
791}
792
Jens Axboe091141a2018-11-21 10:32:39 -0700793struct file *fget_many(unsigned int fd, unsigned int refs)
794{
795 return __fget(fd, FMODE_PATH, refs);
796}
797
Oleg Nesterov1deb46e2014-01-13 16:48:19 +0100798struct file *fget(unsigned int fd)
799{
Jens Axboe091141a2018-11-21 10:32:39 -0700800 return __fget(fd, FMODE_PATH, 1);
Oleg Nesterov1deb46e2014-01-13 16:48:19 +0100801}
Al Viro0ee8cdf2012-08-15 21:12:10 -0400802EXPORT_SYMBOL(fget);
803
804struct file *fget_raw(unsigned int fd)
805{
Jens Axboe091141a2018-11-21 10:32:39 -0700806 return __fget(fd, 0, 1);
Al Viro0ee8cdf2012-08-15 21:12:10 -0400807}
Al Viro0ee8cdf2012-08-15 21:12:10 -0400808EXPORT_SYMBOL(fget_raw);
809
Sargun Dhillon5e876fb2020-01-07 09:59:24 -0800810struct file *fget_task(struct task_struct *task, unsigned int fd)
811{
812 struct file *file = NULL;
813
814 task_lock(task);
815 if (task->files)
816 file = __fget_files(task->files, fd, 0, 1);
817 task_unlock(task);
818
819 return file;
820}
821
Al Viro0ee8cdf2012-08-15 21:12:10 -0400822/*
823 * Lightweight file lookup - no refcnt increment if fd table isn't shared.
824 *
825 * You can use this instead of fget if you satisfy all of the following
826 * conditions:
827 * 1) You must call fput_light before exiting the syscall and returning control
828 * to userspace (i.e. you cannot remember the returned struct file * after
829 * returning to userspace).
830 * 2) You must not call filp_close on the returned struct file * in between
831 * calls to fget_light and fput_light.
832 * 3) You must not clone the current task in between the calls to fget_light
833 * and fput_light.
834 *
835 * The fput_needed flag returned by fget_light should be passed to the
836 * corresponding fput_light.
837 */
Al Virobd2a31d2014-03-04 14:54:22 -0500838static unsigned long __fget_light(unsigned int fd, fmode_t mask)
Al Viro0ee8cdf2012-08-15 21:12:10 -0400839{
Al Viro0ee8cdf2012-08-15 21:12:10 -0400840 struct files_struct *files = current->files;
Oleg Nesterovad461832014-01-13 16:48:40 +0100841 struct file *file;
Al Viro0ee8cdf2012-08-15 21:12:10 -0400842
Al Viro0ee8cdf2012-08-15 21:12:10 -0400843 if (atomic_read(&files->count) == 1) {
Oleg Nesterova8d4b832014-01-11 19:19:32 +0100844 file = __fcheck_files(files, fd);
Al Virobd2a31d2014-03-04 14:54:22 -0500845 if (!file || unlikely(file->f_mode & mask))
846 return 0;
847 return (unsigned long)file;
Al Viro0ee8cdf2012-08-15 21:12:10 -0400848 } else {
Jens Axboe091141a2018-11-21 10:32:39 -0700849 file = __fget(fd, mask, 1);
Al Virobd2a31d2014-03-04 14:54:22 -0500850 if (!file)
851 return 0;
852 return FDPUT_FPUT | (unsigned long)file;
Al Viro0ee8cdf2012-08-15 21:12:10 -0400853 }
Al Viro0ee8cdf2012-08-15 21:12:10 -0400854}
Al Virobd2a31d2014-03-04 14:54:22 -0500855unsigned long __fdget(unsigned int fd)
Oleg Nesterovad461832014-01-13 16:48:40 +0100856{
Al Virobd2a31d2014-03-04 14:54:22 -0500857 return __fget_light(fd, FMODE_PATH);
Oleg Nesterovad461832014-01-13 16:48:40 +0100858}
Al Virobd2a31d2014-03-04 14:54:22 -0500859EXPORT_SYMBOL(__fdget);
Al Viro0ee8cdf2012-08-15 21:12:10 -0400860
Al Virobd2a31d2014-03-04 14:54:22 -0500861unsigned long __fdget_raw(unsigned int fd)
Al Viro0ee8cdf2012-08-15 21:12:10 -0400862{
Al Virobd2a31d2014-03-04 14:54:22 -0500863 return __fget_light(fd, 0);
Al Viro0ee8cdf2012-08-15 21:12:10 -0400864}
Al Virofe17f222012-08-21 11:48:11 -0400865
Al Virobd2a31d2014-03-04 14:54:22 -0500866unsigned long __fdget_pos(unsigned int fd)
867{
Eric Biggers99aea682014-03-16 15:47:48 -0500868 unsigned long v = __fdget(fd);
869 struct file *file = (struct file *)(v & ~3);
Al Virobd2a31d2014-03-04 14:54:22 -0500870
Linus Torvalds2be7d342019-11-26 11:34:06 -0800871 if (file && (file->f_mode & FMODE_ATOMIC_POS)) {
Al Virobd2a31d2014-03-04 14:54:22 -0500872 if (file_count(file) > 1) {
873 v |= FDPUT_POS_UNLOCK;
874 mutex_lock(&file->f_pos_lock);
875 }
876 }
Eric Biggers99aea682014-03-16 15:47:48 -0500877 return v;
Al Virobd2a31d2014-03-04 14:54:22 -0500878}
879
Al Viro63b6df12016-04-20 17:08:21 -0400880void __f_unlock_pos(struct file *f)
881{
882 mutex_unlock(&f->f_pos_lock);
883}
884
Al Virobd2a31d2014-03-04 14:54:22 -0500885/*
886 * We only lock f_pos if we have threads or if the file might be
887 * shared with another process. In both cases we'll have an elevated
888 * file count (done either by fdget() or by fork()).
889 */
890
Al Virofe17f222012-08-21 11:48:11 -0400891void set_close_on_exec(unsigned int fd, int flag)
892{
893 struct files_struct *files = current->files;
894 struct fdtable *fdt;
895 spin_lock(&files->file_lock);
896 fdt = files_fdtable(files);
897 if (flag)
898 __set_close_on_exec(fd, fdt);
899 else
900 __clear_close_on_exec(fd, fdt);
901 spin_unlock(&files->file_lock);
902}
903
904bool get_close_on_exec(unsigned int fd)
905{
906 struct files_struct *files = current->files;
907 struct fdtable *fdt;
908 bool res;
909 rcu_read_lock();
910 fdt = files_fdtable(files);
911 res = close_on_exec(fd, fdt);
912 rcu_read_unlock();
913 return res;
914}
915
Al Viro8280d162012-08-21 12:11:46 -0400916static int do_dup2(struct files_struct *files,
917 struct file *file, unsigned fd, unsigned flags)
Al Viroe9830942014-08-31 14:12:09 -0400918__releases(&files->file_lock)
Al Viro8280d162012-08-21 12:11:46 -0400919{
920 struct file *tofree;
921 struct fdtable *fdt;
922
923 /*
924 * We need to detect attempts to do dup2() over allocated but still
925 * not finished descriptor. NB: OpenBSD avoids that at the price of
926 * extra work in their equivalent of fget() - they insert struct
927 * file immediately after grabbing descriptor, mark it larval if
928 * more work (e.g. actual opening) is needed and make sure that
929 * fget() treats larval files as absent. Potentially interesting,
930 * but while extra work in fget() is trivial, locking implications
931 * and amount of surgery on open()-related paths in VFS are not.
932 * FreeBSD fails with -EBADF in the same situation, NetBSD "solution"
933 * deadlocks in rather amusing ways, AFAICS. All of that is out of
934 * scope of POSIX or SUS, since neither considers shared descriptor
935 * tables and this condition does not arise without those.
936 */
937 fdt = files_fdtable(files);
938 tofree = fdt->fd[fd];
939 if (!tofree && fd_is_open(fd, fdt))
940 goto Ebusy;
941 get_file(file);
942 rcu_assign_pointer(fdt->fd[fd], file);
943 __set_open_fd(fd, fdt);
944 if (flags & O_CLOEXEC)
945 __set_close_on_exec(fd, fdt);
946 else
947 __clear_close_on_exec(fd, fdt);
948 spin_unlock(&files->file_lock);
949
950 if (tofree)
951 filp_close(tofree, files);
952
953 return fd;
954
955Ebusy:
956 spin_unlock(&files->file_lock);
957 return -EBUSY;
958}
959
960int replace_fd(unsigned fd, struct file *file, unsigned flags)
961{
962 int err;
963 struct files_struct *files = current->files;
964
965 if (!file)
966 return __close_fd(files, fd);
967
968 if (fd >= rlimit(RLIMIT_NOFILE))
Al Viro08f05c42012-10-31 03:37:48 +0000969 return -EBADF;
Al Viro8280d162012-08-21 12:11:46 -0400970
971 spin_lock(&files->file_lock);
972 err = expand_files(files, fd);
973 if (unlikely(err < 0))
974 goto out_unlock;
975 return do_dup2(files, file, fd, flags);
976
977out_unlock:
978 spin_unlock(&files->file_lock);
979 return err;
980}
981
Dominik Brodowskic7248322018-03-11 11:34:40 +0100982static int ksys_dup3(unsigned int oldfd, unsigned int newfd, int flags)
Al Virofe17f222012-08-21 11:48:11 -0400983{
984 int err = -EBADF;
Al Viro8280d162012-08-21 12:11:46 -0400985 struct file *file;
986 struct files_struct *files = current->files;
Al Virofe17f222012-08-21 11:48:11 -0400987
988 if ((flags & ~O_CLOEXEC) != 0)
989 return -EINVAL;
990
Richard W.M. Jonesaed97642012-10-09 15:27:43 +0100991 if (unlikely(oldfd == newfd))
992 return -EINVAL;
993
Al Virofe17f222012-08-21 11:48:11 -0400994 if (newfd >= rlimit(RLIMIT_NOFILE))
Al Viro08f05c42012-10-31 03:37:48 +0000995 return -EBADF;
Al Virofe17f222012-08-21 11:48:11 -0400996
997 spin_lock(&files->file_lock);
998 err = expand_files(files, newfd);
999 file = fcheck(oldfd);
1000 if (unlikely(!file))
1001 goto Ebadf;
1002 if (unlikely(err < 0)) {
1003 if (err == -EMFILE)
1004 goto Ebadf;
1005 goto out_unlock;
1006 }
Al Viro8280d162012-08-21 12:11:46 -04001007 return do_dup2(files, file, newfd, flags);
Al Virofe17f222012-08-21 11:48:11 -04001008
1009Ebadf:
1010 err = -EBADF;
1011out_unlock:
1012 spin_unlock(&files->file_lock);
1013 return err;
1014}
1015
Dominik Brodowskic7248322018-03-11 11:34:40 +01001016SYSCALL_DEFINE3(dup3, unsigned int, oldfd, unsigned int, newfd, int, flags)
1017{
1018 return ksys_dup3(oldfd, newfd, flags);
1019}
1020
Al Virofe17f222012-08-21 11:48:11 -04001021SYSCALL_DEFINE2(dup2, unsigned int, oldfd, unsigned int, newfd)
1022{
1023 if (unlikely(newfd == oldfd)) { /* corner case */
1024 struct files_struct *files = current->files;
1025 int retval = oldfd;
1026
1027 rcu_read_lock();
1028 if (!fcheck_files(files, oldfd))
1029 retval = -EBADF;
1030 rcu_read_unlock();
1031 return retval;
1032 }
Dominik Brodowskic7248322018-03-11 11:34:40 +01001033 return ksys_dup3(oldfd, newfd, 0);
Al Virofe17f222012-08-21 11:48:11 -04001034}
1035
Dominik Brodowski74f1a292020-01-01 20:05:03 +01001036int ksys_dup(unsigned int fildes)
Al Virofe17f222012-08-21 11:48:11 -04001037{
1038 int ret = -EBADF;
1039 struct file *file = fget_raw(fildes);
1040
1041 if (file) {
Yann Droneaud8d10a032014-12-10 15:45:44 -08001042 ret = get_unused_fd_flags(0);
Al Virofe17f222012-08-21 11:48:11 -04001043 if (ret >= 0)
1044 fd_install(ret, file);
1045 else
1046 fput(file);
1047 }
1048 return ret;
1049}
1050
Dominik Brodowski74f1a292020-01-01 20:05:03 +01001051SYSCALL_DEFINE1(dup, unsigned int, fildes)
1052{
1053 return ksys_dup(fildes);
1054}
1055
Al Virofe17f222012-08-21 11:48:11 -04001056int f_dupfd(unsigned int from, struct file *file, unsigned flags)
1057{
1058 int err;
1059 if (from >= rlimit(RLIMIT_NOFILE))
1060 return -EINVAL;
1061 err = alloc_fd(from, flags);
1062 if (err >= 0) {
1063 get_file(file);
1064 fd_install(err, file);
1065 }
1066 return err;
1067}
Al Viroc3c073f2012-08-21 22:32:06 -04001068
1069int iterate_fd(struct files_struct *files, unsigned n,
1070 int (*f)(const void *, struct file *, unsigned),
1071 const void *p)
1072{
1073 struct fdtable *fdt;
Al Viroc3c073f2012-08-21 22:32:06 -04001074 int res = 0;
1075 if (!files)
1076 return 0;
1077 spin_lock(&files->file_lock);
Al Viroa77cfcb2012-11-29 22:57:33 -05001078 for (fdt = files_fdtable(files); n < fdt->max_fds; n++) {
1079 struct file *file;
1080 file = rcu_dereference_check_fdtable(files, fdt->fd[n]);
1081 if (!file)
1082 continue;
1083 res = f(p, file, n);
1084 if (res)
1085 break;
Al Viroc3c073f2012-08-21 22:32:06 -04001086 }
1087 spin_unlock(&files->file_lock);
1088 return res;
1089}
1090EXPORT_SYMBOL(iterate_fd);