1 | /*
|
---|
2 | * Copyright (c) 2025 Miroslav Cimerman
|
---|
3 | * Copyright (c) 2024 Vojtech Horky
|
---|
4 | * All rights reserved.
|
---|
5 | *
|
---|
6 | * Redistribution and use in source and binary forms, with or without
|
---|
7 | * modification, are permitted provided that the following conditions
|
---|
8 | * are met:
|
---|
9 | *
|
---|
10 | * - Redistributions of source code must retain the above copyright
|
---|
11 | * notice, this list of conditions and the following disclaimer.
|
---|
12 | * - Redistributions in binary form must reproduce the above copyright
|
---|
13 | * notice, this list of conditions and the following disclaimer in the
|
---|
14 | * documentation and/or other materials provided with the distribution.
|
---|
15 | * - The name of the author may not be used to endorse or promote products
|
---|
16 | * derived from this software without specific prior written permission.
|
---|
17 | *
|
---|
18 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
---|
19 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
---|
20 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
---|
21 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
---|
22 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
---|
23 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
---|
24 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
---|
25 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
---|
26 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
---|
27 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
28 | */
|
---|
29 |
|
---|
30 | /** @addtogroup hr
|
---|
31 | * @{
|
---|
32 | */
|
---|
33 | /**
|
---|
34 | * @file
|
---|
35 | */
|
---|
36 |
|
---|
37 | #ifndef _HR_FGE_H
|
---|
38 | #define _HR_FGE_H
|
---|
39 |
|
---|
40 | #include <adt/bitmap.h>
|
---|
41 | #include <adt/circ_buf.h>
|
---|
42 | #include <errno.h>
|
---|
43 | #include <stddef.h>
|
---|
44 |
|
---|
45 | /* forward declarations */
|
---|
46 | typedef struct hr_fpool hr_fpool_t;
|
---|
47 | typedef struct hr_fgroup hr_fgroup_t;
|
---|
48 | typedef struct fge_fibril_data fge_fibril_data_t;
|
---|
49 | typedef struct wu_queue wu_queue_t;
|
---|
50 |
|
---|
51 | typedef errno_t (*hr_wu_t)(void *);
|
---|
52 |
|
---|
53 | struct fge_fibril_data {
|
---|
54 | hr_wu_t wu; /* work unit function pointer */
|
---|
55 | void *arg; /* work unit function argument */
|
---|
56 | hr_fgroup_t *group; /* back-pointer to group */
|
---|
57 | ssize_t memslot; /* index to pool bitmap slot */
|
---|
58 | };
|
---|
59 |
|
---|
60 | struct wu_queue {
|
---|
61 | fibril_mutex_t lock;
|
---|
62 | fibril_condvar_t not_empty;
|
---|
63 | fibril_condvar_t not_full;
|
---|
64 | fge_fibril_data_t *fexecs; /* circ-buf memory */
|
---|
65 | circ_buf_t cbuf;
|
---|
66 | };
|
---|
67 |
|
---|
68 | struct hr_fpool {
|
---|
69 | fibril_mutex_t lock;
|
---|
70 | bitmap_t bitmap; /* memory slot bitmap */
|
---|
71 | wu_queue_t queue;
|
---|
72 | fid_t *fibrils;
|
---|
73 | uint8_t *wu_storage; /* pre-allocated pool storage */
|
---|
74 | size_t fibril_cnt;
|
---|
75 | size_t max_wus;
|
---|
76 | size_t active_groups;
|
---|
77 | bool stop;
|
---|
78 | size_t wu_size;
|
---|
79 | size_t wu_storage_free_count;
|
---|
80 | fibril_condvar_t all_wus_done;
|
---|
81 | };
|
---|
82 |
|
---|
83 | struct hr_fgroup {
|
---|
84 | hr_fpool_t *pool;/* back-pointer to pool */
|
---|
85 | size_t wu_cnt;/* upper bound of work units */
|
---|
86 | size_t submitted; /* number of submitted jobs */
|
---|
87 | size_t reserved_cnt; /* no. of reserved wu storage slots */
|
---|
88 | size_t reserved_avail;
|
---|
89 | size_t *memslots; /* indices to pool bitmap */
|
---|
90 | void *own_mem; /* own allocated memory */
|
---|
91 | size_t own_used; /* own memory slots used counter */
|
---|
92 | errno_t final_errno; /* agreggated errno */
|
---|
93 | size_t finished_okay; /* no. of wus that ended with EOK */
|
---|
94 | size_t finished_fail; /* no. of wus that ended with != EOK */
|
---|
95 | fibril_mutex_t lock;
|
---|
96 | fibril_condvar_t all_done;
|
---|
97 | };
|
---|
98 |
|
---|
99 | extern hr_fpool_t *hr_fpool_create(size_t, size_t, size_t);
|
---|
100 | extern void hr_fpool_destroy(hr_fpool_t *);
|
---|
101 | extern hr_fgroup_t *hr_fgroup_create(hr_fpool_t *, size_t);
|
---|
102 | extern void *hr_fgroup_alloc(hr_fgroup_t *);
|
---|
103 | extern void hr_fgroup_submit(hr_fgroup_t *, hr_wu_t, void *);
|
---|
104 | extern errno_t hr_fgroup_wait(hr_fgroup_t *, size_t *, size_t *);
|
---|
105 |
|
---|
106 | #endif
|
---|
107 |
|
---|
108 | /** @}
|
---|
109 | */
|
---|