1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#include <linux/module.h>
#include <linux/types.h>
#include <linux/kthread.h>
#include <linux/freezer.h>
#include "trustzone/tz_cross/trustzone.h"
#include "trustzone/tz_cross/ta_icnt.h"
#include "trustzone/kree/system.h"
#include "kree_int.h"
#include "tz_counter.h"
#ifdef ENABLE_INC_ONLY_COUNTER
uint32_t TEECK_Icnt_Counter(KREE_SESSION_HANDLE session, uint32_t* a, uint32_t* b)
{
MTEEC_PARAM param[4];
uint32_t paramTypes;
TZ_RESULT ret;
paramTypes = TZ_ParamTypes2(TZPT_VALUE_OUTPUT, TZPT_VALUE_OUTPUT);
ret = KREE_TeeServiceCall(session, TZCMD_ICNT_COUNT,
paramTypes, param);
if (ret != TZ_RESULT_SUCCESS)
{
printk("ServiceCall error %d\n", ret);
}
*a = param[0].value.a;
*b = param[1].value.a;
return ret;
}
uint32_t TEECK_Icnt_Rate(KREE_SESSION_HANDLE session, uint32_t* a)
{
MTEEC_PARAM param[4];
uint32_t paramTypes;
TZ_RESULT ret;
paramTypes = TZ_ParamTypes1(TZPT_VALUE_OUTPUT);
ret = KREE_TeeServiceCall(session, TZCMD_ICNT_RATE,
paramTypes, param);
if (ret != TZ_RESULT_SUCCESS)
{
printk("ServiceCall error %d\n", ret);
}
*a = param[0].value.a;
return ret;
}
#define THREAD_COUNT_FREQ 10
int update_counter_thread(void *data)
{
TZ_RESULT ret;
KREE_SESSION_HANDLE icnt_session;
uint32_t result;
uint32_t a, b, rate;
uint32_t nsec = THREAD_COUNT_FREQ;
ret = KREE_CreateSession(TZ_TA_ICNT_UUID, &icnt_session);
if (ret != TZ_RESULT_SUCCESS)
{
printk("CreateSession error %d\n", ret);
return 1;
}
result = TEECK_Icnt_Rate(icnt_session, &rate);
if (result == TZ_RESULT_SUCCESS)
{
//printk("(yjdbg) rate: %d\n", rate);
nsec = (0xffffffff / rate);
nsec -= 600;
//printk("(yjdbg) rate: %d\n", nsec);
}
set_freezable();
for (;;) {
if (kthread_should_stop())
break;
if (try_to_freeze())
continue;
result = TEECK_Icnt_Counter(icnt_session, &a, &b);
if (result == TZ_RESULT_SUCCESS)
{
//printk("(yjdbg) tz_test TZCMD_ICNT_COUNT: 0x%x, 0x%x\n", a, b);
}
schedule_timeout_interruptible(HZ * nsec);
}
ret = KREE_CloseSession(icnt_session);
if (ret != TZ_RESULT_SUCCESS)
{
printk("CloseSession error %d\n", ret);
return 1;
}
return 0;
}
#endif
|
