blob: c16879f5436326319957230233e4874f4a95f694 (
plain) (
blame)
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
|
#include <linux/module.h>
#include <litmus/sched_trace.h>
#include <litmus/nvidia_info.h>
static struct module* nvidia_mod = NULL;
int init_nvidia_info(void)
{
mutex_lock(&module_mutex);
nvidia_mod = find_module("nvidia");
mutex_unlock(&module_mutex);
if(nvidia_mod != NULL)
{
TRACE("%s : Found NVIDIA module. Core Code: %p to %p\n", __FUNCTION__,
(void*)(nvidia_mod->module_core),
(void*)(nvidia_mod->module_core) + nvidia_mod->core_size);
return(0);
}
else
{
TRACE("%s : Could not find NVIDIA module! Loaded?\n", __FUNCTION__);
return(-1);
}
}
/* works with pointers to static data inside the module too. */
int is_nvidia_func(void *func_addr)
{
int ret = 0;
if(nvidia_mod)
{
ret = within_module_core((long unsigned int)func_addr, nvidia_mod);
TRACE("%s : %p is in NVIDIA module : %d\n",
__FUNCTION__, func_addr, ret);
}
else
{
TRACE("%s : No NVIDIA module info loaded!\n", __FUNCTION__);
}
return(ret);
}
int is_nvidia_in_callstack(void)
{
/* TODO -- Walk back through the callstack, running each called
function through is_nvidia_func() */
TRACE("%s : NOT IMPLEMENTED!\n", __FUNCTION__);
return(1);
}
u32 get_nv_device_num(const struct tasklet_struct *t)
{
// offset determined though observed behavior of the NV driver.
const int DEVICE_NUM_OFFSET = 0x278;
void* state = (void*)(t->data);
void* device_num_ptr = state + DEVICE_NUM_OFFSET;
return(*((u32*)device_num_ptr));
}
|
