summaryrefslogtreecommitdiffstats
path: root/sys/kern/uipc_sem.c
diff options
context:
space:
mode:
Diffstat (limited to 'sys/kern/uipc_sem.c')
-rw-r--r--sys/kern/uipc_sem.c1091
1 files changed, 1091 insertions, 0 deletions
diff --git a/sys/kern/uipc_sem.c b/sys/kern/uipc_sem.c
new file mode 100644
index 0000000..c219844
--- /dev/null
+++ b/sys/kern/uipc_sem.c
@@ -0,0 +1,1091 @@
+/*-
+ * Copyright (c) 2002 Alfred Perlstein <alfred@FreeBSD.org>
+ * Copyright (c) 2003-2005 SPARTA, Inc.
+ * Copyright (c) 2005 Robert N. M. Watson
+ * All rights reserved.
+ *
+ * This software was developed for the FreeBSD Project in part by Network
+ * Associates Laboratories, the Security Research Division of Network
+ * Associates, Inc. under DARPA/SPAWAR contract N66001-01-C-8035 ("CBOSS"),
+ * as part of the DARPA CHATS research program.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <sys/cdefs.h>
+__FBSDID("$FreeBSD$");
+
+#include "opt_compat.h"
+#include "opt_posix.h"
+
+#include <sys/param.h>
+#include <sys/capability.h>
+#include <sys/condvar.h>
+#include <sys/fcntl.h>
+#include <sys/file.h>
+#include <sys/filedesc.h>
+#include <sys/fnv_hash.h>
+#include <sys/kernel.h>
+#include <sys/ksem.h>
+#include <sys/lock.h>
+#include <sys/malloc.h>
+#include <sys/module.h>
+#include <sys/mutex.h>
+#include <sys/priv.h>
+#include <sys/proc.h>
+#include <sys/posix4.h>
+#include <sys/_semaphore.h>
+#include <sys/stat.h>
+#include <sys/syscall.h>
+#include <sys/syscallsubr.h>
+#include <sys/sysctl.h>
+#include <sys/sysent.h>
+#include <sys/sysproto.h>
+#include <sys/systm.h>
+#include <sys/sx.h>
+#include <sys/vnode.h>
+
+#include <security/mac/mac_framework.h>
+
+FEATURE(p1003_1b_semaphores, "POSIX P1003.1B semaphores support");
+/*
+ * TODO
+ *
+ * - Resource limits?
+ * - Update fstat(1)
+ * - Replace global sem_lock with mtx_pool locks?
+ * - Add a MAC check_create() hook for creating new named semaphores.
+ */
+
+#ifndef SEM_MAX
+#define SEM_MAX 30
+#endif
+
+#ifdef SEM_DEBUG
+#define DP(x) printf x
+#else
+#define DP(x)
+#endif
+
+struct ksem_mapping {
+ char *km_path;
+ Fnv32_t km_fnv;
+ struct ksem *km_ksem;
+ LIST_ENTRY(ksem_mapping) km_link;
+};
+
+static MALLOC_DEFINE(M_KSEM, "ksem", "semaphore file descriptor");
+static LIST_HEAD(, ksem_mapping) *ksem_dictionary;
+static struct sx ksem_dict_lock;
+static struct mtx ksem_count_lock;
+static struct mtx sem_lock;
+static u_long ksem_hash;
+static int ksem_dead;
+
+#define KSEM_HASH(fnv) (&ksem_dictionary[(fnv) & ksem_hash])
+
+static int nsems = 0;
+SYSCTL_DECL(_p1003_1b);
+SYSCTL_INT(_p1003_1b, OID_AUTO, nsems, CTLFLAG_RD, &nsems, 0,
+ "Number of active kernel POSIX semaphores");
+
+static int kern_sem_wait(struct thread *td, semid_t id, int tryflag,
+ struct timespec *abstime);
+static int ksem_access(struct ksem *ks, struct ucred *ucred);
+static struct ksem *ksem_alloc(struct ucred *ucred, mode_t mode,
+ unsigned int value);
+static int ksem_create(struct thread *td, const char *path,
+ semid_t *semidp, mode_t mode, unsigned int value,
+ int flags, int compat32);
+static void ksem_drop(struct ksem *ks);
+static int ksem_get(struct thread *td, semid_t id, cap_rights_t rights,
+ struct file **fpp);
+static struct ksem *ksem_hold(struct ksem *ks);
+static void ksem_insert(char *path, Fnv32_t fnv, struct ksem *ks);
+static struct ksem *ksem_lookup(char *path, Fnv32_t fnv);
+static void ksem_module_destroy(void);
+static int ksem_module_init(void);
+static int ksem_remove(char *path, Fnv32_t fnv, struct ucred *ucred);
+static int sem_modload(struct module *module, int cmd, void *arg);
+
+static fo_rdwr_t ksem_read;
+static fo_rdwr_t ksem_write;
+static fo_truncate_t ksem_truncate;
+static fo_ioctl_t ksem_ioctl;
+static fo_poll_t ksem_poll;
+static fo_kqfilter_t ksem_kqfilter;
+static fo_stat_t ksem_stat;
+static fo_close_t ksem_closef;
+static fo_chmod_t ksem_chmod;
+static fo_chown_t ksem_chown;
+
+/* File descriptor operations. */
+static struct fileops ksem_ops = {
+ .fo_read = ksem_read,
+ .fo_write = ksem_write,
+ .fo_truncate = ksem_truncate,
+ .fo_ioctl = ksem_ioctl,
+ .fo_poll = ksem_poll,
+ .fo_kqfilter = ksem_kqfilter,
+ .fo_stat = ksem_stat,
+ .fo_close = ksem_closef,
+ .fo_chmod = ksem_chmod,
+ .fo_chown = ksem_chown,
+ .fo_flags = DFLAG_PASSABLE
+};
+
+FEATURE(posix_sem, "POSIX semaphores");
+
+static int
+ksem_read(struct file *fp, struct uio *uio, struct ucred *active_cred,
+ int flags, struct thread *td)
+{
+
+ return (EOPNOTSUPP);
+}
+
+static int
+ksem_write(struct file *fp, struct uio *uio, struct ucred *active_cred,
+ int flags, struct thread *td)
+{
+
+ return (EOPNOTSUPP);
+}
+
+static int
+ksem_truncate(struct file *fp, off_t length, struct ucred *active_cred,
+ struct thread *td)
+{
+
+ return (EINVAL);
+}
+
+static int
+ksem_ioctl(struct file *fp, u_long com, void *data,
+ struct ucred *active_cred, struct thread *td)
+{
+
+ return (EOPNOTSUPP);
+}
+
+static int
+ksem_poll(struct file *fp, int events, struct ucred *active_cred,
+ struct thread *td)
+{
+
+ return (EOPNOTSUPP);
+}
+
+static int
+ksem_kqfilter(struct file *fp, struct knote *kn)
+{
+
+ return (EOPNOTSUPP);
+}
+
+static int
+ksem_stat(struct file *fp, struct stat *sb, struct ucred *active_cred,
+ struct thread *td)
+{
+ struct ksem *ks;
+#ifdef MAC
+ int error;
+#endif
+
+ ks = fp->f_data;
+
+#ifdef MAC
+ error = mac_posixsem_check_stat(active_cred, fp->f_cred, ks);
+ if (error)
+ return (error);
+#endif
+
+ /*
+ * Attempt to return sanish values for fstat() on a semaphore
+ * file descriptor.
+ */
+ bzero(sb, sizeof(*sb));
+
+ mtx_lock(&sem_lock);
+ sb->st_atim = ks->ks_atime;
+ sb->st_ctim = ks->ks_ctime;
+ sb->st_mtim = ks->ks_mtime;
+ sb->st_birthtim = ks->ks_birthtime;
+ sb->st_uid = ks->ks_uid;
+ sb->st_gid = ks->ks_gid;
+ sb->st_mode = S_IFREG | ks->ks_mode; /* XXX */
+ mtx_unlock(&sem_lock);
+
+ return (0);
+}
+
+static int
+ksem_chmod(struct file *fp, mode_t mode, struct ucred *active_cred,
+ struct thread *td)
+{
+ struct ksem *ks;
+ int error;
+
+ error = 0;
+ ks = fp->f_data;
+ mtx_lock(&sem_lock);
+#ifdef MAC
+ error = mac_posixsem_check_setmode(active_cred, ks, mode);
+ if (error != 0)
+ goto out;
+#endif
+ error = vaccess(VREG, ks->ks_mode, ks->ks_uid, ks->ks_gid, VADMIN,
+ active_cred, NULL);
+ if (error != 0)
+ goto out;
+ ks->ks_mode = mode & ACCESSPERMS;
+out:
+ mtx_unlock(&sem_lock);
+ return (error);
+}
+
+static int
+ksem_chown(struct file *fp, uid_t uid, gid_t gid, struct ucred *active_cred,
+ struct thread *td)
+{
+ struct ksem *ks;
+ int error;
+
+ error = 0;
+ ks = fp->f_data;
+ mtx_lock(&sem_lock);
+#ifdef MAC
+ error = mac_posixsem_check_setowner(active_cred, ks, uid, gid);
+ if (error != 0)
+ goto out;
+#endif
+ if (uid == (uid_t)-1)
+ uid = ks->ks_uid;
+ if (gid == (gid_t)-1)
+ gid = ks->ks_gid;
+ if (((uid != ks->ks_uid && uid != active_cred->cr_uid) ||
+ (gid != ks->ks_gid && !groupmember(gid, active_cred))) &&
+ (error = priv_check_cred(active_cred, PRIV_VFS_CHOWN, 0)))
+ goto out;
+ ks->ks_uid = uid;
+ ks->ks_gid = gid;
+out:
+ mtx_unlock(&sem_lock);
+ return (error);
+}
+
+static int
+ksem_closef(struct file *fp, struct thread *td)
+{
+ struct ksem *ks;
+
+ ks = fp->f_data;
+ fp->f_data = NULL;
+ ksem_drop(ks);
+
+ return (0);
+}
+
+/*
+ * ksem object management including creation and reference counting
+ * routines.
+ */
+static struct ksem *
+ksem_alloc(struct ucred *ucred, mode_t mode, unsigned int value)
+{
+ struct ksem *ks;
+
+ mtx_lock(&ksem_count_lock);
+ if (nsems == p31b_getcfg(CTL_P1003_1B_SEM_NSEMS_MAX) || ksem_dead) {
+ mtx_unlock(&ksem_count_lock);
+ return (NULL);
+ }
+ nsems++;
+ mtx_unlock(&ksem_count_lock);
+ ks = malloc(sizeof(*ks), M_KSEM, M_WAITOK | M_ZERO);
+ ks->ks_uid = ucred->cr_uid;
+ ks->ks_gid = ucred->cr_gid;
+ ks->ks_mode = mode;
+ ks->ks_value = value;
+ cv_init(&ks->ks_cv, "ksem");
+ vfs_timestamp(&ks->ks_birthtime);
+ ks->ks_atime = ks->ks_mtime = ks->ks_ctime = ks->ks_birthtime;
+ refcount_init(&ks->ks_ref, 1);
+#ifdef MAC
+ mac_posixsem_init(ks);
+ mac_posixsem_create(ucred, ks);
+#endif
+
+ return (ks);
+}
+
+static struct ksem *
+ksem_hold(struct ksem *ks)
+{
+
+ refcount_acquire(&ks->ks_ref);
+ return (ks);
+}
+
+static void
+ksem_drop(struct ksem *ks)
+{
+
+ if (refcount_release(&ks->ks_ref)) {
+#ifdef MAC
+ mac_posixsem_destroy(ks);
+#endif
+ cv_destroy(&ks->ks_cv);
+ free(ks, M_KSEM);
+ mtx_lock(&ksem_count_lock);
+ nsems--;
+ mtx_unlock(&ksem_count_lock);
+ }
+}
+
+/*
+ * Determine if the credentials have sufficient permissions for read
+ * and write access.
+ */
+static int
+ksem_access(struct ksem *ks, struct ucred *ucred)
+{
+ int error;
+
+ error = vaccess(VREG, ks->ks_mode, ks->ks_uid, ks->ks_gid,
+ VREAD | VWRITE, ucred, NULL);
+ if (error)
+ error = priv_check_cred(ucred, PRIV_SEM_WRITE, 0);
+ return (error);
+}
+
+/*
+ * Dictionary management. We maintain an in-kernel dictionary to map
+ * paths to semaphore objects. We use the FNV hash on the path to
+ * store the mappings in a hash table.
+ */
+static struct ksem *
+ksem_lookup(char *path, Fnv32_t fnv)
+{
+ struct ksem_mapping *map;
+
+ LIST_FOREACH(map, KSEM_HASH(fnv), km_link) {
+ if (map->km_fnv != fnv)
+ continue;
+ if (strcmp(map->km_path, path) == 0)
+ return (map->km_ksem);
+ }
+
+ return (NULL);
+}
+
+static void
+ksem_insert(char *path, Fnv32_t fnv, struct ksem *ks)
+{
+ struct ksem_mapping *map;
+
+ map = malloc(sizeof(struct ksem_mapping), M_KSEM, M_WAITOK);
+ map->km_path = path;
+ map->km_fnv = fnv;
+ map->km_ksem = ksem_hold(ks);
+ LIST_INSERT_HEAD(KSEM_HASH(fnv), map, km_link);
+}
+
+static int
+ksem_remove(char *path, Fnv32_t fnv, struct ucred *ucred)
+{
+ struct ksem_mapping *map;
+ int error;
+
+ LIST_FOREACH(map, KSEM_HASH(fnv), km_link) {
+ if (map->km_fnv != fnv)
+ continue;
+ if (strcmp(map->km_path, path) == 0) {
+#ifdef MAC
+ error = mac_posixsem_check_unlink(ucred, map->km_ksem);
+ if (error)
+ return (error);
+#endif
+ error = ksem_access(map->km_ksem, ucred);
+ if (error)
+ return (error);
+ LIST_REMOVE(map, km_link);
+ ksem_drop(map->km_ksem);
+ free(map->km_path, M_KSEM);
+ free(map, M_KSEM);
+ return (0);
+ }
+ }
+
+ return (ENOENT);
+}
+
+static int
+ksem_create_copyout_semid(struct thread *td, semid_t *semidp, int fd,
+ int compat32)
+{
+ semid_t semid;
+#ifdef COMPAT_FREEBSD32
+ int32_t semid32;
+#endif
+ void *ptr;
+ size_t ptrs;
+
+#ifdef COMPAT_FREEBSD32
+ if (compat32) {
+ semid32 = fd;
+ ptr = &semid32;
+ ptrs = sizeof(semid32);
+ } else {
+#endif
+ semid = fd;
+ ptr = &semid;
+ ptrs = sizeof(semid);
+ compat32 = 0; /* silence gcc */
+#ifdef COMPAT_FREEBSD32
+ }
+#endif
+
+ return (copyout(ptr, semidp, ptrs));
+}
+
+/* Other helper routines. */
+static int
+ksem_create(struct thread *td, const char *name, semid_t *semidp, mode_t mode,
+ unsigned int value, int flags, int compat32)
+{
+ struct filedesc *fdp;
+ struct ksem *ks;
+ struct file *fp;
+ char *path;
+ Fnv32_t fnv;
+ int error, fd;
+
+ if (value > SEM_VALUE_MAX)
+ return (EINVAL);
+
+ fdp = td->td_proc->p_fd;
+ mode = (mode & ~fdp->fd_cmask) & ACCESSPERMS;
+ error = falloc(td, &fp, &fd, 0);
+ if (error) {
+ if (name == NULL)
+ error = ENOSPC;
+ return (error);
+ }
+
+ /*
+ * Go ahead and copyout the file descriptor now. This is a bit
+ * premature, but it is a lot easier to handle errors as opposed
+ * to later when we've possibly created a new semaphore, etc.
+ */
+ error = ksem_create_copyout_semid(td, semidp, fd, compat32);
+ if (error) {
+ fdclose(fdp, fp, fd, td);
+ fdrop(fp, td);
+ return (error);
+ }
+
+ if (name == NULL) {
+ /* Create an anonymous semaphore. */
+ ks = ksem_alloc(td->td_ucred, mode, value);
+ if (ks == NULL)
+ error = ENOSPC;
+ else
+ ks->ks_flags |= KS_ANONYMOUS;
+ } else {
+ path = malloc(MAXPATHLEN, M_KSEM, M_WAITOK);
+ error = copyinstr(name, path, MAXPATHLEN, NULL);
+
+ /* Require paths to start with a '/' character. */
+ if (error == 0 && path[0] != '/')
+ error = EINVAL;
+ if (error) {
+ fdclose(fdp, fp, fd, td);
+ fdrop(fp, td);
+ free(path, M_KSEM);
+ return (error);
+ }
+
+ fnv = fnv_32_str(path, FNV1_32_INIT);
+ sx_xlock(&ksem_dict_lock);
+ ks = ksem_lookup(path, fnv);
+ if (ks == NULL) {
+ /* Object does not exist, create it if requested. */
+ if (flags & O_CREAT) {
+ ks = ksem_alloc(td->td_ucred, mode, value);
+ if (ks == NULL)
+ error = ENFILE;
+ else {
+ ksem_insert(path, fnv, ks);
+ path = NULL;
+ }
+ } else
+ error = ENOENT;
+ } else {
+ /*
+ * Object already exists, obtain a new
+ * reference if requested and permitted.
+ */
+ if ((flags & (O_CREAT | O_EXCL)) ==
+ (O_CREAT | O_EXCL))
+ error = EEXIST;
+ else {
+#ifdef MAC
+ error = mac_posixsem_check_open(td->td_ucred,
+ ks);
+ if (error == 0)
+#endif
+ error = ksem_access(ks, td->td_ucred);
+ }
+ if (error == 0)
+ ksem_hold(ks);
+#ifdef INVARIANTS
+ else
+ ks = NULL;
+#endif
+ }
+ sx_xunlock(&ksem_dict_lock);
+ if (path)
+ free(path, M_KSEM);
+ }
+
+ if (error) {
+ KASSERT(ks == NULL, ("ksem_create error with a ksem"));
+ fdclose(fdp, fp, fd, td);
+ fdrop(fp, td);
+ return (error);
+ }
+ KASSERT(ks != NULL, ("ksem_create w/o a ksem"));
+
+ finit(fp, FREAD | FWRITE, DTYPE_SEM, ks, &ksem_ops);
+
+ FILEDESC_XLOCK(fdp);
+ if (fdp->fd_ofiles[fd] == fp)
+ fdp->fd_ofileflags[fd] |= UF_EXCLOSE;
+ FILEDESC_XUNLOCK(fdp);
+ fdrop(fp, td);
+
+ return (0);
+}
+
+static int
+ksem_get(struct thread *td, semid_t id, cap_rights_t rights, struct file **fpp)
+{
+ struct ksem *ks;
+ struct file *fp;
+ int error;
+
+ error = fget(td, id, rights, &fp);
+ if (error)
+ return (EINVAL);
+ if (fp->f_type != DTYPE_SEM) {
+ fdrop(fp, td);
+ return (EINVAL);
+ }
+ ks = fp->f_data;
+ if (ks->ks_flags & KS_DEAD) {
+ fdrop(fp, td);
+ return (EINVAL);
+ }
+ *fpp = fp;
+ return (0);
+}
+
+/* System calls. */
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_init_args {
+ unsigned int value;
+ semid_t *idp;
+};
+#endif
+int
+sys_ksem_init(struct thread *td, struct ksem_init_args *uap)
+{
+
+ return (ksem_create(td, NULL, uap->idp, S_IRWXU | S_IRWXG, uap->value,
+ 0, 0));
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_open_args {
+ char *name;
+ int oflag;
+ mode_t mode;
+ unsigned int value;
+ semid_t *idp;
+};
+#endif
+int
+sys_ksem_open(struct thread *td, struct ksem_open_args *uap)
+{
+
+ DP((">>> ksem_open start, pid=%d\n", (int)td->td_proc->p_pid));
+
+ if ((uap->oflag & ~(O_CREAT | O_EXCL)) != 0)
+ return (EINVAL);
+ return (ksem_create(td, uap->name, uap->idp, uap->mode, uap->value,
+ uap->oflag, 0));
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_unlink_args {
+ char *name;
+};
+#endif
+int
+sys_ksem_unlink(struct thread *td, struct ksem_unlink_args *uap)
+{
+ char *path;
+ Fnv32_t fnv;
+ int error;
+
+ path = malloc(MAXPATHLEN, M_TEMP, M_WAITOK);
+ error = copyinstr(uap->name, path, MAXPATHLEN, NULL);
+ if (error) {
+ free(path, M_TEMP);
+ return (error);
+ }
+
+ fnv = fnv_32_str(path, FNV1_32_INIT);
+ sx_xlock(&ksem_dict_lock);
+ error = ksem_remove(path, fnv, td->td_ucred);
+ sx_xunlock(&ksem_dict_lock);
+ free(path, M_TEMP);
+
+ return (error);
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_close_args {
+ semid_t id;
+};
+#endif
+int
+sys_ksem_close(struct thread *td, struct ksem_close_args *uap)
+{
+ struct ksem *ks;
+ struct file *fp;
+ int error;
+
+ /* No capability rights required to close a semaphore. */
+ error = ksem_get(td, uap->id, 0, &fp);
+ if (error)
+ return (error);
+ ks = fp->f_data;
+ if (ks->ks_flags & KS_ANONYMOUS) {
+ fdrop(fp, td);
+ return (EINVAL);
+ }
+ error = kern_close(td, uap->id);
+ fdrop(fp, td);
+ return (error);
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_post_args {
+ semid_t id;
+};
+#endif
+int
+sys_ksem_post(struct thread *td, struct ksem_post_args *uap)
+{
+ struct file *fp;
+ struct ksem *ks;
+ int error;
+
+ error = ksem_get(td, uap->id, CAP_SEM_POST, &fp);
+ if (error)
+ return (error);
+ ks = fp->f_data;
+
+ mtx_lock(&sem_lock);
+#ifdef MAC
+ error = mac_posixsem_check_post(td->td_ucred, fp->f_cred, ks);
+ if (error)
+ goto err;
+#endif
+ if (ks->ks_value == SEM_VALUE_MAX) {
+ error = EOVERFLOW;
+ goto err;
+ }
+ ++ks->ks_value;
+ if (ks->ks_waiters > 0)
+ cv_signal(&ks->ks_cv);
+ error = 0;
+ vfs_timestamp(&ks->ks_ctime);
+err:
+ mtx_unlock(&sem_lock);
+ fdrop(fp, td);
+ return (error);
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_wait_args {
+ semid_t id;
+};
+#endif
+int
+sys_ksem_wait(struct thread *td, struct ksem_wait_args *uap)
+{
+
+ return (kern_sem_wait(td, uap->id, 0, NULL));
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_timedwait_args {
+ semid_t id;
+ const struct timespec *abstime;
+};
+#endif
+int
+sys_ksem_timedwait(struct thread *td, struct ksem_timedwait_args *uap)
+{
+ struct timespec abstime;
+ struct timespec *ts;
+ int error;
+
+ /*
+ * We allow a null timespec (wait forever).
+ */
+ if (uap->abstime == NULL)
+ ts = NULL;
+ else {
+ error = copyin(uap->abstime, &abstime, sizeof(abstime));
+ if (error != 0)
+ return (error);
+ if (abstime.tv_nsec >= 1000000000 || abstime.tv_nsec < 0)
+ return (EINVAL);
+ ts = &abstime;
+ }
+ return (kern_sem_wait(td, uap->id, 0, ts));
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_trywait_args {
+ semid_t id;
+};
+#endif
+int
+sys_ksem_trywait(struct thread *td, struct ksem_trywait_args *uap)
+{
+
+ return (kern_sem_wait(td, uap->id, 1, NULL));
+}
+
+static int
+kern_sem_wait(struct thread *td, semid_t id, int tryflag,
+ struct timespec *abstime)
+{
+ struct timespec ts1, ts2;
+ struct timeval tv;
+ struct file *fp;
+ struct ksem *ks;
+ int error;
+
+ DP((">>> kern_sem_wait entered! pid=%d\n", (int)td->td_proc->p_pid));
+ error = ksem_get(td, id, CAP_SEM_WAIT, &fp);
+ if (error)
+ return (error);
+ ks = fp->f_data;
+ mtx_lock(&sem_lock);
+ DP((">>> kern_sem_wait critical section entered! pid=%d\n",
+ (int)td->td_proc->p_pid));
+#ifdef MAC
+ error = mac_posixsem_check_wait(td->td_ucred, fp->f_cred, ks);
+ if (error) {
+ DP(("kern_sem_wait mac failed\n"));
+ goto err;
+ }
+#endif
+ DP(("kern_sem_wait value = %d, tryflag %d\n", ks->ks_value, tryflag));
+ vfs_timestamp(&ks->ks_atime);
+ while (ks->ks_value == 0) {
+ ks->ks_waiters++;
+ if (tryflag != 0)
+ error = EAGAIN;
+ else if (abstime == NULL)
+ error = cv_wait_sig(&ks->ks_cv, &sem_lock);
+ else {
+ for (;;) {
+ ts1 = *abstime;
+ getnanotime(&ts2);
+ timespecsub(&ts1, &ts2);
+ TIMESPEC_TO_TIMEVAL(&tv, &ts1);
+ if (tv.tv_sec < 0) {
+ error = ETIMEDOUT;
+ break;
+ }
+ error = cv_timedwait_sig(&ks->ks_cv,
+ &sem_lock, tvtohz(&tv));
+ if (error != EWOULDBLOCK)
+ break;
+ }
+ }
+ ks->ks_waiters--;
+ if (error)
+ goto err;
+ }
+ ks->ks_value--;
+ DP(("kern_sem_wait value post-decrement = %d\n", ks->ks_value));
+ error = 0;
+err:
+ mtx_unlock(&sem_lock);
+ fdrop(fp, td);
+ DP(("<<< kern_sem_wait leaving, pid=%d, error = %d\n",
+ (int)td->td_proc->p_pid, error));
+ return (error);
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_getvalue_args {
+ semid_t id;
+ int *val;
+};
+#endif
+int
+sys_ksem_getvalue(struct thread *td, struct ksem_getvalue_args *uap)
+{
+ struct file *fp;
+ struct ksem *ks;
+ int error, val;
+
+ error = ksem_get(td, uap->id, CAP_SEM_GETVALUE, &fp);
+ if (error)
+ return (error);
+ ks = fp->f_data;
+
+ mtx_lock(&sem_lock);
+#ifdef MAC
+ error = mac_posixsem_check_getvalue(td->td_ucred, fp->f_cred, ks);
+ if (error) {
+ mtx_unlock(&sem_lock);
+ fdrop(fp, td);
+ return (error);
+ }
+#endif
+ val = ks->ks_value;
+ vfs_timestamp(&ks->ks_atime);
+ mtx_unlock(&sem_lock);
+ fdrop(fp, td);
+ error = copyout(&val, uap->val, sizeof(val));
+ return (error);
+}
+
+#ifndef _SYS_SYSPROTO_H_
+struct ksem_destroy_args {
+ semid_t id;
+};
+#endif
+int
+sys_ksem_destroy(struct thread *td, struct ksem_destroy_args *uap)
+{
+ struct file *fp;
+ struct ksem *ks;
+ int error;
+
+ /* No capability rights required to close a semaphore. */
+ error = ksem_get(td, uap->id, 0, &fp);
+ if (error)
+ return (error);
+ ks = fp->f_data;
+ if (!(ks->ks_flags & KS_ANONYMOUS)) {
+ fdrop(fp, td);
+ return (EINVAL);
+ }
+ mtx_lock(&sem_lock);
+ if (ks->ks_waiters != 0) {
+ mtx_unlock(&sem_lock);
+ error = EBUSY;
+ goto err;
+ }
+ ks->ks_flags |= KS_DEAD;
+ mtx_unlock(&sem_lock);
+
+ error = kern_close(td, uap->id);
+err:
+ fdrop(fp, td);
+ return (error);
+}
+
+static struct syscall_helper_data ksem_syscalls[] = {
+ SYSCALL_INIT_HELPER(ksem_init),
+ SYSCALL_INIT_HELPER(ksem_open),
+ SYSCALL_INIT_HELPER(ksem_unlink),
+ SYSCALL_INIT_HELPER(ksem_close),
+ SYSCALL_INIT_HELPER(ksem_post),
+ SYSCALL_INIT_HELPER(ksem_wait),
+ SYSCALL_INIT_HELPER(ksem_timedwait),
+ SYSCALL_INIT_HELPER(ksem_trywait),
+ SYSCALL_INIT_HELPER(ksem_getvalue),
+ SYSCALL_INIT_HELPER(ksem_destroy),
+ SYSCALL_INIT_LAST
+};
+
+#ifdef COMPAT_FREEBSD32
+#include <compat/freebsd32/freebsd32.h>
+#include <compat/freebsd32/freebsd32_proto.h>
+#include <compat/freebsd32/freebsd32_signal.h>
+#include <compat/freebsd32/freebsd32_syscall.h>
+#include <compat/freebsd32/freebsd32_util.h>
+
+int
+freebsd32_ksem_init(struct thread *td, struct freebsd32_ksem_init_args *uap)
+{
+
+ return (ksem_create(td, NULL, uap->idp, S_IRWXU | S_IRWXG, uap->value,
+ 0, 1));
+}
+
+int
+freebsd32_ksem_open(struct thread *td, struct freebsd32_ksem_open_args *uap)
+{
+
+ if ((uap->oflag & ~(O_CREAT | O_EXCL)) != 0)
+ return (EINVAL);
+ return (ksem_create(td, uap->name, uap->idp, uap->mode, uap->value,
+ uap->oflag, 1));
+}
+
+int
+freebsd32_ksem_timedwait(struct thread *td,
+ struct freebsd32_ksem_timedwait_args *uap)
+{
+ struct timespec32 abstime32;
+ struct timespec *ts, abstime;
+ int error;
+
+ /*
+ * We allow a null timespec (wait forever).
+ */
+ if (uap->abstime == NULL)
+ ts = NULL;
+ else {
+ error = copyin(uap->abstime, &abstime32, sizeof(abstime32));
+ if (error != 0)
+ return (error);
+ CP(abstime32, abstime, tv_sec);
+ CP(abstime32, abstime, tv_nsec);
+ if (abstime.tv_nsec >= 1000000000 || abstime.tv_nsec < 0)
+ return (EINVAL);
+ ts = &abstime;
+ }
+ return (kern_sem_wait(td, uap->id, 0, ts));
+}
+
+static struct syscall_helper_data ksem32_syscalls[] = {
+ SYSCALL32_INIT_HELPER(freebsd32_ksem_init),
+ SYSCALL32_INIT_HELPER(freebsd32_ksem_open),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_unlink),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_close),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_post),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_wait),
+ SYSCALL32_INIT_HELPER(freebsd32_ksem_timedwait),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_trywait),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_getvalue),
+ SYSCALL32_INIT_HELPER_COMPAT(ksem_destroy),
+ SYSCALL_INIT_LAST
+};
+#endif
+
+static int
+ksem_module_init(void)
+{
+ int error;
+
+ mtx_init(&sem_lock, "sem", NULL, MTX_DEF);
+ mtx_init(&ksem_count_lock, "ksem count", NULL, MTX_DEF);
+ sx_init(&ksem_dict_lock, "ksem dictionary");
+ ksem_dictionary = hashinit(1024, M_KSEM, &ksem_hash);
+ p31b_setcfg(CTL_P1003_1B_SEMAPHORES, 200112L);
+ p31b_setcfg(CTL_P1003_1B_SEM_NSEMS_MAX, SEM_MAX);
+ p31b_setcfg(CTL_P1003_1B_SEM_VALUE_MAX, SEM_VALUE_MAX);
+
+ error = syscall_helper_register(ksem_syscalls);
+ if (error)
+ return (error);
+#ifdef COMPAT_FREEBSD32
+ error = syscall32_helper_register(ksem32_syscalls);
+ if (error)
+ return (error);
+#endif
+ return (0);
+}
+
+static void
+ksem_module_destroy(void)
+{
+
+#ifdef COMPAT_FREEBSD32
+ syscall32_helper_unregister(ksem32_syscalls);
+#endif
+ syscall_helper_unregister(ksem_syscalls);
+
+ p31b_setcfg(CTL_P1003_1B_SEMAPHORES, 0);
+ hashdestroy(ksem_dictionary, M_KSEM, ksem_hash);
+ sx_destroy(&ksem_dict_lock);
+ mtx_destroy(&ksem_count_lock);
+ mtx_destroy(&sem_lock);
+ p31b_unsetcfg(CTL_P1003_1B_SEM_VALUE_MAX);
+ p31b_unsetcfg(CTL_P1003_1B_SEM_NSEMS_MAX);
+}
+
+static int
+sem_modload(struct module *module, int cmd, void *arg)
+{
+ int error = 0;
+
+ switch (cmd) {
+ case MOD_LOAD:
+ error = ksem_module_init();
+ if (error)
+ ksem_module_destroy();
+ break;
+
+ case MOD_UNLOAD:
+ mtx_lock(&ksem_count_lock);
+ if (nsems != 0) {
+ error = EOPNOTSUPP;
+ mtx_unlock(&ksem_count_lock);
+ break;
+ }
+ ksem_dead = 1;
+ mtx_unlock(&ksem_count_lock);
+ ksem_module_destroy();
+ break;
+
+ case MOD_SHUTDOWN:
+ break;
+ default:
+ error = EINVAL;
+ break;
+ }
+ return (error);
+}
+
+static moduledata_t sem_mod = {
+ "sem",
+ &sem_modload,
+ NULL
+};
+
+DECLARE_MODULE(sem, sem_mod, SI_SUB_SYSV_SEM, SI_ORDER_FIRST);
+MODULE_VERSION(sem, 1);
OpenPOWER on IntegriCloud