// SPDX-License-Identifier: ISC
/*
 * Copyright (c) 2019 Broadcom
 */

#include <linux/unaligned.h>

#include <linux/math.h>
#include <linux/string.h>
#include <linux/bug.h>

#include "xtlv.h"

static int brcmf_xtlv_header_size(u16 opts)
{
 int len = (int)offsetof(struct brcmf_xtlv, data);

 if (opts & BRCMF_XTLV_OPTION_IDU8)
  --len;
 if (opts & BRCMF_XTLV_OPTION_LENU8)
  --len;

 return len;
}

int brcmf_xtlv_data_size(int dlen, u16 opts)
{
 int hsz;

 hsz = brcmf_xtlv_header_size(opts);
 if (opts & BRCMF_XTLV_OPTION_ALIGN32)
  return roundup(dlen + hsz, 4);

 return dlen + hsz;
}

void brcmf_xtlv_pack_header(struct brcmf_xtlv *xtlv, u16 id, u16 len,
       const u8 *data, u16 opts)
{
 u8 *data_buf;
 u16 mask = BRCMF_XTLV_OPTION_IDU8 | BRCMF_XTLV_OPTION_LENU8;

 if (!(opts & mask)) {
  u8 *idp = (u8 *)xtlv;
  u8 *lenp = idp + sizeof(xtlv->id);

  put_unaligned_le16(id, idp);
  put_unaligned_le16(len, lenp);
  data_buf = lenp + sizeof(u16);
 } else if ((opts & mask) == mask) { /* u8 id and u8 len */
  u8 *idp = (u8 *)xtlv;
  u8 *lenp = idp + 1;

  *idp = (u8)id;
  *lenp = (u8)len;
  data_buf = lenp + sizeof(u8);
 } else if (opts & BRCMF_XTLV_OPTION_IDU8) { /* u8 id, u16 len */
  u8 *idp = (u8 *)xtlv;
  u8 *lenp = idp + 1;

  *idp = (u8)id;
  put_unaligned_le16(len, lenp);
  data_buf = lenp + sizeof(u16);
 } else if (opts & BRCMF_XTLV_OPTION_LENU8) { /* u16 id, u8 len */
  u8 *idp = (u8 *)xtlv;
  u8 *lenp = idp + sizeof(u16);

  put_unaligned_le16(id, idp);
  *lenp = (u8)len;
  data_buf = lenp + sizeof(u8);
 } else {
  WARN(true, "Unexpected xtlv option");
  return;
 }

 if (opts & BRCMF_XTLV_OPTION_LENU8) {
  WARN_ON(len > 0x00ff);
  len &= 0xff;
 }

 if (data)
  memcpy(data_buf, data, len);
}