mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-18 04:12:19 +00:00
b79ead9c84
* Replicate poco into base/poco/ * De-register poco submodule * Build poco from ClickHouse * Exclude poco from stylecheck * Exclude poco from whitespace check * Exclude poco from typo check * Remove x bit from sources/headers (the style check complained) * Exclude poco from duplicate include check * Fix fasttest * Remove contrib/poco-cmake/* * Simplify poco build descriptions * Remove poco stuff not used by ClickHouse * Glob poco sources * Exclude poco from clang-tidy
179 lines
2.5 KiB
C++
179 lines
2.5 KiB
C++
//
|
|
// HTTPHeaderStream.cpp
|
|
//
|
|
// Library: Net
|
|
// Package: HTTP
|
|
// Module: HTTPHeaderStream
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Net/HTTPHeaderStream.h"
|
|
#include "Poco/Net/HTTPSession.h"
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
//
|
|
// HTTPHeaderStreamBuf
|
|
//
|
|
|
|
|
|
HTTPHeaderStreamBuf::HTTPHeaderStreamBuf(HTTPSession& session, openmode mode):
|
|
HTTPBasicStreamBuf(HTTPBufferAllocator::BUFFER_SIZE, mode),
|
|
_session(session),
|
|
_end(false)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPHeaderStreamBuf::~HTTPHeaderStreamBuf()
|
|
{
|
|
}
|
|
|
|
|
|
int HTTPHeaderStreamBuf::readFromDevice(char* buffer, std::streamsize length)
|
|
{
|
|
// read line-by-line; an empty line denotes the end of the headers.
|
|
static const int eof = std::char_traits<char>::eof();
|
|
|
|
if (_end) return 0;
|
|
|
|
int n = 0;
|
|
int ch = _session.get();
|
|
while (ch != eof && ch != '\n' && n < length - 1)
|
|
{
|
|
*buffer++ = (char) ch; ++n;
|
|
ch = _session.get();
|
|
}
|
|
if (ch != eof)
|
|
{
|
|
*buffer++ = (char) ch; ++n;
|
|
if (n == 2) _end = true;
|
|
}
|
|
return n;
|
|
}
|
|
|
|
|
|
int HTTPHeaderStreamBuf::writeToDevice(const char* buffer, std::streamsize length)
|
|
{
|
|
return _session.write(buffer, length);
|
|
}
|
|
|
|
|
|
//
|
|
// HTTPHeaderIOS
|
|
//
|
|
|
|
|
|
HTTPHeaderIOS::HTTPHeaderIOS(HTTPSession& session, HTTPHeaderStreamBuf::openmode mode):
|
|
_buf(session, mode)
|
|
{
|
|
poco_ios_init(&_buf);
|
|
}
|
|
|
|
|
|
HTTPHeaderIOS::~HTTPHeaderIOS()
|
|
{
|
|
try
|
|
{
|
|
_buf.sync();
|
|
}
|
|
catch (...)
|
|
{
|
|
}
|
|
}
|
|
|
|
|
|
HTTPHeaderStreamBuf* HTTPHeaderIOS::rdbuf()
|
|
{
|
|
return &_buf;
|
|
}
|
|
|
|
|
|
//
|
|
// HTTPHeaderInputStream
|
|
//
|
|
|
|
|
|
Poco::MemoryPool HTTPHeaderInputStream::_pool(sizeof(HTTPHeaderInputStream));
|
|
|
|
|
|
HTTPHeaderInputStream::HTTPHeaderInputStream(HTTPSession& session):
|
|
HTTPHeaderIOS(session, std::ios::in),
|
|
std::istream(&_buf)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPHeaderInputStream::~HTTPHeaderInputStream()
|
|
{
|
|
}
|
|
|
|
|
|
void* HTTPHeaderInputStream::operator new(std::size_t size)
|
|
{
|
|
return _pool.get();
|
|
}
|
|
|
|
|
|
void HTTPHeaderInputStream::operator delete(void* ptr)
|
|
{
|
|
try
|
|
{
|
|
_pool.release(ptr);
|
|
}
|
|
catch (...)
|
|
{
|
|
poco_unexpected();
|
|
}
|
|
}
|
|
|
|
|
|
//
|
|
// HTTPHeaderOutputStream
|
|
//
|
|
|
|
|
|
Poco::MemoryPool HTTPHeaderOutputStream::_pool(sizeof(HTTPHeaderOutputStream));
|
|
|
|
|
|
HTTPHeaderOutputStream::HTTPHeaderOutputStream(HTTPSession& session):
|
|
HTTPHeaderIOS(session, std::ios::out),
|
|
std::ostream(&_buf)
|
|
{
|
|
}
|
|
|
|
|
|
HTTPHeaderOutputStream::~HTTPHeaderOutputStream()
|
|
{
|
|
}
|
|
|
|
|
|
void* HTTPHeaderOutputStream::operator new(std::size_t size)
|
|
{
|
|
return _pool.get();
|
|
}
|
|
|
|
|
|
void HTTPHeaderOutputStream::operator delete(void* ptr)
|
|
{
|
|
try
|
|
{
|
|
_pool.release(ptr);
|
|
}
|
|
catch (...)
|
|
{
|
|
poco_unexpected();
|
|
}
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|