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 <condition_variable>
#include <deque>
#include <iostream>
#include <memory>
#include <stdint.h>
#include <string>
#include <thread>
#include <vector>
 
#include <boost/asio/io_context.hpp>
#include <boost/asio/executor_work_guard.hpp>
#include <boost/asio/ip/udp.hpp>
#include <boost/asio/deadline_timer.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 ( std::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:
      size_t mMaxPayloadSize;
 
      boost::asio::io_context 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::executor_work_guard<boost::asio::io_context::executor_type> mIOserviceWork;
 
      std::thread mDispatchThread;
 
      std::mutex mTransportLayerMutex;
 
      std::deque < std::shared_ptr< Buffers > > mDispatchQueue;
      std::deque < std::shared_ptr< Buffers > > mReplyQueue;
 
      uint32_t mPacketsInFlight;
 
      std::mutex mConditionalVariableMutex;
      std::condition_variable mConditionalVariable;
      bool mFlushDone;
 
      std::shared_ptr< Buffers > mDispatchBuffers;
      std::shared_ptr< Buffers > mReplyBuffers;
 
      uhal::exception::exception* mAsynchronousException;
 
  };
 
 
}
 
 
#endif