yubikey-val/ykval-import

#!/usr/bin/php
<?php

# Copyright (c) 2010-2015 Yubico AB
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#   * Redistributions of source code must retain the above copyright
#     notice, this list of conditions and the following disclaimer.
#
#   * 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 COPYRIGHT HOLDERS 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 COPYRIGHT
# OWNER 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.

set_include_path(implode(PATH_SEPARATOR, array(
	get_include_path(),
	'/usr/share/yubikey-val',
	'/etc/yubico/val',
)));

require_once 'ykval-config.php';
require_once 'ykval-db.php';


$logname="ykval-import";
$myLog = new Log($logname);

$db = Db::GetDatabaseHandle($baseParams, $logname);

if (!$db->connect()) {
  $myLog->log(LOG_WARNING, "Could not connect to database");
  error_log("Could not connect to database");
  exit(1);
}


while ($res=fgetcsv(STDIN, 0, ",")) {

  if ($res[0]===null || strpos($res[0], '#')===0)
    continue;

  $params=array(
    "active"=>$res[0],
    "created"=>$res[1],
    "modified"=>$res[2],
    "yk_publicname"=>$res[3],
    "yk_counter"=>$res[4],
    "yk_use"=>$res[5],
    "yk_low"=>$res[6],
    "yk_high"=>$res[7],
    "nonce"=>$res[8],
    "notes"=>$res[9]
  );

  # FIXME
  # INSERT first, if duplicate error, UPDATE.

  $query="SELECT * FROM yubikeys WHERE yk_publicname='" . $params['yk_publicname'] . "'";
  $result=$db->customQuery($query);

  if ($db->rowCount($result)) {
    $query="UPDATE yubikeys SET " .
      "active='" . $params["active"] . "' " .
      ",created='" . $params["created"] . "' " .
      ",modified='" . $params["modified"] . "' " .
      ",yk_counter='" . $params["yk_counter"] . "' " .
      ",yk_use='" . $params["yk_use"] . "' " .
      ",yk_low='" . $params["yk_low"] . "' " .
      ",yk_high='" . $params["yk_high"] . "' " .
      ",nonce='" . $params["nonce"] . "' " .
      ",notes='" . $params["notes"] . "' " .
      "WHERE yk_publicname='" . $params['yk_publicname'] . "' AND " .
      "(".$params['yk_counter'].">yk_counter or (".$params['yk_counter']."=yk_counter and " .
      $params['yk_use'] . ">yk_use))";

    if(!$db->customQuery($query)) {
      $myLog->log(LOG_ERR, "Failed to update yk_publicname with query " . $query);
      error_log("Failed to update yk_publicname with query " . $query);
      exit(1);
    }
  }
  else {
    // We didn't have the yk_publicname in database so we need to do insert instead
    $query="INSERT INTO yubikeys " .
      "(active,created,modified,yk_publicname,yk_counter,yk_use,yk_low,yk_high,nonce,notes) VALUES " .
      "('" . $params["active"] . "', " .
      "'" . $params['created'] . "'," .
      "'" . $params['modified'] . "'," .
      "'" . $params['yk_publicname'] . "'," .
      "'" . $params['yk_counter'] . "'," .
      "'" . $params['yk_use'] . "'," .
      "'" . $params['yk_low'] . "'," .
      "'" . $params['yk_high'] . "'," .
      "'" . $params['nonce'] . "'," .
      "'" . $params['notes'] . "')";

    if (!$db->customQuery($query)) {
      $myLog->log(LOG_ERR, "Failed to insert new yk_publicname with query " . $query);
      error_log("Failed to insert new yk_publicname with query " . $query);
      exit(1);
    }
  }

  $db->closeCursor($result);
}

$myLog->log(LOG_NOTICE, "Successfully imported yubikeys to database");
echo "Successfully imported yubikeys to database\n";