#include <network.hpp>

#include "dbg_rpc_engine.h"

//-------------------------------------------------------------------------
dbg_rpc_engine_t::dbg_rpc_engine_t(bool _is_client)
  : rpc_engine_t(_is_client),
    has_pending_event(false),
    poll_debug_events(false)
{
}

//--------------------------------------------------------------------------
// sends a request and waits for a reply
// may occasionally send another request based on the reply
rpc_packet_t *dbg_rpc_engine_t::send_request_and_receive_reply(bytevec_t &req, int flags)
{
  bool displayed = false;
  rpc_packet_t *result = NULL;

  while ( true )
  {
    if ( displayed && user_cancelled() )
      req = prepare_rpc_packet(RPC_CANCELLED);

    if ( !req.empty() )
    {
      int code = send_data(req);
      if ( code != 0 || (flags & PREQ_GET_EVENT) != 0 )
        break;

      rpc_packet_t *reqp = (rpc_packet_t *)req.begin();
      if ( reqp->code == RPC_ERROR )
        qexit(1); // sent error packet, may die now
    }

    rpc_packet_t *rp = recv_packet();
    if ( rp == NULL )
      break;

    switch ( rp->code )
    {
      case RPC_UNK:
        dwarning("rpc: remote did not understand our request");
        goto FAILURE;
      case RPC_MEM:
        dwarning("rpc: no remote memory");
        goto FAILURE;
      case RPC_CANCELLED:
        msg("rpc: user cancelled the operation\n");
        goto FAILURE;
      case RPC_OK:
        result = rp;
        goto END;
      default:
        // other replies are passed to the handler
        break;
    }

    if ( !logged_in )
    {
      lprintf("Exploit packet has been detected and ignored\n");
FAILURE:
      qfree(rp);
      break;
    }
    //handle actual command in the request
    //FIXME: use a better function name
    req = on_send_request_interrupt(rp);
    qfree(rp);
  }

END:
  on_send_request_end(result);

  return result;
}

//-------------------------------------------------------------------------
int dbg_rpc_engine_t::_send_request_get_int_result(
        bytevec_t &req,
        int failure_code,
        qstring *errbuf)
{
  rpc_packet_t *rp = send_request_and_receive_reply(req);
  if ( rp == NULL )
    return failure_code;

  memory_deserializer_t mmdsr(rp+1, rp->length);
  int rc = mmdsr.unpack_dd();
  if ( rc < 0 && errbuf != NULL )
    *errbuf = mmdsr.unpack_str();

  qfree(rp);
  return rc;
}