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
105
106
107
108
109
110
111
112
|
// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "projfsproviderinterface.h"
#if ZEN_WITH_VFS
# include <zencore/compactbinarybuilder.h>
# include <zencore/except.h>
# include <zencore/string.h>
# include <zencore/thread.h>
ZEN_THIRD_PARTY_INCLUDES_START
# if ZEN_PLATFORM_WINDOWS
# include <zencore/windows.h>
# include <projectedfslib.h>
# endif
ZEN_THIRD_PARTY_INCLUDES_END
# include <unordered_map>
# include <vector>
namespace zen {
class VfsTree
{
public:
VfsTree();
~VfsTree();
void Initialize(VfsTreeDataSource& DataProvider);
void SetTree(VfsTreeNode&& TreeRoot, VfsTreeDataSource& DataProvider);
VfsTreeNode& GetTree();
VfsTreeNode* FindNode(std::wstring NodeName);
void ReadChunkData(Oid ChunkId, void* DataBuffer, uint64_t ByteOffset, uint64_t Length);
private:
struct Impl;
Impl* m_Impl;
};
//////////////////////////////////////////////////////////////////////////
struct VfsProvider : public ProjFsProviderInterface
{
struct EnumerationState
{
std::string Path;
const VfsTreeNode* Node = nullptr;
std::vector<std::unique_ptr<VfsTreeNode>>::const_iterator Iterator;
std::vector<std::unique_ptr<VfsTreeNode>>::const_iterator EndIterator;
void RestartScan();
inline bool IsCurrentValid() { return Iterator != EndIterator; }
bool MatchesPattern(PCWSTR SearchExpression);
void MoveToNext();
const WCHAR* GetCurrentFileName();
PRJ_FILE_BASIC_INFO& GetCurrentBasicInfo();
PRJ_FILE_BASIC_INFO BasicInfo;
std::wstring CurrentName;
};
//////////////////////////////////////////////////////////////////////////
VfsProvider(std::string_view RootPath);
~VfsProvider();
void Initialize();
void Cleanup();
void AddMount(std::string_view Mountpoint, Ref<VfsTreeDataSource>&& DataSource);
void RequestStop();
// ProjFsProviderInterface implementation
virtual void Run() override;
virtual HRESULT StartDirEnum(_In_ const PRJ_CALLBACK_DATA* CallbackData, _In_ const GUID* EnumerationId) override;
virtual HRESULT EndDirEnum(_In_ const PRJ_CALLBACK_DATA* CallbackData, _In_ const GUID* EnumerationId) override;
virtual HRESULT GetDirEnum(_In_ const PRJ_CALLBACK_DATA* CallbackData,
_In_ const GUID* EnumerationId,
_In_opt_ PCWSTR SearchExpression,
_In_ PRJ_DIR_ENTRY_BUFFER_HANDLE DirEntryBufferHandle) override;
virtual HRESULT GetPlaceholderInfo(_In_ const PRJ_CALLBACK_DATA* CallbackData) override;
virtual HRESULT GetFileData(_In_ const PRJ_CALLBACK_DATA* CallbackData, _In_ UINT64 ByteOffset, _In_ UINT32 Length) override;
virtual HRESULT Notify(_In_ const PRJ_CALLBACK_DATA* CallbackData,
_In_ BOOLEAN IsDirectory,
_In_ PRJ_NOTIFICATION NotificationType,
_In_opt_ PCWSTR DestinationFileName,
_Inout_ PRJ_NOTIFICATION_PARAMETERS* NotificationParameters) override;
virtual HRESULT QueryFileName(_In_ const PRJ_CALLBACK_DATA* CallbackData) override;
virtual void CancelCommand(_In_ const PRJ_CALLBACK_DATA* CallbackData) override;
private:
std::string m_RootPath;
Guid m_RootGuid;
VfsTree* m_Tree = nullptr;
RwLock m_EnumerationsLock;
std::unordered_map<zen::Guid, EnumerationState> m_ActiveEnumerations;
Event m_StopRunningEvent;
std::atomic_flag m_IsRunning;
std::atomic_flag m_IsInitialized;
};
} // namespace zen
#endif
|