summaryrefslogtreecommitdiff
path: root/driver/sspt/ip.c
blob: c6e59ceb404a0467bb61691965c74332c74291a7 (plain)
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
/*
 *  Dynamic Binary Instrumentation Module based on KProbes
 *  modules/driver/sspt/ip.c
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * Copyright (C) Samsung Electronics, 2013
 *
 * 2013         Vyacheslav Cherkashin <v.cherkashin@samsung.com>
 *
 */

#include "ip.h"
#include "sspt_page.h"
#include "sspt_file.h"

struct us_ip *create_ip(unsigned long offset)
{
	struct us_ip *ip = kmalloc(sizeof(*ip), GFP_ATOMIC);
	memset(ip, 0, sizeof(*ip));

	INIT_LIST_HEAD(&ip->list);
	ip->offset = offset;

	return ip;
}

void free_ip(struct us_ip *ip)
{
	kfree(ip);
}

static inline void set_ip_jp_handler(struct us_ip *ip, kprobe_pre_entry_handler_t per_entry, void *entry)
{
	ip->jprobe.pre_entry = per_entry;
	ip->jprobe.entry = entry;
}

static inline void set_ip_rp_handler(struct us_ip *ip, kretprobe_handler_t handler)
{
	ip->flag_retprobe = 1;
	ip->retprobe.handler = handler;
}

static inline void set_ip_got_addr(struct us_ip *ip, unsigned long got_addr)
{
	ip->got_addr = got_addr;
}

struct us_ip *copy_ip(const struct us_ip *ip)
{
	struct us_ip *new_ip = create_ip(ip->offset);

	if (new_ip == NULL) {
		printk("us_proc_ip_copy: No enough memory\n");
		return NULL;
	}

	// jprobe
	set_ip_jp_handler(new_ip, ip->jprobe.pre_entry, ip->jprobe.entry);

	// retprobe
	if (ip->flag_retprobe) {
		set_ip_rp_handler(new_ip, ip->retprobe.handler);
	}

	return new_ip;
}

struct us_ip *create_ip_by_ip_data(struct ip_data *ip_d)
{
	struct us_ip *ip = create_ip(ip_d->offset);
	set_ip_jp_handler(ip, ip_d->pre_handler, (void *)ip_d->jp_handler);

	if (ip_d->flag_retprobe) {
		set_ip_rp_handler(ip, ip_d->rp_handler);
	}

	set_ip_got_addr(ip, ip_d->got_addr);

	return ip;
}

void sspt_set_ip_addr(struct us_ip *ip, const struct sspt_page *page, const struct sspt_file *file)
{
	unsigned long addr = file->vm_start + page->offset + ip->offset;
	ip->retprobe.kp.addr = ip->jprobe.kp.addr = (kprobe_opcode_t *)addr;
}