Repository navigation
Expand file tree
/
Copy pathminieap.c
More file actions
177 lines (146 loc) · 4.11 KB
/
Copy pathminieap.c
File metadata and controls
177 lines (146 loc) · 4.11 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
#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 <stdlib.h>
#include <errno.h>
#include <arpa/inet.h>
#include <linux/if_ether.h>
#include <signal.h>
#include <time.h>
/*
* Initialize the settings.
* Note: Override values in config file with cmdline.
*/
static int init_program_config(int argc, char* argv[]) {
PROG_CONFIG* cfg;
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_ERR("配置文件内容解析出错");
goto err;
}
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();
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) {
select_packet_plugin((const char* )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();
load_default_params();
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_load_default_params();
if (IS_FAIL(init_plugin_config(argc, argv))) {
PR_ERR("插件初始化错误");
return FAILURE;
}
if (cfg->run_in_background && IS_FAIL(go_background())) {
PR_WARN("无法应用后台运行设置");
}
/* Apply log destination */
close_log();
start_log();
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 exit_handler() {
free_config();
free_if_impl();
packet_plugin_destroy();
eap_state_machine_destroy();
sched_alarm_destroy();
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;
}
switch_to_state(EAP_STATE_PREPARING, NULL);
return 0;
}