/src/exiv2/src/basicio.cpp

Source
// SPDX-License-Identifier: GPL-2.0-or-later

// included header files
#include "basicio.hpp"
#include "config.h"
#include "datasets.hpp"
#include "enforce.hpp"
#include "error.hpp"
#include "futils.hpp"
#include "http.hpp"
#include "image_int.hpp"
#include "types.hpp"

#include <algorithm>
#include <cstdio>   // for remove, rename
#include <cstdlib>  // for alloc, realloc, free
#include <cstring>  // std::memcpy
#include <ctime>    // timestamp for the name of temporary file
#include <fstream>  // write the temporary file
#include <iostream>
#include <stdexcept>  // std::logic_error from std::stoll

#if __has_include(<sys/mman.h>)
#include <sys/mman.h>  // for mmap and munmap
#endif
#if __has_include(<process.h>)
#include <process.h>
#endif
#if __has_include(<unistd.h>)
#include <unistd.h>
#endif

#ifdef EXV_USE_CURL
#include <curl/curl.h>
#endif

#ifdef EXV_ENABLE_FILESYSTEM
#include <filesystem>
#ifdef _WIN32
#include <fcntl.h>  // _O_BINARY in FileIo::FileIo
#include <io.h>
#include <windows.h>
#endif
namespace fs = std::filesystem;
#endif

#ifndef _WIN32
#define _fileno fileno
#define _isatty isatty
#endif

// When RemoteIo is used to access a file, the remote server can claim
// that the file is ludicrously large, which will cause a large allocation
// on the client side (Exiv2). To avoid this, we will throw an exception if
// the file size is larger than this limit.
static constexpr int MAX_REMOTE_FILE_SIZE = 0x8000000;

namespace Exiv2 {

BasicIo::~BasicIo() = default;

void BasicIo::readOrThrow(byte* buf, size_t rcount, ErrorCode err) {
  const size_t nread = read(buf, rcount);
  Internal::enforce(nread == rcount, err);
  Internal::enforce(!error(), err);
}

void BasicIo::seekOrThrow(int64_t offset, Position pos, ErrorCode err) {
  const int r = seek(offset, pos);
  Internal::enforce(r == 0, err);
}

#ifdef EXV_ENABLE_FILESYSTEM
//! Internal Pimpl structure of class FileIo.
class FileIo::Impl {
 public:
  //! Constructor
  explicit Impl(std::string path);
#ifdef _WIN32
  explicit Impl(std::wstring path);
#endif
  ~Impl() = default;
  // Enumerations
  //! Mode of operation
  enum OpMode { opRead, opWrite, opSeek };
  // DATA
  std::string path_;  //!< (Standard) path
#ifdef _WIN32
  std::wstring wpath_;  //!< UCS2 path
#endif
  std::string openMode_;   //!< File open mode
  FILE* fp_{};             //!< File stream pointer
  OpMode opMode_{opSeek};  //!< File open mode

#ifdef _WIN32
  HANDLE hFile_{};  //!< Duplicated fd
  HANDLE hMap_{};   //!< Handle from CreateFileMapping
#endif
  byte* pMappedArea_{};    //!< Pointer to the memory-mapped area
  size_t mappedLength_{};  //!< Size of the memory-mapped area
  bool isWriteable_{};     //!< Can the mapped area be written to?
  // TYPES
  //! Simple struct stat wrapper for internal use
  struct StructStat {
    fs::perms st_mode{};       //!< Permissions
    std::uintmax_t st_size{};  //!< Size
  };
  // #endif
  // METHODS
  /*!
    @brief Switch to a new access mode, reopening the file if needed.
        Optimized to only reopen the file when it is really necessary.
    @param opMode The mode to switch to.
    @return 0 if successful
   */
  int switchMode(OpMode opMode);
  //! stat wrapper for internal use
  int stat(StructStat& buf) const;
  // NOT IMPLEMENTED
  Impl(const Impl&) = delete;             //!< Copy constructor
  Impl& operator=(const Impl&) = delete;  //!< Assignment
};

FileIo::Impl::Impl(std::string path) : path_(std::move(path)) {
#ifdef _WIN32
  wchar_t t[512];
  const auto nw = MultiByteToWideChar(CP_UTF8, 0, path_.data(), static_cast<int>(path_.size()), t, 512);
  wpath_.assign(t, nw);
#endif
}
#ifdef _WIN32
FileIo::Impl::Impl(std::wstring path) : wpath_(std::move(path)) {
  char t[1024];
  const auto nc =
      WideCharToMultiByte(CP_UTF8, 0, wpath_.data(), static_cast<int>(wpath_.size()), t, 1024, nullptr, nullptr);
  path_.assign(t, nc);
}
#endif

int FileIo::Impl::switchMode(OpMode opMode) {
  if (opMode_ == opMode)
    return 0;
  OpMode oldOpMode = opMode_;
  opMode_ = opMode;

  bool reopen = true;
  switch (opMode) {
    case opRead:
      // Flush if current mode allows reading, else reopen (in mode "r+b"
      // as in this case we know that we can write to the file)
      if (openMode_.front() == 'r' || openMode_.at(1) == '+')
        reopen = false;
      break;
    case opWrite:
      // Flush if current mode allows writing, else reopen
      if (openMode_.front() != 'r' || openMode_.at(1) == '+')
        reopen = false;
      break;
    case opSeek:
      reopen = false;
      break;
  }

  if (!reopen) {
    // Don't do anything when switching _from_ opSeek mode; we
    // flush when switching _to_ opSeek.
    if (oldOpMode == opSeek)
      return 0;

    // Flush. On msvcrt fflush does not do the job
    std::fseek(fp_, 0, SEEK_CUR);
    return 0;
  }

  // Reopen the file
#ifdef _WIN32
  auto offset = _ftelli64(fp_);
#else
  auto offset = ftello(fp_);
#endif
  if (offset == -1)
    return -1;
  // 'Manual' open("r+b") to avoid munmap()
  std::fclose(fp_);
  openMode_ = "r+b";
  opMode_ = opSeek;
#ifdef _WIN32
  if (_wfopen_s(&fp_, wpath_.c_str(), L"r+b"))
    return 1;
  return _fseeki64(fp_, offset, SEEK_SET);
#else
  fp_ = std::fopen(path_.c_str(), openMode_.c_str());
  if (!fp_)
    return 1;
  return fseeko(fp_, offset, SEEK_SET);
#endif
}  // FileIo::Impl::switchMode

int FileIo::Impl::stat(StructStat& buf) const {
#ifdef _WIN32
  const auto& file = wpath_;
#else
  const auto& file = path_;
#endif
  try {
    buf.st_size = fs::file_size(file);
    buf.st_mode = fs::status(file).permissions();
    return 0;
  } catch (const fs::filesystem_error&) {
    return -1;
  }
}  // FileIo::Impl::stat

FileIo::FileIo(const std::string& path) : p_(std::make_unique<Impl>(path)) {
}
#ifdef _WIN32
FileIo::FileIo(const std::wstring& path) : p_(std::make_unique<Impl>(path)) {
}
#endif

FileIo::~FileIo() {
  close();
}

int FileIo::munmap() {
  int rc = 0;
  if (p_->pMappedArea_) {
#ifdef _WIN32
    UnmapViewOfFile(p_->pMappedArea_);
    CloseHandle(p_->hMap_);
    p_->hMap_ = nullptr;
    CloseHandle(p_->hFile_);
    p_->hFile_ = nullptr;
#elif __has_include(<sys/mman.h>)
    if (::munmap(p_->pMappedArea_, p_->mappedLength_) != 0) {
      rc = 1;
    }
#else
#error Platforms without mmap are not supported. See https://github.com/Exiv2/exiv2/issues/2380
    if (p_->isWriteable_) {
      seek(0, BasicIo::beg);
      write(p_->pMappedArea_, p_->mappedLength_);
    }
    delete[] p_->pMappedArea_;
#endif
  }
  if (p_->isWriteable_) {
    if (p_->fp_)
      p_->switchMode(Impl::opRead);
    p_->isWriteable_ = false;
  }
  p_->pMappedArea_ = nullptr;
  p_->mappedLength_ = 0;
  return rc;
}

byte* FileIo::mmap(bool isWriteable) {
  if (munmap() != 0) {
    throw Error(ErrorCode::kerCallFailed, path(), strError(), "munmap");
  }
  p_->mappedLength_ = size();
  p_->isWriteable_ = isWriteable;
  if (p_->isWriteable_ && p_->switchMode(Impl::opWrite) != 0) {
    throw Error(ErrorCode::kerFailedToMapFileForReadWrite, path(), strError());
  }
#if __has_include(<sys/mman.h>)
  int prot = PROT_READ;
  if (p_->isWriteable_) {
    prot |= PROT_WRITE;
  }
  void* rc = ::mmap(nullptr, p_->mappedLength_, prot, MAP_SHARED, _fileno(p_->fp_), 0);
  if (MAP_FAILED == rc) {
    throw Error(ErrorCode::kerCallFailed, path(), strError(), "mmap");
  }
  p_->pMappedArea_ = static_cast<byte*>(rc);

#elif defined _WIN32
  // Windows implementation

  // TODO: An attempt to map a file with a length of 0 (zero) fails with
  // an error code of ERROR_FILE_INVALID.
  // Applications should test for files with a length of 0 (zero) and
  // reject those files.

  DWORD dwAccess = FILE_MAP_READ;
  DWORD flProtect = PAGE_READONLY;
  if (isWriteable) {
    dwAccess = FILE_MAP_WRITE;
    flProtect = PAGE_READWRITE;
  }
  HANDLE hPh = GetCurrentProcess();
  auto hFd = reinterpret_cast<HANDLE>(_get_osfhandle(_fileno(p_->fp_)));
  if (hFd == INVALID_HANDLE_VALUE) {
    throw Error(ErrorCode::kerCallFailed, path(), "MSG1", "_get_osfhandle");
  }
  if (!DuplicateHandle(hPh, hFd, hPh, &p_->hFile_, 0, false, DUPLICATE_SAME_ACCESS)) {
    throw Error(ErrorCode::kerCallFailed, path(), "MSG2", "DuplicateHandle");
  }
  // For the call to CreateFileMapping(), p_->mappedLength_ needs to be split into high and low parts:
  ULARGE_INTEGER mappedLength{};
  mappedLength.QuadPart = p_->mappedLength_;
  p_->hMap_ = CreateFileMapping(p_->hFile_, nullptr, flProtect, mappedLength.HighPart, mappedLength.LowPart, nullptr);
  if (p_->hMap_ == nullptr) {
    throw Error(ErrorCode::kerCallFailed, path(), "MSG3", "CreateFileMapping");
  }
  void* rc = MapViewOfFile(p_->hMap_, dwAccess, 0, 0, 0);
  if (rc == nullptr) {
    throw Error(ErrorCode::kerCallFailed, path(), "MSG4", "CreateFileMapping");
  }
  p_->pMappedArea_ = static_cast<byte*>(rc);
#else
#error Platforms without mmap are not supported. See https://github.com/Exiv2/exiv2/issues/2380
  // Workaround for platforms without mmap: Read the file into memory
  byte* buf = new byte[p_->mappedLength_];
  const long offset = std::ftell(p_->fp_);
  std::fseek(p_->fp_, 0, SEEK_SET);
  if (read(buf, p_->mappedLength_) != p_->mappedLength_) {
    delete[] buf;
    throw Error(ErrorCode::kerCallFailed, path(), strError(), "FileIo::read");
  }
  std::fseek(p_->fp_, offset, SEEK_SET);
  if (error()) {
    delete[] buf;
    throw Error(ErrorCode::kerCallFailed, path(), strError(), "FileIo::mmap");
  }
  p_->pMappedArea_ = buf;
#endif
  return p_->pMappedArea_;
}

void FileIo::setPath(const std::string& path) {
  close();
  p_->path_ = path;
#ifdef _WIN32
  wchar_t t[512];
  const auto nw = MultiByteToWideChar(CP_UTF8, 0, p_->path_.data(), static_cast<int>(p_->path_.size()), t, 512);
  p_->wpath_.assign(t, nw);
#endif
}

#ifdef _WIN32
void FileIo::setPath(const std::wstring& path) {
  close();
  p_->wpath_ = path;
  char t[1024];
  const auto nc = WideCharToMultiByte(CP_UTF8, 0, p_->wpath_.data(), static_cast<int>(p_->wpath_.size()), t, 1024,
                                      nullptr, nullptr);
  p_->path_.assign(t, nc);
}
#endif

size_t FileIo::write(const byte* data, size_t wcount) {
  if (p_->switchMode(Impl::opWrite) != 0)
    return 0;
  return std::fwrite(data, 1, wcount, p_->fp_);
}

size_t FileIo::write(BasicIo& src) {
  if (static_cast<BasicIo*>(this) == &src)
    return 0;
  if (!src.isopen())
    return 0;
  if (p_->switchMode(Impl::opWrite) != 0)
    return 0;

  byte buf[4096];
  size_t writeTotal = 0;
  size_t readCount = src.read(buf, sizeof(buf));
  while (readCount != 0) {
    size_t writeCount = std::fwrite(buf, 1, readCount, p_->fp_);
    writeTotal += writeCount;
    if (writeCount != readCount) {
      // try to reset back to where write stopped
      src.seek(writeCount - readCount, BasicIo::cur);
      break;
    }
    readCount = src.read(buf, sizeof(buf));
  }

  return writeTotal;
}

void FileIo::transfer(BasicIo& src) {
  const bool wasOpen = (p_->fp_ != nullptr);
  const std::string lastMode(p_->openMode_);

  if (auto fileIo = dynamic_cast<FileIo*>(&src)) {
    // Optimization if src is another instance of FileIo
    fileIo->close();
    // Check if the file can be written to, if it already exists
    if (open("a+b") != 0) {
      // Remove the (temporary) file
      fs::remove(fileIo->path());
      throw Error(ErrorCode::kerFileOpenFailed, path(), "a+b", strError());
    }
    close();

    bool statOk = true;
    fs::perms origStMode;
    const auto& pf = path();

    Impl::StructStat buf1;
    if (p_->stat(buf1) == -1) {
      statOk = false;
    }
    origStMode = buf1.st_mode;

    {
#if defined(_WIN32) && defined(REPLACEFILE_IGNORE_MERGE_ERRORS)
      // Windows implementation that deals with the fact that ::rename fails
      // if the target filename still exists, which regularly happens when
      // that file has been opened with FILE_SHARE_DELETE by another process,
      // like a virus scanner or disk indexer
      // (see also http://stackoverflow.com/a/11023068)
      auto ret =
          ReplaceFileA(pf.c_str(), fileIo->path().c_str(), nullptr, REPLACEFILE_IGNORE_MERGE_ERRORS, nullptr, nullptr);
      if (ret == 0) {
        if (GetLastError() != ERROR_FILE_NOT_FOUND)
          throw Error(ErrorCode::kerFileRenameFailed, fileIo->path(), pf, strError());
        fs::rename(fileIo->path(), pf);
        fs::remove(fileIo->path());
      } else {
        if (fileExists(pf) && fs::remove(pf) != 0)
          throw Error(ErrorCode::kerCallFailed, pf, strError(), "fs::remove");
        fs::rename(fileIo->path(), pf);
        fs::remove(fileIo->path());
      }
#else
      if (fileExists(pf) && fs::remove(pf) != 0) {
        throw Error(ErrorCode::kerCallFailed, pf, strError(), "fs::remove");
      }
      fs::rename(fileIo->path(), pf);
      fs::remove(fileIo->path());
#endif
      // Check permissions of new file
      auto newStMode = fs::status(pf).permissions();
      // Set original file permissions
      if (statOk && origStMode != newStMode) {
        fs::permissions(pf, origStMode);
#ifndef SUPPRESS_WARNINGS
        EXV_WARNING << Error(ErrorCode::kerCallFailed, pf, strError(), "::chmod") << "\n";
#endif
      }
    }
  }  // if (fileIo)
  else {
    // Generic handling, reopen both to reset to start
    if (open("w+b") != 0) {
      throw Error(ErrorCode::kerFileOpenFailed, path(), "w+b", strError());
    }
    if (src.open() != 0) {
      throw Error(ErrorCode::kerDataSourceOpenFailed, src.path(), strError());
    }
    write(src);
    src.close();
  }

  if (wasOpen) {
    if (open(lastMode) != 0) {
      throw Error(ErrorCode::kerFileOpenFailed, path(), lastMode, strError());
    }
  } else
    close();

  if (error() || src.error()) {
    throw Error(ErrorCode::kerTransferFailed, path(), strError());
  }
}  // FileIo::transfer

int FileIo::putb(byte data) {
  if (p_->switchMode(Impl::opWrite) != 0)
    return EOF;
  return putc(data, p_->fp_);
}

int FileIo::seek(int64_t offset, Position pos) {
  int fileSeek = 0;
  switch (pos) {
    case BasicIo::cur:
      fileSeek = SEEK_CUR;
      break;
    case BasicIo::beg:
      fileSeek = SEEK_SET;
      break;
    case BasicIo::end:
      fileSeek = SEEK_END;
      break;
  }

  if (p_->switchMode(Impl::opSeek) != 0)
    return 1;
#ifdef _WIN32
  return _fseeki64(p_->fp_, offset, fileSeek);
#else
  return fseeko(p_->fp_, offset, fileSeek);
#endif
}

size_t FileIo::tell() const {
#ifdef _WIN32
  auto pos = _ftelli64(p_->fp_);
#else
  auto pos = ftello(p_->fp_);
#endif
  Internal::enforce(pos >= 0, ErrorCode::kerInputDataReadFailed);
  return static_cast<size_t>(pos);
}

size_t FileIo::size() const {
  // Flush and commit only if the file is open for writing
  if (p_->fp_ && (p_->openMode_.front() != 'r' || p_->openMode_.at(1) == '+')) {
    std::fflush(p_->fp_);
#ifdef _MSC_VER
    // This is required on msvcrt before stat after writing to a file
    _commit(_fileno(p_->fp_));
#endif
  }

  Impl::StructStat buf;
  if (p_->stat(buf))
    return std::numeric_limits<size_t>::max();
  return static_cast<size_t>(buf.st_size);
}

int FileIo::open() {
  // Default open is in read-only binary mode
  return open("rb");
}

int FileIo::open(const std::string& mode) {
  close();
  p_->openMode_ = mode;
  p_->opMode_ = Impl::opSeek;
#ifdef _WIN32
  wchar_t wmode[10];
  MultiByteToWideChar(CP_UTF8, 0, mode.c_str(), -1, wmode, 10);
  if (_wfopen_s(&p_->fp_, p_->wpath_.c_str(), wmode))
    return 1;
#else
  p_->fp_ = ::fopen(path().c_str(), mode.c_str());
  if (!p_->fp_)
    return 1;
#endif
  return 0;
}

bool FileIo::isopen() const {
  return p_->fp_ != nullptr;
}

int FileIo::close() {
  int rc = 0;
  if (munmap() != 0)
    rc = 2;
  if (p_->fp_) {
    if (std::fclose(p_->fp_) != 0)
      rc |= 1;
    p_->fp_ = nullptr;
  }
  return rc;
}

DataBuf FileIo::read(size_t rcount) {
  if (rcount > size())
    throw Error(ErrorCode::kerInvalidMalloc);
  DataBuf buf(rcount);
  size_t readCount = read(buf.data(), buf.size());
  if (readCount == 0) {
    throw Error(ErrorCode::kerInputDataReadFailed);
  }
  buf.resize(readCount);
  return buf;
}

size_t FileIo::read(byte* buf, size_t rcount) {
  if (p_->switchMode(Impl::opRead) != 0) {
    return 0;
  }
  return std::fread(buf, 1, rcount, p_->fp_);
}

int FileIo::getb() {
  if (p_->switchMode(Impl::opRead) != 0)
    return EOF;
  return getc(p_->fp_);
}

int FileIo::error() const {
  return p_->fp_ ? ferror(p_->fp_) : 0;
}

bool FileIo::eof() const {
  return std::feof(p_->fp_) != 0;
}

const std::string& FileIo::path() const noexcept {
  return p_->path_;
}

void FileIo::populateFakeData() {
}
#endif

//! Internal Pimpl structure of class MemIo.
class MemIo::Impl final {
 public:
  Impl() = default;                     //!< Default constructor
  Impl(const byte* data, size_t size);  //!< Constructor 2
  ~Impl() = default;

  // DATA
  byte* data_{nullptr};     //!< Pointer to the start of the memory area
  size_t idx_{0};           //!< Index into the memory area
  size_t size_{0};          //!< Size of the memory area
  size_t sizeAlloced_{0};   //!< Size of the allocated buffer
  bool isMalloced_{false};  //!< Was the buffer allocated?
  bool eof_{false};         //!< EOF indicator

  // METHODS
  void reserve(size_t wcount);  //!< Reserve memory

  // NOT IMPLEMENTED
  Impl(const Impl&) = delete;             //!< Copy constructor
  Impl& operator=(const Impl&) = delete;  //!< Assignment
};

MemIo::Impl::Impl(const byte* data, size_t size) : data_(const_cast<byte*>(data)), size_(size) {
}

/*!
  @brief Utility class provides the block mapping to the part of data. This avoids allocating
        a single contiguous block of memory to the big data.
 */
class BlockMap {
 public:
  //! the status of the block.
  enum blockType_e { bNone, bKnown, bMemory };

  //! @brief Populate the block.
  //! @param source The data populate to the block
  //! @param num The size of data
  void populate(const byte* source, size_t num) {
    size_ = num;
    data_ = Blob(source, source + num);
    type_ = bMemory;
  }

  /*!
    @brief Change the status to bKnow. bKnow blocks do not contain the data,
          but they keep the size of data. This avoids allocating memory for parts
          of the file that contain image-date (non-metadata/pixel data) which never change in exiv2.
    @param num The size of the data
   */
  void markKnown(size_t num) {
    type_ = bKnown;
    size_ = num;
  }

  [[nodiscard]] bool isNone() const {
    return type_ == bNone;
  }

  [[nodiscard]] bool isKnown() const {
    return type_ == bKnown;
  }

  [[nodiscard]] auto getData() const {
    return data_.data();
  }

  [[nodiscard]] size_t getSize() const {
    return size_;
  }

 private:
  blockType_e type_{bNone};
  Blob data_;
  size_t size_{};
};

void MemIo::Impl::reserve(size_t wcount) {
  const size_t need = wcount + idx_;
  size_t blockSize = 32 * 1024;  // 32768
  const size_t maxBlockSize = 4 * 1024 * 1024;

  if (!isMalloced_) {
    // Minimum size for 1st block
    auto size = std::max<size_t>(blockSize * (1 + need / blockSize), size_);
    auto data = static_cast<byte*>(std::malloc(size));
    if (!data) {
      throw Error(ErrorCode::kerMallocFailed);
    }
    if (data_) {
      std::memcpy(data, data_, size_);
    }
    data_ = data;
    sizeAlloced_ = size;
    isMalloced_ = true;
  }

  if (need > size_) {
    if (need > sizeAlloced_) {
      blockSize = std::min(2 * sizeAlloced_, maxBlockSize);
      // Allocate in blocks
      size_t want = blockSize * (1 + need / blockSize);
      data_ = static_cast<byte*>(std::realloc(data_, want));
      if (!data_) {
        throw Error(ErrorCode::kerMallocFailed);
      }
      sizeAlloced_ = want;
    }
    size_ = need;
  }
}

MemIo::MemIo() : p_(std::make_unique<Impl>()) {
}

MemIo::MemIo(const byte* data, size_t size) : p_(std::make_unique<Impl>(data, size)) {
}

MemIo::~MemIo() {
  if (p_->isMalloced_) {
    std::free(p_->data_);
  }
}

size_t MemIo::write(const byte* data, size_t wcount) {
  p_->reserve(wcount);
  if (data) {
    std::memcpy(&p_->data_[p_->idx_], data, wcount);
  }
  p_->idx_ += wcount;
  return wcount;
}

void MemIo::transfer(BasicIo& src) {
  if (auto memIo = dynamic_cast<MemIo*>(&src)) {
    // Optimization if src is another instance of MemIo
    if (p_->isMalloced_) {
      std::free(p_->data_);
    }
    p_->idx_ = 0;
    p_->data_ = memIo->p_->data_;
    p_->size_ = memIo->p_->size_;
    p_->isMalloced_ = memIo->p_->isMalloced_;
    memIo->p_->idx_ = 0;
    memIo->p_->data_ = nullptr;
    memIo->p_->size_ = 0;
    memIo->p_->isMalloced_ = false;
  } else {
    // Generic reopen to reset position to start
    if (src.open() != 0) {
      throw Error(ErrorCode::kerDataSourceOpenFailed, src.path(), strError());
    }
    p_->idx_ = 0;
    write(src);
    src.close();
  }
  if (error() || src.error())
    throw Error(ErrorCode::kerMemoryTransferFailed, strError());
}

size_t MemIo::write(BasicIo& src) {
  if (this == &src)
    return 0;
  if (!src.isopen())
    return 0;

  byte buf[4096];
  size_t writeTotal = 0;
  size_t readCount = src.read(buf, sizeof(buf));
  while (readCount != 0) {
    write(buf, readCount);
    writeTotal += readCount;
    readCount = src.read(buf, sizeof(buf));
  }

  return writeTotal;
}

int MemIo::putb(byte data) {
  p_->reserve(1);
  p_->data_[p_->idx_++] = data;
  return data;
}

int MemIo::seek(int64_t offset, Position pos) {
  int64_t newIdx = 0;

  switch (pos) {
    case BasicIo::cur:
      newIdx = p_->idx_ + offset;
      break;
    case BasicIo::beg:
      newIdx = offset;
      break;
    case BasicIo::end:
      newIdx = p_->size_ + offset;
      break;
  }

  if (newIdx < 0)
    return 1;

  if (newIdx > static_cast<int64_t>(p_->size_)) {
    p_->eof_ = true;
    return 1;
  }

  p_->idx_ = static_cast<size_t>(newIdx);
  p_->eof_ = false;
  return 0;
}

byte* MemIo::mmap(bool /*isWriteable*/) {
  return p_->data_;
}

int MemIo::munmap() {
  return 0;
}

size_t MemIo::tell() const {
  return p_->idx_;
}

size_t MemIo::size() const {
  return p_->size_;
}

int MemIo::open() {
  p_->idx_ = 0;
  p_->eof_ = false;
  return 0;
}

bool MemIo::isopen() const {
  return true;
}

int MemIo::close() {
  return 0;
}

DataBuf MemIo::read(size_t rcount) {
  DataBuf buf(rcount);
  size_t readCount = read(buf.data(), buf.size());
  buf.resize(readCount);
  return buf;
}

size_t MemIo::read(byte* buf, size_t rcount) {
  const auto avail = std::max<size_t>(p_->size_ - p_->idx_, 0);
  const auto allow = std::min<size_t>(rcount, avail);
  if (allow > 0) {
    std::memcpy(buf, &p_->data_[p_->idx_], allow);
  }
  p_->idx_ += allow;
  if (rcount > avail) {
    p_->eof_ = true;
  }
  return allow;
}

int MemIo::getb() {
  if (p_->idx_ >= p_->size_) {
    p_->eof_ = true;
    return EOF;
  }
  return p_->data_[p_->idx_++];
}

int MemIo::error() const {
  return 0;
}

bool MemIo::eof() const {
  return p_->eof_;
}

const std::string& MemIo::path() const noexcept {
  static std::string _path{"MemIo"};
  return _path;
}

void MemIo::populateFakeData() {
}

#ifdef EXV_ENABLE_FILESYSTEM
XPathIo::XPathIo(const std::string& orgPath) : FileIo(XPathIo::writeDataToFile(orgPath)), tempFilePath_(path()) {
}

XPathIo::~XPathIo() {
  if (isTemp_ && !fs::remove(tempFilePath_)) {
    // error when removing file
    // printf ("Warning: Unable to remove the temp file %s.\n", tempFilePath_.c_str());
  }
}

void XPathIo::transfer(BasicIo& src) {
  if (isTemp_) {
    // replace temp path to gent path.
    auto currentPath = path();

    // replace each substring of the subject that matches the given search string with the given replacement.
    auto ReplaceStringInPlace = [](std::string& subject, std::string_view search, std::string_view replace) {
      auto pos = subject.find(search);
      while (pos != std::string::npos) {
        subject.replace(pos, search.length(), replace);
        pos += subject.find(search, pos + replace.length());
      }
    };

    ReplaceStringInPlace(currentPath, XPathIo::TEMP_FILE_EXT, XPathIo::GEN_FILE_EXT);
    setPath(currentPath);

    tempFilePath_ = path();
    fs::rename(currentPath, tempFilePath_);
    isTemp_ = false;
    // call super class method
    FileIo::transfer(src);
  }
}

std::string XPathIo::writeDataToFile(const std::string& orgPath) {
  Protocol prot = fileProtocol(orgPath);

  // generating the name for temp file.
  std::time_t timestamp = std::time(nullptr);
  auto path = stringFormat("{}{}", timestamp, XPathIo::TEMP_FILE_EXT);

  if (prot == pStdin) {
    if (_isatty(_fileno(stdin)))
      throw Error(ErrorCode::kerInputDataReadFailed);
#ifdef _WIN32
    // convert stdin to binary
    if (_setmode(_fileno(stdin), _O_BINARY) == -1)
      throw Error(ErrorCode::kerInputDataReadFailed);
#endif
    std::ofstream fs(path, std::ios::out | std::ios::binary | std::ios::trunc);
    // read stdin and write to the temp file.
    auto readBuf = std::make_unique<char[]>(100 * 1024);
    std::streamsize readBufSize = 0;
    do {
      std::cin.read(readBuf.get(), 100 * 1024);
      readBufSize = std::cin.gcount();
      if (readBufSize > 0) {
        fs.write(readBuf.get(), readBufSize);
      }
    } while (readBufSize);
    fs.close();
  } else if (prot == pDataUri) {
    std::ofstream fs(path, std::ios::out | std::ios::binary | std::ios::trunc);
    // read data uri and write to the temp file.
    size_t base64Pos = orgPath.find("base64,");
    if (base64Pos == std::string::npos) {
      fs.close();
      throw Error(ErrorCode::kerErrorMessage, "No base64 data");
    }

    std::string data = orgPath.substr(base64Pos + 7);
    auto decodeData = std::make_unique<char[]>(data.length());
    auto size = base64decode(data.c_str(), decodeData.get(), data.length());
    if (size > 0) {
      fs.write(decodeData.get(), size);
      fs.close();
    } else {
      fs.close();
      throw Error(ErrorCode::kerErrorMessage, "Unable to decode base 64.");
    }
  }

  return path;
}

#endif

//! Internal Pimpl abstract structure of class RemoteIo.
class RemoteIo::Impl {
 public:
  //! Constructor
  Impl(const std::string& url, size_t blockSize);
  //! Destructor. Releases all managed memory.
  virtual ~Impl() = default;

  // DATA
  std::string path_;                 //!< (Standard) path
  size_t blockSize_;                 //!< Size of the block memory.
  std::vector<BlockMap> blocksMap_;  //!< An array contains all blocksMap
  size_t size_{0};                   //!< The file size
  size_t idx_{0};                    //!< Index into the memory area
  bool isOpen_{false};               //!< Is the IO open?
  bool eof_{false};                  //!< EOF indicator
  Protocol protocol_;                //!< the protocol of url
  size_t totalRead_{0};              //!< total number of bytes read from host

  // METHODS
  /*!
    @brief Get the length (in bytes) of the remote file.
    @return Return -1 if the size is unknown. Otherwise it returns the length of remote file (in bytes).
    @throw Error if the server returns the error code.
   */
  [[nodiscard]] virtual int64_t getFileLength() const = 0;
  /*!
    @brief Get the data by range.
    @param startBlock The start block index.
    @param stopBlock The stop block index.
    @param response The data from the server.
    @throw Error if the server returns the error code.
    @note Set startBlock = -1 and stopBlock = -1 to get the whole file content.
   */
  virtual void getDataByRange(size_t startBlock, size_t stopBlock, std::string& response) const = 0;
  /*!
    @brief Submit the data to the remote machine. The data replace a part of the remote file.
          The replaced part of remote file is indicated by from and to parameters.
    @param data The data are submitted to the remote machine.
    @param size The size of data.
    @param from The start position in the remote file where the data replace.
    @param to The end position in the remote file where the data replace.
    @note The write access is available on some protocols. HTTP and HTTPS require the script file
          on the remote machine to handle the data. SSH requires the permission to edit the file.
    @throw Error if it fails.
   */
  virtual void writeRemote(const byte* data, size_t size, size_t from, size_t to) = 0;
  /*!
    @brief Get the data from the remote machine and write them to the memory blocks.
    @param startBlock The start block index.
    @param stopBlock The stop block index.
    @return Number of bytes written to the memory block successfully
    @throw Error if it fails.
   */
  virtual size_t populateBlocks(size_t startBlock, size_t stopBlock);
};

RemoteIo::Impl::Impl(const std::string& url, size_t blockSize) :
    path_(url), blockSize_(blockSize), protocol_(fileProtocol(url)) {
}

size_t RemoteIo::Impl::populateBlocks(size_t startBlock, size_t stopBlock) {
  // optimize: ignore all true blocks on left & right sides.
  while (startBlock < stopBlock && !blocksMap_.at(startBlock).isNone())
    startBlock++;
  while (startBlock < stopBlock && !blocksMap_.at(stopBlock - 1).isNone())
    stopBlock--;
  if (startBlock >= stopBlock) {
    return 0;
  }

  size_t rcount = 0;
  std::string data;
  getDataByRange(startBlock, stopBlock, data);
  rcount = data.length();
  size_t iBlock;
  size_t iStop;
  if (rcount == 0) {
    throw Error(ErrorCode::kerErrorMessage, "Data By Range is empty. Please check the permission.");
  } else if (rcount > size_) {
    throw Error(ErrorCode::kerErrorMessage, "Remote server returned more bytes than the specified file size.");
  } else if (rcount == size_) {
    // The remote server is returning the entire file, rather than the subset that we asked for.
    iBlock = 0;
    iStop = (size_ + blockSize_ - 1) / blockSize_;
  } else {
    iBlock = startBlock;
    iStop = stopBlock;
  }

  auto source = reinterpret_cast<byte*>(const_cast<char*>(data.c_str()));
  size_t remain = rcount;
  size_t totalRead = 0;

  while (iBlock < iStop && remain) {
    auto allow = std::min<size_t>(remain, blockSize_);
    blocksMap_.at(iBlock).populate(&source[totalRead], allow);
    remain -= allow;
    totalRead += allow;
    iBlock++;
  }

  return rcount;
}

RemoteIo::RemoteIo() = default;

RemoteIo::~RemoteIo() {
  if (p_) {
    close();
  }
}

int RemoteIo::open() {
  close();  // reset the IO position
  bigBlock_ = nullptr;
  if (!p_->isOpen_) {
    const auto length = p_->getFileLength();
    if (length < 0) {  // unable to get the length of remote file, get the whole file content.
      std::string data;
      p_->getDataByRange(std::numeric_limits<size_t>::max(), std::numeric_limits<size_t>::max(), data);
      p_->size_ = data.length();
      size_t nBlocks = (p_->size_ + p_->blockSize_ - 1) / p_->blockSize_;
      p_->blocksMap_.resize(nBlocks);
      p_->isOpen_ = true;
      auto source = reinterpret_cast<const byte*>(data.c_str());
      size_t remain = p_->size_;
      size_t iBlock = 0;
      size_t totalRead = 0;
      while (remain) {
        auto allow = std::min<size_t>(remain, p_->blockSize_);
        p_->blocksMap_.at(iBlock).populate(&source[totalRead], allow);
        remain -= allow;
        totalRead += allow;
        iBlock++;
      }
    } else if (length == 0) {  // file is empty
      throw Error(ErrorCode::kerErrorMessage, "the file length is 0");
    } else if (length > MAX_REMOTE_FILE_SIZE) {
      throw Error(ErrorCode::kerErrorMessage, "the remote file is too large");
    } else {
      p_->size_ = static_cast<size_t>(length);
      size_t nBlocks = (p_->size_ + p_->blockSize_ - 1) / p_->blockSize_;
      p_->blocksMap_.resize(nBlocks);
      p_->isOpen_ = true;
    }
  }
  return 0;  // means OK
}

int RemoteIo::close() {
  if (p_->isOpen_) {
    p_->eof_ = false;
    p_->idx_ = 0;
    p_->isOpen_ = false;
  }
#ifdef EXIV2_DEBUG_MESSAGES
  std::cerr << "RemoteIo::close totalRead_ = " << p_->totalRead_ << '\n';
#endif
  if (bigBlock_) {
    delete[] bigBlock_;
    bigBlock_ = nullptr;
  }
  return 0;
}

size_t RemoteIo::write(const byte* /* unused data*/, size_t /* unused wcount*/) {
  return 0;  // means failure
}

size_t RemoteIo::write(BasicIo& src) {
  if (!src.isopen())
    return 0;

  /*
   * The idea is to compare the file content, find the different bytes and submit them to the remote machine.
   * To simplify it, it:
   *      + goes from the left, find the first different position -> $left
   *      + goes from the right, find the first different position -> $right
   * The different bytes are [$left-$right] part.
   */
  size_t left = 0;
  size_t right = 0;
  size_t blockIndex = 0;
  auto buf = std::make_unique<byte[]>(p_->blockSize_);
  size_t nBlocks = (p_->size_ + p_->blockSize_ - 1) / p_->blockSize_;

  // find $left
  src.seek(0, BasicIo::beg);
  bool findDiff = false;
  while (blockIndex < nBlocks && !src.eof() && !findDiff) {
    size_t blockSize = p_->blocksMap_.at(blockIndex).getSize();
    bool isFakeData = p_->blocksMap_.at(blockIndex).isKnown();  // fake data
    size_t readCount = src.read(buf.get(), blockSize);
    auto blockData = p_->blocksMap_.at(blockIndex).getData();
    for (size_t i = 0; (i < readCount) && (i < blockSize) && !findDiff; i++) {
      if ((!isFakeData && buf[i] != blockData[i]) || (isFakeData && buf[i] != 0)) {
        findDiff = true;
      } else {
        left++;
      }
    }
    blockIndex++;
  }

  // find $right
  findDiff = false;
  blockIndex = nBlocks;
  while (blockIndex > 0 && right < src.size() && !findDiff) {
    blockIndex--;
    size_t blockSize = p_->blocksMap_.at(blockIndex).getSize();
    if (src.seek(-1 * (blockSize + right), BasicIo::end)) {
      findDiff = true;
    } else {
      bool isFakeData = p_->blocksMap_.at(blockIndex).isKnown();  // fake data
      size_t readCount = src.read(buf.get(), blockSize);
      auto blockData = p_->blocksMap_.at(blockIndex).getData();
      for (size_t i = 0; (i < readCount) && (i < blockSize) && !findDiff; i++) {
        if ((!isFakeData && buf[readCount - i - 1] != blockData[blockSize - i - 1]) ||
            (isFakeData && buf[readCount - i - 1] != 0)) {
          findDiff = true;
        } else {
          right++;
        }
      }
    }
  }

  // submit to the remote machine.
  if (auto dataSize = src.size() - left - right) {
    auto data = std::make_unique<byte[]>(dataSize);
    src.seek(left, BasicIo::beg);
    src.read(data.get(), dataSize);
    p_->writeRemote(data.get(), dataSize, left, p_->size_ - right);
  }
  return src.size();
}

int RemoteIo::putb(byte /*unused data*/) {
  return 0;
}

DataBuf RemoteIo::read(size_t rcount) {
  DataBuf buf(rcount);
  size_t readCount = read(buf.data(), buf.size());
  if (readCount == 0) {
    throw Error(ErrorCode::kerInputDataReadFailed);
  }
  buf.resize(readCount);
  return buf;
}

size_t RemoteIo::read(byte* buf, size_t rcount) {
  if (p_->eof_)
    return 0;

  auto allow = std::min<size_t>(rcount, (p_->size_ - p_->idx_));
  if (allow == 0) {
    return 0;
  }
  size_t startBlock = p_->idx_ / p_->blockSize_;
  size_t stopBlock = (p_->idx_ + allow + p_->blockSize_ - 1) / p_->blockSize_;

  // connect to the remote machine & populate the blocks just in time.
  p_->populateBlocks(startBlock, stopBlock);
  auto fakeData = static_cast<byte*>(std::calloc(p_->blockSize_, sizeof(byte)));
  if (!fakeData) {
    throw Error(ErrorCode::kerErrorMessage, "Unable to allocate data");
  }

  size_t iBlock = startBlock;
  size_t startPos = p_->idx_ - (startBlock * p_->blockSize_);
  size_t totalRead = 0;
  do {
    auto data = p_->blocksMap_.at(iBlock++).getData();
    if (!data)
      data = fakeData;
    auto blockR = std::min<size_t>(allow, p_->blockSize_ - startPos);
    std::memcpy(&buf[totalRead], &data[startPos], blockR);
    totalRead += blockR;
    startPos = 0;
    allow -= blockR;
  } while (allow);

  std::free(fakeData);

  p_->idx_ += totalRead;
  p_->eof_ = (p_->idx_ == p_->size_);
  p_->totalRead_ += totalRead;

  return totalRead;
}

int RemoteIo::getb() {
  if (p_->idx_ == p_->size_) {
    p_->eof_ = true;
    return EOF;
  }

  size_t expectedBlock = p_->idx_ / p_->blockSize_;
  // connect to the remote machine & populate the blocks just in time.
  p_->populateBlocks(expectedBlock, expectedBlock + 1);

  auto data = p_->blocksMap_.at(expectedBlock).getData();
  return data[p_->idx_++ - (expectedBlock * p_->blockSize_)];
}

void RemoteIo::transfer(BasicIo& src) {
  if (src.open() != 0) {
    throw Error(ErrorCode::kerErrorMessage, "unable to open src when transferring");
  }
  write(src);
  src.close();
}

int RemoteIo::seek(int64_t offset, Position pos) {
  int64_t newIdx = 0;

  switch (pos) {
    case BasicIo::cur:
      newIdx = p_->idx_ + offset;
      break;
    case BasicIo::beg:
      newIdx = offset;
      break;
    case BasicIo::end:
      newIdx = p_->size_ + offset;
      break;
  }

  // #1198.  Don't return 1 when asked to seek past EOF.  Stay calm and set eof_
  // if (newIdx < 0 || newIdx > (long) p_->size_) return 1;
  p_->idx_ = static_cast<size_t>(newIdx);
  p_->eof_ = newIdx > static_cast<int64_t>(p_->size_);
  p_->idx_ = std::min(p_->idx_, p_->size_);
  return 0;
}

byte* RemoteIo::mmap(bool /*isWriteable*/) {
  if (!bigBlock_) {
    size_t blockSize = p_->blockSize_;
    size_t blocks = (p_->size_ + blockSize - 1) / blockSize;
    bigBlock_ = new byte[blocks * blockSize]{};
    for (size_t block = 0; block < blocks; block++) {
      auto& b = p_->blocksMap_.at(block);
      if (!b.isNone()) {
        memcpy(bigBlock_ + (block * blockSize), b.getData(), b.getSize());
      }
    }
  }

  return bigBlock_;
}

int RemoteIo::munmap() {
  return 0;
}

size_t RemoteIo::tell() const {
  return p_->idx_;
}

size_t RemoteIo::size() const {
  return p_->size_;
}

bool RemoteIo::isopen() const {
  return p_->isOpen_;
}

int RemoteIo::error() const {
  return 0;
}

bool RemoteIo::eof() const {
  return p_->eof_;
}

const std::string& RemoteIo::path() const noexcept {
  return p_->path_;
}

void RemoteIo::populateFakeData() {
  size_t nBlocks = (p_->size_ + p_->blockSize_ - 1) / p_->blockSize_;
  for (size_t i = 0; i < nBlocks; i++) {
    if (p_->blocksMap_.at(i).isNone())
      p_->blocksMap_.at(i).markKnown(p_->blockSize_);
  }
}

#ifdef EXV_ENABLE_WEBREADY
//! Internal Pimpl structure of class HttpIo.
class HttpIo::HttpImpl : public Impl {
 public:
  //! Constructor
  HttpImpl(const std::string& url, size_t blockSize);
  Exiv2::Uri hostInfo_;  //!< the host information extracted from the path

  // METHODS
  /*!
    @brief Get the length (in bytes) of the remote file.
    @return Return -1 if the size is unknown. Otherwise it returns the length of remote file (in bytes).
    @throw Error if the server returns the error code.
   */
  [[nodiscard]] int64_t getFileLength() const override;
  /*!
    @brief Get the data by range.
    @param startBlock The start block index.
    @param stopBlock The stop block index.
    @param response The data from the server.
    @throw Error if the server returns the error code.
    @note Set startBlock = -1 and stopBlock = -1 to get the whole file content.
   */
  void getDataByRange(size_t startBlock, size_t stopBlock, std::string& response) const override;
  /*!
    @brief Submit the data to the remote machine. The data replace a part of the remote file.
          The replaced part of remote file is indicated by from and to parameters.
    @param data The data are submitted to the remote machine.
    @param size The size of data.
    @param from The start position in the remote file where the data replace.
    @param to The end position in the remote file where the data replace.
    @note The data are submitted to the remote machine via POST. This requires the script file
          on the remote machine to receive the data and edit the remote file. The server-side
          script may be specified with the environment string EXIV2_HTTP_POST. The default value is
          "/exiv2.php". More info is available at http://dev.exiv2.org/wiki/exiv2
    @throw Error if it fails.
   */
  void writeRemote(const byte* data, size_t size, size_t from, size_t to) override;
};

HttpIo::HttpImpl::HttpImpl(const std::string& url, size_t blockSize) : Impl(url, blockSize) {
  hostInfo_ = Exiv2::Uri::Parse(url);
  Exiv2::Uri::Decode(hostInfo_);
}

int64_t HttpIo::HttpImpl::getFileLength() const {
  Exiv2::Dictionary response;
  Exiv2::Dictionary request;
  std::string errors;
  request["server"] = hostInfo_.Host;
  request["page"] = hostInfo_.Path;
  if (!hostInfo_.Port.empty())
    request["port"] = hostInfo_.Port;
  request["verb"] = "HEAD";
  int serverCode = http(request, response, errors);
  if (serverCode < 0 || serverCode >= 400 || !errors.empty()) {
    throw Error(ErrorCode::kerFileOpenFailed, "http", serverCode, hostInfo_.Path);
  }

  auto lengthIter = response.find("Content-Length");
  if (lengthIter == response.end())
    return -1;
  try {
    return std::stoll(lengthIter->second);
  } catch (const std::logic_error&) {
    // the server returned a non-numeric or out-of-range Content-Length; treat the size as unknown
    return -1;
  }
}

void HttpIo::HttpImpl::getDataByRange(size_t startBlock, size_t stopBlock, std::string& response) const {
  Exiv2::Dictionary responseDic;
  Exiv2::Dictionary request;
  request["server"] = hostInfo_.Host;
  request["page"] = hostInfo_.Path;
  if (!hostInfo_.Port.empty())
    request["port"] = hostInfo_.Port;
  request["verb"] = "GET";
  std::string errors;
  if (startBlock != std::numeric_limits<size_t>::max() && stopBlock != std::numeric_limits<size_t>::max()) {
    request["header"] = stringFormat("Range: bytes={}-{}", startBlock * blockSize_, stopBlock * blockSize_ - 1);
  }

  int serverCode = http(request, responseDic, errors);
  if (serverCode < 0 || serverCode >= 400 || !errors.empty()) {
    throw Error(ErrorCode::kerFileOpenFailed, "http", serverCode, hostInfo_.Path);
  }
  response = responseDic["body"];
}

void HttpIo::HttpImpl::writeRemote(const byte* data, size_t size, size_t from, size_t to) {
  std::string scriptPath(getEnv(envHTTPPOST));
  if (scriptPath.empty()) {
    throw Error(ErrorCode::kerErrorMessage,
                "Please set the path of the server script to handle http post data to EXIV2_HTTP_POST "
                "environmental variable.");
  }

  // standardize the path without "/" at the beginning.
  if (scriptPath.find("://") == std::string::npos && scriptPath.front() != '/') {
    scriptPath = "/" + scriptPath;
  }

  Exiv2::Dictionary response;
  Exiv2::Dictionary request;
  std::string errors;

  Uri scriptUri = Exiv2::Uri::Parse(scriptPath);
  request["server"] = scriptUri.Host.empty() ? hostInfo_.Host : scriptUri.Host;
  if (!scriptUri.Port.empty())
    request["port"] = scriptUri.Port;
  request["page"] = scriptUri.Path;
  request["verb"] = "POST";

  // encode base64
  size_t encodeLength = (((size + 2) / 3) * 4) + 1;
  auto encodeData = std::make_unique<char[]>(encodeLength);
  base64encode(data, size, encodeData.get(), encodeLength);
  // url encode
  const std::string urlencodeData = urlencode(encodeData.get());

  auto postData = stringFormat("path={}&from={}&to={}&data={}", hostInfo_.Path, from, to, urlencodeData);

  // create the header
  auto header = stringFormat(
      "Content-Length: {}\n"
      "Content-Type: application/x-www-form-urlencoded\n"
      "\n{}\r\n",
      postData.length(), postData);
  request["header"] = std::move(header);

  int serverCode = http(request, response, errors);
  if (serverCode < 0 || serverCode >= 400 || !errors.empty()) {
    throw Error(ErrorCode::kerFileOpenFailed, "http", serverCode, hostInfo_.Path);
  }
}

HttpIo::HttpIo(const std::string& url, size_t blockSize) {
  p_ = std::make_unique<HttpImpl>(url, blockSize);
}

HttpIo::~HttpIo() = default;
#endif

#ifdef EXV_USE_CURL
//! Internal Pimpl structure of class RemoteIo.
class CurlIo::CurlImpl : public Impl {
 public:
  //! Constructor
  CurlImpl(const std::string& url, size_t blockSize);

  std::unique_ptr<CURL, decltype(&curl_easy_cleanup)> curl_;  //!< libcurl pointer

  // METHODS
  /*!
    @brief Get the length (in bytes) of the remote file.
    @return Return -1 if the size is unknown. Otherwise it returns the length of remote file (in bytes).
    @throw Error if the server returns the error code.
   */
  [[nodiscard]] int64_t getFileLength() const override;
  /*!
    @brief Get the data by range.
    @param startBlock The start block index.
    @param stopBlock The stop block index.
    @param response The data from the server.
    @throw Error if the server returns the error code.
    @note Set startBlock = -1 and stopBlock = -1 to get the whole file content.
   */
  void getDataByRange(size_t startBlock, size_t stopBlock, std::string& response) const override;
  /*!
    @brief Submit the data to the remote machine. The data replace a part of the remote file.
          The replaced part of remote file is indicated by from and to parameters.
    @param data The data are submitted to the remote machine.
    @param size The size of data.
    @param from The start position in the remote file where the data replace.
    @param to The end position in the remote file where the data replace.
    @throw Error if it fails.
    @note The write access is only available on HTTP & HTTPS protocols. The data are submitted to server
          via POST method. It requires the script file on the remote machine to receive the data
          and edit the remote file. The server-side script may be specified with the environment
          string EXIV2_HTTP_POST. The default value is "/exiv2.php". More info is available at
          http://dev.exiv2.org/wiki/exiv2
   */
  void writeRemote(const byte* data, size_t size, size_t from, size_t to) override;

 private:
  long timeout_;  //!< The number of seconds to wait while trying to connect.
};

CurlIo::CurlImpl::CurlImpl(const std::string& url, size_t blockSize) :
    Impl(url, blockSize), curl_(curl_easy_init(), curl_easy_cleanup) {
  // The default block size for FTP is much larger than other protocols
  // the reason is that getDataByRange() in FTP always creates the new connection,
  // so we need the large block size to reduce the overhead of creating the connection.
  if (blockSize_ == 0) {
    blockSize_ = protocol_ == pFtp ? 102400 : 1024;
  }

  std::string timeout = getEnv(envTIMEOUT);
  timeout_ = std::stol(timeout);
  if (timeout_ == 0) {
    throw Error(ErrorCode::kerErrorMessage, "Timeout Environmental Variable must be a positive integer.");
  }
}

int64_t CurlIo::CurlImpl::getFileLength() const {
  curl_easy_reset(curl_.get());  // reset all options
  curl_easy_setopt(curl_.get(), CURLOPT_URL, path_.c_str());
  curl_easy_setopt(curl_.get(), CURLOPT_NOBODY, 1);  // HEAD
  curl_easy_setopt(curl_.get(), CURLOPT_WRITEFUNCTION, curlWriter);
  curl_easy_setopt(curl_.get(), CURLOPT_CONNECTTIMEOUT, timeout_);
  // curl_easy_setopt(curl_.get(), CURLOPT_VERBOSE, 1); // debugging mode

  /* Perform the request, res will get the return code */
  if (auto res = curl_easy_perform(curl_.get()); res != CURLE_OK) {  // error happened
    throw Error(ErrorCode::kerErrorMessage, curl_easy_strerror(res));
  }
  // get status
  int serverCode;
  curl_easy_getinfo(curl_.get(), CURLINFO_RESPONSE_CODE, &serverCode);  // get code
  if (serverCode >= 400 || serverCode < 0) {
    throw Error(ErrorCode::kerFileOpenFailed, "http", serverCode, path_);
  }
  // get length
  curl_off_t temp;
  curl_easy_getinfo(curl_.get(), CURLINFO_CONTENT_LENGTH_DOWNLOAD_T, &temp);  // return -1 if unknown
  return temp;
}

void CurlIo::CurlImpl::getDataByRange(size_t startBlock, size_t stopBlock, std::string& response) const {
  curl_easy_reset(curl_.get());  // reset all options
  curl_easy_setopt(curl_.get(), CURLOPT_URL, path_.c_str());
  curl_easy_setopt(curl_.get(), CURLOPT_NOPROGRESS, 1L);  // no progress meter please
  curl_easy_setopt(curl_.get(), CURLOPT_WRITEFUNCTION, curlWriter);
  curl_easy_setopt(curl_.get(), CURLOPT_WRITEDATA, &response);
  curl_easy_setopt(curl_.get(), CURLOPT_CONNECTTIMEOUT, timeout_);

  // curl_easy_setopt(curl_.get(), CURLOPT_VERBOSE, 1); // debugging mode

  if (startBlock != std::numeric_limits<size_t>::max() && stopBlock != std::numeric_limits<size_t>::max()) {
    auto range = stringFormat("{}-{}", startBlock * blockSize_, (stopBlock * blockSize_) - 1);
    curl_easy_setopt(curl_.get(), CURLOPT_RANGE, range.c_str());
  }

  /* Perform the request, res will get the return code */
  if (auto res = curl_easy_perform(curl_.get()); res != CURLE_OK) {
    throw Error(ErrorCode::kerErrorMessage, curl_easy_strerror(res));
  }
  int serverCode;
  curl_easy_getinfo(curl_.get(), CURLINFO_RESPONSE_CODE, &serverCode);  // get code
  if (serverCode >= 400 || serverCode < 0) {
    throw Error(ErrorCode::kerFileOpenFailed, "http", serverCode, path_);
  }
}

void CurlIo::CurlImpl::writeRemote(const byte* data, size_t size, size_t from, size_t to) {
  std::string scriptPath(getEnv(envHTTPPOST));
  if (scriptPath.empty()) {
    throw Error(ErrorCode::kerErrorMessage,
                "Please set the path of the server script to handle http post data to EXIV2_HTTP_POST "
                "environmental variable.");
  }

  Exiv2::Uri hostInfo = Exiv2::Uri::Parse(path_);

  // add the protocol and host to the path
  if (scriptPath.find("://") == std::string::npos) {
    if (scriptPath.front() != '/')
      scriptPath = "/" + scriptPath;
    scriptPath = hostInfo.Protocol + "://" + hostInfo.Host + scriptPath;
  }

  curl_easy_reset(curl_.get());                           // reset all options
  curl_easy_setopt(curl_.get(), CURLOPT_NOPROGRESS, 1L);  // no progress meter please
  // curl_easy_setopt(curl_.get(), CURLOPT_VERBOSE, 1); // debugging mode
  curl_easy_setopt(curl_.get(), CURLOPT_URL, scriptPath.c_str());

  // encode base64
  size_t encodeLength = (((size + 2) / 3) * 4) + 1;
  auto encodeData = std::make_unique<char[]>(encodeLength);
  base64encode(data, size, encodeData.get(), encodeLength);
  // url encode
  const std::string urlencodeData = urlencode(encodeData.get());
  auto postData = stringFormat("path={}&from={}&to={}&data={}", hostInfo.Path, from, to, urlencodeData);

  curl_easy_setopt(curl_.get(), CURLOPT_POSTFIELDS, postData.c_str());
  // Perform the request, res will get the return code.
  if (auto res = curl_easy_perform(curl_.get()); res != CURLE_OK) {
    throw Error(ErrorCode::kerErrorMessage, curl_easy_strerror(res));
  }
  int serverCode;
  curl_easy_getinfo(curl_.get(), CURLINFO_RESPONSE_CODE, &serverCode);
  if (serverCode >= 400 || serverCode < 0) {
    throw Error(ErrorCode::kerFileOpenFailed, "http", serverCode, path_);
  }
}

size_t CurlIo::write(const byte* data, size_t wcount) {
  if (p_->protocol_ == pHttp || p_->protocol_ == pHttps) {
    return RemoteIo::write(data, wcount);
  }
  throw Error(ErrorCode::kerErrorMessage, "does not support write for this protocol.");
}

size_t CurlIo::write(BasicIo& src) {
  if (p_->protocol_ == pHttp || p_->protocol_ == pHttps) {
    return RemoteIo::write(src);
  }
  throw Error(ErrorCode::kerErrorMessage, "does not support write for this protocol.");
}

CurlIo::CurlIo(const std::string& url, size_t blockSize) {
  p_ = std::make_unique<CurlImpl>(url, blockSize);
}

#endif

// *************************************************************************
// free functions
#ifdef EXV_ENABLE_FILESYSTEM
DataBuf readFile(const std::string& path) {
  FileIo file(path);
  if (file.open("rb") != 0) {
    throw Error(ErrorCode::kerFileOpenFailed, path, "rb", strError());
  }
  DataBuf buf(static_cast<size_t>(fs::file_size(path)));
  if (file.read(buf.data(), buf.size()) != buf.size()) {
    throw Error(ErrorCode::kerCallFailed, path, strError(), "FileIo::read");
  }
  return buf;
}

size_t writeFile(const DataBuf& buf, const std::string& path) {
  FileIo file(path);
  if (file.open("wb") != 0) {
    throw Error(ErrorCode::kerFileOpenFailed, path, "wb", strError());
  }
  return file.write(buf.c_data(), buf.size());
}
#endif

#ifdef EXV_USE_CURL
size_t curlWriter(char* data, size_t size, size_t nmemb, std::string* writerData) {
  if (!writerData)
    return 0;
  writerData->append(data, size * nmemb);
  return size * nmemb;
}
#endif
}  // namespace Exiv2

Coverage Report

Created: 2026-10-28 07:09