main.pm 27 KB

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