1 // SPDX-License-Identifier: GPL-2.0
5 * Copyright (C) 1998-1999, Stephen Tweedie and Bill Hawes
7 * Manage the dynamic fd arrays in the process files_struct.
10 #include <linux/syscalls.h>
11 #include <linux/export.h>
13 #include <linux/kernel.h>
15 #include <linux/sched/signal.h>
16 #include <linux/slab.h>
17 #include <linux/file.h>
18 #include <linux/fdtable.h>
19 #include <linux/bitops.h>
20 #include <linux/spinlock.h>
21 #include <linux/rcupdate.h>
22 #include <linux/close_range.h>
27 unsigned int sysctl_nr_open __read_mostly
= 1024*1024;
28 unsigned int sysctl_nr_open_min
= BITS_PER_LONG
;
29 /* our min() is unusable in constant expressions ;-/ */
30 #define __const_min(x, y) ((x) < (y) ? (x) : (y))
31 unsigned int sysctl_nr_open_max
=
32 __const_min(INT_MAX
, ~(size_t)0/sizeof(void *)) & -BITS_PER_LONG
;
34 static void __free_fdtable(struct fdtable
*fdt
)
37 kvfree(fdt
->open_fds
);
41 static void free_fdtable_rcu(struct rcu_head
*rcu
)
43 __free_fdtable(container_of(rcu
, struct fdtable
, rcu
));
46 #define BITBIT_NR(nr) BITS_TO_LONGS(BITS_TO_LONGS(nr))
47 #define BITBIT_SIZE(nr) (BITBIT_NR(nr) * sizeof(long))
50 * Copy 'count' fd bits from the old table to the new table and clear the extra
51 * space if any. This does not copy the file pointers. Called with the files
52 * spinlock held for write.
54 static void copy_fd_bitmaps(struct fdtable
*nfdt
, struct fdtable
*ofdt
,
57 unsigned int cpy
, set
;
59 cpy
= count
/ BITS_PER_BYTE
;
60 set
= (nfdt
->max_fds
- count
) / BITS_PER_BYTE
;
61 memcpy(nfdt
->open_fds
, ofdt
->open_fds
, cpy
);
62 memset((char *)nfdt
->open_fds
+ cpy
, 0, set
);
63 memcpy(nfdt
->close_on_exec
, ofdt
->close_on_exec
, cpy
);
64 memset((char *)nfdt
->close_on_exec
+ cpy
, 0, set
);
66 cpy
= BITBIT_SIZE(count
);
67 set
= BITBIT_SIZE(nfdt
->max_fds
) - cpy
;
68 memcpy(nfdt
->full_fds_bits
, ofdt
->full_fds_bits
, cpy
);
69 memset((char *)nfdt
->full_fds_bits
+ cpy
, 0, set
);
73 * Copy all file descriptors from the old table to the new, expanded table and
74 * clear the extra space. Called with the files spinlock held for write.
76 static void copy_fdtable(struct fdtable
*nfdt
, struct fdtable
*ofdt
)
80 BUG_ON(nfdt
->max_fds
< ofdt
->max_fds
);
82 cpy
= ofdt
->max_fds
* sizeof(struct file
*);
83 set
= (nfdt
->max_fds
- ofdt
->max_fds
) * sizeof(struct file
*);
84 memcpy(nfdt
->fd
, ofdt
->fd
, cpy
);
85 memset((char *)nfdt
->fd
+ cpy
, 0, set
);
87 copy_fd_bitmaps(nfdt
, ofdt
, ofdt
->max_fds
);
91 * Note how the fdtable bitmap allocations very much have to be a multiple of
92 * BITS_PER_LONG. This is not only because we walk those things in chunks of
93 * 'unsigned long' in some places, but simply because that is how the Linux
94 * kernel bitmaps are defined to work: they are not "bits in an array of bytes",
95 * they are very much "bits in an array of unsigned long".
97 * The ALIGN(nr, BITS_PER_LONG) here is for clarity: since we just multiplied
98 * by that "1024/sizeof(ptr)" before, we already know there are sufficient
99 * clear low bits. Clang seems to realize that, gcc ends up being confused.
101 * On a 128-bit machine, the ALIGN() would actually matter. In the meantime,
102 * let's consider it documentation (and maybe a test-case for gcc to improve
103 * its code generation ;)
105 static struct fdtable
* alloc_fdtable(unsigned int nr
)
111 * Figure out how many fds we actually want to support in this fdtable.
112 * Allocation steps are keyed to the size of the fdarray, since it
113 * grows far faster than any of the other dynamic data. We try to fit
114 * the fdarray into comfortable page-tuned chunks: starting at 1024B
115 * and growing in powers of two from there on.
117 nr
/= (1024 / sizeof(struct file
*));
118 nr
= roundup_pow_of_two(nr
+ 1);
119 nr
*= (1024 / sizeof(struct file
*));
120 nr
= ALIGN(nr
, BITS_PER_LONG
);
122 * Note that this can drive nr *below* what we had passed if sysctl_nr_open
123 * had been set lower between the check in expand_files() and here. Deal
124 * with that in caller, it's cheaper that way.
126 * We make sure that nr remains a multiple of BITS_PER_LONG - otherwise
127 * bitmaps handling below becomes unpleasant, to put it mildly...
129 if (unlikely(nr
> sysctl_nr_open
))
130 nr
= ((sysctl_nr_open
- 1) | (BITS_PER_LONG
- 1)) + 1;
132 fdt
= kmalloc(sizeof(struct fdtable
), GFP_KERNEL_ACCOUNT
);
136 data
= kvmalloc_array(nr
, sizeof(struct file
*), GFP_KERNEL_ACCOUNT
);
141 data
= kvmalloc(max_t(size_t,
142 2 * nr
/ BITS_PER_BYTE
+ BITBIT_SIZE(nr
), L1_CACHE_BYTES
),
146 fdt
->open_fds
= data
;
147 data
+= nr
/ BITS_PER_BYTE
;
148 fdt
->close_on_exec
= data
;
149 data
+= nr
/ BITS_PER_BYTE
;
150 fdt
->full_fds_bits
= data
;
163 * Expand the file descriptor table.
164 * This function will allocate a new fdtable and both fd array and fdset, of
166 * Return <0 error code on error; 1 on successful completion.
167 * The files->file_lock should be held on entry, and will be held on exit.
169 static int expand_fdtable(struct files_struct
*files
, unsigned int nr
)
170 __releases(files
->file_lock
)
171 __acquires(files
->file_lock
)
173 struct fdtable
*new_fdt
, *cur_fdt
;
175 spin_unlock(&files
->file_lock
);
176 new_fdt
= alloc_fdtable(nr
);
178 /* make sure all fd_install() have seen resize_in_progress
179 * or have finished their rcu_read_lock_sched() section.
181 if (atomic_read(&files
->count
) > 1)
184 spin_lock(&files
->file_lock
);
188 * extremely unlikely race - sysctl_nr_open decreased between the check in
189 * caller and alloc_fdtable(). Cheaper to catch it here...
191 if (unlikely(new_fdt
->max_fds
<= nr
)) {
192 __free_fdtable(new_fdt
);
195 cur_fdt
= files_fdtable(files
);
196 BUG_ON(nr
< cur_fdt
->max_fds
);
197 copy_fdtable(new_fdt
, cur_fdt
);
198 rcu_assign_pointer(files
->fdt
, new_fdt
);
199 if (cur_fdt
!= &files
->fdtab
)
200 call_rcu(&cur_fdt
->rcu
, free_fdtable_rcu
);
201 /* coupled with smp_rmb() in fd_install() */
208 * This function will expand the file structures, if the requested size exceeds
209 * the current capacity and there is room for expansion.
210 * Return <0 error code on error; 0 when nothing done; 1 when files were
211 * expanded and execution may have blocked.
212 * The files->file_lock should be held on entry, and will be held on exit.
214 static int expand_files(struct files_struct
*files
, unsigned int nr
)
215 __releases(files
->file_lock
)
216 __acquires(files
->file_lock
)
222 fdt
= files_fdtable(files
);
224 /* Do we need to expand? */
225 if (nr
< fdt
->max_fds
)
229 if (nr
>= sysctl_nr_open
)
232 if (unlikely(files
->resize_in_progress
)) {
233 spin_unlock(&files
->file_lock
);
235 wait_event(files
->resize_wait
, !files
->resize_in_progress
);
236 spin_lock(&files
->file_lock
);
240 /* All good, so we try */
241 files
->resize_in_progress
= true;
242 expanded
= expand_fdtable(files
, nr
);
243 files
->resize_in_progress
= false;
245 wake_up_all(&files
->resize_wait
);
249 static inline void __set_close_on_exec(unsigned int fd
, struct fdtable
*fdt
)
251 __set_bit(fd
, fdt
->close_on_exec
);
254 static inline void __clear_close_on_exec(unsigned int fd
, struct fdtable
*fdt
)
256 if (test_bit(fd
, fdt
->close_on_exec
))
257 __clear_bit(fd
, fdt
->close_on_exec
);
260 static inline void __set_open_fd(unsigned int fd
, struct fdtable
*fdt
)
262 __set_bit(fd
, fdt
->open_fds
);
264 if (!~fdt
->open_fds
[fd
])
265 __set_bit(fd
, fdt
->full_fds_bits
);
268 static inline void __clear_open_fd(unsigned int fd
, struct fdtable
*fdt
)
270 __clear_bit(fd
, fdt
->open_fds
);
271 __clear_bit(fd
/ BITS_PER_LONG
, fdt
->full_fds_bits
);
274 static unsigned int count_open_files(struct fdtable
*fdt
)
276 unsigned int size
= fdt
->max_fds
;
279 /* Find the last open fd */
280 for (i
= size
/ BITS_PER_LONG
; i
> 0; ) {
281 if (fdt
->open_fds
[--i
])
284 i
= (i
+ 1) * BITS_PER_LONG
;
289 * Note that a sane fdtable size always has to be a multiple of
290 * BITS_PER_LONG, since we have bitmaps that are sized by this.
292 * 'max_fds' will normally already be properly aligned, but it
293 * turns out that in the close_range() -> __close_range() ->
294 * unshare_fd() -> dup_fd() -> sane_fdtable_size() we can end
295 * up having a 'max_fds' value that isn't already aligned.
297 * Rather than make close_range() have to worry about this,
298 * just make that BITS_PER_LONG alignment be part of a sane
299 * fdtable size. Becuase that's really what it is.
301 static unsigned int sane_fdtable_size(struct fdtable
*fdt
, unsigned int max_fds
)
305 count
= count_open_files(fdt
);
306 if (max_fds
< NR_OPEN_DEFAULT
)
307 max_fds
= NR_OPEN_DEFAULT
;
308 return ALIGN(min(count
, max_fds
), BITS_PER_LONG
);
312 * Allocate a new files structure and copy contents from the
313 * passed in files structure.
314 * errorp will be valid only when the returned files_struct is NULL.
316 struct files_struct
*dup_fd(struct files_struct
*oldf
, unsigned int max_fds
, int *errorp
)
318 struct files_struct
*newf
;
319 struct file
**old_fds
, **new_fds
;
320 unsigned int open_files
, i
;
321 struct fdtable
*old_fdt
, *new_fdt
;
324 newf
= kmem_cache_alloc(files_cachep
, GFP_KERNEL
);
328 atomic_set(&newf
->count
, 1);
330 spin_lock_init(&newf
->file_lock
);
331 newf
->resize_in_progress
= false;
332 init_waitqueue_head(&newf
->resize_wait
);
334 new_fdt
= &newf
->fdtab
;
335 new_fdt
->max_fds
= NR_OPEN_DEFAULT
;
336 new_fdt
->close_on_exec
= newf
->close_on_exec_init
;
337 new_fdt
->open_fds
= newf
->open_fds_init
;
338 new_fdt
->full_fds_bits
= newf
->full_fds_bits_init
;
339 new_fdt
->fd
= &newf
->fd_array
[0];
341 spin_lock(&oldf
->file_lock
);
342 old_fdt
= files_fdtable(oldf
);
343 open_files
= sane_fdtable_size(old_fdt
, max_fds
);
346 * Check whether we need to allocate a larger fd array and fd set.
348 while (unlikely(open_files
> new_fdt
->max_fds
)) {
349 spin_unlock(&oldf
->file_lock
);
351 if (new_fdt
!= &newf
->fdtab
)
352 __free_fdtable(new_fdt
);
354 new_fdt
= alloc_fdtable(open_files
- 1);
360 /* beyond sysctl_nr_open; nothing to do */
361 if (unlikely(new_fdt
->max_fds
< open_files
)) {
362 __free_fdtable(new_fdt
);
368 * Reacquire the oldf lock and a pointer to its fd table
369 * who knows it may have a new bigger fd table. We need
370 * the latest pointer.
372 spin_lock(&oldf
->file_lock
);
373 old_fdt
= files_fdtable(oldf
);
374 open_files
= sane_fdtable_size(old_fdt
, max_fds
);
377 copy_fd_bitmaps(new_fdt
, old_fdt
, open_files
);
379 old_fds
= old_fdt
->fd
;
380 new_fds
= new_fdt
->fd
;
382 for (i
= open_files
; i
!= 0; i
--) {
383 struct file
*f
= *old_fds
++;
388 * The fd may be claimed in the fd bitmap but not yet
389 * instantiated in the files array if a sibling thread
390 * is partway through open(). So make sure that this
391 * fd is available to the new process.
393 __clear_open_fd(open_files
- i
, new_fdt
);
395 rcu_assign_pointer(*new_fds
++, f
);
397 spin_unlock(&oldf
->file_lock
);
399 /* clear the remainder */
400 memset(new_fds
, 0, (new_fdt
->max_fds
- open_files
) * sizeof(struct file
*));
402 rcu_assign_pointer(newf
->fdt
, new_fdt
);
407 kmem_cache_free(files_cachep
, newf
);
412 static struct fdtable
*close_files(struct files_struct
* files
)
415 * It is safe to dereference the fd table without RCU or
416 * ->file_lock because this is the last reference to the
419 struct fdtable
*fdt
= rcu_dereference_raw(files
->fdt
);
420 unsigned int i
, j
= 0;
424 i
= j
* BITS_PER_LONG
;
425 if (i
>= fdt
->max_fds
)
427 set
= fdt
->open_fds
[j
++];
430 struct file
* file
= xchg(&fdt
->fd
[i
], NULL
);
432 filp_close(file
, files
);
444 void put_files_struct(struct files_struct
*files
)
446 if (atomic_dec_and_test(&files
->count
)) {
447 struct fdtable
*fdt
= close_files(files
);
449 /* free the arrays if they are not embedded */
450 if (fdt
!= &files
->fdtab
)
452 kmem_cache_free(files_cachep
, files
);
456 void exit_files(struct task_struct
*tsk
)
458 struct files_struct
* files
= tsk
->files
;
464 put_files_struct(files
);
468 struct files_struct init_files
= {
469 .count
= ATOMIC_INIT(1),
470 .fdt
= &init_files
.fdtab
,
472 .max_fds
= NR_OPEN_DEFAULT
,
473 .fd
= &init_files
.fd_array
[0],
474 .close_on_exec
= init_files
.close_on_exec_init
,
475 .open_fds
= init_files
.open_fds_init
,
476 .full_fds_bits
= init_files
.full_fds_bits_init
,
478 .file_lock
= __SPIN_LOCK_UNLOCKED(init_files
.file_lock
),
479 .resize_wait
= __WAIT_QUEUE_HEAD_INITIALIZER(init_files
.resize_wait
),
482 static unsigned int find_next_fd(struct fdtable
*fdt
, unsigned int start
)
484 unsigned int maxfd
= fdt
->max_fds
;
485 unsigned int maxbit
= maxfd
/ BITS_PER_LONG
;
486 unsigned int bitbit
= start
/ BITS_PER_LONG
;
488 bitbit
= find_next_zero_bit(fdt
->full_fds_bits
, maxbit
, bitbit
) * BITS_PER_LONG
;
493 return find_next_zero_bit(fdt
->open_fds
, maxfd
, start
);
497 * allocate a file descriptor, mark it busy.
499 static int alloc_fd(unsigned start
, unsigned end
, unsigned flags
)
501 struct files_struct
*files
= current
->files
;
506 spin_lock(&files
->file_lock
);
508 fdt
= files_fdtable(files
);
510 if (fd
< files
->next_fd
)
513 if (fd
< fdt
->max_fds
)
514 fd
= find_next_fd(fdt
, fd
);
517 * N.B. For clone tasks sharing a files structure, this test
518 * will limit the total number of files that can be opened.
524 error
= expand_files(files
, fd
);
529 * If we needed to expand the fs array we
530 * might have blocked - try again.
535 if (start
<= files
->next_fd
)
536 files
->next_fd
= fd
+ 1;
538 __set_open_fd(fd
, fdt
);
539 if (flags
& O_CLOEXEC
)
540 __set_close_on_exec(fd
, fdt
);
542 __clear_close_on_exec(fd
, fdt
);
546 if (rcu_access_pointer(fdt
->fd
[fd
]) != NULL
) {
547 printk(KERN_WARNING
"alloc_fd: slot %d not NULL!\n", fd
);
548 rcu_assign_pointer(fdt
->fd
[fd
], NULL
);
553 spin_unlock(&files
->file_lock
);
557 int __get_unused_fd_flags(unsigned flags
, unsigned long nofile
)
559 return alloc_fd(0, nofile
, flags
);
562 int get_unused_fd_flags(unsigned flags
)
564 return __get_unused_fd_flags(flags
, rlimit(RLIMIT_NOFILE
));
566 EXPORT_SYMBOL(get_unused_fd_flags
);
568 static void __put_unused_fd(struct files_struct
*files
, unsigned int fd
)
570 struct fdtable
*fdt
= files_fdtable(files
);
571 __clear_open_fd(fd
, fdt
);
572 if (fd
< files
->next_fd
)
576 void put_unused_fd(unsigned int fd
)
578 struct files_struct
*files
= current
->files
;
579 spin_lock(&files
->file_lock
);
580 __put_unused_fd(files
, fd
);
581 spin_unlock(&files
->file_lock
);
584 EXPORT_SYMBOL(put_unused_fd
);
587 * Install a file pointer in the fd array.
589 * The VFS is full of places where we drop the files lock between
590 * setting the open_fds bitmap and installing the file in the file
591 * array. At any such point, we are vulnerable to a dup2() race
592 * installing a file in the array before us. We need to detect this and
593 * fput() the struct file we are about to overwrite in this case.
595 * It should never happen - if we allow dup2() do it, _really_ bad things
598 * This consumes the "file" refcount, so callers should treat it
599 * as if they had called fput(file).
602 void fd_install(unsigned int fd
, struct file
*file
)
604 struct files_struct
*files
= current
->files
;
607 rcu_read_lock_sched();
609 if (unlikely(files
->resize_in_progress
)) {
610 rcu_read_unlock_sched();
611 spin_lock(&files
->file_lock
);
612 fdt
= files_fdtable(files
);
613 BUG_ON(fdt
->fd
[fd
] != NULL
);
614 rcu_assign_pointer(fdt
->fd
[fd
], file
);
615 spin_unlock(&files
->file_lock
);
618 /* coupled with smp_wmb() in expand_fdtable() */
620 fdt
= rcu_dereference_sched(files
->fdt
);
621 BUG_ON(fdt
->fd
[fd
] != NULL
);
622 rcu_assign_pointer(fdt
->fd
[fd
], file
);
623 rcu_read_unlock_sched();
626 EXPORT_SYMBOL(fd_install
);
629 * pick_file - return file associatd with fd
630 * @files: file struct to retrieve file from
631 * @fd: file descriptor to retrieve file for
633 * Context: files_lock must be held.
635 * Returns: The file associated with @fd (NULL if @fd is not open)
637 static struct file
*pick_file(struct files_struct
*files
, unsigned fd
)
639 struct fdtable
*fdt
= files_fdtable(files
);
642 if (fd
>= fdt
->max_fds
)
647 rcu_assign_pointer(fdt
->fd
[fd
], NULL
);
648 __put_unused_fd(files
, fd
);
653 int close_fd(unsigned fd
)
655 struct files_struct
*files
= current
->files
;
658 spin_lock(&files
->file_lock
);
659 file
= pick_file(files
, fd
);
660 spin_unlock(&files
->file_lock
);
664 return filp_close(file
, files
);
666 EXPORT_SYMBOL(close_fd
); /* for ksys_close() */
669 * last_fd - return last valid index into fd table
670 * @cur_fds: files struct
672 * Context: Either rcu read lock or files_lock must be held.
674 * Returns: Last valid index into fdtable.
676 static inline unsigned last_fd(struct fdtable
*fdt
)
678 return fdt
->max_fds
- 1;
681 static inline void __range_cloexec(struct files_struct
*cur_fds
,
682 unsigned int fd
, unsigned int max_fd
)
686 /* make sure we're using the correct maximum value */
687 spin_lock(&cur_fds
->file_lock
);
688 fdt
= files_fdtable(cur_fds
);
689 max_fd
= min(last_fd(fdt
), max_fd
);
691 bitmap_set(fdt
->close_on_exec
, fd
, max_fd
- fd
+ 1);
692 spin_unlock(&cur_fds
->file_lock
);
695 static inline void __range_close(struct files_struct
*cur_fds
, unsigned int fd
,
701 n
= last_fd(files_fdtable(cur_fds
));
703 max_fd
= min(max_fd
, n
);
705 while (fd
<= max_fd
) {
708 spin_lock(&cur_fds
->file_lock
);
709 file
= pick_file(cur_fds
, fd
++);
710 spin_unlock(&cur_fds
->file_lock
);
713 /* found a valid file to close */
714 filp_close(file
, cur_fds
);
721 * __close_range() - Close all file descriptors in a given range.
723 * @fd: starting file descriptor to close
724 * @max_fd: last file descriptor to close
726 * This closes a range of file descriptors. All file descriptors
727 * from @fd up to and including @max_fd are closed.
729 int __close_range(unsigned fd
, unsigned max_fd
, unsigned int flags
)
731 struct task_struct
*me
= current
;
732 struct files_struct
*cur_fds
= me
->files
, *fds
= NULL
;
734 if (flags
& ~(CLOSE_RANGE_UNSHARE
| CLOSE_RANGE_CLOEXEC
))
740 if (flags
& CLOSE_RANGE_UNSHARE
) {
742 unsigned int max_unshare_fds
= NR_OPEN_MAX
;
745 * If the caller requested all fds to be made cloexec we always
746 * copy all of the file descriptors since they still want to
749 if (!(flags
& CLOSE_RANGE_CLOEXEC
)) {
751 * If the requested range is greater than the current
752 * maximum, we're closing everything so only copy all
753 * file descriptors beneath the lowest file descriptor.
756 if (max_fd
>= last_fd(files_fdtable(cur_fds
)))
757 max_unshare_fds
= fd
;
761 ret
= unshare_fd(CLONE_FILES
, max_unshare_fds
, &fds
);
766 * We used to share our file descriptor table, and have now
767 * created a private one, make sure we're using it below.
773 if (flags
& CLOSE_RANGE_CLOEXEC
)
774 __range_cloexec(cur_fds
, fd
, max_fd
);
776 __range_close(cur_fds
, fd
, max_fd
);
780 * We're done closing the files we were supposed to. Time to install
781 * the new file descriptor table and drop the old one.
786 put_files_struct(fds
);
793 * See close_fd_get_file() below, this variant assumes current->files->file_lock
796 struct file
*__close_fd_get_file(unsigned int fd
)
798 return pick_file(current
->files
, fd
);
802 * variant of close_fd that gets a ref on the file for later fput.
803 * The caller must ensure that filp_close() called on the file.
805 struct file
*close_fd_get_file(unsigned int fd
)
807 struct files_struct
*files
= current
->files
;
810 spin_lock(&files
->file_lock
);
811 file
= pick_file(files
, fd
);
812 spin_unlock(&files
->file_lock
);
817 void do_close_on_exec(struct files_struct
*files
)
822 /* exec unshares first */
823 spin_lock(&files
->file_lock
);
826 unsigned fd
= i
* BITS_PER_LONG
;
827 fdt
= files_fdtable(files
);
828 if (fd
>= fdt
->max_fds
)
830 set
= fdt
->close_on_exec
[i
];
833 fdt
->close_on_exec
[i
] = 0;
834 for ( ; set
; fd
++, set
>>= 1) {
841 rcu_assign_pointer(fdt
->fd
[fd
], NULL
);
842 __put_unused_fd(files
, fd
);
843 spin_unlock(&files
->file_lock
);
844 filp_close(file
, files
);
846 spin_lock(&files
->file_lock
);
850 spin_unlock(&files
->file_lock
);
853 static inline struct file
*__fget_files_rcu(struct files_struct
*files
,
854 unsigned int fd
, fmode_t mask
)
858 struct fdtable
*fdt
= rcu_dereference_raw(files
->fdt
);
859 struct file __rcu
**fdentry
;
861 if (unlikely(fd
>= fdt
->max_fds
))
864 fdentry
= fdt
->fd
+ array_index_nospec(fd
, fdt
->max_fds
);
865 file
= rcu_dereference_raw(*fdentry
);
869 if (unlikely(file
->f_mode
& mask
))
873 * Ok, we have a file pointer. However, because we do
874 * this all locklessly under RCU, we may be racing with
875 * that file being closed.
877 * Such a race can take two forms:
879 * (a) the file ref already went down to zero,
880 * and get_file_rcu() fails. Just try again:
882 if (unlikely(!get_file_rcu(file
)))
886 * (b) the file table entry has changed under us.
887 * Note that we don't need to re-check the 'fdt->fd'
888 * pointer having changed, because it always goes
889 * hand-in-hand with 'fdt'.
891 * If so, we need to put our ref and try again.
893 if (unlikely(rcu_dereference_raw(files
->fdt
) != fdt
) ||
894 unlikely(rcu_dereference_raw(*fdentry
) != file
)) {
900 * Ok, we have a ref to the file, and checked that it
907 static struct file
*__fget_files(struct files_struct
*files
, unsigned int fd
,
913 file
= __fget_files_rcu(files
, fd
, mask
);
919 static inline struct file
*__fget(unsigned int fd
, fmode_t mask
)
921 return __fget_files(current
->files
, fd
, mask
);
924 struct file
*fget(unsigned int fd
)
926 return __fget(fd
, FMODE_PATH
);
930 struct file
*fget_raw(unsigned int fd
)
932 return __fget(fd
, 0);
934 EXPORT_SYMBOL(fget_raw
);
936 struct file
*fget_task(struct task_struct
*task
, unsigned int fd
)
938 struct file
*file
= NULL
;
942 file
= __fget_files(task
->files
, fd
, 0);
948 struct file
*task_lookup_fd_rcu(struct task_struct
*task
, unsigned int fd
)
950 /* Must be called with rcu_read_lock held */
951 struct files_struct
*files
;
952 struct file
*file
= NULL
;
957 file
= files_lookup_fd_rcu(files
, fd
);
963 struct file
*task_lookup_next_fd_rcu(struct task_struct
*task
, unsigned int *ret_fd
)
965 /* Must be called with rcu_read_lock held */
966 struct files_struct
*files
;
967 unsigned int fd
= *ret_fd
;
968 struct file
*file
= NULL
;
973 for (; fd
< files_fdtable(files
)->max_fds
; fd
++) {
974 file
= files_lookup_fd_rcu(files
, fd
);
985 * Lightweight file lookup - no refcnt increment if fd table isn't shared.
987 * You can use this instead of fget if you satisfy all of the following
989 * 1) You must call fput_light before exiting the syscall and returning control
990 * to userspace (i.e. you cannot remember the returned struct file * after
991 * returning to userspace).
992 * 2) You must not call filp_close on the returned struct file * in between
993 * calls to fget_light and fput_light.
994 * 3) You must not clone the current task in between the calls to fget_light
997 * The fput_needed flag returned by fget_light should be passed to the
998 * corresponding fput_light.
1000 static unsigned long __fget_light(unsigned int fd
, fmode_t mask
)
1002 struct files_struct
*files
= current
->files
;
1005 if (atomic_read(&files
->count
) == 1) {
1006 file
= files_lookup_fd_raw(files
, fd
);
1007 if (!file
|| unlikely(file
->f_mode
& mask
))
1009 return (unsigned long)file
;
1011 file
= __fget(fd
, mask
);
1014 return FDPUT_FPUT
| (unsigned long)file
;
1017 unsigned long __fdget(unsigned int fd
)
1019 return __fget_light(fd
, FMODE_PATH
);
1021 EXPORT_SYMBOL(__fdget
);
1023 unsigned long __fdget_raw(unsigned int fd
)
1025 return __fget_light(fd
, 0);
1028 unsigned long __fdget_pos(unsigned int fd
)
1030 unsigned long v
= __fdget(fd
);
1031 struct file
*file
= (struct file
*)(v
& ~3);
1033 if (file
&& (file
->f_mode
& FMODE_ATOMIC_POS
)) {
1034 if (file_count(file
) > 1) {
1035 v
|= FDPUT_POS_UNLOCK
;
1036 mutex_lock(&file
->f_pos_lock
);
1042 void __f_unlock_pos(struct file
*f
)
1044 mutex_unlock(&f
->f_pos_lock
);
1048 * We only lock f_pos if we have threads or if the file might be
1049 * shared with another process. In both cases we'll have an elevated
1050 * file count (done either by fdget() or by fork()).
1053 void set_close_on_exec(unsigned int fd
, int flag
)
1055 struct files_struct
*files
= current
->files
;
1056 struct fdtable
*fdt
;
1057 spin_lock(&files
->file_lock
);
1058 fdt
= files_fdtable(files
);
1060 __set_close_on_exec(fd
, fdt
);
1062 __clear_close_on_exec(fd
, fdt
);
1063 spin_unlock(&files
->file_lock
);
1066 bool get_close_on_exec(unsigned int fd
)
1068 struct files_struct
*files
= current
->files
;
1069 struct fdtable
*fdt
;
1072 fdt
= files_fdtable(files
);
1073 res
= close_on_exec(fd
, fdt
);
1078 static int do_dup2(struct files_struct
*files
,
1079 struct file
*file
, unsigned fd
, unsigned flags
)
1080 __releases(&files
->file_lock
)
1082 struct file
*tofree
;
1083 struct fdtable
*fdt
;
1086 * We need to detect attempts to do dup2() over allocated but still
1087 * not finished descriptor. NB: OpenBSD avoids that at the price of
1088 * extra work in their equivalent of fget() - they insert struct
1089 * file immediately after grabbing descriptor, mark it larval if
1090 * more work (e.g. actual opening) is needed and make sure that
1091 * fget() treats larval files as absent. Potentially interesting,
1092 * but while extra work in fget() is trivial, locking implications
1093 * and amount of surgery on open()-related paths in VFS are not.
1094 * FreeBSD fails with -EBADF in the same situation, NetBSD "solution"
1095 * deadlocks in rather amusing ways, AFAICS. All of that is out of
1096 * scope of POSIX or SUS, since neither considers shared descriptor
1097 * tables and this condition does not arise without those.
1099 fdt
= files_fdtable(files
);
1100 tofree
= fdt
->fd
[fd
];
1101 if (!tofree
&& fd_is_open(fd
, fdt
))
1104 rcu_assign_pointer(fdt
->fd
[fd
], file
);
1105 __set_open_fd(fd
, fdt
);
1106 if (flags
& O_CLOEXEC
)
1107 __set_close_on_exec(fd
, fdt
);
1109 __clear_close_on_exec(fd
, fdt
);
1110 spin_unlock(&files
->file_lock
);
1113 filp_close(tofree
, files
);
1118 spin_unlock(&files
->file_lock
);
1122 int replace_fd(unsigned fd
, struct file
*file
, unsigned flags
)
1125 struct files_struct
*files
= current
->files
;
1128 return close_fd(fd
);
1130 if (fd
>= rlimit(RLIMIT_NOFILE
))
1133 spin_lock(&files
->file_lock
);
1134 err
= expand_files(files
, fd
);
1135 if (unlikely(err
< 0))
1137 return do_dup2(files
, file
, fd
, flags
);
1140 spin_unlock(&files
->file_lock
);
1145 * __receive_fd() - Install received file into file descriptor table
1146 * @file: struct file that was received from another process
1147 * @ufd: __user pointer to write new fd number to
1148 * @o_flags: the O_* flags to apply to the new fd entry
1150 * Installs a received file into the file descriptor table, with appropriate
1151 * checks and count updates. Optionally writes the fd number to userspace, if
1154 * This helper handles its own reference counting of the incoming
1157 * Returns newly install fd or -ve on error.
1159 int __receive_fd(struct file
*file
, int __user
*ufd
, unsigned int o_flags
)
1164 error
= security_file_receive(file
);
1168 new_fd
= get_unused_fd_flags(o_flags
);
1173 error
= put_user(new_fd
, ufd
);
1175 put_unused_fd(new_fd
);
1180 fd_install(new_fd
, get_file(file
));
1181 __receive_sock(file
);
1185 int receive_fd_replace(int new_fd
, struct file
*file
, unsigned int o_flags
)
1189 error
= security_file_receive(file
);
1192 error
= replace_fd(new_fd
, file
, o_flags
);
1195 __receive_sock(file
);
1199 int receive_fd(struct file
*file
, unsigned int o_flags
)
1201 return __receive_fd(file
, NULL
, o_flags
);
1203 EXPORT_SYMBOL_GPL(receive_fd
);
1205 static int ksys_dup3(unsigned int oldfd
, unsigned int newfd
, int flags
)
1209 struct files_struct
*files
= current
->files
;
1211 if ((flags
& ~O_CLOEXEC
) != 0)
1214 if (unlikely(oldfd
== newfd
))
1217 if (newfd
>= rlimit(RLIMIT_NOFILE
))
1220 spin_lock(&files
->file_lock
);
1221 err
= expand_files(files
, newfd
);
1222 file
= files_lookup_fd_locked(files
, oldfd
);
1223 if (unlikely(!file
))
1225 if (unlikely(err
< 0)) {
1230 return do_dup2(files
, file
, newfd
, flags
);
1235 spin_unlock(&files
->file_lock
);
1239 SYSCALL_DEFINE3(dup3
, unsigned int, oldfd
, unsigned int, newfd
, int, flags
)
1241 return ksys_dup3(oldfd
, newfd
, flags
);
1244 SYSCALL_DEFINE2(dup2
, unsigned int, oldfd
, unsigned int, newfd
)
1246 if (unlikely(newfd
== oldfd
)) { /* corner case */
1247 struct files_struct
*files
= current
->files
;
1251 if (!files_lookup_fd_rcu(files
, oldfd
))
1256 return ksys_dup3(oldfd
, newfd
, 0);
1259 SYSCALL_DEFINE1(dup
, unsigned int, fildes
)
1262 struct file
*file
= fget_raw(fildes
);
1265 ret
= get_unused_fd_flags(0);
1267 fd_install(ret
, file
);
1274 int f_dupfd(unsigned int from
, struct file
*file
, unsigned flags
)
1276 unsigned long nofile
= rlimit(RLIMIT_NOFILE
);
1280 err
= alloc_fd(from
, nofile
, flags
);
1283 fd_install(err
, file
);
1288 int iterate_fd(struct files_struct
*files
, unsigned n
,
1289 int (*f
)(const void *, struct file
*, unsigned),
1292 struct fdtable
*fdt
;
1296 spin_lock(&files
->file_lock
);
1297 for (fdt
= files_fdtable(files
); n
< fdt
->max_fds
; n
++) {
1299 file
= rcu_dereference_check_fdtable(files
, fdt
->fd
[n
]);
1302 res
= f(p
, file
, n
);
1306 spin_unlock(&files
->file_lock
);
1309 EXPORT_SYMBOL(iterate_fd
);