Server.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 SERVER_HH
#define SERVER_HH

#include <map>
#include <string>
#include <utility>
#include <vector>
#include <stdint.h>
#include <pthread.h>

namespace XrdClTests {

//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
class ClientHandler
{
  public:
    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    ClientHandler();

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    virtual ~ClientHandler();

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    virtual void HandleConnection( int socket ) = 0;

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    void UpdateSentData( char *buffer, uint32_t size );

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    void UpdateReceivedData( char *buffer, uint32_t size );

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    uint64_t GetSentBytes() const
    {
      return pSentBytes;
    }

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    uint32_t GetSentChecksum() const
    {
      return pSentChecksum;
    }

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    uint64_t GetReceivedBytes() const
    {
      return pReceivedBytes;
    }

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    uint32_t GetReceivedChecksum() const
    {
      return pReceivedChecksum;
    }


  private:
    uint64_t    pSentBytes;
    uint64_t    pReceivedBytes;
    uint32_t    pSentChecksum;
    uint32_t    pReceivedChecksum;
};

//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
class ClientHandlerFactory
{
  public:
    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    virtual ~ClientHandlerFactory() {};

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    virtual ClientHandler *CreateHandler() = 0;
};

//------------------------------------------------------------------------------
// Forward declaration
//------------------------------------------------------------------------------
struct ClientHelper;

//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
class Server
{
  public:
    typedef std::map<std::string, std::pair<uint64_t, uint32_t> > TransferMap;

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    Server();

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    ~Server();

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    bool Setup( int port, int accept, ClientHandlerFactory *factory );

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    bool Start();

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    bool Stop();

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    std::pair<uint64_t, uint32_t> GetSentStats( const std::string host ) const;

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    std::pair<uint64_t, uint32_t> GetReceivedStats( const std::string host ) const;

    //--------------------------------------------------------------------------
    //--------------------------------------------------------------------------
    int HandleConnections();

  private:

    TransferMap                 pSent;
    TransferMap                 pReceived;
    pthread_t                   pServerThread;
    std::vector<ClientHelper*>  pClients;
    int                         pListenSocket;
    ClientHandlerFactory       *pHandlerFactory;
};

}

#endif // SERVER_HH