the Restructuring

4 files changed, 567 insertions, 0 deletions

diff --git a/xmpp/base64.lua b/xmpp/base64.lua
new file mode 100644
index 0000000..4f9a967
--- /dev/null
+++ b/xmpp/base64.lua
@@ -0,0 +1,38 @@
+local alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
+
+-- really bad
+local function encode(s)
+	local x = 0
+	local n = 0
+	local out = {}
+	local outn = 1
+	local pad = ""
+	local function si(v, m) x = (x << m) | v; n = n + m end
+	local function so(m)
+		local rem = n - m
+		local res = x >> rem
+		x = x & ((1 << rem) - 1)
+		n = n - m
+		return res
+	end
+	local function o()
+		while n >= 6 do
+			local i = so(6)+1
+			out[outn] = alphabet:sub(i,i)
+			outn = outn + 1
+		end
+	end
+	for i = 1, #s do
+		si(s:byte(i), 8)
+		o()
+	end
+	if n > 0 then
+		local on = n
+		si(0, 6-n)
+		o()
+		pad = ('='):rep(3-on/2) -- bad
+	end
+	return table.concat(out)..pad
+end
+
+return {encode=encode}
diff --git a/xmpp/pylon.lua b/xmpp/pylon.lua
new file mode 100644
index 0000000..ba9c252
--- /dev/null
+++ b/xmpp/pylon.lua
@@ -0,0 +1,187 @@
+local cqueues = require'cqueues'
+local cqaux = require'cqueues.auxlib'
+local socket = require'cqueues.socket'
+local xml = require'xmpp.xml'
+local X = xml.X
+local xmlify = xml.xmlify
+local pprint=require'pprint'
+local Queue = require'util.queue'
+local base64 = require'xmpp.base64'
+local sha1 = require'xmpp.sha1'
+local Channel = require'util.channel'
+
+local Xmpp = {}
+
+local function make_auth(authz, authn, password)
+	-- sasl plain (RFC4616)
+	return base64.encode(authz..'\0'..authn..'\0'..password)
+end
+
+function Xmpp.make(wilson, conf)
+	local self = {
+		name = conf.name,
+		wilson = wilson,
+		inbox = Queue.make(),
+		cq = wilson.cq, -- todo
+		nicks_inuse = {}, -- todo
+	}
+	local function conf_var(name)
+		assert(conf[name] ~= nil, 'missing conf field '..name)
+		self[name] = conf[name]
+	end
+	-- conf_var 'jid'
+	conf_var 'server'
+	-- conf_var 'resource'
+	conf_var 'component'
+	conf_var 'component_secret'
+
+	setmetatable(self, {__index=Xmpp})
+	return self
+end
+
+function Xmpp._connect_c2s(self)
+	local sock = assert(socket.connect(self.server, 5222))
+	self.sock = sock
+	sock:setmode('bn','bn')
+
+	local start = ([[
+<?xml version='1.0'?><stream:stream from='%s' to='%s' version='1.0' xml:lang='en' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams'>]]):format(self.jid, self.server)
+
+	-- state of the art xml parser
+	local function check_and_send(test, text)
+		local x = sock:read('-2048')
+		assert(x:find(test))
+		if text then sock:write(text) end
+	end
+	local function ietf_urn(v) return 'urn:ietf:params:xml:ns:xmpp-'..v end
+
+	sock:write(start)
+	check_and_send('starttls', xmlify(X.starttls{xmlns=ietf_urn"tls"}))
+	check_and_send('proceed', nil)
+	sock:starttls()
+	sock:write(start)
+	local auth = make_auth('', self.jid:match"(.*)@", self.password)
+	check_and_send('PLAIN',
+		xmlify(X.auth{xmlns=ietf_urn"sasl", mechanism='PLAIN', auth}))
+	check_and_send('success',start)
+	check_and_send('bind',
+		xmlify(X.iq{type='set', id='aaaa',
+			X.bind{xmlns=ietf_urn"bind", X.resource{self.resource}}}))
+	check_and_send('jid',X.presence{X.show{'chat'}})
+
+	return sock
+end
+
+-- this sucks! no tls! no security! only use on local connections!
+function Xmpp._connect_component(self)
+	local sock = assert(socket.connect(self.server, 5347))
+	self.sock = sock
+	sock:setmode('bn','bn')
+	-- yes, our component name goes in the 'to' field. don't ask me why
+	local start = ([[<stream:stream to='%s' xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams'>]]):format(self.component)
+
+	print(start)
+
+	-- state of the art xml parser
+	local function check_and_send(test, text)
+		local x = sock:read('-2048')
+		assert(x:find(test))
+		if text then sock:write(text) end
+	end
+
+	sock:write(start)
+	local streamhead = sock:read('-2048')
+	print('streamhead', streamhead)
+	assert(streamhead:find'accept')
+	local streamid = streamhead:match"id='(.-)'"
+	sock:write(xmlify(X.handshake{sha1.sha1(streamid..self.component_secret)}))
+	check_and_send('<handshake/>',nil)
+
+	return sock
+end
+
+local THE_MUC = 'd@conference.ubq323.website'
+
+function Xmpp.run(self)
+	self:_connect_component()
+	self.cq:wrap(self.recving, self)
+	self.cq:wrap(self.sending, self)
+end
+
+function Xmpp.recving(self)
+	local function getmore()
+		local function t(...)
+			pprint(...)
+			return ...
+		end
+		return t(self.sock:read'-2048')
+	end
+	for x in xml.stanzae(getmore) do
+		pprint(x)
+		print(xmlify(x))
+		local body = x'body' and x'body'[1]
+		if x.label == 'message' then
+			local fr = x.xarg.from
+			local t = x.xarg.to
+			local from_nick = fr:match("/(.*)")
+			if not self.nicks_inuse[from_nick] and body and fr:match"/" and t == 'wilson@'..self.component then
+				self.wilson:deliver(Channel(self, THE_MUC), {
+					body = body,
+					sender = '[x]'..from_nick
+				})
+			end
+		end
+	end
+end
+
+function Xmpp.sending(self)
+	local users_inuse = {}
+	local function ensure_joined(muc,user,nick)
+		if self.nicks_inuse[nick] then return end
+		user = user:gsub("[^a-zA-Z0-9%.]","."):match("^%.*(.-)%.*$")
+		while users_inuse[user] do user = user..'-' end
+		local jid = user..'@'..self.component
+		local mucjid = muc..'/'..nick
+		self.nicks_inuse[nick] = true
+		users_inuse[user] = true
+
+		self.sock:write(xmlify(
+			X.presence{from=jid, to=mucjid,
+				X.x{xmlns='http://jabber.org/protocol/muc',
+					X.history{maxstanzas='0'}}}))
+	end
+	ensure_joined(THE_MUC, 'wilson', 'wilson')
+	for dest_channel, message in self.inbox:iter() do
+		pprint(dest_channel, message)
+		ensure_joined(THE_MUC, message.sender, message.sender)
+		local user = message.sender:gsub("[^a-zA-Z0-9%.]","."):match("^%.*(.-)%.*$")
+		local jid = user..'@'..self.component
+		self.sock:write(xmlify(
+			X.message{to=THE_MUC, type='groupchat', from=jid,
+				X.body{message.body}}))
+	end
+end
+
+function Xmpp.post(self, dest_channel, message)
+	self.inbox:enqueue(dest_channel, message)
+end
+
+return Xmpp
+
+-- local cq = cqueues.new()
+-- local conf = {
+-- 	jid='wilson@ubq323.website',
+-- 	server='ubq323.website',
+-- 	password='gregory<3',
+-- 	resource='cheese',
+-- }
+-- local dummy_network = {
+-- 	post = function(self, pylonname, channel, message)
+-- 		pprint(pylonname, channel, message)
+-- 	end
+-- }
+-- local pylon = Xmpp.makepylon('xmpptest',conf, cq, dummy_network)
+-- pylon:run()
+
+-- pprint('peas', cq:loop())
+
diff --git a/xmpp/sha1.lua b/xmpp/sha1.lua
new file mode 100644
index 0000000..7451595
--- /dev/null
+++ b/xmpp/sha1.lua
@@ -0,0 +1,169 @@
+-- from https://github.com/mpeterv/sha1
+-- by Enrique García Cota, Eike Decker, Jeffrey Friedl, Peter Melnichenko
+-- (MIT license)
+
+local sha1 = {}
+
+-- Merges four bytes into a uint32 number.
+local function bytes_to_uint32(a, b, c, d)
+   return a * 0x1000000 + b * 0x10000 + c * 0x100 + d
+end
+
+-- Splits a uint32 number into four bytes.
+local function uint32_to_bytes(a)
+   local a4 = a % 256
+   a = (a - a4) / 256
+   local a3 = a % 256
+   a = (a - a3) / 256
+   local a2 = a % 256
+   local a1 = (a - a2) / 256
+   return a1, a2, a3, a4
+end
+
+local function uint32_lrot(a, bits)
+   return ((a << bits) & 0xFFFFFFFF) | (a >> (32 - bits))
+end
+
+local function uint32_ternary(a, b, c)
+   -- c ~ (a & (b ~ c)) has less bitwise operations than (a & b) | (~a & c).
+   return c ~ (a & (b ~ c))
+end
+
+local function uint32_majority(a, b, c)
+   -- (a & (b | c)) | (b & c) has less bitwise operations than (a & b) | (a & c) | (b & c).
+   return (a & (b | c)) | (b & c)
+end
+
+local sbyte = string.byte
+local schar = string.char
+local sformat = string.format
+local srep = string.rep
+
+local function hex_to_binary(hex)
+   return (hex:gsub("..", function(hexval)
+      return schar(tonumber(hexval, 16))
+   end))
+end
+
+-- Calculates SHA1 for a string, returns it encoded as 40 hexadecimal digits.
+function sha1.sha1(str)
+   -- Input preprocessing.
+   -- First, append a `1` bit and seven `0` bits.
+   local first_append = schar(0x80)
+
+   -- Next, append some zero bytes to make the length of the final message a multiple of 64.
+   -- Eight more bytes will be added next.
+   local non_zero_message_bytes = #str + 1 + 8
+   local second_append = srep(schar(0), -non_zero_message_bytes % 64)
+
+   -- Finally, append the length of the original message in bits as a 64-bit number.
+   -- Assume that it fits into the lower 32 bits.
+   local third_append = schar(0, 0, 0, 0, uint32_to_bytes(#str * 8))
+
+   str = str .. first_append .. second_append .. third_append
+   assert(#str % 64 == 0)
+
+   -- Initialize hash value.
+   local h0 = 0x67452301
+   local h1 = 0xEFCDAB89
+   local h2 = 0x98BADCFE
+   local h3 = 0x10325476
+   local h4 = 0xC3D2E1F0
+
+   local w = {}
+
+   -- Process the input in successive 64-byte chunks.
+   for chunk_start = 1, #str, 64 do
+      -- Load the chunk into W[0..15] as uint32 numbers.
+      local uint32_start = chunk_start
+
+      for i = 0, 15 do
+         w[i] = bytes_to_uint32(sbyte(str, uint32_start, uint32_start + 3))
+         uint32_start = uint32_start + 4
+      end
+
+      -- Extend the input vector.
+      for i = 16, 79 do
+         w[i] = uint32_lrot(w[i - 3] ~ w[i - 8] ~ w[i - 14] ~ w[i - 16], 1)
+      end
+
+      -- Initialize hash value for this chunk.
+      local a = h0
+      local b = h1
+      local c = h2
+      local d = h3
+      local e = h4
+
+      -- Main loop.
+      for i = 0, 79 do
+         local f
+         local k
+
+         if i <= 19 then
+            f = uint32_ternary(b, c, d)
+            k = 0x5A827999
+         elseif i <= 39 then
+            f = b ~ c ~ d
+            k = 0x6ED9EBA1
+         elseif i <= 59 then
+            f = uint32_majority(b, c, d)
+            k = 0x8F1BBCDC
+         else
+            f = b ~ c ~ d
+            k = 0xCA62C1D6
+         end
+
+         local temp = (uint32_lrot(a, 5) + f + e + k + w[i]) % 4294967296
+         e = d
+         d = c
+         c = uint32_lrot(b, 30)
+         b = a
+         a = temp
+      end
+
+      -- Add this chunk's hash to result so far.
+      h0 = (h0 + a) % 4294967296
+      h1 = (h1 + b) % 4294967296
+      h2 = (h2 + c) % 4294967296
+      h3 = (h3 + d) % 4294967296
+      h4 = (h4 + e) % 4294967296
+   end
+
+   return sformat("%08x%08x%08x%08x%08x", h0, h1, h2, h3, h4)
+end
+
+function sha1.binary(str)
+   return hex_to_binary(sha1.sha1(str))
+end
+
+-- Precalculate replacement tables.
+local xor_with_0x5c = {}
+local xor_with_0x36 = {}
+
+for i = 0, 0xff do
+   xor_with_0x5c[schar(i)] = schar(0x5c ~ i)
+   xor_with_0x36[schar(i)] = schar(0x36 ~ i)
+end
+
+-- 512 bits.
+local BLOCK_SIZE = 64
+
+function sha1.hmac(key, text)
+   if #key > BLOCK_SIZE then
+      key = sha1.binary(key)
+   end
+
+   local key_xord_with_0x36 = key:gsub('.', xor_with_0x36) .. srep(schar(0x36), BLOCK_SIZE - #key)
+   local key_xord_with_0x5c = key:gsub('.', xor_with_0x5c) .. srep(schar(0x5c), BLOCK_SIZE - #key)
+
+   return sha1.sha1(key_xord_with_0x5c .. sha1.binary(key_xord_with_0x36 .. text))
+end
+
+function sha1.hmac_binary(key, text)
+   return hex_to_binary(sha1.hmac(key, text))
+end
+
+setmetatable(sha1, {__call = function(_, str) return sha1.sha1(str) end})
+
+return sha1
+
diff --git a/xmpp/xml.lua b/xmpp/xml.lua
new file mode 100644
index 0000000..68fa741
--- /dev/null
+++ b/xmpp/xml.lua
@@ -0,0 +1,173 @@
+-- originally from http://lua-users.org/wiki/LuaXml
+-- modified by me a bit
+
+
+entity_escapes = {
+		["<"]="&lt;",
+		[">"]="&gt;",
+		["&"]="&amp;",
+		['"']="&quot;",
+		["'"]="&apos;"
+}
+entity_unescapes = {}
+for k,v in pairs(entity_escapes) do entity_unescapes[v]=k end
+
+function escape(s)
+	return s:gsub("[<>&'\"]",entity_escapes)
+end
+function unescape(s)
+	return s:gsub("&[a-z]+;",entity_unescapes)
+end
+
+local function parseargs(s)
+  local arg = {}
+  string.gsub(s, "([%-%w]+)=([\"'])(.-)%2", function (w, _, a)
+    arg[w] = a
+  end)
+  return arg
+end
+
+local function tag(x)
+	return setmetatable(x, {__call=function(x, label)
+		-- search for child with given label
+		for _,c in ipairs(x) do
+			if c.label == label then
+				return c
+			end
+		end
+		return nil
+	end})
+end
+
+local psingle
+
+local function pmulti(s, i, parent)
+	::again::
+	local nexti, child = psingle(s, i)
+	if child.close and child.label == parent.label then
+		return nexti, parent
+	else
+		table.insert(parent, child)
+		i = nexti
+		goto again
+	end
+end
+
+psingle = function(s, i)
+	i = i or 1
+	local ts,j,c,label,xarg,empty = s:find("<(%/?)([%w:]+)(.-)(%/?)>", i)
+	if not ts then
+		local rest = s:sub(i)
+		if rest:find("<",i) then
+			error('ill formed (eof?)')
+		elseif #rest == 0 then
+			error('empty string')
+		else
+			return i+#rest, unescape(rest)
+		end
+	end
+	local nexti = j+1
+
+	local pretext = s:sub(i, ts-1)
+	if not pretext:find("^%s*$") then -- not entirely whitespace
+		return ts, unescape(pretext)
+	end
+
+	if empty == "/" then
+		return nexti, tag{label=label, xarg=parseargs(xarg), empty=true}
+	elseif c == "" then -- start tag
+		return pmulti(s, nexti, tag{label=label, xarg=parseargs(xarg)})
+	else -- end tag
+		return nexti, tag{label=label, close=true}
+	end
+end
+
+local wrap
+do
+	local _, cqa = pcall(require, 'cqueues.auxlib')
+	wrap = cqa and cqa.wrap or coroutine.wrap
+end
+
+local function stanzae(getmore)
+	return wrap(function()
+		local buf = ''
+		while true do
+			local ok, ni, el = pcall(psingle, buf)
+			if ok then
+				coroutine.yield(el)
+				buf = buf:sub(ni)
+			else
+				local more = assert(getmore())
+				buf = buf .. more
+			end
+		end
+	end)
+end
+
+
+local safestr_mt = {name='SAFESTR', __tostring=function(x) return x.s end}
+local function safestr(s) return setmetatable({s=s}, safestr_mt) end
+
+local function _xmlify(x)
+	if getmetatable(x) == safestr_mt then
+		return x
+	elseif type(x) == 'string' then
+		return safestr(escape(x))
+	elseif type(x) == 'table' then -- must be a tag
+		local function argstr(t)
+			local argstring = ''
+			for k,v in pairs(t) do
+				argstring = argstring .. (" %s='%s'"):format(k, _xmlify(v))
+			end
+			return argstring
+		end
+		if x.empty then
+			return safestr(("<%s%s/>"):format(x.label, argstr(x.xarg)))
+		else
+			local open = ("<%s%s>"):format(x.label, argstr(x.xarg))
+			local close = ("</%s>"):format(x.label)
+			local children = {}
+			for _,child in ipairs(x) do
+				table.insert(children, tostring(_xmlify(child)))
+			end
+			return safestr(open .. table.concat(children, '') .. close)
+		end
+	end
+end
+
+local function xmlify(...)
+	local n = select('#',...)
+	if n <= 1 then return _xmlify(...) end
+
+	local out = {}
+	for i = 1, n do
+		out[i] = tostring(_xmlify(select(i, ...)))
+	end
+	return table.concat(out, '\n')
+end
+
+-- dsl for input to xmlify. doesn't do escaping or anything.
+-- THERE IS NO ESCAPE
+local X = setmetatable({}, {__index=function(_,label)
+	return function(tab)
+		local out = {}
+		local xarg = {}
+		local empty = true
+
+		for k, v in pairs(tab) do
+			if type(k) == 'number' then -- child
+				empty = nil
+				out[k] = v
+			elseif type(k) == 'string' then -- attrib
+				xarg[k] = v
+			end
+		end
+
+		out.label = label
+		out.xarg = xarg
+		out.empty = empty
+		return out
+	end
+end})
+
+return { psingle = psingle, stanzae = stanzae, wrap=wrap, xmlify=xmlify, X=X }