/*

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

 *

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

 */

#include "config.h"

#include "fu-redfish-common.h"

#include "fu-redfish-smbios.h"

#include "fu-redfish-struct.h"

struct _FuRedfishSmbios {

  FuFirmware parent_instance;

  FuRedfishSmbiosInterfaceType interface_type;

  guint16 port;

  gchar *hostname;

  gchar *mac_addr;

  gchar *ip_addr;

  guint16 vid;

  guint16 pid;

};

G_DEFINE_TYPE(FuRedfishSmbios, fu_redfish_smbios, FU_TYPE_FIRMWARE)

FuRedfishSmbiosInterfaceType

fu_redfish_smbios_get_interface_type(FuRedfishSmbios *self)

{

  return self->interface_type;

}

guint16

fu_redfish_smbios_get_port(FuRedfishSmbios *self)

{

  return self->port;

}

guint16

fu_redfish_smbios_get_vid(FuRedfishSmbios *self)

{

  return self->vid;

}

guint16

fu_redfish_smbios_get_pid(FuRedfishSmbios *self)

{

  return self->pid;

}

const gchar *

fu_redfish_smbios_get_hostname(FuRedfishSmbios *self)

{

  return self->hostname;

}

const gchar *

fu_redfish_smbios_get_mac_addr(FuRedfishSmbios *self)

{

  return self->mac_addr;

}

const gchar *

fu_redfish_smbios_get_ip_addr(FuRedfishSmbios *self)

{

  return self->ip_addr;

}

static void

fu_redfish_smbios_set_hostname(FuRedfishSmbios *self, const gchar *hostname)

{

  g_set_str(&self->hostname, hostname);

}

static void

fu_redfish_smbios_set_mac_addr(FuRedfishSmbios *self, const gchar *mac_addr)

{

  g_set_str(&self->mac_addr, mac_addr);

}

static void

fu_redfish_smbios_set_ip_addr(FuRedfishSmbios *self, const gchar *ip_addr)

{

  g_set_str(&self->ip_addr, ip_addr);

}

static void

fu_redfish_smbios_set_port(FuRedfishSmbios *self, guint16 port)

{

  self->port = port;

}

static void

fu_redfish_smbios_export(FuFirmware *firmware, FuFirmwareExportFlags flags, XbBuilderNode *bn)

{

  FuRedfishSmbios *self = FU_REDFISH_SMBIOS(firmware);

  fu_xmlb_builder_insert_kv(bn,

          "interface_type",

          fu_redfish_smbios_interface_type_to_string(self->interface_type));

  fu_xmlb_builder_insert_kx(bn, "port", self->port);

  fu_xmlb_builder_insert_kv(bn, "hostname", self->hostname);

  fu_xmlb_builder_insert_kv(bn, "mac_addr", self->mac_addr);

  fu_xmlb_builder_insert_kv(bn, "ip_addr", self->ip_addr);

  fu_xmlb_builder_insert_kx(bn, "vid", self->vid);

  fu_xmlb_builder_insert_kx(bn, "pid", self->pid);

}

static gboolean

fu_redfish_smbios_build(FuFirmware *firmware, XbNode *n, GError **error)

{

  FuRedfishSmbios *self = FU_REDFISH_SMBIOS(firmware);

  const gchar *tmp;

  guint64 tmpu;

  /* optional properties */

  tmpu = xb_node_query_text_as_uint(n, "port", NULL);

  if (tmpu != G_MAXUINT64)

    fu_redfish_smbios_set_port(self, (guint16)tmpu);

  tmpu = xb_node_query_text_as_uint(n, "vid", NULL);

  if (tmpu != G_MAXUINT64)

    self->vid = (guint16)tmpu;

  tmpu = xb_node_query_text_as_uint(n, "pid", NULL);

  if (tmpu != G_MAXUINT64)

    self->pid = (guint16)tmpu;

  tmp = xb_node_query_text(n, "hostname", NULL);

  if (tmp != NULL)

    fu_redfish_smbios_set_hostname(self, tmp);

  tmp = xb_node_query_text(n, "mac_addr", NULL);

  if (tmp != NULL)

    fu_redfish_smbios_set_mac_addr(self, tmp);

  tmp = xb_node_query_text(n, "ip_addr", NULL);

  if (tmp != NULL)

    fu_redfish_smbios_set_ip_addr(self, tmp);

  /* success */

  return TRUE;

}

static gboolean

fu_redfish_smbios_parse_interface_data(FuRedfishSmbios *self,

               FuInputStream *stream,

               gsize offset,

               GError **error)

{

  gsize offset_mac_addr = G_MAXSIZE;

  gsize offset_vid_pid = G_MAXSIZE;

  guint8 interface_type = 0x0;

  /* parse the data depending on the interface type */

  if (!fu_input_stream_read_u8(stream, offset, &interface_type, error))

    return FALSE;

  g_debug("interface_type: %s [0x%x]",

    fu_redfish_interface_type_to_string(interface_type),

    interface_type);

  offset++;

  switch (interface_type) {

  case FU_REDFISH_INTERFACE_TYPE_USB_NETWORK:

  case FU_REDFISH_INTERFACE_TYPE_PCI_NETWORK:

    offset_vid_pid = 0x00;

    break;

  case FU_REDFISH_INTERFACE_TYPE_USB_NETWORK_V2:

    offset_vid_pid = 0x01;

    offset_mac_addr = 0x06;

    break;

  case FU_REDFISH_INTERFACE_TYPE_PCI_NETWORK_V2:

    offset_vid_pid = 0x01;

    offset_mac_addr = 0x09;

    break;

  default:

    g_debug("unknown Network Interface");

    break;

  }

  /* MAC address */

  if (offset_mac_addr != G_MAXSIZE) {

    guint8 mac_addr[6] = {0x0};

    g_autofree gchar *mac_addr_str = NULL;

    if (!fu_input_stream_read_safe(stream,

                 mac_addr,

                 sizeof(mac_addr),

                 0x0,      /* dst */

                 offset + offset_mac_addr, /* src */

                 sizeof(mac_addr),

                 error))

      return FALSE;

    mac_addr_str = fu_redfish_common_buffer_to_mac(mac_addr);

    fu_redfish_smbios_set_mac_addr(self, mac_addr_str);

  }

  /* VID:PID */

  if (offset_vid_pid != G_MAXSIZE) {

    if (!fu_input_stream_read_u16(stream,

                offset + offset_vid_pid,

                &self->vid,

                G_LITTLE_ENDIAN,

                error))

      return FALSE;

    if (!fu_input_stream_read_u16(stream,

                offset + offset_vid_pid + 0x02,

                &self->pid,

                G_LITTLE_ENDIAN,

                error))

      return FALSE;

  }

  /* success */

  return TRUE;

}

static gboolean

fu_redfish_smbios_parse_over_ip(FuRedfishSmbios *self,

        FuInputStream *stream,

        gsize offset,

        GError **error)

{

  guint8 hostname_length;

  guint8 service_ip_address_format;

  g_autoptr(FuStructRedfishProtocolOverIp) st = NULL;

  /* port + IP address */

  st = fu_struct_redfish_protocol_over_ip_parse_stream(stream, offset, error);

  if (st == NULL)

    return FALSE;

  fu_redfish_smbios_set_port(self,

           fu_struct_redfish_protocol_over_ip_get_service_ip_port(st));

  service_ip_address_format =

      fu_struct_redfish_protocol_over_ip_get_service_ip_address_format(st);

  if (service_ip_address_format == FU_REDFISH_IP_ADDRESS_FORMAT_V4) {

    const guint8 *ip_address =

        fu_struct_redfish_protocol_over_ip_get_service_ip_address(st, NULL);

    g_autofree gchar *tmp = fu_redfish_common_buffer_to_ipv4(ip_address);

    fu_redfish_smbios_set_ip_addr(self, tmp);

  } else if (service_ip_address_format == FU_REDFISH_IP_ADDRESS_FORMAT_V6) {

    const guint8 *ip_address =

        fu_struct_redfish_protocol_over_ip_get_service_ip_address(st, NULL);

    g_autofree gchar *tmp = fu_redfish_common_buffer_to_ipv6(ip_address);

    fu_redfish_smbios_set_ip_addr(self, tmp);

  } else {

    g_set_error_literal(error,

            FWUPD_ERROR,

            FWUPD_ERROR_INVALID_FILE,

            "address format is invalid");

    return FALSE;

  }

  /* hostname */

  hostname_length = fu_struct_redfish_protocol_over_ip_get_service_hostname_len(st);

  if (hostname_length > 0) {

    g_autofree gchar *hostname = g_malloc0(hostname_length + 1);

    if (!fu_input_stream_read_safe(stream,

                 (guint8 *)hostname,

                 hostname_length,

                 0x0,         /* dst */

                 offset + st->buf->len, /* seek */

                 hostname_length,

                 error))

      return FALSE;

    fu_redfish_smbios_set_hostname(self, hostname);

  }

  /* success */

  return TRUE;

}

static gboolean

fu_redfish_smbios_parse(FuFirmware *firmware,

      FuInputStream *stream,

      FuFirmwareParseFlags flags,

      GError **error)

{

  FuRedfishSmbios *self = FU_REDFISH_SMBIOS(firmware);

  gsize offset = 0;

  gsize streamsz = 0;

  g_autoptr(FuStructRedfishSmbiosType42) st = NULL;

  /* check size */

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

    return FALSE;

  if (streamsz < 0x09) {

    g_set_error(error,

          FWUPD_ERROR,

          FWUPD_ERROR_INVALID_FILE,

          "SMBIOS entry too small: %zu",

          streamsz);

    return FALSE;

  }

  /* parse */

  st = fu_struct_redfish_smbios_type42_parse_stream(stream, offset, error);

  if (st == NULL)

    return FALSE;

  /* check length */

  if (fu_struct_redfish_smbios_type42_get_length(st) != streamsz) {

    g_set_error(error,

          FWUPD_ERROR,

          FWUPD_ERROR_INVALID_FILE,

          "size of table 0x%x does not match binary 0x%x",

          fu_struct_redfish_smbios_type42_get_length(st),

          (guint)streamsz);

    return FALSE;

  }

  /* check length */

  if (!fu_size_checked_inc(&offset, FU_STRUCT_REDFISH_SMBIOS_TYPE42_SIZE, error)) {

    g_prefix_error_literal(error, "header offset overflow: ");

    return FALSE;

  }

  if (fu_struct_redfish_smbios_type42_get_data_length(st) > 0) {

    if (!fu_redfish_smbios_parse_interface_data(self, stream, offset, error))

      return FALSE;

  }

  if (!fu_size_checked_inc(&offset,

         fu_struct_redfish_smbios_type42_get_data_length(st),

         error))

    return FALSE;

  /* parse protocol records */

  self->interface_type = fu_struct_redfish_smbios_type42_get_interface_type(st);

  if (self->interface_type == FU_REDFISH_SMBIOS_INTERFACE_TYPE_NETWORK) {

    guint8 protocol_rcds = 0;

    if (!fu_input_stream_read_u8(stream, offset, &protocol_rcds, error))

      return FALSE;

    if (!fu_size_checked_inc(&offset, 1, error)) {

      g_prefix_error_literal(error, "protocol records offset overflow: ");

      return FALSE;

    }

    g_debug("protocol_rcds: %u", protocol_rcds);

    for (guint i = 0; i < protocol_rcds; i++) {

      guint8 protocol_id = 0;

      guint8 protocol_sz = 0;

      if (!fu_input_stream_read_u8(stream, offset, &protocol_id, error))

        return FALSE;

      if (!fu_input_stream_read_u8(stream, offset + 0x1, &protocol_sz, error))

        return FALSE;

      if (protocol_id == REDFISH_PROTOCOL_REDFISH_OVER_IP) {

        if (!fu_redfish_smbios_parse_over_ip(self,

                     stream,

                     offset + 0x2,

                     error))

          return FALSE;

      } else {

        g_debug("ignoring protocol ID 0x%02x", protocol_id);

      }

      if (!fu_size_checked_inc(&offset, protocol_sz + 1, error))

        return FALSE;

    }

  }

  /* success */

  return TRUE;

}

static GByteArray *

fu_redfish_smbios_write(FuFirmware *firmware, GError **error)

{

  FuRedfishSmbios *self = FU_REDFISH_SMBIOS(firmware);

  gsize hostname_sz = 0;

  g_autoptr(FuStructRedfishProtocolOverIp) st = fu_struct_redfish_protocol_over_ip_new();

  g_autoptr(GByteArray) buf = g_byte_array_new();

  if (self->hostname != NULL)

    hostname_sz = strlen(self->hostname);

  fu_byte_array_append_uint8(buf, REDFISH_SMBIOS_TABLE_TYPE);

  fu_byte_array_append_uint8(buf, 0x6D + hostname_sz);     /* length */

  fu_byte_array_append_uint16(buf, 0x1234, G_LITTLE_ENDIAN); /* handle */

  fu_byte_array_append_uint8(buf, FU_REDFISH_SMBIOS_INTERFACE_TYPE_NETWORK);

  fu_byte_array_append_uint8(buf, 0x09);          /* iface datalen */

  fu_byte_array_append_uint8(buf, FU_REDFISH_INTERFACE_TYPE_USB_NETWORK); /* iface */

  fu_byte_array_append_uint16(buf, self->vid, G_LITTLE_ENDIAN);    /* iface:VID */

  fu_byte_array_append_uint16(buf, self->pid, G_LITTLE_ENDIAN);    /* iface:PID */

  fu_byte_array_append_uint8(buf, 0x02);          /* iface:serialsz */

  fu_byte_array_append_uint8(buf, 0x03);          /* iType */

  fu_byte_array_append_uint8(buf, 'S');         /* iface:serial */

  fu_byte_array_append_uint8(buf, 'n');         /* iface:serial */

  fu_byte_array_append_uint8(buf, 0x1); /* nr protocol rcds */

  /* protocol record */

  fu_byte_array_append_uint8(buf, REDFISH_PROTOCOL_REDFISH_OVER_IP);

  fu_byte_array_append_uint8(buf, st->buf->len + hostname_sz);

  if (self->hostname != NULL)

    hostname_sz = strlen(self->hostname);

  fu_struct_redfish_protocol_over_ip_set_service_ip_port(st, self->port);

  fu_struct_redfish_protocol_over_ip_set_service_ip_address_format(

      st,

      FU_REDFISH_IP_ADDRESS_FORMAT_V4);

  fu_struct_redfish_protocol_over_ip_set_service_ip_assignment_type(

      st,

      FU_REDFISH_IP_ASSIGNMENT_TYPE_STATIC);

  fu_struct_redfish_protocol_over_ip_set_service_hostname_len(st, hostname_sz);

  g_byte_array_append(buf, st->buf->data, st->buf->len);

  if (hostname_sz > 0)

    g_byte_array_append(buf, (guint8 *)self->hostname, hostname_sz);

  return g_steal_pointer(&buf);

}

static void

fu_redfish_smbios_finalize(GObject *object)

{

  FuRedfishSmbios *self = FU_REDFISH_SMBIOS(object);

  g_free(self->hostname);

  g_free(self->mac_addr);

  g_free(self->ip_addr);

  G_OBJECT_CLASS(fu_redfish_smbios_parent_class)->finalize(object);

}

static void

fu_redfish_smbios_init(FuRedfishSmbios *self)

{

}

static void

fu_redfish_smbios_class_init(FuRedfishSmbiosClass *klass)

{

  GObjectClass *object_class = G_OBJECT_CLASS(klass);

  FuFirmwareClass *firmware_class = FU_FIRMWARE_CLASS(klass);

  object_class->finalize = fu_redfish_smbios_finalize;

  firmware_class->parse = fu_redfish_smbios_parse;

  firmware_class->write = fu_redfish_smbios_write;

  firmware_class->build = fu_redfish_smbios_build;

  firmware_class->export = fu_redfish_smbios_export;

  fu_firmware_set_size_max(firmware_class, 1 * FU_MB);

}

FuRedfishSmbios *

fu_redfish_smbios_new(void)

{

  return FU_REDFISH_SMBIOS(g_object_new(FU_TYPE_REDFISH_SMBIOS, NULL));

}