2023-03-06 14:42:04 +01:00
|
|
|
// SPDX-License-Identifier: BSD-2-Clause
|
|
|
|
// author: Max Kellermann <max.kellermann@gmail.com>
|
2014-08-07 18:10:23 +02:00
|
|
|
|
2023-09-12 09:47:56 +02:00
|
|
|
#pragma once
|
2014-08-07 18:10:23 +02:00
|
|
|
|
|
|
|
#include "Reader.hxx"
|
|
|
|
#include "fs/AllocatedPath.hxx"
|
|
|
|
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifdef _WIN32
|
2021-05-19 17:24:02 +02:00
|
|
|
#include <fileapi.h>
|
|
|
|
#include <handleapi.h> // for INVALID_HANDLE_VALUE
|
|
|
|
#include <windef.h> // for HWND (needed by winbase.h)
|
|
|
|
#include <winbase.h> // for FILE_CURRENT
|
2018-08-21 18:24:02 +02:00
|
|
|
#else
|
2020-05-05 14:11:13 +02:00
|
|
|
#include "io/UniqueFileDescriptor.hxx"
|
2014-08-07 18:10:23 +02:00
|
|
|
#endif
|
|
|
|
|
2023-01-02 14:27:44 +01:00
|
|
|
#include <cstdint>
|
|
|
|
|
2014-08-07 18:10:23 +02:00
|
|
|
class Path;
|
2015-03-03 20:34:02 +01:00
|
|
|
class FileInfo;
|
2014-08-07 18:10:23 +02:00
|
|
|
|
|
|
|
class FileReader final : public Reader {
|
|
|
|
AllocatedPath path;
|
|
|
|
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifdef _WIN32
|
2014-08-07 18:10:23 +02:00
|
|
|
HANDLE handle;
|
|
|
|
#else
|
2018-08-21 18:23:43 +02:00
|
|
|
UniqueFileDescriptor fd;
|
2014-08-07 18:10:23 +02:00
|
|
|
#endif
|
|
|
|
|
|
|
|
public:
|
2017-01-04 10:37:55 +01:00
|
|
|
explicit FileReader(Path _path);
|
2014-08-07 18:10:23 +02:00
|
|
|
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifdef _WIN32
|
2018-08-21 18:24:30 +02:00
|
|
|
FileReader(FileReader &&other) noexcept
|
2015-03-03 20:09:34 +01:00
|
|
|
:path(std::move(other.path)),
|
2018-08-21 18:25:14 +02:00
|
|
|
handle(std::exchange(other.handle, INVALID_HANDLE_VALUE)) {}
|
2015-03-03 20:09:34 +01:00
|
|
|
|
2018-08-21 18:24:30 +02:00
|
|
|
~FileReader() noexcept {
|
2023-09-12 09:47:56 +02:00
|
|
|
if (handle != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(handle);
|
2014-08-07 18:10:23 +02:00
|
|
|
}
|
2018-08-21 18:23:43 +02:00
|
|
|
#else
|
|
|
|
FileReader(FileReader &&other) noexcept
|
|
|
|
:path(std::move(other.path)),
|
|
|
|
fd(std::move(other.fd)) {}
|
|
|
|
#endif
|
2014-08-07 18:10:23 +02:00
|
|
|
|
|
|
|
|
2015-12-16 11:05:33 +01:00
|
|
|
protected:
|
2018-08-21 18:24:30 +02:00
|
|
|
bool IsDefined() const noexcept {
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifdef _WIN32
|
2014-08-07 18:10:23 +02:00
|
|
|
return handle != INVALID_HANDLE_VALUE;
|
|
|
|
#else
|
2015-03-03 17:03:21 +01:00
|
|
|
return fd.IsDefined();
|
2014-08-07 18:10:23 +02:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2015-12-16 11:05:33 +01:00
|
|
|
public:
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifndef _WIN32
|
2018-08-21 18:24:30 +02:00
|
|
|
FileDescriptor GetFD() const noexcept {
|
2015-03-03 18:10:02 +01:00
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2015-12-16 11:05:33 +01:00
|
|
|
FileInfo GetFileInfo() const;
|
2015-03-03 20:34:02 +01:00
|
|
|
|
2021-10-13 11:28:04 +02:00
|
|
|
[[gnu::pure]]
|
2017-05-08 14:44:49 +02:00
|
|
|
uint64_t GetSize() const noexcept {
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifdef _WIN32
|
2016-08-27 12:20:59 +02:00
|
|
|
LARGE_INTEGER size;
|
|
|
|
return GetFileSizeEx(handle, &size)
|
|
|
|
? size.QuadPart
|
|
|
|
: 0;
|
|
|
|
#else
|
|
|
|
return fd.GetSize();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2021-10-13 11:28:04 +02:00
|
|
|
[[gnu::pure]]
|
2017-05-08 14:44:49 +02:00
|
|
|
uint64_t GetPosition() const noexcept {
|
2017-12-12 10:22:20 +01:00
|
|
|
#ifdef _WIN32
|
2016-08-27 12:20:59 +02:00
|
|
|
LARGE_INTEGER zero;
|
|
|
|
zero.QuadPart = 0;
|
|
|
|
LARGE_INTEGER position;
|
|
|
|
return SetFilePointerEx(handle, zero, &position, FILE_CURRENT)
|
|
|
|
? position.QuadPart
|
|
|
|
: 0;
|
|
|
|
#else
|
|
|
|
return fd.Tell();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2016-08-27 12:09:17 +02:00
|
|
|
void Rewind() {
|
|
|
|
Seek(0);
|
|
|
|
}
|
|
|
|
|
2015-12-16 11:05:33 +01:00
|
|
|
void Seek(off_t offset);
|
2016-02-19 18:18:25 +01:00
|
|
|
void Skip(off_t offset);
|
2015-03-03 14:29:36 +01:00
|
|
|
|
2014-08-07 18:10:23 +02:00
|
|
|
/* virtual methods from class Reader */
|
2021-12-08 19:39:50 +01:00
|
|
|
std::size_t Read(void *data, std::size_t size) override;
|
2014-08-07 18:10:23 +02:00
|
|
|
};
|