scan_ipcam.pl 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. #!/usr/bin/perl
  2. use utf8;
  3. use warnings;
  4. use Encode;
  5. use open qw(:std :encoding(UTF-8));
  6. no warnings 'utf8';
  7. use FindBin '$Bin';
  8. use lib "/opt/Eye/scripts";
  9. use DBI;
  10. use File::Basename;
  11. use File::Find;
  12. use File::stat qw(:FIELDS);
  13. use File::Spec::Functions;
  14. use Sys::Hostname;
  15. use DirHandle;
  16. use Time::localtime;
  17. use Fcntl;
  18. use Tie::File;
  19. use Data::Dumper;
  20. use Net::Ping;
  21. use Net::SNMP qw(ticks_to_time TRANSLATE_NONE);
  22. use eyelib::config;
  23. use eyelib::main;
  24. use eyelib::database;
  25. use eyelib::common;
  26. use Fcntl qw(:flock);
  27. open(SELF,"<",$0) or die "Cannot open $0 - $!";
  28. flock(SELF, LOCK_EX|LOCK_NB) or exit 1;
  29. my %hik_snmp_oids=(
  30. '.1.3.6.1.4.1.39165.1.1.0'=>'Model',
  31. '.1.3.6.1.4.1.39165.1.3.0'=>'Firmware',
  32. '.1.3.6.1.4.1.39165.1.5.0'=>'Number',
  33. '.1.3.6.1.4.1.39165.1.6.0'=>'Vendor',
  34. );
  35. my @hik_oids=();
  36. foreach my $oid (keys %hik_snmp_oids) {
  37. push (@hik_oids,$oid);
  38. }
  39. sub scan_ipcam {
  40. my $ip = shift;
  41. my $community = shift;
  42. my $version = shift;
  43. my $result;
  44. #print "$ip $community $version\n";
  45. eval {
  46. my ($session, $error) = Net::SNMP->session(
  47. -hostname => $ip,
  48. -community => $community,
  49. -port => 161,
  50. -version => $version
  51. );
  52. $session->translate(TRANSLATE_NONE);
  53. my $ret = $session->get_request( -varbindlist => [@hik_oids] );
  54. $result->{text} = 'ip: '.$ip;
  55. foreach my $oid (keys %hik_snmp_oids) {
  56. $result->{text} = $result->{text}." ".$hik_snmp_oids{$oid}.": ".$ret->{$oid};
  57. if ($hik_snmp_oids{$oid}=~/Model/i) { $result->{model_name}=$ret->{$oid}; }
  58. if ($hik_snmp_oids{$oid}=~/Firmware/i) { $result->{firmware}=$ret->{$oid}; }
  59. }
  60. $result->{text} = trim($result->{text});
  61. };
  62. return $result;
  63. }
  64. my @auth_list=get_records_sql($dbh,'SELECT * FROM user_auth WHERE deleted=0 and nagios=1');
  65. ##################################### User auth analyze ################################################
  66. if (scalar(@auth_list)>0) {
  67. foreach my $auth (@auth_list) {
  68. next if (!$auth);
  69. my $ip = $auth->{'ip'};
  70. $ip =~s/\/\d+$//g;
  71. #get user
  72. my $login = get_record_sql($dbh,"SELECT * FROM user_list WHERE id=".$auth->{'user_id'});
  73. next if ($login->{ou_id} ne 5);
  74. my $cam_dev = get_record_sql($dbh,"SELECT * FROM devices WHERE user_id=".$auth->{'user_id'});
  75. my $d_community=undef;
  76. my $d_version=undef;
  77. if ($cam_dev) {
  78. if ($cam_dev->{snmp_version}) { $d_version=$cam_dev->{snmp_version}; }
  79. if ($cam_dev->{community}) { $d_community=$cam_dev->{community}; }
  80. }
  81. if (!$d_version) { $d_version=$config_ref{snmp_default_version}; }
  82. if (!$d_community) { $d_community=$config_ref{snmp_default_community}; }
  83. my $snmp_info = scan_ipcam($auth->{ip},$d_community,$d_version);
  84. if ($snmp_info) {
  85. if (!$cam_dev) {
  86. $cam_dev->{user_id}=$auth->{'user_id'};
  87. $cam_dev->{device_type}=5;
  88. $cam_dev->{ip}=$auth->{'ip'};
  89. $cam_dev->{device_name}=$login->{login};
  90. $cam_dev->{snmp_version}=$config_ref{snmp_default_version};
  91. $cam_dev->{community}=$config_ref{snmp_default_community};
  92. my $new_dev_id = insert_record($dbh,"devices",$cam_dev);
  93. $cam_dev = get_record_sql($dbh,"SELECT * FROM devices WHERE user_id=".$auth->{'user_id'});
  94. }
  95. if ($snmp_info->{firmware} and ($cam_dev->{firmware} ne $snmp_info->{firmware})) {
  96. do_sql($dbh,"UPDATE devices SET firmware='".$snmp_info->{firmware}."' WHERE id=".$cam_dev->{'id'});
  97. }
  98. if ($snmp_info->{model_name}) {
  99. my $model = get_record_sql($dbh,"SELECT id FROM device_models WHERE model_name='".$snmp_info->{model_name}."'");
  100. if ($model and $cam_dev->{device_model_id} ne $model->{id}) {
  101. print "Updated: ".$snmp_info->{text}."\n";
  102. do_sql($dbh,"UPDATE devices SET device_model_id=".$model->{id}." WHERE id=".$cam_dev->{'id'});
  103. } else {
  104. print "OK: ".$snmp_info->{text}."\n";
  105. }
  106. }
  107. } else { print "Fail: ".$ip."\n"; }
  108. }
  109. }
  110. exit;