From 542d2f5e32e0c0d5bcf40a82cf9ecdf211fe825b Mon Sep 17 00:00:00 2001
From: Mirek Kratochvil <exa.exa@gmail.com>
Date: Sat, 29 Dec 2012 19:40:14 +0100
Subject: [PATCH] rc4-based random number generator with seeding

---
 src/generator.cpp | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 src/generator.h   | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 src/generator.cpp
 create mode 100644 src/generator.h

diff --git a/src/generator.cpp b/src/generator.cpp
new file mode 100644
index 0000000..c694916
--- /dev/null
+++ b/src/generator.cpp
@@ -0,0 +1,47 @@
+
+/*
+ * This file is part of Codecrypt.
+ *
+ * Codecrypt is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Codecrypt is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
+ * License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Codecrypt. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "generator.h"
+
+#include <fstream>
+#include <vector>
+
+using namespace std;
+
+static inline uint bytes (uint bits)
+{
+	return (bits >> 3) + ( (bits & 7) ? 1 : 0);
+}
+
+void arcfour_rng::seed (uint bits, bool quick)
+{
+	vector<byte> s;
+	ifstream f;
+
+	uint b = bytes (bits);
+	if (b > 256) b = 256;
+
+	f.open (quick ? "/dev/urandom" : "/dev/random", ios::in | ios::binary);
+	s.resize (b);
+	for (uint i = 0; i < b; ++i) f >> s[i];
+	f.close();
+
+	r.load_key (s);
+	r.discard (256);
+}
+
diff --git a/src/generator.h b/src/generator.h
new file mode 100644
index 0000000..a42b6bf
--- /dev/null
+++ b/src/generator.h
@@ -0,0 +1,46 @@
+
+/*
+ * This file is part of Codecrypt.
+ *
+ * Codecrypt is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Codecrypt is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
+ * License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Codecrypt. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef _generator_h_
+#define _generator_h_
+
+#include "arcfour.h"
+#include "prng.h"
+
+class arcfour_rng : public prng
+{
+public:
+	arcfour<byte> r;
+
+	arcfour_rng() {
+		r.init (8);
+	}
+
+	~arcfour_rng() {
+		r.clear();
+	}
+
+	void seed (uint bits, bool quick);
+
+	uint random (uint n) {
+		//rand_max is 2^32.
+		return ( (r.gen() << 24) | (r.gen() << 16) | (r.gen() << 8) | r.gen() ) % n;
+	}
+};
+
+#endif