server.h

Go to the documentation of this file.

/*
   Copyright 2011 John Selbie

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/


#ifndef STUN_SERVER_H
#define STUN_SERVER_H

#include "stunsocket.h"
#include "stunsocketthread.h"
#include "stunauth.h"
#include "messagehandler.h"




class CStunServerConfig
{
public:
    
    bool fHasPP; // PP: Primary ip,   Primary port
    bool fHasPA; // PA: Primary ip,   Alternate port
    bool fHasAP; // AP: Alternate ip, Primary port
    bool fHasAA; // AA: Alternate ip, Alternate port

    bool fMultiThreadedMode;  // if true, one thread for each socket
    
    bool fTCP; // if true, then use TCP instead of UDP
    uint32_t nMaxConnections; // only valid for TCP (on a per-thread basis)

    CSocketAddress addrPP; // address for PP
    CSocketAddress addrPA; // address for PA
    CSocketAddress addrAP; // address for AP
    CSocketAddress addrAA; // address for AA
    
    CSocketAddress addrPrimaryAdvertised;    // public-IP for PP and PA (port is ignored)
    CSocketAddress addrAlternateAdvertised;  // public-IP for AP and AA (port is ignored)
    
    bool fEnableDosProtection; // enable denial of service protection

    CStunServerConfig();
};


class CStunServer :
public CBasicRefCount,
public CObjectFactory<CStunServer>,
public IRefCounted
{
private:
    CStunSocket _arrSockets[4];

    std::vector<CStunSocketThread*> _threads;

    CStunServer();
    ~CStunServer();

    friend class CObjectFactory<CStunServer>;

    CRefCountedPtr<IStunAuth> _spAuth;
    
    HRESULT AddSocket(TransportAddressSet* pTSA, SocketRole role, const CSocketAddress& addrListen, const CSocketAddress& addrAdvertise);
    
public:

    HRESULT Initialize(const CStunServerConfig& config);
    HRESULT Shutdown();

    HRESULT Start();
    HRESULT Stop();

    ADDREF_AND_RELEASE_IMPL();
};



#endif  /* SERVER_H */