ObjFW  Diff

Differences From Artifact [6f5c36ea25]:

To Artifact [da0a13a473]:


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
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







+
-
+
-
-
+
-




-
-
+
+



-
-
+
+





-
-
+
+



-
+









-
+


-
+







-
+








-
+




-
+








#include <stdio.h>
#include <stdlib.h>

#import "runtime.h"
#import "runtime-private.h"

#import "globals.h"
static struct objc_dtable_level2 *empty_level2 = NULL;
#define empty_dtable_level2 objc_globals.empty_dtable_level2
#ifdef OF_SELUID24
static struct objc_dtable_level3 *empty_level3 = NULL;
#define empty_dtable_level3 objc_globals.empty_dtable_level3
#endif

static void
init(void)
{
	empty_level2 = malloc(sizeof(struct objc_dtable_level2));
	if (empty_level2 == NULL)
	empty_dtable_level2 = malloc(sizeof(struct objc_dtable_level2));
	if (empty_dtable_level2 == NULL)
		OBJC_ERROR("Not enough memory to allocate dtable!");

#ifdef OF_SELUID24
	empty_level3 = malloc(sizeof(struct objc_dtable_level3));
	if (empty_level3 == NULL)
	empty_dtable_level3 = malloc(sizeof(struct objc_dtable_level3));
	if (empty_dtable_level3 == NULL)
		OBJC_ERROR("Not enough memory to allocate dtable!");
#endif

#ifdef OF_SELUID24
	for (uint_fast16_t i = 0; i < 256; i++) {
		empty_level2->buckets[i] = empty_level3;
		empty_level3->buckets[i] = (IMP)0;
		empty_dtable_level2->buckets[i] = empty_dtable_level3;
		empty_dtable_level3->buckets[i] = (IMP)0;
	}
#else
	for (uint_fast16_t i = 0; i < 256; i++)
		empty_level2->buckets[i] = (IMP)0;
		empty_dtable_level2->buckets[i] = (IMP)0;
#endif
}

struct objc_dtable *
objc_dtable_new(void)
{
	struct objc_dtable *dtable;

#ifdef OF_SELUID24
	if (empty_level2 == NULL || empty_level3 == NULL)
	if (empty_dtable_level2 == NULL || empty_dtable_level3 == NULL)
		init();
#else
	if (empty_level2 == NULL)
	if (empty_dtable_level2 == NULL)
		init();
#endif

	if ((dtable = malloc(sizeof(struct objc_dtable))) == NULL)
		OBJC_ERROR("Not enough memory to allocate dtable!");

	for (uint_fast16_t i = 0; i < 256; i++)
		dtable->buckets[i] = empty_level2;
		dtable->buckets[i] = empty_dtable_level2;

	return dtable;
}

void
objc_dtable_copy(struct objc_dtable *dst, struct objc_dtable *src)
{
	for (uint_fast16_t i = 0; i < 256; i++) {
		if (src->buckets[i] == empty_level2)
		if (src->buckets[i] == empty_dtable_level2)
			continue;

#ifdef OF_SELUID24
		for (uint_fast16_t j = 0; j < 256; j++) {
			if (src->buckets[i]->buckets[j] == empty_level3)
			if (src->buckets[i]->buckets[j] == empty_dtable_level3)
				continue;

			for (uint_fast16_t k = 0; k < 256; k++) {
				IMP obj;
				uint32_t idx;

				obj = src->buckets[i]->buckets[j]->buckets[k];
122
123
124
125
126
127
128
129

130
131
132
133
134
135
136
137
138

139
140
141
142
143
144
145
146
147

148
149
150
151
152
153
154
121
122
123
124
125
126
127

128
129
130
131
132
133
134
135
136

137
138
139
140
141
142
143
144
145

146
147
148
149
150
151
152
153







-
+








-
+








-
+







	uint8_t j = idx >> 8;
	uint8_t k = idx;
#else
	uint8_t i = idx >> 8;
	uint8_t j = idx;
#endif

	if (dtable->buckets[i] == empty_level2) {
	if (dtable->buckets[i] == empty_dtable_level2) {
		struct objc_dtable_level2 *level2 =
		    malloc(sizeof(struct objc_dtable_level2));

		if (level2 == NULL)
			OBJC_ERROR("Not enough memory to insert into dtable!");

		for (uint_fast16_t l = 0; l < 256; l++)
#ifdef OF_SELUID24
			level2->buckets[l] = empty_level3;
			level2->buckets[l] = empty_dtable_level3;
#else
			level2->buckets[l] = (IMP)0;
#endif

		dtable->buckets[i] = level2;
	}

#ifdef OF_SELUID24
	if (dtable->buckets[i]->buckets[j] == empty_level3) {
	if (dtable->buckets[i]->buckets[j] == empty_dtable_level3) {
		struct objc_dtable_level3 *level3 =
		    malloc(sizeof(struct objc_dtable_level3));

		if (level3 == NULL)
			OBJC_ERROR("Not enough memory to insert into dtable!");

		for (uint_fast16_t l = 0; l < 256; l++)
163
164
165
166
167
168
169
170

171
172
173
174
175


176
177
178
179
180
181
182
183
184
185
186
187
188
189


190
191
192


193
194
195

196
197

198
199
162
163
164
165
166
167
168

169
170
171
172
173

174
175
176
177
178
179
180
181
182
183
184
185
186
187


188
189
190


191
192
193
194

195
196

197
198
199







-
+




-
+
+












-
-
+
+

-
-
+
+


-
+

-
+


#endif
}

void
objc_dtable_free(struct objc_dtable *dtable)
{
	for (uint_fast16_t i = 0; i < 256; i++) {
		if (dtable->buckets[i] == empty_level2)
		if (dtable->buckets[i] == empty_dtable_level2)
			continue;

#ifdef OF_SELUID24
		for (uint_fast16_t j = 0; j < 256; j++)
			if (dtable->buckets[i]->buckets[j] != empty_level3)
			if (dtable->buckets[i]->buckets[j] !=
			    empty_dtable_level3)
				free(dtable->buckets[i]->buckets[j]);
#endif

		free(dtable->buckets[i]);
	}

	free(dtable);
}

void
objc_dtable_cleanup(void)
{
	if (empty_level2 != NULL)
		free(empty_level2);
	if (empty_dtable_level2 != NULL)
		free(empty_dtable_level2);
#ifdef OF_SELUID24
	if (empty_level3 != NULL)
		free(empty_level3);
	if (empty_dtable_level3 != NULL)
		free(empty_dtable_level3);
#endif

	empty_level2 = NULL;
	empty_dtable_level2 = NULL;
#ifdef OF_SELUID24
	empty_level3 = NULL;
	empty_dtable_level3 = NULL;
#endif
}