ponymap/plugins/socks.lua

102 lines
2.5 KiB
Lua

--
-- socks.lua: SOCKS service detection plugin
--
-- Copyright (c) 2015, Přemysl Janouch <p.janouch@gmail.com>
--
-- Permission to use, copy, modify, and/or distribute this software for any
-- purpose with or without fee is hereby granted, provided that the above
-- copyright notice and this permission notice appear in all copies.
--
-- THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
-- WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
-- MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
-- SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
-- WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
-- OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
-- CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
--
-- This plugin serves as an example of how to write Lua plugins for ponymap
ponymap.check_api_version (1)
-- SOCKS 4
local Socks4 = {}
Socks4.__index = Socks4
function Socks4.new (unit)
unit:write (string.pack ("> I1 I1 I2 I1 I1 I1 I1 z",
4, 1, 80, 127, 0, 0, 1, ""))
return setmetatable ({ unit = unit, buf = "" }, Socks4)
end
function Socks4:on_data (data)
self.buf = self.buf .. data
if #self.buf >= 8 then
null, code = string.unpack ("> I1 I1", self.buf)
if null == 0 and code >= 90 and code <= 93 then
self.unit:set_success (true)
end
self.unit:stop ()
end
end
-- SOCKS 4A
local Socks4A = {}
Socks4A.__index = Socks4A
function Socks4A.new (unit)
unit:write (string.pack ("> I1 I1 I2 I4 z z",
4, 1, 80, 1, "", "google.com"))
return setmetatable ({ unit = unit, buf = "" }, Socks4A)
end
Socks4A.on_data = Socks4.on_data
-- SOCKS 5
local Socks5 = {}
Socks5.__index = Socks5
function Socks5.new (unit)
unit:write (string.pack ("> I1 I1 I1", 5, 1, 0))
return setmetatable ({ unit = unit, buf = "" }, Socks5)
end
function Socks5:on_data (data)
self.buf = self.buf .. data
if #self.buf >= 2 then
version, result = string.unpack ("> I1 I1", self.buf)
if version == 5 and (result == 0 or result == 255) then
if result == 0 then
self.unit:add_info ("no authentication required")
end
self.unit:set_success (true)
end
self.unit:stop ()
end
end
-- Register everything
ponymap.register_service ({
name = "SOCKS4",
flags = 0,
new_scan = Socks4.new
})
-- At the moment this is nearly useless
-- ponymap.register_service ({
-- name = "SOCKS4A",
-- flags = 0,
-- new_scan = Socks4A.new
-- })
ponymap.register_service ({
name = "SOCKS5",
flags = 0,
new_scan = Socks5.new
})