src/m_raw_socket.cc

   1 ///
   2 /// \file       m_raw_socket.cc
   3 ///             Mode class for the raw socket
   4 ///
   5
   6 /*
   7     Copyright (C) 2005-2010, Net Direct Inc. (http://www.netdirect.ca/)
   8     Portions Copyright (C) 2010 RealVNC Ltd.
   9
  10     This program is free software; you can redistribute it and/or modify
  11     it under the terms of the GNU General Public License as published by
  12     the Free Software Foundation; either version 2 of the License, or
  13     (at your option) any later version.
  14
  15     This program is distributed in the hope that it will be useful,
  16     but WITHOUT ANY WARRANTY; without even the implied warranty of
  17     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  18
  19     See the GNU General Public License in the COPYING file at the
  20     root directory of this project for more details.
  21 */
  22
  23 #include "m_raw_socket.h"
  24 #include "data.h"
  25 #include "protocol.h"
  26 #include "protostructs.h"
  27 #include "packet.h"
  28 #include "endian.h"
  29 #include "error.h"
  30 #include "usbwrap.h"
  31 #include "controller.h"
  32 #include <stdexcept>
  33 #include <sstream>
  34 #include <cstring>
  35
  36 #include "debug.h"
  37
  38 namespace Barry { namespace Mode {
  39
  40 static void HandleReceivedDataCallback(void* ctx, Data* data)
  41 {
  42     ((RawSocket*)ctx)->HandleReceivedData(*data);
  43 }
  44
  45 ///////////////////////////////////////////////////////////////////////////////
  46 // RawSocket Mode class
  47
  48 RawSocket::RawSocket(Controller &con, RawSocketDataCallback& callback)
  49         : Mode(con, Controller::RawSocket),
  50       Callback(callback)
  51 {
  52 }
  53
  54 RawSocket::~RawSocket()
  55 {
  56 }
  57
  58 void RawSocket::OnOpen()
  59 {
  60     m_socket->RegisterInterest(HandleReceivedDataCallback, this);
  61 }
  62
  63 ///////////////////////////////////////////////////////////////////////////////
  64 // public API
  65
  66 void RawSocket::Send(Data& data)
  67 {
  68     Data toReceive;
  69     try
  70     {
  71         size_t packetSize = 4 + data.GetSize();
  72         Barry::Protocol::Packet* packet = (Barry::Protocol::Packet*)m_sendBuffer;
  73         packet->socket = htobs(m_socket->GetSocket());
  74         packet->size = htobs(packetSize);
  75         std::memcpy(&(m_sendBuffer[4]), data.GetData(), data.GetSize());
  76
  77         Data toSend(m_sendBuffer, packetSize);
  78         m_socket->PacketData(toSend, toReceive, 0); // timeout immediately
  79         if (toReceive.GetSize() != 0)
  80             HandleReceivedData(toReceive);
  81     }
  82     catch (Usb::Error& err)
  83     {
  84     }
  85 }
  86
  87 void RawSocket::HandleReceivedData(Data& data)
  88 {
  89     // Remove packet headers
  90     Data partial(data.GetData() + 4, data.GetSize() - 4);
  91     Callback.DataReceived(partial);
  92 }
  93
  94 }} // namespace Barry::Mode
  95