initial commit

include/drives/winfsp/remotewinfsp/remote_winfsp_drive.hpp

@@ -0,0 +1,140 @@
+/*
+  Copyright <2018-2022> <scott.e.graves@protonmail.com>
+
+  Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
+  associated documentation files (the "Software"), to deal in the Software without restriction,
+  including without limitation the rights to use, copy, modify, merge, publish, distribute,
+  sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
+  furnished to do so, subject to the following conditions:
+
+  The above copyright notice and this permission notice shall be included in all copies or
+  substantial portions of the Software.
+
+  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT
+  NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
+  DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT
+  OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+*/
+#ifndef INCLUDE_DRIVES_WINFSP_REMOTEWINFSP_REMOTE_WINFSP_DRIVE_HPP_
+#define INCLUDE_DRIVES_WINFSP_REMOTEWINFSP_REMOTE_WINFSP_DRIVE_HPP_
+#ifdef _WIN32
+
+#include "common.hpp"
+#include "drives/winfsp/remotewinfsp/i_remote_instance.hpp"
+#include "events/event_system.hpp"
+
+namespace repertory {
+class app_config;
+class lock_data;
+class server;
+namespace remote_winfsp {
+class remote_winfsp_drive final : public virtual FileSystemBase {
+  E_CONSUMER();
+
+public:
+  remote_winfsp_drive(app_config &config, lock_data &lockData,
+                      remote_instance_factory remoteInstanceFactory);
+
+  ~remote_winfsp_drive() override { E_CONSUMER_RELEASE(); }
+
+public:
+  class winfsp_service : virtual public Service {
+  public:
+    winfsp_service(lock_data &lock, remote_winfsp_drive &drive, std::vector<std::string> drive_args,
+                   app_config &config);
+
+    ~winfsp_service() override = default;
+
+  private:
+    app_config &config_;
+    lock_data &lock_;
+    remote_winfsp_drive &drive_;
+    const std::vector<std::string> drive_args_;
+    FileSystemHost host_;
+
+  protected:
+    NTSTATUS OnStart(ULONG, PWSTR *) override;
+
+    NTSTATUS OnStop() override;
+  };
+
+private:
+  app_config &config_;
+  lock_data &lock_;
+  remote_instance_factory factory_;
+  std::unique_ptr<i_remote_instance> remote_instance_;
+  std::unique_ptr<server> server_;
+  std::string mount_location_;
+
+private:
+  void PopulateFileInfo(const json &item, FSP_FSCTL_FILE_INFO &file_info);
+
+  static void SetFileInfo(FileInfo &dest, const remote::file_info &src);
+
+public:
+  NTSTATUS CanDelete(PVOID file_node, PVOID file_desc, PWSTR file_name) override;
+
+  VOID Cleanup(PVOID file_node, PVOID file_desc, PWSTR file_name, ULONG flags) override;
+
+  VOID Close(PVOID file_node, PVOID file_desc) override;
+
+  NTSTATUS Create(PWSTR file_name, UINT32 create_options, UINT32 granted_access, UINT32 attributes,
+                  PSECURITY_DESCRIPTOR descriptor, UINT64 allocation_size, PVOID *file_node,
+                  PVOID *file_desc, OpenFileInfo *ofi) override;
+
+  NTSTATUS Flush(PVOID file_node, PVOID file_desc, FileInfo *file_info) override;
+
+  NTSTATUS GetFileInfo(PVOID file_node, PVOID file_desc, FileInfo *file_info) override;
+
+  NTSTATUS GetSecurityByName(PWSTR file_name, PUINT32 attributes, PSECURITY_DESCRIPTOR descriptor,
+                             SIZE_T *descriptor_size) override;
+
+  NTSTATUS GetVolumeInfo(VolumeInfo *volume_info) override;
+
+  NTSTATUS Init(PVOID host) override;
+
+  int mount(const std::vector<std::string> &drive_args);
+
+  NTSTATUS Mounted(PVOID host) override;
+
+  NTSTATUS Open(PWSTR file_name, UINT32 create_options, UINT32 granted_access, PVOID *file_node,
+                PVOID *file_desc, OpenFileInfo *ofi) override;
+
+  NTSTATUS Overwrite(PVOID file_node, PVOID file_desc, UINT32 attributes,
+                     BOOLEAN replace_attributes, UINT64 allocation_size,
+                     FileInfo *file_info) override;
+
+  NTSTATUS Read(PVOID file_node, PVOID file_desc, PVOID buffer, UINT64 offset, ULONG length,
+                PULONG bytes_transferred) override;
+
+  NTSTATUS ReadDirectory(PVOID file_node, PVOID file_desc, PWSTR pattern, PWSTR marker,
+                         PVOID buffer, ULONG buffer_length, PULONG bytes_transferred) override;
+
+  NTSTATUS Rename(PVOID file_node, PVOID file_desc, PWSTR file_name, PWSTR new_file_name,
+                  BOOLEAN replace_if_exists) override;
+
+  NTSTATUS SetBasicInfo(PVOID file_node, PVOID file_desc, UINT32 attributes, UINT64 creation_time,
+                        UINT64 last_access_time, UINT64 last_write_time, UINT64 change_time,
+                        FileInfo *file_info) override;
+
+  NTSTATUS SetFileSize(PVOID file_node, PVOID file_desc, UINT64 new_size,
+                       BOOLEAN set_allocation_size, FileInfo *file_info) override;
+
+  VOID Unmounted(PVOID host) override;
+
+  NTSTATUS Write(PVOID file_node, PVOID file_desc, PVOID buffer, UINT64 offset, ULONG length,
+                 BOOLEAN write_to_end, BOOLEAN constrained_io, PULONG bytes_transferred,
+                 FileInfo *file_info) override;
+
+  void shutdown() { ::GenerateConsoleCtrlEvent(CTRL_C_EVENT, 0); }
+
+  static void display_options(int argc, char *argv[]) {}
+
+  static void display_version_information(int argc, char *argv[]) {}
+};
+} // namespace remote_winfsp
+} // namespace repertory
+
+#endif // _WIN32
+#endif // INCLUDE_DRIVES_WINFSP_REMOTEWINFSP_REMOTE_WINFSP_DRIVE_HPP_