/* packet-drb.c

 *

 * Routines for Ruby Marshal Object

 *

 * Copyright 2018, Dario Lombardo (lomato@gmail.com)

 *

 * Wireshark - Network traffic analyzer

 * By Gerald Combs <gerald@wireshark.org>

 * Copyright 1998 Gerald Combs

 *

 * SPDX-License-Identifier: GPL-2.0-or-later

 */

#include "config.h"

#include <epan/packet.h>

#include <file-rbm.h>

static dissector_handle_t drb_handle;

static int proto_drb;

static int hf_drb_len;

static int ett_drb;

static int ett_ref;

void proto_register_drb(void);

void proto_reg_handoff_drb(void);

static void dissect_drb_object(tvbuff_t* tvb, packet_info* pinfo, proto_tree* tree, unsigned* offset, const char* label)

{

  uint32_t len;

  proto_tree* obj_tree;

  char* type;

  char* value;

  len = tvb_get_uint32(tvb, *offset, ENC_BIG_ENDIAN);

  obj_tree = proto_tree_add_subtree(tree, tvb, *offset, 4 + len, ett_ref, NULL, label);

  proto_tree_add_item(obj_tree, hf_drb_len, tvb, *offset, 4, ENC_BIG_ENDIAN);

  *offset += 4;

  dissect_rbm_inline(tvb, pinfo, obj_tree, offset, &type, &value);

  if (type)

    proto_item_append_text(obj_tree, "Type: %s", type);

  if (value)

    proto_item_append_text(obj_tree, "Value: %s", value);

}

static void dissect_drb_response(tvbuff_t* tvb, packet_info* pinfo, proto_tree* tree, unsigned* offset)

{

  col_append_str(pinfo->cinfo, COL_INFO, " (response)");

  dissect_drb_object(tvb, pinfo, tree, offset, "Success");

  dissect_drb_object(tvb, pinfo, tree, offset, "Response");

}

static void dissect_drb_request(tvbuff_t* tvb, packet_info* pinfo, proto_tree* tree, unsigned* offset)

{

  int32_t nargs;

  int32_t i;

  int len;

  char* loop_label;

  col_append_str(pinfo->cinfo, COL_INFO, " (request)");

  dissect_drb_object(tvb, pinfo, tree, offset, "Ref");

  dissect_drb_object(tvb, pinfo, tree, offset, "Msg ID");

  get_rbm_integer(tvb, *offset + 4 + 3, &nargs, &len);

  dissect_drb_object(tvb, pinfo, tree, offset, "Arg length");

  for (i = 0; i < nargs; i++) {

    loop_label = wmem_strdup_printf(pinfo->pool, "Arg %d", i + 1);

    dissect_drb_object(tvb, pinfo, tree, offset, loop_label);

  }

  dissect_drb_object(tvb, pinfo, tree, offset, "Block");

}

static int dissect_drb(tvbuff_t* tvb, packet_info* pinfo, proto_tree* tree, void* data _U_)

{

  unsigned offset = 0;

  proto_tree* ti;

  proto_tree* drb_tree;

  uint8_t type;

  col_set_str(pinfo->cinfo, COL_PROTOCOL, "DRb");

  col_clear(pinfo->cinfo, COL_INFO);

  col_set_str(pinfo->cinfo, COL_INFO, "Distributed Ruby");

  ti = proto_tree_add_item(tree, proto_drb, tvb, 0, -1, ENC_NA);

  drb_tree = proto_item_add_subtree(ti, ett_drb);

  type = tvb_get_uint8(tvb, 6);

  if (type == 'T' || type == 'F') {

    dissect_drb_response(tvb, pinfo, drb_tree, &offset);

  } else {

    dissect_drb_request(tvb, pinfo, drb_tree, &offset);

  }

  return offset;

}

void proto_register_drb(void)

{

  static hf_register_info hf[] = {

    { &hf_drb_len,

      { "Length", "drb.length", FT_UINT32, BASE_DEC, NULL, 0x00, NULL, HFILL }

    }

  };

  /* Setup protocol subtree array */

  static int* ett[] = {

    &ett_drb,

    &ett_ref

  };

  proto_drb = proto_register_protocol("Distributed Ruby", "DRb", "drb");

  drb_handle = register_dissector("drb", dissect_drb, proto_drb);

  proto_register_field_array(proto_drb, hf, array_length(hf));

  proto_register_subtree_array(ett, array_length(ett));

}

void proto_reg_handoff_drb(void)

{

  dissector_add_for_decode_as_with_preference("tcp.port", drb_handle);

}

/*

 * Editor modelines  -  https://www.wireshark.org/tools/modelines.html

 *

 * Local variables:

 * c-basic-offset: 8

 * tab-width: 8

 * indent-tabs-mode: t

 * End:

 *

 * vi: set shiftwidth=8 tabstop=8 noexpandtab:

 * :indentSize=8:tabSize=8:noTabs=false:

 */