137 lines
3.2 KiB
PHP
Executable File
137 lines
3.2 KiB
PHP
Executable File
#!/usr/bin/env php
|
|
<?php
|
|
/*
|
|
* Functional test for TFTPServer
|
|
*
|
|
* Copyright (c) 2011 <mattias.wadman@gmail.com>
|
|
*
|
|
* MIT License:
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*
|
|
*/
|
|
require_once("lib/tftp.php");
|
|
|
|
class TestTFTPServer extends TFTPServer
|
|
{
|
|
private $_files = array();
|
|
private $_debug;
|
|
|
|
function __construct($server_url, $logger = NULL, $debug = false)
|
|
{
|
|
parent::__construct($server_url, $logger);
|
|
$this->_debug = $debug;
|
|
$this->max_put_size = 60000000;
|
|
}
|
|
|
|
private function log($peer, $level, $message)
|
|
{
|
|
echo
|
|
date("H:i:s") . " " .
|
|
$level . " " .
|
|
$peer . " " .
|
|
$message . "\n";
|
|
}
|
|
|
|
public function log_debug($peer, $message)
|
|
{
|
|
if(!$this->_debug)
|
|
return;
|
|
|
|
$this->log($peer, "D", $message);
|
|
}
|
|
|
|
public function log_info($peer, $message)
|
|
{
|
|
$this->log($peer, "I", $message);
|
|
}
|
|
|
|
public function log_warning($peer, $message)
|
|
{
|
|
$this->log($peer, "W", $message);
|
|
}
|
|
|
|
public function log_error($peer, $message)
|
|
{
|
|
$this->log($peer, "E", $message);
|
|
}
|
|
|
|
public function exists($peer, $filename)
|
|
{
|
|
if($filename == "not_writable" || $filename == "not_readable")
|
|
return true;
|
|
|
|
return isset($this->_files[$filename]);
|
|
}
|
|
|
|
public function readable($peer, $filename)
|
|
{
|
|
if($filename == "not_readable")
|
|
return false;
|
|
|
|
return isset($this->_files[$filename]);
|
|
}
|
|
|
|
public function get($peer, $filename, $mode)
|
|
{
|
|
if(isset($this->_files[$filename]))
|
|
return $this->_files[$filename];
|
|
else
|
|
return false;
|
|
}
|
|
|
|
public function writable($peer, $filename)
|
|
{
|
|
if($filename == "not_writable")
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
public function put($peer, $filename, $mode, $content)
|
|
{
|
|
$this->_files[$filename] = $content;
|
|
}
|
|
}
|
|
|
|
$host = "127.0.0.1";
|
|
$port = 1196;
|
|
$url = "udp://$host:$port";
|
|
|
|
if(count($_SERVER["argv"]) > 1) {
|
|
$server = new TestTFTPServer($url, true);
|
|
if(!$server->loop($error))
|
|
die("$error\n");
|
|
} else {
|
|
|
|
$pid = pcntl_fork();
|
|
$logger = new Logger_Stdout('LOG_DEBUG');
|
|
if($pid == 0) {
|
|
$server = new TestTFTPServer($url, $logger);
|
|
if(!$server->loop($error))
|
|
die("$error\n");
|
|
exit(0);
|
|
}
|
|
usleep(100000);
|
|
// kill server
|
|
posix_kill($pid, SIGINT);
|
|
}
|
|
|
|
?>
|
|
|