1 files changed, 36 insertions, 0 deletions
diff --git a/include/pso/PSOV3Encryption.h b/include/pso/PSOV3Encryption.h
new file mode 100644
index 0000000..5def0f2
--- /dev/null
+++ b/include/pso/PSOV3Encryption.h
@@ -0,0 +1,36 @@
+#ifndef PSOV3ENCRYPTION_H
+#define PSOV3ENCRYPTION_H
+
+#include <global_types.h>
+#include <string.h>
+#include <pso/macros.h>
+#include <pso/TArray.h>
+
+class PSOEncryption {
+public:
+	PSOEncryption();
+	virtual void update_stream() = 0;
+	virtual ~PSOEncryption();
+	virtual void encrypt(u32 seed) = 0;
+	virtual u32 next() = 0;
+};
+
+class PSOV3Encryption : public PSOEncryption {
+public:
+	PSOV3Encryption();
+	virtual void update_stream() override;
+	virtual ~PSOV3Encryption();
+	virtual void encrypt(u32 seed) override;
+	virtual u32 next() override;
+
+	PRIVATE_MEMBER_ACCESSORS_ARRAY(u32, buffer, 522);
+	PRIVATE_MEMBER_ACCESSORS(u32 *, buffer_start);
+	PRIVATE_MEMBER_ACCESSORS(u32 *, buffer_end);
+
+private:
+	TArray<u32, 522> m_buffer;
+	u32 *m_buffer_start;
+	u32 *m_buffer_end;
+};
+
+#endif