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
97
98
99
100
101
102
103
104
|
/**
* Copyright (C) 2021 Saturneric <[email protected]>
*
* This file is part of GpgFrontend.
*
* GpgFrontend is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* GpgFrontend is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GpgFrontend. If not, see <https://www.gnu.org/licenses/>.
*
* The initial version of the source code is inherited from
* the gpg4usb project, which is under GPL-3.0-or-later.
*
* All the source code of GpgFrontend was modified and released by
* Saturneric <[email protected]> starting on May 12, 2021.
*
* SPDX-License-Identifier: GPL-3.0-or-later
*
*/
#include "core/model/GpgData.h"
#include "core/typedef/GpgTypedef.h"
namespace GpgFrontend {
constexpr size_t kBufferSize = 32 * 1024;
GpgFrontend::GpgData::GpgData() {
gpgme_data_t data;
auto err = gpgme_data_new(&data);
assert(gpgme_err_code(err) == GPG_ERR_NO_ERROR);
data_ref_ = std::unique_ptr<struct gpgme_data, DataRefDeleter>(data);
}
GpgFrontend::GpgData::GpgData(const void* buffer, size_t size, bool copy) {
gpgme_data_t data;
auto err = gpgme_data_new_from_mem(&data, static_cast<const char*>(buffer),
size, copy);
assert(gpgme_err_code(err) == GPG_ERR_NO_ERROR);
data_ref_ = std::unique_ptr<struct gpgme_data, DataRefDeleter>(data);
}
auto GpgFrontend::GpgData::Read2Buffer() -> GpgFrontend::ByteArrayPtr {
gpgme_off_t ret = gpgme_data_seek(*this, 0, SEEK_SET);
ByteArrayPtr out_buffer = std::make_unique<std::string>();
if (ret != 0) {
GpgError const err = gpgme_err_code_from_errno(errno);
assert(gpgme_err_code(err) == GPG_ERR_NO_ERROR);
} else {
std::array<std::byte, kBufferSize + 2> buf;
while ((ret = gpgme_data_read(*this, buf.data(), kBufferSize)) > 0) {
const size_t size = out_buffer->size();
out_buffer->resize(static_cast<int>(size + ret));
memcpy(out_buffer->data() + size, buf.data(), ret);
}
if (ret < 0) {
GpgError const err = gpgme_err_code_from_errno(errno);
assert(gpgme_err_code(err) == GPG_ERR_NO_ERROR);
}
}
return out_buffer;
}
auto GpgData::Read2GFBuffer() -> GFBuffer {
gpgme_off_t ret = gpgme_data_seek(*this, 0, SEEK_SET);
GFBuffer out_buffer;
if (ret != 0) {
const GpgError err = gpgme_err_code_from_errno(errno);
assert(gpgme_err_code(err) == GPG_ERR_NO_ERROR);
} else {
std::array<std::byte, kBufferSize + 2> buf;
while ((ret = gpgme_data_read(*this, buf.data(), kBufferSize)) > 0) {
const size_t size = out_buffer.Size();
out_buffer.Resize(static_cast<int>(size + ret));
memcpy(out_buffer.Data() + size, buf.data(), ret);
}
if (ret < 0) {
const GpgError err = gpgme_err_code_from_errno(errno);
assert(gpgme_err_code(err) == GPG_ERR_NO_ERROR);
}
}
return out_buffer;
}
GpgFrontend::GpgData::operator gpgme_data_t() { return data_ref_.get(); }
} // namespace GpgFrontend
|
