/*

 * Copyright 2023 Adam.Chen <Adam.Chen@genesyslogic.com.tw>

 *

 * SPDX-License-Identifier: LGPL-2.1-or-later

 */

#include "config.h"

#include "fu-genesys-common.h"

#include "fu-genesys-usbhub-dev-firmware.h"

#include "fu-genesys-usbhub-firmware.h"

#include "fu-genesys-usbhub-struct.h"

struct _FuGenesysUsbhubDevFirmware {

  FuFirmware parent_instance;

};

G_DEFINE_TYPE(FuGenesysUsbhubDevFirmware, fu_genesys_usbhub_dev_firmware, FU_TYPE_FIRMWARE)

static gboolean

fu_genesys_usbhub_dev_firmware_validate(FuFirmware *firmware,

          FuInputStream *stream,

          gsize offset,

          GError **error)

{

  return fu_struct_genesys_dev_firmware_hdr_validate_stream(stream, offset, error);

}

static gboolean

fu_genesys_usbhub_dev_firmware_parse(FuFirmware *firmware,

             FuInputStream *stream,

             FuFirmwareParseFlags flags,

             GError **error)

{

  gsize code_size = 0;

  g_autoptr(FuInputStream) stream_trunc = NULL;

  fu_firmware_set_id(firmware, fu_genesys_fw_type_to_string(FU_GENESYS_FW_TYPE_DEV_BRIDGE));

  fu_firmware_set_idx(firmware, FU_GENESYS_FW_TYPE_DEV_BRIDGE);

  fu_firmware_set_alignment(firmware, FU_FIRMWARE_ALIGNMENT_1K);

  /* truncate to correct size */

  if (!fu_genesys_usbhub_firmware_calculate_size(stream, &code_size, error)) {

    g_prefix_error_literal(error, "not valid for dev: ");

    return FALSE;

  }

  stream_trunc = fu_partial_input_stream_new(stream, 0x0, code_size, error);

  if (stream_trunc == NULL)

    return FALSE;

  if (!fu_firmware_set_stream(firmware, stream_trunc, error))

    return FALSE;

  /* calculate checksum */

  if ((flags & FU_FIRMWARE_PARSE_FLAG_IGNORE_CHECKSUM) == 0) {

    if (!fu_genesys_usbhub_firmware_verify_checksum(stream_trunc, error)) {

      g_prefix_error_literal(error, "not valid for dev: ");

      return FALSE;

    }

  }

  /* get firmware version */

  if (!fu_genesys_usbhub_firmware_ensure_version(firmware, error)) {

    g_prefix_error_literal(error, "not valid for dev: ");

    return FALSE;

  }

  /* success */

  return TRUE;

}

static void

fu_genesys_usbhub_dev_firmware_init(FuGenesysUsbhubDevFirmware *self)

{

  fu_firmware_add_flag(FU_FIRMWARE(self), FU_FIRMWARE_FLAG_HAS_CHECKSUM);

}

static void

fu_genesys_usbhub_dev_firmware_class_init(FuGenesysUsbhubDevFirmwareClass *klass)

{

  FuFirmwareClass *firmware_class = FU_FIRMWARE_CLASS(klass);

  firmware_class->validate = fu_genesys_usbhub_dev_firmware_validate;

  firmware_class->parse = fu_genesys_usbhub_dev_firmware_parse;

  fu_firmware_set_size_max(firmware_class, 16 * FU_MB);

}