Repository navigation
Expand file tree
/
Copy pathminieap.c
More file actions
219 lines (176 loc) · 4.98 KB
/
Copy pathminieap.c
File metadata and controls
219 lines (176 loc) · 4.98 KB
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
#include "config.h"
#include "if_impl.h"
#include "packet_plugin.h"
#include "packet_builder.h"
#include "logging.h"
#include "eap_state_machine.h"
#include "sched_alarm.h"
#include "misc.h"
#include "conf_parser.h"
#include "pid_lock.h"
#include <stdlib.h>
#include <errno.h>
#include <arpa/inet.h>
#include <signal.h>
#include <time.h>
#ifdef __linux__
#include <linux/if_ether.h>
#else
#define ETH_P_PAE 0x888e
#endif
/*
* Initialize the settings.
* Note: Override values in config file with cmdline.
*/
static int init_program_config(int argc, char* argv[]) {
PROG_CONFIG* cfg;
load_default_params();
if (IS_FAIL(parse_cmdline_conf_file(argc, argv))) {
PR_ERR("配置文件路径解析出错");
goto err;
}
cfg = get_program_config();
if (IS_FAIL(parse_config_file(cfg->conffile))) {
PR_WARN("配置文件解析出错，请注意命令行参数是否完整");
}
if (IS_FAIL(parse_cmdline_opts(argc, argv))) {
PR_ERR("命令行参数解析出错");
goto err;
}
return validate_params();
err:
return FAILURE;
}
/*
* Initialize the settings. Called after plugins being selected.
* Note: Override values in config file with cmdline.
*/
static int init_plugin_config(int argc, char* argv[]) {
PROG_CONFIG *cfg = get_program_config();
packet_plugin_load_default_params();
if (IS_FAIL(packet_plugin_process_config_file(cfg->conffile))) {
PR_ERR("插件配置文件内容解析出错");
goto err;
}
if (IS_FAIL(packet_plugin_process_cmdline_opts(argc, argv))) {
PR_ERR("插件命令行参数解析出错");
goto err;
}
return packet_plugin_validate_params();
err:
return FAILURE;
}
static void packet_plugin_list_select(void* name, void* unused) {
if (IS_FAIL(select_packet_plugin((const char*)name))) {
PR_WARN("%s 插件未找到，请检查插件名称是否拼写正确", name);
}
}
/*
* Do all the initialization tasks
*/
static int init_cfg(int argc, char* argv[]) {
PROG_CONFIG* cfg = get_program_config();
/* Temporaory logs */
set_log_destination(LOG_TO_CONSOLE);
start_log();
init_if_impl_list();
init_packet_plugin_list();
PR_RAW("MiniEAP " VERSION "\n"
"Hamster Tian, 2016\n\n");
if (IS_FAIL(init_program_config(argc, argv))) {
PR_ERR("参数初始化错误");
return FAILURE;
}
list_traverse(cfg->packet_plugin_list, packet_plugin_list_select, NULL);
if (IS_FAIL(select_if_impl(cfg->if_impl))) {
PR_ERR("网络驱动插件启用失败，请检查插件名称是否拼写正确");
return FAILURE;
}
packet_plugin_print_banner();
if (IS_FAIL(init_plugin_config(argc, argv))) {
PR_ERR("插件初始化错误");
return FAILURE;
}
if (cfg->save_now) {
save_config_file();
}
/* Parsed in parse_config_file(). This is no longer needed */
conf_parser_free();
if (IS_FAIL(pid_lock_init(cfg->pidfile))) {
return FAILURE;
}
return SUCCESS;
}
static int init_if() {
PROG_CONFIG* cfg = get_program_config();
IF_IMPL* if_impl;
if_impl = get_if_impl();
if (IS_FAIL(if_impl->set_ifname(if_impl,cfg->ifname))) {
PR_ERR("设置接口名称失败");
return FAILURE;
}
if (IS_FAIL(if_impl->setup_capture_params(if_impl, ETH_P_PAE, FALSE))) {
PR_ERR("设置捕获参数失败");
return FAILURE;
}
if (IS_FAIL(if_impl->prepare_interface(if_impl))) {
PR_ERR("捕获准备失败");
return FAILURE;
}
if_impl->set_frame_handler(if_impl, eap_state_machine_recv_handler);
return SUCCESS;
}
static void apply_log_daemon_params() {
PROG_CONFIG* cfg = get_program_config();
if (cfg->daemon_type != DAEMON_FOREGROUND) {
PR_INFO("正在转入后台运行……");
if (IS_FAIL(go_background())) {
PR_WARN("无法转入后台运行");
}
}
/* Apply log destination */
close_log();
start_log();
}
static void exit_handler() {
free_if_impl();
packet_plugin_destroy();
eap_state_machine_destroy();
sched_alarm_destroy();
pid_lock_destroy();
free_config();
PR_INFO("MiniEAP 已退出");
close_log();
};
static void signal_handler(int signal) {
exit(0);
}
/*
* Detailed errors are printed where they happen, not here ...
*/
int main(int argc, char* argv[]) {
srand(time(0));
atexit(exit_handler);
signal(SIGHUP, signal_handler);
signal(SIGINT, signal_handler);
signal(SIGTERM, signal_handler);
if (IS_FAIL(init_cfg(argc, argv))) {
return FAILURE;
}
if (IS_FAIL(init_if())) {
return FAILURE;
}
if (IS_FAIL(eap_state_machine_init())) {
return FAILURE;
}
if (IS_FAIL(sched_alarm_init())) {
return FAILURE;
}
if (IS_FAIL(pid_lock_lock())) {
return FAILURE;
}
apply_log_daemon_params();
pid_lock_save_pid();
switch_to_state(EAP_STATE_PREPARING, NULL);
return 0;
}