/*
   BAREOS® - Backup Archiving REcovery Open Sourced

   Copyright (C) 2000-2012 Free Software Foundation Europe e.V.
   Copyright (C) 2016-2026 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
   Affero 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, May MM
/**
 * @file
 * Append code for Storage daemon
 */

#include "stored/append.h"
#include "stored/askdir.h"
#include "stored/stored.h"
#include "stored/acquire.h"
#include "stored/checkpoint_handler.h"
#include "stored/fd_cmds.h"
#include "stored/stored_globals.h"
#include "stored/stored_jcr_impl.h"
#include "stored/label.h"
#include "stored/spool.h"
#include "lib/bget_msg.h"
#include "lib/edit.h"
#include "include/jcr.h"
#include "include/streams.h"
#include "lib/berrno.h"
#include "lib/crypto.h"
#include "lib/berrno.h"
#include <algorithm>

#include <algorithm>
#include <thread>
#include <variant>
#include <deque>
#include <utility>
#include <condition_variable>
#include "lib/channel.h"

namespace {
/* Responses sent to the daemon */
inline constexpr const char OK_data[] = "3000 OK data\n";
inline constexpr const char OK_append[] = "3000 OK append data\n";
inline constexpr const char OK_replicate[] = "3000 OK replicate data\n";
}  // namespace

namespace storagedaemon {
ProcessedFileData::ProcessedFileData(DeviceRecord* record)
    : volsessionid_(record->VolSessionId)
    , volsessiontime_(record->VolSessionTime)
    , fileindex_(record->FileIndex)
    , stream_(record->Stream)
    , data_len_(record->data_len)
    , data_(record->data, record->data + record->data_len)
{
}

DeviceRecord ProcessedFileData::GetData()
{
  DeviceRecord devicerecord{};
  devicerecord.VolSessionId = volsessionid_;
  devicerecord.VolSessionTime = volsessiontime_;
  devicerecord.FileIndex = fileindex_;
  devicerecord.Stream = stream_;
  devicerecord.data_len = data_len_;
  devicerecord.data = data_.data();

  return devicerecord;
}

ProcessedFile::ProcessedFile(int32_t fileindex) : fileindex_(fileindex) {}

void ProcessedFile::SendAttributesToDirector(JobControlRecord* jcr)
{
  std::for_each(attributes_.begin(), attributes_.end(),
                [jcr](ProcessedFileData& attribute) {
                  DeviceRecord devicerecord = attribute.GetData();
                  SendAttrsToDir(jcr, &devicerecord);
                });
}

void ProcessedFile::AddAttribute(DeviceRecord* record)
{
  attributes_.emplace_back(ProcessedFileData(record));
}

bool IsAttribute(DeviceRecord* record)
{
  return record->maskedStream == STREAM_UNIX_ATTRIBUTES
         || record->maskedStream == STREAM_UNIX_ATTRIBUTES_EX
         || record->maskedStream == STREAM_RESTORE_OBJECT
         || CryptoDigestStreamType(record->maskedStream) != CRYPTO_DIGEST_NONE;
}

static bool SaveFullyProcessedFilesAttributes(
    JobControlRecord* jcr,
    std::vector<ProcessedFile>& processed_files)
{
  if (!processed_files.empty()) {
    std::for_each(
        processed_files.begin(), processed_files.end(),
        [jcr](ProcessedFile& file) { file.SendAttributesToDirector(jcr); });
    jcr->JobFiles = processed_files.back().GetFileIndex();
    processed_files.clear();
    return true;
  }
  return false;
}

class MessageHandler {
 public:
  using signal_type = int;

  struct message_type {
    std::size_t size;
    PoolMem data;
  };

  struct error_type {
    enum class type
    {
      HARDEOF,
      // both ERROR and SOCKET_ERROR are taken by windows.h
      INTERNAL_ERROR,
    } type;

    std::string msg;
  };

  using result_type = std::variant<signal_type, message_type, error_type>;

  MessageHandler(BareosSocket* t_fd)
      : MessageHandler{t_fd,
                       // 500 msg reserves at most 256MB in size
                       // probably much less because of signals
                       channel::CreateBufferedChannel<result_type>(500)}
  {
  }

  std::optional<result_type> get_msg() { return output.get(); }

  const char* error()
  {
    if (fd->IsError()) { return fd->bstrerror(); }
    return nullptr;
  }

  BareosSocket* close_and_get_sock()
  {
    output.close();
    receive_thread.join();
    return fd;
  }

 private:
  MessageHandler(BareosSocket* t_fd,
                 std::pair<channel::input<result_type>,
                           channel::output<result_type>> chan_pair)
      : fd{t_fd}
      , input{std::move(chan_pair.first)}
      , output{std::move(chan_pair.second)}
      , receive_thread{enlist, this}
  {
  }

  BareosSocket* fd;
  channel::input<result_type> input;
  channel::output<result_type> output;

  // receive_thread has to be defined last!
  // The thread created will try to access this class immediately after
  // being created!  As such everything else has to be initialized.
  std::thread receive_thread;
  void do_work()
  {
    POOLMEM* save = fd->msg;
    bool cont = true;
    for (int res = 0; cont; res = fd->WaitData(0, 100'000)) {
      if (res == fd->DataAvailable) {
        PoolMem msg(PM_MESSAGE);
        fd->msg = msg.addr();
        result_type result;
        int n = BgetMsg(fd);
        // fd->msg might have been relocated
        msg.addr() = fd->msg;
        if (n < 0) {
          if (n == BNET_SIGNAL) {
            result = signal_type{fd->message_length};
            // break; /* end of data */
          } else if (n == BNET_HARDEOF) {
            result = error_type{error_type::type::HARDEOF, fd->bstrerror()};
            cont = false;
          } else {
            result
                = error_type{error_type::type::INTERNAL_ERROR, fd->bstrerror()};
            cont = false;
          }
        } else {
          std::size_t length = n;
          result = message_type{length, std::move(msg)};
        }
        fd->msg = nullptr;

        if (!input.emplace(std::move(result))) {
          if (input.closed()) {
            Dmsg1(20, "Tried to put message into closed queue.\n");
          } else {
            Dmsg1(20,
                  "Tried to put message into queue; but it did not succeed.\n");
          }
          cont = false;
        }
      } else if (res == fd->Error) {
        cont = false;
      } else {
        ASSERT(res == fd->Timeout);
        input.try_update_status();
      }

      if (input.closed()) { cont = false; }
    }

    input.close();

    fd->msg = save;
  }

  static void enlist(MessageHandler* handler) { handler->do_work(); }
};

static bool SetupDCR(JobControlRecord* jcr,
                     std::int64_t& volid,
                     uint32_t& blocknum)
{
  if (!jcr->sd_impl->dcr) {
    jcr->sendJobStatus(JS_WaitSD);
    if (TryReserveAfterUse(jcr, true)) {
      Jmsg(jcr, M_INFO, 0, T_("Using Device %s to write.\n"),
           jcr->sd_impl->dcr->dev->print_name());
    } else {
      FreeDeviceControlRecord(jcr->sd_impl->dcr);
      jcr->sd_impl->dcr = nullptr;
      Jmsg(jcr, M_FATAL, 0, T_("Could not reserve any device for this job.\n"));
      return false;
    }
  }

  if (!jcr->sd_impl->dcr) {
    Jmsg0(jcr, M_FATAL, 0, T_("DeviceControlRecord is NULL!!!\n"));
    return false;
  }
  auto* dev = jcr->sd_impl->dcr->dev;
  if (!dev) {
    Jmsg0(jcr, M_FATAL, 0, T_("Device is NULL!!!\n"));
    return false;
  }

  Dmsg1(100, "Start append data. res=%d\n", dev->NumReserved());

  if (!AcquireDeviceForAppend(jcr->sd_impl->dcr)) {
    jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
    return false;
  }

  if (GeneratePluginEvent(jcr, bSdEventSetupRecordTranslation,
                          jcr->sd_impl->dcr)
      != bRC_OK) {
    jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
    return false;
  }

  jcr->sendJobStatus(JS_Running);

  if (dev->VolCatInfo.VolCatName[0] == 0) {
    Pmsg0(000, T_("NULL Volume name. This shouldn't happen!!!\n"));
    return false;
  }
  Dmsg1(50, "Begin append device=%s\n", dev->print_name());

  if (!BeginDataSpool(jcr->sd_impl->dcr)) {
    jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
    return false;
  }

  if (!BeginAttributeSpool(jcr)) {
    DiscardDataSpool(jcr->sd_impl->dcr);
    jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
    return false;
  }

  Dmsg0(100, "Just after AcquireDeviceForAppend\n");
  if (dev->VolCatInfo.VolCatName[0] == 0) {
    Pmsg0(000, T_("NULL Volume name. This shouldn't happen!!!\n"));
    return false;
  }

  // Write Begin Session Record
  if (!WriteSessionLabel(jcr->sd_impl->dcr, SOS_LABEL)) {
    Jmsg1(jcr, M_FATAL, 0, T_("Write session label failed. ERR=%s\n"),
          dev->bstrerror());
    jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
  }

  if (dev->VolCatInfo.VolCatName[0] == 0) {
    Pmsg0(000, T_("NULL Volume name. This shouldn't happen!!!\n"));
    return false;
  }

  jcr->sd_impl->dcr->VolFirstIndex = jcr->sd_impl->dcr->VolLastIndex = 0;

  volid = jcr->sd_impl->dcr->VolMediaId;

  blocknum = jcr->sd_impl->dcr->block->BlockNumber;

  return true;
}

// Append Data sent from File daemon
bool DoAppendData(JobControlRecord* jcr, BareosSocket* bs, const char* what)
{
  int32_t n, file_index, stream, last_file_index, job_elapsed;
  bool ok = true;
  char buf1[100];
  char ec[50];

  if (me->enable_ktls) {
    bool ktls_enabled = bs->KtlsForRecv();
    Jmsg(jcr, M_INFO, 0, "Receiving via kTLS: %s\n",
         ktls_enabled ? "yes" : "no");
  }

  int64_t current_volumeid = 0;
  uint32_t current_block_number = 0;

  if (jcr->sd_impl->dcr) {
    // if the device was already reserved, we will now try to acquire it

    // first we set the buffer size; this is only done with advance reservation
    // since it does not make sense to do it after data starts arriving

    if (!bs->SetBufferSize(me->max_network_buffer_size, BNET_SETBUF_WRITE)) {
      Jmsg0(jcr, M_FATAL, 0, T_("Unable to set network buffer size.\n"));
      jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
      return false;
    }

    if (!SetupDCR(jcr, current_volumeid, current_block_number)) {
      Jmsg(jcr, M_FATAL, 0,
           T_("Unable to setup pre reserved dcr for this job.\n"));
      return false;
    }
  }

  // Tell daemon to send data
  if (!bs->fsend(OK_data)) {
    BErrNo be;
    Jmsg2(jcr, M_FATAL, 0, T_("Network send error to %s. ERR=%s\n"), what,
          be.bstrerror(bs->b_errno));
    ok = false;
  }

  /* Get Data from daemon, write to device.  To clarify what is
   * going on here.  We expect:
   * - A stream header
   * - Multiple records of data
   * - EOD record
   *
   * The Stream header is just used to synchronize things, and
   * none of the stream header is written to tape.
   * The Multiple records of data, contain first the Attributes,
   * then after another stream header, the file data, then
   * after another stream header, the MD5 data if any.
   *
   * So we get the (stream header, data, EOD) three time for each
   * file. 1. for the Attributes, 2. for the file data if any,
   * and 3. for the MD5 if any. */
  jcr->run_time = time(NULL); /* start counting time for rates */

  const bool checkpoints_enabled = me->checkpoint_interval > 0;
  CheckpointHandler checkpoint_handler(me->checkpoint_interval);

  std::vector<ProcessedFile> processed_files{};

  ProcessedFile file_currently_processed;

  // we need to clone here as we need to send heartbeats in case of
  // just in time reservation.  Keep in mind that from this point forward
  // trying to read from the filedaemon socked is still forbidden.  Only
  // writing is ok!


  auto* cloned = bs->clone();
  if (cloned->fd_ < 0) {
    Jmsg2(jcr, M_FATAL, 0, T_("Could not clone socket (fd = %d): %s\n"),
          cloned->fd_, cloned->errmsg);
    return false;
  }
  MessageHandler handler(cloned);

  for (last_file_index = 0; ok && !jcr->IsJobCanceled();) {
    /* Read Stream header from the daemon.
     *
     * The stream header consists of the following:
     * - file_index (sequential Bareos file index, base 1)
     * - stream     (Bareos number to distinguish parts of data)
     * - info       (Info for Storage daemon -- compressed, encrypted, ...)
     *               info is not currently used, so is read, but ignored! */
    auto msg = handler.get_msg();
    if (!msg) {
      Jmsg2(jcr, M_FATAL, 0,
            T_("Internal Error reading data header from %s.\n"), what);
      ok = false;
      break;
    }

    using signal_type = MessageHandler::signal_type;
    using message_type = MessageHandler::message_type;
    using error_type = MessageHandler::error_type;

    if (auto* error = std::get_if<error_type>(&msg.value())) {
      Jmsg2(jcr, M_FATAL, 0, T_("Error reading data header from %s. ERR=%s\n"),
            what, error->msg.c_str());
      ok = false;
      break;
    }

    if (auto* signal = std::get_if<signal_type>(&msg.value())) {
      if (*signal != BNET_EOD) {
        Jmsg2(jcr, M_FATAL, 0, T_("Unexpected signal from %s: %d\n"), what,
              *signal);
        ok = false;
      }
      break;
    }

    auto content = std::get<message_type>(std::move(msg).value());
    n = content.size;

    if (sscanf(content.data.c_str(), "%ld %ld", &file_index, &stream) != 2) {
      Jmsg2(jcr, M_FATAL, 0, T_("Malformed data header from %s: %s\n"), what,
            content.data.c_str());
      ok = false;
      break;
    }

    Dmsg2(890, "<filed: Header FilInx=%d stream=%d\n", file_index, stream);

    /* We make sure the file_index is advancing sequentially.
     * An incomplete job can start the file_index at any number.
     * otherwise, it must start at 1. */

    bool incomplete_job_rerun_fileindex_positive
        = jcr->rerunning && file_index > 0 && last_file_index == 0;
    bool fileindex_is_sequential = file_index > 0
                                   && (file_index == last_file_index
                                       || file_index == last_file_index + 1);

    if (!incomplete_job_rerun_fileindex_positive && !fileindex_is_sequential) {
      Jmsg3(jcr, M_FATAL, 0,
            T_("FileIndex=%d from %s not positive or sequential=%d\n"),
            file_index, what, last_file_index);
      ok = false;
      break;
    }

    if (file_index != last_file_index) {
      last_file_index = file_index;
      if (file_currently_processed.GetFileIndex() > 0) {
        processed_files.push_back(std::move(file_currently_processed));
      }
      file_currently_processed = ProcessedFile{file_index};
    }

    /* Read data stream from the daemon. The data stream is just raw bytes.
     * We save the original data pointer from the record so we can restore
     * that after the loop ends. */
    POOLMEM* rec_data = nullptr;
    while (!jcr->IsJobCanceled()) {
      auto msg2 = handler.get_msg();

      if (!msg2) {
        Jmsg2(jcr, M_FATAL, 0, T_("Internal Error reading data from %s.\n"),
              what);
        ok = false;
        break;
      }

      if (auto* error = std::get_if<error_type>(&msg2.value())) {
        Jmsg2(jcr, M_FATAL, 0,
              T_("Error reading data header from %s. ERR=%s\n"), what,
              error->msg.c_str());
        ok = false;
        break;
      }

      if (auto* signal = std::get_if<signal_type>(&msg2.value())) {
        if (*signal != BNET_EOD) {
          Jmsg2(jcr, M_FATAL, 0, T_("Unexpected signal from %s: %d\n"), what,
                *signal);
          ok = false;
        }
        break;
      }

      auto content2 = std::get<message_type>(std::move(msg2).value());
      n = content2.size;

      if (!jcr->sd_impl->dcr) {
        Jmsg(jcr, M_INFO, 0,
             T_("JustInTime Reservation: Finding drive to reserve.\n"));
        if (!SetupDCR(jcr, current_volumeid, current_block_number)) {
          Jmsg(jcr, M_FATAL, 0, T_("Unable to setup device for this job.\n"));
          ok = false;
          break;
        } else {
          // we just now got a device for the first time this job,
          // but the checkpoint timer was running for this whole time already
          // this does not really make sense, so we should "restart"
          // the timer.

          if (checkpoints_enabled) { checkpoint_handler.ResetTimer(); }
        }
      }

      if (rec_data == nullptr) { rec_data = jcr->sd_impl->dcr->rec->data; }
      jcr->sd_impl->dcr->rec->VolSessionId = jcr->VolSessionId;
      jcr->sd_impl->dcr->rec->VolSessionTime = jcr->VolSessionTime;
      jcr->sd_impl->dcr->rec->FileIndex = file_index;
      jcr->sd_impl->dcr->rec->Stream = stream;
      jcr->sd_impl->dcr->rec->maskedStream
          = stream & STREAMMASK_TYPE; /* strip high bits */
      jcr->sd_impl->dcr->rec->data_len = content2.size;
      jcr->sd_impl->dcr->rec->data
          = content2.data.addr(); /* use message buffer */

      Dmsg4(850, "before writ_rec FI=%d SessId=%d Strm=%s len=%d\n",
            jcr->sd_impl->dcr->rec->FileIndex,
            jcr->sd_impl->dcr->rec->VolSessionId,
            stream_to_ascii(buf1, jcr->sd_impl->dcr->rec->Stream,
                            jcr->sd_impl->dcr->rec->FileIndex),
            jcr->sd_impl->dcr->rec->data_len);

      ok = jcr->sd_impl->dcr->WriteRecord();
      if (!ok) {
        Dmsg2(90, "Got WriteBlockToDev error on device %s. %s\n",
              jcr->sd_impl->dcr->dev->print_name(),
              jcr->sd_impl->dcr->dev->bstrerror());
        break;
      }

      if (IsAttribute(jcr->sd_impl->dcr->rec)) {
        file_currently_processed.AddAttribute(jcr->sd_impl->dcr->rec);
      }

      const bool block_changed
          = current_block_number != jcr->sd_impl->dcr->block->BlockNumber;
      const bool volume_changed
          = jcr->sd_impl->dcr->VolMediaId != current_volumeid && block_changed;

      if (AttributesAreSpooled(jcr)) {
        SaveFullyProcessedFilesAttributes(jcr, processed_files);
      } else {
        if (block_changed) {
          current_block_number = jcr->sd_impl->dcr->block->BlockNumber;
          if (SaveFullyProcessedFilesAttributes(jcr, processed_files)) {
            if (checkpoints_enabled) {
              checkpoint_handler.SetReadyForCheckpoint();
            }
          }
        }

        if (checkpoints_enabled && checkpoint_handler.IsReadyForCheckpoint()) {
          if (volume_changed) {
            checkpoint_handler.DoVolumeChangeBackupCheckpoint(jcr);
            current_volumeid = jcr->sd_impl->dcr->VolMediaId;
          } else {
            checkpoint_handler.DoTimedCheckpoint(jcr);
          }
        }
      }

      Dmsg0(650, "Enter bnet_get\n");
    }
    Dmsg2(650, "End read loop with %s. Stat=%d\n", what, n);

    // Restore the original data pointer.
    if (rec_data) { jcr->sd_impl->dcr->rec->data = rec_data; }

    if (auto* error = handler.error()) {
      if (!jcr->IsJobCanceled()) {
        Dmsg2(350, "Network read error from %s. ERR=%s\n", what, error);
        Jmsg2(jcr, M_FATAL, 0, T_("Network error reading from %s. ERR=%s\n"),
              what, error);
      }

      ok = false;
      break;
    }
  }

  {
    // delete the copy
    auto* copy = handler.close_and_get_sock();
    copy->close();
    delete copy;
  }

  // Create Job status for end of session label
  jcr->setJobStatusWithPriorityCheck(ok ? JS_Terminated : JS_ErrorTerminated);

  if (ok && bs == jcr->file_bsock) {
    // Terminate connection with FD
    bs->fsend(OK_append);
    DoFdCommands(jcr); /* finish dialog with FD */
  } else if (bs == jcr->store_bsock) {
    bs->fsend(OK_replicate);
  } else {
    bs->fsend("3999 Failed append\n");
  }

  if (jcr->sd_impl->dcr) {
    Dmsg1(200, "Write EOS label JobStatus=%c\n", jcr->getJobStatus());

    /* Check if we can still write. This may not be the case
     * if we are at the end of the tape or we got a fatal I/O error. */
    if (ok || jcr->sd_impl->dcr->dev->CanWrite()) {
      // take into account the fact that GetFileIndex() may return -1
      // if no file is currently getting processed.
      // This should only happen if no file was send to begin with!
      jcr->JobFiles = std::max(file_currently_processed.GetFileIndex(), 0);
      if (!WriteSessionLabel(jcr->sd_impl->dcr, EOS_LABEL)) {
        // Print only if ok and not cancelled to avoid spurious messages
        if (ok && !jcr->IsJobCanceled()) {
          Jmsg1(jcr, M_FATAL, 0,
                T_("Error writing end session label. ERR=%s\n"),
                jcr->sd_impl->dcr->dev->bstrerror());
        }
        jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
        ok = false;
      }
      Dmsg0(90, "back from write_end_session_label()\n");

      // Flush out final partial block of this session
      if (!jcr->sd_impl->dcr->WriteBlockToDevice()) {
        // Print only if ok and not cancelled to avoid spurious messages
        if (ok && !jcr->IsJobCanceled()) {
          Jmsg2(jcr, M_FATAL, 0,
                T_("Fatal append error on device %s: ERR=%s\n"),
                jcr->sd_impl->dcr->dev->print_name(),
                jcr->sd_impl->dcr->dev->bstrerror());
          Dmsg0(100, T_("Set ok=FALSE after WriteBlockToDevice.\n"));
        }
        jcr->setJobStatusWithPriorityCheck(JS_ErrorTerminated);
        ok = false;
      } else if (ok && !jcr->IsJobCanceled()) {
        // Send attributes of the final partial block of the session
        if (file_currently_processed.GetFileIndex() > 0) {
          processed_files.push_back(std::move(file_currently_processed));
        }
        SaveFullyProcessedFilesAttributes(jcr, processed_files);
      }
    }

    if (!ok && !jcr->is_JobStatus(JS_Incomplete)) {
      DiscardDataSpool(jcr->sd_impl->dcr);
    } else {
      // Note: if commit is OK, the device will remain blocked
      CommitDataSpool(jcr->sd_impl->dcr);
    }

    // Release the device -- and send final Vol info to DIR and unlock it.
    ReleaseDevice(jcr->sd_impl->dcr);
  } else if (ok) {
    Jmsg(jcr, M_INFO, 0,
         "Because no backup data was received, no device was reserved. As such "
         "no Session Labels were written for this job.\n");
    Dmsg0(50, "No data for job %d => no data written.\n", jcr->JobId);
  }

  if (!DeleteNullJobmediaRecords(jcr)) {
    Jmsg(jcr, M_WARNING, 0,
         T_("Could not delete placeholder media records.\n"));
  }

  /* Don't use time_t for job_elapsed as time_t can be 32 or 64 bits,
   * and the subsequent Jmsg() editing will break */
  job_elapsed = time(NULL) - jcr->run_time;
  if (job_elapsed <= 0) { job_elapsed = 1; }

  Jmsg(jcr, M_INFO, 0,
       T_("Elapsed time=%02d:%02d:%02d, Transfer rate=%s Bytes/second\n"),
       job_elapsed / 3600, job_elapsed % 3600 / 60, job_elapsed % 60,
       edit_uint64_with_suffix(jcr->JobBytes / job_elapsed, ec));

  if ((!ok || jcr->IsJobCanceled()) && !jcr->is_JobStatus(JS_Incomplete)) {
    DiscardAttributeSpool(jcr);
  } else {
    CommitAttributeSpool(jcr);
  }

  jcr->sendJobStatus(); /* update director */

  Dmsg1(100, "return from DoAppendData() ok=%d\n", ok);
  return ok;
}

// Send attributes and digest to Director for Catalog
bool SendAttrsToDir(JobControlRecord* jcr, DeviceRecord* rec)
{
  if (!jcr->sd_impl->no_attributes) {
    BareosSocket* dir = jcr->dir_bsock;
    if (AttributesAreSpooled(jcr)) { dir->SetSpooling(); }
    Dmsg0(850, "Send attributes to dir.\n");
    if (!jcr->sd_impl->dcr->DirUpdateFileAttributes(rec)) {
      Jmsg(jcr, M_FATAL, 0, T_("Error updating file attributes. ERR=%s\n"),
           dir->bstrerror());
      dir->ClearSpooling();
      return false;
    }
    dir->ClearSpooling();
  }
  return true;
}
} /* namespace storagedaemon */
