Import util.random from Prosody (using SHA-1)

Tue, 25 Aug 2015 16:01:52 +0200

author
Kim Alvefur <zash@zash.se>
date
Tue, 25 Aug 2015 16:01:52 +0200
changeset 388
d963c8a5d89c
parent 387
508bf163502b
child 389
bf3a4fcdcb76

Import util.random from Prosody (using SHA-1)

squishy file | annotate | diff | comparison | revisions
util/random.lua file | annotate | diff | comparison | revisions
--- a/squishy	Wed Aug 19 16:41:47 2015 +0200
+++ b/squishy	Tue Aug 25 16:01:52 2015 +0200
@@ -31,6 +31,7 @@
 Module "util.json"		"util/json.lua"
 Module "util.xml"		"util/xml.lua"
 Module "util.rsm"       "util/rsm.lua"
+Module "util.random"       "util/random.lua"
 
 Module "util.sasl.scram" "util/sasl/scram.lua"
 Module "util.sasl.plain" "util/sasl/plain.lua"
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/util/random.lua	Tue Aug 25 16:01:52 2015 +0200
@@ -0,0 +1,43 @@
+-- Prosody IM
+-- Copyright (C) 2008-2014 Matthew Wild
+-- Copyright (C) 2008-2014 Waqas Hussain
+--
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
+local tostring = tostring;
+local os_time = os.time;
+local os_clock = os.clock;
+local ceil = math.ceil;
+local H = require "util.hashes".sha1;
+
+local last_uniq_time = 0;
+local function uniq_time()
+	local new_uniq_time = os_time();
+	if last_uniq_time >= new_uniq_time then new_uniq_time = last_uniq_time + 1; end
+	last_uniq_time = new_uniq_time;
+	return new_uniq_time;
+end
+
+local function new_random(x)
+	return H(x..os_clock()..tostring({}));
+end
+
+local buffer = new_random(uniq_time());
+
+local function seed(x)
+	buffer = new_random(buffer..x);
+end
+
+local function bytes(n)
+	if #buffer < n+4 then seed(uniq_time()); end
+	local r = buffer:sub(1, n);
+	buffer = buffer:sub(n+1);
+	return r;
+end
+
+return {
+	seed = seed;
+	bytes = bytes;
+};

mercurial