switch-ztp/www/export.php

<?php
#-
# Copyright (c) 2011-2015 Joe Clarke <jclarke@cisco.com>
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
#
#

include_once 'db.inc.php';
include_once 'swreg/swreg.inc.php';
include_once 'swreg_web.inc.php';
include_once 'swreg_creds.inc.php';
require_once 'MDB2.php';
require_once 'Log.php';
require_once 'functions.php';

$dsn = "mysql://$db_user:$db_pass@$db_host/$db_name";
$options = array('result_buffering', false);

$dbh = MDB2::factory($dsn, $options);
if (PEAR::isError($dbh)) {
	die($dbh->getMessage());
}
$dbh->setFetchMode(MDB2_FETCHMODE_ASSOC);

$logger = Log::singleton('file', LOGFILE, TOOL_NAME . ' : Exporter');
if ($logger === FALSE) {
	die("Failed to open logfile.\n");
}
$mask = Log::UPTO(LOG_LEVEL);
$logger->setMask($mask);

$base = get_base($_SERVER['SCRIPT_NAME']);

$sql = 'SELECT * FROM DEVICE_MAP WHERE checked_out=1';
$res = $dbh->query($sql);
if (PEAR::isError($res)) {
	echo "<p><font color=\"red\">Error querying for physical switches: {$res->getUserInfo()}</font></p>\r\n";
	exit(1);
}

$p['authentication']['user'] = PRIME_API_USER;
$p['authentication']['password'] = PRIME_API_PASS;

$url = PRIME_URL . '/webacs/api/v1/op/devices/bulkImport.json';
$d['devicesImport']['devices']['device'] = array();
$dns_devices = array();

//$csv_file = array("ip_address,snmp_version,snmp_community,snmp_retries,snmp_timeout,protocol,cli_username,cli_password,cli_enable_password,cli_timeout,snmpv3_user_name,snmpv3_auth_type,snmpv3_auth_password,snmpv3_privacy_type,snmpv3_privacy_password,http_server,http_port,http_config_username,http_config_password,http_monitor_username,http_monitor_password,udf:Serial Number");
while ($row = $res->fetchRow()) {
	$sql = "SELECT ip_address FROM SWITCHES WHERE name = '{$row['assigned_switch']}'";
	$sw_res = $dbh->query($sql);
	if (PEAR::isError($sw_res)) {
		echo "<p><font color=\"red\">Error querying for IP address for {$row['assigned_switch']}: {$sw_res->getUserInfo()}</font></p>\r\n";
		continue;
	}
	$sw_row = $sw_res->fetchRow();
	$device['ipAddress'] = $sw_row['ip_address'];
	$device['credentialProfileName'] = PRIME_CRED_PROFILE;
	$device['udfs']['udf'] = array();
	$dns_device['ip'] = $sw_row['ip_address'];
	$dns_device['name'] = $row['assigned_switch'];
	$sn['name'] = 'Serial Number';
	$sn['value'] = $row['serial_number'];
	array_push($device['udfs']['udf'], $sn);
	array_push($d['devicesImport']['devices']['device'], $device);
	array_push($dns_devices, $dns_device);
	unset($device);
}

$json = json_encode($d);
$p['put'] = $json;
$p['header']['Content-Type'] = 'application/json';

$res = get_PI_REST($url, $p);
$errors = array();
$dns_pass = DNS_PASS;
$dns_user = DNS_USER;
$dns_server = DNS_SERVER;
$dns_zone = DNS_ZONE;
$ztp_user = ZTP_USER;
if (!$res) {
	echo "<font color=\"red\">ERROR: Failed to export to PI</font>\n";
} else {
	foreach ($dns_devices as $device) {
		$out = `/usr/local/bin/sudo -u $ztp_user /usr/bin/ssh -2 root@$dns_server /opt/nwreg2/local/usrbin/nrcmd -N $dns_user -P $dns_pass zone $dns_zone showHost {$device['name']}`;
		if (preg_match("/302 Not Found/", $out)) {
			$out = `/usr/local/bin/sudo -u $ztp_user /usr/bin/ssh -2 root@$dns_server /opt/nwreg2/local/usrbin/nrcmd -N $dns_user -P $dns_pass zone $dns_zone addHost {$device['name']} {$device['ip']}`;
			if (!preg_match("/100 Ok/", $out)) {
				array_push($errors, "Error adding entry for {$device['name']}: '$out'");
			}
		} else {
			preg_match("/([^:]+): addr=([^;]+)/", $out, $matches);
			$h = $matches[1];
			$a = $matches[2];
			if ($a != $device['ip']) {
				$out = `/usr/local/bin/sudo -u $ztp_user /usr/bin/ssh -2 root@$dns_server /opt/nwreg2/local/usrbin/nrcmd -N $dns_user -P $dns_pass zone $dns_zone removeHost {$device['name']}`;
				if (!preg_match("/100 Ok/", $out)) {
					array_push($errors, "Failed to remove host {$device['name']}: '$out'");
					continue;
				}

				$out = `/usr/local/bin/sudo -u $ztp_user /usr/bin/ssh -2 root@$dns_server /opt/nwreg2/local/usrbin/nrcmd -N $dns_user -P $dns_pass zone $dns_zone addHost {$device['name']} {$device['ip']}`;
				if (!preg_match("/100 Ok/", $out)) {
					array_push($errors, "Error adding entry for {$device['name']}: '$out'");
				}
			}
		}
	}
	cleanup();
	if (count($errors) > 0) {
		echo "<font color=\"red\">ERROR: Failed to add entries to DNS:<br/></font>\n";
		echo (implode("<br/>", $errors));
	} else {
?>
	<script language="Javascript">
	window.close();
	</script>
<?php
	}
}

?>