xref: /OK3568_Linux_fs/u-boot/lib/optee_clientApi/OpteeClientRkNewFs_v2.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun  * Copyright 2020, Rockchip Electronics Co., Ltd
3*4882a593Smuzhiyun  * hisping lin, <hisping.lin@rock-chips.com>
4*4882a593Smuzhiyun  *
5*4882a593Smuzhiyun  * SPDX-License-Identifier:	GPL-2.0+
6*4882a593Smuzhiyun  */
7*4882a593Smuzhiyun 
8*4882a593Smuzhiyun #include <common.h>
9*4882a593Smuzhiyun #include <stdlib.h>
10*4882a593Smuzhiyun #include <command.h>
11*4882a593Smuzhiyun #include <boot_rkimg.h>
12*4882a593Smuzhiyun #include <part.h>
13*4882a593Smuzhiyun #include <optee_include/OpteeClientRkNewFs.h>
14*4882a593Smuzhiyun #include <optee_include/tee_client_api.h>
15*4882a593Smuzhiyun 
16*4882a593Smuzhiyun /*
17*4882a593Smuzhiyun  *#define DEBUG_RKSS
18*4882a593Smuzhiyun  *#define DEBUG_CLEAN_RKSS
19*4882a593Smuzhiyun  */
20*4882a593Smuzhiyun 
21*4882a593Smuzhiyun /*
22*4882a593Smuzhiyun  *	RK Secure Storage Version 2
23*4882a593Smuzhiyun  *		Area0 Backup 0 Size : 256 kb	<---->	Area0 Backup 1 Size : 256 kb
24*4882a593Smuzhiyun  *		Area1 Backup 0 Size : 256 kb	<---->	Area1 Backup 1 Size : 256 kb
25*4882a593Smuzhiyun  *
26*4882a593Smuzhiyun  *	------ 1 section is 512 bytes -----
27*4882a593Smuzhiyun  *	------ Area0 Backup0 section from 0 to 511 --------
28*4882a593Smuzhiyun  *	1 section for file header		[0]
29*4882a593Smuzhiyun  *	1 section for used flags		[1]
30*4882a593Smuzhiyun  *		- 1 byte = 2 flag
31*4882a593Smuzhiyun  *	62 section for file tables		[2-63]
32*4882a593Smuzhiyun  *		- size of table 128 bytes
33*4882a593Smuzhiyun  *	447 section for data			[64-510]
34*4882a593Smuzhiyun  *	1 section for file footer		[511]
35*4882a593Smuzhiyun  *
36*4882a593Smuzhiyun  * 	------ Area0 Backup1 section from 512 to 1023 --------
37*4882a593Smuzhiyun  *	1 section for file header		[512]
38*4882a593Smuzhiyun  *	1 section for used flags		[513]
39*4882a593Smuzhiyun  *		- 1 byte = 2 flag
40*4882a593Smuzhiyun  *	62 section for file tables		[514-575]
41*4882a593Smuzhiyun  *		- size of table 128 bytes
42*4882a593Smuzhiyun  *	447 section for data			[576-1022]
43*4882a593Smuzhiyun  *	1 section for file footer		[1023]
44*4882a593Smuzhiyun  *
45*4882a593Smuzhiyun  * 	------ Area1 Backup0 section from 1024 to 1535 --------
46*4882a593Smuzhiyun  *	1 section for file header		[1024]
47*4882a593Smuzhiyun  *	1 section for used flags		[1025]
48*4882a593Smuzhiyun  *		- 1 byte = 2 flag
49*4882a593Smuzhiyun  *	62 section for file tables		[1026-1087]
50*4882a593Smuzhiyun  *		- size of table 128 bytes
51*4882a593Smuzhiyun  *	447 section for data			[1088-1534]
52*4882a593Smuzhiyun  *	1 section for file footer		[1535]
53*4882a593Smuzhiyun  *
54*4882a593Smuzhiyun  * 	------ Area1 Backup1 section from 1536 to 2047 --------
55*4882a593Smuzhiyun  *	1 section for file header		[1536]
56*4882a593Smuzhiyun  *	1 section for used flags		[1537]
57*4882a593Smuzhiyun  *		- 1 byte = 2 flag
58*4882a593Smuzhiyun  *	62 section for file tables		[1538-1599]
59*4882a593Smuzhiyun  *		- size of table 128 bytes
60*4882a593Smuzhiyun  *	447 section for data			[1600-2046]
61*4882a593Smuzhiyun  *	1 section for file footer		[2047]
62*4882a593Smuzhiyun  */
63*4882a593Smuzhiyun 
64*4882a593Smuzhiyun /* define for backup */
65*4882a593Smuzhiyun #define RKSS_HEADER_INDEX		0
66*4882a593Smuzhiyun #define RKSS_HEADER_COUNT		1
67*4882a593Smuzhiyun #define RKSS_USEDFLAGS_INDEX		1
68*4882a593Smuzhiyun #define RKSS_USEDFLAGS_COUNT		1
69*4882a593Smuzhiyun #define RKSS_TABLE_INDEX		2
70*4882a593Smuzhiyun #define RKSS_TABLE_COUNT		62
71*4882a593Smuzhiyun #define RKSS_DATA_INDEX			64
72*4882a593Smuzhiyun #define RKSS_DATA_COUNT			447
73*4882a593Smuzhiyun #define RKSS_FOOTER_INDEX		511
74*4882a593Smuzhiyun #define RKSS_FOOTER_COUNT		1
75*4882a593Smuzhiyun #define RKSS_SECTION_COUNT		512
76*4882a593Smuzhiyun 
77*4882a593Smuzhiyun #define RKSS_MAX_AREA_NUM		8
78*4882a593Smuzhiyun #define RKSS_ACTIVE_AREA_NUM		2
79*4882a593Smuzhiyun #define RKSS_DATA_LEN			512
80*4882a593Smuzhiyun #define RKSS_EACH_FILEFOLDER_COUNT	4
81*4882a593Smuzhiyun #define RKSS_TABLE_SIZE			128
82*4882a593Smuzhiyun #define RKSS_NAME_MAX_LENGTH		112
83*4882a593Smuzhiyun #define RKSS_BACKUP_NUM			2
84*4882a593Smuzhiyun #define RKSS_TAG			0x524B5353
85*4882a593Smuzhiyun 
86*4882a593Smuzhiyun struct rkss_file_header {
87*4882a593Smuzhiyun 	uint32_t	tag;
88*4882a593Smuzhiyun 	uint32_t	version;
89*4882a593Smuzhiyun 	uint32_t	backup_count;
90*4882a593Smuzhiyun 	uint16_t	backup_index;
91*4882a593Smuzhiyun 	uint16_t	backup_dirty;
92*4882a593Smuzhiyun 	uint8_t		reserve[496];
93*4882a593Smuzhiyun };
94*4882a593Smuzhiyun struct rkss_file_table {
95*4882a593Smuzhiyun 	uint32_t	size;
96*4882a593Smuzhiyun 	uint16_t	index;
97*4882a593Smuzhiyun 	uint8_t		flags;
98*4882a593Smuzhiyun 	uint8_t		used;
99*4882a593Smuzhiyun 	char		name[RKSS_NAME_MAX_LENGTH];
100*4882a593Smuzhiyun 	uint8_t		reserve[8];
101*4882a593Smuzhiyun };
102*4882a593Smuzhiyun struct rkss_file_footer {
103*4882a593Smuzhiyun 	uint8_t		reserve[508];
104*4882a593Smuzhiyun 	uint32_t	backup_count;
105*4882a593Smuzhiyun };
106*4882a593Smuzhiyun struct rkss_file {
107*4882a593Smuzhiyun 	struct rkss_file_header *header;
108*4882a593Smuzhiyun 	uint8_t	*flags;
109*4882a593Smuzhiyun 	struct rkss_file_table *table;
110*4882a593Smuzhiyun 	uint8_t	*data;
111*4882a593Smuzhiyun 	struct rkss_file_footer *footer;
112*4882a593Smuzhiyun };
113*4882a593Smuzhiyun 
114*4882a593Smuzhiyun /* RK Secure Storage Calls */
115*4882a593Smuzhiyun static char dir_cache[RKSS_NAME_MAX_LENGTH][12];
116*4882a593Smuzhiyun static int dir_num;
117*4882a593Smuzhiyun static int dir_seek;
118*4882a593Smuzhiyun static uint8_t *rkss_buffer[RKSS_MAX_AREA_NUM];
119*4882a593Smuzhiyun static struct rkss_file rkss_info[RKSS_MAX_AREA_NUM];
120*4882a593Smuzhiyun 
121*4882a593Smuzhiyun static struct blk_desc *dev_desc;
122*4882a593Smuzhiyun static disk_partition_t part_info;
123*4882a593Smuzhiyun 
check_security_exist(int print_flag)124*4882a593Smuzhiyun static int check_security_exist(int print_flag)
125*4882a593Smuzhiyun {
126*4882a593Smuzhiyun 	if (!dev_desc) {
127*4882a593Smuzhiyun 		dev_desc = rockchip_get_bootdev();
128*4882a593Smuzhiyun 		if (!dev_desc) {
129*4882a593Smuzhiyun 			printf("TEEC: %s: Could not find device\n", __func__);
130*4882a593Smuzhiyun 			return -1;
131*4882a593Smuzhiyun 		}
132*4882a593Smuzhiyun 
133*4882a593Smuzhiyun 		if (part_get_info_by_name(dev_desc,
134*4882a593Smuzhiyun 					  "security", &part_info) < 0) {
135*4882a593Smuzhiyun 			dev_desc = NULL;
136*4882a593Smuzhiyun 			if (print_flag != 0)
137*4882a593Smuzhiyun 				printf("TEEC: Could not find security partition\n");
138*4882a593Smuzhiyun 			return -1;
139*4882a593Smuzhiyun 		}
140*4882a593Smuzhiyun 	}
141*4882a593Smuzhiyun 	return 0;
142*4882a593Smuzhiyun }
143*4882a593Smuzhiyun 
rkss_verify_usedflags(unsigned int area_index)144*4882a593Smuzhiyun static int rkss_verify_usedflags(unsigned int area_index)
145*4882a593Smuzhiyun {
146*4882a593Smuzhiyun 	uint8_t *flags;
147*4882a593Smuzhiyun 	int i, duel, flag, n, value;
148*4882a593Smuzhiyun 	uint8_t *flagw;
149*4882a593Smuzhiyun 	int used_count;
150*4882a593Smuzhiyun 
151*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
152*4882a593Smuzhiyun 		printf("TEEC: Not support area_index 0x%x\n", area_index);
153*4882a593Smuzhiyun 		return -1;
154*4882a593Smuzhiyun 	}
155*4882a593Smuzhiyun 
156*4882a593Smuzhiyun 	flags = rkss_info[area_index].flags;
157*4882a593Smuzhiyun 	if (flags == NULL) {
158*4882a593Smuzhiyun 		printf("TEEC: %s flags is null\n", __func__);
159*4882a593Smuzhiyun 		return -1;
160*4882a593Smuzhiyun 	}
161*4882a593Smuzhiyun 
162*4882a593Smuzhiyun 	used_count = RKSS_HEADER_COUNT +
163*4882a593Smuzhiyun 		RKSS_USEDFLAGS_COUNT +
164*4882a593Smuzhiyun 		RKSS_TABLE_COUNT;
165*4882a593Smuzhiyun 
166*4882a593Smuzhiyun 	for (i = 0; i < used_count; i++) {
167*4882a593Smuzhiyun 		duel = *(flags + (int)i/2);
168*4882a593Smuzhiyun 		flag = i & 0x1 ? duel & 0x0F : (duel & 0xF0) >> 4;
169*4882a593Smuzhiyun 		if (flag != 0x1)
170*4882a593Smuzhiyun 			goto init;
171*4882a593Smuzhiyun 	}
172*4882a593Smuzhiyun 
173*4882a593Smuzhiyun 	for (i = RKSS_FOOTER_INDEX; i < RKSS_USEDFLAGS_COUNT * RKSS_DATA_LEN * 2; i++) {
174*4882a593Smuzhiyun 		duel = *(flags + (int)i/2);
175*4882a593Smuzhiyun 		flag = i & 0x1 ? duel & 0x0F : (duel & 0xF0) >> 4;
176*4882a593Smuzhiyun 		if (flag != 0x1)
177*4882a593Smuzhiyun 			goto init;
178*4882a593Smuzhiyun 	}
179*4882a593Smuzhiyun 
180*4882a593Smuzhiyun 	debug("TEEC: %s: success.\n", __func__);
181*4882a593Smuzhiyun 	return 0;
182*4882a593Smuzhiyun 
183*4882a593Smuzhiyun init:
184*4882a593Smuzhiyun 	debug("TEEC: init usedflags section ...\n");
185*4882a593Smuzhiyun 	memset(flags, 0, RKSS_USEDFLAGS_COUNT * RKSS_DATA_LEN);
186*4882a593Smuzhiyun 	for (n = 0; n < used_count; n++) {
187*4882a593Smuzhiyun 		flagw = flags + (int)n/2;
188*4882a593Smuzhiyun 		value = 0x1;
189*4882a593Smuzhiyun 		*flagw = n & 0x1 ? (*flagw & 0xF0) | (value & 0x0F) :
190*4882a593Smuzhiyun 				(*flagw & 0x0F) | (value << 4);
191*4882a593Smuzhiyun 	}
192*4882a593Smuzhiyun 
193*4882a593Smuzhiyun 	for (n = RKSS_FOOTER_INDEX; n < RKSS_USEDFLAGS_COUNT * RKSS_DATA_LEN * 2; n++) {
194*4882a593Smuzhiyun 		flagw = flags + (int)n/2;
195*4882a593Smuzhiyun 		value = 0x1;
196*4882a593Smuzhiyun 		*flagw = n & 0x1 ? (*flagw & 0xF0) | (value & 0x0F) :
197*4882a593Smuzhiyun 				(*flagw & 0x0F) | (value << 4);
198*4882a593Smuzhiyun 	}
199*4882a593Smuzhiyun 	return 0;
200*4882a593Smuzhiyun }
201*4882a593Smuzhiyun 
202*4882a593Smuzhiyun #ifdef DEBUG_CLEAN_RKSS
rkss_storage_delete(uint32_t area_index)203*4882a593Smuzhiyun static int rkss_storage_delete(uint32_t area_index)
204*4882a593Smuzhiyun {
205*4882a593Smuzhiyun 	int ret;
206*4882a593Smuzhiyun 	uint32_t size;
207*4882a593Smuzhiyun 	uint8_t *delete_buff;
208*4882a593Smuzhiyun 
209*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
210*4882a593Smuzhiyun 		printf("TEEC: Not support area_index 0x%x\n", area_index);
211*4882a593Smuzhiyun 		return -1;
212*4882a593Smuzhiyun 	}
213*4882a593Smuzhiyun 
214*4882a593Smuzhiyun 	printf("TEEC: delete area index 0x%x!\n", area_index);
215*4882a593Smuzhiyun 	size = RKSS_SECTION_COUNT * RKSS_BACKUP_NUM * RKSS_DATA_LEN;
216*4882a593Smuzhiyun 	delete_buff = (uint8_t *)memalign(CONFIG_SYS_CACHELINE_SIZE, size);
217*4882a593Smuzhiyun 	if (!delete_buff) {
218*4882a593Smuzhiyun 		printf("TEEC: Malloc failed!\n");
219*4882a593Smuzhiyun 		return -1;
220*4882a593Smuzhiyun 	}
221*4882a593Smuzhiyun 	memset(delete_buff, 0, size);
222*4882a593Smuzhiyun 	ret = blk_dwrite(dev_desc,
223*4882a593Smuzhiyun 		part_info.start + area_index * RKSS_SECTION_COUNT * RKSS_BACKUP_NUM,
224*4882a593Smuzhiyun 		RKSS_SECTION_COUNT * RKSS_BACKUP_NUM, delete_buff);
225*4882a593Smuzhiyun 	if (ret != RKSS_SECTION_COUNT * RKSS_BACKUP_NUM) {
226*4882a593Smuzhiyun 		free(delete_buff);
227*4882a593Smuzhiyun 		printf("TEEC: blk_dwrite fail\n");
228*4882a593Smuzhiyun 		return -1;
229*4882a593Smuzhiyun 	}
230*4882a593Smuzhiyun 
231*4882a593Smuzhiyun 	if (delete_buff)
232*4882a593Smuzhiyun 		free(delete_buff);
233*4882a593Smuzhiyun 	printf("TEEC: delete area success!\n");
234*4882a593Smuzhiyun 	return 0;
235*4882a593Smuzhiyun }
236*4882a593Smuzhiyun 
rkss_storage_reset(void)237*4882a593Smuzhiyun static int rkss_storage_reset(void)
238*4882a593Smuzhiyun {
239*4882a593Smuzhiyun 	if (rkss_storage_delete(0) < 0)
240*4882a593Smuzhiyun 		return -1;
241*4882a593Smuzhiyun 	if (rkss_storage_delete(1) < 0)
242*4882a593Smuzhiyun 		return -1;
243*4882a593Smuzhiyun 	return 0;
244*4882a593Smuzhiyun }
245*4882a593Smuzhiyun #endif
246*4882a593Smuzhiyun 
247*4882a593Smuzhiyun #ifdef DEBUG_RKSS
rkss_dump(void * data,unsigned int len)248*4882a593Smuzhiyun static void rkss_dump(void *data, unsigned int len)
249*4882a593Smuzhiyun {
250*4882a593Smuzhiyun 	char *p = (char *)data;
251*4882a593Smuzhiyun 	unsigned int i = 0;
252*4882a593Smuzhiyun 
253*4882a593Smuzhiyun 	printf("-------------- DUMP %d --------------\n", len);
254*4882a593Smuzhiyun 	for (i = 0; i < len; i++) {
255*4882a593Smuzhiyun 		if (i % 32 == 0)
256*4882a593Smuzhiyun 			printf("\n");
257*4882a593Smuzhiyun 		printf("%02x ", *(p + i));
258*4882a593Smuzhiyun 	}
259*4882a593Smuzhiyun 	printf("\n");
260*4882a593Smuzhiyun 	printf("------------- DUMP END -------------\n");
261*4882a593Smuzhiyun }
262*4882a593Smuzhiyun 
rkss_dump_ptable(void)263*4882a593Smuzhiyun static void rkss_dump_ptable(void)
264*4882a593Smuzhiyun {
265*4882a593Smuzhiyun 	int i, j, n;
266*4882a593Smuzhiyun 	struct rkss_file_table *ptable;
267*4882a593Smuzhiyun 
268*4882a593Smuzhiyun 	printf("-------------- DUMP ptable --------------\n");
269*4882a593Smuzhiyun 
270*4882a593Smuzhiyun 	for (i = 0; i < RKSS_MAX_AREA_NUM; i++) {
271*4882a593Smuzhiyun 		ptable = rkss_info[i].table;
272*4882a593Smuzhiyun 		if (rkss_info[i].table == NULL)
273*4882a593Smuzhiyun 			continue;
274*4882a593Smuzhiyun 		printf("--------------- area[%d] tables ------------\n", i);
275*4882a593Smuzhiyun 		for (j = 0; j < RKSS_TABLE_COUNT; j++) {
276*4882a593Smuzhiyun 			for (n = 0; n < RKSS_EACH_FILEFOLDER_COUNT; n++) {
277*4882a593Smuzhiyun 				printf("[%02d][%c] %s , inx:%d, size:%d\n",
278*4882a593Smuzhiyun 						j * RKSS_EACH_FILEFOLDER_COUNT + n,
279*4882a593Smuzhiyun 						ptable->used == 0 ? 'F':'T', ptable->name,
280*4882a593Smuzhiyun 						ptable->index, ptable->size);
281*4882a593Smuzhiyun 
282*4882a593Smuzhiyun 				ptable++;
283*4882a593Smuzhiyun 			}
284*4882a593Smuzhiyun 		}
285*4882a593Smuzhiyun 	}
286*4882a593Smuzhiyun 	printf("-------------- DUMP END --------------\n");
287*4882a593Smuzhiyun }
288*4882a593Smuzhiyun 
rkss_dump_usedflags(void)289*4882a593Smuzhiyun static void rkss_dump_usedflags(void)
290*4882a593Smuzhiyun {
291*4882a593Smuzhiyun 	int i;
292*4882a593Smuzhiyun 
293*4882a593Smuzhiyun 	for (i = 0; i < RKSS_MAX_AREA_NUM; i++) {
294*4882a593Smuzhiyun 		if (rkss_info[i].flags == NULL)
295*4882a593Smuzhiyun 			continue;
296*4882a593Smuzhiyun 		printf("--------------- area[%d] flags ------------\n", i);
297*4882a593Smuzhiyun 		rkss_dump(rkss_info[i].flags, RKSS_USEDFLAGS_COUNT * RKSS_DATA_LEN);
298*4882a593Smuzhiyun 	}
299*4882a593Smuzhiyun }
300*4882a593Smuzhiyun #endif
301*4882a593Smuzhiyun 
rkss_read_multi_sections(unsigned int area_index,unsigned char * data,unsigned long index,unsigned int num)302*4882a593Smuzhiyun static int rkss_read_multi_sections(unsigned int area_index,
303*4882a593Smuzhiyun 		unsigned char *data, unsigned long index, unsigned int num)
304*4882a593Smuzhiyun {
305*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
306*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
307*4882a593Smuzhiyun 		return -1;
308*4882a593Smuzhiyun 	}
309*4882a593Smuzhiyun 	if (index >= RKSS_SECTION_COUNT || num > RKSS_SECTION_COUNT ||
310*4882a593Smuzhiyun 		(index + num) > RKSS_SECTION_COUNT) {
311*4882a593Smuzhiyun 		printf("TEEC: %s index num invalid\n", __func__);
312*4882a593Smuzhiyun 		return -1;
313*4882a593Smuzhiyun 	}
314*4882a593Smuzhiyun 	if (rkss_buffer[area_index] == NULL) {
315*4882a593Smuzhiyun 		printf("TEEC: %s rkss_buffer is null\n", __func__);
316*4882a593Smuzhiyun 		return -1;
317*4882a593Smuzhiyun 	}
318*4882a593Smuzhiyun 	memcpy(data, rkss_buffer[area_index] + index * RKSS_DATA_LEN, num * RKSS_DATA_LEN);
319*4882a593Smuzhiyun 	return 0;
320*4882a593Smuzhiyun }
321*4882a593Smuzhiyun 
rkss_write_multi_sections(unsigned int area_index,unsigned char * data,unsigned long index,unsigned int num)322*4882a593Smuzhiyun static int rkss_write_multi_sections(unsigned int area_index,
323*4882a593Smuzhiyun 		unsigned char *data, unsigned long index, unsigned int num)
324*4882a593Smuzhiyun {
325*4882a593Smuzhiyun 	if (num == 0)
326*4882a593Smuzhiyun 		return 0;
327*4882a593Smuzhiyun 
328*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
329*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
330*4882a593Smuzhiyun 		return -1;
331*4882a593Smuzhiyun 	}
332*4882a593Smuzhiyun 
333*4882a593Smuzhiyun 	if (index >= RKSS_SECTION_COUNT || num > RKSS_SECTION_COUNT ||
334*4882a593Smuzhiyun 		(index + num) > RKSS_SECTION_COUNT) {
335*4882a593Smuzhiyun 		printf("TEEC: %s index num invalid\n", __func__);
336*4882a593Smuzhiyun 		return -1;
337*4882a593Smuzhiyun 	}
338*4882a593Smuzhiyun 
339*4882a593Smuzhiyun 	if (rkss_buffer[area_index] == NULL) {
340*4882a593Smuzhiyun 		printf("TEEC: %s rkss_buffer is null\n", __func__);
341*4882a593Smuzhiyun 		return -1;
342*4882a593Smuzhiyun 	}
343*4882a593Smuzhiyun 
344*4882a593Smuzhiyun 	memcpy(rkss_buffer[area_index] + index * RKSS_DATA_LEN, data, num * RKSS_DATA_LEN);
345*4882a593Smuzhiyun 	rkss_info[area_index].header->backup_dirty = 1;
346*4882a593Smuzhiyun 	return 0;
347*4882a593Smuzhiyun }
348*4882a593Smuzhiyun 
rkss_get_fileinfo_by_index(int fd,struct rkss_file_table * ptable,unsigned int * out_area_index)349*4882a593Smuzhiyun static int rkss_get_fileinfo_by_index(int fd,
350*4882a593Smuzhiyun 		struct rkss_file_table *ptable, unsigned int *out_area_index)
351*4882a593Smuzhiyun {
352*4882a593Smuzhiyun 	struct rkss_file_table *p;
353*4882a593Smuzhiyun 	unsigned int area_index;
354*4882a593Smuzhiyun 
355*4882a593Smuzhiyun 	area_index = fd / (RKSS_TABLE_COUNT * RKSS_EACH_FILEFOLDER_COUNT);
356*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
357*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
358*4882a593Smuzhiyun 		return -1;
359*4882a593Smuzhiyun 	}
360*4882a593Smuzhiyun 
361*4882a593Smuzhiyun 	p = rkss_info[area_index].table;
362*4882a593Smuzhiyun 	if (p == NULL) {
363*4882a593Smuzhiyun 		printf("TEEC: %s table is null\n", __func__);
364*4882a593Smuzhiyun 		return -1;
365*4882a593Smuzhiyun 	}
366*4882a593Smuzhiyun 
367*4882a593Smuzhiyun 	p += fd % (RKSS_TABLE_COUNT * RKSS_EACH_FILEFOLDER_COUNT);
368*4882a593Smuzhiyun 	if (p->used != 1) {
369*4882a593Smuzhiyun 		debug("TEEC: %s unused table!\n", __func__);
370*4882a593Smuzhiyun 		return -1;
371*4882a593Smuzhiyun 	}
372*4882a593Smuzhiyun 	debug("TEEC: %s p->used = %d p->name=%s p->index=%d p->size=%d\n",
373*4882a593Smuzhiyun 		__func__, p->used, p->name, p->index, p->size);
374*4882a593Smuzhiyun 	memcpy(ptable, p, sizeof(struct rkss_file_table));
375*4882a593Smuzhiyun 	*out_area_index = area_index;
376*4882a593Smuzhiyun 	return 0;
377*4882a593Smuzhiyun }
378*4882a593Smuzhiyun 
rkss_get_fileinfo_by_name(char * filename,struct rkss_file_table * ptable,unsigned int * out_area_index)379*4882a593Smuzhiyun static int rkss_get_fileinfo_by_name(char *filename,
380*4882a593Smuzhiyun 		struct rkss_file_table *ptable, unsigned int *out_area_index)
381*4882a593Smuzhiyun {
382*4882a593Smuzhiyun 	int ret;
383*4882a593Smuzhiyun 	unsigned int i, j, n, len;
384*4882a593Smuzhiyun 	struct rkss_file_table *p;
385*4882a593Smuzhiyun 
386*4882a593Smuzhiyun 	len = strlen(filename);
387*4882a593Smuzhiyun 	if (len > RKSS_NAME_MAX_LENGTH - 1) {
388*4882a593Smuzhiyun 		printf("TEEC: filename is too long. length:%u\n", len);
389*4882a593Smuzhiyun 		return -1;
390*4882a593Smuzhiyun 	}
391*4882a593Smuzhiyun 
392*4882a593Smuzhiyun 	for (i = 0; i < RKSS_MAX_AREA_NUM; i++) {
393*4882a593Smuzhiyun 		if (rkss_info[i].table == NULL)
394*4882a593Smuzhiyun 			continue;
395*4882a593Smuzhiyun 		for (j = 0; j < RKSS_TABLE_COUNT; j++) {
396*4882a593Smuzhiyun 			for (n = 0; n < RKSS_EACH_FILEFOLDER_COUNT; n++) {
397*4882a593Smuzhiyun 				p = rkss_info[i].table + j * RKSS_EACH_FILEFOLDER_COUNT + n;
398*4882a593Smuzhiyun 
399*4882a593Smuzhiyun 				if (p->used == 0)
400*4882a593Smuzhiyun 					continue;
401*4882a593Smuzhiyun 
402*4882a593Smuzhiyun 				if (!strcmp(p->name, filename)) {
403*4882a593Smuzhiyun 					debug("TEEC: %s: area%d hit table[%d/%d], index[%d/%d]\n",
404*4882a593Smuzhiyun 						__func__, i, j, RKSS_TABLE_COUNT, n, RKSS_EACH_FILEFOLDER_COUNT);
405*4882a593Smuzhiyun 					memcpy(ptable, p, sizeof(struct rkss_file_table));
406*4882a593Smuzhiyun 					*out_area_index = i;
407*4882a593Smuzhiyun 					ret = i * RKSS_TABLE_COUNT * RKSS_EACH_FILEFOLDER_COUNT +
408*4882a593Smuzhiyun 						j * RKSS_EACH_FILEFOLDER_COUNT + n;
409*4882a593Smuzhiyun 					return ret;
410*4882a593Smuzhiyun 				}
411*4882a593Smuzhiyun 
412*4882a593Smuzhiyun 				// Folder Matching
413*4882a593Smuzhiyun 				const char *split = "/";
414*4882a593Smuzhiyun 				char *last_inpos = filename;
415*4882a593Smuzhiyun 				char *last_svpos = p->name;
416*4882a593Smuzhiyun 				char *cur_inpos = NULL;
417*4882a593Smuzhiyun 				char *cur_svpos = NULL;
418*4882a593Smuzhiyun 
419*4882a593Smuzhiyun 				do {
420*4882a593Smuzhiyun 					cur_inpos = strstr(last_inpos, split);
421*4882a593Smuzhiyun 					cur_svpos = strstr(last_svpos, split);
422*4882a593Smuzhiyun 					int size_in = cur_inpos == NULL ?
423*4882a593Smuzhiyun 							(int)strlen(last_inpos) : cur_inpos - last_inpos;
424*4882a593Smuzhiyun 					int size_sv = cur_svpos == NULL ?
425*4882a593Smuzhiyun 							(int)strlen(last_svpos) : cur_svpos - last_svpos;
426*4882a593Smuzhiyun 
427*4882a593Smuzhiyun 					ret = memcmp(last_inpos, last_svpos, size_in);
428*4882a593Smuzhiyun 
429*4882a593Smuzhiyun 					last_inpos = cur_inpos + 1;
430*4882a593Smuzhiyun 					last_svpos = cur_svpos + 1;
431*4882a593Smuzhiyun 
432*4882a593Smuzhiyun 					if (size_in != size_sv || ret)
433*4882a593Smuzhiyun 						goto UNMATCHFOLDER;
434*4882a593Smuzhiyun 
435*4882a593Smuzhiyun 				} while (cur_inpos && cur_svpos);
436*4882a593Smuzhiyun 
437*4882a593Smuzhiyun 				debug("TEEC: Matched folder: %s\n", p->name);
438*4882a593Smuzhiyun 				return -100;
439*4882a593Smuzhiyun UNMATCHFOLDER:
440*4882a593Smuzhiyun 				debug("TEEC: Unmatched ...");
441*4882a593Smuzhiyun 			}
442*4882a593Smuzhiyun 		}
443*4882a593Smuzhiyun 	}
444*4882a593Smuzhiyun 	debug("TEEC: %s: file or dir no found!\n", __func__);
445*4882a593Smuzhiyun 	return -1;
446*4882a593Smuzhiyun }
447*4882a593Smuzhiyun 
rkss_get_dirs_by_name(char * filename)448*4882a593Smuzhiyun static int rkss_get_dirs_by_name(char *filename)
449*4882a593Smuzhiyun {
450*4882a593Smuzhiyun 	int ret;
451*4882a593Smuzhiyun 	unsigned int i, j, n, len;
452*4882a593Smuzhiyun 	struct rkss_file_table *p;
453*4882a593Smuzhiyun 
454*4882a593Smuzhiyun 	len = strlen(filename);
455*4882a593Smuzhiyun 	if (len > RKSS_NAME_MAX_LENGTH - 1) {
456*4882a593Smuzhiyun 		printf("TEEC: filename is too long. length:%u\n", len);
457*4882a593Smuzhiyun 		return -1;
458*4882a593Smuzhiyun 	}
459*4882a593Smuzhiyun 
460*4882a593Smuzhiyun 	dir_num = 0;
461*4882a593Smuzhiyun 	for (i = 0; i < RKSS_MAX_AREA_NUM; i++) {
462*4882a593Smuzhiyun 		if (rkss_info[i].table == NULL)
463*4882a593Smuzhiyun 			continue;
464*4882a593Smuzhiyun 		for (j = 0; j < RKSS_TABLE_COUNT; j++) {
465*4882a593Smuzhiyun 			for (n = 0; n < RKSS_EACH_FILEFOLDER_COUNT; n++) {
466*4882a593Smuzhiyun 				p = rkss_info[i].table + j * RKSS_EACH_FILEFOLDER_COUNT + n;
467*4882a593Smuzhiyun 
468*4882a593Smuzhiyun 				if (p->used == 0)
469*4882a593Smuzhiyun 					continue;
470*4882a593Smuzhiyun 
471*4882a593Smuzhiyun 				// Full Matching
472*4882a593Smuzhiyun 				ret = memcmp(p->name, filename, strlen(filename));
473*4882a593Smuzhiyun 				debug("TEEC: comparing [fd:%d] : %s ?= %s, ret: %d\n",
474*4882a593Smuzhiyun 					(i * RKSS_TABLE_COUNT + j) * RKSS_EACH_FILEFOLDER_COUNT + n,
475*4882a593Smuzhiyun 					p->name, filename, ret);
476*4882a593Smuzhiyun 				if (!ret && strlen(p->name) > strlen(filename)) {
477*4882a593Smuzhiyun 					char *chk = p->name + strlen(filename);
478*4882a593Smuzhiyun 					if (*chk == '/') {
479*4882a593Smuzhiyun 						char *file = p->name + strlen(filename) + 1;
480*4882a593Smuzhiyun 						char *subdir = strtok(file, "/");
481*4882a593Smuzhiyun 						debug("TEEC: found: %s\n", subdir);
482*4882a593Smuzhiyun 						strcpy(dir_cache[dir_num], subdir);
483*4882a593Smuzhiyun 						++dir_num;
484*4882a593Smuzhiyun 					}
485*4882a593Smuzhiyun 				}
486*4882a593Smuzhiyun 			}
487*4882a593Smuzhiyun 		}
488*4882a593Smuzhiyun 	}
489*4882a593Smuzhiyun 	return dir_num;
490*4882a593Smuzhiyun }
491*4882a593Smuzhiyun 
rkss_get_empty_section_from_usedflags(unsigned int area_index,int section_size)492*4882a593Smuzhiyun static int rkss_get_empty_section_from_usedflags(
493*4882a593Smuzhiyun 		unsigned int area_index, int section_size)
494*4882a593Smuzhiyun {
495*4882a593Smuzhiyun 	int i = 0;
496*4882a593Smuzhiyun 	int count0 = 0;
497*4882a593Smuzhiyun 
498*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
499*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
500*4882a593Smuzhiyun 		return -1;
501*4882a593Smuzhiyun 	}
502*4882a593Smuzhiyun 	if (rkss_info[area_index].flags == NULL) {
503*4882a593Smuzhiyun 		printf("TEEC: %s flags is null\n", __func__);
504*4882a593Smuzhiyun 		return -1;
505*4882a593Smuzhiyun 	}
506*4882a593Smuzhiyun 	for (i = 0; i < RKSS_SECTION_COUNT; i++) {
507*4882a593Smuzhiyun 		uint8_t *flag = rkss_info[area_index].flags + (int)i/2;
508*4882a593Smuzhiyun 		uint8_t value = i & 0x1 ? *flag & 0x0F : (*flag & 0xF0) >> 4;
509*4882a593Smuzhiyun 
510*4882a593Smuzhiyun 		if (value == 0x0) {
511*4882a593Smuzhiyun 			if (++count0 == section_size)
512*4882a593Smuzhiyun 				return (i + 1 - section_size);
513*4882a593Smuzhiyun 		} else {
514*4882a593Smuzhiyun 			count0 = 0;
515*4882a593Smuzhiyun 		}
516*4882a593Smuzhiyun 	}
517*4882a593Smuzhiyun 
518*4882a593Smuzhiyun 	printf("TEEC: Not enough space available in secure storage !\n");
519*4882a593Smuzhiyun 	return -10;
520*4882a593Smuzhiyun }
521*4882a593Smuzhiyun 
rkss_incref_multi_usedflags_sections(unsigned int area_index,unsigned int index,unsigned int num)522*4882a593Smuzhiyun static int rkss_incref_multi_usedflags_sections(
523*4882a593Smuzhiyun 	unsigned int area_index, unsigned int index, unsigned int num)
524*4882a593Smuzhiyun {
525*4882a593Smuzhiyun 	int value, i;
526*4882a593Smuzhiyun 	uint8_t *flag;
527*4882a593Smuzhiyun 
528*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
529*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
530*4882a593Smuzhiyun 		return -1;
531*4882a593Smuzhiyun 	}
532*4882a593Smuzhiyun 
533*4882a593Smuzhiyun 	if (index >= RKSS_SECTION_COUNT || num > RKSS_SECTION_COUNT ||
534*4882a593Smuzhiyun 		(index + num) > RKSS_SECTION_COUNT) {
535*4882a593Smuzhiyun 		printf("TEEC: index[%d] out of range.\n", index);
536*4882a593Smuzhiyun 		return -1;
537*4882a593Smuzhiyun 	}
538*4882a593Smuzhiyun 	if (rkss_info[area_index].flags == NULL) {
539*4882a593Smuzhiyun 		printf("TEEC: %s flags is null\n", __func__);
540*4882a593Smuzhiyun 		return -1;
541*4882a593Smuzhiyun 	}
542*4882a593Smuzhiyun 
543*4882a593Smuzhiyun 	for (i = 0; i < num; i++, index++) {
544*4882a593Smuzhiyun 		flag = rkss_info[area_index].flags + (int)index / 2;
545*4882a593Smuzhiyun 		value = index & 0x1 ? *flag & 0x0F : (*flag & 0xF0) >> 4;
546*4882a593Smuzhiyun 		if (++value > 0xF) {
547*4882a593Smuzhiyun 			printf("TEEC: reference out of data: %d\n", value);
548*4882a593Smuzhiyun 			value = 0xF;
549*4882a593Smuzhiyun 		}
550*4882a593Smuzhiyun 		*flag = index & 0x1 ? (*flag & 0xF0) | (value & 0x0F) :
551*4882a593Smuzhiyun 				(*flag & 0x0F) | (value << 4);
552*4882a593Smuzhiyun 	}
553*4882a593Smuzhiyun 	rkss_info[area_index].header->backup_dirty = 1;
554*4882a593Smuzhiyun 	return 0;
555*4882a593Smuzhiyun }
556*4882a593Smuzhiyun 
rkss_decref_multi_usedflags_sections(unsigned int area_index,unsigned int index,unsigned int num)557*4882a593Smuzhiyun static int rkss_decref_multi_usedflags_sections(
558*4882a593Smuzhiyun 	unsigned int area_index, unsigned int index, unsigned int num)
559*4882a593Smuzhiyun {
560*4882a593Smuzhiyun 	int value, i;
561*4882a593Smuzhiyun 	uint8_t *flag;
562*4882a593Smuzhiyun 
563*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
564*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
565*4882a593Smuzhiyun 		return -1;
566*4882a593Smuzhiyun 	}
567*4882a593Smuzhiyun 
568*4882a593Smuzhiyun 	if (index >= RKSS_SECTION_COUNT || num > RKSS_SECTION_COUNT ||
569*4882a593Smuzhiyun 		(index + num) > RKSS_SECTION_COUNT) {
570*4882a593Smuzhiyun 		printf("TEEC: index[%d] out of range.\n", index);
571*4882a593Smuzhiyun 		return -1;
572*4882a593Smuzhiyun 	}
573*4882a593Smuzhiyun 	if (rkss_info[area_index].flags == NULL) {
574*4882a593Smuzhiyun 		printf("TEEC: %s flags is null\n", __func__);
575*4882a593Smuzhiyun 		return -1;
576*4882a593Smuzhiyun 	}
577*4882a593Smuzhiyun 
578*4882a593Smuzhiyun 	for (i = 0; i < num; i++, index++) {
579*4882a593Smuzhiyun 		flag = rkss_info[area_index].flags + (int)index / 2;
580*4882a593Smuzhiyun 		value = index & 0x1 ? *flag & 0x0F : (*flag & 0xF0) >> 4;
581*4882a593Smuzhiyun 		if (--value < 0) {
582*4882a593Smuzhiyun 			printf("TEEC: reference out of data: %d\n", value);
583*4882a593Smuzhiyun 			value = 0x0;
584*4882a593Smuzhiyun 		}
585*4882a593Smuzhiyun 		*flag = index & 0x1 ? (*flag & 0xF0) | (value & 0x0F) :
586*4882a593Smuzhiyun 				(*flag & 0x0F) | (value << 4);
587*4882a593Smuzhiyun 	}
588*4882a593Smuzhiyun 	rkss_info[area_index].header->backup_dirty = 1;
589*4882a593Smuzhiyun 	return 0;
590*4882a593Smuzhiyun }
591*4882a593Smuzhiyun 
rkss_get_remain_tables(struct rkss_file_table * p)592*4882a593Smuzhiyun static int rkss_get_remain_tables(struct rkss_file_table *p)
593*4882a593Smuzhiyun {
594*4882a593Smuzhiyun 	unsigned int i, n;
595*4882a593Smuzhiyun 	int count = 0;
596*4882a593Smuzhiyun 
597*4882a593Smuzhiyun 	if (p == NULL)
598*4882a593Smuzhiyun 		return -1;
599*4882a593Smuzhiyun 
600*4882a593Smuzhiyun 	for (i = 0; i < RKSS_TABLE_COUNT; i++) {
601*4882a593Smuzhiyun 		for (n = 0; n < RKSS_EACH_FILEFOLDER_COUNT; n++) {
602*4882a593Smuzhiyun 			if (p->used == 0)
603*4882a593Smuzhiyun 				count++;
604*4882a593Smuzhiyun 			p++;
605*4882a593Smuzhiyun 		}
606*4882a593Smuzhiyun 	}
607*4882a593Smuzhiyun 	return count;
608*4882a593Smuzhiyun }
609*4882a593Smuzhiyun 
rkss_get_remain_flags(uint8_t * flags)610*4882a593Smuzhiyun static int rkss_get_remain_flags(uint8_t *flags)
611*4882a593Smuzhiyun {
612*4882a593Smuzhiyun 	unsigned int i, value;
613*4882a593Smuzhiyun 	uint8_t *flag;
614*4882a593Smuzhiyun 	int count = 0;
615*4882a593Smuzhiyun 
616*4882a593Smuzhiyun 	if (flags == NULL)
617*4882a593Smuzhiyun 		return -1;
618*4882a593Smuzhiyun 
619*4882a593Smuzhiyun 	for (i = 0; i < RKSS_SECTION_COUNT; i++) {
620*4882a593Smuzhiyun 		flag = flags + (int)i / 2;
621*4882a593Smuzhiyun 		value = i & 0x1 ? *flag & 0x0F : (*flag & 0xF0) >> 4;
622*4882a593Smuzhiyun 		if (value == 0)
623*4882a593Smuzhiyun 			count++;
624*4882a593Smuzhiyun 	}
625*4882a593Smuzhiyun 	return count;
626*4882a593Smuzhiyun }
627*4882a593Smuzhiyun 
rkss_get_larger_area(void)628*4882a593Smuzhiyun static int rkss_get_larger_area(void)
629*4882a593Smuzhiyun {
630*4882a593Smuzhiyun 	int i, tables, flags, max_flags = 0;
631*4882a593Smuzhiyun 	int area_index = -1;
632*4882a593Smuzhiyun 
633*4882a593Smuzhiyun 	for (i = 0; i < RKSS_MAX_AREA_NUM; i++) {
634*4882a593Smuzhiyun 		if (rkss_info[i].table == NULL ||
635*4882a593Smuzhiyun 		rkss_info[i].flags == NULL)
636*4882a593Smuzhiyun 			continue;
637*4882a593Smuzhiyun 		tables = rkss_get_remain_tables(rkss_info[i].table);
638*4882a593Smuzhiyun 		flags = rkss_get_remain_flags(rkss_info[i].flags);
639*4882a593Smuzhiyun 		if (tables > 0 && flags > 0 && flags > max_flags) {
640*4882a593Smuzhiyun 			max_flags = flags;
641*4882a593Smuzhiyun 			area_index = i;
642*4882a593Smuzhiyun 		}
643*4882a593Smuzhiyun 	}
644*4882a593Smuzhiyun 	return area_index;
645*4882a593Smuzhiyun }
646*4882a593Smuzhiyun 
rkss_write_area_empty_ptable(unsigned int area_index,struct rkss_file_table * pfile_table)647*4882a593Smuzhiyun static int rkss_write_area_empty_ptable(
648*4882a593Smuzhiyun 	unsigned int area_index, struct rkss_file_table *pfile_table)
649*4882a593Smuzhiyun {
650*4882a593Smuzhiyun 	int i, n, ret;
651*4882a593Smuzhiyun 	struct rkss_file_table *p;
652*4882a593Smuzhiyun 
653*4882a593Smuzhiyun 	if (rkss_info[area_index].table == NULL) {
654*4882a593Smuzhiyun 		printf("TEEC: %s table is null\n", __func__);
655*4882a593Smuzhiyun 		return -1;
656*4882a593Smuzhiyun 	}
657*4882a593Smuzhiyun 	for (i = 0; i < RKSS_TABLE_COUNT; i++) {
658*4882a593Smuzhiyun 		for (n = 0; n < RKSS_EACH_FILEFOLDER_COUNT; n++) {
659*4882a593Smuzhiyun 			p = rkss_info[area_index].table + i * RKSS_EACH_FILEFOLDER_COUNT + n;
660*4882a593Smuzhiyun 			if (p->used == 0) {
661*4882a593Smuzhiyun 				memcpy(p, pfile_table, sizeof(struct rkss_file_table));
662*4882a593Smuzhiyun 				p->used = 1;
663*4882a593Smuzhiyun 				debug("TEEC: write emt ptable : [%d,%d] name:%s, index:%d, size:%d, used:%d\n",
664*4882a593Smuzhiyun 						i, n, p->name, p->index, p->size, p->used);
665*4882a593Smuzhiyun 				rkss_info[area_index].header->backup_dirty = 1;
666*4882a593Smuzhiyun 				ret = area_index * RKSS_TABLE_COUNT * RKSS_EACH_FILEFOLDER_COUNT +
667*4882a593Smuzhiyun 					i * RKSS_EACH_FILEFOLDER_COUNT + n;
668*4882a593Smuzhiyun 				return ret;
669*4882a593Smuzhiyun 			}
670*4882a593Smuzhiyun 		}
671*4882a593Smuzhiyun 	}
672*4882a593Smuzhiyun 	printf("TEEC: No enough ptable space available in secure storage.\n");
673*4882a593Smuzhiyun 	return -1;
674*4882a593Smuzhiyun }
675*4882a593Smuzhiyun 
rkss_write_empty_ptable(struct rkss_file_table * pfile_table)676*4882a593Smuzhiyun static int rkss_write_empty_ptable(struct rkss_file_table *pfile_table)
677*4882a593Smuzhiyun {
678*4882a593Smuzhiyun 	int area_index;
679*4882a593Smuzhiyun 
680*4882a593Smuzhiyun 	area_index = rkss_get_larger_area();
681*4882a593Smuzhiyun 	if (area_index < 0) {
682*4882a593Smuzhiyun 		printf("TEEC: get area index fail\n");
683*4882a593Smuzhiyun 		return -1;
684*4882a593Smuzhiyun 	}
685*4882a593Smuzhiyun 
686*4882a593Smuzhiyun 	return rkss_write_area_empty_ptable(area_index, pfile_table);
687*4882a593Smuzhiyun }
688*4882a593Smuzhiyun 
rkss_write_back_ptable(int fd,struct rkss_file_table * pfile_table)689*4882a593Smuzhiyun static int rkss_write_back_ptable(
690*4882a593Smuzhiyun 	int fd, struct rkss_file_table *pfile_table)
691*4882a593Smuzhiyun {
692*4882a593Smuzhiyun 	struct rkss_file_table *p;
693*4882a593Smuzhiyun 	unsigned int area_index;
694*4882a593Smuzhiyun 
695*4882a593Smuzhiyun 	area_index = fd / (RKSS_TABLE_COUNT * RKSS_EACH_FILEFOLDER_COUNT);
696*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
697*4882a593Smuzhiyun 		printf("TEEC: %s area_index invalid\n", __func__);
698*4882a593Smuzhiyun 		return -1;
699*4882a593Smuzhiyun 	}
700*4882a593Smuzhiyun 
701*4882a593Smuzhiyun 	p = rkss_info[area_index].table;
702*4882a593Smuzhiyun 	if (p == NULL) {
703*4882a593Smuzhiyun 		printf("TEEC: %s table is null\n", __func__);
704*4882a593Smuzhiyun 		return -1;
705*4882a593Smuzhiyun 	}
706*4882a593Smuzhiyun 
707*4882a593Smuzhiyun 	p += fd % (RKSS_TABLE_COUNT * RKSS_EACH_FILEFOLDER_COUNT);
708*4882a593Smuzhiyun 
709*4882a593Smuzhiyun 	memcpy(p, pfile_table, sizeof(struct rkss_file_table));
710*4882a593Smuzhiyun 	debug("TEEC: write ptable : name:%s, index:%d, size:%d, used:%d\n",
711*4882a593Smuzhiyun 		p->name, p->index, p->size, p->used);
712*4882a593Smuzhiyun 
713*4882a593Smuzhiyun 	rkss_info[area_index].header->backup_dirty = 1;
714*4882a593Smuzhiyun 	return 0;
715*4882a593Smuzhiyun }
716*4882a593Smuzhiyun 
rkss_storage_write(void)717*4882a593Smuzhiyun static int rkss_storage_write(void)
718*4882a593Smuzhiyun {
719*4882a593Smuzhiyun 	int ret, i;
720*4882a593Smuzhiyun 
721*4882a593Smuzhiyun 	for (i = 0; i < RKSS_MAX_AREA_NUM; i++) {
722*4882a593Smuzhiyun 		if (rkss_info[i].header != NULL && rkss_info[i].header->backup_dirty == 1) {
723*4882a593Smuzhiyun 			rkss_info[i].header->backup_count++;
724*4882a593Smuzhiyun 			rkss_info[i].footer->backup_count = rkss_info[i].header->backup_count;
725*4882a593Smuzhiyun 			rkss_info[i].header->backup_index++;
726*4882a593Smuzhiyun 			if (rkss_info[i].header->backup_index >= RKSS_BACKUP_NUM)
727*4882a593Smuzhiyun 				rkss_info[i].header->backup_index = 0;
728*4882a593Smuzhiyun 			rkss_info[i].header->backup_dirty = 0;
729*4882a593Smuzhiyun 
730*4882a593Smuzhiyun 			if (rkss_info[i].header->backup_count == 0xffffffff) {
731*4882a593Smuzhiyun 				rkss_info[i].header->backup_count = 1;
732*4882a593Smuzhiyun 				rkss_info[i].footer->backup_count = 1;
733*4882a593Smuzhiyun 				ret = blk_dwrite(dev_desc,
734*4882a593Smuzhiyun 					part_info.start + i * RKSS_SECTION_COUNT * RKSS_BACKUP_NUM +
735*4882a593Smuzhiyun 					rkss_info[i].header->backup_index * RKSS_SECTION_COUNT,
736*4882a593Smuzhiyun 					RKSS_SECTION_COUNT, rkss_buffer[i]);
737*4882a593Smuzhiyun 				if (ret != RKSS_SECTION_COUNT) {
738*4882a593Smuzhiyun 					printf("TEEC: blk_dwrite fail\n");
739*4882a593Smuzhiyun 					return -1;
740*4882a593Smuzhiyun 				}
741*4882a593Smuzhiyun 
742*4882a593Smuzhiyun 				rkss_info[i].header->backup_count = 2;
743*4882a593Smuzhiyun 				rkss_info[i].footer->backup_count = 2;
744*4882a593Smuzhiyun 				rkss_info[i].header->backup_index++;
745*4882a593Smuzhiyun 				if (rkss_info[i].header->backup_index >= RKSS_BACKUP_NUM)
746*4882a593Smuzhiyun 					rkss_info[i].header->backup_index = 0;
747*4882a593Smuzhiyun 				ret = blk_dwrite(dev_desc,
748*4882a593Smuzhiyun 					part_info.start + i * RKSS_SECTION_COUNT * RKSS_BACKUP_NUM +
749*4882a593Smuzhiyun 					rkss_info[i].header->backup_index * RKSS_SECTION_COUNT,
750*4882a593Smuzhiyun 					RKSS_SECTION_COUNT, rkss_buffer[i]);
751*4882a593Smuzhiyun 				if (ret != RKSS_SECTION_COUNT) {
752*4882a593Smuzhiyun 					printf("TEEC: blk_dwrite fail\n");
753*4882a593Smuzhiyun 					return -1;
754*4882a593Smuzhiyun 				}
755*4882a593Smuzhiyun 			} else {
756*4882a593Smuzhiyun 				ret = blk_dwrite(dev_desc,
757*4882a593Smuzhiyun 					part_info.start + i * RKSS_SECTION_COUNT * RKSS_BACKUP_NUM +
758*4882a593Smuzhiyun 					rkss_info[i].header->backup_index * RKSS_SECTION_COUNT,
759*4882a593Smuzhiyun 					RKSS_SECTION_COUNT, rkss_buffer[i]);
760*4882a593Smuzhiyun 				if (ret != RKSS_SECTION_COUNT) {
761*4882a593Smuzhiyun 					printf("TEEC: blk_dwrite fail\n");
762*4882a593Smuzhiyun 					return -1;
763*4882a593Smuzhiyun 				}
764*4882a593Smuzhiyun 			}
765*4882a593Smuzhiyun 		}
766*4882a593Smuzhiyun 	}
767*4882a593Smuzhiyun 	return 0;
768*4882a593Smuzhiyun }
769*4882a593Smuzhiyun 
rkss_storage_init(uint32_t area_index)770*4882a593Smuzhiyun static int rkss_storage_init(uint32_t area_index)
771*4882a593Smuzhiyun {
772*4882a593Smuzhiyun 	unsigned long ret = 0;
773*4882a593Smuzhiyun 	uint32_t size, i;
774*4882a593Smuzhiyun 	uint32_t max_ver = 0;
775*4882a593Smuzhiyun 	uint32_t max_index = 0;
776*4882a593Smuzhiyun 	uint32_t flags_offset, table_offset, data_offset, footer_offset;
777*4882a593Smuzhiyun 
778*4882a593Smuzhiyun 	if (area_index >= RKSS_MAX_AREA_NUM) {
779*4882a593Smuzhiyun 		printf("TEEC: Not support index=0x%x\n", area_index);
780*4882a593Smuzhiyun 		return -1;
781*4882a593Smuzhiyun 	}
782*4882a593Smuzhiyun 
783*4882a593Smuzhiyun 	size = RKSS_SECTION_COUNT * RKSS_DATA_LEN;
784*4882a593Smuzhiyun 	flags_offset = RKSS_USEDFLAGS_INDEX * RKSS_DATA_LEN;
785*4882a593Smuzhiyun 	table_offset = RKSS_TABLE_INDEX * RKSS_DATA_LEN;
786*4882a593Smuzhiyun 	data_offset = RKSS_DATA_INDEX * RKSS_DATA_LEN;
787*4882a593Smuzhiyun 	footer_offset = RKSS_FOOTER_INDEX * RKSS_DATA_LEN;
788*4882a593Smuzhiyun 
789*4882a593Smuzhiyun 	if (rkss_buffer[area_index] == NULL) {
790*4882a593Smuzhiyun 		/* Always use, no need to release */
791*4882a593Smuzhiyun 		rkss_buffer[area_index] = (uint8_t *)memalign(CONFIG_SYS_CACHELINE_SIZE, size);
792*4882a593Smuzhiyun 		if (!(rkss_buffer[area_index])) {
793*4882a593Smuzhiyun 			printf("TEEC: Malloc failed!\n");
794*4882a593Smuzhiyun 			return -1;
795*4882a593Smuzhiyun 		}
796*4882a593Smuzhiyun 
797*4882a593Smuzhiyun 		/* Pointer initialization */
798*4882a593Smuzhiyun 		rkss_info[area_index].header = (struct rkss_file_header *)(rkss_buffer[area_index]);
799*4882a593Smuzhiyun 		rkss_info[area_index].flags = (uint8_t *)(rkss_buffer[area_index] + flags_offset);
800*4882a593Smuzhiyun 		rkss_info[area_index].table = (struct rkss_file_table *)(rkss_buffer[area_index] + table_offset);
801*4882a593Smuzhiyun 		rkss_info[area_index].data = (uint8_t *)(rkss_buffer[area_index] + data_offset);
802*4882a593Smuzhiyun 		rkss_info[area_index].footer = (struct rkss_file_footer *)(rkss_buffer[area_index] + footer_offset);
803*4882a593Smuzhiyun 
804*4882a593Smuzhiyun 		/* Find valid from (backup0 - backup1) */
805*4882a593Smuzhiyun 		for (i = 0; i < RKSS_BACKUP_NUM; i++) {
806*4882a593Smuzhiyun 			ret = blk_dread(dev_desc,
807*4882a593Smuzhiyun 				part_info.start +
808*4882a593Smuzhiyun 				area_index * RKSS_SECTION_COUNT * RKSS_BACKUP_NUM +
809*4882a593Smuzhiyun 				i * RKSS_SECTION_COUNT,
810*4882a593Smuzhiyun 				RKSS_SECTION_COUNT, rkss_buffer[area_index]);
811*4882a593Smuzhiyun 			if (ret != RKSS_SECTION_COUNT) {
812*4882a593Smuzhiyun 				printf("TEEC: blk_dread fail\n");
813*4882a593Smuzhiyun 				return -1;
814*4882a593Smuzhiyun 			}
815*4882a593Smuzhiyun 
816*4882a593Smuzhiyun 			if ((rkss_info[area_index].header->tag == RKSS_TAG) &&
817*4882a593Smuzhiyun 			    (rkss_info[area_index].footer->backup_count == rkss_info[area_index].header->backup_count)) {
818*4882a593Smuzhiyun 				if (max_ver < rkss_info[area_index].header->backup_count) {
819*4882a593Smuzhiyun 					max_index = i;
820*4882a593Smuzhiyun 					max_ver = rkss_info[area_index].header->backup_count;
821*4882a593Smuzhiyun 				}
822*4882a593Smuzhiyun 			}
823*4882a593Smuzhiyun 		}
824*4882a593Smuzhiyun 
825*4882a593Smuzhiyun 		if (max_ver) {
826*4882a593Smuzhiyun 			debug("TEEC: max_ver=%d, max_index=%d.\n",
827*4882a593Smuzhiyun 				max_ver, max_index);
828*4882a593Smuzhiyun 
829*4882a593Smuzhiyun 			if (max_index != (RKSS_BACKUP_NUM - 1)) {
830*4882a593Smuzhiyun 				ret = blk_dread(dev_desc,
831*4882a593Smuzhiyun 					part_info.start +
832*4882a593Smuzhiyun 					area_index * RKSS_SECTION_COUNT * RKSS_BACKUP_NUM +
833*4882a593Smuzhiyun 					max_index * RKSS_SECTION_COUNT,
834*4882a593Smuzhiyun 					RKSS_SECTION_COUNT, rkss_buffer[area_index]);
835*4882a593Smuzhiyun 				if (ret != RKSS_SECTION_COUNT) {
836*4882a593Smuzhiyun 					printf("TEEC: blk_dread fail\n");
837*4882a593Smuzhiyun 					return -1;
838*4882a593Smuzhiyun 				}
839*4882a593Smuzhiyun 			}
840*4882a593Smuzhiyun 
841*4882a593Smuzhiyun 			if (rkss_info[area_index].header->version == RKSS_VERSION_V2) {
842*4882a593Smuzhiyun 				debug("TEEC: data version equal to image version, do nothing!\n");
843*4882a593Smuzhiyun 			} else if (rkss_info[area_index].header->version < RKSS_VERSION_V2) {
844*4882a593Smuzhiyun 				printf("TEEC: data version lower than image version!\n");
845*4882a593Smuzhiyun 				/* convert rkss version 2 to higher rkss version */
846*4882a593Smuzhiyun 				free(rkss_buffer[area_index]);
847*4882a593Smuzhiyun 				rkss_buffer[area_index] = NULL;
848*4882a593Smuzhiyun 				return -1;
849*4882a593Smuzhiyun 			} else {
850*4882a593Smuzhiyun 				printf("TEEC: data version higher than image version!\n");
851*4882a593Smuzhiyun 				printf("TEEC: please update image!\n");
852*4882a593Smuzhiyun 				free(rkss_buffer[area_index]);
853*4882a593Smuzhiyun 				rkss_buffer[area_index] = NULL;
854*4882a593Smuzhiyun 				return -1;
855*4882a593Smuzhiyun 			}
856*4882a593Smuzhiyun 		} else {
857*4882a593Smuzhiyun 			printf("TEEC: Reset area[%d] info...\n", area_index);
858*4882a593Smuzhiyun 			memset(rkss_buffer[area_index], 0, size);
859*4882a593Smuzhiyun 			rkss_info[area_index].header->tag = RKSS_TAG;
860*4882a593Smuzhiyun 			rkss_info[area_index].header->version = RKSS_VERSION_V2;
861*4882a593Smuzhiyun 			rkss_info[area_index].header->backup_count = 1;
862*4882a593Smuzhiyun 			rkss_info[area_index].footer->backup_count = 1;
863*4882a593Smuzhiyun 			/* Verify Usedflags Section */
864*4882a593Smuzhiyun 			if (rkss_verify_usedflags(area_index) < 0) {
865*4882a593Smuzhiyun 				printf("TEEC: rkss_verify_usedflags fail !\n");
866*4882a593Smuzhiyun 				return -1;
867*4882a593Smuzhiyun 			}
868*4882a593Smuzhiyun 		}
869*4882a593Smuzhiyun 	}
870*4882a593Smuzhiyun 	return 0;
871*4882a593Smuzhiyun }
872*4882a593Smuzhiyun 
ree_fs_new_open(size_t num_params,struct tee_ioctl_param * params)873*4882a593Smuzhiyun static uint32_t ree_fs_new_open(size_t num_params,
874*4882a593Smuzhiyun 				   struct tee_ioctl_param *params)
875*4882a593Smuzhiyun {
876*4882a593Smuzhiyun 	char *filename;
877*4882a593Smuzhiyun 	int fd;
878*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
879*4882a593Smuzhiyun 	unsigned int area_index;
880*4882a593Smuzhiyun 
881*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
882*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
883*4882a593Smuzhiyun 
884*4882a593Smuzhiyun 	filename = (char *)(size_t)params[1].u.memref.shm_id;
885*4882a593Smuzhiyun 	if (!filename)
886*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
887*4882a593Smuzhiyun 
888*4882a593Smuzhiyun 	if (strlen(filename) > RKSS_NAME_MAX_LENGTH) {
889*4882a593Smuzhiyun 		printf("TEEC: %s: file name too long. %s\n", __func__, filename);
890*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
891*4882a593Smuzhiyun 	}
892*4882a593Smuzhiyun 
893*4882a593Smuzhiyun 	debug("TEEC: %s open file: %s, len: %zu\n",
894*4882a593Smuzhiyun 		__func__, filename, strlen(filename));
895*4882a593Smuzhiyun 	fd = rkss_get_fileinfo_by_name(filename, &p, &area_index);
896*4882a593Smuzhiyun 	if (fd < 0) {
897*4882a593Smuzhiyun 		printf("TEEC: %s : no such file. %s\n", __func__, filename);
898*4882a593Smuzhiyun 		return TEEC_ERROR_ITEM_NOT_FOUND;
899*4882a593Smuzhiyun 	}
900*4882a593Smuzhiyun 
901*4882a593Smuzhiyun 	params[2].u.value.a = fd;
902*4882a593Smuzhiyun 	return TEEC_SUCCESS;
903*4882a593Smuzhiyun }
904*4882a593Smuzhiyun 
ree_fs_new_create(size_t num_params,struct tee_ioctl_param * params)905*4882a593Smuzhiyun static TEEC_Result ree_fs_new_create(size_t num_params,
906*4882a593Smuzhiyun 				     struct tee_ioctl_param *params)
907*4882a593Smuzhiyun {
908*4882a593Smuzhiyun 	char *filename;
909*4882a593Smuzhiyun 	int fd;
910*4882a593Smuzhiyun 	int ret, num;
911*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
912*4882a593Smuzhiyun 	unsigned int area_index;
913*4882a593Smuzhiyun 	/* file open flags: O_RDWR | O_CREAT | O_TRUNC
914*4882a593Smuzhiyun 	 * if file exists, we must remove it first.
915*4882a593Smuzhiyun 	 */
916*4882a593Smuzhiyun 	filename = (char *)(size_t)params[1].u.memref.shm_id;
917*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
918*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
919*4882a593Smuzhiyun 	if (!filename)
920*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
921*4882a593Smuzhiyun 
922*4882a593Smuzhiyun 	if (strlen(filename) > RKSS_NAME_MAX_LENGTH) {
923*4882a593Smuzhiyun 		printf("TEEC: %s: file name too long. %s\n", __func__, filename);
924*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
925*4882a593Smuzhiyun 	}
926*4882a593Smuzhiyun 
927*4882a593Smuzhiyun 	debug("TEEC: %s create file: %s, len: %zu\n",
928*4882a593Smuzhiyun 		__func__, filename, strlen(filename));
929*4882a593Smuzhiyun 	fd = rkss_get_fileinfo_by_name(filename, &p, &area_index);
930*4882a593Smuzhiyun 	if (fd >= 0) {
931*4882a593Smuzhiyun 		printf("TEEC: %s : file exist, clear it. %s\n", __func__, filename);
932*4882a593Smuzhiyun 		/* decrease ref from usedflags */
933*4882a593Smuzhiyun 		num = p.size / RKSS_DATA_LEN + 1;
934*4882a593Smuzhiyun 		ret = rkss_decref_multi_usedflags_sections(area_index, p.index, num);
935*4882a593Smuzhiyun 		if (ret < 0) {
936*4882a593Smuzhiyun 			printf("TEEC: rkss_decref_multi_usedflags_sections error !\n");
937*4882a593Smuzhiyun 			return TEEC_ERROR_GENERIC;
938*4882a593Smuzhiyun 		}
939*4882a593Smuzhiyun 
940*4882a593Smuzhiyun 		/* rm from ptable */
941*4882a593Smuzhiyun 		memset(&p, 0, sizeof(struct rkss_file_table));
942*4882a593Smuzhiyun 		ret = rkss_write_back_ptable(fd, &p);
943*4882a593Smuzhiyun 		if (ret < 0) {
944*4882a593Smuzhiyun 			printf("TEEC: %s : write back error %d\n", __func__, ret);
945*4882a593Smuzhiyun 			return TEEC_ERROR_GENERIC;
946*4882a593Smuzhiyun 		}
947*4882a593Smuzhiyun 	}
948*4882a593Smuzhiyun 
949*4882a593Smuzhiyun 	strcpy(p.name, filename);
950*4882a593Smuzhiyun 	p.index = 0;
951*4882a593Smuzhiyun 	p.size = 0;
952*4882a593Smuzhiyun 	p.used = 1;
953*4882a593Smuzhiyun 	p.flags = RK_FS_R | RK_FS_W;
954*4882a593Smuzhiyun 	fd = rkss_write_empty_ptable(&p);
955*4882a593Smuzhiyun 	if (fd < 0) {
956*4882a593Smuzhiyun 		printf("TEEC: %s : write empty ptable error. %s\n", __func__, filename);
957*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
958*4882a593Smuzhiyun 	}
959*4882a593Smuzhiyun 
960*4882a593Smuzhiyun 	params[2].u.value.a = fd;
961*4882a593Smuzhiyun 
962*4882a593Smuzhiyun 	return TEEC_SUCCESS;
963*4882a593Smuzhiyun }
964*4882a593Smuzhiyun 
ree_fs_new_close(size_t num_params,struct tee_ioctl_param * params)965*4882a593Smuzhiyun static TEEC_Result ree_fs_new_close(size_t num_params,
966*4882a593Smuzhiyun 				    struct tee_ioctl_param *params)
967*4882a593Smuzhiyun {
968*4882a593Smuzhiyun 	debug("TEEC: %s !\n", __func__);
969*4882a593Smuzhiyun 	UNREFERENCED_PARAMETER(params);
970*4882a593Smuzhiyun 	UNREFERENCED_PARAMETER(num_params);
971*4882a593Smuzhiyun 	return TEEC_SUCCESS;
972*4882a593Smuzhiyun }
973*4882a593Smuzhiyun 
ree_fs_new_read(size_t num_params,struct tee_ioctl_param * params)974*4882a593Smuzhiyun static TEEC_Result ree_fs_new_read(size_t num_params,
975*4882a593Smuzhiyun 				   struct tee_ioctl_param *params)
976*4882a593Smuzhiyun {
977*4882a593Smuzhiyun 	uint8_t *data;
978*4882a593Smuzhiyun 	size_t len;
979*4882a593Smuzhiyun 	off_t offs;
980*4882a593Smuzhiyun 	int fd;
981*4882a593Smuzhiyun 	int ret;
982*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
983*4882a593Smuzhiyun 	int di, section_num;
984*4882a593Smuzhiyun 	uint8_t *temp_file_data;
985*4882a593Smuzhiyun 	unsigned int area_index;
986*4882a593Smuzhiyun 
987*4882a593Smuzhiyun 	fd = params[0].u.value.b;
988*4882a593Smuzhiyun 	offs = params[0].u.value.c;
989*4882a593Smuzhiyun 
990*4882a593Smuzhiyun 	data = (uint8_t *)(size_t)params[1].u.memref.shm_id;
991*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
992*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
993*4882a593Smuzhiyun 
994*4882a593Smuzhiyun 	if (!data)
995*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
996*4882a593Smuzhiyun 	len = params[1].u.memref.size;
997*4882a593Smuzhiyun 
998*4882a593Smuzhiyun 	debug("TEEC: %s! fd:%d, len:%zu, offs:%ld\n",
999*4882a593Smuzhiyun 		__func__, fd, len, offs);
1000*4882a593Smuzhiyun 
1001*4882a593Smuzhiyun 	ret = rkss_get_fileinfo_by_index(fd, &p, &area_index);
1002*4882a593Smuzhiyun 	if (ret < 0) {
1003*4882a593Smuzhiyun 		printf("TEEC: unavailable fd: %d!\n", fd);
1004*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1005*4882a593Smuzhiyun 	}
1006*4882a593Smuzhiyun 
1007*4882a593Smuzhiyun 	if (offs >= p.size)
1008*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1009*4882a593Smuzhiyun 
1010*4882a593Smuzhiyun 	section_num = p.size / RKSS_DATA_LEN + 1;
1011*4882a593Smuzhiyun 	temp_file_data = malloc(section_num * RKSS_DATA_LEN);
1012*4882a593Smuzhiyun 	ret = rkss_read_multi_sections(area_index, temp_file_data, p.index, section_num);
1013*4882a593Smuzhiyun 	if (ret < 0) {
1014*4882a593Smuzhiyun 		printf("TEEC: unavailable file index!\n");
1015*4882a593Smuzhiyun 		free(temp_file_data);
1016*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1017*4882a593Smuzhiyun 	}
1018*4882a593Smuzhiyun 	di = (offs + len) > p.size ? (p.size - offs) : len;
1019*4882a593Smuzhiyun 	memcpy(data, temp_file_data + offs, di);
1020*4882a593Smuzhiyun 	free(temp_file_data);
1021*4882a593Smuzhiyun 	temp_file_data = 0;
1022*4882a593Smuzhiyun 	params[1].u.memref.size = di;
1023*4882a593Smuzhiyun 
1024*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1025*4882a593Smuzhiyun }
1026*4882a593Smuzhiyun 
ree_fs_new_write(size_t num_params,struct tee_ioctl_param * params)1027*4882a593Smuzhiyun static TEEC_Result ree_fs_new_write(size_t num_params,
1028*4882a593Smuzhiyun 				    struct tee_ioctl_param *params)
1029*4882a593Smuzhiyun {
1030*4882a593Smuzhiyun 	uint8_t *data;
1031*4882a593Smuzhiyun 	size_t len;
1032*4882a593Smuzhiyun 	off_t offs;
1033*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
1034*4882a593Smuzhiyun 	int ret, fd, new_size;
1035*4882a593Smuzhiyun 	int section_num;
1036*4882a593Smuzhiyun 	uint8_t *file_data = 0, *temp_file_data = 0;
1037*4882a593Smuzhiyun 	unsigned int area_index;
1038*4882a593Smuzhiyun 
1039*4882a593Smuzhiyun 	fd = params[0].u.value.b;
1040*4882a593Smuzhiyun 	offs = params[0].u.value.c;
1041*4882a593Smuzhiyun 
1042*4882a593Smuzhiyun 	data = (uint8_t *)(size_t)params[1].u.memref.shm_id;
1043*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
1044*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
1045*4882a593Smuzhiyun 	if (!data)
1046*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1047*4882a593Smuzhiyun 	len = params[1].u.memref.size;
1048*4882a593Smuzhiyun 
1049*4882a593Smuzhiyun 	debug("TEEC: %s ! fd:%d, len:%zu, offs:%ld\n",
1050*4882a593Smuzhiyun 		__func__, fd, len, offs);
1051*4882a593Smuzhiyun 
1052*4882a593Smuzhiyun 	ret = rkss_get_fileinfo_by_index(fd, &p, &area_index);
1053*4882a593Smuzhiyun 	if (ret < 0) {
1054*4882a593Smuzhiyun 		printf("TEEC: %s: fd:%d unvailable!\n", __func__, fd);
1055*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1056*4882a593Smuzhiyun 	}
1057*4882a593Smuzhiyun 
1058*4882a593Smuzhiyun 	new_size = offs + len > p.size ? offs + len : p.size;
1059*4882a593Smuzhiyun 	file_data = malloc(new_size);
1060*4882a593Smuzhiyun 	if (!file_data)
1061*4882a593Smuzhiyun 		return TEEC_ERROR_OUT_OF_MEMORY;
1062*4882a593Smuzhiyun 
1063*4882a593Smuzhiyun 	if (p.size != 0) {
1064*4882a593Smuzhiyun 		/* Read old file data out */
1065*4882a593Smuzhiyun 		section_num = p.size / RKSS_DATA_LEN + 1;
1066*4882a593Smuzhiyun 		temp_file_data = malloc(section_num * RKSS_DATA_LEN);
1067*4882a593Smuzhiyun 		ret = rkss_read_multi_sections(area_index, temp_file_data, p.index, section_num);
1068*4882a593Smuzhiyun 		if (ret < 0) {
1069*4882a593Smuzhiyun 			printf("TEEC: unavailable file index %d section_num %d\n", p.index, section_num);
1070*4882a593Smuzhiyun 			ret = TEEC_ERROR_GENERIC;
1071*4882a593Smuzhiyun 			goto out;
1072*4882a593Smuzhiyun 		}
1073*4882a593Smuzhiyun 		memcpy(file_data, temp_file_data, p.size);
1074*4882a593Smuzhiyun 		free(temp_file_data);
1075*4882a593Smuzhiyun 		temp_file_data = 0;
1076*4882a593Smuzhiyun 		ret = rkss_decref_multi_usedflags_sections(area_index, p.index, section_num);
1077*4882a593Smuzhiyun 		if (ret < 0) {
1078*4882a593Smuzhiyun 			printf("TEEC: rkss_decref_multi_usedflags_sections error !\n");
1079*4882a593Smuzhiyun 			ret = TEEC_ERROR_GENERIC;
1080*4882a593Smuzhiyun 			goto out;
1081*4882a593Smuzhiyun 		}
1082*4882a593Smuzhiyun 	}
1083*4882a593Smuzhiyun 
1084*4882a593Smuzhiyun 	/* update new file info */
1085*4882a593Smuzhiyun 	memcpy(file_data + offs, data, len);
1086*4882a593Smuzhiyun 	p.size = new_size;
1087*4882a593Smuzhiyun 	section_num = new_size / RKSS_DATA_LEN + 1;
1088*4882a593Smuzhiyun 	p.index = rkss_get_empty_section_from_usedflags(area_index, section_num);
1089*4882a593Smuzhiyun 	debug("TEEC: Get Empty section in %d\n", p.index);
1090*4882a593Smuzhiyun 	p.used = 1;
1091*4882a593Smuzhiyun 	ret = rkss_incref_multi_usedflags_sections(area_index, p.index, section_num);
1092*4882a593Smuzhiyun 	if (ret < 0) {
1093*4882a593Smuzhiyun 		printf("TEEC: rkss_incref_multi_usedflags_sections error !\n");
1094*4882a593Smuzhiyun 		ret = TEEC_ERROR_GENERIC;
1095*4882a593Smuzhiyun 		goto out;
1096*4882a593Smuzhiyun 	}
1097*4882a593Smuzhiyun 
1098*4882a593Smuzhiyun 	ret = rkss_write_back_ptable(fd, &p);
1099*4882a593Smuzhiyun 	if (ret < 0) {
1100*4882a593Smuzhiyun 		printf("TEEC: %s: write ptable error!\n", __func__);
1101*4882a593Smuzhiyun 		ret = TEEC_ERROR_GENERIC;
1102*4882a593Smuzhiyun 		goto out;
1103*4882a593Smuzhiyun 	}
1104*4882a593Smuzhiyun 
1105*4882a593Smuzhiyun 	/* write new file data */
1106*4882a593Smuzhiyun 	temp_file_data = malloc(section_num * RKSS_DATA_LEN);
1107*4882a593Smuzhiyun 	memset(temp_file_data, 0, section_num * RKSS_DATA_LEN);
1108*4882a593Smuzhiyun 	memcpy(temp_file_data, file_data, p.size);
1109*4882a593Smuzhiyun 	rkss_write_multi_sections(area_index, temp_file_data, p.index, section_num);
1110*4882a593Smuzhiyun 	free(temp_file_data);
1111*4882a593Smuzhiyun 	temp_file_data = 0;
1112*4882a593Smuzhiyun 
1113*4882a593Smuzhiyun out:
1114*4882a593Smuzhiyun 	if (file_data)
1115*4882a593Smuzhiyun 		free(file_data);
1116*4882a593Smuzhiyun 	if (temp_file_data) {
1117*4882a593Smuzhiyun 		free(temp_file_data);
1118*4882a593Smuzhiyun 		temp_file_data = 0;
1119*4882a593Smuzhiyun 	}
1120*4882a593Smuzhiyun 
1121*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1122*4882a593Smuzhiyun }
1123*4882a593Smuzhiyun 
1124*4882a593Smuzhiyun /* TODO: update file data space */
ree_fs_new_truncate(size_t num_params,struct tee_ioctl_param * params)1125*4882a593Smuzhiyun static TEEC_Result ree_fs_new_truncate(size_t num_params,
1126*4882a593Smuzhiyun 				       struct tee_ioctl_param *params)
1127*4882a593Smuzhiyun {
1128*4882a593Smuzhiyun 	size_t len;
1129*4882a593Smuzhiyun 	int fd, ret;
1130*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
1131*4882a593Smuzhiyun 	unsigned int section_num_old, section_num_new;
1132*4882a593Smuzhiyun 	unsigned int area_index;
1133*4882a593Smuzhiyun 
1134*4882a593Smuzhiyun 	fd = params[0].u.value.b;
1135*4882a593Smuzhiyun 	len = params[0].u.value.c;
1136*4882a593Smuzhiyun 
1137*4882a593Smuzhiyun 	debug("TEEC: %s: fd:%d, lenth:%zu\n", __func__, fd, len);
1138*4882a593Smuzhiyun 
1139*4882a593Smuzhiyun 	ret = rkss_get_fileinfo_by_index(fd, &p, &area_index);
1140*4882a593Smuzhiyun 	if (ret < 0) {
1141*4882a593Smuzhiyun 		printf("TEEC: fd:%d unvailable!\n", fd);
1142*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1143*4882a593Smuzhiyun 	}
1144*4882a593Smuzhiyun 	if (len > p.size) {
1145*4882a593Smuzhiyun 		printf("TEEC: truncate error!\n");
1146*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1147*4882a593Smuzhiyun 	}
1148*4882a593Smuzhiyun 	section_num_old = p.size / RKSS_DATA_LEN + 1;
1149*4882a593Smuzhiyun 	section_num_new = len / RKSS_DATA_LEN + 1;
1150*4882a593Smuzhiyun 	ret = rkss_decref_multi_usedflags_sections(area_index, p.index + section_num_new, section_num_old - section_num_new);
1151*4882a593Smuzhiyun 	if (ret < 0) {
1152*4882a593Smuzhiyun 		printf("TEEC: rkss_decref_multi_usedflags_sections error !\n");
1153*4882a593Smuzhiyun 		ret = TEEC_ERROR_GENERIC;
1154*4882a593Smuzhiyun 	}
1155*4882a593Smuzhiyun 	p.size = len;
1156*4882a593Smuzhiyun 	ret = rkss_write_back_ptable(fd, &p);
1157*4882a593Smuzhiyun 	if (ret < 0) {
1158*4882a593Smuzhiyun 		printf("TEEC: %s: write ptable error!\n", __func__);
1159*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1160*4882a593Smuzhiyun 	}
1161*4882a593Smuzhiyun 
1162*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1163*4882a593Smuzhiyun }
1164*4882a593Smuzhiyun 
ree_fs_new_remove(size_t num_params,struct tee_ioctl_param * params)1165*4882a593Smuzhiyun static TEEC_Result ree_fs_new_remove(size_t num_params,
1166*4882a593Smuzhiyun 				     struct tee_ioctl_param *params)
1167*4882a593Smuzhiyun {
1168*4882a593Smuzhiyun 	char *filename;
1169*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
1170*4882a593Smuzhiyun 	int ret, fd, num;
1171*4882a593Smuzhiyun 	unsigned int area_index;
1172*4882a593Smuzhiyun 
1173*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
1174*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
1175*4882a593Smuzhiyun 
1176*4882a593Smuzhiyun 	filename = (char *)(size_t)params[1].u.memref.shm_id;
1177*4882a593Smuzhiyun 	if (!filename)
1178*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1179*4882a593Smuzhiyun 
1180*4882a593Smuzhiyun 	ret = rkss_get_fileinfo_by_name(filename, &p, &area_index);
1181*4882a593Smuzhiyun 	if (ret < 0) {
1182*4882a593Smuzhiyun 		printf("TEEC: %s: no such file. %s\n", __func__, filename);
1183*4882a593Smuzhiyun 		return 0;
1184*4882a593Smuzhiyun 	}
1185*4882a593Smuzhiyun 	fd = ret;
1186*4882a593Smuzhiyun 
1187*4882a593Smuzhiyun 	debug("TEEC: %s! %s fd:%d index:%d size:%d\n",
1188*4882a593Smuzhiyun 		__func__, filename, fd, p.index, p.size);
1189*4882a593Smuzhiyun 
1190*4882a593Smuzhiyun 	/* decrease ref from usedflags */
1191*4882a593Smuzhiyun 	num = p.size / RKSS_DATA_LEN + 1;
1192*4882a593Smuzhiyun 	ret = rkss_decref_multi_usedflags_sections(area_index, p.index, num);
1193*4882a593Smuzhiyun 	if (ret < 0) {
1194*4882a593Smuzhiyun 		printf("TEEC: rkss_decref_multi_usedflags_sections error !\n");
1195*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1196*4882a593Smuzhiyun 	}
1197*4882a593Smuzhiyun 
1198*4882a593Smuzhiyun 	/* rm from ptable */
1199*4882a593Smuzhiyun 	memset(&p, 0, sizeof(struct rkss_file_table));
1200*4882a593Smuzhiyun 	ret = rkss_write_back_ptable(fd, &p);
1201*4882a593Smuzhiyun 	if (ret < 0) {
1202*4882a593Smuzhiyun 		printf("TEEC: %s: write back error %d\n", __func__, ret);
1203*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1204*4882a593Smuzhiyun 	}
1205*4882a593Smuzhiyun 
1206*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1207*4882a593Smuzhiyun }
1208*4882a593Smuzhiyun 
ree_fs_new_rename(size_t num_params,struct tee_ioctl_param * params)1209*4882a593Smuzhiyun static TEEC_Result ree_fs_new_rename(size_t num_params,
1210*4882a593Smuzhiyun 				     struct tee_ioctl_param *params)
1211*4882a593Smuzhiyun {
1212*4882a593Smuzhiyun 	char *old_fname;
1213*4882a593Smuzhiyun 	char *new_fname;
1214*4882a593Smuzhiyun 	struct rkss_file_table p = {0};
1215*4882a593Smuzhiyun 	int ret;
1216*4882a593Smuzhiyun 	unsigned int area_index;
1217*4882a593Smuzhiyun 
1218*4882a593Smuzhiyun 	old_fname = (char *)(size_t)params[1].u.memref.shm_id;
1219*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
1220*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
1221*4882a593Smuzhiyun 	if (!old_fname)
1222*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1223*4882a593Smuzhiyun 
1224*4882a593Smuzhiyun 	new_fname = (char *)(size_t)params[2].u.memref.shm_id;
1225*4882a593Smuzhiyun 	debug("TEEC: params[2].u.memref.shm_id = 0x%llx params[2].u.memref.shm_offs = 0x%llx\n",
1226*4882a593Smuzhiyun 		params[2].u.memref.shm_id, params[2].u.memref.shm_offs);
1227*4882a593Smuzhiyun 	if (!new_fname)
1228*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1229*4882a593Smuzhiyun 
1230*4882a593Smuzhiyun 	if (strlen(new_fname) > RKSS_NAME_MAX_LENGTH) {
1231*4882a593Smuzhiyun 		printf("TEEC: new file name too long. %s\n", new_fname);
1232*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1233*4882a593Smuzhiyun 	}
1234*4882a593Smuzhiyun 
1235*4882a593Smuzhiyun 	debug("TEEC: rename: %s -> %s\n", old_fname, new_fname);
1236*4882a593Smuzhiyun 
1237*4882a593Smuzhiyun 	ret = rkss_get_fileinfo_by_name(old_fname, &p, &area_index);
1238*4882a593Smuzhiyun 	if (ret < 0) {
1239*4882a593Smuzhiyun 		printf("TEEC: filename: %s no found.\n", old_fname);
1240*4882a593Smuzhiyun 		return TEEC_ERROR_ITEM_NOT_FOUND;
1241*4882a593Smuzhiyun 	}
1242*4882a593Smuzhiyun 
1243*4882a593Smuzhiyun 	strcpy(p.name, new_fname);
1244*4882a593Smuzhiyun 
1245*4882a593Smuzhiyun 	ret = rkss_write_back_ptable(ret, &p);
1246*4882a593Smuzhiyun 	if (ret < 0) {
1247*4882a593Smuzhiyun 		printf("TEEC: write ptable error!\n");
1248*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1249*4882a593Smuzhiyun 	}
1250*4882a593Smuzhiyun 
1251*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1252*4882a593Smuzhiyun }
1253*4882a593Smuzhiyun 
ree_fs_new_opendir(size_t num_params,struct tee_ioctl_param * params)1254*4882a593Smuzhiyun static TEEC_Result ree_fs_new_opendir(size_t num_params,
1255*4882a593Smuzhiyun 				      struct tee_ioctl_param *params)
1256*4882a593Smuzhiyun {
1257*4882a593Smuzhiyun 	char *dirname;
1258*4882a593Smuzhiyun 	int ret;
1259*4882a593Smuzhiyun 
1260*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
1261*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
1262*4882a593Smuzhiyun 
1263*4882a593Smuzhiyun 	dirname = (char *)(size_t)params[1].u.memref.shm_id;
1264*4882a593Smuzhiyun 	if (!dirname)
1265*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1266*4882a593Smuzhiyun 
1267*4882a593Smuzhiyun 	dir_seek = 0;
1268*4882a593Smuzhiyun 	ret = rkss_get_dirs_by_name(dirname);
1269*4882a593Smuzhiyun 	if (ret < 0) {
1270*4882a593Smuzhiyun 		printf("TEEC: %s: error\n", __func__);
1271*4882a593Smuzhiyun 		return TEEC_ERROR_GENERIC;
1272*4882a593Smuzhiyun 	}
1273*4882a593Smuzhiyun 
1274*4882a593Smuzhiyun 	debug("TEEC: %s: %s, seek/num:%d/%d\n",
1275*4882a593Smuzhiyun 		__func__, dirname, dir_seek, dir_num);
1276*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1277*4882a593Smuzhiyun }
1278*4882a593Smuzhiyun 
ree_fs_new_closedir(size_t num_params,struct tee_ioctl_param * params)1279*4882a593Smuzhiyun static TEEC_Result ree_fs_new_closedir(size_t num_params,
1280*4882a593Smuzhiyun 				       struct tee_ioctl_param *params)
1281*4882a593Smuzhiyun {
1282*4882a593Smuzhiyun 	if (num_params != 1 ||
1283*4882a593Smuzhiyun 	    (params[0].attr & TEE_IOCTL_PARAM_ATTR_TYPE_MASK) !=
1284*4882a593Smuzhiyun 			TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT)
1285*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1286*4882a593Smuzhiyun 
1287*4882a593Smuzhiyun 	dir_seek = 0;
1288*4882a593Smuzhiyun 	dir_num = 0;
1289*4882a593Smuzhiyun 
1290*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1291*4882a593Smuzhiyun }
1292*4882a593Smuzhiyun 
ree_fs_new_readdir(size_t num_params,struct tee_ioctl_param * params)1293*4882a593Smuzhiyun static TEEC_Result ree_fs_new_readdir(size_t num_params,
1294*4882a593Smuzhiyun 				      struct tee_ioctl_param *params)
1295*4882a593Smuzhiyun {
1296*4882a593Smuzhiyun 	char *dirname;
1297*4882a593Smuzhiyun 	size_t len;
1298*4882a593Smuzhiyun 	size_t dirname_len;
1299*4882a593Smuzhiyun 
1300*4882a593Smuzhiyun 	dirname = (char *)(size_t)params[1].u.memref.shm_id;
1301*4882a593Smuzhiyun 	debug("TEEC: params[1].u.memref.shm_id = 0x%llx params[1].u.memref.shm_offs = 0x%llx\n",
1302*4882a593Smuzhiyun 		params[1].u.memref.shm_id, params[1].u.memref.shm_offs);
1303*4882a593Smuzhiyun 	if (!dirname)
1304*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1305*4882a593Smuzhiyun 	len = params[1].u.memref.size;
1306*4882a593Smuzhiyun 
1307*4882a593Smuzhiyun 	debug("TEEC: seek/num:%d/%d\n", dir_seek, dir_num);
1308*4882a593Smuzhiyun 	if (dir_seek == dir_num) {
1309*4882a593Smuzhiyun 		params[1].u.memref.size = 0;
1310*4882a593Smuzhiyun 		debug("TEEC: %s: END\n", __func__);
1311*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_STATE;
1312*4882a593Smuzhiyun 	}
1313*4882a593Smuzhiyun 
1314*4882a593Smuzhiyun 	dirname_len = strlen(dir_cache[dir_seek]) + 1;
1315*4882a593Smuzhiyun 	params[1].u.memref.size = dirname_len;
1316*4882a593Smuzhiyun 	if (dirname_len > len)
1317*4882a593Smuzhiyun 		return TEEC_ERROR_SHORT_BUFFER;
1318*4882a593Smuzhiyun 
1319*4882a593Smuzhiyun 	strcpy(dirname, dir_cache[dir_seek]);
1320*4882a593Smuzhiyun 	++dir_seek;
1321*4882a593Smuzhiyun 
1322*4882a593Smuzhiyun 	debug("TEEC: %s: %s\n", __func__, dirname);
1323*4882a593Smuzhiyun 
1324*4882a593Smuzhiyun 	return TEEC_SUCCESS;
1325*4882a593Smuzhiyun }
1326*4882a593Smuzhiyun 
tee_supp_rk_fs_init_v2(void)1327*4882a593Smuzhiyun int tee_supp_rk_fs_init_v2(void)
1328*4882a593Smuzhiyun {
1329*4882a593Smuzhiyun 	assert(sizeof(struct rkss_file_table) == RKSS_TABLE_SIZE);
1330*4882a593Smuzhiyun 	assert(RKSS_DATA_LEN / sizeof(struct rkss_file_table) == RKSS_EACH_FILEFOLDER_COUNT);
1331*4882a593Smuzhiyun 
1332*4882a593Smuzhiyun 	if (check_security_exist(0) < 0)
1333*4882a593Smuzhiyun 		return 0;
1334*4882a593Smuzhiyun 
1335*4882a593Smuzhiyun 	/* clean secure storage */
1336*4882a593Smuzhiyun #ifdef DEBUG_CLEAN_RKSS
1337*4882a593Smuzhiyun 	if (rkss_storage_reset() < 0)
1338*4882a593Smuzhiyun 		return -1;
1339*4882a593Smuzhiyun #endif
1340*4882a593Smuzhiyun 
1341*4882a593Smuzhiyun 	for (uint32_t i = 0; i < RKSS_ACTIVE_AREA_NUM; i++) {
1342*4882a593Smuzhiyun 		if (rkss_storage_init(i) < 0)
1343*4882a593Smuzhiyun 			return -1;
1344*4882a593Smuzhiyun 	}
1345*4882a593Smuzhiyun 
1346*4882a593Smuzhiyun #ifdef DEBUG_RKSS
1347*4882a593Smuzhiyun 	rkss_dump_ptable();
1348*4882a593Smuzhiyun 	rkss_dump_usedflags();
1349*4882a593Smuzhiyun #endif
1350*4882a593Smuzhiyun 
1351*4882a593Smuzhiyun 	return 0;
1352*4882a593Smuzhiyun }
1353*4882a593Smuzhiyun 
tee_supp_param_is_value(struct tee_ioctl_param * param)1354*4882a593Smuzhiyun static bool tee_supp_param_is_value(struct tee_ioctl_param *param)
1355*4882a593Smuzhiyun {
1356*4882a593Smuzhiyun 	switch (param->attr & TEE_IOCTL_PARAM_ATTR_TYPE_MASK) {
1357*4882a593Smuzhiyun 	case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT:
1358*4882a593Smuzhiyun 	case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_OUTPUT:
1359*4882a593Smuzhiyun 	case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INOUT:
1360*4882a593Smuzhiyun 		return true;
1361*4882a593Smuzhiyun 	default:
1362*4882a593Smuzhiyun 		return false;
1363*4882a593Smuzhiyun 	}
1364*4882a593Smuzhiyun }
1365*4882a593Smuzhiyun 
1366*4882a593Smuzhiyun static int rkss_step;
tee_supp_rk_fs_process_v2(size_t num_params,struct tee_ioctl_param * params)1367*4882a593Smuzhiyun int tee_supp_rk_fs_process_v2(size_t num_params,
1368*4882a593Smuzhiyun 			struct tee_ioctl_param *params)
1369*4882a593Smuzhiyun {
1370*4882a593Smuzhiyun 	uint32_t ret;
1371*4882a593Smuzhiyun 
1372*4882a593Smuzhiyun 	if (!num_params || !tee_supp_param_is_value(params))
1373*4882a593Smuzhiyun 		return TEEC_ERROR_BAD_PARAMETERS;
1374*4882a593Smuzhiyun 
1375*4882a593Smuzhiyun 	switch (params->u.value.a) {
1376*4882a593Smuzhiyun 	case OPTEE_MRF_OPEN:
1377*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_OPEN!\n", rkss_step++);
1378*4882a593Smuzhiyun 		ret = ree_fs_new_open(num_params, params);
1379*4882a593Smuzhiyun 		break;
1380*4882a593Smuzhiyun 	case OPTEE_MRF_CREATE:
1381*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_CREATE!\n", rkss_step++);
1382*4882a593Smuzhiyun 		ret = ree_fs_new_create(num_params, params);
1383*4882a593Smuzhiyun 		break;
1384*4882a593Smuzhiyun 	case OPTEE_MRF_CLOSE:
1385*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_CLOSE!\n", rkss_step++);
1386*4882a593Smuzhiyun 		ret = ree_fs_new_close(num_params, params);
1387*4882a593Smuzhiyun 		break;
1388*4882a593Smuzhiyun 	case OPTEE_MRF_READ:
1389*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_READ!\n", rkss_step++);
1390*4882a593Smuzhiyun 		ret = ree_fs_new_read(num_params, params);
1391*4882a593Smuzhiyun 		break;
1392*4882a593Smuzhiyun 	case OPTEE_MRF_WRITE:
1393*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_WRITE!\n", rkss_step++);
1394*4882a593Smuzhiyun 		ret = ree_fs_new_write(num_params, params);
1395*4882a593Smuzhiyun 		break;
1396*4882a593Smuzhiyun 	case OPTEE_MRF_TRUNCATE:
1397*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_TRUNCATE!\n", rkss_step++);
1398*4882a593Smuzhiyun 		ret = ree_fs_new_truncate(num_params, params);
1399*4882a593Smuzhiyun 		break;
1400*4882a593Smuzhiyun 	case OPTEE_MRF_REMOVE:
1401*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_REMOVE!\n", rkss_step++);
1402*4882a593Smuzhiyun 		ret = ree_fs_new_remove(num_params, params);
1403*4882a593Smuzhiyun 		break;
1404*4882a593Smuzhiyun 	case OPTEE_MRF_RENAME:
1405*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_RENAME!\n", rkss_step++);
1406*4882a593Smuzhiyun 		ret = ree_fs_new_rename(num_params, params);
1407*4882a593Smuzhiyun 		break;
1408*4882a593Smuzhiyun 	case OPTEE_MRF_OPENDIR:
1409*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_OPENDIR!\n", rkss_step++);
1410*4882a593Smuzhiyun 		ret = ree_fs_new_opendir(num_params, params);
1411*4882a593Smuzhiyun 		break;
1412*4882a593Smuzhiyun 	case OPTEE_MRF_CLOSEDIR:
1413*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_CLOSEDIR!\n", rkss_step++);
1414*4882a593Smuzhiyun 		ret = ree_fs_new_closedir(num_params, params);
1415*4882a593Smuzhiyun 		break;
1416*4882a593Smuzhiyun 	case OPTEE_MRF_READDIR:
1417*4882a593Smuzhiyun 		debug(">>>>>>> [%d] OPTEE_MRF_READDIR!\n", rkss_step++);
1418*4882a593Smuzhiyun 		ret = ree_fs_new_readdir(num_params, params);
1419*4882a593Smuzhiyun 		break;
1420*4882a593Smuzhiyun 	default:
1421*4882a593Smuzhiyun 		ret = TEEC_ERROR_BAD_PARAMETERS;
1422*4882a593Smuzhiyun 		break;
1423*4882a593Smuzhiyun 	}
1424*4882a593Smuzhiyun 	rkss_storage_write();
1425*4882a593Smuzhiyun 	return ret;
1426*4882a593Smuzhiyun }
1427