mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-01-31 09:57:14 +01:00
4a6bfc086c
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.
214 lines
3.5 KiB
C++
214 lines
3.5 KiB
C++
//
|
|
// HTTPChunkedStream.cpp
|
|
//
|
|
// Library: Net
|
|
// Package: HTTP
|
|
// Module: HTTPChunkedStream
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Net/HTTPChunkedStream.h"
|
|
#include "Poco/Net/HTTPSession.h"
|
|
#include "Poco/NumberFormatter.h"
|
|
#include "Poco/NumberParser.h"
|
|
#include "Poco/Ascii.h"
|
|
|
|
|
|
using Poco::NumberFormatter;
|
|
using Poco::NumberParser;
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
//
|
|
// HTTPChunkedStreamBuf
|
|
//
|
|
|
|
|
|
HTTPChunkedStreamBuf::HTTPChunkedStreamBuf(HTTPSession& session, openmode mode):
|
|
HTTPBasicStreamBuf(HTTPBufferAllocator::BUFFER_SIZE, mode),
|
|
_session(session),
|
|
_mode(mode),
|
|
_chunk(0)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPChunkedStreamBuf::~HTTPChunkedStreamBuf()
|
|
{
|
|
}
|
|
|
|
|
|
void HTTPChunkedStreamBuf::close()
|
|
{
|
|
if (_mode & std::ios::out)
|
|
{
|
|
sync();
|
|
_session.write("0\r\n\r\n", 5);
|
|
}
|
|
}
|
|
|
|
|
|
int HTTPChunkedStreamBuf::readFromDevice(char* buffer, std::streamsize length)
|
|
{
|
|
static const int eof = std::char_traits<char>::eof();
|
|
|
|
if (_chunk == 0)
|
|
{
|
|
int ch = _session.get();
|
|
while (Poco::Ascii::isSpace(ch)) ch = _session.get();
|
|
std::string chunkLen;
|
|
while (Poco::Ascii::isHexDigit(ch) && chunkLen.size() < 8) { chunkLen += (char) ch; ch = _session.get(); }
|
|
if (ch != eof && !(Poco::Ascii::isSpace(ch) || ch == ';')) return eof;
|
|
while (ch != eof && ch != '\n') ch = _session.get();
|
|
unsigned chunk;
|
|
if (NumberParser::tryParseHex(chunkLen, chunk))
|
|
_chunk = (std::streamsize) chunk;
|
|
else
|
|
return eof;
|
|
}
|
|
if (_chunk > 0)
|
|
{
|
|
if (length > _chunk) length = _chunk;
|
|
int n = _session.read(buffer, length);
|
|
if (n > 0) _chunk -= n;
|
|
return n;
|
|
}
|
|
else
|
|
{
|
|
int ch = _session.get();
|
|
while (ch != eof && ch != '\n') ch = _session.get();
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
|
|
int HTTPChunkedStreamBuf::writeToDevice(const char* buffer, std::streamsize length)
|
|
{
|
|
_chunkBuffer.clear();
|
|
NumberFormatter::appendHex(_chunkBuffer, length);
|
|
_chunkBuffer.append("\r\n", 2);
|
|
_chunkBuffer.append(buffer, static_cast<std::string::size_type>(length));
|
|
_chunkBuffer.append("\r\n", 2);
|
|
_session.write(_chunkBuffer.data(), static_cast<std::streamsize>(_chunkBuffer.size()));
|
|
return static_cast<int>(length);
|
|
}
|
|
|
|
|
|
//
|
|
// HTTPChunkedIOS
|
|
//
|
|
|
|
|
|
HTTPChunkedIOS::HTTPChunkedIOS(HTTPSession& session, HTTPChunkedStreamBuf::openmode mode):
|
|
_buf(session, mode)
|
|
{
|
|
poco_ios_init(&_buf);
|
|
}
|
|
|
|
|
|
HTTPChunkedIOS::~HTTPChunkedIOS()
|
|
{
|
|
try
|
|
{
|
|
_buf.close();
|
|
}
|
|
catch (...)
|
|
{
|
|
}
|
|
}
|
|
|
|
|
|
HTTPChunkedStreamBuf* HTTPChunkedIOS::rdbuf()
|
|
{
|
|
return &_buf;
|
|
}
|
|
|
|
|
|
//
|
|
// HTTPChunkedInputStream
|
|
//
|
|
|
|
|
|
Poco::MemoryPool HTTPChunkedInputStream::_pool(sizeof(HTTPChunkedInputStream));
|
|
|
|
|
|
HTTPChunkedInputStream::HTTPChunkedInputStream(HTTPSession& session):
|
|
HTTPChunkedIOS(session, std::ios::in),
|
|
std::istream(&_buf)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPChunkedInputStream::~HTTPChunkedInputStream()
|
|
{
|
|
}
|
|
|
|
|
|
void* HTTPChunkedInputStream::operator new(std::size_t size)
|
|
{
|
|
return _pool.get();
|
|
}
|
|
|
|
|
|
void HTTPChunkedInputStream::operator delete(void* ptr)
|
|
{
|
|
try
|
|
{
|
|
_pool.release(ptr);
|
|
}
|
|
catch (...)
|
|
{
|
|
poco_unexpected();
|
|
}
|
|
}
|
|
|
|
|
|
//
|
|
// HTTPChunkedOutputStream
|
|
//
|
|
|
|
|
|
Poco::MemoryPool HTTPChunkedOutputStream::_pool(sizeof(HTTPChunkedOutputStream));
|
|
|
|
|
|
HTTPChunkedOutputStream::HTTPChunkedOutputStream(HTTPSession& session):
|
|
HTTPChunkedIOS(session, std::ios::out),
|
|
std::ostream(&_buf)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPChunkedOutputStream::~HTTPChunkedOutputStream()
|
|
{
|
|
}
|
|
|
|
|
|
void* HTTPChunkedOutputStream::operator new(std::size_t size)
|
|
{
|
|
return _pool.get();
|
|
}
|
|
|
|
|
|
void HTTPChunkedOutputStream::operator delete(void* ptr)
|
|
{
|
|
try
|
|
{
|
|
_pool.release(ptr);
|
|
}
|
|
catch (...)
|
|
{
|
|
poco_unexpected();
|
|
}
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|