file.c 22.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8
/*
 *  linux/fs/file.c
 *
 *  Copyright (C) 1998-1999, Stephen Tweedie and Bill Hawes
 *
 *  Manage the dynamic fd arrays in the process files_struct.
 */

9
#include <linux/syscalls.h>
10
#include <linux/export.h>
L
Linus Torvalds 已提交
11 12
#include <linux/fs.h>
#include <linux/mm.h>
13
#include <linux/mmzone.h>
L
Linus Torvalds 已提交
14
#include <linux/time.h>
15
#include <linux/sched.h>
L
Linus Torvalds 已提交
16 17 18
#include <linux/slab.h>
#include <linux/vmalloc.h>
#include <linux/file.h>
A
Al Viro 已提交
19
#include <linux/fdtable.h>
L
Linus Torvalds 已提交
20
#include <linux/bitops.h>
21 22 23 24 25 26 27 28 29 30 31
#include <linux/interrupt.h>
#include <linux/spinlock.h>
#include <linux/rcupdate.h>
#include <linux/workqueue.h>

struct fdtable_defer {
	spinlock_t lock;
	struct work_struct wq;
	struct fdtable *next;
};

32
int sysctl_nr_open __read_mostly = 1024*1024;
33 34
int sysctl_nr_open_min = BITS_PER_LONG;
int sysctl_nr_open_max = 1024 * 1024; /* raised later */
35

36 37 38 39 40 41 42
/*
 * We use this list to defer free fdtables that have vmalloced
 * sets/arrays. By keeping a per-cpu list, we avoid having to embed
 * the work_struct in fdtable itself which avoids a 64 byte (i386) increase in
 * this per-task structure.
 */
static DEFINE_PER_CPU(struct fdtable_defer, fdtable_defer_list);
L
Linus Torvalds 已提交
43

44
static void *alloc_fdmem(size_t size)
L
Linus Torvalds 已提交
45
{
46 47 48 49 50 51 52 53 54
	/*
	 * Very large allocations can stress page reclaim, so fall back to
	 * vmalloc() if the allocation size will be considered "large" by the VM.
	 */
	if (size <= (PAGE_SIZE << PAGE_ALLOC_COSTLY_ORDER)) {
		void *data = kmalloc(size, GFP_KERNEL|__GFP_NOWARN);
		if (data != NULL)
			return data;
	}
55
	return vmalloc(size);
L
Linus Torvalds 已提交
56 57
}

58
static void free_fdmem(void *ptr)
L
Linus Torvalds 已提交
59
{
60
	is_vmalloc_addr(ptr) ? vfree(ptr) : kfree(ptr);
L
Linus Torvalds 已提交
61 62
}

63
static void __free_fdtable(struct fdtable *fdt)
L
Linus Torvalds 已提交
64
{
65 66 67
	free_fdmem(fdt->fd);
	free_fdmem(fdt->open_fds);
	kfree(fdt);
68
}
L
Linus Torvalds 已提交
69

70
static void free_fdtable_work(struct work_struct *work)
71
{
72 73
	struct fdtable_defer *f =
		container_of(work, struct fdtable_defer, wq);
74
	struct fdtable *fdt;
L
Linus Torvalds 已提交
75

76 77 78 79 80 81
	spin_lock_bh(&f->lock);
	fdt = f->next;
	f->next = NULL;
	spin_unlock_bh(&f->lock);
	while(fdt) {
		struct fdtable *next = fdt->next;
82 83

		__free_fdtable(fdt);
84 85 86
		fdt = next;
	}
}
L
Linus Torvalds 已提交
87

88
static void free_fdtable_rcu(struct rcu_head *rcu)
89 90 91
{
	struct fdtable *fdt = container_of(rcu, struct fdtable, rcu);
	struct fdtable_defer *fddef;
L
Linus Torvalds 已提交
92

93
	BUG_ON(!fdt);
A
Al Viro 已提交
94
	BUG_ON(fdt->max_fds <= NR_OPEN_DEFAULT);
95

96
	if (!is_vmalloc_addr(fdt->fd) && !is_vmalloc_addr(fdt->open_fds)) {
97
		kfree(fdt->fd);
98
		kfree(fdt->open_fds);
99
		kfree(fdt);
L
Linus Torvalds 已提交
100
	} else {
101 102 103 104
		fddef = &get_cpu_var(fdtable_defer_list);
		spin_lock(&fddef->lock);
		fdt->next = fddef->next;
		fddef->next = fdt;
105 106
		/* vmallocs are handled from the workqueue context */
		schedule_work(&fddef->wq);
107 108
		spin_unlock(&fddef->lock);
		put_cpu_var(fdtable_defer_list);
L
Linus Torvalds 已提交
109
	}
110 111 112 113 114 115
}

/*
 * Expand the fdset in the files_struct.  Called with the files spinlock
 * held for write.
 */
116
static void copy_fdtable(struct fdtable *nfdt, struct fdtable *ofdt)
117
{
118
	unsigned int cpy, set;
119

120 121 122 123 124 125 126 127 128 129 130 131 132
	BUG_ON(nfdt->max_fds < ofdt->max_fds);

	cpy = ofdt->max_fds * sizeof(struct file *);
	set = (nfdt->max_fds - ofdt->max_fds) * sizeof(struct file *);
	memcpy(nfdt->fd, ofdt->fd, cpy);
	memset((char *)(nfdt->fd) + cpy, 0, set);

	cpy = ofdt->max_fds / BITS_PER_BYTE;
	set = (nfdt->max_fds - ofdt->max_fds) / BITS_PER_BYTE;
	memcpy(nfdt->open_fds, ofdt->open_fds, cpy);
	memset((char *)(nfdt->open_fds) + cpy, 0, set);
	memcpy(nfdt->close_on_exec, ofdt->close_on_exec, cpy);
	memset((char *)(nfdt->close_on_exec) + cpy, 0, set);
L
Linus Torvalds 已提交
133 134
}

135
static struct fdtable * alloc_fdtable(unsigned int nr)
L
Linus Torvalds 已提交
136
{
137
	struct fdtable *fdt;
138
	void *data;
L
Linus Torvalds 已提交
139

140
	/*
141 142 143 144 145
	 * Figure out how many fds we actually want to support in this fdtable.
	 * Allocation steps are keyed to the size of the fdarray, since it
	 * grows far faster than any of the other dynamic data. We try to fit
	 * the fdarray into comfortable page-tuned chunks: starting at 1024B
	 * and growing in powers of two from there on.
146
	 */
147 148 149
	nr /= (1024 / sizeof(struct file *));
	nr = roundup_pow_of_two(nr + 1);
	nr *= (1024 / sizeof(struct file *));
A
Al Viro 已提交
150 151 152 153 154 155 156 157 158 159
	/*
	 * Note that this can drive nr *below* what we had passed if sysctl_nr_open
	 * had been set lower between the check in expand_files() and here.  Deal
	 * with that in caller, it's cheaper that way.
	 *
	 * We make sure that nr remains a multiple of BITS_PER_LONG - otherwise
	 * bitmaps handling below becomes unpleasant, to put it mildly...
	 */
	if (unlikely(nr > sysctl_nr_open))
		nr = ((sysctl_nr_open - 1) | (BITS_PER_LONG - 1)) + 1;
160

161 162
	fdt = kmalloc(sizeof(struct fdtable), GFP_KERNEL);
	if (!fdt)
163
		goto out;
164 165 166 167
	fdt->max_fds = nr;
	data = alloc_fdmem(nr * sizeof(struct file *));
	if (!data)
		goto out_fdt;
168 169 170
	fdt->fd = data;

	data = alloc_fdmem(max_t(size_t,
171 172 173
				 2 * nr / BITS_PER_BYTE, L1_CACHE_BYTES));
	if (!data)
		goto out_arr;
174
	fdt->open_fds = data;
175
	data += nr / BITS_PER_BYTE;
176
	fdt->close_on_exec = data;
177 178
	fdt->next = NULL;

179
	return fdt;
180 181

out_arr:
182
	free_fdmem(fdt->fd);
183
out_fdt:
184
	kfree(fdt);
185
out:
186 187
	return NULL;
}
L
Linus Torvalds 已提交
188

189
/*
190 191 192 193 194
 * Expand the file descriptor table.
 * This function will allocate a new fdtable and both fd array and fdset, of
 * the given size.
 * Return <0 error code on error; 1 on successful completion.
 * The files->file_lock should be held on entry, and will be held on exit.
195 196 197 198 199
 */
static int expand_fdtable(struct files_struct *files, int nr)
	__releases(files->file_lock)
	__acquires(files->file_lock)
{
200
	struct fdtable *new_fdt, *cur_fdt;
201 202

	spin_unlock(&files->file_lock);
203
	new_fdt = alloc_fdtable(nr);
204
	spin_lock(&files->file_lock);
205 206
	if (!new_fdt)
		return -ENOMEM;
A
Al Viro 已提交
207 208 209 210 211
	/*
	 * extremely unlikely race - sysctl_nr_open decreased between the check in
	 * caller and alloc_fdtable().  Cheaper to catch it here...
	 */
	if (unlikely(new_fdt->max_fds <= nr)) {
212
		__free_fdtable(new_fdt);
A
Al Viro 已提交
213 214
		return -EMFILE;
	}
215
	/*
216 217
	 * Check again since another task may have expanded the fd table while
	 * we dropped the lock
218
	 */
219
	cur_fdt = files_fdtable(files);
220
	if (nr >= cur_fdt->max_fds) {
221 222 223
		/* Continue as planned */
		copy_fdtable(new_fdt, cur_fdt);
		rcu_assign_pointer(files->fdt, new_fdt);
224
		if (cur_fdt->max_fds > NR_OPEN_DEFAULT)
A
Al Viro 已提交
225
			call_rcu(&cur_fdt->rcu, free_fdtable_rcu);
226
	} else {
227
		/* Somebody else expanded, so undo our attempt */
228
		__free_fdtable(new_fdt);
229
	}
230
	return 1;
L
Linus Torvalds 已提交
231 232 233 234
}

/*
 * Expand files.
235 236 237 238 239
 * This function will expand the file structures, if the requested size exceeds
 * the current capacity and there is room for expansion.
 * Return <0 error code on error; 0 when nothing done; 1 when files were
 * expanded and execution may have blocked.
 * The files->file_lock should be held on entry, and will be held on exit.
L
Linus Torvalds 已提交
240
 */
241
static int expand_files(struct files_struct *files, int nr)
L
Linus Torvalds 已提交
242
{
243
	struct fdtable *fdt;
L
Linus Torvalds 已提交
244

245
	fdt = files_fdtable(files);
A
Al Viro 已提交
246

247
	/* Do we need to expand? */
248
	if (nr < fdt->max_fds)
249
		return 0;
A
Al Viro 已提交
250

251
	/* Can we expand? */
252
	if (nr >= sysctl_nr_open)
253 254 255 256
		return -EMFILE;

	/* All good, so we try */
	return expand_fdtable(files, nr);
L
Linus Torvalds 已提交
257
}
258

259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
static inline void __set_close_on_exec(int fd, struct fdtable *fdt)
{
	__set_bit(fd, fdt->close_on_exec);
}

static inline void __clear_close_on_exec(int fd, struct fdtable *fdt)
{
	__clear_bit(fd, fdt->close_on_exec);
}

static inline void __set_open_fd(int fd, struct fdtable *fdt)
{
	__set_bit(fd, fdt->open_fds);
}

static inline void __clear_open_fd(int fd, struct fdtable *fdt)
{
	__clear_bit(fd, fdt->open_fds);
}

A
Al Viro 已提交
279 280 281 282 283 284
static int count_open_files(struct fdtable *fdt)
{
	int size = fdt->max_fds;
	int i;

	/* Find the last open fd */
285 286
	for (i = size / BITS_PER_LONG; i > 0; ) {
		if (fdt->open_fds[--i])
A
Al Viro 已提交
287 288
			break;
	}
289
	i = (i + 1) * BITS_PER_LONG;
A
Al Viro 已提交
290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305
	return i;
}

/*
 * Allocate a new files structure and copy contents from the
 * passed in files structure.
 * errorp will be valid only when the returned files_struct is NULL.
 */
struct files_struct *dup_fd(struct files_struct *oldf, int *errorp)
{
	struct files_struct *newf;
	struct file **old_fds, **new_fds;
	int open_files, size, i;
	struct fdtable *old_fdt, *new_fdt;

	*errorp = -ENOMEM;
A
Al Viro 已提交
306
	newf = kmem_cache_alloc(files_cachep, GFP_KERNEL);
A
Al Viro 已提交
307 308 309
	if (!newf)
		goto out;

A
Al Viro 已提交
310 311 312 313 314 315
	atomic_set(&newf->count, 1);

	spin_lock_init(&newf->file_lock);
	newf->next_fd = 0;
	new_fdt = &newf->fdtab;
	new_fdt->max_fds = NR_OPEN_DEFAULT;
316 317
	new_fdt->close_on_exec = newf->close_on_exec_init;
	new_fdt->open_fds = newf->open_fds_init;
A
Al Viro 已提交
318 319 320
	new_fdt->fd = &newf->fd_array[0];
	new_fdt->next = NULL;

A
Al Viro 已提交
321 322 323 324 325 326 327
	spin_lock(&oldf->file_lock);
	old_fdt = files_fdtable(oldf);
	open_files = count_open_files(old_fdt);

	/*
	 * Check whether we need to allocate a larger fd array and fd set.
	 */
A
Al Viro 已提交
328
	while (unlikely(open_files > new_fdt->max_fds)) {
A
Al Viro 已提交
329
		spin_unlock(&oldf->file_lock);
A
Al Viro 已提交
330

331 332
		if (new_fdt != &newf->fdtab)
			__free_fdtable(new_fdt);
A
Al Viro 已提交
333

A
Al Viro 已提交
334 335 336 337 338 339 340 341
		new_fdt = alloc_fdtable(open_files - 1);
		if (!new_fdt) {
			*errorp = -ENOMEM;
			goto out_release;
		}

		/* beyond sysctl_nr_open; nothing to do */
		if (unlikely(new_fdt->max_fds < open_files)) {
342
			__free_fdtable(new_fdt);
A
Al Viro 已提交
343
			*errorp = -EMFILE;
A
Al Viro 已提交
344
			goto out_release;
A
Al Viro 已提交
345 346
		}

A
Al Viro 已提交
347 348 349 350 351 352 353
		/*
		 * Reacquire the oldf lock and a pointer to its fd table
		 * who knows it may have a new bigger fd table. We need
		 * the latest pointer.
		 */
		spin_lock(&oldf->file_lock);
		old_fdt = files_fdtable(oldf);
A
Al Viro 已提交
354
		open_files = count_open_files(old_fdt);
A
Al Viro 已提交
355 356 357 358 359
	}

	old_fds = old_fdt->fd;
	new_fds = new_fdt->fd;

360 361
	memcpy(new_fdt->open_fds, old_fdt->open_fds, open_files / 8);
	memcpy(new_fdt->close_on_exec, old_fdt->close_on_exec, open_files / 8);
A
Al Viro 已提交
362 363 364 365 366 367 368 369 370 371 372 373

	for (i = open_files; i != 0; i--) {
		struct file *f = *old_fds++;
		if (f) {
			get_file(f);
		} else {
			/*
			 * The fd may be claimed in the fd bitmap but not yet
			 * instantiated in the files array if a sibling thread
			 * is partway through open().  So make sure that this
			 * fd is available to the new process.
			 */
374
			__clear_open_fd(open_files - i, new_fdt);
A
Al Viro 已提交
375 376 377 378 379 380 381 382 383 384 385 386
		}
		rcu_assign_pointer(*new_fds++, f);
	}
	spin_unlock(&oldf->file_lock);

	/* compute the remainder to be cleared */
	size = (new_fdt->max_fds - open_files) * sizeof(struct file *);

	/* This is long word aligned thus could use a optimized version */
	memset(new_fds, 0, size);

	if (new_fdt->max_fds > open_files) {
387 388
		int left = (new_fdt->max_fds - open_files) / 8;
		int start = open_files / BITS_PER_LONG;
A
Al Viro 已提交
389

390 391
		memset(&new_fdt->open_fds[start], 0, left);
		memset(&new_fdt->close_on_exec[start], 0, left);
A
Al Viro 已提交
392 393
	}

A
Al Viro 已提交
394 395
	rcu_assign_pointer(newf->fdt, new_fdt);

A
Al Viro 已提交
396 397 398 399 400 401 402 403
	return newf;

out_release:
	kmem_cache_free(files_cachep, newf);
out:
	return NULL;
}

404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457
static void close_files(struct files_struct * files)
{
	int i, j;
	struct fdtable *fdt;

	j = 0;

	/*
	 * It is safe to dereference the fd table without RCU or
	 * ->file_lock because this is the last reference to the
	 * files structure.  But use RCU to shut RCU-lockdep up.
	 */
	rcu_read_lock();
	fdt = files_fdtable(files);
	rcu_read_unlock();
	for (;;) {
		unsigned long set;
		i = j * BITS_PER_LONG;
		if (i >= fdt->max_fds)
			break;
		set = fdt->open_fds[j++];
		while (set) {
			if (set & 1) {
				struct file * file = xchg(&fdt->fd[i], NULL);
				if (file) {
					filp_close(file, files);
					cond_resched();
				}
			}
			i++;
			set >>= 1;
		}
	}
}

struct files_struct *get_files_struct(struct task_struct *task)
{
	struct files_struct *files;

	task_lock(task);
	files = task->files;
	if (files)
		atomic_inc(&files->count);
	task_unlock(task);

	return files;
}

void put_files_struct(struct files_struct *files)
{
	struct fdtable *fdt;

	if (atomic_dec_and_test(&files->count)) {
		close_files(files);
458
		/* not really needed, since nobody can see us */
459 460 461
		rcu_read_lock();
		fdt = files_fdtable(files);
		rcu_read_unlock();
462 463 464 465
		/* free the arrays if they are not embedded */
		if (fdt != &files->fdtab)
			__free_fdtable(fdt);
		kmem_cache_free(files_cachep, files);
466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492
	}
}

void reset_files_struct(struct files_struct *files)
{
	struct task_struct *tsk = current;
	struct files_struct *old;

	old = tsk->files;
	task_lock(tsk);
	tsk->files = files;
	task_unlock(tsk);
	put_files_struct(old);
}

void exit_files(struct task_struct *tsk)
{
	struct files_struct * files = tsk->files;

	if (files) {
		task_lock(tsk);
		tsk->files = NULL;
		task_unlock(tsk);
		put_files_struct(files);
	}
}

493 494 495 496
static void __devinit fdtable_defer_list_init(int cpu)
{
	struct fdtable_defer *fddef = &per_cpu(fdtable_defer_list, cpu);
	spin_lock_init(&fddef->lock);
497
	INIT_WORK(&fddef->wq, free_fdtable_work);
498 499 500 501 502 503
	fddef->next = NULL;
}

void __init files_defer_init(void)
{
	int i;
504
	for_each_possible_cpu(i)
505
		fdtable_defer_list_init(i);
506 507
	sysctl_nr_open_max = min((size_t)INT_MAX, ~(size_t)0/sizeof(void *)) &
			     -BITS_PER_LONG;
508
}
A
Al Viro 已提交
509 510 511 512 513 514 515

struct files_struct init_files = {
	.count		= ATOMIC_INIT(1),
	.fdt		= &init_files.fdtab,
	.fdtab		= {
		.max_fds	= NR_OPEN_DEFAULT,
		.fd		= &init_files.fd_array[0],
516 517
		.close_on_exec	= init_files.close_on_exec_init,
		.open_fds	= init_files.open_fds_init,
A
Al Viro 已提交
518 519 520
	},
	.file_lock	= __SPIN_LOCK_UNLOCKED(init_task.file_lock),
};
521 522 523 524

/*
 * allocate a file descriptor, mark it busy.
 */
A
Al Viro 已提交
525 526
int __alloc_fd(struct files_struct *files,
	       unsigned start, unsigned end, unsigned flags)
527 528 529 530 531 532 533 534 535 536 537 538 539
{
	unsigned int fd;
	int error;
	struct fdtable *fdt;

	spin_lock(&files->file_lock);
repeat:
	fdt = files_fdtable(files);
	fd = start;
	if (fd < files->next_fd)
		fd = files->next_fd;

	if (fd < fdt->max_fds)
540
		fd = find_next_zero_bit(fdt->open_fds, fdt->max_fds, fd);
541

542 543 544 545 546 547 548 549
	/*
	 * N.B. For clone tasks sharing a files structure, this test
	 * will limit the total number of files that can be opened.
	 */
	error = -EMFILE;
	if (fd >= end)
		goto out;

550 551 552 553 554 555 556 557 558 559 560 561 562 563
	error = expand_files(files, fd);
	if (error < 0)
		goto out;

	/*
	 * If we needed to expand the fs array we
	 * might have blocked - try again.
	 */
	if (error)
		goto repeat;

	if (start <= files->next_fd)
		files->next_fd = fd + 1;

564
	__set_open_fd(fd, fdt);
565
	if (flags & O_CLOEXEC)
566
		__set_close_on_exec(fd, fdt);
567
	else
568
		__clear_close_on_exec(fd, fdt);
569 570 571
	error = fd;
#if 1
	/* Sanity check */
572
	if (rcu_dereference_raw(fdt->fd[fd]) != NULL) {
573 574 575 576 577 578 579 580 581 582
		printk(KERN_WARNING "alloc_fd: slot %d not NULL!\n", fd);
		rcu_assign_pointer(fdt->fd[fd], NULL);
	}
#endif

out:
	spin_unlock(&files->file_lock);
	return error;
}

583
static int alloc_fd(unsigned start, unsigned flags)
A
Al Viro 已提交
584 585 586 587
{
	return __alloc_fd(current->files, start, rlimit(RLIMIT_NOFILE), flags);
}

A
Al Viro 已提交
588
int get_unused_fd_flags(unsigned flags)
589
{
A
Al Viro 已提交
590
	return __alloc_fd(current->files, 0, rlimit(RLIMIT_NOFILE), flags);
591
}
A
Al Viro 已提交
592
EXPORT_SYMBOL(get_unused_fd_flags);
593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622

static void __put_unused_fd(struct files_struct *files, unsigned int fd)
{
	struct fdtable *fdt = files_fdtable(files);
	__clear_open_fd(fd, fdt);
	if (fd < files->next_fd)
		files->next_fd = fd;
}

void put_unused_fd(unsigned int fd)
{
	struct files_struct *files = current->files;
	spin_lock(&files->file_lock);
	__put_unused_fd(files, fd);
	spin_unlock(&files->file_lock);
}

EXPORT_SYMBOL(put_unused_fd);

/*
 * Install a file pointer in the fd array.
 *
 * The VFS is full of places where we drop the files lock between
 * setting the open_fds bitmap and installing the file in the file
 * array.  At any such point, we are vulnerable to a dup2() race
 * installing a file in the array before us.  We need to detect this and
 * fput() the struct file we are about to overwrite in this case.
 *
 * It should never happen - if we allow dup2() do it, _really_ bad things
 * will follow.
623 624 625 626 627 628 629
 *
 * NOTE: __fd_install() variant is really, really low-level; don't
 * use it unless you are forced to by truly lousy API shoved down
 * your throat.  'files' *MUST* be either current->files or obtained
 * by get_files_struct(current) done by whoever had given it to you,
 * or really bad things will happen.  Normally you want to use
 * fd_install() instead.
630 631
 */

632 633
void __fd_install(struct files_struct *files, unsigned int fd,
		struct file *file)
634 635 636 637 638 639 640 641 642
{
	struct fdtable *fdt;
	spin_lock(&files->file_lock);
	fdt = files_fdtable(files);
	BUG_ON(fdt->fd[fd] != NULL);
	rcu_assign_pointer(fdt->fd[fd], file);
	spin_unlock(&files->file_lock);
}

643 644 645 646 647
void fd_install(unsigned int fd, struct file *file)
{
	__fd_install(current->files, fd, file);
}

648
EXPORT_SYMBOL(fd_install);
A
Al Viro 已提交
649

650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675
/*
 * The same warnings as for __alloc_fd()/__fd_install() apply here...
 */
int __close_fd(struct files_struct *files, unsigned fd)
{
	struct file *file;
	struct fdtable *fdt;

	spin_lock(&files->file_lock);
	fdt = files_fdtable(files);
	if (fd >= fdt->max_fds)
		goto out_unlock;
	file = fdt->fd[fd];
	if (!file)
		goto out_unlock;
	rcu_assign_pointer(fdt->fd[fd], NULL);
	__clear_close_on_exec(fd, fdt);
	__put_unused_fd(files, fd);
	spin_unlock(&files->file_lock);
	return filp_close(file, files);

out_unlock:
	spin_unlock(&files->file_lock);
	return -EBADF;
}

676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712
void do_close_on_exec(struct files_struct *files)
{
	unsigned i;
	struct fdtable *fdt;

	/* exec unshares first */
	BUG_ON(atomic_read(&files->count) != 1);
	spin_lock(&files->file_lock);
	for (i = 0; ; i++) {
		unsigned long set;
		unsigned fd = i * BITS_PER_LONG;
		fdt = files_fdtable(files);
		if (fd >= fdt->max_fds)
			break;
		set = fdt->close_on_exec[i];
		if (!set)
			continue;
		fdt->close_on_exec[i] = 0;
		for ( ; set ; fd++, set >>= 1) {
			struct file *file;
			if (!(set & 1))
				continue;
			file = fdt->fd[fd];
			if (!file)
				continue;
			rcu_assign_pointer(fdt->fd[fd], NULL);
			__put_unused_fd(files, fd);
			spin_unlock(&files->file_lock);
			filp_close(file, files);
			cond_resched();
			spin_lock(&files->file_lock);
		}

	}
	spin_unlock(&files->file_lock);
}

A
Al Viro 已提交
713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817
struct file *fget(unsigned int fd)
{
	struct file *file;
	struct files_struct *files = current->files;

	rcu_read_lock();
	file = fcheck_files(files, fd);
	if (file) {
		/* File object ref couldn't be taken */
		if (file->f_mode & FMODE_PATH ||
		    !atomic_long_inc_not_zero(&file->f_count))
			file = NULL;
	}
	rcu_read_unlock();

	return file;
}

EXPORT_SYMBOL(fget);

struct file *fget_raw(unsigned int fd)
{
	struct file *file;
	struct files_struct *files = current->files;

	rcu_read_lock();
	file = fcheck_files(files, fd);
	if (file) {
		/* File object ref couldn't be taken */
		if (!atomic_long_inc_not_zero(&file->f_count))
			file = NULL;
	}
	rcu_read_unlock();

	return file;
}

EXPORT_SYMBOL(fget_raw);

/*
 * Lightweight file lookup - no refcnt increment if fd table isn't shared.
 *
 * You can use this instead of fget if you satisfy all of the following
 * conditions:
 * 1) You must call fput_light before exiting the syscall and returning control
 *    to userspace (i.e. you cannot remember the returned struct file * after
 *    returning to userspace).
 * 2) You must not call filp_close on the returned struct file * in between
 *    calls to fget_light and fput_light.
 * 3) You must not clone the current task in between the calls to fget_light
 *    and fput_light.
 *
 * The fput_needed flag returned by fget_light should be passed to the
 * corresponding fput_light.
 */
struct file *fget_light(unsigned int fd, int *fput_needed)
{
	struct file *file;
	struct files_struct *files = current->files;

	*fput_needed = 0;
	if (atomic_read(&files->count) == 1) {
		file = fcheck_files(files, fd);
		if (file && (file->f_mode & FMODE_PATH))
			file = NULL;
	} else {
		rcu_read_lock();
		file = fcheck_files(files, fd);
		if (file) {
			if (!(file->f_mode & FMODE_PATH) &&
			    atomic_long_inc_not_zero(&file->f_count))
				*fput_needed = 1;
			else
				/* Didn't get the reference, someone's freed */
				file = NULL;
		}
		rcu_read_unlock();
	}

	return file;
}

struct file *fget_raw_light(unsigned int fd, int *fput_needed)
{
	struct file *file;
	struct files_struct *files = current->files;

	*fput_needed = 0;
	if (atomic_read(&files->count) == 1) {
		file = fcheck_files(files, fd);
	} else {
		rcu_read_lock();
		file = fcheck_files(files, fd);
		if (file) {
			if (atomic_long_inc_not_zero(&file->f_count))
				*fput_needed = 1;
			else
				/* Didn't get the reference, someone's freed */
				file = NULL;
		}
		rcu_read_unlock();
	}

	return file;
}
818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843

void set_close_on_exec(unsigned int fd, int flag)
{
	struct files_struct *files = current->files;
	struct fdtable *fdt;
	spin_lock(&files->file_lock);
	fdt = files_fdtable(files);
	if (flag)
		__set_close_on_exec(fd, fdt);
	else
		__clear_close_on_exec(fd, fdt);
	spin_unlock(&files->file_lock);
}

bool get_close_on_exec(unsigned int fd)
{
	struct files_struct *files = current->files;
	struct fdtable *fdt;
	bool res;
	rcu_read_lock();
	fdt = files_fdtable(files);
	res = close_on_exec(fd, fdt);
	rcu_read_unlock();
	return res;
}

A
Al Viro 已提交
844 845
static int do_dup2(struct files_struct *files,
	struct file *file, unsigned fd, unsigned flags)
846
{
A
Al Viro 已提交
847
	struct file *tofree;
848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864
	struct fdtable *fdt;

	/*
	 * We need to detect attempts to do dup2() over allocated but still
	 * not finished descriptor.  NB: OpenBSD avoids that at the price of
	 * extra work in their equivalent of fget() - they insert struct
	 * file immediately after grabbing descriptor, mark it larval if
	 * more work (e.g. actual opening) is needed and make sure that
	 * fget() treats larval files as absent.  Potentially interesting,
	 * but while extra work in fget() is trivial, locking implications
	 * and amount of surgery on open()-related paths in VFS are not.
	 * FreeBSD fails with -EBADF in the same situation, NetBSD "solution"
	 * deadlocks in rather amusing ways, AFAICS.  All of that is out of
	 * scope of POSIX or SUS, since neither considers shared descriptor
	 * tables and this condition does not arise without those.
	 */
	fdt = files_fdtable(files);
A
Al Viro 已提交
865 866 867
	tofree = fdt->fd[fd];
	if (!tofree && fd_is_open(fd, fdt))
		goto Ebusy;
868
	get_file(file);
A
Al Viro 已提交
869 870
	rcu_assign_pointer(fdt->fd[fd], file);
	__set_open_fd(fd, fdt);
871
	if (flags & O_CLOEXEC)
A
Al Viro 已提交
872
		__set_close_on_exec(fd, fdt);
873
	else
A
Al Viro 已提交
874
		__clear_close_on_exec(fd, fdt);
875 876 877 878 879
	spin_unlock(&files->file_lock);

	if (tofree)
		filp_close(tofree, files);

A
Al Viro 已提交
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931
	return fd;

Ebusy:
	spin_unlock(&files->file_lock);
	return -EBUSY;
}

int replace_fd(unsigned fd, struct file *file, unsigned flags)
{
	int err;
	struct files_struct *files = current->files;

	if (!file)
		return __close_fd(files, fd);

	if (fd >= rlimit(RLIMIT_NOFILE))
		return -EMFILE;

	spin_lock(&files->file_lock);
	err = expand_files(files, fd);
	if (unlikely(err < 0))
		goto out_unlock;
	return do_dup2(files, file, fd, flags);

out_unlock:
	spin_unlock(&files->file_lock);
	return err;
}

SYSCALL_DEFINE3(dup3, unsigned int, oldfd, unsigned int, newfd, int, flags)
{
	int err = -EBADF;
	struct file *file;
	struct files_struct *files = current->files;

	if ((flags & ~O_CLOEXEC) != 0)
		return -EINVAL;

	if (newfd >= rlimit(RLIMIT_NOFILE))
		return -EMFILE;

	spin_lock(&files->file_lock);
	err = expand_files(files, newfd);
	file = fcheck(oldfd);
	if (unlikely(!file))
		goto Ebadf;
	if (unlikely(err < 0)) {
		if (err == -EMFILE)
			goto Ebadf;
		goto out_unlock;
	}
	return do_dup2(files, file, newfd, flags);
932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981

Ebadf:
	err = -EBADF;
out_unlock:
	spin_unlock(&files->file_lock);
	return err;
}

SYSCALL_DEFINE2(dup2, unsigned int, oldfd, unsigned int, newfd)
{
	if (unlikely(newfd == oldfd)) { /* corner case */
		struct files_struct *files = current->files;
		int retval = oldfd;

		rcu_read_lock();
		if (!fcheck_files(files, oldfd))
			retval = -EBADF;
		rcu_read_unlock();
		return retval;
	}
	return sys_dup3(oldfd, newfd, 0);
}

SYSCALL_DEFINE1(dup, unsigned int, fildes)
{
	int ret = -EBADF;
	struct file *file = fget_raw(fildes);

	if (file) {
		ret = get_unused_fd();
		if (ret >= 0)
			fd_install(ret, file);
		else
			fput(file);
	}
	return ret;
}

int f_dupfd(unsigned int from, struct file *file, unsigned flags)
{
	int err;
	if (from >= rlimit(RLIMIT_NOFILE))
		return -EINVAL;
	err = alloc_fd(from, flags);
	if (err >= 0) {
		get_file(file);
		fd_install(err, file);
	}
	return err;
}