file.c 22.5 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 242
 */
int expand_files(struct files_struct *files, int nr)
{
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

A
Al Viro 已提交
259 260 261 262 263 264
static int count_open_files(struct fdtable *fdt)
{
	int size = fdt->max_fds;
	int i;

	/* Find the last open fd */
265 266
	for (i = size / BITS_PER_LONG; i > 0; ) {
		if (fdt->open_fds[--i])
A
Al Viro 已提交
267 268
			break;
	}
269
	i = (i + 1) * BITS_PER_LONG;
A
Al Viro 已提交
270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
	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 已提交
286
	newf = kmem_cache_alloc(files_cachep, GFP_KERNEL);
A
Al Viro 已提交
287 288 289
	if (!newf)
		goto out;

A
Al Viro 已提交
290 291 292 293 294 295
	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;
296 297
	new_fdt->close_on_exec = newf->close_on_exec_init;
	new_fdt->open_fds = newf->open_fds_init;
A
Al Viro 已提交
298 299 300
	new_fdt->fd = &newf->fd_array[0];
	new_fdt->next = NULL;

A
Al Viro 已提交
301 302 303 304 305 306 307
	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 已提交
308
	while (unlikely(open_files > new_fdt->max_fds)) {
A
Al Viro 已提交
309
		spin_unlock(&oldf->file_lock);
A
Al Viro 已提交
310

311 312
		if (new_fdt != &newf->fdtab)
			__free_fdtable(new_fdt);
A
Al Viro 已提交
313

A
Al Viro 已提交
314 315 316 317 318 319 320 321
		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)) {
322
			__free_fdtable(new_fdt);
A
Al Viro 已提交
323
			*errorp = -EMFILE;
A
Al Viro 已提交
324
			goto out_release;
A
Al Viro 已提交
325 326
		}

A
Al Viro 已提交
327 328 329 330 331 332 333
		/*
		 * 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 已提交
334
		open_files = count_open_files(old_fdt);
A
Al Viro 已提交
335 336 337 338 339
	}

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

340 341
	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 已提交
342 343 344 345 346 347 348 349 350 351 352 353

	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.
			 */
354
			__clear_open_fd(open_files - i, new_fdt);
A
Al Viro 已提交
355 356 357 358 359 360 361 362 363 364 365 366
		}
		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) {
367 368
		int left = (new_fdt->max_fds - open_files) / 8;
		int start = open_files / BITS_PER_LONG;
A
Al Viro 已提交
369

370 371
		memset(&new_fdt->open_fds[start], 0, left);
		memset(&new_fdt->close_on_exec[start], 0, left);
A
Al Viro 已提交
372 373
	}

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

A
Al Viro 已提交
376 377 378 379 380 381 382 383
	return newf;

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

384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 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
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);
438
		/* not really needed, since nobody can see us */
439 440 441
		rcu_read_lock();
		fdt = files_fdtable(files);
		rcu_read_unlock();
442 443 444 445
		/* free the arrays if they are not embedded */
		if (fdt != &files->fdtab)
			__free_fdtable(fdt);
		kmem_cache_free(files_cachep, files);
446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
	}
}

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);
	}
}

473 474 475 476
static void __devinit fdtable_defer_list_init(int cpu)
{
	struct fdtable_defer *fddef = &per_cpu(fdtable_defer_list, cpu);
	spin_lock_init(&fddef->lock);
477
	INIT_WORK(&fddef->wq, free_fdtable_work);
478 479 480 481 482 483
	fddef->next = NULL;
}

void __init files_defer_init(void)
{
	int i;
484
	for_each_possible_cpu(i)
485
		fdtable_defer_list_init(i);
486 487
	sysctl_nr_open_max = min((size_t)INT_MAX, ~(size_t)0/sizeof(void *)) &
			     -BITS_PER_LONG;
488
}
A
Al Viro 已提交
489 490 491 492 493 494 495

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],
496 497
		.close_on_exec	= init_files.close_on_exec_init,
		.open_fds	= init_files.open_fds_init,
A
Al Viro 已提交
498 499 500
	},
	.file_lock	= __SPIN_LOCK_UNLOCKED(init_task.file_lock),
};
501 502 503 504

/*
 * allocate a file descriptor, mark it busy.
 */
A
Al Viro 已提交
505 506
int __alloc_fd(struct files_struct *files,
	       unsigned start, unsigned end, unsigned flags)
507 508 509 510 511 512 513 514 515 516 517 518 519
{
	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)
520
		fd = find_next_zero_bit(fdt->open_fds, fdt->max_fds, fd);
521

522 523 524 525 526 527 528 529
	/*
	 * 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;

530 531 532 533 534 535 536 537 538 539 540 541 542 543
	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;

544
	__set_open_fd(fd, fdt);
545
	if (flags & O_CLOEXEC)
546
		__set_close_on_exec(fd, fdt);
547
	else
548
		__clear_close_on_exec(fd, fdt);
549 550 551
	error = fd;
#if 1
	/* Sanity check */
552
	if (rcu_dereference_raw(fdt->fd[fd]) != NULL) {
553 554 555 556 557 558 559 560 561 562
		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;
}

A
Al Viro 已提交
563 564 565 566 567
int alloc_fd(unsigned start, unsigned flags)
{
	return __alloc_fd(current->files, start, rlimit(RLIMIT_NOFILE), flags);
}

A
Al Viro 已提交
568
int get_unused_fd_flags(unsigned flags)
569
{
A
Al Viro 已提交
570
	return __alloc_fd(current->files, 0, rlimit(RLIMIT_NOFILE), flags);
571
}
A
Al Viro 已提交
572
EXPORT_SYMBOL(get_unused_fd_flags);
573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602

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.
603 604 605 606 607 608 609
 *
 * 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.
610 611
 */

612 613
void __fd_install(struct files_struct *files, unsigned int fd,
		struct file *file)
614 615 616 617 618 619 620 621 622
{
	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);
}

623 624 625 626 627
void fd_install(unsigned int fd, struct file *file)
{
	__fd_install(current->files, fd, file);
}

628
EXPORT_SYMBOL(fd_install);
A
Al Viro 已提交
629

630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655
/*
 * 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;
}

656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692
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 已提交
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 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
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;
}
798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823

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 已提交
824 825
static int do_dup2(struct files_struct *files,
	struct file *file, unsigned fd, unsigned flags)
826
{
A
Al Viro 已提交
827
	struct file *tofree;
828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844
	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 已提交
845 846 847
	tofree = fdt->fd[fd];
	if (!tofree && fd_is_open(fd, fdt))
		goto Ebusy;
848
	get_file(file);
A
Al Viro 已提交
849 850
	rcu_assign_pointer(fdt->fd[fd], file);
	__set_open_fd(fd, fdt);
851
	if (flags & O_CLOEXEC)
A
Al Viro 已提交
852
		__set_close_on_exec(fd, fdt);
853
	else
A
Al Viro 已提交
854
		__clear_close_on_exec(fd, fdt);
855 856 857 858 859
	spin_unlock(&files->file_lock);

	if (tofree)
		filp_close(tofree, files);

A
Al Viro 已提交
860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 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
	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);
912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 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

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;
}