blob: 48e4c5122ab6916d456adde09f9a98e506f8a365 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright (c) 2016 Egor Tensin <Egor.Tensin@gmail.com>
// This file is part of the "AES tools" project.
// For details, see https://github.com/egor-tensin/aes-tools.
// Distributed under the MIT License.
#include <cstddef>
#include <fstream>
#include <iterator>
#include <limits>
#include <stdexcept>
#include <string>
#include <type_traits>
#include <vector>
namespace file
{
inline std::size_t cast_to_size_t(std::streamoff size)
{
if (size < 0)
throw std::range_error{"file::cast_to_size_t: something went really wrong"};
typedef std::make_unsigned<std::streamoff>::type unsigned_streamoff;
if (static_cast<unsigned_streamoff>(size) > std::numeric_limits<std::size_t>::max())
throw std::range_error{"file::cast_to_size_t: this file is too large"};
return static_cast<std::size_t>(size);
}
inline std::size_t get_file_size(const std::string& path)
{
std::ifstream ifs;
ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit);
ifs.open(path, std::ifstream::binary | std::ifstream::ate);
return cast_to_size_t(ifs.tellg());
}
inline std::vector<char> read_file(const std::string& path)
{
const auto size = get_file_size(path);
std::ifstream ifs;
ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit);
ifs.open(path, std::ifstream::binary);
std::vector<char> src_buf;
src_buf.reserve(size);
src_buf.assign(
std::istreambuf_iterator<char>{ifs},
std::istreambuf_iterator<char>{});
return src_buf;
}
inline void write_file(
const std::string& path,
const void* buffer,
const std::size_t size)
{
std::ofstream ofs;
ofs.exceptions(std::ofstream::badbit | std::ofstream::failbit);
ofs.open(path, std::ofstream::binary);
ofs.write(reinterpret_cast<const char*>(buffer), size);
}
inline void write_file(
const std::string& path,
const std::vector<unsigned char>& src)
{
write_file(path, src.data(), src.size());
}
}
|
