Source code

Revision control

Copy as Markdown

Other Tools

/*
* DataSource
* (C) 1999-2007 Jack Lloyd
* 2005 Matthew Gregan
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include <botan/data_src.h>
#include <botan/exceptn.h>
#include <algorithm>
#include <istream>
#if defined(BOTAN_TARGET_OS_HAS_FILESYSTEM)
#include <fstream>
#endif
namespace Botan {
/*
* Read a single byte from the DataSource
*/
size_t DataSource::read_byte(uint8_t& out)
{
return read(&out, 1);
}
/*
* Peek a single byte from the DataSource
*/
size_t DataSource::peek_byte(uint8_t& out) const
{
return peek(&out, 1, 0);
}
/*
* Discard the next N bytes of the data
*/
size_t DataSource::discard_next(size_t n)
{
uint8_t buf[64] = { 0 };
size_t discarded = 0;
while(n)
{
const size_t got = this->read(buf, std::min(n, sizeof(buf)));
discarded += got;
n -= got;
if(got == 0)
break;
}
return discarded;
}
/*
* Read from a memory buffer
*/
size_t DataSource_Memory::read(uint8_t out[], size_t length)
{
const size_t got = std::min<size_t>(m_source.size() - m_offset, length);
copy_mem(out, m_source.data() + m_offset, got);
m_offset += got;
return got;
}
bool DataSource_Memory::check_available(size_t n)
{
return (n <= (m_source.size() - m_offset));
}
/*
* Peek into a memory buffer
*/
size_t DataSource_Memory::peek(uint8_t out[], size_t length,
size_t peek_offset) const
{
const size_t bytes_left = m_source.size() - m_offset;
if(peek_offset >= bytes_left) return 0;
const size_t got = std::min(bytes_left - peek_offset, length);
copy_mem(out, &m_source[m_offset + peek_offset], got);
return got;
}
/*
* Check if the memory buffer is empty
*/
bool DataSource_Memory::end_of_data() const
{
return (m_offset == m_source.size());
}
/*
* DataSource_Memory Constructor
*/
DataSource_Memory::DataSource_Memory(const std::string& in) :
m_source(cast_char_ptr_to_uint8(in.data()),
cast_char_ptr_to_uint8(in.data()) + in.length()),
m_offset(0)
{
}
/*
* Read from a stream
*/
size_t DataSource_Stream::read(uint8_t out[], size_t length)
{
m_source.read(cast_uint8_ptr_to_char(out), length);
if(m_source.bad())
throw Stream_IO_Error("DataSource_Stream::read: Source failure");
const size_t got = static_cast<size_t>(m_source.gcount());
m_total_read += got;
return got;
}
bool DataSource_Stream::check_available(size_t n)
{
const std::streampos orig_pos = m_source.tellg();
m_source.seekg(0, std::ios::end);
const size_t avail = static_cast<size_t>(m_source.tellg() - orig_pos);
m_source.seekg(orig_pos);
return (avail >= n);
}
/*
* Peek into a stream
*/
size_t DataSource_Stream::peek(uint8_t out[], size_t length, size_t offset) const
{
if(end_of_data())
throw Invalid_State("DataSource_Stream: Cannot peek when out of data");
size_t got = 0;
if(offset)
{
secure_vector<uint8_t> buf(offset);
m_source.read(cast_uint8_ptr_to_char(buf.data()), buf.size());
if(m_source.bad())
throw Stream_IO_Error("DataSource_Stream::peek: Source failure");
got = static_cast<size_t>(m_source.gcount());
}
if(got == offset)
{
m_source.read(cast_uint8_ptr_to_char(out), length);
if(m_source.bad())
throw Stream_IO_Error("DataSource_Stream::peek: Source failure");
got = static_cast<size_t>(m_source.gcount());
}
if(m_source.eof())
m_source.clear();
m_source.seekg(m_total_read, std::ios::beg);
return got;
}
/*
* Check if the stream is empty or in error
*/
bool DataSource_Stream::end_of_data() const
{
return (!m_source.good());
}
/*
* Return a human-readable ID for this stream
*/
std::string DataSource_Stream::id() const
{
return m_identifier;
}
#if defined(BOTAN_TARGET_OS_HAS_FILESYSTEM)
/*
* DataSource_Stream Constructor
*/
DataSource_Stream::DataSource_Stream(const std::string& path,
bool use_binary) :
m_identifier(path),
m_source_memory(new std::ifstream(path, use_binary ? std::ios::binary : std::ios::in)),
m_source(*m_source_memory),
m_total_read(0)
{
if(!m_source.good())
{
throw Stream_IO_Error("DataSource: Failure opening file " + path);
}
}
#endif
/*
* DataSource_Stream Constructor
*/
DataSource_Stream::DataSource_Stream(std::istream& in,
const std::string& name) :
m_identifier(name),
m_source(in),
m_total_read(0)
{
}
DataSource_Stream::~DataSource_Stream()
{
// for ~unique_ptr
}
}