/* SPDX-License-Identifier: GPL-2.0+ */
/*
 * MACsec netdev header, used for h/w accelerated implementations.
 *
 * Copyright (c) 2015 Sabrina Dubroca <sd@queasysnail.net>
 */
#ifndef _NET_MACSEC_H_
#define _NET_MACSEC_H_

#include <linux/u64_stats_sync.h>
#include <linux/if_vlan.h>
#include <uapi/linux/if_link.h>
#include <uapi/linux/if_macsec.h>

#define MACSEC_DEFAULT_PN_LEN 4
#define MACSEC_XPN_PN_LEN 8

#define MACSEC_NUM_AN 4 /* 2 bits for the association number */

#define MACSEC_SCI_LEN 8
#define MACSEC_PORT_ES (htons(0x0001))

#define MACSEC_TCI_VERSION 0x80
#define MACSEC_TCI_ES      0x40 /* end station */
#define MACSEC_TCI_SC      0x20 /* SCI present */
#define MACSEC_TCI_SCB     0x10 /* epon */
#define MACSEC_TCI_E       0x08 /* encryption */
#define MACSEC_TCI_C       0x04 /* changed text */
#define MACSEC_AN_MASK     0x03 /* association number */
#define MACSEC_TCI_CONFID  (MACSEC_TCI_E | MACSEC_TCI_C)

#define MACSEC_DEFAULT_ICV_LEN 16

typedef u64 __bitwise sci_t;
typedef u32 __bitwise ssci_t;

struct metadata_dst;

typedef union salt {
 struct {
  ssci_t ssci;
  __be64 pn;
 } __packed;
 u8 bytes[MACSEC_SALT_LEN];
} __packed salt_t;

typedef union pn {
 struct {
#if defined(__LITTLE_ENDIAN_BITFIELD)
  u32 lower;
  u32 upper;
#elif defined(__BIG_ENDIAN_BITFIELD)
  u32 upper;
  u32 lower;
#else
#error "Please fix <asm/byteorder.h>"
#endif
 };
 u64 full64;
} pn_t;

/**
 * struct macsec_key - SA key
 * @id: user-provided key identifier
 * @tfm: crypto struct, key storage
 * @salt: salt used to generate IV in XPN cipher suites
 */
struct macsec_key {
 u8 id[MACSEC_KEYID_LEN];
 struct crypto_aead *tfm;
 salt_t salt;
};

struct macsec_rx_sc_stats {
 __u64 InOctetsValidated;
 __u64 InOctetsDecrypted;
 __u64 InPktsUnchecked;
 __u64 InPktsDelayed;
 __u64 InPktsOK;
 __u64 InPktsInvalid;
 __u64 InPktsLate;
 __u64 InPktsNotValid;
 __u64 InPktsNotUsingSA;
 __u64 InPktsUnusedSA;
};

struct macsec_rx_sa_stats {
 __u32 InPktsOK;
 __u32 InPktsInvalid;
 __u32 InPktsNotValid;
 __u32 InPktsNotUsingSA;
 __u32 InPktsUnusedSA;
};

struct macsec_tx_sa_stats {
 __u32 OutPktsProtected;
 __u32 OutPktsEncrypted;
};

struct macsec_tx_sc_stats {
 __u64 OutPktsProtected;
 __u64 OutPktsEncrypted;
 __u64 OutOctetsProtected;
 __u64 OutOctetsEncrypted;
};

struct macsec_dev_stats {
 __u64 OutPktsUntagged;
 __u64 InPktsUntagged;
 __u64 OutPktsTooLong;
 __u64 InPktsNoTag;
 __u64 InPktsBadTag;
 __u64 InPktsUnknownSCI;
 __u64 InPktsNoSCI;
 __u64 InPktsOverrun;
};

/**
 * struct macsec_rx_sa - receive secure association
 * @active:
 * @next_pn: packet number expected for the next packet
 * @lock: protects next_pn manipulations
 * @key: key structure
 * @ssci: short secure channel identifier
 * @stats: per-SA stats
 */
struct macsec_rx_sa {
 struct macsec_key key;
 ssci_t ssci;
 spinlock_t lock;
 union {
  pn_t next_pn_halves;
  u64 next_pn;
 };
 refcount_t refcnt;
 bool active;
 struct macsec_rx_sa_stats __percpu *stats;
 struct macsec_rx_sc *sc;
 struct rcu_head rcu;
};

struct pcpu_rx_sc_stats {
 struct macsec_rx_sc_stats stats;
 struct u64_stats_sync syncp;
};

struct pcpu_tx_sc_stats {
 struct macsec_tx_sc_stats stats;
 struct u64_stats_sync syncp;
};

/**
 * struct macsec_rx_sc - receive secure channel
 * @sci: secure channel identifier for this SC
 * @active: channel is active
 * @sa: array of secure associations
 * @stats: per-SC stats
 */
struct macsec_rx_sc {
 struct macsec_rx_sc __rcu *next;
 sci_t sci;
 bool active;
 struct macsec_rx_sa __rcu *sa[MACSEC_NUM_AN];
 struct pcpu_rx_sc_stats __percpu *stats;
 refcount_t refcnt;
 struct rcu_head rcu_head;
};

/**
 * struct macsec_tx_sa - transmit secure association
 * @active:
 * @next_pn: packet number to use for the next packet
 * @lock: protects next_pn manipulations
 * @key: key structure
 * @ssci: short secure channel identifier
 * @stats: per-SA stats
 */
struct macsec_tx_sa {
 struct macsec_key key;
 ssci_t ssci;
 spinlock_t lock;
 union {
  pn_t next_pn_halves;
  u64 next_pn;
 };
 refcount_t refcnt;
 bool active;
 struct macsec_tx_sa_stats __percpu *stats;
 struct rcu_head rcu;
};

/**
 * struct macsec_tx_sc - transmit secure channel
 * @active:
 * @encoding_sa: association number of the SA currently in use
 * @encrypt: encrypt packets on transmit, or authenticate only
 * @send_sci: always include the SCI in the SecTAG
 * @end_station:
 * @scb: single copy broadcast flag
 * @sa: array of secure associations
 * @stats: stats for this TXSC
 * @md_dst: MACsec offload metadata dst
 */
struct macsec_tx_sc {
 bool active;
 u8 encoding_sa;
 bool encrypt;
 bool send_sci;
 bool end_station;
 bool scb;
 struct macsec_tx_sa __rcu *sa[MACSEC_NUM_AN];
 struct pcpu_tx_sc_stats __percpu *stats;
 struct metadata_dst *md_dst;
};

/**
 * struct macsec_secy - MACsec Security Entity
 * @netdev: netdevice for this SecY
 * @n_rx_sc: number of receive secure channels configured on this SecY
 * @sci: secure channel identifier used for tx
 * @key_len: length of keys used by the cipher suite
 * @icv_len: length of ICV used by the cipher suite
 * @validate_frames: validation mode
 * @xpn: enable XPN for this SecY
 * @operational: MAC_Operational flag
 * @protect_frames: enable protection for this SecY
 * @replay_protect: enable packet number checks on receive
 * @replay_window: size of the replay window
 * @tx_sc: transmit secure channel
 * @rx_sc: linked list of receive secure channels
 */
struct macsec_secy {
 struct net_device *netdev;
 unsigned int n_rx_sc;
 sci_t sci;
 u16 key_len;
 u16 icv_len;
 enum macsec_validation_type validate_frames;
 bool xpn;
 bool operational;
 bool protect_frames;
 bool replay_protect;
 u32 replay_window;
 struct macsec_tx_sc tx_sc;
 struct macsec_rx_sc __rcu *rx_sc;
};

/**
 * struct macsec_context - MACsec context for hardware offloading
 * @netdev: a valid pointer to a struct net_device if @offload ==
 * MACSEC_OFFLOAD_MAC
 * @phydev: a valid pointer to a struct phy_device if @offload ==
 * MACSEC_OFFLOAD_PHY
 * @offload: MACsec offload status
 * @secy: pointer to a MACsec SecY
 * @rx_sc: pointer to a RX SC
 * @update_pn: when updating the SA, update the next PN
 * @assoc_num: association number of the target SA
 * @key: key of the target SA
 * @rx_sa: pointer to an RX SA if a RX SA is added/updated/removed
 * @tx_sa: pointer to an TX SA if a TX SA is added/updated/removed
 * @tx_sc_stats: pointer to TX SC stats structure
 * @tx_sa_stats: pointer to TX SA stats structure
 * @rx_sc_stats: pointer to RX SC stats structure
 * @rx_sa_stats: pointer to RX SA stats structure
 * @dev_stats: pointer to dev stats structure
 */
struct macsec_context {
 union {
  struct net_device *netdev;
  struct phy_device *phydev;
 };
 enum macsec_offload offload;

 struct macsec_secy *secy;
 struct macsec_rx_sc *rx_sc;
 struct {
  bool update_pn;
  unsigned char assoc_num;
  u8 key[MACSEC_MAX_KEY_LEN];
  union {
   struct macsec_rx_sa *rx_sa;
   struct macsec_tx_sa *tx_sa;
  };
 } sa;
 union {
  struct macsec_tx_sc_stats *tx_sc_stats;
  struct macsec_tx_sa_stats *tx_sa_stats;
  struct macsec_rx_sc_stats *rx_sc_stats;
  struct macsec_rx_sa_stats *rx_sa_stats;
  struct macsec_dev_stats  *dev_stats;
 } stats;
};

/**
 * struct macsec_ops - MACsec offloading operations
 * @mdo_dev_open: called when the MACsec interface transitions to the up state
 * @mdo_dev_stop: called when the MACsec interface transitions to the down
 * state
 * @mdo_add_secy: called when a new SecY is added
 * @mdo_upd_secy: called when the SecY flags are changed or the MAC address of
 * the MACsec interface is changed
 * @mdo_del_secy: called when the hw offload is disabled or the MACsec
 * interface is removed
 * @mdo_add_rxsc: called when a new RX SC is added
 * @mdo_upd_rxsc: called when a certain RX SC is updated
 * @mdo_del_rxsc: called when a certain RX SC is removed
 * @mdo_add_rxsa: called when a new RX SA is added
 * @mdo_upd_rxsa: called when a certain RX SA is updated
 * @mdo_del_rxsa: called when a certain RX SA is removed
 * @mdo_add_txsa: called when a new TX SA is added
 * @mdo_upd_txsa: called when a certain TX SA is updated
 * @mdo_del_txsa: called when a certain TX SA is removed
 * @mdo_get_dev_stats: called when dev stats are read
 * @mdo_get_tx_sc_stats: called when TX SC stats are read
 * @mdo_get_tx_sa_stats: called when TX SA stats are read
 * @mdo_get_rx_sc_stats: called when RX SC stats are read
 * @mdo_get_rx_sa_stats: called when RX SA stats are read
 * @mdo_insert_tx_tag: called to insert the TX tag
 * @needed_headroom: number of bytes reserved at the beginning of the sk_buff
 * for the TX tag
 * @needed_tailroom: number of bytes reserved at the end of the sk_buff for the
 * TX tag
 * @rx_uses_md_dst: whether MACsec device offload supports sk_buff md_dst
 */
struct macsec_ops {
 /* Device wide */
 int (*mdo_dev_open)(struct macsec_context *ctx);
 int (*mdo_dev_stop)(struct macsec_context *ctx);
 /* SecY */
 int (*mdo_add_secy)(struct macsec_context *ctx);
 int (*mdo_upd_secy)(struct macsec_context *ctx);
 int (*mdo_del_secy)(struct macsec_context *ctx);
 /* Security channels */
 int (*mdo_add_rxsc)(struct macsec_context *ctx);
 int (*mdo_upd_rxsc)(struct macsec_context *ctx);
 int (*mdo_del_rxsc)(struct macsec_context *ctx);
 /* Security associations */
 int (*mdo_add_rxsa)(struct macsec_context *ctx);
 int (*mdo_upd_rxsa)(struct macsec_context *ctx);
 int (*mdo_del_rxsa)(struct macsec_context *ctx);
 int (*mdo_add_txsa)(struct macsec_context *ctx);
 int (*mdo_upd_txsa)(struct macsec_context *ctx);
 int (*mdo_del_txsa)(struct macsec_context *ctx);
 /* Statistics */
 int (*mdo_get_dev_stats)(struct macsec_context *ctx);
 int (*mdo_get_tx_sc_stats)(struct macsec_context *ctx);
 int (*mdo_get_tx_sa_stats)(struct macsec_context *ctx);
 int (*mdo_get_rx_sc_stats)(struct macsec_context *ctx);
 int (*mdo_get_rx_sa_stats)(struct macsec_context *ctx);
 /* Offload tag */
 int (*mdo_insert_tx_tag)(struct phy_device *phydev,
     struct sk_buff *skb);
 unsigned int needed_headroom;
 unsigned int needed_tailroom;
 bool rx_uses_md_dst;
};

void macsec_pn_wrapped(struct macsec_secy *secy, struct macsec_tx_sa *tx_sa);
static inline bool macsec_send_sci(const struct macsec_secy *secy)
{
 const struct macsec_tx_sc *tx_sc = &secy->tx_sc;

 return tx_sc->send_sci ||
  (secy->n_rx_sc > 1 && !tx_sc->end_station && !tx_sc->scb);
}
struct net_device *macsec_get_real_dev(const struct net_device *dev);
bool macsec_netdev_is_offloaded(struct net_device *dev);

static inline void *macsec_netdev_priv(const struct net_device *dev)
{
#if IS_ENABLED(CONFIG_VLAN_8021Q)
 if (is_vlan_dev(dev))
  return netdev_priv(vlan_dev_priv(dev)->real_dev);
#endif
 return netdev_priv(dev);
}

static inline u64 sci_to_cpu(sci_t sci)
{
 return be64_to_cpu((__force __be64)sci);
}

#endif /* _NET_MACSEC_H_ */