TheCloud/dwarfs
1
0
Fork 0
mirror of https://github.com/mhx/dwarfs.git synced 2025-09-09 04:19:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(dwarfsck): use iolayer for mapping files / writing header

Browse Source
This commit is contained in:
Marcus Holland-Moritz 2024-01-03 12:42:42 +01:00
parent accece15fd
commit 2881a921ba
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/dwarfsck_main.cpp
View File

@ -40,6 +40,7 @@
#include "dwarfs/logger.h"
#include "dwarfs/mmap.h"
#include "dwarfs/options.h"
#include "dwarfs/os_access.h"
#include "dwarfs/tool.h"
#include "dwarfs_tool_main.h"
@ -149,14 +150,17 @@ int dwarfsck_main(int argc, sys_char** argv, iolayer const& iol) {
DWARFS_THROW(runtime_error, "failed to parse offset: " + image_offset);
}
auto mm = std::make_shared<mmap>(input);
std::shared_ptr<mmif> mm = iol.os->map_file(input);
if (print_header) {
if (auto hdr = filesystem_v2::header(mm, fsopts.image_offset)) {
#ifdef _WIN32
::_setmode(STDOUT_FILENO, _O_BINARY);
if (&iol.out == &std::cout) {
::_setmode(::_fileno(stdout), _O_BINARY);
}
#endif
if (::write(STDOUT_FILENO, hdr->data(), hdr->size()) < 0) {
iol.out.write(reinterpret_cast<char const*>(hdr->data()), hdr->size());
if (iol.out.bad() || iol.out.fail()) {
LOG_ERROR << "error writing header: " << ::strerror(errno);
return 1;
}