1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*
* Copyright (c) 2006-2023, RT-Thread Development Team
*
* SPDX-License-Identifier: Apache-2.0
*
* Change Logs:
* Date Author Notes
* 2023-12-07 Shell First version
*/
#include "rid_bitmap.h"
#include <bitmap.h>
#include <rtdef.h>
void rid_bitmap_init(rid_bitmap_t mgr, int min_id, int total_id_count,
rt_bitmap_t *set, struct rt_mutex *id_lock)
{
mgr->min_id = min_id;
mgr->total_id_count = total_id_count;
mgr->bitset = set;
mgr->id_lock = id_lock;
return;
}
long rid_bitmap_get(rid_bitmap_t mgr)
{
long id;
long overflow;
if (mgr->id_lock)
{
rt_mutex_take(mgr->id_lock, RT_WAITING_FOREVER);
}
overflow = mgr->total_id_count;
id = rt_bitmap_next_clear_bit(mgr->bitset, 0, overflow);
if (id == overflow)
{
id = -1;
}
else
{
rt_bitmap_set_bit(mgr->bitset, id);
id += mgr->min_id;
}
if (mgr->id_lock)
{
rt_mutex_release(mgr->id_lock);
}
return id;
}
long rid_bitmap_get_named(rid_bitmap_t mgr, long no)
{
long id_relative;
long overflow;
long min;
if (mgr->id_lock)
{
rt_mutex_take(mgr->id_lock, RT_WAITING_FOREVER);
}
min = mgr->min_id;
id_relative = no - min;
overflow = mgr->total_id_count;
if (id_relative >= min && id_relative < overflow)
{
if (rt_bitmap_test_bit(mgr->bitset, id_relative))
{
id_relative = -RT_EBUSY;
}
else
{
rt_bitmap_set_bit(mgr->bitset, id_relative);
id_relative += min;
}
}
else
{
id_relative = -1;
}
if (mgr->id_lock)
{
rt_mutex_release(mgr->id_lock);
}
return id_relative;
}
void rid_bitmap_put(rid_bitmap_t mgr, long no)
{
long id_relative;
long overflow;
long min;
if (mgr->id_lock)
{
rt_mutex_take(mgr->id_lock, RT_WAITING_FOREVER);
}
min = mgr->min_id;
id_relative = no - min;
overflow = mgr->total_id_count;
if (id_relative >= min && id_relative < overflow &&
rt_bitmap_test_bit(mgr->bitset, id_relative))
{
rt_bitmap_clear_bit(mgr->bitset, id_relative);
}
if (mgr->id_lock)
{
rt_mutex_release(mgr->id_lock);
}
}
|
