XrdClAsyncSocketHandler.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// Copyright (c) 2011-2012 by European Organization for Nuclear Research (CERN)
// Author: Lukasz Janyst <ljanyst@cern.ch>
//------------------------------------------------------------------------------
// XRootD is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// XRootD 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 Lesser General Public License
// along with XRootD.  If not, see <http://www.gnu.org/licenses/>.
//------------------------------------------------------------------------------
 
#ifndef __XRD_CL_ASYNC_SOCKET_HANDLER_HH__
#define __XRD_CL_ASYNC_SOCKET_HANDLER_HH__
 
#include "XrdCl/XrdClSocket.hh"
#include "XrdCl/XrdClDefaultEnv.hh"
#include "XrdCl/XrdClPoller.hh"
#include "XrdCl/XrdClPostMasterInterfaces.hh"
#include "XrdCl/XrdClTaskManager.hh"
#include "XrdCl/XrdClXRootDResponses.hh"
#include "XrdCl/XrdClURL.hh"
#include "XrdCl/XrdClAsyncMsgReader.hh"
#include "XrdCl/XrdClAsyncHSReader.hh"
#include "XrdCl/XrdClAsyncMsgWriter.hh"
#include "XrdCl/XrdClAsyncHSWriter.hh"
#include "XrdOuc/XrdOucCompiler.hh"
 
namespace XrdCl
{
  class Stream;
 
  //----------------------------------------------------------------------------
  //----------------------------------------------------------------------------
  class AsyncSocketHandler: public SocketHandler
  {
    public:
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      AsyncSocketHandler( const URL        &url,
                          Poller           *poller,
                          TransportHandler *transport,
                          AnyObject        *channelData,
                          uint16_t          subStreamNum,
                          Stream           *strm );
 
      AsyncSocketHandler( const AsyncSocketHandler &other ):
        AsyncSocketHandler( other.pUrl, other.pPoller, other.pTransport,
                            other.pChannelData, other.pSubStreamNum,
                            other.pStream )
      {
        pLastActivity = other.pLastActivity;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      ~AsyncSocketHandler();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetAddress( const XrdNetAddr &address )
      {
        pSockAddr = address;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      const XrdNetAddr &GetAddress() const
      {
        return pSockAddr;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      XRootDStatus Connect( time_t timeout );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      XRootDStatus PreClose();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      XRootDStatus Close();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      virtual void Event( uint8_t type, XrdCl::Socket */*socket*/ );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      XRootDStatus EnableUplink()
      {
        if( !pPoller->EnableWriteNotification( pSocket, true, pTimeoutResolution ) )
          return XRootDStatus( stFatal, errPollerError );
        return XRootDStatus();
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      XRootDStatus DisableUplink()
      {
        if( !pPoller->EnableWriteNotification( pSocket, false ) )
          return XRootDStatus( stFatal, errPollerError );
        return XRootDStatus();
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      const std::string &GetStreamName()
      {
        return pStreamName;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      time_t GetLastActivity()
      {
        return pLastActivity;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      std::string GetIpStack() const;
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      std::string GetIpAddr();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      std::string GetHostName();
 
    protected:
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      static std::string ToStreamName( const URL &url, uint16_t strmnb );
 
      //------------------------------------------------------------------------
      // Connect returned
      //------------------------------------------------------------------------
      virtual bool OnConnectionReturn() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Got a write readiness event
      //------------------------------------------------------------------------
      bool OnWrite() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Got a write readiness event while handshaking
      //------------------------------------------------------------------------
      bool OnWriteWhileHandshaking() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Got a read readiness event
      //------------------------------------------------------------------------
      bool OnRead() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Got a read readiness event while handshaking
      //------------------------------------------------------------------------
      bool OnReadWhileHandshaking() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handle the handshake message
      //------------------------------------------------------------------------
      bool HandleHandShake( std::unique_ptr<Message> msg )
                            XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Prepare the next step of the hand-shake procedure
      //------------------------------------------------------------------------
      bool HandShakeNextStep( bool done ) XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handle fault
      //------------------------------------------------------------------------
      void OnFault( XRootDStatus st );
 
      //------------------------------------------------------------------------
      // Handle fault while handshaking
      //------------------------------------------------------------------------
      void OnFaultWhileHandshaking( XRootDStatus st );
 
      //------------------------------------------------------------------------
      // Handle write timeout event
      //------------------------------------------------------------------------
      bool OnWriteTimeout() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handle read timeout event
      //------------------------------------------------------------------------
      bool OnReadTimeout() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handle timeout event while handshaking
      //------------------------------------------------------------------------
      bool OnTimeoutWhileHandshaking() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handle header corruption in case of kXR_status response
      //------------------------------------------------------------------------
      void OnHeaderCorruption();
 
      //------------------------------------------------------------------------
      // Carry out the TLS hand-shake
      //
      // The TLS hand-shake is being initiated in HandleHandShake() by calling
      // Socket::TlsHandShake(), however it returns suRetry the TLS hand-shake
      // needs to be followed up by OnTlsHandShake().
      //
      // However, once the TLS connection has been established the server may
      // decide to redo the TLS hand-shake at any time, this operation is handled
      // under the hood by read and write requests and facilitated by
      // Socket::MapEvent()
      //------------------------------------------------------------------------
      XRootDStatus DoTlsHandShake();
 
      //------------------------------------------------------------------------
      // Handle read/write event if we are in the middle of a TLS hand-shake
      //------------------------------------------------------------------------
      // Handle read/write event if we are in the middle of a TLS hand-shake
      bool OnTLSHandShake() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Prepare a HS writer for sending and enable uplink
      //------------------------------------------------------------------------
      bool SendHSMsg() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Extract the value of a wait response
      //
      // @param rsp : the server response
      // @return    : if rsp is a wait response then its value
      //              otherwise -1
      //------------------------------------------------------------------------
      inline kXR_int32 HandleWaitRsp( Message *rsp );
 
      //------------------------------------------------------------------------
      // Check if HS wait time elapsed
      //------------------------------------------------------------------------
      bool CheckHSWait() XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handler for read related socket events
      //------------------------------------------------------------------------
      inline bool EventRead( uint8_t type ) XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Handler for write related socket events
      //------------------------------------------------------------------------
      inline bool EventWrite( uint8_t type ) XRD_WARN_UNUSED_RESULT;
 
      //------------------------------------------------------------------------
      // Data members
      //------------------------------------------------------------------------
      Poller                        *pPoller;
      TransportHandler              *pTransport;
      AnyObject                     *pChannelData;
      uint16_t                       pSubStreamNum;
      Stream                        *pStream;
      std::string                    pStreamName;
      Socket                        *pSocket;
      XrdNetAddr                     pSockAddr;
      std::unique_ptr<HandShakeData> pHandShakeData;
      bool                           pHandShakeDone;
      time_t                         pTimeoutResolution;
      time_t                         pConnectionStarted;
      time_t                         pConnectionTimeout;
      time_t                         pLastActivity;
      time_t                         pHSWaitStarted;
      time_t                         pHSWaitSeconds;
      URL                            pUrl;
      bool                           pTlsHandShakeOngoing;
      XRootDStatus                   pReqConnResetError;
      bool                           pDoTransportDisc;
 
      std::unique_ptr<AsyncHSWriter>  hswriter;
      std::unique_ptr<AsyncMsgReader> rspreader;
      std::unique_ptr<AsyncHSReader>  hsreader;
      std::unique_ptr<AsyncMsgWriter> reqwriter;
 
      std::shared_ptr<Channel>       pOpenChannel;
  };
}
 
#endif // __XRD_CL_ASYNC_SOCKET_HANDLER_HH__