|
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
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
1
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
;
}
|
因篇幅问题不能全部显示，请点此查看更多更全内容