rev |
line source |
devi@52
|
1 #!/usr/bin/perl -w
|
devi@23
|
2
|
devi@23
|
3 #
|
devi@62
|
4 # (c) Igor Chubin, igor@chub.in, 2004-2006
|
devi@23
|
5 #
|
devi@23
|
6
|
devi@50
|
7
|
devi@50
|
8 ## Эта строчка добавлена из блокнота Windows
|
devi@50
|
9 ## Надо отдать должное, он каким-то образом научился понимать кодировку
|
devi@50
|
10
|
devi@23
|
11 use strict;
|
devi@25
|
12 use POSIX;
|
devi@23
|
13 use Term::VT102;
|
devi@23
|
14 use Text::Iconv;
|
devi@23
|
15 use Time::Local 'timelocal_nocheck';
|
devi@27
|
16 use IO::Socket;
|
devi@23
|
17
|
devi@32
|
18 use lib "/usr/local/bin";
|
devi@23
|
19 use l3config;
|
devi@23
|
20
|
devi@23
|
21
|
devi@23
|
22 our @Command_Lines;
|
devi@23
|
23 our @Command_Lines_Index;
|
devi@28
|
24 our %Diffs;
|
devi@27
|
25 our %Sessions;
|
devi@23
|
26
|
devi@62
|
27 our %Script_Files; # Информация о позициях в скрипт-файлах,
|
devi@62
|
28 # до которых уже выполнен разбор
|
devi@62
|
29 # и информация о времени модификации файла
|
devi@62
|
30 # $Script_Files{$file}->{size}
|
devi@62
|
31 # $Script_Files{$file}->{tell}
|
devi@23
|
32
|
devi@62
|
33 our $Killed =0; # В режиме демона -- процесс получил сигнал о завершении
|
devi@23
|
34
|
devi@23
|
35 sub init_variables;
|
devi@23
|
36 sub main;
|
devi@23
|
37
|
devi@23
|
38 sub load_diff_files;
|
devi@23
|
39 sub bind_diff;
|
devi@62
|
40 sub extract_commands_from_cline;
|
devi@23
|
41 sub load_command_lines;
|
devi@23
|
42 sub sort_command_lines;
|
devi@23
|
43 sub print_command_lines;
|
devi@23
|
44 sub printq;
|
devi@23
|
45
|
devi@25
|
46 sub save_cache_stat;
|
devi@25
|
47 sub load_cache_stat;
|
devi@27
|
48 sub print_session;
|
devi@25
|
49
|
devi@23
|
50 sub load_diff_files
|
devi@23
|
51 {
|
devi@62
|
52 my @pathes = @_;
|
devi@62
|
53
|
devi@62
|
54 for my $path (@pathes) {
|
devi@62
|
55 my $template = "*.diff";
|
devi@62
|
56 my @files = <$path/$template>;
|
devi@62
|
57 my $i=0;
|
devi@62
|
58 for my $file (@files) {
|
devi@28
|
59
|
devi@62
|
60 next if defined($Diffs{$file});
|
devi@62
|
61
|
devi@62
|
62 my %diff;
|
devi@62
|
63
|
devi@62
|
64 $diff{"path"}=$path;
|
devi@62
|
65 $diff{"uid"}="SET THIS";
|
devi@23
|
66
|
devi@23
|
67 # Сейчас UID определяется из названия каталога
|
devi@23
|
68 # откуда берутся diff-файлы
|
devi@23
|
69 # Это неправильно
|
devi@23
|
70 #
|
devi@23
|
71 # ВАРИАНТ:
|
devi@23
|
72 # К файлам жураналам должны прилагаться ситемны файлы,
|
devi@23
|
73 # мз которых и будет определяться соответствие
|
devi@23
|
74 # имён пользователей их uid'ам
|
devi@23
|
75 #
|
devi@62
|
76 $diff{"uid"} = 0 if $path =~ m@/root/@;
|
devi@62
|
77
|
devi@62
|
78 $diff{"bind_to"}="";
|
devi@62
|
79 $diff{"time_range"}=-1;
|
devi@62
|
80
|
devi@62
|
81 next if not $file=~m@/(D?[0-9][0-9]?[0-9]?)[^/]*?([0-9]*):([0-9]*):?([0-9]*)@;
|
devi@62
|
82 $diff{"day"}=$1 || "";
|
devi@62
|
83 $diff{"hour"}=$2;
|
devi@62
|
84 $diff{"min"}=$3;
|
devi@62
|
85 $diff{"sec"}=$4 || 0;
|
devi@62
|
86
|
devi@62
|
87 $diff{"index"}=$i;
|
devi@23
|
88
|
devi@62
|
89 print "diff loaded: $diff{day} $diff{hour}:$diff{min}:$diff{sec}\n";
|
devi@62
|
90
|
devi@62
|
91 local $/;
|
devi@62
|
92 open (F, "$file")
|
devi@62
|
93 or return "Can't open file $file ($_[0]) for reading";
|
devi@62
|
94 my $text = <F>;
|
devi@62
|
95 if ($Config{"encoding"} && $Config{"encoding"} !~ /^utf-8$/i) {
|
devi@62
|
96 my $converter = Text::Iconv->new($Config{"encoding"}, "utf-8");
|
devi@62
|
97 $text = $converter->convert($text);
|
devi@62
|
98 }
|
devi@62
|
99 close(F);
|
devi@62
|
100 $diff{"text"}=$text;
|
devi@62
|
101 #print "$file loaded ($diff{day})\n";
|
devi@23
|
102
|
devi@62
|
103 #push @Diffs, \%diff;
|
devi@62
|
104 $Diffs{$file} = \%diff;
|
devi@62
|
105 $i++;
|
devi@62
|
106 }
|
devi@62
|
107 }
|
devi@23
|
108 }
|
devi@23
|
109
|
devi@23
|
110
|
devi@23
|
111 sub bind_diff
|
devi@23
|
112 {
|
devi@62
|
113 # my $path = shift;
|
devi@62
|
114 # my $pid = shift;
|
devi@62
|
115 # my $day = shift;
|
devi@62
|
116 # my $lab = shift;
|
devi@23
|
117
|
devi@62
|
118 print "Trying to bind diff...\n";
|
devi@23
|
119
|
devi@62
|
120 my $cl = shift;
|
devi@62
|
121 my $hour = $cl->{"hour"};
|
devi@62
|
122 my $min = $cl->{"min"};
|
devi@62
|
123 my $sec = $cl->{"sec"};
|
devi@23
|
124
|
devi@62
|
125 my $min_dt = 10000;
|
devi@23
|
126
|
devi@62
|
127 for my $diff_key (keys %Diffs) {
|
devi@62
|
128 my $diff = $Diffs{$diff_key};
|
devi@62
|
129 # Check here date, time and user
|
devi@62
|
130 next if ($diff->{"day"} && $cl->{"day"} && ($cl->{"day"} ne $diff->{"day"}));
|
devi@62
|
131 #next if (!$diff->{"uid"} && $cl->{"euid"} != $diff->{"uid"});
|
devi@62
|
132
|
devi@62
|
133 my $dt=($diff->{"hour"}-$hour)*3600 +($diff->{"min"}-$min)*60 + ($diff->{"sec"}-$sec);
|
devi@62
|
134 if ($dt >0 && $dt < $min_dt && ($diff->{"time_range"} <0 || $dt < $diff->{"time_range"})) {
|
devi@62
|
135 print "Approppriate diff found: dt=$dt\n";
|
devi@62
|
136 if ($diff->{"bind_to"}) {
|
devi@62
|
137 undef $diff->{"bind_to"}->{"diff"};
|
devi@62
|
138 };
|
devi@62
|
139 $diff->{"time_range"}=$dt;
|
devi@62
|
140 $diff->{"bind_to"}=$cl;
|
devi@23
|
141
|
devi@62
|
142 $cl->{"diff"} = $diff_key;
|
devi@62
|
143 $min_dt = $dt;
|
devi@62
|
144 }
|
devi@62
|
145 }
|
devi@23
|
146 }
|
devi@23
|
147
|
devi@23
|
148
|
devi@62
|
149 sub extract_commands_from_cline
|
devi@23
|
150 # Разобрать командную строку $_[1] и возвратить хэш, содержащий
|
devi@23
|
151 # номер первого появление команды в строке:
|
devi@62
|
152 # команда => первая позиция
|
devi@23
|
153 {
|
devi@62
|
154 my $cline = $_[0];
|
devi@62
|
155 my @lists = split /\;/, $cline;
|
devi@62
|
156
|
devi@62
|
157
|
devi@62
|
158 my @commands = ();
|
devi@62
|
159 for my $list (@lists) {
|
devi@62
|
160 push @commands, split /\|/, $list;
|
devi@62
|
161 }
|
devi@23
|
162
|
devi@62
|
163 my %commands;
|
devi@62
|
164 my %files;
|
devi@62
|
165 my $i=0;
|
devi@62
|
166 for my $command (@commands) {
|
devi@62
|
167 $command =~ /\s*(\S+)\s*(.*)/;
|
devi@62
|
168 if ($1 && $1 eq "sudo" ) {
|
devi@62
|
169 $commands{"$1"}=$i++;
|
devi@62
|
170 $command =~ s/\s*sudo\s+//;
|
devi@62
|
171 }
|
devi@62
|
172 $command =~ /\s*(\S+)\s*(.*)/;
|
devi@62
|
173 if ($1 && !defined $commands{"$1"}) {
|
devi@62
|
174 $commands{"$1"}=$i++;
|
devi@62
|
175 };
|
devi@62
|
176 }
|
devi@62
|
177 return %commands;
|
devi@23
|
178 }
|
devi@23
|
179
|
devi@23
|
180 sub load_command_lines
|
devi@23
|
181 {
|
devi@62
|
182 my $lab_scripts_path = $_[0];
|
devi@62
|
183 my $lab_scripts_mask = $_[1];
|
devi@23
|
184
|
devi@62
|
185 my $cline_re_base = qq'
|
devi@62
|
186 (
|
devi@62
|
187 (?:\\^?([0-9]*C?)) # exitcode
|
devi@62
|
188 (?:_([0-9]+)_)? # uid
|
devi@62
|
189 (?:_([0-9]+)_) # pid
|
devi@62
|
190 (...?) # day
|
devi@62
|
191 (.?.?) # lab
|
devi@62
|
192 \\s # space separator
|
devi@62
|
193 ([0-9][0-9]):([0-9][0-9]):([0-9][0-9]) # time
|
devi@62
|
194 .\\[50D.\\[K # killing symbols
|
devi@62
|
195 (.*?([\$\#]\\s?)) # prompt
|
devi@62
|
196 (.*) # command line
|
devi@62
|
197 )
|
devi@62
|
198 ';
|
devi@62
|
199 #my $cline_re = qr/$cline_re_base(?:$cline_re_base|$)/x;
|
devi@62
|
200 #my $cline_re = qr/(?:$cline_re_base)*$cline_re_base$/x;
|
devi@62
|
201 my $cline_re = qr/$cline_re_base/sx;
|
devi@62
|
202 my $cline_re1 = qr/$cline_re_base\x0D/sx;
|
devi@62
|
203 my $cline_re2 = qr/$cline_re_base$/sx;
|
devi@23
|
204
|
devi@62
|
205 my $vt = Term::VT102->new ( 'cols' => $Config{"terminal_width"},
|
devi@62
|
206 'rows' => $Config{"terminal_height"});
|
devi@62
|
207 my $cline_vt = Term::VT102->new ('cols' => $Config{"terminal_width"},
|
devi@62
|
208 'rows' => $Config{"terminal_height"});
|
devi@23
|
209
|
devi@62
|
210 my $converter = Text::Iconv->new($Config{"encoding"}, "utf-8")
|
devi@62
|
211 if ($Config{"encoding"} && $Config{"encoding"} !~ /^utf-8$/i);
|
devi@62
|
212
|
devi@62
|
213 print "Loading lm-scripts...\n" if $Config{"verbose"} =~ /y/;
|
devi@23
|
214
|
devi@62
|
215 my $file;
|
devi@62
|
216 my $skip_info;
|
devi@23
|
217
|
devi@62
|
218 my $commandlines_loaded =0;
|
devi@62
|
219 my $commandlines_processed =0;
|
devi@23
|
220
|
devi@62
|
221 my @lab_scripts = <$lab_scripts_path/$lab_scripts_mask>;
|
devi@62
|
222 for $file (@lab_scripts){
|
devi@23
|
223
|
devi@62
|
224 # Пропускаем файл, если он не изменялся со времени нашего предудущего прохода
|
devi@62
|
225 my $size = (stat($file))[7];
|
devi@62
|
226 next if ($Script_Files{$file} && $Script_Files{$file}->{size} && $Script_Files{$file}->{size} >= $size);
|
devi@27
|
227
|
devi@27
|
228
|
devi@62
|
229 my $local_session_id;
|
devi@62
|
230 # Начальное значение идентификатора текущего сеанса определяем из имени скрипта
|
devi@62
|
231 # Впоследствии оно может быть уточнено
|
devi@62
|
232 $file =~ m@.*/([^/]*)\.script$@;
|
devi@62
|
233 $local_session_id = $1;
|
devi@27
|
234
|
devi@62
|
235 #Если файл только что появился,
|
devi@62
|
236 #пытаемся найти и загрузить информацию о соответствующей ему сессии
|
devi@62
|
237 if (!$Script_Files{$file}) {
|
devi@62
|
238 my $session_file = $file;
|
devi@62
|
239 $session_file =~ s/\.script/.info/;
|
devi@62
|
240 if (open(SESSION, $session_file)) {
|
devi@62
|
241 local $/;
|
devi@62
|
242 my $data = <SESSION>;
|
devi@62
|
243 close(SESSION);
|
devi@27
|
244
|
devi@62
|
245 for my $session_data ($data =~ m@<session>(.*?)</session>@sg) {
|
devi@62
|
246 my %session;
|
devi@62
|
247 while ($session_data =~ m@<([^>]*?)>(.*?)</\1>@sg) {
|
devi@62
|
248 $session{$1} = $2;
|
devi@62
|
249 }
|
devi@62
|
250 $local_session_id = $session{"local_session_id"} if $session{"local_session_id"};
|
devi@62
|
251 $Sessions{$local_session_id}=\%session;
|
devi@62
|
252 }
|
devi@25
|
253
|
devi@62
|
254 #Загруженную информацию сразу же отправляем в поток
|
devi@62
|
255 print_session($Config{cache}, $local_session_id);
|
devi@62
|
256 }
|
devi@62
|
257 }
|
devi@25
|
258
|
devi@62
|
259 open (FILE, "$file");
|
devi@62
|
260 binmode FILE;
|
devi@23
|
261
|
devi@62
|
262 # Переходим к тому месту, где мы окончили разбор
|
devi@62
|
263 seek (FILE, $Script_Files{$file}->{tell}, 0) if $Script_Files{$file}->{tell};
|
devi@62
|
264 $Script_Files{$file}->{size} = $size;
|
devi@62
|
265 $Script_Files{$file}->{tell} = 0 unless $Script_Files{$file}->{tell};
|
devi@32
|
266
|
devi@62
|
267 $file =~ m@.*/(.*?)-.*@;
|
devi@23
|
268
|
devi@62
|
269 my $tty = $1;
|
devi@62
|
270 my $first_pass = 1;
|
devi@62
|
271 my %cl;
|
devi@62
|
272 my $last_output_length=0;
|
devi@62
|
273 while (<FILE>) {
|
devi@62
|
274 $commandlines_processed++;
|
devi@23
|
275
|
devi@62
|
276 next if s/^Script started on.*?\n//s;
|
devi@23
|
277
|
devi@62
|
278 if (/[0-9][0-9]:[0-9][0-9]:[0-9][0-9].\[[0-9][0-9]D.\[K/ && m/$cline_re/) {
|
devi@62
|
279 s/.*\x0d(?!\x0a)//;
|
devi@62
|
280 m/$cline_re2/gs;
|
devi@23
|
281
|
devi@62
|
282 $commandlines_loaded++;
|
devi@62
|
283 $last_output_length=0;
|
devi@23
|
284
|
devi@62
|
285 # Previous command
|
devi@62
|
286 my %last_cl = %cl;
|
devi@62
|
287 my $err = $2 || "";
|
devi@23
|
288
|
devi@62
|
289 $cl{"local_session_id"} = $local_session_id;
|
devi@62
|
290 # Parse new command
|
devi@62
|
291 $cl{"uid"} = $3;
|
devi@62
|
292 $cl{"euid"} = $cl{"uid"}; # Если в команде обнаружится sudo, euid поменяем на 0
|
devi@62
|
293 $cl{"pid"} = $4;
|
devi@62
|
294 $cl{"day"} = $5;
|
devi@62
|
295 $cl{"lab"} = $6;
|
devi@62
|
296 $cl{"hour"} = $7;
|
devi@62
|
297 $cl{"min"} = $8;
|
devi@62
|
298 $cl{"sec"} = $9;
|
devi@62
|
299 $cl{"fullprompt"} = $10;
|
devi@62
|
300 $cl{"prompt"} = $11;
|
devi@62
|
301 $cl{"raw_cline"} = $12;
|
devi@23
|
302
|
devi@62
|
303 {
|
devi@62
|
304 use bytes;
|
devi@62
|
305 $cl{"raw_start"} = tell (FILE) - length($1);
|
devi@62
|
306 $cl{"raw_output_start"} = tell FILE;
|
devi@62
|
307 }
|
devi@62
|
308 $cl{"raw_file"} = $file;
|
devi@23
|
309
|
devi@62
|
310 $cl{"err"} = 0;
|
devi@62
|
311 $cl{"output"} = "";
|
devi@62
|
312 $cl{"tty"} = $tty;
|
devi@23
|
313
|
devi@62
|
314 $cline_vt->process($cl{"raw_cline"}."\n");
|
devi@62
|
315 $cl{"cline"} = $cline_vt->row_plaintext (1);
|
devi@62
|
316 $cl{"cline"} =~ s/\s*$//;
|
devi@62
|
317 $cline_vt->reset();
|
devi@23
|
318
|
devi@62
|
319 my %commands = extract_commands_from_cline($cl{"cline"});
|
devi@62
|
320 $cl{"euid"}=0 if defined $commands{"sudo"};
|
devi@62
|
321 my @comms = sort { $commands{$a} cmp $commands{$b} } keys %commands;
|
devi@62
|
322 $cl{"last_command"} = $comms[$#comms] || "";
|
devi@23
|
323
|
devi@62
|
324 if (
|
devi@62
|
325 $Config{"suppress_editors"} =~ /^y/i
|
devi@62
|
326 && grep ($_ eq $cl{"last_command"}, @{$Config{"editors"}})
|
devi@62
|
327 || $Config{"suppress_pagers"} =~ /^y/i
|
devi@62
|
328 && grep ($_ eq $cl{"last_command"}, @{$Config{"pagers"}})
|
devi@62
|
329 || $Config{"suppress_terminal"}=~ /^y/i
|
devi@62
|
330 && grep ($_ eq $cl{"last_command"}, @{$Config{"terminal"}})
|
devi@62
|
331 ) {
|
devi@62
|
332 $cl{"suppress_output"} = "1";
|
devi@62
|
333 }
|
devi@62
|
334 else {
|
devi@62
|
335 $cl{"suppress_output"} = "0";
|
devi@62
|
336 }
|
devi@62
|
337 $skip_info = 0;
|
devi@23
|
338
|
devi@23
|
339
|
devi@62
|
340 print " ",$cl{"last_command"};
|
devi@23
|
341
|
devi@62
|
342 # Processing previous command line
|
devi@62
|
343 if ($first_pass) {
|
devi@62
|
344 $first_pass = 0;
|
devi@62
|
345 next;
|
devi@62
|
346 }
|
devi@23
|
347
|
devi@62
|
348 # Error code
|
devi@62
|
349 $last_cl{"raw_end"} = $cl{"raw_start"};
|
devi@62
|
350 $last_cl{"err"}=$err;
|
devi@62
|
351 $last_cl{"err"}=130 if $err eq "^C";
|
devi@23
|
352
|
devi@62
|
353 if (grep ($_ eq $last_cl{"last_command"}, @{$Config{"editors"}})) {
|
devi@62
|
354 bind_diff(\%last_cl);
|
devi@62
|
355 }
|
devi@23
|
356
|
devi@62
|
357 # Output
|
devi@62
|
358 if (!$last_cl{"suppress_output"} || $last_cl{"err"}) {
|
devi@62
|
359 for (my $i=0; $i<$Config{"terminal_height"}; $i++) {
|
devi@62
|
360 my $line= $vt->row_plaintext($i);
|
devi@62
|
361 next if !defined ($line) ; #|| $line =~ /^\s*$/;
|
devi@62
|
362 $line =~ s/\s*$//;
|
devi@62
|
363 $line .= "\n" unless $line =~ /^\s*$/;
|
devi@62
|
364 $last_cl{"output"} .= $line;
|
devi@62
|
365 }
|
devi@62
|
366 }
|
devi@62
|
367 else {
|
devi@62
|
368 $last_cl{"output"}= "";
|
devi@62
|
369 }
|
devi@23
|
370
|
devi@62
|
371 $vt->reset();
|
devi@23
|
372
|
devi@23
|
373
|
devi@62
|
374 # Classifying the command line
|
devi@23
|
375
|
devi@23
|
376
|
devi@62
|
377 # Save
|
devi@62
|
378 if (!$Config{"lab"} || $cl{"lab"} eq $Config{"lab"}) {
|
devi@62
|
379 # Changing encoding
|
devi@62
|
380 for (keys %last_cl) {
|
devi@62
|
381 next if /raw/;
|
devi@62
|
382 $last_cl{$_} = $converter->convert($last_cl{$_})
|
devi@62
|
383 if ($Config{"encoding"} &&
|
devi@62
|
384 $Config{"encoding"} !~ /^utf-8$/i);
|
devi@62
|
385 }
|
devi@62
|
386 push @Command_Lines, \%last_cl;
|
devi@23
|
387
|
devi@62
|
388 # Сохранение позиции в файле, до которой выполнен
|
devi@62
|
389 # успешный разбор
|
devi@62
|
390 $Script_Files{$file}->{tell} = $last_cl{raw_end};
|
devi@62
|
391 }
|
devi@62
|
392 next;
|
devi@62
|
393 }
|
devi@62
|
394 $last_output_length+=length($_);
|
devi@62
|
395 #if (!$cl{"suppress_output"} || $last_output_length < 5000) {
|
devi@62
|
396 if ($last_output_length < 50000) {
|
devi@62
|
397 $vt->process("$_"."\n")
|
devi@62
|
398 }
|
devi@62
|
399 else
|
devi@62
|
400 {
|
devi@62
|
401 if (!$skip_info) {
|
devi@62
|
402 print "($cl{last_command})";
|
devi@62
|
403 $skip_info = 1;
|
devi@62
|
404 }
|
devi@62
|
405 }
|
devi@62
|
406 }
|
devi@62
|
407 close(FILE);
|
devi@23
|
408
|
devi@62
|
409 }
|
devi@62
|
410 if ($Config{"verbose"} =~ /y/) {
|
devi@62
|
411 print "...finished." ;
|
devi@62
|
412 print "Lines loaded: $commandlines_processed\n";
|
devi@62
|
413 print "Command lines: $commandlines_loaded\n";
|
devi@62
|
414 }
|
devi@23
|
415 }
|
devi@23
|
416
|
devi@23
|
417
|
devi@23
|
418
|
devi@62
|
419
|
devi@62
|
420 sub sort_command_lines
|
devi@62
|
421 {
|
devi@62
|
422 print "Sorting command lines...\n" if $Config{"verbose"} =~ /y/;
|
devi@62
|
423
|
devi@62
|
424 # Sort Command_Lines
|
devi@62
|
425 # Write Command_Lines to Command_Lines_Index
|
devi@62
|
426
|
devi@62
|
427 my @index;
|
devi@62
|
428 for (my $i=0;$i<=$#Command_Lines;$i++) {
|
devi@62
|
429 $index[$i]=$i;
|
devi@62
|
430 }
|
devi@62
|
431
|
devi@62
|
432 @Command_Lines_Index = sort {
|
devi@62
|
433 $Command_Lines[$index[$a]]->{"day"} cmp $Command_Lines[$index[$b]]->{"day"} ||
|
devi@62
|
434 $Command_Lines[$index[$a]]->{"hour"} <=> $Command_Lines[$index[$b]]->{"hour"} ||
|
devi@62
|
435 $Command_Lines[$index[$a]]->{"min"} <=> $Command_Lines[$index[$b]]->{"min"} ||
|
devi@62
|
436 $Command_Lines[$index[$a]]->{"sec"} <=> $Command_Lines[$index[$b]]->{"sec"}
|
devi@62
|
437 } @index;
|
devi@62
|
438
|
devi@62
|
439 print "...finished\n" if $Config{"verbose"} =~ /y/;
|
devi@62
|
440
|
devi@62
|
441 }
|
devi@62
|
442
|
devi@23
|
443 sub printq
|
devi@23
|
444 {
|
devi@62
|
445 my $TO = shift;
|
devi@62
|
446 my $text = join "", @_;
|
devi@62
|
447 $text =~ s/&/&/g;
|
devi@62
|
448 $text =~ s/</</g;
|
devi@62
|
449 $text =~ s/>/>/g;
|
devi@62
|
450 print $TO $text;
|
devi@23
|
451 }
|
devi@23
|
452
|
devi@23
|
453
|
devi@23
|
454 =cut
|
devi@23
|
455 Вывести результат обработки журнала.
|
devi@23
|
456 =cut
|
devi@23
|
457
|
devi@23
|
458 sub print_command_lines
|
devi@23
|
459 {
|
devi@62
|
460 my $output_filename=$_[0];
|
devi@62
|
461 my $mode = ">";
|
devi@62
|
462 $mode =">>" if $Config{mode} eq "daemon";
|
devi@62
|
463 open(OUT, $mode, $output_filename)
|
devi@62
|
464 or die "Can't open $output_filename for writing\n";
|
devi@23
|
465
|
devi@23
|
466
|
devi@62
|
467 my $cl;
|
devi@62
|
468 my $in_range=0;
|
devi@62
|
469 for my $i (@Command_Lines_Index) {
|
devi@62
|
470 $cl = $Command_Lines[$i];
|
devi@23
|
471
|
devi@62
|
472 if ($Config{"from"} && $cl->{"cline"} =~ /$Config{"signature"}\s*$Config{"from"}/) {
|
devi@62
|
473 $in_range=1;
|
devi@62
|
474 next;
|
devi@62
|
475 }
|
devi@62
|
476 if ($Config{"to"} && $cl->{"cline"} =~ /$Config{"signature"}\s*$Config{"to"}/) {
|
devi@62
|
477 $in_range=0;
|
devi@62
|
478 next;
|
devi@62
|
479 }
|
devi@62
|
480 next if ($Config{"from"} && $Config{"to"} && !$in_range)
|
devi@62
|
481 ||
|
devi@62
|
482 ($Config{"skip_empty"} =~ /^y/i && $cl->{"cline"} =~ /^\s*$/ )
|
devi@62
|
483 ||
|
devi@62
|
484 ($Config{"skip_wrong"} =~ /^y/i && $cl->{"err"} != 0)
|
devi@62
|
485 ||
|
devi@62
|
486 ($Config{"skip_interrupted"} =~ /^y/i && $cl->{"err"} == 130);
|
devi@62
|
487
|
devi@62
|
488 # Вырезаем из вывода только нужное количество строк
|
devi@23
|
489
|
devi@62
|
490 my $output="";
|
devi@62
|
491 if ($Config{"head_lines"} || $Config{"tail_lines"}) {
|
devi@62
|
492 # Partialy output
|
devi@62
|
493 my @lines = split '\n', $cl->{"output"};
|
devi@62
|
494 # head
|
devi@62
|
495 my $mark=1;
|
devi@62
|
496 for (my $i=0; $i<= $#lines && $i < $Config{"cache_head_lines"}; $i++) {
|
devi@62
|
497 $output .= $lines[$i]."\n";
|
devi@62
|
498 }
|
devi@62
|
499 # tail
|
devi@62
|
500 my $start=$#lines-$Config{"cache_tail_lines"}+1;
|
devi@62
|
501 if ($start < 0) {
|
devi@62
|
502 $start=0;
|
devi@62
|
503 $mark=0;
|
devi@62
|
504 }
|
devi@62
|
505 if ($start < $Config{"cache_head_lines"}) {
|
devi@62
|
506 $start=$Config{"cache_head_lines"};
|
devi@62
|
507 $mark=0;
|
devi@62
|
508 }
|
devi@62
|
509 $output .= $Config{"skip_text"}."\n" if $mark;
|
devi@62
|
510 for (my $i=$start; $i<= $#lines; $i++) {
|
devi@62
|
511 $output .= $lines[$i]."\n";
|
devi@62
|
512 }
|
devi@62
|
513 }
|
devi@62
|
514 else {
|
devi@62
|
515 # Full output
|
devi@62
|
516 $output .= $cl->{"output"};
|
devi@62
|
517 }
|
devi@23
|
518
|
devi@62
|
519 # Совместимость с labmaker
|
devi@23
|
520
|
devi@62
|
521 # Переводим в секунды Эпохи
|
devi@62
|
522 # В labmaker'е данные хранились в неудобной форме: hour, min, sec, day of year
|
devi@62
|
523 # Информация о годе отсутствовала
|
devi@62
|
524 # Её можно внести:
|
devi@62
|
525 # Декабрь 2004 год; остальные -- 2005 год.
|
devi@23
|
526
|
devi@62
|
527 my $year = 2005;
|
devi@62
|
528 #$year = 2004 if ( $cl->{day} > 330 );
|
devi@62
|
529 $year = $Config{year} if $Config{year};
|
devi@62
|
530 # timelocal( $sec, $min, $hour, $mday,$mon,$year);
|
devi@62
|
531 $cl->{time} = timelocal_nocheck($cl->{sec},$cl->{min},$cl->{hour},$cl->{day},0,$year);
|
devi@23
|
532
|
devi@23
|
533
|
devi@62
|
534 # Начинаем вывод команды
|
devi@62
|
535 print OUT "<command>\n";
|
devi@62
|
536 for my $element (qw(
|
devi@62
|
537 local_session_id
|
devi@62
|
538 time
|
devi@62
|
539 raw_start
|
devi@62
|
540 raw_output_start
|
devi@62
|
541 raw_end
|
devi@62
|
542 raw_file
|
devi@62
|
543 tty
|
devi@62
|
544 uid
|
devi@62
|
545 err
|
devi@62
|
546 last_command
|
devi@62
|
547 )) {
|
devi@65
|
548 next unless defined($cl->{"$element"});
|
devi@62
|
549 print OUT "<$element>".$cl->{$element}."</$element>\n";
|
devi@62
|
550 }
|
devi@62
|
551 for my $element (qw(
|
devi@62
|
552 prompt
|
devi@62
|
553 cline
|
devi@62
|
554 note
|
devi@62
|
555 note_title
|
devi@62
|
556 )) {
|
devi@65
|
557 next unless defined($cl->{"$element"});
|
devi@62
|
558 print OUT "<$element>";
|
devi@62
|
559 printq(\*OUT,$cl->{"$element"});
|
devi@62
|
560 print OUT "</$element>\n";
|
devi@62
|
561 }
|
devi@62
|
562 print OUT "<output>";
|
devi@62
|
563 printq(\*OUT,$output);
|
devi@62
|
564 print OUT "</output>\n";
|
devi@62
|
565 if ($cl->{"diff"}) {
|
devi@62
|
566 print OUT "<diff>";
|
devi@62
|
567 printq(\*OUT,${$Diffs{$cl->{"diff"}}}{"text"});
|
devi@62
|
568 print OUT "</diff>\n";
|
devi@62
|
569 }
|
devi@62
|
570 print OUT "</command>\n";
|
devi@23
|
571
|
devi@62
|
572 }
|
devi@23
|
573
|
devi@62
|
574 close(OUT);
|
devi@27
|
575 }
|
devi@27
|
576
|
devi@27
|
577 sub print_session
|
devi@27
|
578 {
|
devi@62
|
579 my $output_filename = $_[0];
|
devi@62
|
580 my $local_session_id = $_[1];
|
devi@62
|
581 return if not defined($Sessions{$local_session_id});
|
devi@27
|
582
|
devi@62
|
583 open(OUT, ">>", $output_filename)
|
devi@62
|
584 or die "Can't open $output_filename for writing\n";
|
devi@62
|
585 print OUT "<session>\n";
|
devi@62
|
586 my %session = %{$Sessions{$local_session_id}};
|
devi@62
|
587 for my $key (keys %session) {
|
devi@62
|
588 print OUT "<$key>".$session{$key}."</$key>\n"
|
devi@62
|
589 }
|
devi@62
|
590 print OUT "</session>\n";
|
devi@62
|
591 close(OUT);
|
devi@27
|
592 }
|
devi@27
|
593
|
devi@27
|
594 sub send_cache
|
devi@27
|
595 {
|
devi@62
|
596 # Если в кэше что-то накопилось,
|
devi@62
|
597 # попытаемся отправить это на сервер
|
devi@62
|
598 #
|
devi@62
|
599 my $cache_was_sent=0;
|
devi@62
|
600
|
devi@62
|
601 if (open(CACHE, $Config{cache})) {
|
devi@62
|
602 local $/;
|
devi@62
|
603 my $cache = <CACHE>;
|
devi@62
|
604 close(CACHE);
|
devi@27
|
605
|
devi@62
|
606 my $socket = IO::Socket::INET->new(
|
devi@62
|
607 PeerAddr => $Config{backend_address},
|
devi@62
|
608 PeerPort => $Config{backend_port},
|
devi@62
|
609 proto => "tcp",
|
devi@62
|
610 Type => SOCK_STREAM
|
devi@62
|
611 );
|
devi@27
|
612
|
devi@62
|
613 if ($socket) {
|
devi@62
|
614 print $socket $cache;
|
devi@62
|
615 close($socket);
|
devi@62
|
616 $cache_was_sent = 1;
|
devi@62
|
617 }
|
devi@62
|
618 }
|
devi@62
|
619 return $cache_was_sent;
|
devi@23
|
620 }
|
devi@23
|
621
|
devi@25
|
622 sub save_cache_stat
|
devi@25
|
623 {
|
devi@62
|
624 open (CACHE, ">$Config{cache_stat}");
|
devi@62
|
625 for my $f (keys %Script_Files) {
|
devi@62
|
626 print CACHE "$f\t",$Script_Files{$f}->{size},"\t",$Script_Files{$f}->{tell},"\n";
|
devi@62
|
627 }
|
devi@62
|
628 close(CACHE);
|
devi@25
|
629 }
|
devi@25
|
630
|
devi@25
|
631 sub load_cache_stat
|
devi@25
|
632 {
|
devi@62
|
633 if (open (CACHE, "$Config{cache_stat}")) {
|
devi@62
|
634 while(<CACHE>) {
|
devi@62
|
635 chomp;
|
devi@62
|
636 my ($f, $size, $tell) = split /\t/;
|
devi@62
|
637 $Script_Files{$f}->{size} = $size;
|
devi@62
|
638 $Script_Files{$f}->{tell} = $tell;
|
devi@62
|
639 }
|
devi@62
|
640 close(CACHE);
|
devi@62
|
641 };
|
devi@25
|
642 }
|
devi@23
|
643
|
devi@23
|
644
|
devi@23
|
645 main();
|
devi@23
|
646
|
devi@25
|
647 sub process_was_killed
|
devi@25
|
648 {
|
devi@62
|
649 $Killed = 1;
|
devi@25
|
650 }
|
devi@25
|
651
|
devi@23
|
652 sub main
|
devi@23
|
653 {
|
devi@23
|
654
|
devi@62
|
655 $| = 1;
|
devi@23
|
656
|
devi@62
|
657 init_variables();
|
devi@62
|
658 init_config();
|
devi@23
|
659
|
devi@27
|
660
|
devi@62
|
661 if ($Config{"mode"} ne "daemon") {
|
devi@27
|
662
|
devi@27
|
663 =cut
|
devi@62
|
664 В нормальном режиме работы нужно
|
devi@62
|
665 считать скрипты, обработать их и записать
|
devi@62
|
666 результат выполнения в результирующий файл.
|
devi@62
|
667 После этого завершить работу.
|
devi@27
|
668 =cut
|
devi@62
|
669 for my $lab_log (split (/\s+/, $Config{"diffs"} || $Config{"input"})) {
|
devi@62
|
670 load_diff_files($lab_log);
|
devi@62
|
671 }
|
devi@62
|
672 load_command_lines($Config{"input"}, $Config{"input_mask"});
|
devi@67
|
673 sort_command_lines;
|
devi@62
|
674 #process_command_lines;
|
devi@62
|
675 print_command_lines($Config{"cache"});
|
devi@62
|
676 }
|
devi@62
|
677 else {
|
devi@62
|
678 if (open(PIDFILE, $Config{agent_pidfile})) {
|
devi@62
|
679 my $pid = <PIDFILE>;
|
devi@62
|
680 close(PIDFILE);
|
devi@62
|
681 if ($^O eq 'linux' && $pid &&(! -e "/proc/$pid" || !`grep $Config{"l3-agent"} /proc/$pid/cmdline && grep "uid:.*\b$<\b" /proc/$pid/status`)) {
|
devi@62
|
682 print "Removing stale pidfile\n";
|
devi@62
|
683 unlink $Config{agent_pidfile}
|
devi@62
|
684 or die "Can't remove stale pidfile ". $Config{agent_pidfile}. " : $!";
|
devi@62
|
685 }
|
devi@62
|
686 elsif ($^O eq 'freebsd' && $pid && `ps axo uid,pid,command | grep '$<\\s*$pid\\s*$Config{"l3-agent"}' 2> /dev/null`) {
|
devi@62
|
687 print "Removing stale pidfile\n";
|
devi@62
|
688 unlink $Config{agent_pidfile}
|
devi@62
|
689 or die "Can't remove stale pidfile ". $Config{agent_pidfile}. " : $!";
|
devi@62
|
690 }
|
devi@62
|
691 elsif ($^O eq 'linux' || $^O eq 'freebsd' ) {
|
devi@62
|
692 print "l3-agent is already running: pid=$pid; pidfile=$Config{agent_pidfile}\n";
|
devi@62
|
693 exit(0);
|
devi@62
|
694 }
|
devi@62
|
695 else {
|
devi@62
|
696 print "Unknown operating system";
|
devi@62
|
697 exit(0);
|
devi@62
|
698 }
|
devi@62
|
699 }
|
devi@62
|
700 if ($Config{detach} =~ /^y/i) {
|
devi@62
|
701 #$Config{verbose} = "no";
|
devi@62
|
702 my $pid = fork;
|
devi@62
|
703 exit if $pid;
|
devi@62
|
704 die "Couldn't fork: $!" unless defined ($pid);
|
devi@27
|
705
|
devi@62
|
706 open(PIDFILE, ">", $Config{agent_pidfile})
|
devi@62
|
707 or die "Can't open pidfile ". $Config{agent_pidfile}. " for wrting: $!";
|
devi@62
|
708 print PIDFILE $$;
|
devi@62
|
709 close(PIDFILE);
|
devi@27
|
710
|
devi@62
|
711 for my $handle (*STDIN, *STDOUT, *STDERR) {
|
devi@62
|
712 open ($handle, "+<", "/dev/null")
|
devi@62
|
713 or die "can't reopen $handle to /dev/null: $!"
|
devi@62
|
714 }
|
devi@27
|
715
|
devi@62
|
716 POSIX::setsid()
|
devi@62
|
717 or die "Can't start a new session: $!";
|
devi@27
|
718
|
devi@62
|
719 $0 = $Config{"l3-agent"};
|
devi@62
|
720
|
devi@62
|
721 $SIG{INT} = $SIG{TERM} = $SIG{HUP} = \&process_was_killed;
|
devi@62
|
722 }
|
devi@62
|
723 while (not $Killed) {
|
devi@62
|
724 @Command_Lines = ();
|
devi@62
|
725 @Command_Lines_Index = ();
|
devi@62
|
726 for my $lab_log (split (/\s+/, $Config{"diffs"} || $Config{"input"})) {
|
devi@62
|
727 load_diff_files($lab_log);
|
devi@62
|
728 }
|
devi@62
|
729 load_cache_stat();
|
devi@62
|
730 load_command_lines($Config{"input"}, $Config{"input_mask"});
|
devi@62
|
731 if (@Command_Lines) {
|
devi@65
|
732 #sort_command_lines;
|
devi@65
|
733 #process_command_lines;
|
devi@62
|
734 print_command_lines($Config{"cache"});
|
devi@62
|
735 }
|
devi@62
|
736 save_cache_stat();
|
devi@62
|
737 if (-e $Config{cache} && (stat($Config{cache}))[7]) {
|
devi@62
|
738 send_cache() && unlink($Config{cache});
|
devi@62
|
739 }
|
devi@62
|
740 sleep($Config{"daemon_sleep_interval"} || 1);
|
devi@62
|
741 }
|
devi@62
|
742
|
devi@62
|
743 unlink $Config{agent_pidfile};
|
devi@62
|
744 }
|
devi@23
|
745
|
devi@23
|
746 }
|
devi@23
|
747
|
devi@23
|
748 sub init_variables
|
devi@23
|
749 {
|
devi@23
|
750 }
|