fsfw/src/fsfw/cfdp/handler/SourceHandler.cpp

#include "SourceHandler.h"

#include <etl/crc32.h>

#include <array>

#include "fsfw/filesystem/HasFileSystemIF.h"

using namespace returnvalue;

cfdp::SourceHandler::SourceHandler(SourceHandlerParams params, FsfwParams fsfwParams)
    : transactionParams(params.maxFilePathSize),
      sourceParams(std::move(params)),
      fsfwParams(fsfwParams) {}

cfdp::SourceHandler::FsmResult& cfdp::SourceHandler::fsmNacked() {
  ReturnValue_t result;
  if (step == TransactionStep::IDLE) {
    step = TransactionStep::TRANSACTION_START;
  }
  if (step == TransactionStep::TRANSACTION_START) {
    step = TransactionStep::CRC_PROCEDURE;
  }
  if (step == TransactionStep::CRC_PROCEDURE) {
    result = checksumGeneration();
    if (result != OK) {
      // TODO: Some error handling
    }
    step = TransactionStep::SENDING_METADATA;
  }
  if (step == TransactionStep::SENDING_METADATA) {
    result = prepareAndSendMetadataPdu();
    if (result != OK) {
      // TODO: Error handling
    }
    return fsmResult;
  }
  if (step == TransactionStep::SENDING_FILE_DATA) {
    result = prepareAndSendNextFileDataPdu();
    if (result != OK) {
      // TODO: Error handling
    }
    return fsmResult;
  }
  if (step == TransactionStep::SENDING_EOF) {
    result = prepareAndSendEofPdu();
    if (result != OK) {
      // TODO: Error handling
    }
    return fsmResult;
  }
  if (step == TransactionStep::WAIT_FOR_FINISH) {
    // TODO: In case this is a request with closure, wait for finish.
    // Done, issue notice of completion
    step = TransactionStep::NOTICE_OF_COMPLETION;
  }
  if (step == TransactionStep::NOTICE_OF_COMPLETION) {
    // TODO: Notice of completion
    // We are done, go back to idle state.
    // TODO: Possible reset state?
    step = TransactionStep::IDLE;
    state = CfdpState::IDLE;
  }
  return fsmResult;
}

cfdp::SourceHandler::FsmResult& cfdp::SourceHandler::stateMachine() {
  if (state == cfdp::CfdpState::IDLE) {
    return fsmResult;
  }
  if (state == cfdp::CfdpState::BUSY_CLASS_1_NACKED) {
    return fsmNacked();
  }
  return fsmResult;
}

ReturnValue_t cfdp::SourceHandler::checksumGeneration() {
  std::array<uint8_t, 1024> buf{};
  etl::crc32 crcCalc;
  uint64_t currentOffset = 0;
  FileOpParams params(transactionParams.destName.data(), transactionParams.fileSize.value());
  while (currentOffset < transactionParams.fileSize.value()) {
    uint64_t readLen;
    if (currentOffset + buf.size() > transactionParams.fileSize.value()) {
      readLen = transactionParams.fileSize.value() - currentOffset;
    } else {
      readLen = buf.size();
    }
    if (readLen > 0) {
      params.offset = currentOffset;
      params.size = readLen;
      auto result = sourceParams.user.vfs.readFromFile(params, buf.data(), buf.size());
      if (result != OK) {
        // TODO: I think this is a case for a filestore rejection, but it might sense to print
        //       a warning or trigger an event because this should generally not happen
        return FAILED;
      }
      crcCalc.add(buf.begin(), buf.begin() + readLen);
    }
    currentOffset += readLen;
  }

  transactionParams.crc = crcCalc.value();
  return OK;
}

ReturnValue_t cfdp::SourceHandler::putRequest(PutRequestFull& putRequest, RemoteEntityCfg& cfg) {
  if (state != CfdpState::IDLE) {
    return SOURCE_TRANSACTION_PENDING;
  }
  transactionParams.closureRequested = putRequest.closureRequested;
  transactionParams.destId = putRequest.destId;
  transactionParams.transmissionMode = putRequest.transmissionMode;
  if (transactionParams.transmissionMode == TransmissionMode::ACKNOWLEDGED) {
    state = cfdp::CfdpState::BUSY_CLASS_2_ACKED;
  } else if (transactionParams.transmissionMode == TransmissionMode::UNACKNOWLEDGED) {
    state = cfdp::CfdpState::BUSY_CLASS_1_NACKED;
  }
  step = TransactionStep::IDLE;
  if (transactionParams.transmissionMode == TransmissionMode::ACKNOWLEDGED) {
  }
  if (putRequest.sourceNameSize > transactionParams.sourceName.size()) {
    return FAILED;
  }
  std::memcpy(transactionParams.sourceName.data(), putRequest.sourceName,
              putRequest.sourceNameSize);
  if (putRequest.destNameSize > transactionParams.destName.size()) {
    return FAILED;
  }
  std::memcpy(transactionParams.destName.data(), putRequest.destName, putRequest.destNameSize);
  currentRemoteCfg = cfg;
  return OK;
}

ReturnValue_t cfdp::SourceHandler::prepareAndSendMetadataPdu() {
  // TODO: Implement
  // Advance FSM if everythings works
  step = TransactionStep::SENDING_FILE_DATA;
  return OK;
}

ReturnValue_t cfdp::SourceHandler::prepareAndSendNextFileDataPdu() {
  // TODO: Implement
  // Advance FSM after all file data PDUs were sent
  step = TransactionStep::SENDING_EOF;
  return OK;
}

ReturnValue_t cfdp::SourceHandler::prepareAndSendEofPdu() {
  // TODO: Implement
  step = TransactionStep::WAIT_FOR_FINISH;
  return OK;
}