// SPDX-License-Identifier: GPL-2.0-only
/* Copyright (C) 2012, 2020 Oliver Hartkopp <socketcan@hartkopp.net>
 */

#include <linux/can/dev.h>

/* CAN DLC to real data length conversion helpers */

static const u8 dlc2len[] = {
 0, 1, 2, 3, 4, 5, 6, 7,
 8, 12, 16, 20, 24, 32, 48, 64
};

/* get data length from raw data length code (DLC) */
u8 can_fd_dlc2len(u8 dlc)
{
 return dlc2len[dlc & 0x0F];
}
EXPORT_SYMBOL_GPL(can_fd_dlc2len);

static const u8 len2dlc[] = {
 0, 1, 2, 3, 4, 5, 6, 7, 8, /* 0 - 8 */
 9, 9, 9, 9,   /* 9 - 12 */
 10, 10, 10, 10,   /* 13 - 16 */
 11, 11, 11, 11,   /* 17 - 20 */
 12, 12, 12, 12,   /* 21 - 24 */
 13, 13, 13, 13, 13, 13, 13, 13, /* 25 - 32 */
 14, 14, 14, 14, 14, 14, 14, 14, /* 33 - 40 */
 14, 14, 14, 14, 14, 14, 14, 14, /* 41 - 48 */
 15, 15, 15, 15, 15, 15, 15, 15, /* 49 - 56 */
 15, 15, 15, 15, 15, 15, 15, 15 /* 57 - 64 */
};

/* map the sanitized data length to an appropriate data length code */
u8 can_fd_len2dlc(u8 len)
{
 /* check for length mapping table size at build time */
 BUILD_BUG_ON(ARRAY_SIZE(len2dlc) != CANFD_MAX_DLEN + 1);

 if (unlikely(len > CANFD_MAX_DLEN))
  return CANFD_MAX_DLC;

 return len2dlc[len];
}
EXPORT_SYMBOL_GPL(can_fd_len2dlc);

/**
 * can_skb_get_frame_len() - Calculate the CAN Frame length in bytes
 *  of a given skb.
 * @skb: socket buffer of a CAN message.
 *
 * Do a rough calculation: bit stuffing is ignored and length in bits
 * is rounded up to a length in bytes.
 *
 * Rationale: this function is to be used for the BQL functions
 * (netdev_sent_queue() and netdev_completed_queue()) which expect a
 * value in bytes. Just using skb->len is insufficient because it will
 * return the constant value of CAN(FD)_MTU. Doing the bit stuffing
 * calculation would be too expensive in term of computing resources
 * for no noticeable gain.
 *
 * Remarks: The payload of CAN FD frames with BRS flag are sent at a
 * different bitrate. Currently, the can-utils canbusload tool does
 * not support CAN-FD yet and so we could not run any benchmark to
 * measure the impact. There might be possible improvement here.
 *
 * Return: length in bytes.
 */
unsigned int can_skb_get_frame_len(const struct sk_buff *skb)
{
 const struct canfd_frame *cf = (const struct canfd_frame *)skb->data;
 u8 len;

 if (can_is_canfd_skb(skb))
  len = canfd_sanitize_len(cf->len);
 else if (cf->can_id & CAN_RTR_FLAG)
  len = 0;
 else
  len = cf->len;

 return can_frame_bytes(can_is_canfd_skb(skb), cf->can_id & CAN_EFF_FLAG,
          false, len);
}
EXPORT_SYMBOL_GPL(can_skb_get_frame_len);