上篇已经讲过client端的CODE
这部分code主要用来接收client端发送来的日志,从数据库中读取reglar然后去匹配.
如果出现匹配则判断为XSS攻击.
server端的SOCKET接收用了coro相关的模块.
配置文件仿照前一篇博客读取即可.
|
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
|
#!/usr/bin/perl
use
warnings;
use
strict;
use
AnyEvent;
use
AnyEvent::DBI::MySQL;
use
Config::Tiny;
use
FindBin;
use
utf8;
use
Coro;
use
Coro::Socket;
use
Coro::Handle;
use
lib
"$FindBin::Bin/../module"
;
my
$server_config_file
=
"$FindBin::Bin/../etc/config.ini"
;
my
$config
= Config::Tiny->new;
my
$server_config
=
$config
->
read
(
$server_config_file
);
my
$server_log_info
=
$server_config
->{
'server_config_info'
};
my
$username
=
$server_log_info
->{
'username'
};
my
$password
=
$server_log_info
->{
'password'
};
my
$port
=
$server_log_info
->{
'server_port'
};
my
$host
=
$server_log_info
->{
'host'
};
my
$database
=
$server_log_info
->{
'database'
};
my
$server_ip
=
$server_log_info
->{
'server_ip'
};
$|++;
print
"Start listening Port:$port"
,
"\n"
;
my
$s
= Coro::Socket->new(
LocalAddr
=>
$server_ip
,
# 创建一个侦听socket
LocalPort
=>
$port
,
Listen
=> 5,
Proto
=>
'tcp'
) or
die
$@;
my
@coro
;
while
(1) {
my
(
$fh
,
$peername
) =
$s
->
accept
;
next
unless
$peername
;
&doit
(
$fh
);
}
sub
doit {
my
$dbh
= AnyEvent::DBI::MySQL->
connect
(
"dbi:mysql:database=$database"
,
"$username"
,
"$password"
);
my
$fh
=
shift
;
push
@coro
, async {
$fh
->autoflush(1);
while
(
my
$line
=
$fh
->
readline
() ) {
log_regex_do(
$line
,
$dbh
)
}
$fh
->
close
;
}
}
sub
log_regex_do {
my
(
$log
,
$dbh
,
$cv
) =
@_
;
my
(
$log_type
,
$url
,
$source
,
$local
,
$date
,
$option
,
$offer
,
$user
) =
$log
=~
/t:(.*)\|me:(.*)\|so:(.*)\|lo:(.*)\|date:(.*)\|opt:(.*)\|of:(.*)\|u:(.*)$/;
my
$log_class
= type_result(
$log_type
,
$dbh
,
$cv
);
if
(
defined
bool(
$local
,
$user
,
$dbh
) ) {
my
(
$log_result
) = log_result(
$url
,
$dbh
);
if
(
$log_result
) {
get_result_db(
$log_type
,
$log_result
,
$url
,
$source
,
$user
,
$date
,
$option
,
$offer
,
$dbh
);
}
}
}
sub
type_result {
my
(
$method
,
$dbh
) =
@_
;
my
$cvs
= AnyEvent->condvar;
my
$type
;
$dbh
->
do
(
"set names utf8"
);
$dbh
->selectall_hashref(
"select * from w3a_log_monitor_type"
,
'id'
,
sub
{
my
(
$ary_ref
) =
@_
;
for
my
$id
(
keys
%
$ary_ref
) {
$type
=
$ary_ref
->{
$id
}->{
'id'
}
if
(
$method
eq
$ary_ref
->{
$id
}->{
'log_type_name'
} );
}
$cvs
->
send
;
}
);
$cvs
->
recv
;
return
$type
;
}
sub
bool {
my
(
$local
,
$user
,
$dbh
) =
@_
;
my
$cv
= AnyEvent->condvar;
my
$count
;
# $dbh->do("set names utf8");
$dbh
->selectcol_arrayref(
"select * from w3a_log_monitor where task_name='$user' and task_url='$local'"
,
sub
{
my
(
$ref_ary
) =
@_
;
$count
= @
$ref_ary
;
$cv
->
send
;
}
);
$cv
->
recv
;
return
$count
;
}
sub
get_result_db {
my
(
$type
,
$method_id
,
$method_url
,
$method_source
,
$method_user
,
$method_date
,
$method_option
,
$method_offer
,
$dbh
) =
@_
;
my
$cv
= AnyEvent->condvar;
my
$sth
=
$dbh
->prepare( "
insert into w3a_log_monitor_attack (
method_name,
method_url,
attack_source,
attack_user,
attack_date,
attack_option,
attack_offer,
log_type
)
values
(?,?,?,?,?,?,?,?)
" );
$sth
->bind_param( 1,
$method_id
);
$sth
->bind_param( 2,
$method_url
);
$sth
->bind_param( 3,
$method_source
);
$sth
->bind_param( 4,
$method_user
);
$sth
->bind_param( 5,
$method_date
);
$sth
->bind_param( 6,
$method_option
);
$sth
->bind_param( 7,
$method_offer
);
$sth
->bind_param( 8,
$type
);
$sth
->execute(
sub
{
my
(
$rv
) =
@_
;
$cv
->
send
;
}
);
$cv
->
recv
;
}
sub
log_result {
my
(
$method
,
$dbh
) =
@_
;
my
$cv
= AnyEvent->condvar;
my
$sum_dbh
=
$dbh
;
my
@target_id
;
$dbh
->
do
(
"set names utf8"
);
$dbh
->selectall_hashref(
"select * from w3a_log_method"
,
'id'
,
sub
{
my
(
$ary_ref
) =
@_
;
for
my
$id
(
keys
%
$ary_ref
) {
$cv
->begin;
my
$switch
=
$ary_ref
->{
$id
}->{
'method_switch'
};
unless
(
$switch
== 0 ) {
if
(
$method
=~ /
$ary_ref
->{
$id
}->{
'method_regex'
}/i ) {
print
"Match regular is: "
,
$ary_ref
->{
$id
}->{
'method_regex'
},
"\n"
;
push
@target_id
,
$ary_ref
->{
$id
}->{
'id'
};
}
}
$cv
->end;
}
}
);
$cv
->
recv
;
attack_update(
$_
,
$dbh
)
for
@target_id
;
return
@target_id
;
}
sub
attack_update {
my
(
$id
,
$dbh
) =
@_
;
my
$cv
= AnyEvent->condvar;
$dbh
->selectcol_arrayref(
"select attack_sum from w3a_log_method where id='$id' "
,
sub
{
my
(
$ref_ary
) =
@_
;
my
$sum
=
$ref_ary
->[0] + 1;
$dbh
->
do
(
"update w3a_log_method set attack_sum='$sum' where id='$id'"
);
$cv
->
send
;
}
);
$cv
->
recv
;
}
|
本文转自 mcshell 51CTO博客，原文链接：http://blog.51cto.com/mcshell/1362689，如需转载请自行联系原作者
