<?php
// error_reporting(0);
session_start();
ob_start();
require_once 'tcp_mcrypt.php';
require_once 'FirePHPCore/fb.php';
require_once 'tcp_mcrypt.php';
$PORT;
$HOST;
$AES_KEY;
$SocketOnline = true;
FB::info("Started Execution...");
if (isset($_POST['set_host'])) {
    $_SESSION['HOST'] = $_POST['hostname'];
    $_SESSION['PORT'] = $_POST['port'];
    $HOST = $_POST['host'];
    $PORT = $_POST['port'];
} else {
    $HOST = (isset($_SESSION['HOST'])) ? $_SESSION['HOST'] : "localhost";
    $PORT = (isset($_SESSION['PORT'])) ? $_SESSION['PORT'] : 9876;
}

if (isset($_POST['set_aes_key'])) {
    $_SESSION['AES_KEY'] = $_POST['aes_key'];
    $AES_KEY = $_POST['aes_key'];
} else {
    $AES_KEY = (isset($_SESSION['AES_KEY'])) ? $_SESSION['AES_KEY'] : "123456789";
}

if (isset($_POST['write']) || isset($_POST['write_aes'])) {
// Create socket
    FB::info("Creating Socket");
    $errno = "";
    $errstr = "";
    $fp = stream_socket_client("tcp://{$HOST}:{$PORT}", $errno, $errstr, 30);
    if (!$fp) {
        echo "<h1>Could not create socket! \"{$errstr}\"</h1>";
        FB::error("Socket failed");
    }
    FB::info("Connecting socket");

    if (isset($_POST['write'])) {
        FB::info("Sending via plaintext");
        $text = $_POST['message_plaintext'];
//        socket_write($socket, base64_encode($text) . "\n", strlen($text) + 1);
        fwrite($fp, $text."\n");
        fflush($fp);
    }

    if (isset($_POST['write_aes'])) {
        FB::info("Sending via aes encryption");
        $text = $_POST['message_aes'];
        $text = Encryption::pkcs5_pad($text, 8);
        $encrypted = Encryption::encrypt($text, $AES_KEY);
        echo "Encrypted (base64): ".base64_encode($encrypted);
        fwrite($fp, base64_encode($encrypted)."\n");
        fflush($fp);
    }
}
FB::warn("Reached end of header block")
?>
<!DOCTYPE html>
<html>
    <body>
        <p>
        <h4>Credentials</h4>
        <?php
        echo "<ul>";
        echo "<li><b>Hostname: </b> {$HOST}</li>";
        echo "<li><b>Port: </b> {$PORT}</li>";
        echo "<li><b>AES Encryption Key:</b> {$AES_KEY}</li>";
        echo "</ul>";
        ?>
    </p>
    <form action="tcp.php" method="post">
        <input type="text" name="aes_key" placeholder="Set AES Key"/>
        <input type="submit" name="set_aes_key" value="Set AES Key"/>
    </form>
    <form action="tcp.php" method="post">
        <input type="text" name="message_plaintext" placeholder="Message"/>
        <input type="submit" name="write" value="Write to socket (Plaintext)"/>
    </form>
    <br>
    <form action="tcp.php" method="post">
        <input type="text" name="message_aes" placeholder="Message to encrypt"/>
        <input type="submit" name="write_aes" value="Write to socket (AES)"/>
    </form>
    <p>
        <?php
        FB::info("Began reading from socket");
        if (isset($_POST['write']) || isset($_POST['write_aes']))
            echo fread($fp, 500);
        FB::info("Finished reading from socket!");
        ?>
    </p>
</body>
</html>