import_auth.pl 5.1 KB

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