Mercurial > xmpp-fingerprint / file comparison
comparison: xmpp-fingerprint.lua
xmpp-fingerprint.lua
- changeset 0
- ba1cc6ad649d
- child 2
- 1eabe33d7af7
equal
deleted
inserted
replaced
| -1:000000000000 | 0:ba1cc6ad649d |
|---|---|
| 1 #!/usr/bin/env lua | |
| 2 | |
| 3 local socket = require "socket"; | |
| 4 | |
| 5 -- Defaults -- | |
| 6 default_port = 5222; | |
| 7 default_namespace = "jabber:client"; | |
| 8 default_timeout = 1.5; | |
| 9 | |
| 10 -- Print debug output -- | |
| 11 function debug() end -- Keep debug output quiet by default | |
| 12 | |
| 13 -- Command-line flag handlers -- | |
| 14 local flags = {}; | |
| 15 function flags.d() -- Show debug | |
| 16 debug = print; | |
| 17 end | |
| 18 | |
| 19 function flags.s() -- Use s2s | |
| 20 -- Use s2s | |
| 21 default_port = 5269; | |
| 22 default_namespace = "jabber:server"; | |
| 23 end | |
| 24 | |
| 25 function flags.t(arg) | |
| 26 default_timeout = tonumber(arg[1] or 1); | |
| 27 return 1; | |
| 28 end | |
| 29 | |
| 30 -- Process flags -- | |
| 31 local function process_args(arg) | |
| 32 local f = arg[1] and arg[1]:match("^%-(%S+)"); | |
| 33 if f then | |
| 34 table.remove(arg, 1); | |
| 35 if flags[f] then | |
| 36 local used = flags[f](arg); | |
| 37 if used then | |
| 38 for i=1,used do | |
| 39 table.remove(arg, 1); | |
| 40 end | |
| 41 end | |
| 42 end | |
| 43 return process_args(arg); | |
| 44 end | |
| 45 end | |
| 46 process_args(arg); | |
| 47 | |
| 48 local received_cache = {}; | |
| 49 local function _test(server, port, question, answer) | |
| 50 local received = received_cache[question]; | |
| 51 if not received then | |
| 52 debug("||", "Connecting to "..server..":"..port.."..."); | |
| 53 local conn = assert(socket.connect(server, port), "Failed to connect to "..arg[1]); | |
| 54 conn:send(question); | |
| 55 debug(">>", tostring(question)); | |
| 56 conn:settimeout(default_timeout); | |
| 57 local data, err, partial = conn:receive("*a"); | |
| 58 received = data or partial or ""; | |
| 59 conn:close(); | |
| 60 received_cache[question] = received; | |
| 61 debug("<<", #received ~= "" and tostring(received) or nil); | |
| 62 end | |
| 63 return received ~= "" and received:match(answer); | |
| 64 end | |
| 65 | |
| 66 function test(question, answer) | |
| 67 return _test(arg[1], tonumber(arg[2]) or default_port, question, answer); | |
| 68 end | |
| 69 | |
| 70 function server_comment(comment) | |
| 71 table.insert(getfenv(2).server_comments, comment); | |
| 72 end | |
| 73 | |
| 74 function literal(s) | |
| 75 return s:gsub("%p", "%%%1"); | |
| 76 end | |
| 77 | |
| 78 ----- | |
| 79 q_invalid_xml = "abc\n"; | |
| 80 q_empty_message_tag = "<message/>"; | |
| 81 q_invalid_host = "<stream:stream xmlns='"..default_namespace.."' xmlns:stream='http://etherx.jabber.org/streams' version='1.0' id='abc' to='0.0.0.0'>"; | |
| 82 ----- | |
| 83 | |
| 84 testers = {}; | |
| 85 | |
| 86 dofile "servers/ejabberd.lua" | |
| 87 dofile "servers/google_talk.lua" | |
| 88 dofile "servers/isode_mlink.lua" | |
| 89 dofile "servers/jabberd14.lua" | |
| 90 dofile "servers/jabberd2.lua" | |
| 91 dofile "servers/prosody.lua" | |
| 92 dofile "servers/tigase.lua" | |
| 93 dofile "servers/djabberd.lua" | |
| 94 dofile "servers/openfire.lua" | |
| 95 dofile "servers/jabber_xcp.lua" | |
| 96 | |
| 97 function test_fingerprint(server) | |
| 98 local tester = testers[server]; | |
| 99 local env = setmetatable({server_comments = {}}, { __index = _G }); | |
| 100 setfenv(tester, env); | |
| 101 local ok, err = pcall(tester); | |
| 102 if not ok then | |
| 103 print("Error while testing for "..server..": "..err); | |
| 104 end | |
| 105 return env; | |
| 106 end | |
| 107 | |
| 108 ------------------------------------------------------------------ | |
| 109 | |
| 110 local results = {}; | |
| 111 for server in pairs(testers) do | |
| 112 local res = test_fingerprint(server); | |
| 113 if res.server_name then | |
| 114 table.insert(results, res); | |
| 115 end | |
| 116 end | |
| 117 | |
| 118 if #results > 1 then | |
| 119 print("Note... multiple matches, could be any:"); | |
| 120 end | |
| 121 | |
| 122 for _, result in ipairs(results) do | |
| 123 io.write(result.server_name); | |
| 124 if result.server_min_version then | |
| 125 io.write(", version "..result.server_min_version.." or later"); | |
| 126 end | |
| 127 io.write("\n"); | |
| 128 if #result.server_comments > 0 then | |
| 129 print(" Notes:\n", table.concat(result.server_comments, "\n ")); | |
| 130 end | |
| 131 end |
