Source for file epo_sync.php

Documentation is available at epo_sync.php

  1. #!/usr/bin/php
  2. <?php
  3. /**
  4.  * enterprise/epo_sync
  5.  *
  6.  * Long description for file:
  7.  * 
  8.  * PHP version 5
  9.  *
  10.  * LICENSE: This program is free software; you can redistribute it and/or
  11.  * modify it under the terms of the GNU General Public License as published
  12.  * by the Free Software Foundation.
  13.  *
  14.  * @package                     FreeNAC
  15.  * @author                      Wolfram Strauss, Sean Boran (FreeNAC Core Team)
  16.  * @copyright                   2006 FreeNAC
  17.  * @license                     http://www.gnu.org/copyleft/gpl.html   GNU Public License Version 2
  18.  * @version                     CVS: $Id:$
  19.  * @link                        http://www.freenac.net
  20.  *
  21.  */
  22.  
  23.  
  24.  
  25.  
  26. /*************************************
  27.         MAIN
  28. **************************************/
  29.  
  30. $EPO_VERSION = 4;    // either 3 or 4
  31.  
  32.  
  33. // Php weirdness: change to script dir, then look for includes
  34. chdir(dirname(__FILE__));
  35. set_include_path("../:./");
  36.  
  37.  
  38. /**
  39. * Load settings and common functions
  40. */
  41. require_once "./funcs.inc.php";     
  42.  
  43. global $logger;
  44. $logger->setLogToStdOut(false);
  45. $output=true;
  46. set_time_limit(0);
  47. $logger->setDebugLevel(0);
  48.  
  49. #  epo version checking
  50. if( $EPO_VERSION == 3 )
  51. {
  52.     $epo_db_prefix = '';
  53. }
  54. elseif( $EPO_VERSION == 4 )
  55. {
  56.     $epo_db_prefix = 'EPO';
  57. }
  58. else
  59. {
  60.     $logger->logit("Version $EPO_VERSION is not supported!", LOG_ERR);
  61.     exit;
  62. }
  63.  
  64. // this timestamp is used as the sync date for data synced from the wsus db to the freenac db
  65. $timestamp=date('Y-m-d H:i:s');
  66. $logger->logit("Starting EPO sync job");
  67.  
  68. db_connect();    // vmps db connection (mysql)
  69. $enabled=v_sql_1_select("select value from config where name='epo_enabled'");
  70.  
  71. if( $enabled )
  72. {
  73.     dbepo_connect();    // epo db connection (mssql)
  74.  
  75.     // warning!! this should run within a transaction ...
  76.     
  77.     if( !empty_tables() )
  78.     {
  79.         $logger->logit("Failed to empty all epo tables, logical status may be inconsistend!", LOG_ERR);
  80.         cleanup();
  81.     }
  82.     
  83.     if( !get_systems() )
  84.     {
  85.         $logger->logit("Failed to sync systems list", LOG_ERR);
  86.         cleanup;
  87.     }
  88.     
  89.     if( !get_versions() )
  90.     {
  91.         $logger->logit("Failed to sync versions list", LOG_ERR);
  92.         cleanup;
  93.     }
  94.     
  95.     // ... end of suggested transaction
  96.     
  97.     cleanup();
  98.         
  99. }
  100. else
  101. {
  102.     $logger->logit("EPO support not enabled", LOG_WARNING);
  103. }
  104.  
  105.  
  106.  
  107. /*******************************
  108.     FUNCTIONS
  109. *******************************/
  110.     
  111.  
  112. /**
  113. *  This function converts the datetime retrieved from MSSQL into MySQL datetime format
  114. */
  115. function convert_date($date)
  116. {
  117.    $date_array=getdate(strtotime($date));
  118.    $date=$date_array['year'].'-';
  119.    $date_array['mon'] < 10 ? $date.='0'.$date_array['mon'].'-' : $date.=$date_array['mon'].'-';
  120.    $date_array['mday'] < 10 ? $date.='0'.$date_array['mday'].' ' : $date.=$date_array['mday'].' ';
  121.    $date_array['hours'] < 10 ? $date.='0'.$date_array['hours'].':' : $date.=$date_array['hours'].':';
  122.    $date_array['minutes'] < 10 ? $date.='0'.$date_array['minutes'].':' : $date.=$date_array['minutes'].':';
  123.    $date_array['seconds'] < 10 ? $date.='0'.$date_array['seconds'] : $date.=$date_array['seconds'];
  124.    return $date;
  125. }
  126.  
  127.  
  128. /**
  129. * Ensures that $string is mysql safe
  130. */
  131. function validate($string)
  132. {
  133.    rtrim($string,' ');
  134.    if (get_magic_quotes_gpc()) {
  135.       $value=stripslashes($string);
  136.    }
  137.    if (!is_numeric($string)) {
  138.       $string= mysql_real_escape_string($string);
  139.    }
  140.    return $string;
  141. }
  142.  
  143.  
  144. /**
  145. * Connect to the EPO server
  146. */
  147. function dbepo_connect()
  148. {
  149.    global $conf,$epo_dbuser,$epo_dbpass, $logger;
  150.    $logger->debug("Connecting to ".$conf->epo_dbalias." ".$conf->epo_db, 2);
  151.    $msconnect = mssql_connect($conf->epo_dbalias, $epo_dbuser, $epo_dbpass);
  152.    if ( !$msconnect )
  153.    {
  154.      $logger->logit("Cannot connect to EPO server ".$conf->epo_dbalias.":" . mssql_get_last_message(), LOG_ERR);
  155.      return false;
  156.    }
  157.    $db = mssql_select_db($conf->epo_db, $msconnect);
  158.    if ( !$db )
  159.    {
  160.       $logger->logit("Couldn't open database ".$conf->epo_db." ".mssql_get_last_message(), LOG_ERR);
  161.       return false;
  162.    }
  163. }
  164.  
  165.  
  166. /**
  167. * Returns the hostname part of an fqdn thus everything before the first dot
  168. */
  169. function get_hostname($fqdn)
  170. {
  171.     global $logger;
  172.     
  173.     $dot_pos = strpos($fqdn, '.');
  174.     $hostname;
  175.     if( $dot_pos )
  176.     {
  177.         $hostname = substr($fqdn, 0, $dot_pos);
  178.     }
  179.     else
  180.     {
  181.         $hostname = $fqdn;
  182.     }
  183.     
  184.     $logger->debug("Converting $fqdn to $hostname", 2);
  185.     
  186.     return strtolower($hostname);
  187. }
  188.  
  189. /**
  190. * Look up a wsus hostname in the vmps table and return the vmps id if and only if there's exactly one entry
  191. */
  192. function get_vmps_id($mac)
  193. {
  194.     global $logger;
  195.     
  196.     $query = "select id from systems where mac = '$mac';";
  197.     $logger->debug("Executing $query", 3);
  198.     $result = mysql_query($query);
  199.     if( !$result )
  200.     {
  201.         $logger->logit("Could not obtain vmps id for mac $mac, " . mysql_error(), LOG_WARNING);
  202.         return false;
  203.     }
  204.     $num_rows = mysql_num_rows($result);    //TODO: exception handling
  205.     if( $num_rows == 0 )
  206.     {
  207.         #$logger->logit("No vmps id for mac $mac found", LOG_WARNING);
  208.         $logger->logit("No vmps id for mac $mac found"); // don't flag as warning until 100% right, its flooding logcheck
  209.         return false;
  210.     }
  211.     elseif( $num_rows == 1 )
  212.     {
  213.         $row = mysql_fetch_row($result);    //TODO: exception handling
  214.         $logger->debug("System with mac $mac matches vmps id $row[0]", 2);
  215.         return $row[0];    
  216.     }
  217.     else
  218.     {
  219.         $logger->logit("mac $mac is not unique in vmps", LOG_WARNING);
  220.         return false;
  221.     }
  222. }
  223.  
  224.  
  225. /**
  226. * Convert mac address obtain from epo into vmps format
  227. */
  228. function convert_mac($epo_mac)
  229. {
  230.     global $logger;
  231.     
  232.     $vmps_mac = preg_replace('/(\w{4})(\w{4})(\w{4})/', '$1.$2.$3', $epo_mac);
  233.     
  234.     $logger->debug("Mac format conversion: $epo_mac -> $vmps_mac", 2);
  235.     
  236.     return $vmps_mac;
  237. }
  238.  
  239. /**
  240. * Empty all epo tables to get ready for fresh sync
  241. */
  242. function empty_tables()
  243. {
  244.     global $logger;
  245.  
  246.     $logger->debug("Emptying tables", 1);
  247.     if( !mysql_query('truncate table epo_systems;') ) {
  248.         $logger->logit("Could not empty epo_systems, " . mysql_error(), LOG_ERR);
  249.         return false;
  250.     }
  251.     if( !mysql_query('truncate table epo_versions;') ) {
  252.         $logger->logit("Could not empty epo_versions, " . mysql_error(), LOG_ERR);
  253.         return false;
  254.     }
  255.     
  256.         
  257.     return true;
  258. }
  259.  
  260.  
  261. /**
  262. * Obtain list of systems managed by the epo server
  263. */
  264. function get_systems()
  265. {
  266.     global $logger, $timestamp, $epo_db_prefix;
  267.  
  268.     $query = "select l.nodename, l.lastupdate, l.agentversion, c.ostype, c.netaddress, c.ipaddress, c.freediskspace, p.enginever, p.datver, p. productversion, p.hotfix, c.domainname, c.ostype, c.osversion, c.osservicepackver, c.osbuildnum, c.username from dbo.${epo_db_prefix}leafnode l left join dbo.${epo_db_prefix}computerproperties c on l.autoid = c.parentid left join dbo.${epo_db_prefix}productproperties p on l.autoid = p.parentid where p.productcode = 'VIRUSCAN8600'";
  269.     
  270.     $logger->debug("Executing $query", 3);
  271.     $result = mssql_query($query);
  272.     if( !$result )
  273.     {
  274.         $logger->logit("Failed to obtain systems from epo, " . mssql_get_last_message(), LOG_ERR);
  275.         return false;
  276.     }
  277.     
  278.     while( $sys_row = mssql_fetch_assoc($result) )
  279.     {
  280.         // TODO: exception handling
  281.         $hostname = get_hostname($sys_row['nodename']);
  282.         $mac = convert_mac($sys_row['netaddress']);
  283.         $id = get_vmps_id($mac);
  284.         if( !$id )
  285.         {
  286.             continue;
  287.         }
  288.         
  289.         // insert system into epo_systems
  290.         $query = sprintf("insert into epo_systems values('%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s') on duplicate key update lastsync=now();", $id, $hostname, validate($sys_row['domainname']), validate($sys_row['ipaddress']), $mac, validate($sys_row['agentversion']), convert_date($sys_row['lastupdate']), validate($sys_row['productversion']), validate($sys_row['enginever']), validate($sys_row['datver']), validate($sys_row['hotfix']), validate($sys_row['ostype']), validate($sys_row['osversion']), validate($sys_row['osservicepackver']), validate($sys_row['osbuildnum']), validate($sys_row['freediskspace']), validate($sys_row['username']), $timestamp);
  291.         
  292.         $logger->debug("Executing: $query", 3);
  293.         if( !mysql_query($query) )
  294.         {
  295.             $logger->logit("Could not insert system $hostname, " . mysql_error(), LOG_WARNING);
  296.             continue;
  297.         }
  298.         
  299.     }
  300.     
  301.     return true;
  302. }
  303.  
  304.  
  305. /**
  306. * Obtain list of product actual product version checked into the epo server
  307. */
  308. function get_versions()
  309. {
  310.     global $logger, $timestamp, $EPO_VERSION;
  311.     
  312.     if( $EPO_VERSION == 3 )
  313.     {
  314.         $query = "select type, version from dbo.latestupdates";
  315.         
  316.         $logger->debug("Executing: $query", 3);
  317.         $result = mssql_query($query);
  318.         
  319.         if( !$result )
  320.         {
  321.             $logger->logit("Failed to obtain product versions from epo, " . mssql_get_last_message(), LOG_ERR);
  322.             return false;
  323.         }
  324.     
  325.         while( $row = mssql_fetch_assoc($result) )
  326.         {
  327.             $query = sprintf("insert into epo_versions(product, version, hotfix, lastsync) values('%s', '%s', '%s', '%s') on duplicate key update lastsync=now();", validate($row['type']), validate($row['version']), '', $timestamp);
  328.         
  329.             $logger->debug("Executing: $query", 3);
  330.             if( !mysql_query($query) )
  331.             {
  332.                 $logger->logit("Could not insert product " . $row['type'] . ", " . mysql_error(), LOG_WARNING);
  333.                 continue;
  334.             }
  335.         }
  336.     }
  337.     elseif( $EPO_VERSION == 4 )
  338.     {
  339.         $query = "select productcode, productversion, hotfixversion from dbo.epomastercatalog";
  340.         
  341.         $logger->debug("Executing: $query", 3);
  342.         $result = mssql_query($query);
  343.         
  344.         if( !$result )
  345.         {
  346.             $logger->logit("Failed to obtain product versions from epo, " . mssql_get_last_message(), LOG_ERR);
  347.             return false;
  348.         }
  349.     
  350.         while( $row = mssql_fetch_assoc($result) )
  351.         {
  352.             $query = sprintf("insert into epo_versions(product, version, hotfix, lastsync) values('%s', '%s', '%s', '%s') on duplicate key update lastsync=now();", validate($row['productcode']), validate($row['productversion']), validate($row['hotfixversion']), $timestamp);
  353.         
  354.             $logger->debug("Executing: $query", 3);
  355.             if( !mysql_query($query) )
  356.             {
  357.                 $logger->logit("Could not insert product " . $row['productcode'] . ",  " . mysql_error(), LOG_WARNING);
  358.                 continue;
  359.             }
  360.         }
  361.     }
  362.     else
  363.     {
  364.         $logger->logit("Wrong epo version $EPO_VERSION. The variable must have changed during execution -> very bad!", LOG_ERR);
  365.         exit(1);
  366.     }
  367.     
  368.     return true;
  369. }
  370.  
  371.  
  372. /**
  373. *
  374. */
  375. function cleanup($exit_code = 0)
  376. {
  377.     global $logger;
  378.  
  379.     // TODO: exception handling    
  380.     
  381.     mssql_close();
  382.     mysql_close();
  383.     
  384.     $logger->logit("Done syncing EPO");
  385.     
  386.     exit($exit_code);
  387. }
  388.     
  389.     
  390. ?>

Documentation generated on Mon, 01 Dec 2008 01:10:27 +0100 by phpDocumentor 1.4.0