blob: e767bbceea86f5c311e1b766320b3b0c2afbb39a (
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
|
// 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 <cassert>
#include <cstddef>
#include <fstream>
#include <iterator>
#include <limits>
#include <string>
#include <type_traits>
#include <vector>
namespace file
{
inline std::size_t cast_to_size_t(std::streamoff size)
{
assert(size >= 0);
assert(static_cast<std::make_unsigned<std::streamoff>::type>(size) <= std::numeric_limits<std::size_t>::max());
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());
}
}
|
