rpc.cpp

#include <iostream> // cout
#include "rpc.h"
#include "profiler.h"


CRpcIoNull RpcIoNull;

void rpcMessage::Create( uint16_t cmd )
{
  m_type = RPC_TYPE_DTB;
  m_cmd = cmd;
  m_size = 0;
  m_pos = 0;
}


void rpcMessage::Send( CRpcIo & rpc_io )
{
  rpc_io.Write( &m_type, 1 );
  rpc_io.Write( &m_cmd, 2 );
  rpc_io.Write( &m_size, 1 );
  if( m_size )
    rpc_io.Write( m_par, m_size );
}


void rpcMessage::Receive( CRpcIo & rpc_io )
{
  m_pos = 0;
  rpc_io.Read( &m_type, 1 );
  if( m_type == RPC_TYPE_DTB ) {
  }
  else if( m_type == RPC_TYPE_DTB_DATA ) { // remove unexpected data message from queue
    uint16_t size = 0;
    rpc_io.Read( &size, 3 );
    rpc_DataSink( rpc_io, size );
    throw CRpcError( CRpcError::NO_CMD_MSG );
  }
  else if( m_type == RPC_TYPE_DTB_DATA_OLD ) { // remove unexpected old data message from queue
    uint8_t chn;
    uint16_t size;
    rpc_io.Read( &chn, 1 );
    rpc_io.Read( &size, 2 );
    rpc_DataSink( rpc_io, size );
    throw CRpcError( CRpcError::NO_CMD_MSG );
  }
  else
    throw CRpcError( CRpcError::WRONG_MSG_TYPE );

  rpc_io.Read( &m_cmd, 2 );
  rpc_io.Read( &m_size, 1 );
 //std::cout << "[rpcMessage::Receive] size " << int(m_size) << std::endl;
  if( m_size )
    rpc_io.Read( m_par, m_size ); // usb Read
}


// === data =================================================================

void CDataHeader::RecvHeader( CRpcIo & rpc_io )
{
  rpc_io.Read( &m_type, 1 );
  if( m_type == RPC_TYPE_DTB_DATA ) {
  }
  else if( m_type == RPC_TYPE_DTB ) { // remove unexpected command message from queue
    uint16_t cmd;
    uint8_t size;
    rpc_io.Read( &cmd, 2 );
    rpc_io.Read( &size, 1 );
    rpc_DataSink( rpc_io, size );
    throw CRpcError( CRpcError::NO_DATA_MSG );
  }
  else if( m_type == RPC_TYPE_DTB_DATA_OLD ) { // remove unexpected old data message from queue
    uint8_t chn;
    uint16_t size;
    rpc_io.Read( &chn, 1 );
    rpc_io.Read( &size, 2 );
    rpc_DataSink( rpc_io, size );
    throw CRpcError( CRpcError::NO_CMD_MSG );
  }
  else
    throw CRpcError( CRpcError::WRONG_MSG_TYPE );

  m_size = 0;
  rpc_io.Read( &m_size, 3 );
}



void rpc_SendRaw( CRpcIo & rpc_io, const void *x, uint32_t size )
{
  uint8_t value = RPC_TYPE_DTB_DATA;
  rpc_io.Write( &value, 1 );
  rpc_io.Write( &size, 3 );
  if( size )
    rpc_io.Write( x, size );
 //    printf("Send Data [%i]\n", int(size));
}


void rpc_DataSink( CRpcIo & rpc_io, uint32_t size )
{
  if( size == 0 )
    return;
  CBuffer buffer( size );
  rpc_io.Read( &buffer, size );
}


void rpc_Receive( CRpcIo & rpc_io, string & x )
{
  CDataHeader msg;
  msg.RecvHeader( rpc_io );
  x.clear(  );
  x.reserve( msg.m_size );
 //std::cout << "rpc_Receive " << msg.m_size << " bytes" << std::endl;
  char ch;
  for( uint32_t i = 0; i < msg.m_size; i++ ) {
    rpc_io.Read( &ch, 1 ); // usb.cpp Read
    x.push_back( ch );
  }
}


// === tools ================================================================

void rpc_TranslateCallName( const string & in, string & out )
{
  out.clear(  );

  unsigned int size = in.rfind( '$' );
  unsigned int pos = size;

  try {

    if( pos == string::npos )
      throw int ( 1 );

    pos++;

    int i = 0;

    while( pos < in.size(  ) ) {

      int comp = -1;
      if( in[pos] >= '0' && in[pos] <= '5' ) {
        comp = in[pos] - '0';
        pos++;
        if( pos >= in.size(  ) )
          throw int ( 2 );
      }

      const char *type;
      switch ( in[pos] ) {
      case 'v':
        type = "void";
        break;
      case 'b':
        type = "bool";
        break;
      case 'c':
        type = "int8_t";
        break;
      case 'C':
        type = "uint8_t";
        break;
      case 's':
        type = "int16_t";
        break;
      case 'S':
        type = "uint16_t";
        break;
      case 'i':
        type = "int32_t";
        break;
      case 'I':
        type = "uint32_t";
        break;
      case 'l':
        type = "int64_t";
        break;
      case 'L':
        type = "uint64_t";
        break;
      default:
        type = "?";
      }

      if( i > 1 )
        out += ", ";

      switch ( comp ) {
      case 0:
        out += type;
        out += "&";
        break;
      case 1:
        out += "vector<";
        out += type;
        out += ">&";
        break;
      case 2:
        out += "vectorR<";
        out += type;
        out += ">&";
        break;
      case 3:
        out += "string&";
        break;
      case 4:
        out += "stringR&";
        break;
      case 5:
        out += "HWvectorR&";
        break;
      default:
        out += type;
      }

      if( i == 0 ) {
        out += ' ';
        out += in.substr( 0, size );
        out += '(';
      }

      pos++;
      i++;
    }
    out += ");";
  }
  catch( int ) {
    out = in;
  }
}