-
Notifications
You must be signed in to change notification settings - Fork 8
/
lldb-auto.h
121 lines (106 loc) · 2.59 KB
/
lldb-auto.h
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#ifndef LLDBAUTOMATION_H
#define LLDBAUTOMATION_H
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <unistd.h>
#include <condition_variable>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <mutex>
#include <queue>
#include <sstream>
#include <string>
#include "api.h"
struct WPInfo
{
unsigned long long address = 0;
int bpsize = 0;
char type;
bool switch_ = false;
bool enabled = false;
};
struct CustomDebugEvent
{
int debugevent;
int threadid;
unsigned long long address;
std::vector<uint64_t> register_;
};
class ThreadSafeQueue
{
private:
std::queue<std::pair<int, int>> q;
std::mutex m;
std::condition_variable cv;
public:
void put(std::pair<int, int> item)
{
std::unique_lock<std::mutex> lock(m);
q.push(item);
cv.notify_one();
}
std::pair<int, int> get()
{
std::unique_lock<std::mutex> lock(m);
while (q.empty())
{
cv.wait(lock);
}
auto val = q.front();
q.pop();
return val;
}
};
template <typename T>
T hexStringToNumber(const std::string &hexString, bool is_reverse)
{
T result;
std::string HexString;
if (is_reverse)
{
for (std::size_t i = 0; i < hexString.length(); i += 2)
{
HexString = hexString.substr(i, 2) + HexString;
}
}
else
{
HexString = hexString;
}
std::istringstream iss(HexString);
iss >> std::hex >> result;
return result;
}
class LLDBAutomation
{
private:
int s;
sockaddr_in serv_addr;
pid_t attach_pid;
std::string calc_checksum(const std::string &message);
std::mutex Lock;
public:
std::vector<CustomDebugEvent> debugevent;
std::vector<uint64_t> register_info;
ThreadSafeQueue continue_queue;
std::vector<WPInfo> wpinfo;
bool is_stopped;
LLDBAutomation(const std::string &server_ip, int server_port);
void disable_ack();
std::string send_message(const std::string &message, bool recvflag = true);
void attach(pid_t pid);
std::string cont();
std::string cont2(int signal, int thread);
std::string step(int thread);
std::string readmem(unsigned long long address, size_t size);
bool set_watchpoint(unsigned long long address, int size, char _type);
bool remove_watchpoint(unsigned long long address, int size, char _type);
std::map<std::string, std::string> parse_result(const std::string &result);
void interrupt();
void debugger_thread();
void interrupt_func();
};
#endif // LLDBAUTOMATION_H