Source for file wsus_getinfo.php

Documentation is available at wsus_getinfo.php

  1. #!/usr/bin/php
  2. <?php
  3. /**
  4.  * PHP version 5
  5.  *
  6.  * LICENSE: This program is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU General Public License as published
  8.  * by the Free Software Foundation.
  9.  *
  10.  * @package                     FreeNAC
  11.  * @author                      Hector Ortiz (FreeNAC Core Team)
  12.  * @copyright                   2006 FreeNAC
  13.  * @license                     http://www.gnu.org/copyleft/gpl.html   GNU Public License Version 2
  14.  * @version                     CVS: $Id:$
  15.  * @link                        http://www.freenac.net
  16.  *
  17.  */
  18.  
  19. /**
  20. * Retrieves information from the WSUS database and stored in local mysql
  21. * table.
  22. * Beta test with Wsus V2.???
  23. * Stable release v2.2 testted with Wusus version 3.???
  24. */
  25.  
  26. # Php weirdness: change to script dir, then look for includes
  27. chdir(dirname(__FILE__));
  28. set_include_path("../:./");
  29.  
  30. /**
  31. * Load settings and commong functions
  32. */
  33. require_once "./funcs.inc.php";     
  34.  
  35. $logger->setLogToStdOut();
  36. $output=true;
  37. set_time_limit(0);
  38.  
  39. $verbose=FALSE;
  40.  
  41. /**
  42. * Simple parsing of command line parameters
  43. */
  44. for ($i=1;$i<$argc;$i++)
  45. {
  46.    switch($argv[$i])
  47.    {
  48.       case '-v':
  49.          {
  50.             $logger->setDebugLevel(1);
  51.             $verbose = TRUE;
  52.          }
  53.          break;
  54.       case '-vv':
  55.          {
  56.             $logger->setDebugLevel(2);
  57.             $verbose = TRUE;
  58.          }
  59.          break;
  60.       case '-s':$output = FALSE;
  61.          break;
  62.       case '-h':
  63.       default:usage();
  64.          break;
  65.    }
  66. }
  67.  
  68. $timestamp=date('Y-m-d H:i:s');
  69. message("Program run on $timestamp",1);
  70.  
  71. /**
  72. * Prints script's usage
  73. */
  74. function usage()
  75. {
  76.    $logger->logit( "Usage: wsus_getinfo [-h][-v[v]][-s]\n");
  77.    $logger->logit( "\t-h\tShow this help screen\n");
  78.    $logger->logit( "\t-v\tDebug level 1 enabled (output goes to stdout & syslog)\n");
  79.    $logger->logit( "\t-vv\tDebug level 1 & 2 enabled (output goes to stdout & syslog)\n");
  80.    $logger->logit( "\t-s\tSupress messages to standard output and redirect them to syslog\n");
  81.    exit(1);
  82. }
  83.  
  84. /**
  85. * Connect to the WSUS server
  86. */
  87. function dbwsus_connect()
  88. {
  89.    global $conf,$wsus_dbuser,$wsus_dbpass;
  90.    message("Connect to ".$conf->wsus_dbalias." ".$conf->wsus_db,1);
  91.    $msconnect = mssql_connect($conf->wsus_dbalias, $wsus_dbuser, $wsus_dbpass);
  92.    if (! $msconnect ) 
  93.    {
  94.      message("Cannot connect to WSUS server ".$conf->wsus_dbalias.":" . mssql_get_last_message(),0);
  95.      return false;
  96.    }
  97.    $d = mssql_select_db($conf->wsus_db, $msconnect);
  98.    if (! $d)
  99.    {
  100.       message("Couldn't open database ".$conf->wsus_db." ".mssql_get_last_message(),0);
  101.       return false;
  102.    }
  103.    return true;
  104. }
  105.  
  106. /**
  107. * Ensures that $string is mysql safe
  108. */
  109. function validate($string)
  110. {
  111.    rtrim($string,' ');
  112.    if (get_magic_quotes_gpc()) {
  113.       $value=stripslashes($string);
  114.    }
  115.    if (!is_numeric($string)) {
  116.       $string= mysql_real_escape_string($string);
  117.    }
  118.    return $string;
  119. }
  120.  
  121. /**
  122. * Executes query and displays error message if any
  123. */
  124. function execute_query($query)
  125. {
  126.    db_connect();
  127.    $res=mysql_query($query);
  128.    if (!$res)
  129.    { 
  130.       message("Cannot execute query $query because ".mysql_error(),2);
  131.       return false;
  132.    }
  133.    return $res;
  134. }
  135.  
  136. /**
  137. *  This function converts the datetime retrieved from MSSQL into MySQL datetime format
  138. */
  139. function convert_date($date)
  140. {
  141.    $date_array=getdate(strtotime($date));
  142.    $date=$date_array['year'].'-';
  143.    $date_array['mon'] < 10 ? $date.='0'.$date_array['mon'].'-' : $date.=$date_array['mon'].'-';
  144.    $date_array['mday'] < 10 ? $date.='0'.$date_array['mday'].' ' : $date.=$date_array['mday'].' ';
  145.    $date_array['hours'] < 10 ? $date.='0'.$date_array['hours'].':' : $date.=$date_array['hours'].':';
  146.    $date_array['minutes'] < 10 ? $date.='0'.$date_array['minutes'].':' : $date.=$date_array['minutes'].':';
  147.    $date_array['seconds'] < 10 ? $date.='0'.$date_array['seconds'] : $date.=$date_array['seconds'];
  148.    return $date;
  149. }
  150.  
  151. /**
  152. * Dumps the tbComputerTarget table into our computertarget tabl
  153. */
  154. function wsus_dump_computertarget()
  155. {
  156.    message("Function wsus_dump_computertarget",1);
  157.    db_connect();
  158.    if (dbwsus_connect())
  159.    {
  160.       #This query is for WSUS 2.X
  161.       #$query="select TargetID,IPAddress,FullDomainName,OSMajorVersion,OSMinorVersion,OSBuildNumber,OSServicePackMajorNumber,OSServicePackMinorNumber,OSLocale,ComputerMake,ComputerModel,BiosVersion,BiosName,BiosReleaseDate,ProcessorArchitecture from tbComputerTarget";
  162.       #This query is for WSUS 3.0
  163.       $query="select ct.TargetID, ct.IPAddress, ct.FullDomainName, td.OSMajorVersion, td.OSMinorVersion, td.OSBuildNumber, td.OSServicePackMajorNumber, td.OSServicePackMinorNumber, td.OSLocale, td.ComputerMake, td.ComputerModel, td.BiosVersion, td.BiosName, td.BiosReleaseDate, td.ProcessorArchitecture from tbComputerTarget as ct, tbcomputertargetdetail as td where ct.targetID=td.targetID";
  164.       message("Executing: ".$query,2);
  165.       $res=mssql_fetch_all($query);
  166.       foreach ($res as $row)
  167.       {
  168.      $old_date=$row[BiosReleaseDate];
  169.      $temp=explode('.',$row[FullDomainName]);
  170.          $row[FullDomainName]=$temp[0];
  171.          $row[BiosReleaseDate]=convert_date($row[BiosReleaseDate]);
  172.          message("Converted $old_date into ".$row[BiosReleaseDate]." for ".$row[FullDomainName],1);
  173.          $row[OSLocale]=substr($row[OSLocale],0,2);
  174.          $osid=wsus_get_osid($row[OSMajorVersion],$row[OSMinorVersion],$row[OSBuildNumber],$row[OSServicePackMajorNumber],$row[OSServicePackMinorNumber],$row[ProcessorArchitecture]);
  175.          $query=sprintf("insert into nac_wsuscomputertarget (TargetID, IPAddress, FullDomainName, OSid, OSLocale, ComputerMake, ComputerModel, BiosVersion, BiosName, BiosReleaseDate) values ('%s','%s','%s','%s','%s','%s','%s','%s','%s','%s') on duplicate key update datetime=NOW();", validate($row[TargetID]), validate($row[IPAddress]), validate($row[FullDomainName]), validate($osid), validate($row[OSLocale]), validate($row[ComputerMake]), validate($row[ComputerModel]), validate($row[BiosVersion]), validate($row[BiosName]), validate($row[BiosReleaseDate]));
  176.          message("Executing: ".$query,2);
  177.          execute_query($query);
  178.       }
  179.       return true;
  180.    }
  181.    else return false;
  182. }
  183.  
  184. /**
  185. * Dumps table tbOSMap into our osmap table
  186. */
  187. function wsus_dump_osmap() 
  188. {
  189.    message("Function wsus_dump_osmap",1);
  190.    if (dbwsus_connect())
  191.    {
  192.       $query="select * from tbOSMap";
  193.       message("Executing: ".$query,2);
  194.       $res=mssql_fetch_all($query);
  195.       db_connect();
  196.       $query='delete from nac_wsusosmap'; //Delete everything we previosuly had.
  197.       message("Executing: ".$query,2);
  198.       execute_query($query);
  199.       foreach ($res as $row)
  200.       {
  201.          $query=sprintf("insert into nac_wsusosmap values ('%s','%s','%s','%s','%s','%s','%s','%s','%s');", $row[OSid], $row[OSMajorVersion], $row[OSMinorVersion], $row[OSBuildNumber], $row[OSServicePackMajorNumber], $row[OSServicePackMinorNumber], $row[ProcessorArchitecture],validate($row[OSShortName]),validate($row[OSLongName]));
  202.          message("Executing: ".$query,2);
  203.          execute_query($query);
  204.       }
  205.       return true;
  206.    }
  207.    else return false;
  208. }
  209.  
  210. /**
  211. * Get the OSID
  212. */
  213. function wsus_get_osid($majv,$minv,$build,$spmaj,$spmin,$processor)
  214. {
  215.    message("Function wsus_getosid",1);
  216.    db_connect();
  217.    $query="select pid from nac_wsusprocessor where ProcessorArchitecture='$processor';";
  218.    message("Executing: ".$query,2);
  219.    $my_res=execute_query($query);
  220.    if ($my_res)
  221.    {
  222.       $my_result=mysql_fetch_array($my_res,MYSQL_ASSOC);
  223.       if (dbwsus_connect())
  224.       {
  225.          $query=sprintf("select osid from tbosmap where osmajorversion='%s' and osminorversion='%s' and osbuildnumber='%s' and osservicepackmajornumber='%s' and osservicepackminornumber='%s' and processorarchitecture='%s'",$majv,$minv,$build,$spmaj,$spmin,$my_result['pid']);
  226.          message("Executing: ".$query,2);
  227.          $ms_res=mssql_fetch_all($query);
  228.          if ($ms_res)
  229.          {
  230.             foreach($ms_res as $row)
  231.             {
  232.                $result=$row[osid];   
  233.             }         
  234.          } 
  235.       }
  236.       return $result;
  237.    }
  238.    else return false;
  239. }
  240.  
  241. /**
  242. * Retrieves the installed and needed updates for computer identified by id
  243. */
  244. function wsus_get_updates_per_computer($id) 
  245. {
  246.    message("Function wsus_get_updates_per_computer for $id",1);
  247.    if (dbwsus_connect())
  248.    {
  249.       #Query for WSUS 2.X
  250.       #$query="select SummarizationState, UpdateID, LastChangeTime, LastRefreshTime from tbUpdateStatusPerComputer where TargetID='$id' and SummarizationState=4 order by id asc"; #Installed updates
  251.       #Query for WSUS 3.0
  252.       $query="select SummarizationState, LocalUpdateID, LastChangeTime, LastRefreshTime from tbUpdateStatusPerComputer where TargetID='$id' and SummarizationState=4 order by LocalUpdateID asc"; #Installed updates
  253.       message("Executing: ".$query,2);
  254.       $installed=mssql_fetch_all($query);
  255.       if ($installed)
  256.       {
  257.          db_connect();
  258.          $query="delete from nac_wsusupdatestatuspercomputer where TargetID='$id' and SummarizationState='4';";
  259.          message("Executing: ".$query,2);
  260.          execute_query($query);
  261.          foreach($installed as $row)
  262.          {
  263.             #Query for WSUS 2.X
  264.             #$query=sprintf("insert into nac_wsusupdatestatuspercomputer values('%s','%s','%s','%s','%s');",validate($row[SummarizationState]),validate(mssql_guid_string($row[UpdateID])),validate($id),validate(convert_date($row[LastChangeTime])),validate(convert_date($row[LastRefreshTime])));
  265.             #Query for WSUS 3.0
  266.             $query=sprintf("insert into nac_wsusupdatestatuspercomputer values('%s','%s','%s','%s','%s');",validate($row[SummarizationState]),validate($row[LocalUpdateID]),validate($id),validate(convert_date($row[LastChangeTime])),validate(convert_date($row[LastRefreshTime])));
  267.             message("Executing: ".$query,2);
  268.             execute_query($query);
  269.          }
  270.       }
  271.       #Query for WSUS 2.x
  272.       #$query="select SummarizationState, UpdateID, LastChangeTime, LastRefreshTime from tbUpdateStatusPerComputer where TargetID='$id' and SummarizationState='6' order by id asc";#Needed updates
  273.       #Query for WSUS 3.0
  274.       $query="select SummarizationState, LocalUpdateID, LastChangeTime, LastRefreshTime from tbUpdateStatusPerComputer where TargetID='$id' and SummarizationState=4 order by LocalUpdateID asc"; #Installed updates
  275.       message("Executing: ".$query,2);
  276.       $needed=mssql_fetch_all($query);
  277.       if ($needed)
  278.       {
  279.          db_connect();
  280.          $query="delete from nac_wsusupdatestatuspercomputer where targetID='$id' and SummarizationState='6';";
  281.          message("Executing: ".$query,2);
  282.          execute_query($query);
  283.          $i=0;
  284.          foreach($needed as $row)
  285.          {
  286.             #Query for WSUS 2.X
  287.             #$query=sprintf("insert into nac_wsusupdatestatuspercomputer values('%s','%s','%s','%s','%s');",validate($row[SummarizationState]),validate(mssql_guid_string($row[UpdateID])),validate($id),validate(convert_date($row[LastChangeTime])),validate(convert_date($row[LastRefreshTime])));
  288.             #Query for WSUS 3.0
  289.             $query=sprintf("insert into nac_wsusupdatestatuspercomputer values('%s','%s','%s','%s','%s');",validate($row[SummarizationState]),validate($row[LocalUpdateID]),validate($id),validate(convert_date($row[LastChangeTime])),validate(convert_date($row[LastRefreshTime])));
  290.             message("Executing: ".$query,2);
  291.             execute_query($query);
  292.          }
  293.       }
  294.       #Query for WSUS 3.0
  295.       $query="select LastSyncTime  from tbcomputertarget where TargetID='$id';";
  296.       message("Executing: ".$query,2);
  297.       $result=mssql_fetch_all($query);
  298.       if ($result)
  299.       {
  300.          foreach($result as $row)
  301.          {
  302.             $lastsynctime=validate(convert_date($row[LastSyncTime]));
  303.             $query="update nac_wsuscomputertarget set LastSyncTime='$lastsynctime' where targetid='$id';";
  304.             message("Executing: ".$query,2);
  305.             execute_query($query);  
  306.          }
  307.       }
  308.       return true;
  309.    }
  310.    else return false;
  311. }
  312.  
  313. /**
  314. * This one retrieves the installed and needed updates for every computer
  315. */
  316. function wsus_dump_updates_for_computers()
  317. {
  318.    message("Function wsus_dump_updates_for_computers",1);
  319.    db_connect();
  320.    $query="select TargetID from nac_wsuscomputertarget;";
  321.    message("Executing: ".$query,2);
  322.    $my_res=execute_query($query);
  323.    if ($my_res)
  324.    {
  325.       while ($row=mysql_fetch_array($my_res,MYSQL_ASSOC))
  326.       {
  327.          wsus_get_updates_per_computer($row['TargetID']);
  328.       }
  329.       return true;
  330.    }
  331.    return false;
  332. }
  333.  
  334. /**
  335. * Connect data from WSUS with the data stored in the systems table
  336. */
  337. function connect_data()
  338. {
  339.    db_connect();
  340.    $query="select id,name from systems";
  341.    message("Executing: ".$query,2);
  342.    $my_res=execute_query($query);
  343.    if ($my_res)
  344.    {
  345.       while ($row=mysql_fetch_array($my_res,MYSQL_ASSOC))
  346.       {
  347.          if ((!empty($row['name']))&&(strcasecmp($row['name'],'unknown')!=0))
  348.          {
  349.             $query="update nac_wsuscomputertarget set sid=".$row['id']." where FullDomainName like '".$row['name']."'";
  350.             message("Executing: ".$query,2);
  351.             execute_query($query);
  352.             if (mysql_affected_rows()==1)
  353.                message("Updated patch information for host {$row['name']}",0);
  354.             else if (mysql_affected_rows()>1)
  355.                message("Possible duplicates in database for host {$row['name']}",0);
  356.          }
  357.          else
  358.             continue;
  359.       }
  360.       return true;
  361.    }
  362.    else
  363.       return false;
  364. }
  365.  
  366. /**
  367. * Get a list of updates available on the server
  368. */
  369. function wsus_dump_updates()
  370. {
  371.    global $wsus_language;
  372.    message("Function wsus_dump_updates",1);
  373.    if (dbwsus_connect() && $wsus_language)
  374.    {
  375.       $query="select u.localupdateid as LocalID,u.updateid as UpdateID,pre.title as Title,kb.kbarticleid as Article from tbprecomputedlocalizedproperty pre, tbkbarticleforrevision kb, tbupdate u where pre.updateid=u.updateid and pre.revisionid=kb.revisionid and pre.shortlanguage='$wsus_language' order by (kb.revisionid) asc";
  376.       message("Executing: ".$query,2);
  377.       $ms_res=mssql_fetch_all($query);
  378.       if ($ms_res)
  379.       {
  380.          db_connect();
  381.          /*$query='delete from nac_wsusupdate;';
  382.          message("Executing: ".$query,2);
  383.          execute_query($query);*/
  384.          foreach($ms_res as $row)
  385.          {
  386.             $updateid=mssql_guid_string($row[UpdateID]);
  387.             $query=sprintf("insert into nac_wsusupdate values('%s','%s','%s','%s') on duplicate key update UpdateID='%s';",validate($row[LocalID]),validate($updateid),validate($row[Title]),validate($row[Article]),validate($updateid));
  388.             message("Executing: ".$query,2);
  389.             execute_query($query);
  390.          }
  391.          return true;
  392.       }
  393.       else
  394.          return false;
  395.    }
  396.    else return false;
  397. }
  398.  
  399. /**
  400. * Print a message
  401. */
  402. function message($string,$level)
  403. {
  404.    global $output,$logger;
  405.    
  406.    if ($level==0)
  407.    {
  408.       if ($output)
  409.       {
  410.          $logger->setLogToStdOut();
  411.          $logger->logit("$string");
  412.       }
  413.       else
  414.       {
  415.          $logger->setLogToStdOut(false);
  416.          $logger->logit($string);
  417.       }
  418.    }
  419.    if ($level==1)
  420.    {
  421.       if ($logger->getDebugLevel())
  422.       {
  423.          $logger->debug($string,1);
  424.          if ($output)
  425.          {
  426.             $logger->setLogToStdOut();
  427.             $logger->logit("$string");
  428.          }
  429.       }
  430.    }
  431.    if ($level==2)
  432.    {
  433.       if ($logger->getDebugLevel())
  434.       {
  435.          $logger->debug($string,2);
  436.          if ($output)
  437.          {
  438.             $logger->setLogToStdOut();
  439.             $logger->logit("$string");
  440.          }
  441.       }
  442.    }
  443.    return true;
  444. }
  445.  
  446. $wsus_language='en';
  447. db_connect();
  448. $enabled=v_sql_1_select("select value from config where name='wsus_enabled'");
  449. if ($enabled)
  450. {
  451.    message("Dumping remote table tbOSmap into our osmap table.",0);
  452.    if (wsus_dump_osmap())
  453.    {
  454.       message("Dumping patches available on the server",0);
  455.       if (wsus_dump_updates())
  456.       {
  457.          message("Dumping remote table tbComputerTarget into our computertarget table.",0);
  458.          if (wsus_dump_computertarget())
  459.          {
  460.             message("Retrieving patch information for computers.",0);
  461.             if (wsus_dump_updates_for_computers())
  462.             {
  463.                message("Connecting information from WSUS to the systems table.",0);
  464.            if (connect_data())
  465.                {
  466.                   message("Done!",0);
  467.                   logit("WSUS synchronization was successful.");
  468.                   log2db('info',"WSUS synchronization was successful.");
  469.                   exit(0);
  470.                }
  471.                else message("Function connect_data failed.",0);
  472.             }
  473.             else message("Function wsus_dump_updates_for_computers failed.",0);
  474.          }
  475.          else message("Function wsus_dump_computertarget failed.",0);
  476.       }
  477.       else message("Function wsus_dump_updates failed.",0);
  478.    }
  479.    else message("Function wsus_dump_osmap failed.",0);
  480.    logit("WSUS synchronization failed.");
  481.    log2db('err',"WSUS synchronization failed.");
  482.    exit(1);
  483. }
  484. else
  485. { 
  486.    message("This function is not enabled",0);
  487.    exit(1);
  488. }
  489.  
  490. </