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