protobuf.lua

-- latest development release of Wireshark supports plugin version information
if set_plugin_info then
    local my_info = {
        version   = "1.0",
        author    = "Hadriel Kaplan",
        email     = "hadriel@128technology.com",
        copyright = "Copyright (c) 2015, 128 Technology, Inc.",
        license   = "MIT license",
        details   = "This is a plugin for Wireshark, to dissect Google protobuf messages.",
        help      = [[
    HOW TO RUN THIS SCRIPT:

    Wireshark and Tshark support multiple ways of loading Lua scripts: through
    a dofile() call in init.lua, through the file being in either the global
    or personal plugins directories, or via the command line. The latter two
    methods are the best: either copy this script into your "Personal Plugins"
    directory, or load it from the command line.

    This script also needs to know where to find your '.proto' files. There are
    two ways of accomplishing that:
        1. Add it to the "ProtoFileNames" table inside this script, or...
        2. Add it as a command-line argument(s). This only works if this protobuf.lua
           script is also loaded through the command line. For example:
          wireshark -X lua_script:protobuf.lua -X lua_script1:foo.proto -X lua_script1:bar.proto
    ]]
    }
    set_plugin_info(my_info)
end


-- capture command line arguments
local args = { ... }


-- enable loading of our modules
_G['protbuf_dissector'] = {
    ["__DIR__"] = __DIR__,
    ["__DIR_SEPARATOR__"] = __DIR_SEPARATOR__,
}


-- help wireshark find our modules
package.prepend_path("modules")


-- load our settings
local Settings = require "settings"
Settings:processCmdLine(args)
local dprint  = Settings.dprint
local dprint2 = Settings.dprint2


-- load the rest of our modules
local GenericProto  = require "generic.proto"

local Compiler = require "compiler"


dprint("Protobuf modules loaded")


-- initialize the compiler
Compiler:init()

local proto_files = Settings:getProtoFileNames()


----------------------------------------
-- load the proto file contents
dprint2(#proto_files,"proto files being compiled")


----------------------------------------
-- compile the proto files
local decoder_table = Compiler:compile(proto_files)

-- register global dissector string table
local protobuf_table = DissectorTable.new("protobuf.message", "Protobuf Message", ftypes.STRING)
for _, object in pairs(decoder_table) do
	local proto = object['proto']
	dprint("Register", proto.name, "protobuf message")
	protobuf_table:add(proto.name, proto)
end

-- disable loading of our modules
_G['protbuf_dissector'] = nil