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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// OpenVPN -- An application to securely tunnel IP networks
// over a single port, with support for SSL/TLS-based
// session authentication and key exchange,
// packet encryption, packet authentication, and
// packet compression.
//
// Copyright (C) 2012- OpenVPN Inc.
//
// SPDX-License-Identifier: MPL-2.0 OR AGPL-3.0-only WITH openvpn3-openssl-exception
//
#ifndef OPENVPN_COMMON_PERSISTFILE_H
#define OPENVPN_COMMON_PERSISTFILE_H
#include <sys/types.h> // for open(), lseek(), ftruncate()
#include <sys/stat.h> // for open()
#include <fcntl.h> // for open()
#include <unistd.h> // for write(), lseek(), ftruncate()
#include <errno.h>
#include <string>
#include <openvpn/common/size.hpp>
#include <openvpn/common/exception.hpp>
#include <openvpn/common/scoped_fd.hpp>
#include <openvpn/common/write.hpp>
#include <openvpn/common/strerror.hpp>
#include <openvpn/buffer/buffer.hpp>
namespace openvpn {
class PersistentFile
{
public:
PersistentFile(const std::string &fn_arg)
: fn(fn_arg)
{
const int f = ::open(fn.c_str(), O_WRONLY | O_CREAT | O_CLOEXEC, S_IRUSR | S_IWUSR);
if (f < 0)
syserr("open");
fd.reset(f);
}
void write(const void *buf, size_t count)
{
const off_t off = ::lseek(fd(), 0, SEEK_SET);
if (off < 0)
syserr("seek");
if (off)
err("unexpected seek");
if (::ftruncate(fd(), 0) < 0)
syserr("truncate");
const ssize_t len = write_retry(fd(), buf, count);
if (len < 0)
syserr("write");
if (static_cast<size_t>(len) != count || len != ::lseek(fd(), 0, SEEK_CUR))
err("incomplete write");
if (::ftruncate(fd(), len) < 0)
syserr("truncate");
}
struct stat stat()
{
struct stat s;
if (::fstat(fd(), &s) < 0)
syserr("fstat");
return s;
}
void write(const Buffer &buf)
{
write(buf.c_data(), buf.size());
}
void write(const std::string &str)
{
write(str.c_str(), str.length());
}
private:
void syserr(const char *type)
{
const int eno = errno;
OPENVPN_THROW_EXCEPTION(fn << " : " << type << " error : " << strerror_str(eno));
}
void err(const char *type)
{
OPENVPN_THROW_EXCEPTION(fn << " : " << type << " error");
}
std::string fn;
ScopedFD fd;
};
} // namespace openvpn
#endif
|
