shm.c 31.9 KB
Newer Older
Linus Torvalds's avatar
Linus Torvalds committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * linux/ipc/shm.c
 * Copyright (C) 1992, 1993 Krishna Balasubramanian
 *	 Many improvements/fixes by Bruno Haible.
 * Replaced `struct shm_desc' by `struct vm_area_struct', July 1994.
 * Fixed the shm swap deallocation (shm_unuse()), August 1998 Andrea Arcangeli.
 *
 * /proc/sysvipc/shm support (c) 1999 Dragos Acostachioaie <dragos@iname.com>
 * BIGMEM support, Andrea Arcangeli <andrea@suse.de>
 * SMP thread shm, Jean-Luc Boyard <jean-luc.boyard@siemens.fr>
 * HIGHMEM support, Ingo Molnar <mingo@redhat.com>
 * Make shmmax, shmall, shmmni sysctl'able, Christoph Rohland <cr@sap.com>
 * Shared /dev/zero support, Kanoj Sarcar <kanoj@sgi.com>
 * Move the mm functionality over to mm/shmem.c, Christoph Rohland <cr@sap.com>
 *
Steve Grubb's avatar
Steve Grubb committed
16 17
 * support for audit of ipc object properties and permission changes
 * Dustin Kirkland <dustin.kirkland@us.ibm.com>
Kirill Korotaev's avatar
Kirill Korotaev committed
18 19 20 21
 *
 * namespaces support
 * OpenVZ, SWsoft Inc.
 * Pavel Emelianov <xemul@openvz.org>
22 23 24
 *
 * Better ipc lock (kern_ipc_perm.lock) handling
 * Davidlohr Bueso <davidlohr.bueso@hp.com>, June 2013.
Linus Torvalds's avatar
Linus Torvalds committed
25 26 27 28 29 30 31 32 33 34 35 36 37
 */

#include <linux/slab.h>
#include <linux/mm.h>
#include <linux/hugetlb.h>
#include <linux/shm.h>
#include <linux/init.h>
#include <linux/file.h>
#include <linux/mman.h>
#include <linux/shmem_fs.h>
#include <linux/security.h>
#include <linux/syscalls.h>
#include <linux/audit.h>
38
#include <linux/capability.h>
39
#include <linux/ptrace.h>
40
#include <linux/seq_file.h>
Nadia Derbey's avatar
Nadia Derbey committed
41
#include <linux/rwsem.h>
Kirill Korotaev's avatar
Kirill Korotaev committed
42
#include <linux/nsproxy.h>
43
#include <linux/mount.h>
44
#include <linux/ipc_namespace.h>
45

Linus Torvalds's avatar
Linus Torvalds committed
46 47 48 49
#include <asm/uaccess.h>

#include "util.h"

50 51 52 53 54 55 56 57 58
struct shm_file_data {
	int id;
	struct ipc_namespace *ns;
	struct file *file;
	const struct vm_operations_struct *vm_ops;
};

#define shm_file_data(file) (*((struct shm_file_data **)&(file)->private_data))

59
static const struct file_operations shm_file_operations;
60
static const struct vm_operations_struct shm_vm_ops;
Linus Torvalds's avatar
Linus Torvalds committed
61

62
#define shm_ids(ns)	((ns)->ids[IPC_SHM_IDS])
Linus Torvalds's avatar
Linus Torvalds committed
63

Kirill Korotaev's avatar
Kirill Korotaev committed
64 65
#define shm_unlock(shp)			\
	ipc_unlock(&(shp)->shm_perm)
Linus Torvalds's avatar
Linus Torvalds committed
66

Nadia Derbey's avatar
Nadia Derbey committed
67
static int newseg(struct ipc_namespace *, struct ipc_params *);
68 69
static void shm_open(struct vm_area_struct *vma);
static void shm_close(struct vm_area_struct *vma);
Manfred Spraul's avatar
Manfred Spraul committed
70
static void shm_destroy(struct ipc_namespace *ns, struct shmid_kernel *shp);
Linus Torvalds's avatar
Linus Torvalds committed
71
#ifdef CONFIG_PROC_FS
72
static int sysvipc_shm_proc_show(struct seq_file *s, void *it);
Linus Torvalds's avatar
Linus Torvalds committed
73 74
#endif

75
void shm_init_ns(struct ipc_namespace *ns)
Kirill Korotaev's avatar
Kirill Korotaev committed
76 77 78 79
{
	ns->shm_ctlmax = SHMMAX;
	ns->shm_ctlall = SHMALL;
	ns->shm_ctlmni = SHMMNI;
80
	ns->shm_rmid_forced = 0;
Kirill Korotaev's avatar
Kirill Korotaev committed
81
	ns->shm_tot = 0;
WANG Cong's avatar
WANG Cong committed
82
	ipc_init_ids(&shm_ids(ns));
Kirill Korotaev's avatar
Kirill Korotaev committed
83 84
}

Nadia Derbey's avatar
Nadia Derbey committed
85
/*
Davidlohr Bueso's avatar
Davidlohr Bueso committed
86 87
 * Called with shm_ids.rwsem (writer) and the shp structure locked.
 * Only shm_ids.rwsem remains locked on exit.
Nadia Derbey's avatar
Nadia Derbey committed
88
 */
89
static void do_shm_rmid(struct ipc_namespace *ns, struct kern_ipc_perm *ipcp)
Kirill Korotaev's avatar
Kirill Korotaev committed
90
{
91 92 93
	struct shmid_kernel *shp;
	shp = container_of(ipcp, struct shmid_kernel, shm_perm);

Manfred Spraul's avatar
Manfred Spraul committed
94
	if (shp->shm_nattch) {
Kirill Korotaev's avatar
Kirill Korotaev committed
95 96 97 98 99 100 101 102
		shp->shm_perm.mode |= SHM_DEST;
		/* Do not find it any more */
		shp->shm_perm.key = IPC_PRIVATE;
		shm_unlock(shp);
	} else
		shm_destroy(ns, shp);
}

103
#ifdef CONFIG_IPC_NS
Kirill Korotaev's avatar
Kirill Korotaev committed
104 105
void shm_exit_ns(struct ipc_namespace *ns)
{
106
	free_ipcs(ns, &shm_ids(ns), do_shm_rmid);
107
	idr_destroy(&ns->ids[IPC_SHM_IDS].ipcs_idr);
Kirill Korotaev's avatar
Kirill Korotaev committed
108
}
109
#endif
Linus Torvalds's avatar
Linus Torvalds committed
110

111
static int __init ipc_ns_init(void)
Linus Torvalds's avatar
Linus Torvalds committed
112
{
113
	shm_init_ns(&init_ipc_ns);
114 115 116 117 118
	return 0;
}

pure_initcall(ipc_ns_init);

Manfred Spraul's avatar
Manfred Spraul committed
119
void __init shm_init(void)
120
{
121
	ipc_init_proc_interface("sysvipc/shm",
122 123 124 125 126
#if BITS_PER_LONG <= 32
				"       key      shmid perms       size  cpid  lpid nattch   uid   gid  cuid  cgid      atime      dtime      ctime        rss       swap\n",
#else
				"       key      shmid perms                  size  cpid  lpid nattch   uid   gid  cuid  cgid      atime      dtime      ctime                   rss                  swap\n",
#endif
Kirill Korotaev's avatar
Kirill Korotaev committed
127
				IPC_SHM_IDS, sysvipc_shm_proc_show);
Linus Torvalds's avatar
Linus Torvalds committed
128 129
}

130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149
static inline struct shmid_kernel *shm_obtain_object(struct ipc_namespace *ns, int id)
{
	struct kern_ipc_perm *ipcp = ipc_obtain_object(&shm_ids(ns), id);

	if (IS_ERR(ipcp))
		return ERR_CAST(ipcp);

	return container_of(ipcp, struct shmid_kernel, shm_perm);
}

static inline struct shmid_kernel *shm_obtain_object_check(struct ipc_namespace *ns, int id)
{
	struct kern_ipc_perm *ipcp = ipc_obtain_object_check(&shm_ids(ns), id);

	if (IS_ERR(ipcp))
		return ERR_CAST(ipcp);

	return container_of(ipcp, struct shmid_kernel, shm_perm);
}

Nadia Derbey's avatar
Nadia Derbey committed
150
/*
Davidlohr Bueso's avatar
Davidlohr Bueso committed
151
 * shm_lock_(check_) routines are called in the paths where the rwsem
Nadia Derbey's avatar
Nadia Derbey committed
152
 * is not necessarily held.
Nadia Derbey's avatar
Nadia Derbey committed
153
 */
154
static inline struct shmid_kernel *shm_lock(struct ipc_namespace *ns, int id)
Linus Torvalds's avatar
Linus Torvalds committed
155
{
Nadia Derbey's avatar
Nadia Derbey committed
156 157
	struct kern_ipc_perm *ipcp = ipc_lock(&shm_ids(ns), id);

158 159 160
	if (IS_ERR(ipcp))
		return (struct shmid_kernel *)ipcp;

Nadia Derbey's avatar
Nadia Derbey committed
161
	return container_of(ipcp, struct shmid_kernel, shm_perm);
162 163
}

164 165 166
static inline void shm_lock_by_ptr(struct shmid_kernel *ipcp)
{
	rcu_read_lock();
167
	ipc_lock_object(&ipcp->shm_perm);
168 169
}

Davidlohr Bueso's avatar
Davidlohr Bueso committed
170 171 172 173 174 175 176 177 178
static void shm_rcu_free(struct rcu_head *head)
{
	struct ipc_rcu *p = container_of(head, struct ipc_rcu, rcu);
	struct shmid_kernel *shp = ipc_rcu_to_struct(p);

	security_shm_free(shp);
	ipc_rcu_free(head);
}

Nadia Derbey's avatar
Nadia Derbey committed
179
static inline void shm_rmid(struct ipc_namespace *ns, struct shmid_kernel *s)
Linus Torvalds's avatar
Linus Torvalds committed
180
{
Nadia Derbey's avatar
Nadia Derbey committed
181
	ipc_rmid(&shm_ids(ns), &s->shm_perm);
Linus Torvalds's avatar
Linus Torvalds committed
182 183 184
}


185 186
/* This is called by fork, once for every shm attach. */
static void shm_open(struct vm_area_struct *vma)
Kirill Korotaev's avatar
Kirill Korotaev committed
187
{
188 189
	struct file *file = vma->vm_file;
	struct shm_file_data *sfd = shm_file_data(file);
Linus Torvalds's avatar
Linus Torvalds committed
190 191
	struct shmid_kernel *shp;

192
	shp = shm_lock(sfd->ns, sfd->id);
193
	BUG_ON(IS_ERR(shp));
Linus Torvalds's avatar
Linus Torvalds committed
194
	shp->shm_atim = get_seconds();
195
	shp->shm_lprid = task_tgid_vnr(current);
Linus Torvalds's avatar
Linus Torvalds committed
196 197 198 199 200 201 202
	shp->shm_nattch++;
	shm_unlock(shp);
}

/*
 * shm_destroy - free the struct shmid_kernel
 *
Nadia Derbey's avatar
Nadia Derbey committed
203
 * @ns: namespace
Linus Torvalds's avatar
Linus Torvalds committed
204 205
 * @shp: struct to free
 *
Davidlohr Bueso's avatar
Davidlohr Bueso committed
206
 * It has to be called with shp and shm_ids.rwsem (writer) locked,
Linus Torvalds's avatar
Linus Torvalds committed
207 208
 * but returns with shp unlocked and freed.
 */
Kirill Korotaev's avatar
Kirill Korotaev committed
209
static void shm_destroy(struct ipc_namespace *ns, struct shmid_kernel *shp)
Linus Torvalds's avatar
Linus Torvalds committed
210
{
211 212 213 214
	struct file *shm_file;

	shm_file = shp->shm_file;
	shp->shm_file = NULL;
Kirill Korotaev's avatar
Kirill Korotaev committed
215
	ns->shm_tot -= (shp->shm_segsz + PAGE_SIZE - 1) >> PAGE_SHIFT;
Nadia Derbey's avatar
Nadia Derbey committed
216
	shm_rmid(ns, shp);
Linus Torvalds's avatar
Linus Torvalds committed
217
	shm_unlock(shp);
218 219
	if (!is_file_hugepages(shm_file))
		shmem_lock(shm_file, 0, shp->mlock_user);
220
	else if (shp->mlock_user)
221 222
		user_shm_unlock(file_inode(shm_file)->i_size, shp->mlock_user);
	fput(shm_file);
Davidlohr Bueso's avatar
Davidlohr Bueso committed
223
	ipc_rcu_putref(shp, shm_rcu_free);
Linus Torvalds's avatar
Linus Torvalds committed
224 225
}

226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
/*
 * shm_may_destroy - identifies whether shm segment should be destroyed now
 *
 * Returns true if and only if there are no active users of the segment and
 * one of the following is true:
 *
 * 1) shmctl(id, IPC_RMID, NULL) was called for this shp
 *
 * 2) sysctl kernel.shm_rmid_forced is set to 1.
 */
static bool shm_may_destroy(struct ipc_namespace *ns, struct shmid_kernel *shp)
{
	return (shp->shm_nattch == 0) &&
	       (ns->shm_rmid_forced ||
		(shp->shm_perm.mode & SHM_DEST));
}

Linus Torvalds's avatar
Linus Torvalds committed
243
/*
244
 * remove the attach descriptor vma.
Linus Torvalds's avatar
Linus Torvalds committed
245 246 247 248
 * free memory for segment if it is marked destroyed.
 * The descriptor has already been removed from the current->mm->mmap list
 * and will later be kfree()d.
 */
249
static void shm_close(struct vm_area_struct *vma)
Linus Torvalds's avatar
Linus Torvalds committed
250
{
Manfred Spraul's avatar
Manfred Spraul committed
251
	struct file *file = vma->vm_file;
252
	struct shm_file_data *sfd = shm_file_data(file);
Linus Torvalds's avatar
Linus Torvalds committed
253
	struct shmid_kernel *shp;
254
	struct ipc_namespace *ns = sfd->ns;
Kirill Korotaev's avatar
Kirill Korotaev committed
255

Davidlohr Bueso's avatar
Davidlohr Bueso committed
256
	down_write(&shm_ids(ns).rwsem);
Linus Torvalds's avatar
Linus Torvalds committed
257
	/* remove from the list of attaches of the shm segment */
Nadia Derbey's avatar
Nadia Derbey committed
258
	shp = shm_lock(ns, sfd->id);
259
	BUG_ON(IS_ERR(shp));
260
	shp->shm_lprid = task_tgid_vnr(current);
Linus Torvalds's avatar
Linus Torvalds committed
261 262
	shp->shm_dtim = get_seconds();
	shp->shm_nattch--;
263 264 265 266
	if (shm_may_destroy(ns, shp))
		shm_destroy(ns, shp);
	else
		shm_unlock(shp);
Davidlohr Bueso's avatar
Davidlohr Bueso committed
267
	up_write(&shm_ids(ns).rwsem);
268 269
}

Davidlohr Bueso's avatar
Davidlohr Bueso committed
270
/* Called with ns->shm_ids(ns).rwsem locked */
271 272 273
static int shm_try_destroy_current(int id, void *p, void *data)
{
	struct ipc_namespace *ns = data;
274 275
	struct kern_ipc_perm *ipcp = p;
	struct shmid_kernel *shp = container_of(ipcp, struct shmid_kernel, shm_perm);
276

277
	if (shp->shm_creator != current)
278 279 280 281 282 283 284 285 286 287 288 289 290
		return 0;

	/*
	 * Mark it as orphaned to destroy the segment when
	 * kernel.shm_rmid_forced is changed.
	 * It is noop if the following shm_may_destroy() returns true.
	 */
	shp->shm_creator = NULL;

	/*
	 * Don't even try to destroy it.  If shm_rmid_forced=0 and IPC_RMID
	 * is not set, it shouldn't be deleted here.
	 */
291
	if (!ns->shm_rmid_forced)
292 293
		return 0;

294 295
	if (shm_may_destroy(ns, shp)) {
		shm_lock_by_ptr(shp);
296
		shm_destroy(ns, shp);
297
	}
298 299 300
	return 0;
}

Davidlohr Bueso's avatar
Davidlohr Bueso committed
301
/* Called with ns->shm_ids(ns).rwsem locked */
302 303 304
static int shm_try_destroy_orphaned(int id, void *p, void *data)
{
	struct ipc_namespace *ns = data;
305 306
	struct kern_ipc_perm *ipcp = p;
	struct shmid_kernel *shp = container_of(ipcp, struct shmid_kernel, shm_perm);
307 308 309 310

	/*
	 * We want to destroy segments without users and with already
	 * exit'ed originating process.
311
	 *
Davidlohr Bueso's avatar
Davidlohr Bueso committed
312
	 * As shp->* are changed under rwsem, it's safe to skip shp locking.
313
	 */
314
	if (shp->shm_creator != NULL)
315 316
		return 0;

317 318
	if (shm_may_destroy(ns, shp)) {
		shm_lock_by_ptr(shp);
Kirill Korotaev's avatar
Kirill Korotaev committed
319
		shm_destroy(ns, shp);
320
	}
321 322 323 324 325
	return 0;
}

void shm_destroy_orphaned(struct ipc_namespace *ns)
{
Davidlohr Bueso's avatar
Davidlohr Bueso committed
326
	down_write(&shm_ids(ns).rwsem);
Vasiliy Kulikov's avatar
Vasiliy Kulikov committed
327
	if (shm_ids(ns).in_use)
328
		idr_for_each(&shm_ids(ns).ipcs_idr, &shm_try_destroy_orphaned, ns);
Davidlohr Bueso's avatar
Davidlohr Bueso committed
329
	up_write(&shm_ids(ns).rwsem);
330 331 332 333 334
}


void exit_shm(struct task_struct *task)
{
335
	struct ipc_namespace *ns = task->nsproxy->ipc_ns;
336

Vasiliy Kulikov's avatar
Vasiliy Kulikov committed
337 338 339
	if (shm_ids(ns).in_use == 0)
		return;

340
	/* Destroy all already created segments, but not mapped yet */
Davidlohr Bueso's avatar
Davidlohr Bueso committed
341
	down_write(&shm_ids(ns).rwsem);
Vasiliy Kulikov's avatar
Vasiliy Kulikov committed
342
	if (shm_ids(ns).in_use)
343
		idr_for_each(&shm_ids(ns).ipcs_idr, &shm_try_destroy_current, ns);
Davidlohr Bueso's avatar
Davidlohr Bueso committed
344
	up_write(&shm_ids(ns).rwsem);
Linus Torvalds's avatar
Linus Torvalds committed
345 346
}

Nick Piggin's avatar
Nick Piggin committed
347
static int shm_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
348 349 350 351
{
	struct file *file = vma->vm_file;
	struct shm_file_data *sfd = shm_file_data(file);

Nick Piggin's avatar
Nick Piggin committed
352
	return sfd->vm_ops->fault(vma, vmf);
353 354 355
}

#ifdef CONFIG_NUMA
356
static int shm_set_policy(struct vm_area_struct *vma, struct mempolicy *new)
357 358 359 360 361 362 363 364 365
{
	struct file *file = vma->vm_file;
	struct shm_file_data *sfd = shm_file_data(file);
	int err = 0;
	if (sfd->vm_ops->set_policy)
		err = sfd->vm_ops->set_policy(vma, new);
	return err;
}

366 367
static struct mempolicy *shm_get_policy(struct vm_area_struct *vma,
					unsigned long addr)
368 369 370 371 372 373 374
{
	struct file *file = vma->vm_file;
	struct shm_file_data *sfd = shm_file_data(file);
	struct mempolicy *pol = NULL;

	if (sfd->vm_ops->get_policy)
		pol = sfd->vm_ops->get_policy(vma, addr);
375
	else if (vma->vm_policy)
376
		pol = vma->vm_policy;
377

378 379 380 381
	return pol;
}
#endif

Manfred Spraul's avatar
Manfred Spraul committed
382
static int shm_mmap(struct file *file, struct vm_area_struct *vma)
Linus Torvalds's avatar
Linus Torvalds committed
383
{
384
	struct shm_file_data *sfd = shm_file_data(file);
385 386
	int ret;

387 388 389 390
	ret = sfd->file->f_op->mmap(sfd->file, vma);
	if (ret != 0)
		return ret;
	sfd->vm_ops = vma->vm_ops;
David Howells's avatar
David Howells committed
391
#ifdef CONFIG_MMU
392
	BUG_ON(!sfd->vm_ops->fault);
David Howells's avatar
David Howells committed
393
#endif
394 395
	vma->vm_ops = &shm_vm_ops;
	shm_open(vma);
396 397

	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
398 399
}

Kirill Korotaev's avatar
Kirill Korotaev committed
400 401
static int shm_release(struct inode *ino, struct file *file)
{
402
	struct shm_file_data *sfd = shm_file_data(file);
Kirill Korotaev's avatar
Kirill Korotaev committed
403

404 405 406
	put_ipc_ns(sfd->ns);
	shm_file_data(file) = NULL;
	kfree(sfd);
Kirill Korotaev's avatar
Kirill Korotaev committed
407 408 409
	return 0;
}

410
static int shm_fsync(struct file *file, loff_t start, loff_t end, int datasync)
411 412 413
{
	struct shm_file_data *sfd = shm_file_data(file);

414 415
	if (!sfd->file->f_op->fsync)
		return -EINVAL;
416
	return sfd->file->f_op->fsync(sfd->file, start, end, datasync);
417 418
}

419 420 421 422 423 424 425 426 427 428
static long shm_fallocate(struct file *file, int mode, loff_t offset,
			  loff_t len)
{
	struct shm_file_data *sfd = shm_file_data(file);

	if (!sfd->file->f_op->fallocate)
		return -EOPNOTSUPP;
	return sfd->file->f_op->fallocate(file, mode, offset, len);
}

429 430 431 432 433
static unsigned long shm_get_unmapped_area(struct file *file,
	unsigned long addr, unsigned long len, unsigned long pgoff,
	unsigned long flags)
{
	struct shm_file_data *sfd = shm_file_data(file);
434 435
	return sfd->file->f_op->get_unmapped_area(sfd->file, addr, len,
						pgoff, flags);
436 437
}

438
static const struct file_operations shm_file_operations = {
Kirill Korotaev's avatar
Kirill Korotaev committed
439
	.mmap		= shm_mmap,
440
	.fsync		= shm_fsync,
Kirill Korotaev's avatar
Kirill Korotaev committed
441
	.release	= shm_release,
David Howells's avatar
David Howells committed
442 443 444
#ifndef CONFIG_MMU
	.get_unmapped_area	= shm_get_unmapped_area,
#endif
445
	.llseek		= noop_llseek,
446
	.fallocate	= shm_fallocate,
447 448 449 450 451 452
};

static const struct file_operations shm_file_operations_huge = {
	.mmap		= shm_mmap,
	.fsync		= shm_fsync,
	.release	= shm_release,
453
	.get_unmapped_area	= shm_get_unmapped_area,
454
	.llseek		= noop_llseek,
455
	.fallocate	= shm_fallocate,
Linus Torvalds's avatar
Linus Torvalds committed
456 457
};

458 459 460 461 462
int is_file_shm_hugepages(struct file *file)
{
	return file->f_op == &shm_file_operations_huge;
}

463
static const struct vm_operations_struct shm_vm_ops = {
Linus Torvalds's avatar
Linus Torvalds committed
464 465
	.open	= shm_open,	/* callback for a new vm-area open */
	.close	= shm_close,	/* callback for when the vm-area is released */
466
	.fault	= shm_fault,
467 468 469
#if defined(CONFIG_NUMA)
	.set_policy = shm_set_policy,
	.get_policy = shm_get_policy,
Linus Torvalds's avatar
Linus Torvalds committed
470 471 472
#endif
};

Nadia Derbey's avatar
Nadia Derbey committed
473 474 475 476 477
/**
 * newseg - Create a new shared memory segment
 * @ns: namespace
 * @params: ptr to the structure that contains key, size and shmflg
 *
Davidlohr Bueso's avatar
Davidlohr Bueso committed
478
 * Called with shm_ids.rwsem held as a writer.
Nadia Derbey's avatar
Nadia Derbey committed
479
 */
Nadia Derbey's avatar
Nadia Derbey committed
480
static int newseg(struct ipc_namespace *ns, struct ipc_params *params)
Linus Torvalds's avatar
Linus Torvalds committed
481
{
Nadia Derbey's avatar
Nadia Derbey committed
482 483 484
	key_t key = params->key;
	int shmflg = params->flg;
	size_t size = params->u.size;
Linus Torvalds's avatar
Linus Torvalds committed
485 486
	int error;
	struct shmid_kernel *shp;
487
	size_t numpages = (size + PAGE_SIZE - 1) >> PAGE_SHIFT;
Manfred Spraul's avatar
Manfred Spraul committed
488
	struct file *file;
Linus Torvalds's avatar
Linus Torvalds committed
489 490
	char name[13];
	int id;
491
	vm_flags_t acctflag = 0;
Linus Torvalds's avatar
Linus Torvalds committed
492

Kirill Korotaev's avatar
Kirill Korotaev committed
493
	if (size < SHMMIN || size > ns->shm_ctlmax)
Linus Torvalds's avatar
Linus Torvalds committed
494 495
		return -EINVAL;

496
	if (ns->shm_tot + numpages > ns->shm_ctlall)
Linus Torvalds's avatar
Linus Torvalds committed
497 498 499 500 501 502 503
		return -ENOSPC;

	shp = ipc_rcu_alloc(sizeof(*shp));
	if (!shp)
		return -ENOMEM;

	shp->shm_perm.key = key;
Andrew Morton's avatar
Andrew Morton committed
504
	shp->shm_perm.mode = (shmflg & S_IRWXUGO);
Linus Torvalds's avatar
Linus Torvalds committed
505 506 507 508 509
	shp->mlock_user = NULL;

	shp->shm_perm.security = NULL;
	error = security_shm_alloc(shp);
	if (error) {
Davidlohr Bueso's avatar
Davidlohr Bueso committed
510
		ipc_rcu_putref(shp, ipc_rcu_free);
Linus Torvalds's avatar
Linus Torvalds committed
511 512 513
		return error;
	}

Manfred Spraul's avatar
Manfred Spraul committed
514
	sprintf(name, "SYSV%08x", key);
Linus Torvalds's avatar
Linus Torvalds committed
515
	if (shmflg & SHM_HUGETLB) {
516
		struct hstate *hs;
517 518
		size_t hugesize;

519
		hs = hstate_sizelog((shmflg >> SHM_HUGE_SHIFT) & SHM_HUGE_MASK);
520 521 522 523 524
		if (!hs) {
			error = -EINVAL;
			goto no_file;
		}
		hugesize = ALIGN(size, huge_page_size(hs));
525

526 527 528
		/* hugetlb_file_setup applies strict accounting */
		if (shmflg & SHM_NORESERVE)
			acctflag = VM_NORESERVE;
529
		file = hugetlb_file_setup(name, hugesize, acctflag,
530 531
				  &shp->mlock_user, HUGETLB_SHMFS_INODE,
				(shmflg >> SHM_HUGE_SHIFT) & SHM_HUGE_MASK);
Linus Torvalds's avatar
Linus Torvalds committed
532
	} else {
533 534
		/*
		 * Do not allow no accounting for OVERCOMMIT_NEVER, even
Manfred Spraul's avatar
Manfred Spraul committed
535
		 * if it's asked for.
536 537 538
		 */
		if  ((shmflg & SHM_NORESERVE) &&
				sysctl_overcommit_memory != OVERCOMMIT_NEVER)
539
			acctflag = VM_NORESERVE;
540
		file = shmem_file_setup(name, size, acctflag);
Linus Torvalds's avatar
Linus Torvalds committed
541 542 543 544 545
	}
	error = PTR_ERR(file);
	if (IS_ERR(file))
		goto no_file;

546
	id = ipc_addid(&shm_ids(ns), &shp->shm_perm, ns->shm_ctlmni);
547 548
	if (id < 0) {
		error = id;
Linus Torvalds's avatar
Linus Torvalds committed
549
		goto no_id;
550
	}
Linus Torvalds's avatar
Linus Torvalds committed
551

552
	shp->shm_cprid = task_tgid_vnr(current);
Linus Torvalds's avatar
Linus Torvalds committed
553 554 555 556 557 558
	shp->shm_lprid = 0;
	shp->shm_atim = shp->shm_dtim = 0;
	shp->shm_ctim = get_seconds();
	shp->shm_segsz = size;
	shp->shm_nattch = 0;
	shp->shm_file = file;
559
	shp->shm_creator = current;
560

561 562 563 564
	/*
	 * shmid gets reported as "inode#" in /proc/pid/maps.
	 * proc-ps tools use this. Changing this will break them.
	 */
Al Viro's avatar
Al Viro committed
565
	file_inode(file)->i_ino = shp->shm_perm.id;
566

Kirill Korotaev's avatar
Kirill Korotaev committed
567
	ns->shm_tot += numpages;
Nadia Derbey's avatar
Nadia Derbey committed
568
	error = shp->shm_perm.id;
569

570
	ipc_unlock_object(&shp->shm_perm);
571
	rcu_read_unlock();
Nadia Derbey's avatar
Nadia Derbey committed
572
	return error;
Linus Torvalds's avatar
Linus Torvalds committed
573 574

no_id:
575
	if (is_file_hugepages(file) && shp->mlock_user)
576
		user_shm_unlock(size, shp->mlock_user);
Linus Torvalds's avatar
Linus Torvalds committed
577 578
	fput(file);
no_file:
Davidlohr Bueso's avatar
Davidlohr Bueso committed
579
	ipc_rcu_putref(shp, shm_rcu_free);
Linus Torvalds's avatar
Linus Torvalds committed
580 581 582
	return error;
}

Nadia Derbey's avatar
Nadia Derbey committed
583
/*
Davidlohr Bueso's avatar
Davidlohr Bueso committed
584
 * Called with shm_ids.rwsem and ipcp locked.
Nadia Derbey's avatar
Nadia Derbey committed
585
 */
Nadia Derbey's avatar
Nadia Derbey committed
586
static inline int shm_security(struct kern_ipc_perm *ipcp, int shmflg)
Nadia Derbey's avatar
Nadia Derbey committed
587
{
Nadia Derbey's avatar
Nadia Derbey committed
588 589 590 591
	struct shmid_kernel *shp;

	shp = container_of(ipcp, struct shmid_kernel, shm_perm);
	return security_shm_associate(shp, shmflg);
Nadia Derbey's avatar
Nadia Derbey committed
592 593
}

Nadia Derbey's avatar
Nadia Derbey committed
594
/*
Davidlohr Bueso's avatar
Davidlohr Bueso committed
595
 * Called with shm_ids.rwsem and ipcp locked.
Nadia Derbey's avatar
Nadia Derbey committed
596
 */
Nadia Derbey's avatar
Nadia Derbey committed
597 598
static inline int shm_more_checks(struct kern_ipc_perm *ipcp,
				struct ipc_params *params)
Nadia Derbey's avatar
Nadia Derbey committed
599
{
Nadia Derbey's avatar
Nadia Derbey committed
600 601 602 603
	struct shmid_kernel *shp;

	shp = container_of(ipcp, struct shmid_kernel, shm_perm);
	if (shp->shm_segsz < params->u.size)
Nadia Derbey's avatar
Nadia Derbey committed
604 605 606 607 608
		return -EINVAL;

	return 0;
}

609
SYSCALL_DEFINE3(shmget, key_t, key, size_t, size, int, shmflg)
Linus Torvalds's avatar
Linus Torvalds committed
610
{
Kirill Korotaev's avatar
Kirill Korotaev committed
611
	struct ipc_namespace *ns;
Nadia Derbey's avatar
Nadia Derbey committed
612 613
	struct ipc_ops shm_ops;
	struct ipc_params shm_params;
Kirill Korotaev's avatar
Kirill Korotaev committed
614 615

	ns = current->nsproxy->ipc_ns;
Linus Torvalds's avatar
Linus Torvalds committed
616

Nadia Derbey's avatar
Nadia Derbey committed
617 618 619
	shm_ops.getnew = newseg;
	shm_ops.associate = shm_security;
	shm_ops.more_checks = shm_more_checks;
Nadia Derbey's avatar
Nadia Derbey committed
620

Nadia Derbey's avatar
Nadia Derbey committed
621 622 623
	shm_params.key = key;
	shm_params.flg = shmflg;
	shm_params.u.size = size;
Linus Torvalds's avatar
Linus Torvalds committed
624

Nadia Derbey's avatar
Nadia Derbey committed
625
	return ipcget(ns, &shm_ids(ns), &shm_ops, &shm_params);
Linus Torvalds's avatar
Linus Torvalds committed
626 627 628 629
}

static inline unsigned long copy_shmid_to_user(void __user *buf, struct shmid64_ds *in, int version)
{
Manfred Spraul's avatar
Manfred Spraul committed
630
	switch (version) {
Linus Torvalds's avatar
Linus Torvalds committed
631 632 633 634 635 636
	case IPC_64:
		return copy_to_user(buf, in, sizeof(*in));
	case IPC_OLD:
	    {
		struct shmid_ds out;

637
		memset(&out, 0, sizeof(out));
Linus Torvalds's avatar
Linus Torvalds committed
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653
		ipc64_perm_to_ipc_perm(&in->shm_perm, &out.shm_perm);
		out.shm_segsz	= in->shm_segsz;
		out.shm_atime	= in->shm_atime;
		out.shm_dtime	= in->shm_dtime;
		out.shm_ctime	= in->shm_ctime;
		out.shm_cpid	= in->shm_cpid;
		out.shm_lpid	= in->shm_lpid;
		out.shm_nattch	= in->shm_nattch;

		return copy_to_user(buf, &out, sizeof(out));
	    }
	default:
		return -EINVAL;
	}
}

654 655
static inline unsigned long
copy_shmid_from_user(struct shmid64_ds *out, void __user *buf, int version)
Linus Torvalds's avatar
Linus Torvalds committed
656
{
Manfred Spraul's avatar
Manfred Spraul committed
657
	switch (version) {
Linus Torvalds's avatar
Linus Torvalds committed
658
	case IPC_64:
659
		if (copy_from_user(out, buf, sizeof(*out)))
Linus Torvalds's avatar
Linus Torvalds committed
660 661 662 663 664 665 666 667 668
			return -EFAULT;
		return 0;
	case IPC_OLD:
	    {
		struct shmid_ds tbuf_old;

		if (copy_from_user(&tbuf_old, buf, sizeof(tbuf_old)))
			return -EFAULT;

669 670 671
		out->shm_perm.uid	= tbuf_old.shm_perm.uid;
		out->shm_perm.gid	= tbuf_old.shm_perm.gid;
		out->shm_perm.mode	= tbuf_old.shm_perm.mode;
Linus Torvalds's avatar
Linus Torvalds committed
672 673 674 675 676 677 678 679 680 681

		return 0;
	    }
	default:
		return -EINVAL;
	}
}

static inline unsigned long copy_shminfo_to_user(void __user *buf, struct shminfo64 *in, int version)
{
Manfred Spraul's avatar
Manfred Spraul committed
682
	switch (version) {
Linus Torvalds's avatar
Linus Torvalds committed
683 684 685 686 687 688
	case IPC_64:
		return copy_to_user(buf, in, sizeof(*in));
	case IPC_OLD:
	    {
		struct shminfo out;

Manfred Spraul's avatar
Manfred Spraul committed
689
		if (in->shmmax > INT_MAX)
Linus Torvalds's avatar
Linus Torvalds committed
690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705
			out.shmmax = INT_MAX;
		else
			out.shmmax = (int)in->shmmax;

		out.shmmin	= in->shmmin;
		out.shmmni	= in->shmmni;
		out.shmseg	= in->shmseg;
		out.shmall	= in->shmall; 

		return copy_to_user(buf, &out, sizeof(out));
	    }
	default:
		return -EINVAL;
	}
}

706 707
/*
 * Calculate and add used RSS and swap pages of a shm.
Davidlohr Bueso's avatar
Davidlohr Bueso committed
708
 * Called with shm_ids.rwsem held as a reader
709 710 711 712 713 714
 */
static void shm_add_rss_swap(struct shmid_kernel *shp,
	unsigned long *rss_add, unsigned long *swp_add)
{
	struct inode *inode;

Al Viro's avatar
Al Viro committed
715
	inode = file_inode(shp->shm_file);
716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733

	if (is_file_hugepages(shp->shm_file)) {
		struct address_space *mapping = inode->i_mapping;
		struct hstate *h = hstate_file(shp->shm_file);
		*rss_add += pages_per_huge_page(h) * mapping->nrpages;
	} else {
#ifdef CONFIG_SHMEM
		struct shmem_inode_info *info = SHMEM_I(inode);
		spin_lock(&info->lock);
		*rss_add += inode->i_mapping->nrpages;
		*swp_add += info->swapped;
		spin_unlock(&info->lock);
#else
		*rss_add += inode->i_mapping->nrpages;
#endif
	}
}

Nadia Derbey's avatar
Nadia Derbey committed
734
/*
Davidlohr Bueso's avatar
Davidlohr Bueso committed
735
 * Called with shm_ids.rwsem held as a reader
Nadia Derbey's avatar
Nadia Derbey committed
736
 */
Kirill Korotaev's avatar
Kirill Korotaev committed
737 738
static void shm_get_stat(struct ipc_namespace *ns, unsigned long *rss,
		unsigned long *swp)
Linus Torvalds's avatar
Linus Torvalds committed
739
{
Nadia Derbey's avatar
Nadia Derbey committed
740 741
	int next_id;
	int total, in_use;
Linus Torvalds's avatar
Linus Torvalds committed
742 743 744 745

	*rss = 0;
	*swp = 0;

Nadia Derbey's avatar
Nadia Derbey committed
746 747 748
	in_use = shm_ids(ns).in_use;

	for (total = 0, next_id = 0; total < in_use; next_id++) {
749
		struct kern_ipc_perm *ipc;
Linus Torvalds's avatar
Linus Torvalds committed
750 751
		struct shmid_kernel *shp;

752 753
		ipc = idr_find(&shm_ids(ns).ipcs_idr, next_id);
		if (ipc == NULL)
Linus Torvalds's avatar
Linus Torvalds committed
754
			continue;
755
		shp = container_of(ipc, struct shmid_kernel, shm_perm);
Linus Torvalds's avatar
Linus Torvalds committed
756

757
		shm_add_rss_swap(shp, rss, swp);
Nadia Derbey's avatar
Nadia Derbey committed
758 759

		total++;
Linus Torvalds's avatar
Linus Torvalds committed
760 761 762
	}
}

763
/*
Davidlohr Bueso's avatar
Davidlohr Bueso committed
764
 * This function handles some shmctl commands which require the rwsem
765
 * to be held in write mode.
Davidlohr Bueso's avatar
Davidlohr Bueso committed
766
 * NOTE: no locks must be held, the rwsem is taken inside this function.
767 768 769
 */
static int shmctl_down(struct ipc_namespace *ns, int shmid, int cmd,
		       struct shmid_ds __user *buf, int version)
Linus Torvalds's avatar
Linus Torvalds committed
770
{
771
	struct kern_ipc_perm *ipcp;
772
	struct shmid64_ds shmid64;
773 774 775 776
	struct shmid_kernel *shp;
	int err;

	if (cmd == IPC_SET) {
777
		if (copy_shmid_from_user(&shmid64, buf, version))
778 779 780
			return -EFAULT;
	}

Davidlohr Bueso's avatar
Davidlohr Bueso committed
781
	down_write(&shm_ids(ns).rwsem);
782 783
	rcu_read_lock();

784 785
	ipcp = ipcctl_pre_down_nolock(ns, &shm_ids(ns), shmid, cmd,
				      &shmid64.shm_perm, 0);
786 787 788 789
	if (IS_ERR(ipcp)) {
		err = PTR_ERR(ipcp);
		goto out_unlock1;
	}
790

791
	shp = container_of(ipcp, struct shmid_kernel, shm_perm);
792 793 794

	err = security_shm_shmctl(shp, cmd);
	if (err)
795
		goto out_unlock1;
796

797 798
	switch (cmd) {
	case IPC_RMID:
799
		ipc_lock_object(&shp->shm_perm);
800
		/* do_shm_rmid unlocks the ipc object and rcu */
801 802 803
		do_shm_rmid(ns, ipcp);
		goto out_up;
	case IPC_SET:
804
		ipc_lock_object(&shp->shm_perm);
805 806
		err = ipc_update_perm(&shmid64.shm_perm, ipcp);
		if (err)
807
			goto out_unlock0;
808 809 810 811
		shp->shm_ctim = get_seconds();
		break;
	default:
		err = -EINVAL;
812
		goto out_unlock1;
813
	}
814 815 816 817 818

out_unlock0:
	ipc_unlock_object(&shp->shm_perm);
out_unlock1:
	rcu_read_unlock();
819
out_up:
Davidlohr Bueso's avatar
Davidlohr Bueso committed
820
	up_write(&shm_ids(ns).rwsem);
821 822 823
	return err;
}

824 825
static int shmctl_nolock(struct ipc_namespace *ns, int shmid,
			 int cmd, int version, void __user *buf)
826
{
827
	int err;
Linus Torvalds's avatar
Linus Torvalds committed
828 829
	struct shmid_kernel *shp;

830 831 832 833 834
	/* preliminary security checks for *_INFO */
	if (cmd == IPC_INFO || cmd == SHM_INFO) {
		err = security_shm_shmctl(NULL, cmd);
		if (err)
			return err;
Linus Torvalds's avatar
Linus Torvalds committed
835 836
	}

837
	switch (cmd) {
Linus Torvalds's avatar
Linus Torvalds committed
838 839 840 841
	case IPC_INFO:
	{
		struct shminfo64 shminfo;

WANG Cong's avatar
WANG Cong committed
842
		memset(&shminfo, 0, sizeof(shminfo));
Kirill Korotaev's avatar
Kirill Korotaev committed
843 844 845
		shminfo.shmmni = shminfo.shmseg = ns->shm_ctlmni;
		shminfo.shmmax = ns->shm_ctlmax;
		shminfo.shmall = ns->shm_ctlall;
Linus Torvalds's avatar
Linus Torvalds committed
846 847

		shminfo.shmmin = SHMMIN;
Manfred Spraul's avatar
Manfred Spraul committed
848
		if (copy_shminfo_to_user(buf, &shminfo, version))
Linus Torvalds's avatar
Linus Torvalds committed
849
			return -EFAULT;
Nadia Derbey's avatar
Nadia Derbey committed
850

Davidlohr Bueso's avatar
Davidlohr Bueso committed
851
		down_read(&shm_ids(ns).rwsem);
Nadia Derbey's avatar
Nadia Derbey committed
852
		err = ipc_get_maxid(&shm_ids(ns));
Davidlohr Bueso's avatar
Davidlohr Bueso committed
853
		up_read(&shm_ids(ns).rwsem);
Nadia Derbey's avatar
Nadia Derbey committed
854

Manfred Spraul's avatar
Manfred Spraul committed
855
		if (err < 0)
Linus Torvalds's avatar
Linus Torvalds committed
856 857 858 859 860 861 862
			err = 0;
		goto out;
	}
	case SHM_INFO:
	{
		struct shm_info shm_info;

WANG Cong's avatar
WANG Cong committed
863
		memset(&shm_info, 0, sizeof(shm_info));
Davidlohr Bueso's avatar
Davidlohr Bueso committed
864
		down_read(&shm_ids(ns).rwsem);
Kirill Korotaev's avatar
Kirill Korotaev committed
865
		shm_info.used_ids = shm_ids(ns).in_use;
Manfred Spraul's avatar
Manfred Spraul committed
866
		shm_get_stat(ns, &shm_info.shm_rss, &shm_info.shm_swp);
Kirill Korotaev's avatar
Kirill Korotaev committed
867
		shm_info.shm_tot = ns->shm_tot;
Linus Torvalds's avatar
Linus Torvalds committed
868 869
		shm_info.swap_attempts = 0;
		shm_info.swap_successes = 0;
Nadia Derbey's avatar
Nadia Derbey committed
870
		err = ipc_get_maxid(&shm_ids(ns));
Davidlohr Bueso's avatar
Davidlohr Bueso committed
871
		up_read(&shm_ids(ns).rwsem);
WANG Cong's avatar
WANG Cong committed
872
		if (copy_to_user(buf, &shm_info, sizeof(shm_info))) {
Linus Torvalds's avatar
Linus Torvalds committed
873 874 875 876 877 878 879 880 881 882 883 884
			err = -EFAULT;
			goto out;
		}

		err = err < 0 ? 0 : err;
		goto out;
	}
	case SHM_STAT:
	case IPC_STAT:
	{
		struct shmid64_ds tbuf;
		int result;
885

886
		rcu_read_lock();
887
		if (cmd == SHM_STAT) {
888
			shp = shm_obtain_object(ns, shmid);
889 890
			if (IS_ERR(shp)) {
				err = PTR_ERR(shp);
891
				goto out_unlock;
892
			}
Nadia Derbey's avatar
Nadia Derbey committed
893
			result = shp->shm_perm.id;
Linus Torvalds's avatar
Linus Torvalds committed
894
		} else {
895
			shp = shm_obtain_object_check(ns, shmid);
896 897
			if (IS_ERR(shp)) {
				err = PTR_ERR(shp);
898
				goto out_unlock;
899
			}
Linus Torvalds's avatar
Linus Torvalds committed
900 901
			result = 0;
		}
902

WANG Cong's avatar
WANG Cong committed
903
		err = -EACCES;
904
		if (ipcperms(ns, &shp->shm_perm, S_IRUGO))
Linus Torvalds's avatar
Linus Torvalds committed
905
			goto out_unlock;
906

Linus Torvalds's avatar
Linus Torvalds committed
907 908 909
		err = security_shm_shmctl(shp, cmd);
		if (err)
			goto out_unlock;
910

911
		memset(&tbuf, 0, sizeof(tbuf));
Linus Torvalds's avatar
Linus Torvalds committed
912 913 914 915 916 917 918
		kernel_to_ipc64_perm(&shp->shm_perm, &tbuf.shm_perm);
		tbuf.shm_segsz	= shp->shm_segsz;
		tbuf.shm_atime	= shp->shm_atim;
		tbuf.shm_dtime	= shp->shm_dtim;
		tbuf.shm_ctime	= shp->shm_ctim;
		tbuf.shm_cpid	= shp->shm_cprid;
		tbuf.shm_lpid	= shp->shm_lprid;
919
		tbuf.shm_nattch	= shp->shm_nattch;
920 921 922
		rcu_read_unlock();

		if (copy_shmid_to_user(buf, &tbuf, version))
Linus Torvalds's avatar
Linus Torvalds committed
923 924 925 926 927
			err = -EFAULT;
		else
			err = result;
		goto out;
	}
928 929 930 931 932
	default:
		return -EINVAL;
	}

out_unlock:
933
	rcu_read_unlock();
934 935 936 937 938 939 940 941 942 943
out:
	return err;
}

SYSCALL_DEFINE3(shmctl, int, shmid, int, cmd, struct shmid_ds __user *, buf)
{
	struct shmid_kernel *shp;
	int err, version;
	struct ipc_namespace *ns;

944 945
	if (cmd < 0 || shmid < 0)
		return -EINVAL;
946 947 948 949 950 951 952 953 954 955

	version = ipc_parse_version(&cmd);
	ns = current->nsproxy->ipc_ns;

	switch (cmd) {
	case IPC_INFO:
	case SHM_INFO:
	case SHM_STAT:
	case IPC_STAT:
		return shmctl_nolock(ns, shmid, cmd, version, buf);
956 957 958
	case IPC_RMID:
	case IPC_SET:
		return shmctl_down(ns, shmid, cmd, buf, version);
Linus Torvalds's avatar
Linus Torvalds committed
959 960 961
	case SHM_LOCK:
	case SHM_UNLOCK:
	{
962
		struct file *shm_file;
963

964 965
		rcu_read_lock();
		shp = shm_obtain_object_check(ns, shmid);
966 967
		if (IS_ERR(shp)) {
			err = PTR_ERR(shp);
968
			goto out_unlock1;
Linus Torvalds's avatar
Linus Torvalds committed
969 970
		}

Al Viro's avatar
Al Viro committed
971
		audit_ipc_obj(&(shp->shm_perm));
972 973 974
		err = security_shm_shmctl(shp, cmd);
		if (err)
			goto out_unlock1;
Steve Grubb's avatar
Steve Grubb committed
975

976
		ipc_lock_object(&shp->shm_perm);
977 978 979 980 981 982 983

		/* check if shm_destroy() is tearing down shp */
		if (!ipc_valid_object(&shp->shm_perm)) {
			err = -EIDRM;
			goto out_unlock0;
		}

984
		if (!ns_capable(ns->user_ns, CAP_IPC_LOCK)) {
985 986
			kuid_t euid = current_euid();
			if (!uid_eq(euid, shp->shm_perm.uid) &&
987 988
			    !uid_eq(euid, shp->shm_perm.cuid)) {
				err = -EPERM;
989
				goto out_unlock0;
990 991 992
			}
			if (cmd == SHM_LOCK && !rlimit(RLIMIT_MEMLOCK)) {
				err = -EPERM;
993
				goto out_unlock0;
994
			}
Linus Torvalds's avatar
Linus Torvalds committed
995 996
		}

997 998
		shm_file = shp->shm_file;
		if (is_file_hugepages(shm_file))
999
			goto out_unlock0;
1000 1001

		if (cmd == SHM_LOCK) {
1002
			struct user_struct *user = current_user();
1003 1004 1005 1006
			err = shmem_lock(shm_file, 1, user);
			if (!err && !(shp->shm_perm.mode & SHM_LOCKED)) {
				shp->shm_perm.mode |= SHM_LOCKED;
				shp->mlock_user = user;
Linus Torvalds's avatar
Linus Torvalds committed
1007
			}
1008
			goto out_unlock0;
Linus Torvalds's avatar
Linus Torvalds committed
1009
		}
1010 1011 1012

		/* SHM_UNLOCK */
		if (!(shp->shm_perm.mode & SHM_LOCKED))
1013
			goto out_unlock0;
1014 1015 1016 1017
		shmem_lock(shm_file, 0, shp->mlock_user);
		shp->shm_perm.mode &= ~SHM_LOCKED;
		shp->mlock_user = NULL;
		get_file(shm_file);
1018 1019
		ipc_unlock_object(&shp->shm_perm);
		rcu_read_unlock();
1020
		shmem_unlock_mapping(shm_file->f_mapping);
1021

1022
		fput(shm_file);
1023
		return err;
1024
	}
Linus Torvalds's avatar
Linus Torvalds committed
1025
	default:
1026
		return -EINVAL;
Linus Torvalds's avatar
Linus Torvalds committed
1027 1028
	}

1029 1030 1031 1032
out_unlock0:
	ipc_unlock_object(&shp->shm_perm);
out_unlock1:
	rcu_read_unlock();
Linus Torvalds's avatar
Linus Torvalds committed
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
	return err;
}

/*
 * Fix shmaddr, allocate descriptor, map shm, add attach descriptor to lists.
 *
 * NOTE! Despite the name, this is NOT a direct system call entrypoint. The
 * "raddr" thing points to kernel space, and there has to be a wrapper around
 * this.
 */
Will Deacon's avatar
Will Deacon committed
1043 1044
long do_shmat(int shmid, char __user *shmaddr, int shmflg, ulong *raddr,
	      unsigned long shmlba)
Linus Torvalds's avatar
Linus Torvalds committed
1045 1046 1047 1048
{
	struct shmid_kernel *shp;
	unsigned long addr;
	unsigned long size;
Manfred Spraul's avatar
Manfred Spraul committed
1049
	struct file *file;
Linus Torvalds's avatar
Linus Torvalds committed
1050 1051 1052 1053
	int    err;
	unsigned long flags;
	unsigned long prot;
	int acc_mode;
Kirill Korotaev's avatar
Kirill Korotaev committed
1054
	struct ipc_namespace *ns;
1055 1056
	struct shm_file_data *sfd;
	struct path path;
1057
	fmode_t f_mode;
1058
	unsigned long populate = 0;
Linus Torvalds's avatar
Linus Torvalds committed
1059

1060 1061
	err = -EINVAL;
	if (shmid < 0)
Linus Torvalds's avatar
Linus Torvalds committed
1062
		goto out;
1063
	else if ((addr = (ulong)shmaddr)) {
Will Deacon's avatar
Will Deacon committed
1064
		if (addr & (shmlba - 1)) {
Linus Torvalds's avatar
Linus Torvalds committed
1065
			if (shmflg & SHM_RND)
Will Deacon's avatar
Will Deacon committed
1066
				addr &= ~(shmlba - 1);	   /* round down */
Linus Torvalds's avatar
Linus Torvalds committed
1067 1068 1069 1070
			else
#ifndef __ARCH_FORCE_SHMLBA
				if (addr & ~PAGE_MASK)
#endif
1071
					goto out;
Linus Torvalds's avatar
Linus Torvalds committed
1072 1073 1074 1075
		}
		flags = MAP_SHARED | MAP_FIXED;
	} else {
		if ((shmflg & SHM_REMAP))
1076
			goto out;
Linus Torvalds's avatar
Linus Torvalds committed
1077 1078 1079 1080 1081 1082 1083

		flags = MAP_SHARED;
	}

	if (shmflg & SHM_RDONLY) {
		prot = PROT_READ;
		acc_mode = S_IRUGO;
1084
		f_mode = FMODE_READ;
Linus Torvalds's avatar
Linus Torvalds committed
1085 1086 1087
	} else {
		prot = PROT_READ | PROT_WRITE;
		acc_mode = S_IRUGO | S_IWUGO;