uipc_sem.c revision 1.16 1 1.16 thorpej /* $NetBSD: uipc_sem.c,v 1.16 2006/10/08 04:28:44 thorpej Exp $ */
2 1.3 thorpej
3 1.3 thorpej /*-
4 1.3 thorpej * Copyright (c) 2003 The NetBSD Foundation, Inc.
5 1.3 thorpej * All rights reserved.
6 1.3 thorpej *
7 1.3 thorpej * This code is derived from software contributed to The NetBSD Foundation
8 1.3 thorpej * by Jason R. Thorpe of Wasabi Systems, Inc.
9 1.3 thorpej *
10 1.3 thorpej * Redistribution and use in source and binary forms, with or without
11 1.3 thorpej * modification, are permitted provided that the following conditions
12 1.3 thorpej * are met:
13 1.3 thorpej * 1. Redistributions of source code must retain the above copyright
14 1.3 thorpej * notice, this list of conditions and the following disclaimer.
15 1.3 thorpej * 2. Redistributions in binary form must reproduce the above copyright
16 1.3 thorpej * notice, this list of conditions and the following disclaimer in the
17 1.3 thorpej * documentation and/or other materials provided with the distribution.
18 1.3 thorpej * 3. All advertising materials mentioning features or use of this software
19 1.3 thorpej * must display the following acknowledgement:
20 1.3 thorpej * This product includes software developed by the NetBSD
21 1.3 thorpej * Foundation, Inc. and its contributors.
22 1.3 thorpej * 4. Neither the name of The NetBSD Foundation nor the names of its
23 1.3 thorpej * contributors may be used to endorse or promote products derived
24 1.3 thorpej * from this software without specific prior written permission.
25 1.3 thorpej *
26 1.3 thorpej * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 1.3 thorpej * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 1.3 thorpej * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 1.3 thorpej * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 1.3 thorpej * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 1.3 thorpej * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 1.3 thorpej * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 1.3 thorpej * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 1.3 thorpej * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 1.3 thorpej * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 1.3 thorpej * POSSIBILITY OF SUCH DAMAGE.
37 1.3 thorpej */
38 1.1 christos
39 1.1 christos /*
40 1.1 christos * Copyright (c) 2002 Alfred Perlstein <alfred (at) FreeBSD.org>
41 1.1 christos * All rights reserved.
42 1.1 christos *
43 1.1 christos * Redistribution and use in source and binary forms, with or without
44 1.1 christos * modification, are permitted provided that the following conditions
45 1.1 christos * are met:
46 1.1 christos * 1. Redistributions of source code must retain the above copyright
47 1.1 christos * notice, this list of conditions and the following disclaimer.
48 1.1 christos * 2. Redistributions in binary form must reproduce the above copyright
49 1.1 christos * notice, this list of conditions and the following disclaimer in the
50 1.1 christos * documentation and/or other materials provided with the distribution.
51 1.1 christos *
52 1.1 christos * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
53 1.1 christos * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
54 1.1 christos * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
55 1.1 christos * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
56 1.1 christos * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
57 1.1 christos * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
58 1.1 christos * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
59 1.1 christos * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
60 1.1 christos * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
61 1.1 christos * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
62 1.1 christos * SUCH DAMAGE.
63 1.1 christos */
64 1.9 lukem
65 1.9 lukem #include <sys/cdefs.h>
66 1.16 thorpej __KERNEL_RCSID(0, "$NetBSD: uipc_sem.c,v 1.16 2006/10/08 04:28:44 thorpej Exp $");
67 1.1 christos
68 1.1 christos #include "opt_posix.h"
69 1.1 christos
70 1.1 christos #include <sys/param.h>
71 1.1 christos #include <sys/systm.h>
72 1.1 christos #include <sys/kernel.h>
73 1.1 christos #include <sys/proc.h>
74 1.1 christos #include <sys/lock.h>
75 1.1 christos #include <sys/ksem.h>
76 1.5 matt #include <sys/sa.h>
77 1.1 christos #include <sys/syscall.h>
78 1.1 christos #include <sys/stat.h>
79 1.1 christos #include <sys/malloc.h>
80 1.1 christos #include <sys/fcntl.h>
81 1.14 elad #include <sys/kauth.h>
82 1.1 christos
83 1.1 christos #include <sys/mount.h>
84 1.1 christos
85 1.1 christos #include <sys/syscallargs.h>
86 1.1 christos
87 1.1 christos #ifndef SEM_MAX
88 1.1 christos #define SEM_MAX 30
89 1.1 christos #endif
90 1.1 christos
91 1.1 christos #define SEM_MAX_NAMELEN 14
92 1.1 christos #define SEM_VALUE_MAX (~0U)
93 1.13 cube #define SEM_HASHTBL_SIZE 13
94 1.1 christos
95 1.13 cube #define SEM_TO_ID(x) (((x)->ks_id))
96 1.13 cube #define SEM_HASH(id) ((id) % SEM_HASHTBL_SIZE)
97 1.4 thorpej
98 1.4 thorpej MALLOC_DEFINE(M_SEM, "p1003_1b_sem", "p1003_1b semaphores");
99 1.1 christos
100 1.3 thorpej /*
101 1.3 thorpej * Note: to read the ks_name member, you need either the ks_interlock
102 1.3 thorpej * or the ksem_slock. To write the ks_name member, you need both. Make
103 1.3 thorpej * sure the order is ksem_slock -> ks_interlock.
104 1.3 thorpej */
105 1.1 christos struct ksem {
106 1.1 christos LIST_ENTRY(ksem) ks_entry; /* global list entry */
107 1.13 cube LIST_ENTRY(ksem) ks_hash; /* hash list entry */
108 1.3 thorpej struct simplelock ks_interlock; /* lock on this ksem */
109 1.1 christos char *ks_name; /* if named, this is the name */
110 1.3 thorpej unsigned int ks_ref; /* number of references */
111 1.1 christos mode_t ks_mode; /* protection bits */
112 1.1 christos uid_t ks_uid; /* creator uid */
113 1.1 christos gid_t ks_gid; /* creator gid */
114 1.1 christos unsigned int ks_value; /* current value */
115 1.3 thorpej unsigned int ks_waiters; /* number of waiters */
116 1.13 cube semid_t ks_id; /* unique identifier */
117 1.3 thorpej };
118 1.3 thorpej
119 1.3 thorpej struct ksem_ref {
120 1.3 thorpej LIST_ENTRY(ksem_ref) ksr_list;
121 1.3 thorpej struct ksem *ksr_ksem;
122 1.3 thorpej };
123 1.3 thorpej
124 1.3 thorpej struct ksem_proc {
125 1.3 thorpej struct lock kp_lock;
126 1.3 thorpej LIST_HEAD(, ksem_ref) kp_ksems;
127 1.1 christos };
128 1.1 christos
129 1.13 cube LIST_HEAD(ksem_list, ksem);
130 1.13 cube
131 1.1 christos /*
132 1.3 thorpej * ksem_slock protects ksem_head and nsems. Only named semaphores go
133 1.3 thorpej * onto ksem_head.
134 1.1 christos */
135 1.1 christos static struct simplelock ksem_slock;
136 1.13 cube static struct ksem_list ksem_head = LIST_HEAD_INITIALIZER(&ksem_head);
137 1.13 cube static struct ksem_list ksem_hash[SEM_HASHTBL_SIZE];
138 1.3 thorpej static int nsems = 0;
139 1.1 christos
140 1.13 cube /*
141 1.13 cube * ksem_counter is the last assigned semid_t. It needs to be COMPAT_NETBSD32
142 1.13 cube * friendly, even though semid_t itself is defined as uintptr_t.
143 1.13 cube */
144 1.13 cube static uint32_t ksem_counter = 1;
145 1.13 cube
146 1.16 thorpej static specificdata_key_t ksem_specificdata_key;
147 1.13 cube
148 1.3 thorpej static void
149 1.3 thorpej ksem_free(struct ksem *ks)
150 1.3 thorpej {
151 1.1 christos
152 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
153 1.3 thorpej /*
154 1.3 thorpej * If the ksem is anonymous (or has been unlinked), then
155 1.3 thorpej * this is the end if its life.
156 1.3 thorpej */
157 1.3 thorpej if (ks->ks_name == NULL) {
158 1.3 thorpej simple_unlock(&ks->ks_interlock);
159 1.1 christos
160 1.3 thorpej simple_lock(&ksem_slock);
161 1.3 thorpej nsems--;
162 1.13 cube LIST_REMOVE(ks, ks_hash);
163 1.3 thorpej simple_unlock(&ksem_slock);
164 1.13 cube
165 1.13 cube free(ks, M_SEM);
166 1.3 thorpej return;
167 1.3 thorpej }
168 1.3 thorpej simple_unlock(&ks->ks_interlock);
169 1.3 thorpej }
170 1.1 christos
171 1.12 perry static inline void
172 1.3 thorpej ksem_addref(struct ksem *ks)
173 1.1 christos {
174 1.1 christos
175 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
176 1.1 christos ks->ks_ref++;
177 1.3 thorpej KASSERT(ks->ks_ref != 0); /* XXX KDASSERT */
178 1.1 christos }
179 1.1 christos
180 1.12 perry static inline void
181 1.3 thorpej ksem_delref(struct ksem *ks)
182 1.1 christos {
183 1.1 christos
184 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
185 1.3 thorpej KASSERT(ks->ks_ref != 0); /* XXX KDASSERT */
186 1.3 thorpej if (--ks->ks_ref == 0) {
187 1.1 christos ksem_free(ks);
188 1.3 thorpej return;
189 1.3 thorpej }
190 1.3 thorpej simple_unlock(&ks->ks_interlock);
191 1.3 thorpej }
192 1.3 thorpej
193 1.3 thorpej static struct ksem_proc *
194 1.3 thorpej ksem_proc_alloc(void)
195 1.3 thorpej {
196 1.3 thorpej struct ksem_proc *kp;
197 1.3 thorpej
198 1.3 thorpej kp = malloc(sizeof(*kp), M_SEM, M_WAITOK);
199 1.3 thorpej lockinit(&kp->kp_lock, PWAIT, "ksproc", 0, 0);
200 1.3 thorpej LIST_INIT(&kp->kp_ksems);
201 1.3 thorpej
202 1.3 thorpej return (kp);
203 1.1 christos }
204 1.1 christos
205 1.3 thorpej static void
206 1.16 thorpej ksem_proc_dtor(void *arg)
207 1.16 thorpej {
208 1.16 thorpej struct ksem_proc *kp = arg;
209 1.16 thorpej struct ksem_ref *ksr;
210 1.16 thorpej
211 1.16 thorpej lockmgr(&kp->kp_lock, LK_DRAIN, NULL);
212 1.16 thorpej
213 1.16 thorpej while ((ksr = LIST_FIRST(&kp->kp_ksems)) != NULL) {
214 1.16 thorpej LIST_REMOVE(ksr, ksr_list);
215 1.16 thorpej simple_lock(&ksr->ksr_ksem->ks_interlock);
216 1.16 thorpej ksem_delref(ksr->ksr_ksem);
217 1.16 thorpej free(ksr, M_SEM);
218 1.16 thorpej }
219 1.16 thorpej
220 1.16 thorpej free(kp, M_SEM);
221 1.16 thorpej }
222 1.16 thorpej
223 1.16 thorpej static void
224 1.3 thorpej ksem_add_proc(struct proc *p, struct ksem *ks)
225 1.3 thorpej {
226 1.3 thorpej struct ksem_proc *kp;
227 1.3 thorpej struct ksem_ref *ksr;
228 1.3 thorpej
229 1.16 thorpej kp = proc_getspecific(p, ksem_specificdata_key);
230 1.16 thorpej if (kp == NULL) {
231 1.3 thorpej kp = ksem_proc_alloc();
232 1.16 thorpej proc_setspecific(p, ksem_specificdata_key, kp);
233 1.16 thorpej }
234 1.3 thorpej
235 1.3 thorpej ksr = malloc(sizeof(*ksr), M_SEM, M_WAITOK);
236 1.3 thorpej ksr->ksr_ksem = ks;
237 1.3 thorpej
238 1.3 thorpej lockmgr(&kp->kp_lock, LK_EXCLUSIVE, NULL);
239 1.3 thorpej LIST_INSERT_HEAD(&kp->kp_ksems, ksr, ksr_list);
240 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
241 1.3 thorpej }
242 1.3 thorpej
243 1.3 thorpej /* We MUST have a write lock on the ksem_proc list! */
244 1.3 thorpej static struct ksem_ref *
245 1.3 thorpej ksem_drop_proc(struct ksem_proc *kp, struct ksem *ks)
246 1.1 christos {
247 1.3 thorpej struct ksem_ref *ksr;
248 1.1 christos
249 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
250 1.3 thorpej LIST_FOREACH(ksr, &kp->kp_ksems, ksr_list) {
251 1.3 thorpej if (ksr->ksr_ksem == ks) {
252 1.3 thorpej ksem_delref(ks);
253 1.3 thorpej LIST_REMOVE(ksr, ksr_list);
254 1.3 thorpej return (ksr);
255 1.3 thorpej }
256 1.1 christos }
257 1.3 thorpej #ifdef DIAGNOSTIC
258 1.3 thorpej panic("ksem_drop_proc: ksem_proc %p ksem %p", kp, ks);
259 1.3 thorpej #endif
260 1.1 christos return (NULL);
261 1.1 christos }
262 1.1 christos
263 1.3 thorpej static int
264 1.15 ad ksem_perm(struct lwp *l, struct ksem *ks)
265 1.3 thorpej {
266 1.14 elad kauth_cred_t uc;
267 1.3 thorpej
268 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
269 1.15 ad uc = l->l_cred;
270 1.14 elad if ((kauth_cred_geteuid(uc) == ks->ks_uid && (ks->ks_mode & S_IWUSR) != 0) ||
271 1.14 elad (kauth_cred_getegid(uc) == ks->ks_gid && (ks->ks_mode & S_IWGRP) != 0) ||
272 1.14 elad (ks->ks_mode & S_IWOTH) != 0 ||
273 1.15 ad kauth_authorize_generic(uc, KAUTH_GENERIC_ISSUSER, &l->l_acflag) == 0)
274 1.3 thorpej return (0);
275 1.3 thorpej return (EPERM);
276 1.3 thorpej }
277 1.3 thorpej
278 1.1 christos static struct ksem *
279 1.13 cube ksem_lookup_byid(semid_t id)
280 1.13 cube {
281 1.13 cube struct ksem *ks;
282 1.13 cube
283 1.13 cube LOCK_ASSERT(simple_lock_held(&ksem_slock));
284 1.13 cube LIST_FOREACH(ks, &ksem_hash[SEM_HASH(id)], ks_hash) {
285 1.13 cube if (ks->ks_id == id)
286 1.13 cube return ks;
287 1.13 cube }
288 1.13 cube return NULL;
289 1.13 cube }
290 1.13 cube
291 1.13 cube static struct ksem *
292 1.3 thorpej ksem_lookup_byname(const char *name)
293 1.1 christos {
294 1.1 christos struct ksem *ks;
295 1.1 christos
296 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ksem_slock));
297 1.3 thorpej LIST_FOREACH(ks, &ksem_head, ks_entry) {
298 1.3 thorpej if (strcmp(ks->ks_name, name) == 0) {
299 1.3 thorpej simple_lock(&ks->ks_interlock);
300 1.1 christos return (ks);
301 1.3 thorpej }
302 1.3 thorpej }
303 1.1 christos return (NULL);
304 1.1 christos }
305 1.1 christos
306 1.1 christos static int
307 1.15 ad ksem_create(struct lwp *l, const char *name, struct ksem **ksret,
308 1.3 thorpej mode_t mode, unsigned int value)
309 1.1 christos {
310 1.1 christos struct ksem *ret;
311 1.14 elad kauth_cred_t uc;
312 1.1 christos size_t len;
313 1.1 christos
314 1.15 ad uc = l->l_cred;
315 1.1 christos if (value > SEM_VALUE_MAX)
316 1.1 christos return (EINVAL);
317 1.1 christos ret = malloc(sizeof(*ret), M_SEM, M_WAITOK | M_ZERO);
318 1.1 christos if (name != NULL) {
319 1.1 christos len = strlen(name);
320 1.1 christos if (len > SEM_MAX_NAMELEN) {
321 1.1 christos free(ret, M_SEM);
322 1.1 christos return (ENAMETOOLONG);
323 1.1 christos }
324 1.1 christos /* name must start with a '/' but not contain one. */
325 1.1 christos if (*name != '/' || len < 2 || strchr(name + 1, '/') != NULL) {
326 1.1 christos free(ret, M_SEM);
327 1.1 christos return (EINVAL);
328 1.1 christos }
329 1.1 christos ret->ks_name = malloc(len + 1, M_SEM, M_WAITOK);
330 1.6 itojun strlcpy(ret->ks_name, name, len + 1);
331 1.3 thorpej } else
332 1.1 christos ret->ks_name = NULL;
333 1.1 christos ret->ks_mode = mode;
334 1.1 christos ret->ks_value = value;
335 1.1 christos ret->ks_ref = 1;
336 1.1 christos ret->ks_waiters = 0;
337 1.14 elad ret->ks_uid = kauth_cred_geteuid(uc);
338 1.14 elad ret->ks_gid = kauth_cred_getegid(uc);
339 1.3 thorpej simple_lock_init(&ret->ks_interlock);
340 1.3 thorpej
341 1.1 christos simple_lock(&ksem_slock);
342 1.1 christos if (nsems >= SEM_MAX) {
343 1.3 thorpej simple_unlock(&ksem_slock);
344 1.3 thorpej if (ret->ks_name != NULL)
345 1.3 thorpej free(ret->ks_name, M_SEM);
346 1.3 thorpej free(ret, M_SEM);
347 1.3 thorpej return (ENFILE);
348 1.1 christos }
349 1.3 thorpej nsems++;
350 1.13 cube while (ksem_lookup_byid(ksem_counter) != NULL) {
351 1.13 cube ksem_counter++;
352 1.13 cube /* 0 is a special value for libpthread */
353 1.13 cube if (ksem_counter == 0)
354 1.13 cube ksem_counter++;
355 1.13 cube }
356 1.13 cube ret->ks_id = ksem_counter;
357 1.13 cube LIST_INSERT_HEAD(&ksem_hash[SEM_HASH(ret->ks_id)], ret, ks_hash);
358 1.1 christos simple_unlock(&ksem_slock);
359 1.3 thorpej
360 1.3 thorpej *ksret = ret;
361 1.3 thorpej return (0);
362 1.1 christos }
363 1.1 christos
364 1.1 christos int
365 1.2 christos sys__ksem_init(struct lwp *l, void *v, register_t *retval)
366 1.1 christos {
367 1.2 christos struct sys__ksem_init_args /* {
368 1.1 christos unsigned int value;
369 1.1 christos semid_t *idp;
370 1.1 christos } */ *uap = v;
371 1.13 cube
372 1.13 cube return do_ksem_init(l, SCARG(uap, value), SCARG(uap, idp), copyout);
373 1.13 cube }
374 1.13 cube
375 1.13 cube int
376 1.13 cube do_ksem_init(struct lwp *l, unsigned int value, semid_t *idp,
377 1.13 cube copyout_t docopyout)
378 1.13 cube {
379 1.1 christos struct ksem *ks;
380 1.1 christos semid_t id;
381 1.1 christos int error;
382 1.1 christos
383 1.3 thorpej /* Note the mode does not matter for anonymous semaphores. */
384 1.15 ad error = ksem_create(l, NULL, &ks, 0, value);
385 1.1 christos if (error)
386 1.1 christos return (error);
387 1.1 christos id = SEM_TO_ID(ks);
388 1.13 cube error = (*docopyout)(&id, idp, sizeof(id));
389 1.1 christos if (error) {
390 1.3 thorpej simple_lock(&ks->ks_interlock);
391 1.3 thorpej ksem_delref(ks);
392 1.1 christos return (error);
393 1.1 christos }
394 1.3 thorpej
395 1.3 thorpej ksem_add_proc(l->l_proc, ks);
396 1.3 thorpej
397 1.3 thorpej return (0);
398 1.1 christos }
399 1.1 christos
400 1.1 christos int
401 1.2 christos sys__ksem_open(struct lwp *l, void *v, register_t *retval)
402 1.1 christos {
403 1.2 christos struct sys__ksem_open_args /* {
404 1.1 christos const char *name;
405 1.1 christos int oflag;
406 1.1 christos mode_t mode;
407 1.1 christos unsigned int value;
408 1.10 perry semid_t *idp;
409 1.1 christos } */ *uap = v;
410 1.13 cube
411 1.13 cube return do_ksem_open(l, SCARG(uap, name), SCARG(uap, oflag),
412 1.13 cube SCARG(uap, mode), SCARG(uap, value), SCARG(uap, idp), copyout);
413 1.13 cube }
414 1.13 cube
415 1.13 cube int
416 1.13 cube do_ksem_open(struct lwp *l, const char *semname, int oflag, mode_t mode,
417 1.13 cube unsigned int value, semid_t *idp, copyout_t docopyout)
418 1.13 cube {
419 1.1 christos char name[SEM_MAX_NAMELEN + 1];
420 1.1 christos size_t done;
421 1.1 christos int error;
422 1.1 christos struct ksem *ksnew, *ks;
423 1.1 christos semid_t id;
424 1.1 christos
425 1.13 cube error = copyinstr(semname, name, sizeof(name), &done);
426 1.1 christos if (error)
427 1.1 christos return (error);
428 1.1 christos
429 1.1 christos ksnew = NULL;
430 1.1 christos simple_lock(&ksem_slock);
431 1.1 christos ks = ksem_lookup_byname(name);
432 1.3 thorpej
433 1.3 thorpej /* Found one? */
434 1.3 thorpej if (ks != NULL) {
435 1.3 thorpej /* Check for exclusive create. */
436 1.13 cube if (oflag & O_EXCL) {
437 1.3 thorpej simple_unlock(&ks->ks_interlock);
438 1.1 christos simple_unlock(&ksem_slock);
439 1.3 thorpej return (EEXIST);
440 1.1 christos }
441 1.3 thorpej found_one:
442 1.1 christos /*
443 1.3 thorpej * Verify permissions. If we can access it, add
444 1.3 thorpej * this process's reference.
445 1.1 christos */
446 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
447 1.15 ad error = ksem_perm(l, ks);
448 1.3 thorpej if (error == 0)
449 1.3 thorpej ksem_addref(ks);
450 1.3 thorpej simple_unlock(&ks->ks_interlock);
451 1.1 christos simple_unlock(&ksem_slock);
452 1.1 christos if (error)
453 1.1 christos return (error);
454 1.3 thorpej
455 1.1 christos id = SEM_TO_ID(ks);
456 1.13 cube error = (*docopyout)(&id, idp, sizeof(id));
457 1.1 christos if (error) {
458 1.3 thorpej simple_lock(&ks->ks_interlock);
459 1.3 thorpej ksem_delref(ks);
460 1.1 christos return (error);
461 1.1 christos }
462 1.3 thorpej
463 1.3 thorpej ksem_add_proc(l->l_proc, ks);
464 1.3 thorpej
465 1.3 thorpej return (0);
466 1.3 thorpej }
467 1.3 thorpej
468 1.3 thorpej /*
469 1.3 thorpej * didn't ask for creation? error.
470 1.3 thorpej */
471 1.13 cube if ((oflag & O_CREAT) == 0) {
472 1.1 christos simple_unlock(&ksem_slock);
473 1.3 thorpej return (ENOENT);
474 1.1 christos }
475 1.1 christos
476 1.3 thorpej /*
477 1.3 thorpej * We may block during creation, so drop the lock.
478 1.3 thorpej */
479 1.3 thorpej simple_unlock(&ksem_slock);
480 1.15 ad error = ksem_create(l, name, &ksnew, mode, value);
481 1.3 thorpej if (error != 0)
482 1.3 thorpej return (error);
483 1.3 thorpej
484 1.3 thorpej id = SEM_TO_ID(ksnew);
485 1.13 cube error = (*docopyout)(&id, idp, sizeof(id));
486 1.3 thorpej if (error) {
487 1.3 thorpej free(ksnew->ks_name, M_SEM);
488 1.3 thorpej ksnew->ks_name = NULL;
489 1.1 christos
490 1.3 thorpej simple_lock(&ksnew->ks_interlock);
491 1.3 thorpej ksem_delref(ksnew);
492 1.3 thorpej return (error);
493 1.3 thorpej }
494 1.1 christos
495 1.3 thorpej /*
496 1.3 thorpej * We need to make sure we haven't lost a race while
497 1.3 thorpej * allocating during creation.
498 1.3 thorpej */
499 1.3 thorpej simple_lock(&ksem_slock);
500 1.3 thorpej if ((ks = ksem_lookup_byname(name)) != NULL) {
501 1.13 cube if (oflag & O_EXCL) {
502 1.3 thorpej simple_unlock(&ks->ks_interlock);
503 1.3 thorpej simple_unlock(&ksem_slock);
504 1.1 christos
505 1.3 thorpej free(ksnew->ks_name, M_SEM);
506 1.3 thorpej ksnew->ks_name = NULL;
507 1.1 christos
508 1.3 thorpej simple_lock(&ksnew->ks_interlock);
509 1.3 thorpej ksem_delref(ksnew);
510 1.3 thorpej return (EEXIST);
511 1.3 thorpej }
512 1.3 thorpej goto found_one;
513 1.3 thorpej } else {
514 1.3 thorpej /* ksnew already has its initial reference. */
515 1.10 perry LIST_INSERT_HEAD(&ksem_head, ksnew, ks_entry);
516 1.3 thorpej simple_unlock(&ksem_slock);
517 1.1 christos
518 1.3 thorpej ksem_add_proc(l->l_proc, ksnew);
519 1.1 christos }
520 1.3 thorpej return (error);
521 1.1 christos }
522 1.1 christos
523 1.3 thorpej /* We must have a read lock on the ksem_proc list! */
524 1.3 thorpej static struct ksem *
525 1.3 thorpej ksem_lookup_proc(struct ksem_proc *kp, semid_t id)
526 1.1 christos {
527 1.3 thorpej struct ksem_ref *ksr;
528 1.1 christos
529 1.3 thorpej LIST_FOREACH(ksr, &kp->kp_ksems, ksr_list) {
530 1.13 cube if (id == SEM_TO_ID(ksr->ksr_ksem)) {
531 1.3 thorpej simple_lock(&ksr->ksr_ksem->ks_interlock);
532 1.3 thorpej return (ksr->ksr_ksem);
533 1.3 thorpej }
534 1.1 christos }
535 1.3 thorpej
536 1.3 thorpej return (NULL);
537 1.1 christos }
538 1.1 christos
539 1.1 christos int
540 1.2 christos sys__ksem_unlink(struct lwp *l, void *v, register_t *retval)
541 1.1 christos {
542 1.2 christos struct sys__ksem_unlink_args /* {
543 1.1 christos const char *name;
544 1.1 christos } */ *uap = v;
545 1.3 thorpej char name[SEM_MAX_NAMELEN + 1], *cp;
546 1.1 christos size_t done;
547 1.1 christos struct ksem *ks;
548 1.1 christos int error;
549 1.1 christos
550 1.1 christos error = copyinstr(SCARG(uap, name), name, sizeof(name), &done);
551 1.1 christos if (error)
552 1.1 christos return error;
553 1.1 christos
554 1.1 christos simple_lock(&ksem_slock);
555 1.1 christos ks = ksem_lookup_byname(name);
556 1.3 thorpej if (ks == NULL) {
557 1.3 thorpej simple_unlock(&ksem_slock);
558 1.3 thorpej return (ENOENT);
559 1.1 christos }
560 1.3 thorpej
561 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
562 1.3 thorpej
563 1.3 thorpej LIST_REMOVE(ks, ks_entry);
564 1.3 thorpej cp = ks->ks_name;
565 1.3 thorpej ks->ks_name = NULL;
566 1.3 thorpej
567 1.1 christos simple_unlock(&ksem_slock);
568 1.3 thorpej
569 1.3 thorpej if (ks->ks_ref == 0)
570 1.3 thorpej ksem_free(ks);
571 1.3 thorpej else
572 1.3 thorpej simple_unlock(&ks->ks_interlock);
573 1.3 thorpej
574 1.3 thorpej free(cp, M_SEM);
575 1.3 thorpej
576 1.3 thorpej return (0);
577 1.1 christos }
578 1.1 christos
579 1.1 christos int
580 1.2 christos sys__ksem_close(struct lwp *l, void *v, register_t *retval)
581 1.1 christos {
582 1.2 christos struct sys__ksem_close_args /* {
583 1.1 christos semid_t id;
584 1.1 christos } */ *uap = v;
585 1.3 thorpej struct ksem_proc *kp;
586 1.3 thorpej struct ksem_ref *ksr;
587 1.1 christos struct ksem *ks;
588 1.1 christos
589 1.16 thorpej kp = proc_getspecific(l->l_proc, ksem_specificdata_key);
590 1.16 thorpej if (kp == NULL)
591 1.3 thorpej return (EINVAL);
592 1.3 thorpej
593 1.3 thorpej lockmgr(&kp->kp_lock, LK_EXCLUSIVE, NULL);
594 1.3 thorpej
595 1.3 thorpej ks = ksem_lookup_proc(kp, SCARG(uap, id));
596 1.3 thorpej if (ks == NULL) {
597 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
598 1.3 thorpej return (EINVAL);
599 1.3 thorpej }
600 1.3 thorpej
601 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
602 1.3 thorpej if (ks->ks_name == NULL) {
603 1.3 thorpej simple_unlock(&ks->ks_interlock);
604 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
605 1.3 thorpej return (EINVAL);
606 1.3 thorpej }
607 1.3 thorpej
608 1.3 thorpej ksr = ksem_drop_proc(kp, ks);
609 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
610 1.3 thorpej free(ksr, M_SEM);
611 1.3 thorpej
612 1.3 thorpej return (0);
613 1.1 christos }
614 1.1 christos
615 1.1 christos int
616 1.2 christos sys__ksem_post(struct lwp *l, void *v, register_t *retval)
617 1.1 christos {
618 1.2 christos struct sys__ksem_post_args /* {
619 1.1 christos semid_t id;
620 1.1 christos } */ *uap = v;
621 1.3 thorpej struct ksem_proc *kp;
622 1.1 christos struct ksem *ks;
623 1.1 christos int error;
624 1.1 christos
625 1.16 thorpej kp = proc_getspecific(l->l_proc, ksem_specificdata_key);
626 1.16 thorpej if (kp == NULL)
627 1.3 thorpej return (EINVAL);
628 1.3 thorpej
629 1.3 thorpej lockmgr(&kp->kp_lock, LK_SHARED, NULL);
630 1.3 thorpej ks = ksem_lookup_proc(kp, SCARG(uap, id));
631 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
632 1.3 thorpej if (ks == NULL)
633 1.3 thorpej return (EINVAL);
634 1.3 thorpej
635 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
636 1.1 christos if (ks->ks_value == SEM_VALUE_MAX) {
637 1.1 christos error = EOVERFLOW;
638 1.3 thorpej goto out;
639 1.1 christos }
640 1.1 christos ++ks->ks_value;
641 1.3 thorpej if (ks->ks_waiters)
642 1.1 christos wakeup(ks);
643 1.1 christos error = 0;
644 1.3 thorpej out:
645 1.3 thorpej simple_unlock(&ks->ks_interlock);
646 1.3 thorpej return (error);
647 1.3 thorpej }
648 1.3 thorpej
649 1.3 thorpej static int
650 1.3 thorpej ksem_wait(struct lwp *l, semid_t id, int tryflag)
651 1.3 thorpej {
652 1.3 thorpej struct ksem_proc *kp;
653 1.3 thorpej struct ksem *ks;
654 1.3 thorpej int error;
655 1.3 thorpej
656 1.16 thorpej kp = proc_getspecific(l->l_proc, ksem_specificdata_key);
657 1.16 thorpej if (kp == NULL)
658 1.3 thorpej return (EINVAL);
659 1.3 thorpej
660 1.3 thorpej lockmgr(&kp->kp_lock, LK_SHARED, NULL);
661 1.3 thorpej ks = ksem_lookup_proc(kp, id);
662 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
663 1.3 thorpej if (ks == NULL)
664 1.3 thorpej return (EINVAL);
665 1.3 thorpej
666 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
667 1.3 thorpej ksem_addref(ks);
668 1.3 thorpej while (ks->ks_value == 0) {
669 1.3 thorpej ks->ks_waiters++;
670 1.10 perry error = tryflag ? EAGAIN : ltsleep(ks, PCATCH, "psem", 0,
671 1.3 thorpej &ks->ks_interlock);
672 1.3 thorpej ks->ks_waiters--;
673 1.3 thorpej if (error)
674 1.3 thorpej goto out;
675 1.3 thorpej }
676 1.3 thorpej ks->ks_value--;
677 1.3 thorpej error = 0;
678 1.3 thorpej out:
679 1.3 thorpej ksem_delref(ks);
680 1.1 christos return (error);
681 1.1 christos }
682 1.1 christos
683 1.1 christos int
684 1.2 christos sys__ksem_wait(struct lwp *l, void *v, register_t *retval)
685 1.1 christos {
686 1.2 christos struct sys__ksem_wait_args /* {
687 1.1 christos semid_t id;
688 1.1 christos } */ *uap = v;
689 1.1 christos
690 1.1 christos return ksem_wait(l, SCARG(uap, id), 0);
691 1.1 christos }
692 1.1 christos
693 1.1 christos int
694 1.2 christos sys__ksem_trywait(struct lwp *l, void *v, register_t *retval)
695 1.1 christos {
696 1.2 christos struct sys__ksem_trywait_args /* {
697 1.1 christos semid_t id;
698 1.1 christos } */ *uap = v;
699 1.1 christos
700 1.1 christos return ksem_wait(l, SCARG(uap, id), 1);
701 1.1 christos }
702 1.1 christos
703 1.1 christos int
704 1.2 christos sys__ksem_getvalue(struct lwp *l, void *v, register_t *retval)
705 1.1 christos {
706 1.2 christos struct sys__ksem_getvalue_args /* {
707 1.1 christos semid_t id;
708 1.1 christos unsigned int *value;
709 1.1 christos } */ *uap = v;
710 1.3 thorpej struct ksem_proc *kp;
711 1.1 christos struct ksem *ks;
712 1.1 christos unsigned int val;
713 1.1 christos
714 1.16 thorpej kp = proc_getspecific(l->l_proc, ksem_specificdata_key);
715 1.16 thorpej if (kp == NULL)
716 1.3 thorpej return (EINVAL);
717 1.3 thorpej
718 1.3 thorpej lockmgr(&kp->kp_lock, LK_SHARED, NULL);
719 1.3 thorpej ks = ksem_lookup_proc(kp, SCARG(uap, id));
720 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
721 1.3 thorpej if (ks == NULL)
722 1.1 christos return (EINVAL);
723 1.3 thorpej
724 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
725 1.1 christos val = ks->ks_value;
726 1.3 thorpej simple_unlock(&ks->ks_interlock);
727 1.3 thorpej
728 1.3 thorpej return (copyout(&val, SCARG(uap, value), sizeof(val)));
729 1.1 christos }
730 1.1 christos
731 1.1 christos int
732 1.2 christos sys__ksem_destroy(struct lwp *l, void *v, register_t *retval)
733 1.1 christos {
734 1.2 christos struct sys__ksem_destroy_args /*{
735 1.1 christos semid_t id;
736 1.1 christos } */ *uap = v;
737 1.3 thorpej struct ksem_proc *kp;
738 1.3 thorpej struct ksem_ref *ksr;
739 1.1 christos struct ksem *ks;
740 1.1 christos
741 1.16 thorpej kp = proc_getspecific(l->l_proc, ksem_specificdata_key);
742 1.16 thorpej if (kp == NULL)
743 1.3 thorpej return (EINVAL);
744 1.3 thorpej
745 1.3 thorpej lockmgr(&kp->kp_lock, LK_EXCLUSIVE, NULL);
746 1.3 thorpej
747 1.3 thorpej ks = ksem_lookup_proc(kp, SCARG(uap, id));
748 1.3 thorpej if (ks == NULL) {
749 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
750 1.3 thorpej return (EINVAL);
751 1.3 thorpej }
752 1.3 thorpej
753 1.3 thorpej LOCK_ASSERT(simple_lock_held(&ks->ks_interlock));
754 1.3 thorpej
755 1.3 thorpej /*
756 1.3 thorpej * XXX This misses named semaphores which have been unlink'd,
757 1.3 thorpej * XXX but since behavior of destroying a named semaphore is
758 1.3 thorpej * XXX undefined, this is technically allowed.
759 1.3 thorpej */
760 1.3 thorpej if (ks->ks_name != NULL) {
761 1.3 thorpej simple_unlock(&ks->ks_interlock);
762 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
763 1.3 thorpej return (EINVAL);
764 1.3 thorpej }
765 1.3 thorpej
766 1.3 thorpej if (ks->ks_waiters) {
767 1.3 thorpej simple_unlock(&ks->ks_interlock);
768 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
769 1.3 thorpej return (EBUSY);
770 1.3 thorpej }
771 1.3 thorpej
772 1.3 thorpej ksr = ksem_drop_proc(kp, ks);
773 1.3 thorpej lockmgr(&kp->kp_lock, LK_RELEASE, NULL);
774 1.3 thorpej free(ksr, M_SEM);
775 1.3 thorpej
776 1.3 thorpej return (0);
777 1.3 thorpej }
778 1.3 thorpej
779 1.3 thorpej static void
780 1.3 thorpej ksem_forkhook(struct proc *p2, struct proc *p1)
781 1.3 thorpej {
782 1.3 thorpej struct ksem_proc *kp1, *kp2;
783 1.3 thorpej struct ksem_ref *ksr, *ksr1;
784 1.3 thorpej
785 1.16 thorpej kp1 = proc_getspecific(p1, ksem_specificdata_key);
786 1.16 thorpej if (kp1 == NULL)
787 1.3 thorpej return;
788 1.3 thorpej
789 1.16 thorpej kp2 = ksem_proc_alloc();
790 1.3 thorpej
791 1.3 thorpej lockmgr(&kp1->kp_lock, LK_SHARED, NULL);
792 1.3 thorpej
793 1.3 thorpej if (!LIST_EMPTY(&kp1->kp_ksems)) {
794 1.3 thorpej LIST_FOREACH(ksr, &kp1->kp_ksems, ksr_list) {
795 1.3 thorpej ksr1 = malloc(sizeof(*ksr), M_SEM, M_WAITOK);
796 1.3 thorpej ksr1->ksr_ksem = ksr->ksr_ksem;
797 1.3 thorpej simple_lock(&ksr->ksr_ksem->ks_interlock);
798 1.3 thorpej ksem_addref(ksr->ksr_ksem);
799 1.3 thorpej simple_unlock(&ksr->ksr_ksem->ks_interlock);
800 1.3 thorpej LIST_INSERT_HEAD(&kp2->kp_ksems, ksr1, ksr_list);
801 1.3 thorpej }
802 1.1 christos }
803 1.3 thorpej
804 1.3 thorpej lockmgr(&kp1->kp_lock, LK_RELEASE, NULL);
805 1.16 thorpej
806 1.16 thorpej proc_setspecific(p2, ksem_specificdata_key, kp2);
807 1.1 christos }
808 1.1 christos
809 1.1 christos static void
810 1.16 thorpej ksem_exechook(struct proc *p, void *arg)
811 1.1 christos {
812 1.3 thorpej struct ksem_proc *kp;
813 1.1 christos
814 1.16 thorpej kp = proc_getspecific(p, ksem_specificdata_key);
815 1.16 thorpej if (kp != NULL) {
816 1.16 thorpej proc_setspecific(p, ksem_specificdata_key, NULL);
817 1.16 thorpej ksem_proc_dtor(kp);
818 1.1 christos }
819 1.1 christos }
820 1.1 christos
821 1.1 christos void
822 1.1 christos ksem_init(void)
823 1.1 christos {
824 1.16 thorpej int i, error;
825 1.3 thorpej
826 1.1 christos simple_lock_init(&ksem_slock);
827 1.16 thorpej exechook_establish(ksem_exechook, NULL);
828 1.3 thorpej forkhook_establish(ksem_forkhook);
829 1.13 cube
830 1.13 cube for (i = 0; i < SEM_HASHTBL_SIZE; i++)
831 1.13 cube LIST_INIT(&ksem_hash[i]);
832 1.16 thorpej
833 1.16 thorpej error = proc_specific_key_create(&ksem_specificdata_key,
834 1.16 thorpej ksem_proc_dtor);
835 1.16 thorpej KASSERT(error == 0);
836 1.1 christos }
837