#!/usr/bin/php
<?php

include __DIR__ . '/func.inc.phps';
include __DIR__ . '/kkk_score.inc.phps';
include __DIR__ . '/kkk3_score.inc.phps';
include __DIR__ . '/wiki_score.inc.phps';

# ---

define('SCORE_OPTIMIZE', 2); // 0=kkk, 1=kkk3, 2=markov, 3=sinkov, -1 = 1 score systems need to be greater, -2 = 2 score systems need to be greater, etc.
define('SAVESTATE_FILE', 's19_markov');
define('HILLCLIMB_ENABLED', true);
define('HILLCLIMB_MAX_LOOKAHEAD', 20);
define('HILLCLIMB_MAX_LOOKAHEAD_RANDOM', false);
define('HILLCLIMB_MAX_LOOKAHEAD_MIN', 40);
define('HILLCLIMB_MAX_LOOKAHEAD_MAX', 60);
define('RANDOM_SWAP_SCHRITTE', 1);
define('ONLY_SAVE_GREATERTHAN_MAX', true);
define('START_RANDOM', true); // Wenn false: Starte mit Englisch-Normalverteilung
define('RANDOM_RESET_EVERY', -1); // Zurueck zu Random key; -1 nein
define('RANDOM_RESET_AFTER_X_FAILS', -1); // Zurueck zu Random key; -1 nein
define('RANDOM_RESET_AFTER_X_FAILS_PLUS_STUFF', 100);
define('EN_DEFAULT_RESET_EVERY', 100); // Zurueck zu Englisch-Normalverteilung; -1 nein
define('EN_DEFAULT_RESET_AFTER_X_FAILS', -1); // Zurueck zu Englisch-Normalverteilung; -1 nein
define('EN_DEFAULT_RESET_AFTER_X_FAILS_PLUS_STUFF', 100);
define('MARKOV_N', 2);
define('SINKOV_N', 4);
define('OPTIMIZE_COLUMN', -1); // -1=withLowestScore, -2=random

# ---

$c1 = 'XGQIP RGWZY PUFWO MNBMW XBQTF MRGWG EWMXR MYIVE RPTMP IXYFK WRPWS RFIPW MXEJF PSBPF XREYF XTWZI';
$key_c1 = 'BEFGIJKMNOPQRSTUVWXYZ';

$c2 = 'ZWWLJ WLLAY TKKAO JWJUZ YYUJJ QWLLA AQJTX OKDPU JSAOH KSAQO ZAKJQ LWWVU ZLQHK IZOWS EWTSB WZIKA';
$key_c2 = 'ABDEHIJKLOPQSTUVWXYZ';

$c3 = 'YKOLC DHUYL UJRBJ MXXJV JNJVL DDHUY XLCKY NJFUJ EYHUX LYASV BBKMK DDUJJ CKYHB SAAJV JMWBS AJHVY';
$key_c3 = 'ABCDEFHJKLMNORSUVWXY';

$c4 = 'IFLJH IPIQG SLJHA MMKXB TIFLJ MSLSH SCGSH WBQBH MKBJW SELNS SHMMI ANTJM XJBKB AMTJM KMTBL NSK'; // CC
$key_c4 = 'SCGH-ABEFIJKLMNPQTWX'; // ohne GH, da teilbekannt

$c5 = 'UFXFL HLRLJ CRMLG KMRRC FUFXF RRMLD ZLWCO SGXLZ ZXFUX MSOLC DLLRR PLMSO FXQGW TCQSQ SDCQF';
$key_c5 = 'CDFGHJKLMOPQRSTUWXZ';

$chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";

srand((double)microtime()*1000000);

if (file_exists(SAVESTATE_FILE)) {
	load();

        $z1 = do_replace($key_c1, $x1, $c1);
        $z2 = do_replace($key_c2, $x2, $c2);
        $z3 = do_replace($key_c3, $x3, $c3);
	$z4 = do_replace($key_c4, $x4, $c4);
        $z5 = do_replace($key_c5, $x5, $c5);
	$z = mix5($z1, $z2, $z3, $z4, $z5);

	$kkk = p($z, true, true);
	$kkk3 = p3($z, true, true);
	$markov = markov_score($z, 'en', MARKOV_N);
	$sinkov = sinkov_score($z, 'en', SINKOV_N);
	$max_kkk = $kkk;
	$max_kkk3 = $kkk3;
	$max_markov = $markov;
	$max_sinkov = $sinkov;

	if (HILLCLIMB_ENABLED) {
		$hillclimb_lookahead_counts = 0;
		$hillclimb_last_bak_key = array($x1, $x2, $x3, $x4, $x5);
	}
	echo "Loaded savestate (kkk: $kkk, kkk3: $kkk3, markov: $markov, sinkov".SINKOV_N.": $sinkov)\n";
} else {
	if (START_RANDOM) {
	        $x1 = randomkey(1);
		$x2 = randomkey(2);
	        $x3 = randomkey(3);
		$x4 = randomkey(4);
	        $x5 = randomkey(5);
	} else {
		// Mapping an Eng-Wahrscheinlichkeitstabelle
		$x1 = 'LDORSYVNPBTMAUCFGEIHW';
		$x2 = 'TYPFMUAOISGHDLRWEBCN';
		$x3 = 'LHMRGBOEINUWPYCASFDT';
		$x4 = 'EXTN-DOPFSAHNELYMRUW'; // TN geraten
		$x5 = 'OLTCFPGESDYHANWRMIU';
	}
	$max_kkk = 0;
	$max_kkk3 = 0;
	$max_markov = 0;
	$max_sinkov = 0;
	if (HILLCLIMB_ENABLED) {
		$hillclimb_lookahead_counts = 0;
		$hillclimb_last_bak_key = array($x1, $x2, $x3, $x4, $x5);
	}
}

$z1 = do_replace($key_c1, $x1, $c1);
$z2 = do_replace($key_c2, $x2, $c2);
$z3 = do_replace($key_c3, $x3, $c3);
$z4 = do_replace($key_c4, $x4, $c4);
$z5 = do_replace($key_c5, $x5, $c5);
$z = mix5($z1, $z2, $z3, $z4, $z5);

$hillclimb_max_lookahead = HILLCLIMB_MAX_LOOKAHEAD;

$found_stuff = 0;

$round = 0;
$fail_counter = 0;
while (true) {
	$round++;
	$fail_counter++;

	echo "$round ($fail_counter)                \r";

	if (RANDOM_RESET_AFTER_X_FAILS_PLUS_STUFF != -1) {
		$stuff1 = 0;
	} else {
		$stuff1 = $found_stuff * RANDOM_RESET_AFTER_X_FAILS_PLUS_STUFF;
	}
	if (EN_DEFAULT_RESET_AFTER_X_FAILS_PLUS_STUFF != -1) {
		$stuff2 = 0;
	} else {
		$stuff2 = $found_stuff * EN_DEFAULT_RESET_AFTER_X_FAILS_PLUS_STUFF;
	}

	if (((RANDOM_RESET_EVERY != -1) && ($round % RANDOM_RESET_EVERY == 0)) ||
	    ((RANDOM_RESET_AFTER_X_FAILS != -1) && ($fail_counter % (RANDOM_RESET_AFTER_X_FAILS+$stuff1) == 0))) {
	        $x1 = randomkey(1);
		$x2 = randomkey(2);
	        $x3 = randomkey(3);
		$x4 = randomkey(4);
	        $x5 = randomkey(5);
		if (HILLCLIMB_ENABLED) {
			$hillclimb_lookahead_counts = 0;
			$hillclimb_last_bak_key = array($x1, $x2, $x3, $x4, $x5);
			if (HILLCLIMB_MAX_LOOKAHEAD_RANDOM) {
				$hillclimb_max_lookahead = rand(HILLCLIMB_MAX_LOOKAHEAD_MIN,HILLCLIMB_MAX_LOOKAHEAD_MAX);
			}
		}
		$found_stuff = 0;
		$fail_counter = 0;
	} else if (((EN_DEFAULT_RESET_EVERY != -1) && ($round % EN_DEFAULT_RESET_EVERY == 0)) ||
	         ((EN_DEFAULT_RESET_AFTER_X_FAILS != -1) && ($fail_counter % (EN_DEFAULT_RESET_AFTER_X_FAILS+$stuff2) == 0))) {
		// Mapping an Eng-Wahrscheinlichkeitstabelle
		$x1 = 'LDORSYVNPBTMAUCFGEIHW';
		$x2 = 'TYPFMUAOISGHDLRWEBCN';
		$x3 = 'LHMRGBOEINUWPYCASFDT';
		$x4 = 'EXTN-DOPFSAHNELYMRUW'; // TN geraten
		$x5 = 'OLTCFPGESDYHANWRMIU';
		if (HILLCLIMB_ENABLED) {
			$hillclimb_lookahead_counts = 0;
			$hillclimb_last_bak_key = array($x1, $x2, $x3, $x4, $x5);
			if (HILLCLIMB_MAX_LOOKAHEAD_RANDOM) {
				$hillclimb_max_lookahead = rand(HILLCLIMB_MAX_LOOKAHEAD_MIN,HILLCLIMB_MAX_LOOKAHEAD_MAX);
			}
		}
		$found_stuff = 0;
		$fail_counter = 0;
	} else {
		// Find out which column to swap: Choose the one with the lowest Sinkov score

		if (OPTIMIZE_COLUMN >= 0) {
			// Specific column

			$column_to_swap = OPTIMIZE_COLUMN;

		} else if (OPTIMIZE_COLUMN == -1) {
			// With lowest Sinkov (TODO: lowest score instead?)

			$z1_sin = sinkov_score($z1, 'en', SINKOV_N);
			$z2_sin = sinkov_score($z2, 'en', SINKOV_N);
			$z3_sin = sinkov_score($z3, 'en', SINKOV_N);
			$z4_sin = sinkov_score($z4, 'en', SINKOV_N);
			$z5_sin = sinkov_score($z5, 'en', SINKOV_N);

			$min = $z1_sin;
			if ($z2_sin < $min) $min = $z2_sin;
			if ($z3_sin < $min) $min = $z3_sin;
			if ($z4_sin < $min) $min = $z4_sin;
			if ($z5_sin < $min) $min = $z5_sin;

			if ($min == $z1_sin) $column_to_swap = 1;
			if ($min == $z2_sin) $column_to_swap = 2;
			if ($min == $z3_sin) $column_to_swap = 3;
			if ($min == $z4_sin) $column_to_swap = 4;
			if ($min == $z5_sin) $column_to_swap = 5;

		} else if (OPTIMIZE_COLUMN == -2) {
			// Random

			$column_to_swap = rand(1,5);
		}

		if ($column_to_swap == 1) $x1 = randomswap($x1, RANDOM_SWAP_SCHRITTE);
		if ($column_to_swap == 2) $x2 = randomswap($x2, RANDOM_SWAP_SCHRITTE);
		if ($column_to_swap == 3) $x3 = randomswap($x3, RANDOM_SWAP_SCHRITTE);
		if ($column_to_swap == 4) $x4 = randomswap($x4, RANDOM_SWAP_SCHRITTE, true);
		if ($column_to_swap == 5) $x5 = randomswap($x5, RANDOM_SWAP_SCHRITTE);
	}

	if (isset($column_to_swap)) {
		$x = "K1=$x1; K2=$x2; K3=$x3; K4=$x4; K5=$x5; Swap: $column_to_swap";
	} else {
		$x = "K1=$x1; K2=$x2; K3=$x3; K4=$x4; K5=$x5";
	}

	// ---

	// $c? = original content of column
	// $x? = key of column
	// $z? = decrypted

        $z1 = do_replace($key_c1, $x1, $c1);
        $z2 = do_replace($key_c2, $x2, $c2);
        $z3 = do_replace($key_c3, $x3, $c3);
	$z4 = do_replace($key_c4, $x4, $c4);
        $z5 = do_replace($key_c5, $x5, $c5);
	$z = mix5($z1, $z2, $z3, $z4, $z5);

	$kkk = p($z, true, true);
	$kkk3 = p3($z, true, true);
	$markov = markov_score($z, 'en', MARKOV_N);
	$sinkov = sinkov_score($z, 'en', SINKOV_N);

	if (SCORE_OPTIMIZE == 0) {
		$new_max = ONLY_SAVE_GREATERTHAN_MAX ? $kkk > $max_kkk : $kkk >= $max_kkk;
	}
	else if (SCORE_OPTIMIZE == 1) {
		$new_max = ONLY_SAVE_GREATERTHAN_MAX ? $kkk3 > $max_kkk3 : $kkk3 >= $max_kkk3;
	}
	else if (SCORE_OPTIMIZE == 2) {
		$new_max = ONLY_SAVE_GREATERTHAN_MAX ? $markov > $max_markov : $markov >= $max_markov;
	}
	else if (SCORE_OPTIMIZE == 3) {
		$new_max = ONLY_SAVE_GREATERTHAN_MAX ? $sinkov > $max_sinkov : $sinkov >= $max_sinkov;
	} else {
		$tmp = 0;
		if (ONLY_SAVE_GREATERTHAN_MAX) {
			if ($kkk > $max_kkk) $tmp++;
			if ($kkk3 > $max_kkk3) $tmp++;
			if ($markov > $max_markov) $tmp++;
			if ($sinkov > $max_sinkov) $tmp++;
		} else {
			if ($kkk >= $max_kkk) $tmp++;
			if ($kkk3 >= $max_kkk3) $tmp++;
			if ($markov >= $max_markov) $tmp++;
			if ($sinkov >= $max_sinkov) $tmp++;
		}
		$new_max = $tmp >= -SCORE_OPTIMIZE;
	}

	if (HILLCLIMB_ENABLED) {
		if ($new_max) {
			$max_kkk = $kkk;
			$max_kkk3 = $kkk3;
			$max_markov = $markov;
			$max_sinkov = $sinkov;

			$hillclimb_last_bak_key = array($x1, $x2, $x3, $x4, $x5);
			$z = strtolower($z);
			echo date('d.m.Y H:i:s').' - ';
			echo "($x) $z\n";
			echo "\t--> (kkk: $kkk, kkk3: $kkk3, markov: $markov, sinkov".SINKOV_N.": $sinkov)\n";
			$found_stuff++;
			$fail_counter = 0;
			$hillclimb_lookahead_counts = 0; # Vielleicht haben wir eine Glückssträne; weitermachen
			save();
		} else {
			$hillclimb_lookahead_counts++;
			if ($hillclimb_lookahead_counts > $hillclimb_max_lookahead) {
				// Recover
				list($x1, $x2, $x3, $x4, $x5) = $hillclimb_last_bak_key;
				$hillclimb_lookahead_counts = 0;
			}
		}
	} else {
		if ($new_max) {
			$max_kkk = $kkk;
			$max_kkk3 = $kkk3;
			$max_markov = $markov;
			$max_sinkov = $sinkov;

			$z = strtolower($z);
			echo date('d.m.Y H:i:s').' - ';
			echo "($x) $z\n";
			echo "\t--> (kkk: $kkk, kkk3: $kkk3, markov: $markov, sinkov".SINKOV_N.": $sinkov)\n";
			$found_stuff++;
			$fail_counter = 0;
			save();
		}
	}

	// ---

}

# ---

function save() {
	global $x1, $x2, $x3, $x4, $x5;
	file_put_contents(SAVESTATE_FILE, "$x1\n$x2\n$x3\n$x4\n$x5\n");
}

function load() {
	global $x1, $x2, $x3, $x4, $x5;
	$sav = file(SAVESTATE_FILE);
	$x1 = trim($sav[0]);
	$x2 = trim($sav[1]);
	$x3 = trim($sav[2]);
	$x4 = trim($sav[3]);
	$x5 = trim($sav[4]);
}

function change_keybyte($col, $char) {
	$chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
	$var = 'x'.$col;
	global $$var;

	if ($col == 4) {
		if (strpos($$var, $char) < 4) return $char;
	}

	do {
		$tmp = $chars[rand(0,strlen($chars)-1)];
	} while (strpos($$var, $tmp) !== false);
	$$var = str_replace($char, $tmp, $$var);

	return $tmp;
}