main.pm 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900
  1. package eyelib::main;
  2. #
  3. # Copyright (C) Roman Dmitriev, rnd@rajven.ru
  4. #
  5. use utf8;
  6. use open ":encoding(utf8)";
  7. use Encode;
  8. use strict;
  9. use English;
  10. use FindBin '$Bin';
  11. use lib "/opt/Eye/scripts";
  12. use base 'Exporter';
  13. use vars qw(@EXPORT @ISA);
  14. use eyelib::config;
  15. use Socket;
  16. use IO::Select;
  17. use IO::Handle;
  18. use Crypt::CBC;
  19. use MIME::Base64;
  20. our @ISA = qw(Exporter);
  21. our @EXPORT = qw(
  22. isNotifyCreate
  23. isNotifyUpdate
  24. isNotifyDelete
  25. isNotifyNone
  26. hasNotifyFlag
  27. log_file
  28. write_to_file
  29. wrlog
  30. log_session
  31. log_warning
  32. log_info
  33. log_debug
  34. log_error
  35. log_verbose
  36. log_die
  37. in_array
  38. timestamp
  39. do_exec
  40. do_exec_ref
  41. do_exit
  42. hash_to_kv_csv
  43. sendEmail
  44. IsNotRun
  45. IsMyPID
  46. Add_PID
  47. Remove_PID
  48. IsNotLocked
  49. IsMyLock
  50. Add_Lock
  51. Remove_Lock
  52. DefHash
  53. read_file
  54. uniq
  55. strim
  56. trim
  57. hash_to_text
  58. is_integer
  59. is_float
  60. run_in_parallel
  61. translit
  62. crypt_string
  63. decrypt_string
  64. netdev_set_auth
  65. GetNowTime
  66. GetUnixTimeByStr
  67. GetTimeStrByUnixTime
  68. );
  69. BEGIN
  70. {
  71. #---------------------------------------------------------------------------------------------------------
  72. # Проверяет, установлен ли флаг создания
  73. sub isNotifyCreate {
  74. my ($flags) = @_;
  75. return ($flags & NOTIFY_CREATE) == NOTIFY_CREATE;
  76. }
  77. #---------------------------------------------------------------------------------------------------------
  78. # Проверяет, установлен ли флаг изменения
  79. sub isNotifyUpdate {
  80. my ($flags) = @_;
  81. return ($flags & NOTIFY_UPDATE) == NOTIFY_UPDATE;
  82. }
  83. #---------------------------------------------------------------------------------------------------------
  84. # Проверяет, установлен ли флаг удаления
  85. sub isNotifyDelete {
  86. my ($flags) = @_;
  87. return ($flags & NOTIFY_DELETE) == NOTIFY_DELETE;
  88. }
  89. #---------------------------------------------------------------------------------------------------------
  90. # Проверяет, отключены ли все уведомления
  91. sub isNotifyNone {
  92. my ($flags) = @_;
  93. return $flags == NOTIFY_NONE;
  94. }
  95. #---------------------------------------------------------------------------------------------------------
  96. # Проверяет, установлен ли конкретный флаг
  97. sub hasNotifyFlag {
  98. my ($flags, $flagToCheck) = @_;
  99. return ($flags & $flagToCheck) == $flagToCheck;
  100. }
  101. #---------------------------------------------------------------------------------------------------------
  102. sub log_file {
  103. return if (!$_[0]);
  104. return if (!$_[1]);
  105. return if (!$_[2]);
  106. open (LG,">>$_[0]") || die("Error open log file $_[0]!!! die...");
  107. my ($sec,$min,$hour,$mday,$mon,$year) = (localtime())[0,1,2,3,4,5];
  108. $mon += 1; $year += 1900;
  109. my @msg = split("\n",$_[2]);
  110. foreach my $row (@msg) {
  111. next if (!$row);
  112. printf LG "%04d%02d%02d-%02d%02d%02d %s [%d] %s\n",$year,$mon,$mday,$hour,$min,$sec,$_[1],$$,$row;
  113. }
  114. close (LG);
  115. if ($< ==0) {
  116. my $uid = getpwnam $log_owner_user;
  117. my $gid = getgrnam $log_owner_user;
  118. if (!$gid) { $gid=getgrnam "root"; }
  119. if (!$uid) { $uid=getpwnam "root"; }
  120. chown $uid, $gid, $_[0];
  121. chmod oct("0660"), $_[0];
  122. }
  123. }
  124. #---------------------------------------------------------------------------------------------------------
  125. sub write_to_file {
  126. return if (!$_[0]);
  127. return if (!$_[1]);
  128. my $f_name = shift;
  129. my $cmd = shift;
  130. my $append = shift;
  131. if ($append) {
  132. open (LG,">>$f_name") || die("Error open file $f_name!!! die...");
  133. } else {
  134. open (LG,">$f_name") || die("Error open file $f_name!!! die...");
  135. }
  136. binmode(LG,':utf8');
  137. if (ref($cmd) eq 'ARRAY') {
  138. foreach my $row (@$cmd) {
  139. next if (!$row);
  140. print LG $row."\n";
  141. }
  142. } else {
  143. my @msg = split("\n",$cmd);
  144. foreach my $row (@msg) {
  145. next if (!$row);
  146. print LG $row."\n";
  147. }
  148. }
  149. close (LG);
  150. }
  151. #---------------------------------------------------------------------------------------------------------
  152. sub wrlog {
  153. my $level = shift;
  154. my $string = shift;
  155. my $PRN_LEVEL = 'INFO:';
  156. if ($level == $W_INFO) { log_info($string); }
  157. if ($level == $W_ERROR) { $PRN_LEVEL = 'ERROR:'; log_error($string); }
  158. if ($level == $W_DEBUG) { $PRN_LEVEL = 'DEBUG'; log_debug($string); }
  159. my @msg = split("\n",$string);
  160. foreach my $row (@msg) {
  161. next if (!$row);
  162. print $PRN_LEVEL.' '.$row."\n";
  163. }
  164. }
  165. #---------------------------------------------------------------------------------------------------------
  166. sub log_session { log_file($LOG_COMMON,"SESSION:",$_[0]) if ($log_enable); }
  167. #---------------------------------------------------------------------------------------------------------
  168. sub log_info { log_file($LOG_COMMON,"INFO:",$_[0]) if ($log_enable); }
  169. #---------------------------------------------------------------------------------------------------------
  170. sub log_verbose { log_file($LOG_COMMON,"VERBOSE:",$_[0]) if ($log_enable); }
  171. #---------------------------------------------------------------------------------------------------------
  172. sub log_warning { log_file($LOG_COMMON,"WARN:",$_[0]) if ($log_enable); }
  173. #---------------------------------------------------------------------------------------------------------
  174. sub log_debug { log_file($LOG_DEBUG,"DEBUG:",$_[0]) if $debug; }
  175. #---------------------------------------------------------------------------------------------------------
  176. sub log_error { log_file($LOG_ERR,"ERROR:",$_[0]) if ($log_enable); }
  177. #---------------------------------------------------------------------------------------------------------
  178. sub log_die {
  179. wrlog($W_ERROR,$_[0]);
  180. my $worktime = time()-$BASETIME;
  181. log_info("Script work $worktime sec.");
  182. sendEmail("$HOSTNAME - $MY_NAME die! ","Process: $MY_NAME aborted with error:\n$_[0]");
  183. die ($_[0]);
  184. }
  185. #---------------------------------------------------------------------------------------------------------
  186. sub timestamp {
  187. my $worktime = time()-$BASETIME;
  188. #print "TimeStamp: $worktime sec.\n";
  189. log_info("TimeStamp: $worktime sec.");
  190. }
  191. #---------------------------------------------------------------------------------------------------------
  192. sub in_array {
  193. my $arr = shift;
  194. my @tmp = ();
  195. if (ref($arr)=~'ARRAY') { @tmp = @{$arr}; } else { push(@tmp,$arr); }
  196. my $value = shift;
  197. my %num = map { $_, 1 } @tmp;
  198. return $num{$value} || 0;
  199. }
  200. #---------------------------------------------------------------------------------------------------------.
  201. sub hash_to_kv_csv {
  202. my ($hash_ref, $delimiter) = @_;
  203. $delimiter ||= ',';
  204. return '' unless $hash_ref && %$hash_ref;
  205. # Экранируем специальные символы
  206. my $escape = sub {
  207. my $value = shift;
  208. return '' unless defined $value;
  209. # Если значение содержит кавычки или запятые - заключаем в кавычки
  210. if ($value =~ /["$delimiter]/) {
  211. $value =~ s/"/""/g;
  212. return '"' . $value . '"';
  213. }
  214. return $value;
  215. };
  216. # Формируем пары ключ=>значение
  217. my @pairs;
  218. while (my ($key, $value) = each %$hash_ref) {
  219. push @pairs, $escape->($key) . '=>' . $escape->($value);
  220. }
  221. return join($delimiter, sort @pairs);
  222. }
  223. #---------------------------------------------------------------------------------------------------------.
  224. sub do_exec_ref {
  225. my $ret = `$_[0] 2>&1`;
  226. my $res = $?;
  227. my %result;
  228. chomp($ret);
  229. $result{output}=$ret;
  230. $result{status}=$res;
  231. log_debug("Run: $_[0] Output:\n$ret\nResult code: $res");
  232. if ($res eq "0") { log_info("Run: $_[0] - $ret"); } else { log_error("Run: $_[0] - $ret"); }
  233. return %result;
  234. }
  235. #---------------------------------------------------------------------------------------------------------
  236. sub do_exec {
  237. my $ret = `$_[0]`;
  238. my $res = $?;
  239. log_debug("Run: $_[0] Output:\n$ret\nResult code: $res");
  240. if ($res eq "0") {
  241. log_info("Run: $_[0] - $ret");
  242. } else {
  243. $ret = "Error";
  244. log_error("Run: $_[0] - $ret");
  245. }
  246. return $ret;
  247. }
  248. #---------------------------------------------------------------------------------------------------------
  249. sub do_exit {
  250. my $worktime = time()-$BASETIME;
  251. my $code;
  252. if ($_[0]) { $code = $_[0]; } else { $code = 0; }
  253. log_info("Script work $worktime sec. Exit code: $code");
  254. exit $code;
  255. }
  256. #---------------------------------------------------------------------------------------------------------
  257. sub encode_mime_header {
  258. my ($str) = @_;
  259. return $str if $str =~ /^[[:ascii:]]*$/;
  260. my $b64 = encode_base64($str, '');
  261. $b64 =~ s/\s+$//;
  262. return "=?UTF-8?B?$b64?=";
  263. }
  264. #---------------------------------------------------------------------------------------------------------
  265. sub sendEmail {
  266. my ($subject, $msg, $use_br) = @_;
  267. return unless defined $msg && length $msg;
  268. return unless $send_email;
  269. unless (defined $sender_email && defined $admin_email) {
  270. log_error("Email addresses not defined");
  271. return;
  272. }
  273. # Санитизация (оставляет Unicode буквы/цифры)
  274. $subject =~ s/[^\p{L}\p{N}\s\-\.\,\!\?]//g;
  275. $msg =~ s/\r//g;
  276. my $html_message = <<"END_HTML";
  277. <!DOCTYPE html>
  278. <html xmlns="http://www.w3.org/1999/xhtml">
  279. <head>
  280. <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
  281. <title>$subject</title>
  282. </head>
  283. <body>
  284. <div>
  285. END_HTML
  286. my @lines = split(/\n/, $msg);
  287. foreach my $line (@lines) {
  288. $line = htmlspecialchars($line);
  289. $html_message .= $use_br ? "$line<br>\n" : "$line\n";
  290. }
  291. $html_message .= "</div>\n</body>\n</html>\n";
  292. # Кодируем Unicode-строку в байты UTF-8, затем в base64
  293. my $html_utf8_bytes = encode('UTF-8', $html_message);
  294. my $encoded_html = encode_base64($html_utf8_bytes);
  295. my $boundary = '----=' . time() . int(rand(1000));
  296. my $encoded_subject = encode_mime_header($subject);
  297. my $headers = <<"END_HEADERS";
  298. From: $sender_email
  299. To: $admin_email
  300. Subject: $encoded_subject
  301. MIME-Version: 1.0
  302. Content-Type: multipart/mixed; boundary="$boundary"
  303. END_HEADERS
  304. my $mime_message = <<"END_MIME";
  305. --$boundary
  306. Content-Type: text/html; charset=utf-8
  307. Content-Transfer-Encoding: base64
  308. $encoded_html
  309. --$boundary--
  310. END_MIME
  311. my $sendmail = '/usr/sbin/sendmail';
  312. unless (-x $sendmail) {
  313. log_error("Sendmail not found or not executable at $sendmail");
  314. return;
  315. }
  316. unless (open(MAIL, "|$sendmail -oi -t")) {
  317. log_error("Failed to open sendmail: $!");
  318. return;
  319. }
  320. print MAIL $headers;
  321. print MAIL $mime_message;
  322. close(MAIL) or log_error("Failed to send email: $!");
  323. log_info("Sent email from $sender_email to $admin_email with subject: $subject");
  324. log_debug("Email body:\n$msg");
  325. }
  326. #---------------------------------------------------------------------------------------------------------
  327. sub hash_to_text {
  328. my ($hash_ref, $indent, $seen) = @_;
  329. $indent ||= 0;
  330. $seen ||= {};
  331. return 'undef' unless defined $hash_ref;
  332. if (ref $hash_ref eq 'HASH') {
  333. # Защита от циклических ссылок
  334. my $addr = refaddr($hash_ref);
  335. if ($seen->{$addr}) {
  336. return '';
  337. }
  338. $seen->{$addr} = 1;
  339. my $spaces = ' ' x $indent;
  340. my @lines;
  341. for my $key (sort keys %$hash_ref) {
  342. my $value = $hash_ref->{$key};
  343. my $formatted_key = $key =~ /^[a-zA-Z_]\w*$/ ? $key : "'$key'";
  344. my $formatted_value;
  345. if (ref $value eq 'HASH') {
  346. $formatted_value = ":\n" . hash_to_text($value, $indent + 1, $seen) . "\n$spaces";
  347. }
  348. elsif (ref $value eq 'ARRAY') {
  349. $formatted_value = array_to_text($value, $indent + 1, $seen);
  350. }
  351. elsif (ref $value) {
  352. $formatted_value = '[' . ref($value) . ']';
  353. }
  354. elsif (!defined $value) {
  355. $formatted_value = '';
  356. }
  357. else {
  358. $formatted_value = "'$value'";
  359. }
  360. push @lines, "$spaces $formatted_key => $formatted_value" if ($formatted_value);
  361. }
  362. return join(",\n", @lines) || "$spaces # empty";
  363. }
  364. else {
  365. return "'$hash_ref'";
  366. }
  367. }
  368. #---------------------------------------------------------------------------------------------------------
  369. sub array_to_text {
  370. my ($array_ref, $indent, $seen) = @_;
  371. $indent ||= 0;
  372. $seen ||= {};
  373. return '[]' unless @$array_ref;
  374. my $spaces = ' ' x $indent;
  375. my @lines;
  376. foreach my $item (@$array_ref) {
  377. my $formatted_item;
  378. if (ref $item eq 'HASH') {
  379. $formatted_item = ":\n" . hash_to_text($item, $indent + 1, $seen) . "\n$spaces";
  380. }
  381. elsif (ref $item eq 'ARRAY') {
  382. $formatted_item = array_to_text($item, $indent + 1, $seen);
  383. }
  384. elsif (ref $item) {
  385. $formatted_item = '[' . ref($item) . ']';
  386. }
  387. elsif (!defined $item) {
  388. $formatted_item = '';
  389. }
  390. else {
  391. $formatted_item = "'$item'";
  392. }
  393. push @lines, "$spaces $formatted_item" if ($formatted_item);
  394. }
  395. return "[\n" . join(",\n", @lines) . "\n$spaces]";
  396. }
  397. #---------------------------------------------------------------------------------------------------------
  398. # Вспомогательная функция для получения адреса ссылки
  399. sub refaddr {
  400. my $ref = shift;
  401. return "$ref" =~ /\(0x([0-9a-f]+)\)$/ ? "0x$1" : "$ref";
  402. }
  403. #---------------------------------------------------------------------------------------------------------
  404. ### Check few run script
  405. sub IsNotRun {
  406. my $pname = shift;
  407. my $lockfile = $pname.".pid";
  408. # if pid file not exists - OK
  409. log_debug("Check what pid file $lockfile exists.");
  410. if (! -e $lockfile) { log_debug("pid file not found. Continue."); return 1; }
  411. open (FF,"<$lockfile") or log_die("can't open file $lockfile: $!");
  412. my $lockid = <FF>;
  413. close(FF);
  414. chomp($lockid);
  415. # If the process ID belongs to the current program - OK
  416. if ($lockid eq $$) { log_debug("pid file found, but owner is this process. Continue. "); return 1; }
  417. # if owner of this process ID not exists - OK
  418. my $process_count = `ps -p $lockid | grep \'$lockid\' | wc -l`;
  419. chomp($process_count);
  420. log_debug("Process count with id $lockid is $process_count");
  421. if ($process_count==0) { log_debug("pid file found, but owner process not found. Remove lock file and continue. "); unlink $lockfile; return 1; }
  422. log_debug("Another proceess with name $MY_NAME pid: $lockid already worked. ");
  423. return 0;
  424. }
  425. #---------------------------------------------------------------------------------------------------------
  426. sub IsMyPID {
  427. my $pname = shift;
  428. my $lockfile = $pname.".pid";
  429. log_debug("Check what pid file $lockfile exists.");
  430. if (! -e $lockfile) { log_debug("pid file not found. Continue."); return 1; }
  431. open (FF,"<$lockfile") or log_die "can't open file $lockfile: $!";
  432. my $lockid = <FF>;
  433. close(FF);
  434. chomp($lockid);
  435. if ($lockid eq $$) { log_debug("pid file is my. continue."); return 1; }
  436. log_debug("Another proceess with name $MY_NAME pid: $lockid already worked. ");
  437. return 0;
  438. }
  439. #---------------------------------------------------------------------------------------------------------
  440. sub Add_PID {
  441. my $pname = shift;
  442. my $lockfile = $pname.".pid";
  443. log_debug("Try create lock file $lockfile");
  444. open (FF,">$lockfile") or log_die "can't open file $lockfile: $!";
  445. flock(FF,2) or log_die "can't flock $lockfile: $!";
  446. print FF $$;
  447. close(FF);
  448. log_debug("Ok.");
  449. return 1;
  450. }
  451. #---------------------------------------------------------------------------------------------------------
  452. sub Remove_PID {
  453. my $pname = shift;
  454. my $lockfile = $pname.".pid";
  455. log_debug("Check what pid file $lockfile exists.");
  456. if (! -e $lockfile) { log_debug("pid file not exists. Continue."); return 1; }
  457. unlink $lockfile or return 0;
  458. log_debug("pid file $lockfile removed.");
  459. return 1;
  460. }
  461. #---------------------------------------------------------------------------------------------------------
  462. sub IsNotLocked {
  463. my $lockfile = $_[0] . ".lock";
  464. log_debug("Check what lock file $lockfile exists.");
  465. if (! -e $lockfile) { log_debug("lock file not found. Continue."); return 1; }
  466. open (FF,"<$lockfile") or log_die "can't open file $lockfile: $!";
  467. my $lockid = <FF>;
  468. close(FF);
  469. chomp($lockid);
  470. if ($lockid eq $$) { log_debug("lock file found, but it is owner is this process. Continue. "); return 1; }
  471. my $process_count = `ps -p $lockid | grep \'$lockid\' | wc -l`;
  472. if ($process_count lt 1) { log_debug("lock file found, but owner process not found. Remove lock file and continue. "); unlink $lockfile; return 1; }
  473. log_debug("Another proceess with pid: $lockid already use $_[0]");
  474. return 0;
  475. }
  476. #---------------------------------------------------------------------------------------------------------
  477. sub IsMyLock {
  478. my $lockfile = $_[0] . ".lock";
  479. log_debug("Check what lock file $lockfile exists.");
  480. if (! -e $lockfile) { log_debug("lock file not found. Continue."); return 0; }
  481. open (FF,"<$lockfile") or log_die "can't open file $lockfile: $!";
  482. my $lockid = <FF>;
  483. close(FF);
  484. chomp($lockid);
  485. if ($lockid eq $$) { log_debug("lock file found, but it is owner is this process. Continue. "); return 1; }
  486. log_debug("file $_[0] used by process with pid: $lockid");
  487. return 0;
  488. }
  489. #---------------------------------------------------------------------------------------------------------
  490. sub Add_Lock {
  491. if (!IsNotLocked($_[0])) { return 0; }
  492. my $lockfile = $_[0] . ".lock";
  493. open (FF,">$lockfile") or log_die "can't open file $lockfile: $!";
  494. flock(FF,2) or log_die "can't flock $lockfile: $!";
  495. print FF $$;
  496. close(FF);
  497. log_debug("Create lock file for $_[0]");
  498. return 1;
  499. }
  500. #---------------------------------------------------------------------------------------------------------
  501. sub Remove_Lock {
  502. if (!IsNotLocked($_[0])) { return 0; }
  503. my $lockfile = $_[0] . ".lock";
  504. if (! -e $lockfile) { return 1; }
  505. unlink $lockfile or return 0;
  506. log_debug("Lock file for $_[0] removed");
  507. return 1;
  508. }
  509. #---------------------------------------------------------------------------------------------------------
  510. sub DefHash {
  511. my $hash=$_[0];
  512. my $num_list = $_[1];
  513. my %num_keys;
  514. if ($num_list) {
  515. my @ret_num = split(' ',$num_list);
  516. %num_keys = map { $_, 1 } @ret_num;
  517. }
  518. foreach my $key (keys %$hash) {
  519. my $null_value = "";
  520. $null_value = 0 if (defined $num_keys{$key});
  521. $hash->{$key}=$null_value if (!defined($hash->{$key}));
  522. }
  523. return $hash;
  524. }
  525. #---------------------------------------------------------------------------------------------------------
  526. sub read_file {
  527. my $filename = shift;
  528. return if (!$filename);
  529. return if (!-e $filename);
  530. open (FF,"<$filename") or die "unable to open file $filename!" ;
  531. my @tmp=<FF>;
  532. close(FF);
  533. chomp(@tmp);
  534. return @tmp;
  535. }
  536. #---------------------------------------------------------------------------------------------------------
  537. sub uniq (\@) {
  538. my @tmp = @{(shift)};
  539. if (scalar(@tmp) eq 0) { return @tmp; }
  540. chomp(@tmp);
  541. my %newlist = map { $_, 1 } @tmp;
  542. return keys %newlist;
  543. }
  544. #---------------------------------------------------------------------------------------------------------
  545. sub strim {
  546. my $str=shift;
  547. return if (!$str);
  548. #$str =~ s/.*[^[:print:]]+//g;
  549. #$str =~ s/[^[:print:]]+//g;
  550. #$str =~ s/[^(a-z|A-Z|0-9|\:|\-|\s|\.)]//g;
  551. #$str =~ s/[:^print:]//g;
  552. $str =~ s/[^[:ascii:]]//g;
  553. $str =~ s/^\s+//g;
  554. $str =~ s/\s+$//g;
  555. return $str;
  556. }
  557. #---------------------------------------------------------------------------------------------------------
  558. sub trim {
  559. my $str=shift;
  560. return if (!$str);
  561. $str =~ s/\n/ /g;
  562. $str =~ s/^\s+//g;
  563. $str =~ s/\s+$//g;
  564. return $str;
  565. }
  566. #---------------------------------------------------------------------------------------------------------
  567. sub is_integer {
  568. defined $_[0] && $_[0] =~ /^[+-]?\d+$/;
  569. }
  570. #---------------------------------------------------------------------------------------------------------
  571. sub is_float {
  572. defined $_[0] && $_[0] =~ /^[+-]?\d+(\.\d+)?$/;
  573. }
  574. #---------------------------------------------------------------------------------------------------------
  575. sub run_in_parallel(\@) {
  576. my @commands = @{(shift)};
  577. my @result = ();
  578. return @result if (!@commands or !scalar(@commands));
  579. my $count = scalar(@commands);
  580. my $start = 0;
  581. while ($start<=$count-1) {
  582. my @run_list=();
  583. my $select = IO::Select->new();
  584. my $stop = $start + $parallel_process_count;
  585. $stop=$count-1 if ($stop >=$count);
  586. for (my $index = $start; $index <=$stop; $index++) {
  587. next if (!$commands[$index]);
  588. my $cmd=$commands[$index];
  589. log_info("Starting ".$cmd);
  590. my ($hchild, $hparent, $childid);
  591. socketpair($hchild, $hparent, AF_UNIX, SOCK_STREAM, PF_UNSPEC) or die "socketpair: $!";
  592. $childid = fork;
  593. die "cannot fork" if($childid == -1);
  594. # redirect child Input|Output
  595. unless($childid) {
  596. open STDIN, "<&", $hparent;
  597. open STDOUT, ">&", $hparent;
  598. open STDERR, ">&", $hparent;
  599. close $hparent;
  600. close $hchild;
  601. $select->remove($_) and close $_ for($select->handles);
  602. exec "/bin/nice -n 15 ".$cmd;
  603. }
  604. close $hparent;
  605. $select->add($hchild);
  606. }
  607. while (my @ready = $select->can_read) {
  608. next if (!@ready or !scalar(@ready));
  609. for my $read(@ready) {
  610. if($read->eof || $read->error) {
  611. # child exit
  612. $select->remove($read);
  613. close $read;
  614. next;
  615. }
  616. if(defined(my $str = <$read>)) {
  617. log_info("Read:".$str);
  618. push(@result,$str);
  619. }
  620. }
  621. }
  622. $start = $stop+1;
  623. }
  624. return (@result);
  625. }
  626. #---------------------------------------------------------------------------------
  627. sub translit {
  628. my $textline=shift;
  629. return if (!$textline);
  630. $textline =~ s/А/A/g; $textline =~ s/а/a/g;
  631. $textline =~ s/Б/B/g; $textline =~ s/б/b/g;
  632. $textline =~ s/В/V/g; $textline =~ s/в/v/g;
  633. $textline =~ s/Г/G/g; $textline =~ s/г/g/g;
  634. $textline =~ s/Д/D/g; $textline =~ s/д/d/g;
  635. $textline =~ s/Е/E/g; $textline =~ s/е/e/g;
  636. $textline =~ s/Ё/E/g; $textline =~ s/ё/e/g;
  637. $textline =~ s/Ж/Zh/g; $textline =~ s/ж/zh/g;
  638. $textline =~ s/З/Z/g; $textline =~ s/з/z/g;
  639. $textline =~ s/И/I/g; $textline =~ s/и/i/g;
  640. $textline =~ s/Й/I/g; $textline =~ s/й/i/g;
  641. $textline =~ s/К/K/g; $textline =~ s/к/k/g;
  642. $textline =~ s/Л/L/g; $textline =~ s/л/l/g;
  643. $textline =~ s/М/M/g; $textline =~ s/м/m/g;
  644. $textline =~ s/Н/N/g; $textline =~ s/н/n/g;
  645. $textline =~ s/О/O/g; $textline =~ s/о/o/g;
  646. $textline =~ s/П/P/g; $textline =~ s/п/p/g;
  647. $textline =~ s/Р/R/g; $textline =~ s/р/r/g;
  648. $textline =~ s/ТС/T-S/g; $textline =~ s/Тс/T-s/g; $textline =~ s/тс/t-s/g;
  649. $textline =~ s/С/S/g; $textline =~ s/с/s/g;
  650. $textline =~ s/Т/T/g; $textline =~ s/т/t/g;
  651. $textline =~ s/У/U/g; $textline =~ s/у/u/g;
  652. $textline =~ s/Ф/F/g; $textline =~ s/ф/f/g;
  653. $textline =~ s/Х/Kh/g; $textline =~ s/х/kh/g;
  654. $textline =~ s/Ц/Ts/g; $textline =~ s/ц/ts/g;
  655. $textline =~ s/Ч/Ch/g; $textline =~ s/ч/ch/g;
  656. $textline =~ s/Ш/Sh/g; $textline =~ s/ш/sh/g;
  657. $textline =~ s/Щ/Shch/g; $textline =~ s/щ/shch/g;
  658. #$textline =~ s/Ь/'/g; $textline =~ s/ь/'/g;
  659. #$textline =~ s/Ъ/''/g; $textline =~ s/ъ/''/g;
  660. $textline =~ s/Ь//g; $textline =~ s/ь//g;
  661. $textline =~ s/Ъ//g; $textline =~ s/ъ//g;
  662. $textline =~ s/Ы/Y/g; $textline =~ s/ы/y/g;
  663. $textline =~ s/Э/E/g; $textline =~ s/э/e/g;
  664. $textline =~ s/Ю/Yu/g; $textline =~ s/ю/yu/g;
  665. $textline =~ s/Я/Ya/g; $textline =~ s/я/ya/g;
  666. return $textline;
  667. }
  668. #---------------------------------------------------------------------------------
  669. sub netdev_set_auth {
  670. my $device = shift;
  671. $device->{login}=$config_ref{router_login} if (!$device->{login});
  672. $device->{password}=$config_ref{router_password} if (!$device->{password});
  673. $device->{password}=decrypt_string($device->{password});
  674. $device->{enable_password}='';
  675. #$device->{enable_password}=$device->{passowrd};
  676. $device->{proto} = 'ssh';
  677. $device->{proto} = 'telnet' if ($device->{protocol} eq '1');
  678. #patch for ssh
  679. if ($device->{proto} eq 'ssh' and exists $switch_auth{$device->{vendor_id}}{proto}) {
  680. #set specified ssh type
  681. if ($switch_auth{$device->{vendor_id}}{proto} =~/ssh/i) {
  682. $device->{proto} = $switch_auth{$device->{vendor_id}}{proto};
  683. }
  684. }
  685. $device->{port} = $device->{control_port} if ($device->{control_port});
  686. $device->{prompt} = qr/[\$#>]\s?$/;
  687. if (exists $switch_auth{$device->{vendor_id}}) {
  688. $device->{prompt} = $switch_auth{$device->{vendor_id}}{prompt} if ($switch_auth{$device->{vendor_id}}{prompt});
  689. }
  690. return $device;
  691. }
  692. #---------------------------------------------------------------------------------
  693. sub decrypt_string {
  694. my $crypted_string = shift;
  695. return if (!$crypted_string);
  696. my $cipher_handle = Crypt::CBC->new(
  697. {
  698. 'key' => $config_ref{encryption_key},
  699. 'cipher' => 'Cipher::AES',
  700. 'iv' => $config_ref{encryption_iv},
  701. 'literal_key' => 1,
  702. 'header' => 'none',
  703. keysize => 128 / 8
  704. }
  705. );
  706. my $result = $cipher_handle->decrypt(decode_base64($crypted_string));
  707. return $result;
  708. }
  709. #---------------------------------------------------------------------------------
  710. sub crypt_string {
  711. my $simple_string = shift;
  712. return if (!$simple_string);
  713. my $cipher_handle = Crypt::CBC->new(
  714. {
  715. 'key' => $config_ref{encryption_key},
  716. 'cipher' => 'Cipher::AES',
  717. 'iv' => $config_ref{encryption_iv},
  718. 'literal_key' => 1,
  719. 'header' => 'none',
  720. keysize => 128 / 8
  721. }
  722. );
  723. my $result = encode_base64($cipher_handle->encrypt($simple_string));
  724. return $result;
  725. }
  726. #---------------------------------------------------------------------------------------------------------------
  727. sub GetNowTime {
  728. my ($sec,$min,$hour,$day,$month,$year,$zone) = localtime(time());
  729. $month += 1;
  730. $year += 1900;
  731. my $now_str=sprintf "%04d-%02d-%02d %02d:%02d:%02d",$year,$month,$day,$hour,$min,$sec;
  732. return $now_str;
  733. }
  734. #---------------------------------------------------------------------------------------------------------------
  735. sub GetUnixTimeByStr {
  736. my $time_str = shift;
  737. $time_str =~s/\//-/g;
  738. $time_str = trim($time_str);
  739. my ($sec,$min,$hour,$day,$mon,$year) = (localtime())[0,1,2,3,4,5];
  740. $year+=1900;
  741. $mon++;
  742. if ($time_str =~/^([0-9]{2,4})\-([0-9]{1,2})-([0-9]{1,2})\s+/) {
  743. $year = $1; $mon = $2; $day = $3;
  744. }
  745. if ($time_str =~/([0-9]{1,2})\:([0-9]{1,2})\:([0-9]{1,2})$/) {
  746. $hour = $1; $min = $2; $sec = $3;
  747. }
  748. my $result = mktime($sec,$min,$hour,$day,$mon-1,$year-1900);
  749. return $result;
  750. }
  751. #---------------------------------------------------------------------------------------------------------------
  752. sub GetTimeStrByUnixTime {
  753. my $time = shift || time();
  754. my ($sec, $min, $hour, $mday, $mon, $year) = (localtime($time))[0,1,2,3,4,5];
  755. my $result = strftime("%Y-%m-%d %H:%M:%S",$sec, $min, $hour, $mday, $mon, $year);
  756. return $result;
  757. }
  758. #---------------------------------------------------------------------------------------------------------
  759. # Helper function for HTML escaping
  760. sub htmlspecialchars {
  761. my ($text) = @_;
  762. return '' unless defined $text;
  763. $text =~ s/&/&amp;/g;
  764. $text =~ s/</&lt;/g;
  765. $text =~ s/>/&gt;/g;
  766. $text =~ s/"/&quot;/g;
  767. $text =~ s/'/&#039;/g;
  768. return $text;
  769. }
  770. #---------------------------------------------------------------------------------
  771. #log_file($LOG_COMMON,"INFO:","----------------------------------------------------------------------------------------");
  772. #log_file($LOG_COMMON,"INFO:","Run script $0. Pid: $$ Pid file: $SPID.pid");
  773. #log_file($LOG_COMMON,"INFO:","User uid: $< Effective uid: $>");
  774. #log_file($LOG_COMMON,"INFO:","Status:");
  775. #log_file($LOG_COMMON,"INFO:","Logging enabled: $log_enable");
  776. #log_file($LOG_COMMON,"INFO:","Logging debug: $debug");
  777. 1;
  778. }