1 /*
2 * Copyright 2015 Rockchip Electronics Co. LTD
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #define MODULE_TAG "mpp_buffer_test"
18
19 #include <string.h>
20
21 #if defined(_WIN32)
22 #include "vld.h"
23 #endif
24 #include "mpp_env.h"
25 #include "mpp_debug.h"
26 #include "mpp_common.h"
27 #include "mpp_buffer.h"
28 #include "mpp_allocator.h"
29
30 #define MPP_BUFFER_TEST_DEBUG_FLAG (0xf)
31 #define MPP_BUFFER_TEST_SIZE (SZ_1K*4)
32 #define MPP_BUFFER_TEST_COMMIT_COUNT 10
33 #define MPP_BUFFER_TEST_NORMAL_COUNT 10
34
main()35 int main()
36 {
37 MPP_RET ret = MPP_OK;
38 MppAllocator allocator = NULL;
39 MppAllocatorApi *api = NULL;
40 MppBufferInfo commit;
41 MppBufferGroup group = NULL;
42 MppBuffer commit_buffer[MPP_BUFFER_TEST_COMMIT_COUNT];
43 void *commit_ptr[MPP_BUFFER_TEST_COMMIT_COUNT];
44 MppBuffer normal_buffer[MPP_BUFFER_TEST_NORMAL_COUNT];
45 MppBuffer legacy_buffer = NULL;
46 size_t size = MPP_BUFFER_TEST_SIZE;
47 RK_S32 count = MPP_BUFFER_TEST_COMMIT_COUNT;
48 RK_S32 i;
49 RK_U32 debug = 0;
50
51 mpp_env_set_u32("mpp_buffer_debug", MPP_BUFFER_TEST_DEBUG_FLAG);
52 mpp_env_get_u32("mpp_buffer_debug", &debug, 0);
53
54 mpp_log("mpp_buffer_test start with debug 0x%x\n", debug);
55
56 memset(commit_ptr, 0, sizeof(commit_ptr));
57 memset(commit_buffer, 0, sizeof(commit_buffer));
58 memset(normal_buffer, 0, sizeof(normal_buffer));
59
60 // create group with external type
61 ret = mpp_buffer_group_get_external(&group, MPP_BUFFER_TYPE_ION);
62 if (MPP_OK != ret) {
63 mpp_err("mpp_buffer_test mpp_buffer_group_get failed\n");
64 goto MPP_BUFFER_failed;
65 }
66
67 mpp_log("mpp_buffer_test commit mode with unused status start\n");
68
69 commit.type = MPP_BUFFER_TYPE_ION;
70 commit.size = size;
71
72 // create misc ion buffer and commit to external group
73 for (i = 0; i < count; i++) {
74 // can be change to different buffer allocator here
75 ret = mpp_buffer_get(NULL, &normal_buffer[i], size);
76 if (ret || NULL == normal_buffer[i]) {
77 mpp_err("mpp_buffer_test get misc buffer failed ret %d\n", ret);
78 goto MPP_BUFFER_failed;
79 }
80
81 // NOTE: setup fd / index in necessary
82 commit.ptr = mpp_buffer_get_ptr(normal_buffer[i]);
83 commit.fd = mpp_buffer_get_fd(normal_buffer[i]);
84 commit.index = i;
85
86 ret = mpp_buffer_commit(group, &commit);
87 if (MPP_OK != ret) {
88 mpp_err("mpp_buffer_test mpp_buffer_commit failed\n");
89 goto MPP_BUFFER_failed;
90 }
91 }
92
93 for (i = 0; i < count; i++) {
94 ret = mpp_buffer_get(group, &commit_buffer[i], size);
95 if (MPP_OK != ret) {
96 mpp_err("mpp_buffer_test mpp_buffer_get commit mode failed\n");
97 goto MPP_BUFFER_failed;
98 }
99 }
100
101 for (i = 0; i < count; i++) {
102 if (commit_buffer[i]) {
103 ret = mpp_buffer_put(commit_buffer[i]);
104 if (MPP_OK != ret) {
105 mpp_err("mpp_buffer_test mpp_buffer_put commit mode failed\n");
106 goto MPP_BUFFER_failed;
107 }
108 commit_buffer[i] = NULL;
109 }
110 }
111
112 for (i = 0; i < count; i++) {
113 if (normal_buffer[i]) {
114 mpp_buffer_put(normal_buffer[i]);
115 normal_buffer[i] = NULL;
116 }
117 }
118
119 mpp_buffer_group_put(group);
120
121 mpp_log("mpp_buffer_test commit mode with unused status success\n");
122
123
124 mpp_log("mpp_buffer_test commit mode with used status start\n");
125
126 ret = mpp_allocator_get(&allocator, &api, MPP_BUFFER_TYPE_ION, 0);
127 if (MPP_OK != ret) {
128 mpp_err("mpp_buffer_test mpp_allocator_get ion failed\n");
129 goto MPP_BUFFER_failed;
130 }
131
132 commit.type = MPP_BUFFER_TYPE_ION;
133 commit.size = size;
134
135 for (i = 0; i < count; i++) {
136 ret = api->alloc(allocator, &commit);
137 if (ret) {
138 mpp_err("mpp_buffer_test mpp_allocator_alloc failed\n");
139 goto MPP_BUFFER_failed;
140 }
141
142 mpp_log("allocator get ptr %p with fd %d\n", commit.ptr, commit.fd);
143
144 /*
145 * NOTE: commit buffer info will be directly return within new MppBuffer
146 * This mode allow input group is NULL
147 */
148 ret = mpp_buffer_import(&commit_buffer[i], &commit);
149 if (MPP_OK != ret) {
150 mpp_err("mpp_buffer_test mpp_buffer_commit failed\n");
151 goto MPP_BUFFER_failed;
152 }
153
154 /*
155 * test imported buffer
156 */
157 {
158 void *ptr = mpp_buffer_get_ptr(commit_buffer[i]);
159 if (NULL == ptr) {
160 mpp_err("mpp_buffer_test mpp_buffer_get_ptr failed\n");
161 goto MPP_BUFFER_failed;
162 }
163
164 mpp_log("get ptr %p from fd %d\n", ptr, mpp_buffer_get_fd(commit_buffer[i]));
165
166 memset(ptr, 0, mpp_buffer_get_size(commit_buffer[i]));
167 }
168 }
169
170 for (i = 0; i < count; i++) {
171 if (commit_buffer[i]) {
172 ret = mpp_buffer_info_get(commit_buffer[i], &commit);
173 if (MPP_OK != ret) {
174 mpp_err("mpp_buffer_test mpp_buffer_info_get failed\n");
175 goto MPP_BUFFER_failed;
176 }
177
178 ret = mpp_buffer_put(commit_buffer[i]);
179 if (MPP_OK != ret) {
180 mpp_err("mpp_buffer_test mpp_buffer_put commit mode failed\n");
181 goto MPP_BUFFER_failed;
182 }
183
184 commit_buffer[i] = NULL;
185
186 /* NOTE: buffer info from allocator need to be free directly */
187 ret = api->free(allocator, &commit);
188 if (MPP_OK != ret) {
189 mpp_err("mpp_buffer_test api->free failed\n");
190 goto MPP_BUFFER_failed;
191 }
192 }
193 }
194
195 ret = mpp_allocator_put(&allocator);
196 if (MPP_OK != ret) {
197 mpp_err("mpp_buffer_test mpp_allocator_put failed\n");
198 goto MPP_BUFFER_failed;
199 }
200
201 mpp_log("mpp_buffer_test commit mode with used status success\n");
202
203
204 mpp_log("mpp_buffer_test normal mode start\n");
205
206 ret = mpp_buffer_group_get_internal(&group, MPP_BUFFER_TYPE_ION);
207 if (MPP_OK != ret) {
208 mpp_err("mpp_buffer_test mpp_buffer_group_get failed\n");
209 goto MPP_BUFFER_failed;
210 }
211
212 count = MPP_BUFFER_TEST_NORMAL_COUNT;
213
214 mpp_buffer_group_limit_config(group, 0, count);
215
216 for (i = 0; i < count; i++) {
217 ret = mpp_buffer_get(group, &normal_buffer[i], (i + 1) * SZ_1K);
218 if (MPP_OK != ret) {
219 mpp_err("mpp_buffer_test mpp_buffer_get mode normal failed\n");
220 goto MPP_BUFFER_failed;
221 }
222 }
223
224 for (i = 0; i < count; i++) {
225 if (normal_buffer[i]) {
226 ret = mpp_buffer_put(normal_buffer[i]);
227 if (MPP_OK != ret) {
228 mpp_err("mpp_buffer_test mpp_buffer_get mode normal failed\n");
229 goto MPP_BUFFER_failed;
230 }
231 normal_buffer[i] = NULL;
232 }
233 }
234
235 mpp_log("mpp_buffer_test normal mode success\n");
236
237 if (group) {
238 mpp_buffer_group_put(group);
239 group = NULL;
240 }
241
242 mpp_log("mpp_buffer_test success\n");
243
244 ret = mpp_buffer_get(NULL, &legacy_buffer, MPP_BUFFER_TEST_SIZE);
245 if (MPP_OK != ret) {
246 mpp_log("mpp_buffer_test mpp_buffer_get legacy buffer failed\n");
247 goto MPP_BUFFER_failed;
248 }
249
250 ret = mpp_buffer_put(legacy_buffer);
251 if (MPP_OK != ret) {
252 mpp_log("mpp_buffer_test mpp_buffer_put legacy buffer failed\n");
253 goto MPP_BUFFER_failed;
254 }
255
256 mpp_env_set_u32("mpp_buffer_debug", 0);
257
258 return ret;
259
260 MPP_BUFFER_failed:
261 for (i = 0; i < MPP_BUFFER_TEST_COMMIT_COUNT; i++) {
262 if (commit_buffer[i])
263 mpp_buffer_put(commit_buffer[i]);
264 }
265
266 for (i = 0; i < MPP_BUFFER_TEST_COMMIT_COUNT; i++) {
267 if (commit_ptr[i]) {
268 free(commit_ptr[i]);
269 commit_ptr[i] = NULL;
270 }
271 }
272 for (i = 0; i < MPP_BUFFER_TEST_NORMAL_COUNT; i++) {
273 if (normal_buffer[i])
274 mpp_buffer_put(normal_buffer[i]);
275 }
276
277 if (group) {
278 mpp_buffer_group_put(group);
279 group = NULL;
280 }
281
282 if (legacy_buffer) {
283 mpp_buffer_put(legacy_buffer);
284 legacy_buffer = NULL;
285 }
286
287 if (allocator) {
288 mpp_allocator_put(&allocator);
289 }
290 mpp_assert(NULL == allocator);
291
292 mpp_log("mpp_buffer_test failed\n");
293
294 mpp_env_set_u32("mpp_buffer_debug", 0);
295
296 return ret;
297 }
298
299