-- -- socks.lua: SOCKS service detection plugin -- -- Copyright (c) 2015, Přemysl Janouch -- -- 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:abort () 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:abort () 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 })