1 files changed, 168 insertions, 0 deletions
diff --git a/libs/archive/src/opencallback.cpp b/libs/archive/src/opencallback.cpp
new file mode 100644
index 0000000..9dd0e24
--- /dev/null
+++ b/libs/archive/src/opencallback.cpp
@@ -0,0 +1,168 @@
+/*
+Mod Organizer archive handling
+
+Copyright (C) 2012 Sebastian Herbord, 2020 MO2 Team. All rights reserved.
+
+This library is free software; you can redistribute it and/or
+modify it under the terms of the GNU Lesser General Public
+License as published by the Free Software Foundation; either
+version 3 of the License, or (at your option) any later version.
+
+This library is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+Lesser General Public License for more details.
+
+You should have received a copy of the GNU Lesser General Public
+License along with this library; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+*/
+
+#include "opencallback.h"
+#include "compat.h"
+
+#include "inputstream.h"
+#include "propertyvariant.h"
+
+#include <format>
+#include <memory>
+#include <stdexcept>
+#include <string>
+#include <vector>
+
+#include "fileio.h"
+
+#define UNUSED(x)
+
+CArchiveOpenCallback::CArchiveOpenCallback(Archive::PasswordCallback passwordCallback,
+                                           Archive::LogCallback logCallback,
+                                           std::filesystem::path const& filepath)
+    : m_PasswordCallback(passwordCallback), m_LogCallback(logCallback),
+      m_Path(filepath), m_SubArchiveMode(false)
+{
+  if (!exists(filepath)) {
+    throw std::runtime_error("invalid archive path");
+  }
+
+  if (!IO::FileBase::GetFileInformation(filepath, &m_FileInfo)) {
+    throw std::runtime_error("failed to retrieve file information");
+  }
+}
+
+/* -------------------- IArchiveOpenCallback -------------------- */
+STDMETHODIMP CArchiveOpenCallback::SetTotal(const UInt64* UNUSED(files),
+                                            const UInt64* UNUSED(bytes)) throw()
+{
+  return S_OK;
+}
+
+STDMETHODIMP CArchiveOpenCallback::SetCompleted(const UInt64* UNUSED(files),
+                                                const UInt64* UNUSED(bytes)) throw()
+{
+  return S_OK;
+}
+
+/* -------------------- ICryptoGetTextPassword -------------------- */
+/* This apparently implements ICryptoGetTextPassword but even with a passworded
+ * archive it doesn't seem to be called. There is also another API which isn't
+ * implemented.
+ */
+STDMETHODIMP CArchiveOpenCallback::CryptoGetTextPassword(BSTR* passwordOut) throw()
+{
+  if (!m_PasswordCallback) {
+    return E_ABORT;
+  }
+  m_Password   = m_PasswordCallback();
+  *passwordOut = ::SysAllocString(m_Password.c_str());
+  return *passwordOut != 0 ? S_OK : E_OUTOFMEMORY;
+}
+
+/* -------------------- IArchiveOpenSetSubArchiveName -------------------- */
+/* I don't know what this does or how you call it. */
+STDMETHODIMP CArchiveOpenCallback::SetSubArchiveName(const wchar_t* name) throw()
+{
+  m_SubArchiveMode = true;
+  m_SubArchiveName = name;
+  return S_OK;
+}
+
+/* -------------------- IArchiveOpenVolumeCallback -------------------- */
+
+STDMETHODIMP CArchiveOpenCallback::GetProperty(PROPID propID,
+                                               PROPVARIANT* value) throw()
+{
+  // A scan of the source code indicates that the only things that ever call the
+  // IArchiveOpenVolumeCallback interface ask for the file name and size.
+  PropertyVariant& prop = *static_cast<PropertyVariant*>(value);
+
+  switch (propID) {
+  case kpidName: {
+    if (m_SubArchiveMode) {
+      prop = m_SubArchiveName;
+    } else {
+      // Note: Need to call .native(), otherwize we get a link error because we try to
+      // assign a fs::path to the variant.
+      prop = m_Path.filename().wstring();
+    }
+  } break;
+
+  case kpidIsDir:
+    prop = m_FileInfo.isDir();
+    break;
+  case kpidSize:
+    prop = m_FileInfo.fileSize();
+    break;
+  case kpidAttrib:
+    prop = m_FileInfo.fileAttributes();
+    break;
+  case kpidCTime:
+    prop = m_FileInfo.creationTime();
+    break;
+  case kpidATime:
+    prop = m_FileInfo.lastAccessTime();
+    break;
+  case kpidMTime:
+    prop = m_FileInfo.lastWriteTime();
+    break;
+
+  default:
+    m_LogCallback(Archive::LogLevel::Warning,
+                  std::format(L"Unexpected property {}.", propID));
+  }
+  return S_OK;
+}
+
+STDMETHODIMP CArchiveOpenCallback::GetStream(const wchar_t* name,
+                                             IInStream** inStream) throw()
+{
+  *inStream = nullptr;
+
+  // this function will be called repeatedly for split archives, `name` will
+  // have increasing numbers in the extension and S_FALSE must be returned
+  // when a filename doesn't exist so the search stops
+
+  if (!name) {
+    return S_FALSE;
+  }
+
+  // `name` is just the filename, so build a path from the directory that
+  // contained the last file
+  const auto path = m_Path.parent_path() / name;
+
+  if (!exists(m_FileInfo.path()) || m_FileInfo.isDir()) {
+    return S_FALSE;
+  }
+
+  if (!IO::FileBase::GetFileInformation(path, &m_FileInfo)) {
+    return S_FALSE;
+  }
+
+  CComPtr<InputStream> inFile(new InputStream);
+
+  if (!inFile->Open(m_FileInfo.path())) {
+    return ::GetLastError();
+  }
+
+  *inStream = inFile.Detach();
+  return S_OK;
+}