xref: /freebsd/sys/kern/uipc_sem.c (revision 9454b2d8)
19454b2d8SWarner Losh /*-
2efaa6588SAlfred Perlstein  * Copyright (c) 2002 Alfred Perlstein <alfred@FreeBSD.org>
3efaa6588SAlfred Perlstein  * All rights reserved.
4efaa6588SAlfred Perlstein  *
5efaa6588SAlfred Perlstein  * Redistribution and use in source and binary forms, with or without
6efaa6588SAlfred Perlstein  * modification, are permitted provided that the following conditions
7efaa6588SAlfred Perlstein  * are met:
8efaa6588SAlfred Perlstein  * 1. Redistributions of source code must retain the above copyright
9efaa6588SAlfred Perlstein  *    notice, this list of conditions and the following disclaimer.
10efaa6588SAlfred Perlstein  * 2. Redistributions in binary form must reproduce the above copyright
11efaa6588SAlfred Perlstein  *    notice, this list of conditions and the following disclaimer in the
12efaa6588SAlfred Perlstein  *    documentation and/or other materials provided with the distribution.
13efaa6588SAlfred Perlstein  *
14efaa6588SAlfred Perlstein  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15efaa6588SAlfred Perlstein  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16efaa6588SAlfred Perlstein  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17efaa6588SAlfred Perlstein  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18efaa6588SAlfred Perlstein  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19efaa6588SAlfred Perlstein  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20efaa6588SAlfred Perlstein  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21efaa6588SAlfred Perlstein  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22efaa6588SAlfred Perlstein  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23efaa6588SAlfred Perlstein  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24efaa6588SAlfred Perlstein  * SUCH DAMAGE.
25efaa6588SAlfred Perlstein  */
26efaa6588SAlfred Perlstein 
27677b542eSDavid E. O'Brien #include <sys/cdefs.h>
28677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
29677b542eSDavid E. O'Brien 
30efaa6588SAlfred Perlstein #include "opt_posix.h"
31efaa6588SAlfred Perlstein 
32efaa6588SAlfred Perlstein #include <sys/param.h>
33efaa6588SAlfred Perlstein #include <sys/systm.h>
34efaa6588SAlfred Perlstein #include <sys/sysproto.h>
3575b8b3b2SJohn Baldwin #include <sys/eventhandler.h>
36efaa6588SAlfred Perlstein #include <sys/kernel.h>
37efaa6588SAlfred Perlstein #include <sys/proc.h>
38efaa6588SAlfred Perlstein #include <sys/lock.h>
39efaa6588SAlfred Perlstein #include <sys/mutex.h>
4077409fe1SPoul-Henning Kamp #include <sys/module.h>
41efaa6588SAlfred Perlstein #include <sys/condvar.h>
42efaa6588SAlfred Perlstein #include <sys/sem.h>
43efaa6588SAlfred Perlstein #include <sys/uio.h>
44efaa6588SAlfred Perlstein #include <sys/syscall.h>
45efaa6588SAlfred Perlstein #include <sys/stat.h>
46efaa6588SAlfred Perlstein #include <sys/sysent.h>
47efaa6588SAlfred Perlstein #include <sys/sysctl.h>
48aae94fbbSDaniel Eischen #include <sys/time.h>
49efaa6588SAlfred Perlstein #include <sys/malloc.h>
50efaa6588SAlfred Perlstein #include <sys/fcntl.h>
51efaa6588SAlfred Perlstein 
52efaa6588SAlfred Perlstein #include <posix4/posix4.h>
53efaa6588SAlfred Perlstein #include <posix4/semaphore.h>
54efaa6588SAlfred Perlstein #include <posix4/_semaphore.h>
55efaa6588SAlfred Perlstein 
56efaa6588SAlfred Perlstein static struct ksem *sem_lookup_byname(const char *name);
57efaa6588SAlfred Perlstein static int sem_create(struct thread *td, const char *name,
58efaa6588SAlfred Perlstein     struct ksem **ksret, mode_t mode, unsigned int value);
59efaa6588SAlfred Perlstein static void sem_free(struct ksem *ksnew);
60b2546660SJohn Baldwin static int sem_perm(struct thread *td, struct ksem *ks);
61efaa6588SAlfred Perlstein static void sem_enter(struct proc *p, struct ksem *ks);
62efaa6588SAlfred Perlstein static int sem_leave(struct proc *p, struct ksem *ks);
6375b8b3b2SJohn Baldwin static void sem_exithook(void *arg, struct proc *p);
64b2546660SJohn Baldwin static int sem_hasopen(struct thread *td, struct ksem *ks);
65efaa6588SAlfred Perlstein 
66efaa6588SAlfred Perlstein static int kern_sem_close(struct thread *td, semid_t id);
67efaa6588SAlfred Perlstein static int kern_sem_post(struct thread *td, semid_t id);
68aae94fbbSDaniel Eischen static int kern_sem_wait(struct thread *td, semid_t id, int tryflag,
69aae94fbbSDaniel Eischen     struct timespec *abstime);
70efaa6588SAlfred Perlstein static int kern_sem_init(struct thread *td, int dir, unsigned int value,
71efaa6588SAlfred Perlstein     semid_t *idp);
72efaa6588SAlfred Perlstein static int kern_sem_open(struct thread *td, int dir, const char *name,
73efaa6588SAlfred Perlstein     int oflag, mode_t mode, unsigned int value, semid_t *idp);
74efaa6588SAlfred Perlstein static int kern_sem_unlink(struct thread *td, const char *name);
75efaa6588SAlfred Perlstein 
76efaa6588SAlfred Perlstein #ifndef SEM_MAX
77efaa6588SAlfred Perlstein #define SEM_MAX	30
78efaa6588SAlfred Perlstein #endif
79efaa6588SAlfred Perlstein 
80efaa6588SAlfred Perlstein #define SEM_MAX_NAMELEN	14
81efaa6588SAlfred Perlstein 
82efaa6588SAlfred Perlstein #define SEM_TO_ID(x)	((intptr_t)(x))
83efaa6588SAlfred Perlstein #define ID_TO_SEM(x)	id_to_sem(x)
84efaa6588SAlfred Perlstein 
85efaa6588SAlfred Perlstein struct kuser {
86efaa6588SAlfred Perlstein 	pid_t ku_pid;
87efaa6588SAlfred Perlstein 	LIST_ENTRY(kuser) ku_next;
88efaa6588SAlfred Perlstein };
89efaa6588SAlfred Perlstein 
90efaa6588SAlfred Perlstein struct ksem {
91efaa6588SAlfred Perlstein 	LIST_ENTRY(ksem) ks_entry;	/* global list entry */
92efaa6588SAlfred Perlstein 	int ks_onlist;			/* boolean if on a list (ks_entry) */
93efaa6588SAlfred Perlstein 	char *ks_name;			/* if named, this is the name */
94efaa6588SAlfred Perlstein 	int ks_ref;			/* number of references */
95efaa6588SAlfred Perlstein 	mode_t ks_mode;			/* protection bits */
96efaa6588SAlfred Perlstein 	uid_t ks_uid;			/* creator uid */
97efaa6588SAlfred Perlstein 	gid_t ks_gid;			/* creator gid */
98efaa6588SAlfred Perlstein 	unsigned int ks_value;		/* current value */
99efaa6588SAlfred Perlstein 	struct cv ks_cv;		/* waiters sleep here */
100efaa6588SAlfred Perlstein 	int ks_waiters;			/* number of waiters */
101efaa6588SAlfred Perlstein 	LIST_HEAD(, kuser) ks_users;	/* pids using this sem */
102efaa6588SAlfred Perlstein };
103efaa6588SAlfred Perlstein 
104efaa6588SAlfred Perlstein /*
105efaa6588SAlfred Perlstein  * available semaphores go here, this includes sem_init and any semaphores
106efaa6588SAlfred Perlstein  * created via sem_open that have not yet been unlinked.
107efaa6588SAlfred Perlstein  */
108efaa6588SAlfred Perlstein LIST_HEAD(, ksem) ksem_head = LIST_HEAD_INITIALIZER(&ksem_head);
109efaa6588SAlfred Perlstein /*
110efaa6588SAlfred Perlstein  * semaphores still in use but have been sem_unlink()'d go here.
111efaa6588SAlfred Perlstein  */
112efaa6588SAlfred Perlstein LIST_HEAD(, ksem) ksem_deadhead = LIST_HEAD_INITIALIZER(&ksem_deadhead);
113efaa6588SAlfred Perlstein 
114efaa6588SAlfred Perlstein static struct mtx sem_lock;
115efaa6588SAlfred Perlstein static MALLOC_DEFINE(M_SEM, "sems", "semaphore data");
116efaa6588SAlfred Perlstein 
117efaa6588SAlfred Perlstein static int nsems = 0;
118efaa6588SAlfred Perlstein SYSCTL_DECL(_p1003_1b);
119efaa6588SAlfred Perlstein SYSCTL_INT(_p1003_1b, OID_AUTO, nsems, CTLFLAG_RD, &nsems, 0, "");
120efaa6588SAlfred Perlstein 
12175b8b3b2SJohn Baldwin static eventhandler_tag sem_exit_tag, sem_exec_tag;
12275b8b3b2SJohn Baldwin 
123c814aa3fSAlfred Perlstein #ifdef SEM_DEBUG
124c814aa3fSAlfred Perlstein #define DP(x)	printf x
125c814aa3fSAlfred Perlstein #else
126c814aa3fSAlfred Perlstein #define DP(x)
127c814aa3fSAlfred Perlstein #endif
128c814aa3fSAlfred Perlstein 
129efaa6588SAlfred Perlstein static __inline
130efaa6588SAlfred Perlstein void
131efaa6588SAlfred Perlstein sem_ref(struct ksem *ks)
132efaa6588SAlfred Perlstein {
133efaa6588SAlfred Perlstein 
1340fddf92dSRobert Watson 	mtx_assert(&sem_lock, MA_OWNED);
135efaa6588SAlfred Perlstein 	ks->ks_ref++;
136c814aa3fSAlfred Perlstein 	DP(("sem_ref: ks = %p, ref = %d\n", ks, ks->ks_ref));
137efaa6588SAlfred Perlstein }
138efaa6588SAlfred Perlstein 
139efaa6588SAlfred Perlstein static __inline
140efaa6588SAlfred Perlstein void
141efaa6588SAlfred Perlstein sem_rel(struct ksem *ks)
142efaa6588SAlfred Perlstein {
143efaa6588SAlfred Perlstein 
1440fddf92dSRobert Watson 	mtx_assert(&sem_lock, MA_OWNED);
145c814aa3fSAlfred Perlstein 	DP(("sem_rel: ks = %p, ref = %d\n", ks, ks->ks_ref - 1));
146efaa6588SAlfred Perlstein 	if (--ks->ks_ref == 0)
147efaa6588SAlfred Perlstein 		sem_free(ks);
148efaa6588SAlfred Perlstein }
149efaa6588SAlfred Perlstein 
150efaa6588SAlfred Perlstein static __inline struct ksem *id_to_sem(semid_t id);
151efaa6588SAlfred Perlstein 
152efaa6588SAlfred Perlstein static __inline
153efaa6588SAlfred Perlstein struct ksem *
154efaa6588SAlfred Perlstein id_to_sem(id)
155efaa6588SAlfred Perlstein 	semid_t id;
156efaa6588SAlfred Perlstein {
157efaa6588SAlfred Perlstein 	struct ksem *ks;
158efaa6588SAlfred Perlstein 
159c814aa3fSAlfred Perlstein 	DP(("id_to_sem: id = %0x,%p\n", id, (struct ksem *)id));
160efaa6588SAlfred Perlstein 	LIST_FOREACH(ks, &ksem_head, ks_entry) {
161c814aa3fSAlfred Perlstein 		DP(("id_to_sem: ks = %p\n", ks));
162efaa6588SAlfred Perlstein 		if (ks == (struct ksem *)id)
163efaa6588SAlfred Perlstein 			return (ks);
164efaa6588SAlfred Perlstein 	}
165efaa6588SAlfred Perlstein 	return (NULL);
166efaa6588SAlfred Perlstein }
167efaa6588SAlfred Perlstein 
168c3053131SPoul-Henning Kamp static struct ksem *
169efaa6588SAlfred Perlstein sem_lookup_byname(name)
170efaa6588SAlfred Perlstein 	const char *name;
171efaa6588SAlfred Perlstein {
172efaa6588SAlfred Perlstein 	struct ksem *ks;
173efaa6588SAlfred Perlstein 
174efaa6588SAlfred Perlstein 	LIST_FOREACH(ks, &ksem_head, ks_entry)
175efaa6588SAlfred Perlstein 		if (ks->ks_name != NULL && strcmp(ks->ks_name, name) == 0)
176efaa6588SAlfred Perlstein 			return (ks);
177efaa6588SAlfred Perlstein 	return (NULL);
178efaa6588SAlfred Perlstein }
179efaa6588SAlfred Perlstein 
180c3053131SPoul-Henning Kamp static int
181efaa6588SAlfred Perlstein sem_create(td, name, ksret, mode, value)
182efaa6588SAlfred Perlstein 	struct thread *td;
183efaa6588SAlfred Perlstein 	const char *name;
184efaa6588SAlfred Perlstein 	struct ksem **ksret;
185efaa6588SAlfred Perlstein 	mode_t mode;
186efaa6588SAlfred Perlstein 	unsigned int value;
187efaa6588SAlfred Perlstein {
188efaa6588SAlfred Perlstein 	struct ksem *ret;
189efaa6588SAlfred Perlstein 	struct proc *p;
190efaa6588SAlfred Perlstein 	struct ucred *uc;
191efaa6588SAlfred Perlstein 	size_t len;
192efaa6588SAlfred Perlstein 	int error;
193efaa6588SAlfred Perlstein 
194c814aa3fSAlfred Perlstein 	DP(("sem_create\n"));
195efaa6588SAlfred Perlstein 	p = td->td_proc;
196b2546660SJohn Baldwin 	uc = td->td_ucred;
197efaa6588SAlfred Perlstein 	if (value > SEM_VALUE_MAX)
198efaa6588SAlfred Perlstein 		return (EINVAL);
199a163d034SWarner Losh 	ret = malloc(sizeof(*ret), M_SEM, M_WAITOK | M_ZERO);
200efaa6588SAlfred Perlstein 	if (name != NULL) {
201efaa6588SAlfred Perlstein 		len = strlen(name);
202efaa6588SAlfred Perlstein 		if (len > SEM_MAX_NAMELEN) {
203efaa6588SAlfred Perlstein 			free(ret, M_SEM);
204efaa6588SAlfred Perlstein 			return (ENAMETOOLONG);
205efaa6588SAlfred Perlstein 		}
206efaa6588SAlfred Perlstein 		/* name must start with a '/' but not contain one. */
207efaa6588SAlfred Perlstein 		if (*name != '/' || len < 2 || index(name + 1, '/') != NULL) {
208efaa6588SAlfred Perlstein 			free(ret, M_SEM);
209efaa6588SAlfred Perlstein 			return (EINVAL);
210efaa6588SAlfred Perlstein 		}
211a163d034SWarner Losh 		ret->ks_name = malloc(len + 1, M_SEM, M_WAITOK);
212efaa6588SAlfred Perlstein 		strcpy(ret->ks_name, name);
213efaa6588SAlfred Perlstein 	} else {
214efaa6588SAlfred Perlstein 		ret->ks_name = NULL;
215efaa6588SAlfred Perlstein 	}
216efaa6588SAlfred Perlstein 	ret->ks_mode = mode;
217efaa6588SAlfred Perlstein 	ret->ks_value = value;
218efaa6588SAlfred Perlstein 	ret->ks_ref = 1;
219efaa6588SAlfred Perlstein 	ret->ks_waiters = 0;
220efaa6588SAlfred Perlstein 	ret->ks_uid = uc->cr_uid;
221efaa6588SAlfred Perlstein 	ret->ks_gid = uc->cr_gid;
222efaa6588SAlfred Perlstein 	ret->ks_onlist = 0;
223efaa6588SAlfred Perlstein 	cv_init(&ret->ks_cv, "sem");
224efaa6588SAlfred Perlstein 	LIST_INIT(&ret->ks_users);
225efaa6588SAlfred Perlstein 	if (name != NULL)
226efaa6588SAlfred Perlstein 		sem_enter(td->td_proc, ret);
227efaa6588SAlfred Perlstein 	*ksret = ret;
228efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
229efaa6588SAlfred Perlstein 	if (nsems >= p31b_getcfg(CTL_P1003_1B_SEM_NSEMS_MAX)) {
230efaa6588SAlfred Perlstein 		sem_leave(td->td_proc, ret);
231efaa6588SAlfred Perlstein 		sem_free(ret);
232efaa6588SAlfred Perlstein 		error = ENFILE;
233efaa6588SAlfred Perlstein 	} else {
234efaa6588SAlfred Perlstein 		nsems++;
235efaa6588SAlfred Perlstein 		error = 0;
236efaa6588SAlfred Perlstein 	}
237efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
238efaa6588SAlfred Perlstein 	return (error);
239efaa6588SAlfred Perlstein }
240efaa6588SAlfred Perlstein 
241efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
242efaa6588SAlfred Perlstein struct ksem_init_args {
243efaa6588SAlfred Perlstein 	unsigned int value;
244efaa6588SAlfred Perlstein 	semid_t *idp;
245efaa6588SAlfred Perlstein };
246efaa6588SAlfred Perlstein int ksem_init(struct thread *td, struct ksem_init_args *uap);
247efaa6588SAlfred Perlstein #endif
248efaa6588SAlfred Perlstein int
249efaa6588SAlfred Perlstein ksem_init(td, uap)
250efaa6588SAlfred Perlstein 	struct thread *td;
251efaa6588SAlfred Perlstein 	struct ksem_init_args *uap;
252efaa6588SAlfred Perlstein {
253efaa6588SAlfred Perlstein 	int error;
254efaa6588SAlfred Perlstein 
255efaa6588SAlfred Perlstein 	error = kern_sem_init(td, UIO_USERSPACE, uap->value, uap->idp);
256efaa6588SAlfred Perlstein 	return (error);
257efaa6588SAlfred Perlstein }
258efaa6588SAlfred Perlstein 
259efaa6588SAlfred Perlstein static int
260efaa6588SAlfred Perlstein kern_sem_init(td, dir, value, idp)
261efaa6588SAlfred Perlstein 	struct thread *td;
262efaa6588SAlfred Perlstein 	int dir;
263efaa6588SAlfred Perlstein 	unsigned int value;
264efaa6588SAlfred Perlstein 	semid_t *idp;
265efaa6588SAlfred Perlstein {
266efaa6588SAlfred Perlstein 	struct ksem *ks;
267efaa6588SAlfred Perlstein 	semid_t id;
268efaa6588SAlfred Perlstein 	int error;
269efaa6588SAlfred Perlstein 
270efaa6588SAlfred Perlstein 	error = sem_create(td, NULL, &ks, S_IRWXU | S_IRWXG, value);
271efaa6588SAlfred Perlstein 	if (error)
272efaa6588SAlfred Perlstein 		return (error);
273efaa6588SAlfred Perlstein 	id = SEM_TO_ID(ks);
274efaa6588SAlfred Perlstein 	if (dir == UIO_USERSPACE) {
275efaa6588SAlfred Perlstein 		error = copyout(&id, idp, sizeof(id));
276efaa6588SAlfred Perlstein 		if (error) {
277efaa6588SAlfred Perlstein 			mtx_lock(&sem_lock);
278efaa6588SAlfred Perlstein 			sem_rel(ks);
279efaa6588SAlfred Perlstein 			mtx_unlock(&sem_lock);
280efaa6588SAlfred Perlstein 			return (error);
281efaa6588SAlfred Perlstein 		}
282efaa6588SAlfred Perlstein 	} else {
283efaa6588SAlfred Perlstein 		*idp = id;
284efaa6588SAlfred Perlstein 	}
285efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
286efaa6588SAlfred Perlstein 	LIST_INSERT_HEAD(&ksem_head, ks, ks_entry);
287efaa6588SAlfred Perlstein 	ks->ks_onlist = 1;
288efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
289efaa6588SAlfred Perlstein 	return (error);
290efaa6588SAlfred Perlstein }
291efaa6588SAlfred Perlstein 
292efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
293efaa6588SAlfred Perlstein struct ksem_open_args {
294efaa6588SAlfred Perlstein 	char *name;
295efaa6588SAlfred Perlstein 	int oflag;
296efaa6588SAlfred Perlstein 	mode_t mode;
297efaa6588SAlfred Perlstein 	unsigned int value;
298efaa6588SAlfred Perlstein 	semid_t *idp;
299efaa6588SAlfred Perlstein };
300efaa6588SAlfred Perlstein int ksem_open(struct thread *td, struct ksem_open_args *uap);
301efaa6588SAlfred Perlstein #endif
302efaa6588SAlfred Perlstein int
303efaa6588SAlfred Perlstein ksem_open(td, uap)
304efaa6588SAlfred Perlstein 	struct thread *td;
305efaa6588SAlfred Perlstein 	struct ksem_open_args *uap;
306efaa6588SAlfred Perlstein {
307efaa6588SAlfred Perlstein 	char name[SEM_MAX_NAMELEN + 1];
308efaa6588SAlfred Perlstein 	size_t done;
309efaa6588SAlfred Perlstein 	int error;
310efaa6588SAlfred Perlstein 
311efaa6588SAlfred Perlstein 	error = copyinstr(uap->name, name, SEM_MAX_NAMELEN + 1, &done);
312efaa6588SAlfred Perlstein 	if (error)
313efaa6588SAlfred Perlstein 		return (error);
314c814aa3fSAlfred Perlstein 	DP((">>> sem_open start\n"));
315efaa6588SAlfred Perlstein 	error = kern_sem_open(td, UIO_USERSPACE,
316efaa6588SAlfred Perlstein 	    name, uap->oflag, uap->mode, uap->value, uap->idp);
317c814aa3fSAlfred Perlstein 	DP(("<<< sem_open end\n"));
318efaa6588SAlfred Perlstein 	return (error);
319efaa6588SAlfred Perlstein }
320efaa6588SAlfred Perlstein 
321efaa6588SAlfred Perlstein static int
322efaa6588SAlfred Perlstein kern_sem_open(td, dir, name, oflag, mode, value, idp)
323efaa6588SAlfred Perlstein 	struct thread *td;
324efaa6588SAlfred Perlstein 	int dir;
325efaa6588SAlfred Perlstein 	const char *name;
326efaa6588SAlfred Perlstein 	int oflag;
327efaa6588SAlfred Perlstein 	mode_t mode;
328efaa6588SAlfred Perlstein 	unsigned int value;
329efaa6588SAlfred Perlstein 	semid_t *idp;
330efaa6588SAlfred Perlstein {
331efaa6588SAlfred Perlstein 	struct ksem *ksnew, *ks;
332efaa6588SAlfred Perlstein 	int error;
333efaa6588SAlfred Perlstein 	semid_t id;
334efaa6588SAlfred Perlstein 
335efaa6588SAlfred Perlstein 	ksnew = NULL;
336efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
337efaa6588SAlfred Perlstein 	ks = sem_lookup_byname(name);
338efaa6588SAlfred Perlstein 	/*
339efaa6588SAlfred Perlstein 	 * If we found it but O_EXCL is set, error.
340efaa6588SAlfred Perlstein 	 */
341efaa6588SAlfred Perlstein 	if (ks != NULL && (oflag & O_EXCL) != 0) {
342efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
343efaa6588SAlfred Perlstein 		return (EEXIST);
344efaa6588SAlfred Perlstein 	}
345efaa6588SAlfred Perlstein 	/*
346efaa6588SAlfred Perlstein 	 * If we didn't find it...
347efaa6588SAlfred Perlstein 	 */
348efaa6588SAlfred Perlstein 	if (ks == NULL) {
349efaa6588SAlfred Perlstein 		/*
350efaa6588SAlfred Perlstein 		 * didn't ask for creation? error.
351efaa6588SAlfred Perlstein 		 */
352efaa6588SAlfred Perlstein 		if ((oflag & O_CREAT) == 0) {
353efaa6588SAlfred Perlstein 			mtx_unlock(&sem_lock);
354efaa6588SAlfred Perlstein 			return (ENOENT);
355efaa6588SAlfred Perlstein 		}
356efaa6588SAlfred Perlstein 		/*
357efaa6588SAlfred Perlstein 		 * We may block during creation, so drop the lock.
358efaa6588SAlfred Perlstein 		 */
359efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
360efaa6588SAlfred Perlstein 		error = sem_create(td, name, &ksnew, mode, value);
361efaa6588SAlfred Perlstein 		if (error != 0)
362efaa6588SAlfred Perlstein 			return (error);
363efaa6588SAlfred Perlstein 		id = SEM_TO_ID(ksnew);
364efaa6588SAlfred Perlstein 		if (dir == UIO_USERSPACE) {
365c814aa3fSAlfred Perlstein 			DP(("about to copyout! %d to %p\n", id, idp));
366efaa6588SAlfred Perlstein 			error = copyout(&id, idp, sizeof(id));
367efaa6588SAlfred Perlstein 			if (error) {
368efaa6588SAlfred Perlstein 				mtx_lock(&sem_lock);
369efaa6588SAlfred Perlstein 				sem_leave(td->td_proc, ksnew);
370efaa6588SAlfred Perlstein 				sem_rel(ksnew);
371efaa6588SAlfred Perlstein 				mtx_unlock(&sem_lock);
372efaa6588SAlfred Perlstein 				return (error);
373efaa6588SAlfred Perlstein 			}
374efaa6588SAlfred Perlstein 		} else {
375c814aa3fSAlfred Perlstein 			DP(("about to set! %d to %p\n", id, idp));
376efaa6588SAlfred Perlstein 			*idp = id;
377efaa6588SAlfred Perlstein 		}
378efaa6588SAlfred Perlstein 		/*
379efaa6588SAlfred Perlstein 		 * We need to make sure we haven't lost a race while
380efaa6588SAlfred Perlstein 		 * allocating during creation.
381efaa6588SAlfred Perlstein 		 */
382efaa6588SAlfred Perlstein 		mtx_lock(&sem_lock);
383efaa6588SAlfred Perlstein 		ks = sem_lookup_byname(name);
384efaa6588SAlfred Perlstein 		if (ks != NULL) {
385efaa6588SAlfred Perlstein 			/* we lost... */
386efaa6588SAlfred Perlstein 			sem_leave(td->td_proc, ksnew);
387efaa6588SAlfred Perlstein 			sem_rel(ksnew);
388efaa6588SAlfred Perlstein 			/* we lost and we can't loose... */
389efaa6588SAlfred Perlstein 			if ((oflag & O_EXCL) != 0) {
390efaa6588SAlfred Perlstein 				mtx_unlock(&sem_lock);
391efaa6588SAlfred Perlstein 				return (EEXIST);
392efaa6588SAlfred Perlstein 			}
393efaa6588SAlfred Perlstein 		} else {
394c814aa3fSAlfred Perlstein 			DP(("sem_create: about to add to list...\n"));
395efaa6588SAlfred Perlstein 			LIST_INSERT_HEAD(&ksem_head, ksnew, ks_entry);
396c814aa3fSAlfred Perlstein 			DP(("sem_create: setting list bit...\n"));
397efaa6588SAlfred Perlstein 			ksnew->ks_onlist = 1;
398c814aa3fSAlfred Perlstein 			DP(("sem_create: done, about to unlock...\n"));
399efaa6588SAlfred Perlstein 		}
400efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
401efaa6588SAlfred Perlstein 	} else {
402efaa6588SAlfred Perlstein 		/*
403efaa6588SAlfred Perlstein 		 * if we aren't the creator, then enforce permissions.
404efaa6588SAlfred Perlstein 		 */
405b2546660SJohn Baldwin 		error = sem_perm(td, ks);
406efaa6588SAlfred Perlstein 		if (!error)
407efaa6588SAlfred Perlstein 			sem_ref(ks);
408efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
409efaa6588SAlfred Perlstein 		if (error)
410efaa6588SAlfred Perlstein 			return (error);
411efaa6588SAlfred Perlstein 		id = SEM_TO_ID(ks);
412efaa6588SAlfred Perlstein 		if (dir == UIO_USERSPACE) {
413efaa6588SAlfred Perlstein 			error = copyout(&id, idp, sizeof(id));
414efaa6588SAlfred Perlstein 			if (error) {
415efaa6588SAlfred Perlstein 				mtx_lock(&sem_lock);
416efaa6588SAlfred Perlstein 				sem_rel(ks);
417efaa6588SAlfred Perlstein 				mtx_unlock(&sem_lock);
418efaa6588SAlfred Perlstein 				return (error);
419efaa6588SAlfred Perlstein 			}
420efaa6588SAlfred Perlstein 		} else {
421efaa6588SAlfred Perlstein 			*idp = id;
422efaa6588SAlfred Perlstein 		}
423efaa6588SAlfred Perlstein 		sem_enter(td->td_proc, ks);
424efaa6588SAlfred Perlstein 		mtx_lock(&sem_lock);
425efaa6588SAlfred Perlstein 		sem_rel(ks);
426efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
427efaa6588SAlfred Perlstein 	}
428efaa6588SAlfred Perlstein 	return (error);
429efaa6588SAlfred Perlstein }
430efaa6588SAlfred Perlstein 
431c3053131SPoul-Henning Kamp static int
432b2546660SJohn Baldwin sem_perm(td, ks)
433b2546660SJohn Baldwin 	struct thread *td;
434efaa6588SAlfred Perlstein 	struct ksem *ks;
435efaa6588SAlfred Perlstein {
436efaa6588SAlfred Perlstein 	struct ucred *uc;
437efaa6588SAlfred Perlstein 
438b2546660SJohn Baldwin 	uc = td->td_ucred;
439c814aa3fSAlfred Perlstein 	DP(("sem_perm: uc(%d,%d) ks(%d,%d,%o)\n",
440efaa6588SAlfred Perlstein 	    uc->cr_uid, uc->cr_gid,
441c814aa3fSAlfred Perlstein 	     ks->ks_uid, ks->ks_gid, ks->ks_mode));
442efaa6588SAlfred Perlstein 	if ((uc->cr_uid == ks->ks_uid && (ks->ks_mode & S_IWUSR) != 0) ||
443efaa6588SAlfred Perlstein 	    (uc->cr_gid == ks->ks_gid && (ks->ks_mode & S_IWGRP) != 0) ||
444b2546660SJohn Baldwin 	    (ks->ks_mode & S_IWOTH) != 0 || suser(td) == 0)
445efaa6588SAlfred Perlstein 		return (0);
446efaa6588SAlfred Perlstein 	return (EPERM);
447efaa6588SAlfred Perlstein }
448efaa6588SAlfred Perlstein 
449c3053131SPoul-Henning Kamp static void
450efaa6588SAlfred Perlstein sem_free(struct ksem *ks)
451efaa6588SAlfred Perlstein {
452efaa6588SAlfred Perlstein 
453efaa6588SAlfred Perlstein 	nsems--;
454efaa6588SAlfred Perlstein 	if (ks->ks_onlist)
455efaa6588SAlfred Perlstein 		LIST_REMOVE(ks, ks_entry);
456efaa6588SAlfred Perlstein 	if (ks->ks_name != NULL)
457efaa6588SAlfred Perlstein 		free(ks->ks_name, M_SEM);
458efaa6588SAlfred Perlstein 	cv_destroy(&ks->ks_cv);
459efaa6588SAlfred Perlstein 	free(ks, M_SEM);
460efaa6588SAlfred Perlstein }
461efaa6588SAlfred Perlstein 
462efaa6588SAlfred Perlstein static __inline struct kuser *sem_getuser(struct proc *p, struct ksem *ks);
463efaa6588SAlfred Perlstein 
464efaa6588SAlfred Perlstein static __inline struct kuser *
465efaa6588SAlfred Perlstein sem_getuser(p, ks)
466efaa6588SAlfred Perlstein 	struct proc *p;
467efaa6588SAlfred Perlstein 	struct ksem *ks;
468efaa6588SAlfred Perlstein {
469efaa6588SAlfred Perlstein 	struct kuser *k;
470efaa6588SAlfred Perlstein 
471efaa6588SAlfred Perlstein 	LIST_FOREACH(k, &ks->ks_users, ku_next)
472efaa6588SAlfred Perlstein 		if (k->ku_pid == p->p_pid)
473efaa6588SAlfred Perlstein 			return (k);
474efaa6588SAlfred Perlstein 	return (NULL);
475efaa6588SAlfred Perlstein }
476efaa6588SAlfred Perlstein 
477c3053131SPoul-Henning Kamp static int
478b2546660SJohn Baldwin sem_hasopen(td, ks)
479b2546660SJohn Baldwin 	struct thread *td;
480efaa6588SAlfred Perlstein 	struct ksem *ks;
481efaa6588SAlfred Perlstein {
482efaa6588SAlfred Perlstein 
483aae94fbbSDaniel Eischen 	return ((ks->ks_name == NULL && sem_perm(td, ks) == 0)
484b2546660SJohn Baldwin 	    || sem_getuser(td->td_proc, ks) != NULL);
485efaa6588SAlfred Perlstein }
486efaa6588SAlfred Perlstein 
487c3053131SPoul-Henning Kamp static int
488efaa6588SAlfred Perlstein sem_leave(p, ks)
489efaa6588SAlfred Perlstein 	struct proc *p;
490efaa6588SAlfred Perlstein 	struct ksem *ks;
491efaa6588SAlfred Perlstein {
492efaa6588SAlfred Perlstein 	struct kuser *k;
493efaa6588SAlfred Perlstein 
494c814aa3fSAlfred Perlstein 	DP(("sem_leave: ks = %p\n", ks));
495efaa6588SAlfred Perlstein 	k = sem_getuser(p, ks);
496c814aa3fSAlfred Perlstein 	DP(("sem_leave: ks = %p, k = %p\n", ks, k));
497efaa6588SAlfred Perlstein 	if (k != NULL) {
498efaa6588SAlfred Perlstein 		LIST_REMOVE(k, ku_next);
499efaa6588SAlfred Perlstein 		sem_rel(ks);
500c814aa3fSAlfred Perlstein 		DP(("sem_leave: about to free k\n"));
501efaa6588SAlfred Perlstein 		free(k, M_SEM);
502c814aa3fSAlfred Perlstein 		DP(("sem_leave: returning\n"));
503efaa6588SAlfred Perlstein 		return (0);
504efaa6588SAlfred Perlstein 	}
505b3890a1cSAlfred Perlstein 	return (EINVAL);
506efaa6588SAlfred Perlstein }
507efaa6588SAlfred Perlstein 
508c3053131SPoul-Henning Kamp static void
509efaa6588SAlfred Perlstein sem_enter(p, ks)
510efaa6588SAlfred Perlstein 	struct proc *p;
511efaa6588SAlfred Perlstein 	struct ksem *ks;
512efaa6588SAlfred Perlstein {
513efaa6588SAlfred Perlstein 	struct kuser *ku, *k;
514efaa6588SAlfred Perlstein 
515a163d034SWarner Losh 	ku = malloc(sizeof(*ku), M_SEM, M_WAITOK);
516efaa6588SAlfred Perlstein 	ku->ku_pid = p->p_pid;
517efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
518efaa6588SAlfred Perlstein 	k = sem_getuser(p, ks);
519efaa6588SAlfred Perlstein 	if (k != NULL) {
520efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
521efaa6588SAlfred Perlstein 		free(ku, M_TEMP);
522efaa6588SAlfred Perlstein 		return;
523efaa6588SAlfred Perlstein 	}
524efaa6588SAlfred Perlstein 	LIST_INSERT_HEAD(&ks->ks_users, ku, ku_next);
525efaa6588SAlfred Perlstein 	sem_ref(ks);
526efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
527efaa6588SAlfred Perlstein }
528efaa6588SAlfred Perlstein 
529efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
530efaa6588SAlfred Perlstein struct ksem_unlink_args {
531efaa6588SAlfred Perlstein 	char *name;
532efaa6588SAlfred Perlstein };
533efaa6588SAlfred Perlstein int ksem_unlink(struct thread *td, struct ksem_unlink_args *uap);
534efaa6588SAlfred Perlstein #endif
535efaa6588SAlfred Perlstein 
536efaa6588SAlfred Perlstein int
537efaa6588SAlfred Perlstein ksem_unlink(td, uap)
538efaa6588SAlfred Perlstein 	struct thread *td;
539efaa6588SAlfred Perlstein 	struct ksem_unlink_args *uap;
540efaa6588SAlfred Perlstein {
541efaa6588SAlfred Perlstein 	char name[SEM_MAX_NAMELEN + 1];
542efaa6588SAlfred Perlstein 	size_t done;
543efaa6588SAlfred Perlstein 	int error;
544efaa6588SAlfred Perlstein 
545efaa6588SAlfred Perlstein 	error = copyinstr(uap->name, name, SEM_MAX_NAMELEN + 1, &done);
546efaa6588SAlfred Perlstein 	return (error ? error :
547efaa6588SAlfred Perlstein 	    kern_sem_unlink(td, name));
548efaa6588SAlfred Perlstein }
549efaa6588SAlfred Perlstein 
550efaa6588SAlfred Perlstein static int
551efaa6588SAlfred Perlstein kern_sem_unlink(td, name)
552efaa6588SAlfred Perlstein 	struct thread *td;
553efaa6588SAlfred Perlstein 	const char *name;
554efaa6588SAlfred Perlstein {
555efaa6588SAlfred Perlstein 	struct ksem *ks;
556efaa6588SAlfred Perlstein 	int error;
557efaa6588SAlfred Perlstein 
558efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
559efaa6588SAlfred Perlstein 	ks = sem_lookup_byname(name);
560efaa6588SAlfred Perlstein 	if (ks == NULL)
561efaa6588SAlfred Perlstein 		error = ENOENT;
562efaa6588SAlfred Perlstein 	else
563b2546660SJohn Baldwin 		error = sem_perm(td, ks);
564c814aa3fSAlfred Perlstein 	DP(("sem_unlink: '%s' ks = %p, error = %d\n", name, ks, error));
565efaa6588SAlfred Perlstein 	if (error == 0) {
566efaa6588SAlfred Perlstein 		LIST_REMOVE(ks, ks_entry);
567efaa6588SAlfred Perlstein 		LIST_INSERT_HEAD(&ksem_deadhead, ks, ks_entry);
568efaa6588SAlfred Perlstein 		sem_rel(ks);
569efaa6588SAlfred Perlstein 	}
570efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
571efaa6588SAlfred Perlstein 	return (error);
572efaa6588SAlfred Perlstein }
573efaa6588SAlfred Perlstein 
574efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
575efaa6588SAlfred Perlstein struct ksem_close_args {
576efaa6588SAlfred Perlstein 	semid_t id;
577efaa6588SAlfred Perlstein };
578efaa6588SAlfred Perlstein int ksem_close(struct thread *td, struct ksem_close_args *uap);
579efaa6588SAlfred Perlstein #endif
580efaa6588SAlfred Perlstein 
581efaa6588SAlfred Perlstein int
582efaa6588SAlfred Perlstein ksem_close(struct thread *td, struct ksem_close_args *uap)
583efaa6588SAlfred Perlstein {
584efaa6588SAlfred Perlstein 
585efaa6588SAlfred Perlstein 	return (kern_sem_close(td, uap->id));
586efaa6588SAlfred Perlstein }
587efaa6588SAlfred Perlstein 
588c3053131SPoul-Henning Kamp static int
589efaa6588SAlfred Perlstein kern_sem_close(td, id)
590efaa6588SAlfred Perlstein 	struct thread *td;
591efaa6588SAlfred Perlstein 	semid_t id;
592efaa6588SAlfred Perlstein {
593efaa6588SAlfred Perlstein 	struct ksem *ks;
594efaa6588SAlfred Perlstein 	int error;
595efaa6588SAlfred Perlstein 
596efaa6588SAlfred Perlstein 	error = EINVAL;
597efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
598efaa6588SAlfred Perlstein 	ks = ID_TO_SEM(id);
599efaa6588SAlfred Perlstein 	/* this is not a valid operation for unnamed sems */
600efaa6588SAlfred Perlstein 	if (ks != NULL && ks->ks_name != NULL)
601b3890a1cSAlfred Perlstein 		error = sem_leave(td->td_proc, ks);
602efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
603b3890a1cSAlfred Perlstein 	return (error);
604efaa6588SAlfred Perlstein }
605efaa6588SAlfred Perlstein 
606efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
607efaa6588SAlfred Perlstein struct ksem_post_args {
608efaa6588SAlfred Perlstein 	semid_t id;
609efaa6588SAlfred Perlstein };
610efaa6588SAlfred Perlstein int ksem_post(struct thread *td, struct ksem_post_args *uap);
611efaa6588SAlfred Perlstein #endif
612efaa6588SAlfred Perlstein int
613efaa6588SAlfred Perlstein ksem_post(td, uap)
614efaa6588SAlfred Perlstein 	struct thread *td;
615efaa6588SAlfred Perlstein 	struct ksem_post_args *uap;
616efaa6588SAlfred Perlstein {
617efaa6588SAlfred Perlstein 
618efaa6588SAlfred Perlstein 	return (kern_sem_post(td, uap->id));
619efaa6588SAlfred Perlstein }
620efaa6588SAlfred Perlstein 
621c3053131SPoul-Henning Kamp static int
622efaa6588SAlfred Perlstein kern_sem_post(td, id)
623efaa6588SAlfred Perlstein 	struct thread *td;
624efaa6588SAlfred Perlstein 	semid_t id;
625efaa6588SAlfred Perlstein {
626efaa6588SAlfred Perlstein 	struct ksem *ks;
627efaa6588SAlfred Perlstein 	int error;
628efaa6588SAlfred Perlstein 
629efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
630efaa6588SAlfred Perlstein 	ks = ID_TO_SEM(id);
631b2546660SJohn Baldwin 	if (ks == NULL || !sem_hasopen(td, ks)) {
632efaa6588SAlfred Perlstein 		error = EINVAL;
633efaa6588SAlfred Perlstein 		goto err;
634efaa6588SAlfred Perlstein 	}
635efaa6588SAlfred Perlstein 	if (ks->ks_value == SEM_VALUE_MAX) {
636efaa6588SAlfred Perlstein 		error = EOVERFLOW;
637efaa6588SAlfred Perlstein 		goto err;
638efaa6588SAlfred Perlstein 	}
639efaa6588SAlfred Perlstein 	++ks->ks_value;
640efaa6588SAlfred Perlstein 	if (ks->ks_waiters > 0)
641efaa6588SAlfred Perlstein 		cv_signal(&ks->ks_cv);
642efaa6588SAlfred Perlstein 	error = 0;
643efaa6588SAlfred Perlstein err:
644efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
645efaa6588SAlfred Perlstein 	return (error);
646efaa6588SAlfred Perlstein }
647efaa6588SAlfred Perlstein 
648efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
649efaa6588SAlfred Perlstein struct ksem_wait_args {
650efaa6588SAlfred Perlstein 	semid_t id;
651efaa6588SAlfred Perlstein };
652efaa6588SAlfred Perlstein int ksem_wait(struct thread *td, struct ksem_wait_args *uap);
653efaa6588SAlfred Perlstein #endif
654efaa6588SAlfred Perlstein 
655efaa6588SAlfred Perlstein int
656efaa6588SAlfred Perlstein ksem_wait(td, uap)
657efaa6588SAlfred Perlstein 	struct thread *td;
658efaa6588SAlfred Perlstein 	struct ksem_wait_args *uap;
659efaa6588SAlfred Perlstein {
660efaa6588SAlfred Perlstein 
661aae94fbbSDaniel Eischen 	return (kern_sem_wait(td, uap->id, 0, NULL));
662aae94fbbSDaniel Eischen }
663aae94fbbSDaniel Eischen 
664aae94fbbSDaniel Eischen #ifndef _SYS_SYSPROTO_H_
665aae94fbbSDaniel Eischen struct ksem_timedwait_args {
666aae94fbbSDaniel Eischen 	semid_t id;
667aae94fbbSDaniel Eischen 	struct timespec *abstime;
668aae94fbbSDaniel Eischen };
669aae94fbbSDaniel Eischen int ksem_timedwait(struct thread *td, struct ksem_timedwait_args *uap);
670aae94fbbSDaniel Eischen #endif
671aae94fbbSDaniel Eischen int
672aae94fbbSDaniel Eischen ksem_timedwait(td, uap)
673aae94fbbSDaniel Eischen 	struct thread *td;
674aae94fbbSDaniel Eischen 	struct ksem_timedwait_args *uap;
675aae94fbbSDaniel Eischen {
676aae94fbbSDaniel Eischen 	struct timespec abstime;
677aae94fbbSDaniel Eischen 	struct timespec *ts;
678aae94fbbSDaniel Eischen 	int error;
679aae94fbbSDaniel Eischen 
680aae94fbbSDaniel Eischen 	/* We allow a null timespec (wait forever). */
681aae94fbbSDaniel Eischen 	if (uap->abstime == NULL)
682aae94fbbSDaniel Eischen 		ts = NULL;
683aae94fbbSDaniel Eischen 	else {
684aae94fbbSDaniel Eischen 		error = copyin(uap->abstime, &abstime, sizeof(abstime));
685aae94fbbSDaniel Eischen 		if (error != 0)
686aae94fbbSDaniel Eischen 			return (error);
687aae94fbbSDaniel Eischen 		if (abstime.tv_nsec >= 1000000000 || abstime.tv_nsec < 0)
688aae94fbbSDaniel Eischen 			return (EINVAL);
689aae94fbbSDaniel Eischen 		ts = &abstime;
690aae94fbbSDaniel Eischen 	}
691aae94fbbSDaniel Eischen 	return (kern_sem_wait(td, uap->id, 0, ts));
692efaa6588SAlfred Perlstein }
693efaa6588SAlfred Perlstein 
694efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
695efaa6588SAlfred Perlstein struct ksem_trywait_args {
696efaa6588SAlfred Perlstein 	semid_t id;
697efaa6588SAlfred Perlstein };
698efaa6588SAlfred Perlstein int ksem_trywait(struct thread *td, struct ksem_trywait_args *uap);
699efaa6588SAlfred Perlstein #endif
700efaa6588SAlfred Perlstein int
701efaa6588SAlfred Perlstein ksem_trywait(td, uap)
702efaa6588SAlfred Perlstein 	struct thread *td;
703efaa6588SAlfred Perlstein 	struct ksem_trywait_args *uap;
704efaa6588SAlfred Perlstein {
705efaa6588SAlfred Perlstein 
706aae94fbbSDaniel Eischen 	return (kern_sem_wait(td, uap->id, 1, NULL));
707efaa6588SAlfred Perlstein }
708efaa6588SAlfred Perlstein 
709c3053131SPoul-Henning Kamp static int
710aae94fbbSDaniel Eischen kern_sem_wait(td, id, tryflag, abstime)
711efaa6588SAlfred Perlstein 	struct thread *td;
712efaa6588SAlfred Perlstein 	semid_t id;
713efaa6588SAlfred Perlstein 	int tryflag;
714aae94fbbSDaniel Eischen 	struct timespec *abstime;
715efaa6588SAlfred Perlstein {
716aae94fbbSDaniel Eischen 	struct timespec ts1, ts2;
717aae94fbbSDaniel Eischen 	struct timeval tv;
718efaa6588SAlfred Perlstein 	struct ksem *ks;
719efaa6588SAlfred Perlstein 	int error;
720efaa6588SAlfred Perlstein 
721c814aa3fSAlfred Perlstein 	DP((">>> kern_sem_wait entered!\n"));
722efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
723efaa6588SAlfred Perlstein 	ks = ID_TO_SEM(id);
724efaa6588SAlfred Perlstein 	if (ks == NULL) {
725c814aa3fSAlfred Perlstein 		DP(("kern_sem_wait ks == NULL\n"));
726efaa6588SAlfred Perlstein 		error = EINVAL;
727efaa6588SAlfred Perlstein 		goto err;
728efaa6588SAlfred Perlstein 	}
729efaa6588SAlfred Perlstein 	sem_ref(ks);
730b2546660SJohn Baldwin 	if (!sem_hasopen(td, ks)) {
731c814aa3fSAlfred Perlstein 		DP(("kern_sem_wait hasopen failed\n"));
732efaa6588SAlfred Perlstein 		error = EINVAL;
733efaa6588SAlfred Perlstein 		goto err;
734efaa6588SAlfred Perlstein 	}
735c814aa3fSAlfred Perlstein 	DP(("kern_sem_wait value = %d, tryflag %d\n", ks->ks_value, tryflag));
736efaa6588SAlfred Perlstein 	if (ks->ks_value == 0) {
737efaa6588SAlfred Perlstein 		ks->ks_waiters++;
738aae94fbbSDaniel Eischen 		if (tryflag != 0)
739aae94fbbSDaniel Eischen 			error = EAGAIN;
740aae94fbbSDaniel Eischen 		else if (abstime == NULL)
741aae94fbbSDaniel Eischen 			error = cv_wait_sig(&ks->ks_cv, &sem_lock);
742aae94fbbSDaniel Eischen 		else {
743aae94fbbSDaniel Eischen 			for (;;) {
744aae94fbbSDaniel Eischen 				ts1 = *abstime;
745aae94fbbSDaniel Eischen 				getnanotime(&ts2);
746aae94fbbSDaniel Eischen 				timespecsub(&ts1, &ts2);
747aae94fbbSDaniel Eischen 				TIMESPEC_TO_TIMEVAL(&tv, &ts1);
748aae94fbbSDaniel Eischen 				if (tv.tv_sec < 0) {
749aae94fbbSDaniel Eischen 					error = ETIMEDOUT;
750aae94fbbSDaniel Eischen 					break;
751aae94fbbSDaniel Eischen 				}
752aae94fbbSDaniel Eischen 				error = cv_timedwait_sig(&ks->ks_cv,
753aae94fbbSDaniel Eischen 				    &sem_lock, tvtohz(&tv));
754aae94fbbSDaniel Eischen 				if (error != EWOULDBLOCK)
755aae94fbbSDaniel Eischen 					break;
756aae94fbbSDaniel Eischen 			}
757aae94fbbSDaniel Eischen 		}
758efaa6588SAlfred Perlstein 		ks->ks_waiters--;
759efaa6588SAlfred Perlstein 		if (error)
760efaa6588SAlfred Perlstein 			goto err;
761efaa6588SAlfred Perlstein 	}
762efaa6588SAlfred Perlstein 	ks->ks_value--;
763efaa6588SAlfred Perlstein 	error = 0;
764efaa6588SAlfred Perlstein err:
765efaa6588SAlfred Perlstein 	if (ks != NULL)
766efaa6588SAlfred Perlstein 		sem_rel(ks);
767efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
768c814aa3fSAlfred Perlstein 	DP(("<<< kern_sem_wait leaving, error = %d\n", error));
769efaa6588SAlfred Perlstein 	return (error);
770efaa6588SAlfred Perlstein }
771efaa6588SAlfred Perlstein 
772efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
773efaa6588SAlfred Perlstein struct ksem_getvalue_args {
774efaa6588SAlfred Perlstein 	semid_t id;
775efaa6588SAlfred Perlstein 	int *val;
776efaa6588SAlfred Perlstein };
777efaa6588SAlfred Perlstein int ksem_getvalue(struct thread *td, struct ksem_getvalue_args *uap);
778efaa6588SAlfred Perlstein #endif
779efaa6588SAlfred Perlstein int
780efaa6588SAlfred Perlstein ksem_getvalue(td, uap)
781efaa6588SAlfred Perlstein 	struct thread *td;
782efaa6588SAlfred Perlstein 	struct ksem_getvalue_args *uap;
783efaa6588SAlfred Perlstein {
784efaa6588SAlfred Perlstein 	struct ksem *ks;
785efaa6588SAlfred Perlstein 	int error, val;
786efaa6588SAlfred Perlstein 
787efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
788efaa6588SAlfred Perlstein 	ks = ID_TO_SEM(uap->id);
789b2546660SJohn Baldwin 	if (ks == NULL || !sem_hasopen(td, ks)) {
790efaa6588SAlfred Perlstein 		mtx_unlock(&sem_lock);
791efaa6588SAlfred Perlstein 		return (EINVAL);
792efaa6588SAlfred Perlstein 	}
793efaa6588SAlfred Perlstein 	val = ks->ks_value;
794efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
795efaa6588SAlfred Perlstein 	error = copyout(&val, uap->val, sizeof(val));
796efaa6588SAlfred Perlstein 	return (error);
797efaa6588SAlfred Perlstein }
798efaa6588SAlfred Perlstein 
799efaa6588SAlfred Perlstein #ifndef _SYS_SYSPROTO_H_
800efaa6588SAlfred Perlstein struct ksem_destroy_args {
801efaa6588SAlfred Perlstein 	semid_t id;
802efaa6588SAlfred Perlstein };
803efaa6588SAlfred Perlstein int ksem_destroy(struct thread *td, struct ksem_destroy_args *uap);
804efaa6588SAlfred Perlstein #endif
805efaa6588SAlfred Perlstein int
806efaa6588SAlfred Perlstein ksem_destroy(td, uap)
807efaa6588SAlfred Perlstein 	struct thread *td;
808efaa6588SAlfred Perlstein 	struct ksem_destroy_args *uap;
809efaa6588SAlfred Perlstein {
810efaa6588SAlfred Perlstein 	struct ksem *ks;
811efaa6588SAlfred Perlstein 	int error;
812efaa6588SAlfred Perlstein 
813efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
814efaa6588SAlfred Perlstein 	ks = ID_TO_SEM(uap->id);
815b2546660SJohn Baldwin 	if (ks == NULL || !sem_hasopen(td, ks) ||
816efaa6588SAlfred Perlstein 	    ks->ks_name != NULL) {
817efaa6588SAlfred Perlstein 		error = EINVAL;
818efaa6588SAlfred Perlstein 		goto err;
819efaa6588SAlfred Perlstein 	}
820efaa6588SAlfred Perlstein 	if (ks->ks_waiters != 0) {
821efaa6588SAlfred Perlstein 		error = EBUSY;
822efaa6588SAlfred Perlstein 		goto err;
823efaa6588SAlfred Perlstein 	}
824efaa6588SAlfred Perlstein 	sem_rel(ks);
825efaa6588SAlfred Perlstein 	error = 0;
826efaa6588SAlfred Perlstein err:
827efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
828efaa6588SAlfred Perlstein 	return (error);
829efaa6588SAlfred Perlstein }
830efaa6588SAlfred Perlstein 
831c3053131SPoul-Henning Kamp static void
83275b8b3b2SJohn Baldwin sem_exithook(arg, p)
83375b8b3b2SJohn Baldwin 	void *arg;
834efaa6588SAlfred Perlstein 	struct proc *p;
835efaa6588SAlfred Perlstein {
836efaa6588SAlfred Perlstein 	struct ksem *ks, *ksnext;
837efaa6588SAlfred Perlstein 
838efaa6588SAlfred Perlstein 	mtx_lock(&sem_lock);
839efaa6588SAlfred Perlstein 	ks = LIST_FIRST(&ksem_head);
840efaa6588SAlfred Perlstein 	while (ks != NULL) {
841efaa6588SAlfred Perlstein 		ksnext = LIST_NEXT(ks, ks_entry);
842efaa6588SAlfred Perlstein 		sem_leave(p, ks);
843efaa6588SAlfred Perlstein 		ks = ksnext;
844efaa6588SAlfred Perlstein 	}
845efaa6588SAlfred Perlstein 	ks = LIST_FIRST(&ksem_deadhead);
846efaa6588SAlfred Perlstein 	while (ks != NULL) {
847efaa6588SAlfred Perlstein 		ksnext = LIST_NEXT(ks, ks_entry);
848efaa6588SAlfred Perlstein 		sem_leave(p, ks);
849efaa6588SAlfred Perlstein 		ks = ksnext;
850efaa6588SAlfred Perlstein 	}
851efaa6588SAlfred Perlstein 	mtx_unlock(&sem_lock);
852efaa6588SAlfred Perlstein }
853efaa6588SAlfred Perlstein 
854efaa6588SAlfred Perlstein static int
855efaa6588SAlfred Perlstein sem_modload(struct module *module, int cmd, void *arg)
856efaa6588SAlfred Perlstein {
857efaa6588SAlfred Perlstein         int error = 0;
858efaa6588SAlfred Perlstein 
859efaa6588SAlfred Perlstein         switch (cmd) {
860efaa6588SAlfred Perlstein         case MOD_LOAD:
861efaa6588SAlfred Perlstein 		mtx_init(&sem_lock, "sem", "semaphore", MTX_DEF);
862efaa6588SAlfred Perlstein 		p31b_setcfg(CTL_P1003_1B_SEM_NSEMS_MAX, SEM_MAX);
863efaa6588SAlfred Perlstein 		p31b_setcfg(CTL_P1003_1B_SEM_VALUE_MAX, SEM_VALUE_MAX);
86475b8b3b2SJohn Baldwin 		sem_exit_tag = EVENTHANDLER_REGISTER(process_exit, sem_exithook,
86575b8b3b2SJohn Baldwin 		    NULL, EVENTHANDLER_PRI_ANY);
86675b8b3b2SJohn Baldwin 		sem_exec_tag = EVENTHANDLER_REGISTER(process_exec, sem_exithook,
86775b8b3b2SJohn Baldwin 		    NULL, EVENTHANDLER_PRI_ANY);
868efaa6588SAlfred Perlstein                 break;
869efaa6588SAlfred Perlstein         case MOD_UNLOAD:
870efaa6588SAlfred Perlstein 		if (nsems != 0) {
871efaa6588SAlfred Perlstein 			error = EOPNOTSUPP;
872efaa6588SAlfred Perlstein 			break;
873efaa6588SAlfred Perlstein 		}
87475b8b3b2SJohn Baldwin 		EVENTHANDLER_DEREGISTER(process_exit, sem_exit_tag);
87575b8b3b2SJohn Baldwin 		EVENTHANDLER_DEREGISTER(process_exec, sem_exec_tag);
876efaa6588SAlfred Perlstein 		mtx_destroy(&sem_lock);
877efaa6588SAlfred Perlstein                 break;
878efaa6588SAlfred Perlstein         case MOD_SHUTDOWN:
879efaa6588SAlfred Perlstein                 break;
880efaa6588SAlfred Perlstein         default:
881efaa6588SAlfred Perlstein                 error = EINVAL;
882efaa6588SAlfred Perlstein                 break;
883efaa6588SAlfred Perlstein         }
884efaa6588SAlfred Perlstein         return (error);
885efaa6588SAlfred Perlstein }
886efaa6588SAlfred Perlstein 
887efaa6588SAlfred Perlstein static moduledata_t sem_mod = {
888efaa6588SAlfred Perlstein         "sem",
889efaa6588SAlfred Perlstein         &sem_modload,
890efaa6588SAlfred Perlstein         NULL
891efaa6588SAlfred Perlstein };
892efaa6588SAlfred Perlstein 
893efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_init);
894efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_open);
895efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_unlink);
896efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_close);
897efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_post);
898efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_wait);
899aae94fbbSDaniel Eischen SYSCALL_MODULE_HELPER(ksem_timedwait);
900efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_trywait);
901efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_getvalue);
902efaa6588SAlfred Perlstein SYSCALL_MODULE_HELPER(ksem_destroy);
903efaa6588SAlfred Perlstein 
904efaa6588SAlfred Perlstein DECLARE_MODULE(sem, sem_mod, SI_SUB_SYSV_SEM, SI_ORDER_FIRST);
905efaa6588SAlfred Perlstein MODULE_VERSION(sem, 1);
906