mirror of
https://github.com/openXC7/prjxray.git
synced 2026-08-22 05:57:29 +02:00
Merge 89cad4ca98 into 15fa018ea0
This commit is contained in:
+12
-5
@@ -319,11 +319,18 @@ int main(int argc, char** argv) {
|
|||||||
frame_range_end = strtol(p.second.c_str(), nullptr, 0) + 1;
|
frame_range_end = strtol(p.second.c_str(), nullptr, 0) + 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// The owners of the bytes the span views: they must outlive the
|
||||||
|
// absl::visit below. fcd561fb turned in_bytes from a copying
|
||||||
|
// std::vector into a Span but left both owners scoped to their
|
||||||
|
// if/else blocks -- the mapping was unmapped (and the stdin vector
|
||||||
|
// freed) before the bitstream was parsed, and every invocation
|
||||||
|
// crashed in the sync-word search (use-after-free).
|
||||||
|
std::unique_ptr<prjxray::MemoryMappedFile> in_file;
|
||||||
|
std::vector<uint8_t> stdin_bytes;
|
||||||
absl::Span<uint8_t> in_bytes;
|
absl::Span<uint8_t> in_bytes;
|
||||||
if (argc == 2) {
|
if (argc == 2) {
|
||||||
auto in_file_name = argv[1];
|
auto in_file_name = argv[1];
|
||||||
auto in_file =
|
in_file = prjxray::MemoryMappedFile::InitWithFile(in_file_name);
|
||||||
prjxray::MemoryMappedFile::InitWithFile(in_file_name);
|
|
||||||
if (!in_file) {
|
if (!in_file) {
|
||||||
std::cerr << "Can't open input file '" << in_file_name
|
std::cerr << "Can't open input file '" << in_file_name
|
||||||
<< "' for reading!" << std::endl;
|
<< "' for reading!" << std::endl;
|
||||||
@@ -336,15 +343,15 @@ int main(int argc, char** argv) {
|
|||||||
in_bytes = absl::Span<uint8_t>(
|
in_bytes = absl::Span<uint8_t>(
|
||||||
static_cast<uint8_t*>(in_file->data()), in_file->size());
|
static_cast<uint8_t*>(in_file->data()), in_file->size());
|
||||||
} else {
|
} else {
|
||||||
std::vector<uint8_t> t;
|
|
||||||
while (1) {
|
while (1) {
|
||||||
int c = getchar();
|
int c = getchar();
|
||||||
if (c == EOF)
|
if (c == EOF)
|
||||||
break;
|
break;
|
||||||
t.push_back(c);
|
stdin_bytes.push_back(c);
|
||||||
}
|
}
|
||||||
in_bytes = absl::Span<uint8_t>(
|
in_bytes = absl::Span<uint8_t>(
|
||||||
static_cast<uint8_t*>(t.data()), t.size());
|
static_cast<uint8_t*>(stdin_bytes.data()),
|
||||||
|
stdin_bytes.size());
|
||||||
|
|
||||||
std::cout << "Bitstream size: " << in_bytes.size() << " bytes"
|
std::cout << "Bitstream size: " << in_bytes.size() << " bytes"
|
||||||
<< std::endl;
|
<< std::endl;
|
||||||
|
|||||||
Reference in New Issue
Block a user