blob: 5840de8005da185a2abf98be269ec4c26d55d1d9 (
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
|
// Copyright (c) 2016 Egor Tensin <egor@tensin.name>
// 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_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(const std::string& path) {
const auto size = get_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(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(const std::string& path, const std::vector<unsigned char>& src) {
write(path, src.data(), src.size());
}
} // namespace file
|
