#!/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', 3); // 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', 'savestate_sinkov4');
define('HILLCLIMB_ENABLED', true);
define('HILLCLIMB_MAX_LOOKAHEAD', 10000);
define('RANDOM_SWAP_SCHRITTE', 1);
define('ONLY_SAVE_GREATERTHAN_MAX', true);
define('ONLY_SWAP_COLUMN_NO', -1); // -1 : swap all columns
define('RANDOM_SWAP_AFTER', 50000000);
define('SWAP_COLUMN_EVERY_XTH_ROUND', 100);
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('EN_DEFAULT_RESET_EVERY', -1); // Zurueck zu Englisch-Normalverteilung; -1 nein
define('EN_DEFAULT_RESET_AFTER_X_FAILS', -1); // Zurueck zu Englisch-Normalverteilung; -1 nein
define('MARKOV_N', 2);
define('SINKOV_N', 4);
define('SINGLE_OPTIMIZE_THRESHOLD', 1); // -1 : no optimization
define('SINGLE_OPTIMIZE_MIN_SIN_CNT', 1);
define('SINGLE_OPTIMIZE_MIN_SIN_CNT_PER', 15000);

# ---

$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);

$round = 0;
$fail_counter = 0;
#$column_to_swap = (ONLY_SWAP_COLUMN_NO == -1) ? rand(1,5) : ONLY_SWAP_COLUMN_NO;
while (true) {
	$round++;
	$fail_counter++;

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


	if (((RANDOM_RESET_EVERY != -1) && ($round % RANDOM_RESET_EVERY == 0)) ||
	    ((RANDOM_RESET_AFTER_X_FAILS != -1) && ($fail_counter % RANDOM_RESET_AFTER_X_FAILS == 0))) {
#		echo " *** NEW RANDOM KEY ***\n";
	        $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);
		}
		$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 == 0))) {
		echo " *** NEW EN MAPPING ***\n";

		// 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);
		}
		$fail_counter = 0;
	} else {
		$sort_ary = array();

		for ($i=0; $i<=strlen($z)-SINKOV_N; $i++) {
			$chunk = substr($z,$i,SINKOV_N);

			$sin = sinkov_score($chunk, 'en', SINKOV_N);
			$sin_key = str_pad($sin * 100000, 20, "0", STR_PAD_LEFT);

			if (!isset($sort_ary[$sin_key])) $sort_ary[$sin_key] = array();
			$sort_ary[$sin_key][] = $i;
		}

		// Search for a random quadgram with the lowest score
		ksort($sort_ary);
		$sincnt = 1 + SINGLE_OPTIMIZE_MIN_SIN_CNT * floor($fail_counter / SINGLE_OPTIMIZE_MIN_SIN_CNT_PER);
		$chosen_sin_idx = rand(0,min(count($sort_ary),$sincnt)-1);
		$chosen_sin = array_keys($sort_ary)[$chosen_sin_idx];
		$chosen_idx = rand(0,count($sort_ary[$chosen_sin])-1);
		$chosen_i   = $sort_ary[$chosen_sin][$chosen_idx];

		$chunk = substr($z,$chosen_i,SINKOV_N);

		$col_1 = floor(($chosen_i+0) / 5) % 5 + 1;
		$col_2 = floor(($chosen_i+1) / 5) % 5 + 1;
		$col_3 = floor(($chosen_i+2) / 5) % 5 + 1;
		$col_4 = floor(($chosen_i+3) / 5) % 5 + 1;

		$test_counter = 0;

		do {
			$test_counter++;

			$sin = sinkov_score($chunk, 'en', SINKOV_N) * 100000;

			$chunk2 = $chunk;
			$byte_to_change = rand(0,SINKOV_N-1);
			if ($byte_to_change == 0) $chunk2[0] = change_keybyte($col_1, $chunk[0]);
			if ($byte_to_change == 1) $chunk2[1] = change_keybyte($col_2, $chunk[1]);
			if ($byte_to_change == 2) $chunk2[2] = change_keybyte($col_3, $chunk[2]);
			if ($byte_to_change == 3) $chunk2[3] = change_keybyte($col_4, $chunk[3]);

			$sin2 = sinkov_score($chunk2, 'en', SINKOV_N) * 100000;
		} while (($sin2 < $sin) && ($test_counter < SINGLE_OPTIMIZE_THRESHOLD));
	}

	if ($fail_counter >= RANDOM_SWAP_AFTER) {
		echo "#";
		if ($round % SWAP_COLUMN_EVERY_XTH_ROUND == 0) {
			$column_to_swap = (ONLY_SWAP_COLUMN_NO == -1) ? rand(1,5) : ONLY_SWAP_COLUMN_NO;
#			echo "$round : $column_to_swap\n";
		}
		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);
	} else {
		echo "*";
	}

	$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";
			$fail_counter = 0;
			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";
			$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;
}