| @@ -1,14 +1,14 @@ | | | @@ -1,14 +1,14 @@ |
1 | /* $NetBSD: once.h,v 1.8 2023/09/08 23:25:39 riastradh Exp $ */ | | 1 | /* $NetBSD: once.h,v 1.9 2023/09/09 00:15:09 riastradh Exp $ */ |
2 | | | 2 | |
3 | /*- | | 3 | /*- |
4 | * Copyright (c)2005 YAMAMOTO Takashi, | | 4 | * Copyright (c)2005 YAMAMOTO Takashi, |
5 | * Copyright (c)2008 Antti Kantee, | | 5 | * Copyright (c)2008 Antti Kantee, |
6 | * All rights reserved. | | 6 | * All rights reserved. |
7 | * | | 7 | * |
8 | * Redistribution and use in source and binary forms, with or without | | 8 | * Redistribution and use in source and binary forms, with or without |
9 | * modification, are permitted provided that the following conditions | | 9 | * modification, are permitted provided that the following conditions |
10 | * are met: | | 10 | * are met: |
11 | * 1. Redistributions of source code must retain the above copyright | | 11 | * 1. Redistributions of source code must retain the above copyright |
12 | * notice, this list of conditions and the following disclaimer. | | 12 | * notice, this list of conditions and the following disclaimer. |
13 | * 2. Redistributions in binary form must reproduce the above copyright | | 13 | * 2. Redistributions in binary form must reproduce the above copyright |
14 | * notice, this list of conditions and the following disclaimer in the | | 14 | * notice, this list of conditions and the following disclaimer in the |
| @@ -20,43 +20,40 @@ | | | @@ -20,43 +20,40 @@ |
20 | * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE | | 20 | * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE |
21 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | | 21 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
22 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | | 22 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
23 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | | 23 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
24 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | | 24 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
25 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | | 25 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
26 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | | 26 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
27 | * SUCH DAMAGE. | | 27 | * SUCH DAMAGE. |
28 | */ | | 28 | */ |
29 | | | 29 | |
30 | #ifndef _SYS_ONCE_H_ | | 30 | #ifndef _SYS_ONCE_H_ |
31 | #define _SYS_ONCE_H_ | | 31 | #define _SYS_ONCE_H_ |
32 | | | 32 | |
33 | #include <sys/stdint.h> | | | |
34 | #include <sys/systm.h> | | | |
35 | | | | |
36 | typedef struct { | | 33 | typedef struct { |
37 | int o_error; | | 34 | int o_error; |
38 | uint16_t o_refcnt; | | 35 | uint16_t o_refcnt; |
39 | uint16_t o_status; | | 36 | uint16_t o_status; |
40 | #define ONCE_VIRGIN 0 | | 37 | #define ONCE_VIRGIN 0 |
41 | #define ONCE_RUNNING 1 | | 38 | #define ONCE_RUNNING 1 |
42 | #define ONCE_DONE 2 | | 39 | #define ONCE_DONE 2 |
43 | } once_t; | | 40 | } once_t; |
44 | | | 41 | |
45 | void once_init(void); | | 42 | void once_init(void); |
46 | int _init_once(once_t *, int (*)(void)); | | 43 | int _init_once(once_t *, int (*)(void)); |
47 | void _fini_once(once_t *, void (*)(void)); | | 44 | void _fini_once(once_t *, void (*)(void)); |
48 | | | 45 | |
49 | #define ONCE_DECL(o) \ | | 46 | #define ONCE_DECL(o) \ |
50 | once_t (o) = { \ | | 47 | once_t (o) = { \ |
51 | .o_status = 0, \ | | 48 | .o_status = 0, \ |
52 | .o_refcnt = 0, \ | | 49 | .o_refcnt = 0, \ |
53 | }; | | 50 | }; |
54 | | | 51 | |
55 | #define RUN_ONCE(o, fn) \ | | 52 | #define RUN_ONCE(o, fn) \ |
56 | (ASSERT_SLEEPABLE(), __predict_true((o)->o_status == ONCE_DONE) ? \ | | 53 | (__predict_true((o)->o_status == ONCE_DONE) ? \ |
57 | ((o)->o_error) : _init_once((o), (fn))) | | 54 | ((o)->o_error) : _init_once((o), (fn))) |
58 | | | 55 | |
59 | #define INIT_ONCE(o, fn) _init_once((o), (fn)) | | 56 | #define INIT_ONCE(o, fn) _init_once((o), (fn)) |
60 | #define FINI_ONCE(o, fn) _fini_once((o), (fn)) | | 57 | #define FINI_ONCE(o, fn) _fini_once((o), (fn)) |
61 | | | 58 | |
62 | #endif /* _SYS_ONCE_H_ */ | | 59 | #endif /* _SYS_ONCE_H_ */ |