mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-02-23 13:17:14 +01:00
Switched to POCO library for unified platform/library interface. Deprecated the external module API. It was creating more problems than solving. Removed most built-in libraries in favor of system libraries for easier maintenance. Cleaned and secured code with help from static analyzers.
169 lines
3.1 KiB
C++
169 lines
3.1 KiB
C++
//
|
|
// StreamSocket.cpp
|
|
//
|
|
// Library: Net
|
|
// Package: Sockets
|
|
// Module: StreamSocket
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Net/StreamSocket.h"
|
|
#include "Poco/Net/StreamSocketImpl.h"
|
|
#include "Poco/FIFOBuffer.h"
|
|
#include "Poco/Mutex.h"
|
|
#include "Poco/Exception.h"
|
|
|
|
|
|
using Poco::InvalidArgumentException;
|
|
using Poco::Mutex;
|
|
using Poco::ScopedLock;
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
StreamSocket::StreamSocket(): Socket(new StreamSocketImpl)
|
|
{
|
|
}
|
|
|
|
|
|
StreamSocket::StreamSocket(const SocketAddress& address): Socket(new StreamSocketImpl(address.family()))
|
|
{
|
|
connect(address);
|
|
}
|
|
|
|
|
|
StreamSocket::StreamSocket(SocketAddress::Family family): Socket(new StreamSocketImpl(family))
|
|
{
|
|
}
|
|
|
|
|
|
StreamSocket::StreamSocket(const Socket& socket): Socket(socket)
|
|
{
|
|
if (!dynamic_cast<StreamSocketImpl*>(impl()))
|
|
throw InvalidArgumentException("Cannot assign incompatible socket");
|
|
}
|
|
|
|
|
|
StreamSocket::StreamSocket(SocketImpl* pImpl): Socket(pImpl)
|
|
{
|
|
if (!dynamic_cast<StreamSocketImpl*>(impl()))
|
|
throw InvalidArgumentException("Cannot assign incompatible socket");
|
|
}
|
|
|
|
|
|
StreamSocket::~StreamSocket()
|
|
{
|
|
}
|
|
|
|
|
|
StreamSocket& StreamSocket::operator = (const Socket& socket)
|
|
{
|
|
if (dynamic_cast<StreamSocketImpl*>(socket.impl()))
|
|
Socket::operator = (socket);
|
|
else
|
|
throw InvalidArgumentException("Cannot assign incompatible socket");
|
|
return *this;
|
|
}
|
|
|
|
|
|
void StreamSocket::connect(const SocketAddress& address)
|
|
{
|
|
impl()->connect(address);
|
|
}
|
|
|
|
|
|
void StreamSocket::connect(const SocketAddress& address, const Poco::Timespan& timeout)
|
|
{
|
|
impl()->connect(address, timeout);
|
|
}
|
|
|
|
|
|
void StreamSocket::connectNB(const SocketAddress& address)
|
|
{
|
|
impl()->connectNB(address);
|
|
}
|
|
|
|
|
|
void StreamSocket::shutdownReceive()
|
|
{
|
|
impl()->shutdownReceive();
|
|
}
|
|
|
|
|
|
void StreamSocket::shutdownSend()
|
|
{
|
|
impl()->shutdownSend();
|
|
}
|
|
|
|
|
|
void StreamSocket::shutdown()
|
|
{
|
|
impl()->shutdown();
|
|
}
|
|
|
|
|
|
int StreamSocket::sendBytes(const void* buffer, int length, int flags)
|
|
{
|
|
return impl()->sendBytes(buffer, length, flags);
|
|
}
|
|
|
|
|
|
int StreamSocket::sendBytes(const SocketBufVec& buffers, int flags)
|
|
{
|
|
return impl()->sendBytes(buffers, flags);
|
|
}
|
|
|
|
|
|
int StreamSocket::sendBytes(FIFOBuffer& fifoBuf)
|
|
{
|
|
ScopedLock<Mutex> l(fifoBuf.mutex());
|
|
|
|
int ret = impl()->sendBytes(fifoBuf.begin(), (int) fifoBuf.used());
|
|
if (ret > 0) fifoBuf.drain(ret);
|
|
return ret;
|
|
}
|
|
|
|
|
|
int StreamSocket::receiveBytes(void* buffer, int length, int flags)
|
|
{
|
|
return impl()->receiveBytes(buffer, length, flags);
|
|
}
|
|
|
|
|
|
int StreamSocket::receiveBytes(SocketBufVec& buffers, int flags)
|
|
{
|
|
return impl()->receiveBytes(buffers, flags);
|
|
}
|
|
|
|
|
|
int StreamSocket::receiveBytes(Poco::Buffer<char>& buffer, int flags, const Poco::Timespan& timeout)
|
|
{
|
|
return impl()->receiveBytes(buffer, flags, timeout);
|
|
}
|
|
|
|
|
|
int StreamSocket::receiveBytes(FIFOBuffer& fifoBuf)
|
|
{
|
|
ScopedLock<Mutex> l(fifoBuf.mutex());
|
|
|
|
int ret = impl()->receiveBytes(fifoBuf.next(), (int)fifoBuf.available());
|
|
if (ret > 0) fifoBuf.advance(ret);
|
|
return ret;
|
|
}
|
|
|
|
|
|
void StreamSocket::sendUrgent(unsigned char data)
|
|
{
|
|
impl()->sendUrgent(data);
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|