/*

 * Copyright 2018 Evan Lojewski

 * Copyright 2020 Richard Hughes <richard@hughsie.com>

 *

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

 */

#include "config.h"

#include <string.h>

#include "fu-bcm57xx-common.h"

gboolean

fu_bcm57xx_verify_crc(GInputStream *stream, GError **error)

{

  guint32 crc_actual = 0xFFFFFFFF;

  guint32 crc_file = 0;

  gsize streamsz = 0;

  g_autoptr(GInputStream) stream_tmp = NULL;

  /* expected */

  if (!fu_input_stream_size(stream, &streamsz, error))

    return FALSE;

  if (streamsz < sizeof(guint32)) {

    g_set_error_literal(error,

            FWUPD_ERROR,

            FWUPD_ERROR_INVALID_DATA,

            "image is too small for CRC");

    return FALSE;

  }

  if (!fu_input_stream_read_u32(stream,

              streamsz - sizeof(guint32),

              &crc_file,

              G_LITTLE_ENDIAN,

              error))

    return FALSE;

  /* reality */

  stream_tmp = fu_partial_input_stream_new(stream, 0, streamsz - sizeof(guint32), error);

  if (stream_tmp == NULL)

    return FALSE;

  if (!fu_input_stream_compute_crc32(stream_tmp,

             FU_CRC_KIND_B32_STANDARD,

             &crc_actual,

             error))

    return FALSE;

  if (crc_actual != crc_file) {

    g_set_error(error,

          FWUPD_ERROR,

          FWUPD_ERROR_NOT_SUPPORTED,

          "invalid CRC, expected 0x%08x got: 0x%08x",

          (guint)crc_file,

          (guint)crc_actual);

    return FALSE;

  }

  /* success */

  return TRUE;

}

gboolean

fu_bcm57xx_verify_magic(GInputStream *stream, gsize offset, GError **error)

{

  guint32 magic = 0;

  /* hardcoded value */

  if (!fu_input_stream_read_u32(stream, offset, &magic, G_BIG_ENDIAN, error))

    return FALSE;

  if (magic != BCM_NVRAM_MAGIC) {

    g_set_error(error,

          FWUPD_ERROR,

          FWUPD_ERROR_NOT_SUPPORTED,

          "invalid magic, got: 0x%x",

          (guint)magic);

    return FALSE;

  }

  /* success */

  return TRUE;

}

void

fu_bcm57xx_veritem_free(FuBcm57xxVeritem *veritem)

{

  g_free(veritem->branch);

  g_free(veritem->version);

  g_free(veritem);

}

FuBcm57xxVeritem *

fu_bcm57xx_veritem_new(const guint8 *buf, gsize bufsz)

{

  g_autofree gchar *tmp = NULL;

  g_autoptr(FuBcm57xxVeritem) veritem = g_new0(FuBcm57xxVeritem, 1);

  struct {

    const gchar *prefix;

    const gchar *branch;

    FwupdVersionFormat verfmt;

  } data[] = {{"5719-v", BCM_FW_BRANCH_UNKNOWN, FWUPD_VERSION_FORMAT_PAIR},

        {"stage1-", BCM_FW_BRANCH_OSS_FIRMWARE, FWUPD_VERSION_FORMAT_TRIPLET},

        {NULL, NULL, 0}};

  /* do not assume this is NUL terminated */

  tmp = g_strndup((const gchar *)buf, bufsz);

  if (tmp == NULL || tmp[0] == '\0')

    return NULL;

  /* use prefix to define object */

  for (guint i = 0; data[i].prefix != NULL; i++) {

    if (g_str_has_prefix(tmp, data[i].prefix)) {

      veritem->version = g_strdup(tmp + strlen(data[i].prefix));

      veritem->branch = g_strdup(data[i].branch);

      veritem->verfmt = data[i].verfmt;

      return g_steal_pointer(&veritem);

    }

  }

  veritem->verfmt = FWUPD_VERSION_FORMAT_UNKNOWN;

  veritem->version = g_strdup(tmp);

  return g_steal_pointer(&veritem);

}