update

idasdk75/dbg/dbg_rpc_engine.cpp

@@ -0,0 +1,97 @@
+
+#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;
+}