ProtocolUDP.hpp

Go to the documentation of this file.

/*
---------------------------------------------------------------------------

    This file is part of uHAL.

    uHAL is a hardware access library and programming framework
    originally developed for upgrades of the Level-1 trigger of the CMS
    experiment at CERN.

    uHAL is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    uHAL is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with uHAL.  If not, see <http://www.gnu.org/licenses/>.


      Andrew Rose, Imperial College, London
      email: awr01 <AT> imperial.ac.uk

      Marc Magrans de Abril, CERN
      email: marc.magrans.de.abril <AT> cern.ch

---------------------------------------------------------------------------
*/

#ifndef _uhal_ProtocolUDP_hpp_
#define _uhal_ProtocolUDP_hpp_


#include <deque>
#include <iostream>
#include <stdint.h>
#include <string>
#include <vector>

#include <boost/shared_ptr.hpp>
#include <boost/asio/io_service.hpp>
#include <boost/asio/ip/udp.hpp>
#include <boost/asio/deadline_timer.hpp>

#include <boost/thread/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition_variable.hpp>

#include "uhal/ClientInterface.hpp"
#include "uhal/log/exception.hpp"


namespace boost {
  namespace system { class error_code; }
}

namespace uhal
{
  // Forward declarations
  class Buffers;
  struct URI;

  namespace exception
  {
    UHAL_DEFINE_DERIVED_EXCEPTION_CLASS ( UdpTimeout , ClientTimeout , "Exception class to handle the case where the UDP connection timed out." )
    UHAL_DEFINE_DERIVED_EXCEPTION_CLASS ( ErrorAtUdpSocketCreation , TransportLayerError , "Exception class to handle a failure to create a UDP socket." )
    UHAL_DEFINE_DERIVED_EXCEPTION_CLASS ( ASIOUdpError , TransportLayerError , "Exception class to handle the case where ASIO returned an error." )
  }

  template < typename InnerProtocol >
  class UDP : public InnerProtocol
  {

    private:
      UDP ( const UDP& aUDP ); // non-copyable

      UDP& operator= ( const UDP& aUDP ); // non-assignable

    public:

      UDP ( const std::string& aId, const URI& aUri );

      virtual ~UDP();

    private:
      void implementDispatch ( boost::shared_ptr< Buffers > aBuffers );

      virtual void Flush( );

      virtual void dispatchExceptionHandler();

      uint32_t getMaxSendSize();

      uint32_t getMaxReplySize();

      void connect();
      void write ( );
      void write_callback ( const boost::system::error_code& aErrorCode , std::size_t aBytesTransferred );

      void read ( );

      void read_callback ( const boost::system::error_code& aErrorCode , std::size_t aBytesTransferred );

      void CheckDeadline();

      void NotifyConditionalVariable ( const bool& aValue );

      void WaitOnConditionalVariable();


    private:
      boost::asio::io_service mIOservice;

      boost::asio::ip::udp::socket mSocket;

      boost::asio::ip::udp::endpoint mEndpoint;

      boost::asio::deadline_timer mDeadlineTimer;

      std::vector<uint8_t> mReplyMemory;

      boost::asio::io_service::work mIOserviceWork;

      boost::thread mDispatchThread;

      boost::mutex mTransportLayerMutex;

      std::deque < boost::shared_ptr< Buffers > > mDispatchQueue;
      std::deque < boost::shared_ptr< Buffers > > mReplyQueue;

      uint32_t mPacketsInFlight;

      boost::mutex mConditionalVariableMutex;
      boost::condition_variable mConditionalVariable;
      bool mFlushDone;

      boost::shared_ptr< Buffers > mDispatchBuffers;
      boost::shared_ptr< Buffers > mReplyBuffers;

      uhal::exception::exception* mAsynchronousException;

  };


}


#endif