import_auth.pl 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. #!/usr/bin/perl
  2. #
  3. # Copyright (C) Roman Dmitiriev, rnd@rajven.ru
  4. #
  5. use utf8;
  6. use open ":encoding(utf8)";
  7. use open ':std', ':encoding(UTF-8)';
  8. use Encode;
  9. no warnings 'utf8';
  10. use English;
  11. use FindBin '$Bin';
  12. use lib "/opt/Eye/scripts";
  13. use Data::Dumper;
  14. use eyelib::config;
  15. use eyelib::main;
  16. use eyelib::database;
  17. use eyelib::net_utils;
  18. use Net::Patricia;
  19. use strict;
  20. use warnings;
  21. sub process_file {
  22. my $fh = shift;
  23. my @data;
  24. my @headers;
  25. my $line_num = 0;
  26. while (my $line = <$fh>) {
  27. chomp $line;
  28. $line_num++;
  29. # Пропускаем пустые строки
  30. next if $line =~ /^\s*$/;
  31. # Разбиваем строку по разделителям: пробелы, табы, запятые, точки с запятой
  32. my @fields = split(/[\t,;]+/, $line);
  33. if ($line_num == 1) {
  34. # Первая непустая строка — заголовки
  35. @headers = @fields;
  36. print "Заголовки: " . join(" | ", map { "'$_'" } @headers) . "\n\n";
  37. next;
  38. }
  39. # Проверяем, совпадает ли количество полей с количеством заголовков
  40. if (@fields != @headers) {
  41. warn "Предупреждение: строка $line_num: количество полей (" . scalar(@fields) .
  42. ") не совпадает с количеством заголовков (" . scalar(@headers) . ")\n";
  43. # Дополняем или обрезаем массив, чтобы избежать ошибок
  44. @fields = @fields[0 .. $#headers] if @fields > @headers;
  45. while (@fields < @headers) {
  46. push @fields, '';
  47. }
  48. }
  49. # Создаём ассоциативный массив (hash) для текущей строки
  50. my %row;
  51. for my $i (0 .. $#headers) {
  52. $row{$headers[$i]} = $fields[$i];
  53. }
  54. # Добавляем в общий результат
  55. push @data, \%row;
  56. }
  57. return @data; # возвращаем список ссылок на хеши
  58. }
  59. my @rows=();
  60. # === Основная логика ===
  61. if (@ARGV) {
  62. foreach my $filename (@ARGV) {
  63. open(my $fh, '<', $filename) or die "Не могу открыть файл '$filename': $!";
  64. print "Обработка файла: $filename\n";
  65. @rows = process_file($fh);
  66. close($fh);
  67. }
  68. } else {
  69. @rows = process_file(\*STDIN);
  70. }
  71. foreach my $record (@rows) {
  72. next if (!exists($record->{ip}));
  73. my $auth_network = $office_networks->match_string($record->{ip});
  74. if (!$auth_network) {
  75. log_error("Unknown network in request! IP: $record->{ip}");
  76. next;
  77. }
  78. my $search_sql = 'SELECT * FROM User_auth WHERE ip="'.$record->{ip}.'" and deleted=0 ORDER BY last_found DESC';
  79. $record->{ip_int}=StrToIp($record->{ip});
  80. if (!exists($record->{'mac'})) {
  81. delete $record->{'mac'};
  82. } else {
  83. $record->{mac}=mac_splitted(isc_mac_simplify($record->{mac}));
  84. $search_sql = 'SELECT * FROM User_auth WHERE ip="'.$record->{ip}.'" and mac="'.$record->{mac}.'" and deleted=0 ORDER BY last_found DESC';
  85. }
  86. print "Импортируем:\n";
  87. for my $key (keys %{$record}) {
  88. print "\t\t$key => $record->{$key}\n";
  89. }
  90. print "\n";
  91. if (exists $record->{dns_name}) {
  92. my $auth_dns_name = lc(trim($record->{dns_name}));
  93. $auth_dns_name=~s/\./-/g;
  94. $auth_dns_name=~s/\//-/g;
  95. $record->{dns_name} = $auth_dns_name;
  96. }
  97. #search actual record
  98. my $auth_record = get_record_sql($dbh,$search_sql);
  99. if ($auth_record) {
  100. update_record($dbh,'User_auth',$record,"id=".$auth_record->{id});
  101. print "URL: <a href='".$config_ref{stat_url}."/admin/users/edituser.php?id=".$auth_record->{user_id}."'>".$auth_record->{user_id}."</a><br>\n";
  102. if (exists $record->{dns_name}) {
  103. my $user_info;
  104. $user_info->{login}=$record->{dns_name};
  105. update_record($dbh,'User_list',$user_info,"id=".$auth_record->{user_id});
  106. my $device;
  107. $device->{device_name}=$record->{dns_name};
  108. update_record($dbh,'devices',$device,"user_id=".$auth_record->{user_id});
  109. }
  110. if (exists $record->{comments}) {
  111. my $user_info;
  112. $user_info->{fio}=$record->{comments};
  113. update_record($dbh,'User_list',$user_info,"id=".$auth_record->{user_id});
  114. }
  115. next;
  116. }
  117. my $dhcp_record = {%{$record || {}}};
  118. $dhcp_record->{'type'}='add';
  119. my $res_id = resurrection_auth($dbh,$dhcp_record);
  120. if (!$res_id) {
  121. db_log_error($dbh,"Error creating an ip address record for:\t\t".Dumper($dhcp_record));
  122. next;
  123. }
  124. update_record($dbh,'User_auth',$record,"id=".$res_id);
  125. $auth_record = get_record_sql($dbh,'SELECT * FROM User_auth where id='.$res_id);
  126. if ($auth_record) {
  127. print "URL: <a href='".$config_ref{stat_url}."/admin/users/edituser.php?id=".$auth_record->{user_id}."'>".$auth_record->{user_id}."</a><br>\n";
  128. if (exists $record->{dns_name}) {
  129. my $user_info;
  130. $user_info->{login}=$record->{dns_name};
  131. update_record($dbh,'User_list',$user_info,"id=".$auth_record->{user_id});
  132. my $device;
  133. $device->{device_name}=$record->{dns_name};
  134. update_record($dbh,'devices',$device,"user_id=".$auth_record->{user_id});
  135. }
  136. if (exists $record->{comments}) {
  137. my $user_info;
  138. $user_info->{fio}=$record->{comments};
  139. update_record($dbh,'User_list',$user_info,"id=".$auth_record->{user_id});
  140. }
  141. }
  142. }
  143. exit;