1
0
mirror of https://github.com/Yubico/yubikey-val.git synced 2024-11-28 15:24:14 +01:00
yubikey-val/ykval-gen-clients
2020-03-03 11:07:04 +11:00

127 lines
4.2 KiB
PHP
Executable File

#!/usr/bin/php
<?php
# Copyright (c) 2013 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',
)));
$options = getopt("h", array("help", "email::", "notes::", "otp::", "urandom"));
if(array_key_exists('h', $options) || array_key_exists('help', $options)) {
echo "Usage: ".$argv[0]." [ OPTIONS ] [ num_clients ]\n";
echo " Unless num_clients is defined, one client will be generated.\n\n";
echo "Options supported by ".$argv[0]."\n";
echo " --urandom\n";
echo " Use /dev/urandom instead of /dev/random as entropy source\n";
echo " --email=EMAIL\n";
echo " Sets the e-mail field of the created clients\n";
echo " --notes=NOTES\n";
echo " Sets the notes field of the created clients\n";
echo " --otp=OTP\n";
echo " Sets the otp field of the created clients\n";
exit(1);
}
require_once 'ykval-config.php';
require_once 'ykval-db.php';
$logname="ykval-gen-clients";
$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);
}
$count=1;
if($last_arg = intval(array_pop($argv))) {
$count = $last_arg;
}
$result = $db->customQuery("SELECT id FROM clients ORDER BY id DESC LIMIT 1");
$row = $db->fetchArray($result);
$db->closeCursor($result);
if($row) {
$next_id = $row['id']+1;
} else {
$next_id = 1;
}
$random = array_key_exists('urandom', $options) ? "/dev/urandom" : "/dev/random";
$fh = fopen($random, "r");
if(!$fh) {
die("cannot open ".$random);
}
for ($i=0; $i<$count; $i++) {
$client_id = $next_id++;
if (!($rnd = fread ($fh, 20))) {
die("cannot read from ".$random);
}
$secret = base64_encode($rnd);
$params = array(
"id" => $client_id,
"active" => 1,
"created" => time(),
"secret" => $secret,
"email" => array_key_exists('email', $options) ? $options['email'] : '',
"notes" => array_key_exists('notes', $options) ? $options['notes'] : '',
"otp" => array_key_exists('otp', $options) ? $options['otp'] : ''
);
$query="INSERT INTO clients " .
"(id,active,created,secret,email,notes,otp) VALUES " .
"('" . $params["id"] . "', " .
"'" . $params["active"] . "', " .
"'" . $params['created'] . "'," .
"'" . $params['secret'] . "'," .
"'" . $params['email'] . "'," .
"'" . $params['notes'] . "'," .
"'" . $params['otp'] . "')";
if(!$db->customQuery($query)){
$myLog->log(LOG_ERR, "Failed to insert new client with query " . $query);
error_log("Failed to insert new client with query " . $query);
exit(1);
}
echo $client_id.",".$secret."\n";
}
fclose($fh);
$myLog->log(LOG_NOTICE, "Successfully inserted generated clients into database");