/*
   BAREOS® - Backup Archiving REcovery Open Sourced

   Copyright (C) 2000-2012 Free Software Foundation Europe e.V.
   Copyright (C) 2013-2013 Planets Communications B.V.
   Copyright (C) 2013-2025 Bareos GmbH & Co. KG

   This program is Free Software; you can redistribute it and/or
   modify it under the terms of version three of the GNU Affero General Public
   License as published by the Free Software Foundation and included
   in the file LICENSE.

   This program 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
   General Public License for more details.

   You should have received a copy of the GNU Affero General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
   02110-1301, USA.
*/
/*
 * Kern Sibbald, MM
 * Extracted from other source files Marco van Wieringen, December 2013
 */
/**
 * @file
 * UNIX FILE API device abstraction.
 */

#include <unistd.h>

#include "include/fcntl_def.h"
#include "include/bareos.h"
#include "stored/stored.h"
#include "stored/stored_globals.h"
#include "stored/sd_backends.h"
#include "stored/device_control_record.h"
#include "unix_file_device.h"
#include "lib/berrno.h"
#include "lib/util.h"

namespace storagedaemon {

// (Un)mount the device (For a FILE device)
static bool do_mount(DeviceControlRecord* dcr, bool mount, int dotimeout)
{
  DeviceResource* device_resource = dcr->dev->device_resource;
  PoolMem ocmd(PM_FNAME);
  POOLMEM* results;
  DIR* dp;
  char* icmd;
  struct dirent* result;
#ifdef USE_READDIR_R
  struct dirent* entry;
#endif
  int status, tries, name_max;
  int count = 0;
  BErrNo be;

  if (mount) {
    icmd = device_resource->mount_command;
  } else {
    icmd = device_resource->unmount_command;
  }

  dcr->dev->EditMountCodes(ocmd, icmd);

  Dmsg2(100, "do_mount: cmd=%s mounted=%d\n", ocmd.c_str(),
        dcr->dev->IsMounted());

  if (dotimeout) {
    /* Try at most 10 times to (un)mount the device. This should perhaps be
     * configurable. */
    tries = 10;
  } else {
    tries = 1;
  }
  results = GetMemory(4000);

  /* If busy retry each second */
  Dmsg1(100, "do_mount run_prog=%s\n", ocmd.c_str());
  while ((status = RunProgramFullOutput(ocmd.c_str(),
                                        dcr->dev->max_open_wait / 2, results))
         != 0) {
    /* Doesn't work with internationalization (This is not a problem) */
    if (mount && fnmatch("*is already mounted on*", results, 0) == 0) { break; }
    if (!mount && fnmatch("* not mounted*", results, 0) == 0) { break; }
    if (tries-- > 0) {
      /* Sometimes the device cannot be mounted because it is already mounted.
       * Try to unmount it, then remount it */
      if (mount) {
        Dmsg1(400, "Trying to unmount the device %s...\n",
              dcr->dev->print_name());
        do_mount(dcr, 0, 0);
      }
      Bmicrosleep(1, 0);
      continue;
    }
    Dmsg5(100, "Device %s cannot be %smounted. status=%d result=%s ERR=%s\n",
          dcr->dev->print_name(), (mount ? "" : "un"), status, results,
          be.bstrerror(status));
    Mmsg(dcr->dev->errmsg, T_("Device %s cannot be %smounted. ERR=%s\n"),
         dcr->dev->print_name(), (mount ? "" : "un"), be.bstrerror(status));

    // Now, just to be sure it is not mounted, try to read the filesystem.
    name_max = pathconf(".", _PC_NAME_MAX);
    if (name_max < 1024) { name_max = 1024; }

    if (!(dp = opendir(device_resource->mount_point))) {
      BErrNo be2;
      dcr->dev->dev_errno = errno;
      Dmsg3(100, "do_mount: failed to open dir %s (dev=%s), ERR=%s\n",
            device_resource->mount_point, dcr->dev->print_name(),
            be2.bstrerror());
      goto get_out;
    }

#ifdef USE_READDIR_R
    entry = (struct dirent*)malloc(sizeof(struct dirent) + name_max + 1000);
    while (1) {
      if ((Readdir_r(dp, entry, &result) != 0) || (result == NULL)) {
#else
    while (1) {
      result = readdir(dp);
      if (result == NULL) {
#endif
        dcr->dev->dev_errno = EIO;
        Dmsg2(129,
              "do_mount: failed to find suitable file in dir %s (dev=%s)\n",
              device_resource->mount_point, dcr->dev->print_name());
        break;
      }
      if (!bstrcmp(result->d_name, ".") && !bstrcmp(result->d_name, "..")
          && !bstrcmp(result->d_name, ".keep")) {
        count++; /* result->d_name != ., .. or .keep (Gentoo-specific) */
        break;
      } else {
        Dmsg2(129, "do_mount: ignoring %s in %s\n", result->d_name,
              device_resource->mount_point);
      }
    }
#ifdef USE_READDIR_R
    free(entry);
#endif
    closedir(dp);

    Dmsg1(100,
          "do_mount: got %d files in the mount point (not counting ., .. and "
          ".keep)\n",
          count);

    if (count > 0) {
      /* If we got more than ., .. and .keep */
      /*   there must be something mounted */
      if (mount) {
        Dmsg1(100, "Did Mount by count=%d\n", count);
        break;
      } else {
        /* An unmount request. We failed to unmount - report an error */
        FreePoolMemory(results);
        Dmsg0(200, "== error mount=1 wanted unmount\n");
        return false;
      }
    }
  get_out:
    FreePoolMemory(results);
    Dmsg0(200, "============ mount=0\n");
    return false;
  }

  FreePoolMemory(results);
  Dmsg1(200, "============ mount=%d\n", mount);
  return true;
}

/**
 * Mount the device.
 *
 * If timeout, wait until the mount command returns 0.
 * If !timeout, try to mount the device only once.
 */
bool unix_file_device::MountBackend(DeviceControlRecord* dcr, int timeout)
{
  bool retval = true;

  if (RequiresMount() && device_resource->mount_command) {
    retval = do_mount(dcr, true, timeout);
  }

  return retval;
}

/**
 * Unmount the device
 *
 * If timeout, wait until the unmount command returns 0.
 * If !timeout, try to unmount the device only once.
 */
bool unix_file_device::UnmountBackend(DeviceControlRecord* dcr, int timeout)
{
  bool retval = true;

  if (RequiresMount() && device_resource->unmount_command) {
    retval = do_mount(dcr, false, timeout);
  }

  return retval;
}

bool unix_file_device::ScanForVolumeImpl(DeviceControlRecord* dcr)
{
  return ScanDirectoryForVolume(dcr);
}

int unix_file_device::d_open(const char* pathname, int flags, int mode)
{
  return ::open(pathname, flags, mode);
}

ssize_t unix_file_device::d_read(int t_fd, void* buffer, size_t count)
{
  return ::read(t_fd, buffer, count);
}

ssize_t unix_file_device::d_write(int t_fd, const void* buffer, size_t count)
{
  return ::write(t_fd, buffer, count);
}

int unix_file_device::d_close(int t_fd) { return ::close(t_fd); }

int unix_file_device::d_ioctl(int, ioctl_req_t, char*) { return -1; }

boffset_t unix_file_device::d_lseek(DeviceControlRecord*,
                                    boffset_t offset,
                                    int whence)
{
  return ::lseek(fd, offset, whence);
}

bool unix_file_device::d_truncate(DeviceControlRecord* dcr)
{
  struct stat st;
  PoolMem archive_name(PM_FNAME);

  // When secure erase is configured never truncate the file.
  if (!me->secure_erase_cmdline) {
    if (ftruncate(fd, 0) != 0) {
      BErrNo be;

      Mmsg2(errmsg, T_("Unable to truncate device %s. ERR=%s\n"), prt_name,
            be.bstrerror());
      return false;
    }

    if (fstat(fd, &st) != 0) {
      BErrNo be;

      Mmsg2(errmsg, T_("Unable to stat device %s. ERR=%s\n"), prt_name,
            be.bstrerror());
      return false;
    }

    if (st.st_size == 0) { goto bail_out; }

    Mmsg2(errmsg,
          T_("Device %s doesn't support ftruncate(). Recreating file %s.\n"),
          prt_name, archive_name.c_str());
  } else {
    if (fstat(fd, &st) != 0) {
      BErrNo be;

      Mmsg2(errmsg, T_("Unable to stat device %s. ERR=%s\n"), prt_name,
            be.bstrerror());
      return false;
    }
  }

  /* Workaround supplied by Martin Schmid as a solution to bug #1011.
   * Used when secure erase is configured or when ftruncate() doesn't have the
   * wanted result. As work around we perform the following steps:
   *
   * 1. close file
   * 2. delete file
   * 3. open new file with same mode
   * 4. change ownership to original */
  PmStrcpy(archive_name, archive_device_string);
  if (!IsPathSeparator(
          archive_name.c_str()[strlen(archive_name.c_str()) - 1])) {
    PmStrcat(archive_name, "/");
  }
  PmStrcat(archive_name, dcr->VolumeName);

  // Close file and blow it away
  ::close(fd);
  SecureErase(dcr->jcr, archive_name.c_str());

  // Recreate the file -- of course, empty
  oflags = O_CREAT | O_RDWR | O_BINARY;
  if ((fd = ::open(archive_name.c_str(), oflags, st.st_mode)) < 0) {
    BErrNo be;

    dev_errno = errno;
    Mmsg2(errmsg, T_("Could not reopen: %s, ERR=%s\n"), archive_name.c_str(),
          be.bstrerror());
    Emsg0(M_FATAL, 0, "%s", errmsg);

    return false;
  }

  // Reset proper owner
  (void)!chown(archive_name.c_str(), st.st_uid, st.st_gid);

bail_out:
  return true;
}

REGISTER_SD_BACKEND(file, unix_file_device);

} /* namespace storagedaemon  */
