17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate * CDDL HEADER START
37c478bd9Sstevel@tonic-gate *
47c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the
57257d1b4Sraf * Common Development and Distribution License (the "License").
67257d1b4Sraf * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate *
87c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate * and limitations under the License.
127c478bd9Sstevel@tonic-gate *
137c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate *
197c478bd9Sstevel@tonic-gate * CDDL HEADER END
207c478bd9Sstevel@tonic-gate */
217257d1b4Sraf
227c478bd9Sstevel@tonic-gate /*
237257d1b4Sraf * Copyright 2008 Sun Microsystems, Inc. All rights reserved.
247c478bd9Sstevel@tonic-gate * Use is subject to license terms.
257c478bd9Sstevel@tonic-gate */
267c478bd9Sstevel@tonic-gate
277c478bd9Sstevel@tonic-gate /*
287c478bd9Sstevel@tonic-gate * plock - lock "segments" in physical memory.
297c478bd9Sstevel@tonic-gate *
307c478bd9Sstevel@tonic-gate * Supports SVID-compatible plock, taking into account dynamically linked
317c478bd9Sstevel@tonic-gate * objects (such as shared libraries).
327c478bd9Sstevel@tonic-gate */
337c478bd9Sstevel@tonic-gate
347257d1b4Sraf #include "lint.h"
357c478bd9Sstevel@tonic-gate #include <mtlib.h>
367c478bd9Sstevel@tonic-gate #include <sys/types.h>
377c478bd9Sstevel@tonic-gate #include <sys/mman.h>
387c478bd9Sstevel@tonic-gate #include <sys/lock.h>
397c478bd9Sstevel@tonic-gate #include <errno.h>
407c478bd9Sstevel@tonic-gate #include <stddef.h>
417c478bd9Sstevel@tonic-gate #include <unistd.h>
427c478bd9Sstevel@tonic-gate #include <thread.h>
437c478bd9Sstevel@tonic-gate #include <synch.h>
447c478bd9Sstevel@tonic-gate
457c478bd9Sstevel@tonic-gate /*
467c478bd9Sstevel@tonic-gate * Module-scope variables.
477c478bd9Sstevel@tonic-gate */
487c478bd9Sstevel@tonic-gate static int lock_state = 0; /* lock state */
497c478bd9Sstevel@tonic-gate static pid_t state_pid = -1; /* pid to which state belongs */
507c478bd9Sstevel@tonic-gate static mutex_t plock_lock = DEFAULTMUTEX;
517c478bd9Sstevel@tonic-gate
527c478bd9Sstevel@tonic-gate /*
537c478bd9Sstevel@tonic-gate * plock
547c478bd9Sstevel@tonic-gate */
557c478bd9Sstevel@tonic-gate int
plock(int op)567c478bd9Sstevel@tonic-gate plock(int op) /* desired operation */
577c478bd9Sstevel@tonic-gate {
587c478bd9Sstevel@tonic-gate int e; /* return value */
597c478bd9Sstevel@tonic-gate pid_t pid; /* current pid */
607c478bd9Sstevel@tonic-gate
617c478bd9Sstevel@tonic-gate lmutex_lock(&plock_lock);
627c478bd9Sstevel@tonic-gate
637c478bd9Sstevel@tonic-gate /*
647c478bd9Sstevel@tonic-gate * Validate state of lock's. If parent has forked, then
657c478bd9Sstevel@tonic-gate * the lock state needs to be reset (children do not inherit
667c478bd9Sstevel@tonic-gate * memory locks, and thus do not inherit their state).
677c478bd9Sstevel@tonic-gate */
687c478bd9Sstevel@tonic-gate if ((pid = getpid()) != state_pid) {
697c478bd9Sstevel@tonic-gate lock_state = 0;
707c478bd9Sstevel@tonic-gate state_pid = pid;
717c478bd9Sstevel@tonic-gate }
727c478bd9Sstevel@tonic-gate
737c478bd9Sstevel@tonic-gate /*
747c478bd9Sstevel@tonic-gate * Dispatch on operation. Note: plock and its relatives depend
757c478bd9Sstevel@tonic-gate * upon "op" being bit encoded.
767c478bd9Sstevel@tonic-gate */
777c478bd9Sstevel@tonic-gate switch (op) {
787c478bd9Sstevel@tonic-gate
797c478bd9Sstevel@tonic-gate /*
807c478bd9Sstevel@tonic-gate * UNLOCK: remove all memory locks. Requires that some be set!
817c478bd9Sstevel@tonic-gate */
827c478bd9Sstevel@tonic-gate case UNLOCK:
837c478bd9Sstevel@tonic-gate if (lock_state == 0) {
847c478bd9Sstevel@tonic-gate errno = EINVAL;
857c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
867c478bd9Sstevel@tonic-gate return (-1);
877c478bd9Sstevel@tonic-gate }
887c478bd9Sstevel@tonic-gate e = munlockall();
897c478bd9Sstevel@tonic-gate if (e) {
907c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
917c478bd9Sstevel@tonic-gate return (-1);
927c478bd9Sstevel@tonic-gate } else {
937c478bd9Sstevel@tonic-gate lock_state = 0;
947c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
957c478bd9Sstevel@tonic-gate return (0);
967c478bd9Sstevel@tonic-gate }
977c478bd9Sstevel@tonic-gate /*NOTREACHED*/
987c478bd9Sstevel@tonic-gate
997c478bd9Sstevel@tonic-gate /*
1007c478bd9Sstevel@tonic-gate * TXTLOCK: locks text segments.
1017c478bd9Sstevel@tonic-gate */
1027c478bd9Sstevel@tonic-gate case TXTLOCK:
1037c478bd9Sstevel@tonic-gate
1047c478bd9Sstevel@tonic-gate /*
1057c478bd9Sstevel@tonic-gate * If a text or process lock is already set, then fail.
1067c478bd9Sstevel@tonic-gate */
1077c478bd9Sstevel@tonic-gate if ((lock_state & TXTLOCK) || (lock_state & PROCLOCK)) {
1087c478bd9Sstevel@tonic-gate errno = EINVAL;
1097c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1107c478bd9Sstevel@tonic-gate return (-1);
1117c478bd9Sstevel@tonic-gate }
1127c478bd9Sstevel@tonic-gate
1137c478bd9Sstevel@tonic-gate /*
1147c478bd9Sstevel@tonic-gate * Try to apply the lock(s). If a failure occurs,
1157c478bd9Sstevel@tonic-gate * memcntl backs them out automatically.
1167c478bd9Sstevel@tonic-gate */
1177c478bd9Sstevel@tonic-gate e = memcntl(NULL, 0, MC_LOCKAS, (caddr_t)MCL_CURRENT,
118*e86c3f00SToomas Soome PROC_TEXT|PRIVATE, 0);
1197c478bd9Sstevel@tonic-gate if (!e)
1207c478bd9Sstevel@tonic-gate lock_state |= TXTLOCK;
1217c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1227c478bd9Sstevel@tonic-gate return (e);
1237c478bd9Sstevel@tonic-gate /*NOTREACHED*/
1247c478bd9Sstevel@tonic-gate
1257c478bd9Sstevel@tonic-gate /*
1267c478bd9Sstevel@tonic-gate * DATLOCK: locks data segment(s), including the stack and all
1277c478bd9Sstevel@tonic-gate * future growth in the address space.
1287c478bd9Sstevel@tonic-gate */
1297c478bd9Sstevel@tonic-gate case DATLOCK:
1307c478bd9Sstevel@tonic-gate
1317c478bd9Sstevel@tonic-gate /*
1327c478bd9Sstevel@tonic-gate * If a data or process lock is already set, then fail.
1337c478bd9Sstevel@tonic-gate */
1347c478bd9Sstevel@tonic-gate if ((lock_state & DATLOCK) || (lock_state & PROCLOCK)) {
1357c478bd9Sstevel@tonic-gate errno = EINVAL;
1367c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1377c478bd9Sstevel@tonic-gate return (-1);
1387c478bd9Sstevel@tonic-gate }
1397c478bd9Sstevel@tonic-gate
1407c478bd9Sstevel@tonic-gate /*
1417c478bd9Sstevel@tonic-gate * Try to lock the data and stack segments. On failure
1427c478bd9Sstevel@tonic-gate * memcntl undoes the locks internally.
1437c478bd9Sstevel@tonic-gate */
1447c478bd9Sstevel@tonic-gate e = memcntl(NULL, 0, MC_LOCKAS, (caddr_t)MCL_CURRENT,
145*e86c3f00SToomas Soome PROC_DATA|PRIVATE, 0);
1467c478bd9Sstevel@tonic-gate if (e) {
1477c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1487c478bd9Sstevel@tonic-gate return (-1);
1497c478bd9Sstevel@tonic-gate }
1507c478bd9Sstevel@tonic-gate
1517c478bd9Sstevel@tonic-gate /* try to set a lock for all future mappings. */
1527c478bd9Sstevel@tonic-gate e = mlockall(MCL_FUTURE);
1537c478bd9Sstevel@tonic-gate
1547c478bd9Sstevel@tonic-gate /*
1557c478bd9Sstevel@tonic-gate * If failures have occurred, back out the locks
1567c478bd9Sstevel@tonic-gate * and return failure.
1577c478bd9Sstevel@tonic-gate */
1587c478bd9Sstevel@tonic-gate if (e) {
1597c478bd9Sstevel@tonic-gate e = errno;
1607c478bd9Sstevel@tonic-gate (void) memcntl(NULL, 0, MC_UNLOCKAS,
161*e86c3f00SToomas Soome (caddr_t)MCL_CURRENT, PROC_DATA|PRIVATE, 0);
1627c478bd9Sstevel@tonic-gate errno = e;
1637c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1647c478bd9Sstevel@tonic-gate return (-1);
1657c478bd9Sstevel@tonic-gate }
1667c478bd9Sstevel@tonic-gate
1677c478bd9Sstevel@tonic-gate /*
1687c478bd9Sstevel@tonic-gate * Data, stack, and growth have been locked. Set state
1697c478bd9Sstevel@tonic-gate * and return success.
1707c478bd9Sstevel@tonic-gate */
1717c478bd9Sstevel@tonic-gate lock_state |= DATLOCK;
1727c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1737c478bd9Sstevel@tonic-gate return (0);
1747c478bd9Sstevel@tonic-gate /*NOTREACHED*/
1757c478bd9Sstevel@tonic-gate
1767c478bd9Sstevel@tonic-gate /*
1777c478bd9Sstevel@tonic-gate * PROCLOCK: lock everything, and all future things as well.
1787c478bd9Sstevel@tonic-gate * There should be nothing locked when this is called.
1797c478bd9Sstevel@tonic-gate */
1807c478bd9Sstevel@tonic-gate case PROCLOCK:
1817c478bd9Sstevel@tonic-gate if (lock_state) {
1827c478bd9Sstevel@tonic-gate errno = EINVAL;
1837c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1847c478bd9Sstevel@tonic-gate return (-1);
1857c478bd9Sstevel@tonic-gate }
1867c478bd9Sstevel@tonic-gate if (mlockall(MCL_CURRENT | MCL_FUTURE) == 0) {
1877c478bd9Sstevel@tonic-gate lock_state |= PROCLOCK;
1887c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1897c478bd9Sstevel@tonic-gate return (0);
1907c478bd9Sstevel@tonic-gate } else {
1917c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
1927c478bd9Sstevel@tonic-gate return (-1);
1937c478bd9Sstevel@tonic-gate }
1947c478bd9Sstevel@tonic-gate /*NOTREACHED*/
1957c478bd9Sstevel@tonic-gate
1967c478bd9Sstevel@tonic-gate /*
1977c478bd9Sstevel@tonic-gate * Invalid operation.
1987c478bd9Sstevel@tonic-gate */
1997c478bd9Sstevel@tonic-gate default:
2007c478bd9Sstevel@tonic-gate errno = EINVAL;
2017c478bd9Sstevel@tonic-gate lmutex_unlock(&plock_lock);
2027c478bd9Sstevel@tonic-gate return (-1);
2037c478bd9Sstevel@tonic-gate /*NOTREACHED*/
2047c478bd9Sstevel@tonic-gate }
2057c478bd9Sstevel@tonic-gate }
206