mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-01-19 03:57:14 +01:00
159 lines
3.3 KiB
C++
159 lines
3.3 KiB
C++
//
|
|
// HTTPSessionFactory.cpp
|
|
//
|
|
// Library: Net
|
|
// Package: HTTPClient
|
|
// Module: HTTPSessionFactory
|
|
//
|
|
// Copyright (c) 2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Net/HTTPSessionFactory.h"
|
|
#include "Poco/Net/HTTPSessionInstantiator.h"
|
|
#include "Poco/Exception.h"
|
|
|
|
|
|
using Poco::SingletonHolder;
|
|
using Poco::FastMutex;
|
|
using Poco::NotFoundException;
|
|
using Poco::ExistsException;
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
HTTPSessionFactory::HTTPSessionFactory()
|
|
{
|
|
}
|
|
|
|
|
|
HTTPSessionFactory::HTTPSessionFactory(const std::string& proxyHost, Poco::UInt16 proxyPort)
|
|
{
|
|
_proxyConfig.host = proxyHost;
|
|
_proxyConfig.port = proxyPort;
|
|
}
|
|
|
|
|
|
HTTPSessionFactory::HTTPSessionFactory(const HTTPClientSession::ProxyConfig& proxyConfig):
|
|
_proxyConfig(proxyConfig)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPSessionFactory::~HTTPSessionFactory()
|
|
{
|
|
for (auto& p: _instantiators)
|
|
{
|
|
delete p.second.pIn;
|
|
}
|
|
}
|
|
|
|
|
|
void HTTPSessionFactory::registerProtocol(const std::string& protocol, HTTPSessionInstantiator* pSessionInstantiator)
|
|
{
|
|
poco_assert_dbg(pSessionInstantiator);
|
|
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
std::pair<Instantiators::iterator, bool> tmp = _instantiators.insert(make_pair(protocol, InstantiatorInfo(pSessionInstantiator)));
|
|
if (!tmp.second)
|
|
{
|
|
++tmp.first->second.cnt;
|
|
delete pSessionInstantiator;
|
|
}
|
|
}
|
|
|
|
|
|
void HTTPSessionFactory::unregisterProtocol(const std::string& protocol)
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
|
|
Instantiators::iterator it = _instantiators.find(protocol);
|
|
if (it != _instantiators.end())
|
|
{
|
|
if (it->second.cnt == 1)
|
|
{
|
|
delete it->second.pIn;
|
|
_instantiators.erase(it);
|
|
}
|
|
else --it->second.cnt;
|
|
}
|
|
else throw NotFoundException("No HTTPSessionInstantiator registered for", protocol);
|
|
}
|
|
|
|
|
|
bool HTTPSessionFactory::supportsProtocol(const std::string& protocol)
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
|
|
Instantiators::iterator it = _instantiators.find(protocol);
|
|
return it != _instantiators.end();
|
|
}
|
|
|
|
|
|
HTTPClientSession* HTTPSessionFactory::createClientSession(const Poco::URI& uri)
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
|
|
if (uri.isRelative()) throw Poco::UnknownURISchemeException("Relative URIs are not supported by HTTPSessionFactory.");
|
|
|
|
Instantiators::iterator it = _instantiators.find(uri.getScheme());
|
|
if (it != _instantiators.end())
|
|
{
|
|
it->second.pIn->setProxyConfig(_proxyConfig);
|
|
return it->second.pIn->createClientSession(uri);
|
|
}
|
|
else throw Poco::UnknownURISchemeException(uri.getScheme());
|
|
}
|
|
|
|
|
|
void HTTPSessionFactory::setProxy(const std::string& host, Poco::UInt16 port)
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
|
|
_proxyConfig.host = host;
|
|
_proxyConfig.port = port;
|
|
}
|
|
|
|
|
|
void HTTPSessionFactory::setProxyCredentials(const std::string& username, const std::string& password)
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
|
|
_proxyConfig.username = username;
|
|
_proxyConfig.password = password;
|
|
}
|
|
|
|
|
|
void HTTPSessionFactory::setProxyConfig(const HTTPClientSession::ProxyConfig& proxyConfig)
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
|
|
_proxyConfig = proxyConfig;
|
|
}
|
|
|
|
|
|
namespace
|
|
{
|
|
static SingletonHolder<HTTPSessionFactory> singleton;
|
|
}
|
|
|
|
|
|
HTTPSessionFactory& HTTPSessionFactory::defaultFactory()
|
|
{
|
|
return *singleton.get();
|
|
}
|
|
|
|
|
|
HTTPSessionFactory::InstantiatorInfo::InstantiatorInfo(HTTPSessionInstantiator* pInst): pIn(pInst), cnt(1)
|
|
{
|
|
poco_check_ptr (pIn);
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|