#include <global_types.h>
#include <string.h>
#include "pso/macros.h"
#include "pso/PSOV3Encryption.h"
#include "pso/TArray.h"

u32 PSOV3Encryption::next() {
	if (++m_buffer_start == m_buffer_end) {
		update_stream();
	}
	return *m_buffer_start;
}

void PSOV3Encryption::update_stream() {
	u32 *ptr;
	u32 *first_end;
	u32 *start;

	start = m_buffer.start();
	m_buffer_start = m_buffer.start();

	for (ptr = m_buffer.start(), first_end = &ptr[489]; first_end != m_buffer_end;) {
		*ptr++ ^= *first_end++;
	}

	for (u32 *p = ptr; p != m_buffer_end; *p++ ^= *start++);
}

void PSOV3Encryption::encrypt(u32 seed) {
	const size_t size = m_buffer.size();
	u32 thing;
	m_buffer_end = &m_buffer[size];
	m_buffer_start = m_buffer.start();
	u32 value = 0;
	for (int i = 0; i <= 16; ++i, *m_buffer_start++ = value) {
		for (int j = 32; j; --j) {
			seed *= 0x5d588b65;
			value = (++seed & (1 << 31)) | (value >> 1);
		}
	}

	//u32 *smth = &m_buffer[0];
	//u32 *smth2 = &m_buffer[1];
	--m_buffer_start;
	thing = m_buffer[0xf];
	*m_buffer_start = (m_buffer[0] >> 9) ^ (*m_buffer_start << 23) ^ thing;
	//u32 *buf_val = &m_buffer[0];
	//u32 *next_buf_val = &m_buffer[1];

	for (u32 *buf_val = &m_buffer[0], *next_buf_val = &m_buffer[1], *buf = m_buffer_start++; m_buffer_start != m_buffer_end;) {
		*m_buffer_start++ = (*buf_val++ << 23) ^ (*next_buf_val++ >> 9) ^ *buf++;
	}

	update_stream();
	update_stream();
	update_stream();
	m_buffer_start = m_buffer.end();
}

PSOV3Encryption::~PSOV3Encryption() {
	m_buffer.fill(0);
	m_buffer_start = nullptr;
	m_buffer_end = nullptr;
}

PSOV3Encryption::PSOV3Encryption() : PSOEncryption() {
	encrypt(0);
}


PSOEncryption::~PSOEncryption() {

}

PSOEncryption::PSOEncryption() {

}