/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 *      Linux NET3:     Internet Group Management Protocol  [IGMP]
 *
 *      Authors:
 *              Alan Cox <alan@lxorguk.ukuu.org.uk>
 *
 *      Extended to talk the BSD extended IGMP protocol of mrouted 3.6
 */
#ifndef _LINUX_IGMP_H
#define _LINUX_IGMP_H

#include <linux/skbuff.h>
#include <linux/timer.h>
#include <linux/in.h>
#include <linux/ip.h>
#include <linux/refcount.h>
#include <linux/sockptr.h>
#include <uapi/linux/igmp.h>

static inline struct igmphdr *igmp_hdr(const struct sk_buff *skb)
{
        return (struct igmphdr *)skb_transport_header(skb);
}

static inline struct igmpv3_report *
                        igmpv3_report_hdr(const struct sk_buff *skb)
{
        return (struct igmpv3_report *)skb_transport_header(skb);
}

static inline struct igmpv3_query *
                        igmpv3_query_hdr(const struct sk_buff *skb)
{
        return (struct igmpv3_query *)skb_transport_header(skb);
}

struct ip_sf_socklist {
        unsigned int            sl_max;
        unsigned int            sl_count;
        struct rcu_head         rcu;
        __be32                  sl_addr[] __counted_by(sl_max);
};

#define IP_SFBLOCK      10      /* allocate this many at once */

/* ip_mc_socklist is real list now. Speed is not argument;
   this list never used in fast path code
 */

struct ip_mc_socklist {
        struct ip_mc_socklist __rcu *next_rcu;
        struct ip_mreqn         multi;
        unsigned int            sfmode;         /* MCAST_{INCLUDE,EXCLUDE} */
        struct ip_sf_socklist __rcu     *sflist;
        struct rcu_head         rcu;
};

struct ip_sf_list {
        struct ip_sf_list       *sf_next;
        unsigned long           sf_count[2];    /* include/exclude counts */
        __be32                  sf_inaddr;
        unsigned char           sf_gsresp;      /* include in g & s response? */
        unsigned char           sf_oldin;       /* change state */
        unsigned char           sf_crcount;     /* retrans. left to send */
};

struct ip_mc_list {
        struct in_device        *interface;
        __be32                  multiaddr;
        unsigned int            sfmode;
        struct ip_sf_list       *sources;
        struct ip_sf_list       *tomb;
        unsigned long           sfcount[2];
        union {
                struct ip_mc_list *next;
                struct ip_mc_list __rcu *next_rcu;
        };
        struct ip_mc_list __rcu *next_hash;
        struct timer_list       timer;
        int                     users;
        refcount_t              refcnt;
        spinlock_t              lock;
        char                    tm_running;
        char                    reporter;
        char                    unsolicit_count;
        char                    loaded;
        unsigned char           gsquery;        /* check source marks? */
        unsigned char           crcount;
        unsigned long           mca_cstamp;
        unsigned long           mca_tstamp;
        struct rcu_head         rcu;
};

/* RFC3376, relevant sections:
 *  - 4.1.1. Maximum Response Code
 *  - 4.1.7. QQIC (Querier's Query Interval Code)
 *
 * For both MRC and QQIC, values >= 128 use the same floating-point
 * encoding as follows:
 *
 *  0 1 2 3 4 5 6 7
 * +-+-+-+-+-+-+-+-+
 * |1| exp | mant  |
 * +-+-+-+-+-+-+-+-+
 */
#define IGMPV3_FP_EXP(value)            (((value) >> 4) & 0x07)
#define IGMPV3_FP_MAN(value)            ((value) & 0x0f)

/* IGMPv3 floating-point exponential field min threshold */
#define IGMPV3_EXP_MIN_THRESHOLD        128
/* IGMPv3 FP max threshold (mant = 0xF, exp = 7) -> 31744 */
#define IGMPV3_EXP_MAX_THRESHOLD        31744

/* V3 exponential field encoding */

/* IGMPv3 MRC/QQIC 8-bit exponential field encode
 *
 * RFC3376, 4.1.1 & 4.1.7. defines only the decoding formula:
 *     MRT/QQI = (mant | 0x10) << (exp + 3)
 *
 * but does NOT define the encoding procedure. To derive exponent:
 *
 * For any value of mantissa and exponent, the decoding formula
 * indicates that the "hidden bit" (0x10) is shifted 4 bits left
 * to sit above the 4-bit mantissa. The RFC again shifts this
 * entire block left by (exp + 3) to reconstruct the value.
 * So, 'hidden bit' is the MSB which is shifted by (4 + exp + 3).
 *
 * Total left shift of the 'hidden bit' = 4 + (exp + 3) = exp + 7.
 * This is the MSB at the 0-based bit position: (exp + 7).
 * Since fls() is 1-based, fls(value) - 1 = exp + 7.
 *
 * Therefore:
 *     exp  = fls(value) - 8
 *     mant = (value >> (exp + 3)) & 0x0F
 *
 * Final encoding formula:
 *     0x80 | (exp << 4) | mant
 *
 * Example (value = 3200):
 *  0               1
 *  0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7
 * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 * |0 0 0 0 1 1 0 0 1 0 0 0 0 0 0 0| (value = 3200)
 * |        ^-^-mant^ ^..(exp+3)..^| exp = 4, mant = 9
 * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 *
 * Encoded:
 *   0x80 | (4 << 4) | 9 = 0xC9
 */
static inline u8 igmpv3_exp_field_encode(unsigned long value)
{
        u8 mc_exp, mc_man;

        /* MRC/QQIC < 128 is literal */
        if (value < IGMPV3_EXP_MIN_THRESHOLD)
                return value;

        /* Saturate at max representable (mant = 0xF, exp = 7) -> 31744 */
        if (value >= IGMPV3_EXP_MAX_THRESHOLD)
                return 0xFF;

        mc_exp  = fls(value) - 8;
        mc_man = (value >> (mc_exp + 3)) & 0x0F;

        return 0x80 | (mc_exp << 4) | mc_man;
}

/* Calculate Maximum Response Code from Max Resp Time
 *
 * RFC3376, relevant sections:
 *  - 4.1.1. Maximum Response Code
 *  - 8.3. Query Response Interval
 *
 * MRC represents the encoded form of Max Resp Time (MRT); once
 * decoded, the resulting value is in units of 0.1 seconds (100 ms).
 */
static inline u8 igmpv3_mrc(unsigned long mrt)
{
        return igmpv3_exp_field_encode(mrt);
}

/* Calculate Querier's Query Interval Code from Querier's Query Interval
 *
 * RFC3376, relevant sections:
 *  - 4.1.7. QQIC (Querier's Query Interval Code)
 *  - 8.2. Query Interval
 *  - 8.12. Older Version Querier Present Timeout
 *    (the [Query Interval] in the last Query received)
 *
 * QQIC represents the encoded form of Querier's Query Interval (QQI);
 * once decoded, the resulting value is in units of seconds.
 */
static inline u8 igmpv3_qqic(unsigned long qi)
{
        return igmpv3_exp_field_encode(qi);
}

/* V3 exponential field decoding */

/* IGMPv3 MRC/QQIC 8-bit exponential field decode
 *
 * RFC3376, 4.1.1 & 4.1.7. defines the decoding formula:
 *      0 1 2 3 4 5 6 7
 *     +-+-+-+-+-+-+-+-+
 *     |1| exp | mant  |
 *     +-+-+-+-+-+-+-+-+
 * Max Resp Time = (mant | 0x10) << (exp + 3)
 * QQI = (mant | 0x10) << (exp + 3)
 */
static inline unsigned long igmpv3_exp_field_decode(const u8 code)
{
        if (code < IGMPV3_EXP_MIN_THRESHOLD) {
                return code;
        } else {
                unsigned long mc_man, mc_exp;

                mc_exp = IGMPV3_FP_EXP(code);
                mc_man = IGMPV3_FP_MAN(code);

                return (mc_man | 0x10) << (mc_exp + 3);
        }
}

/* Calculate Max Resp Time from Maximum Response Code
 *
 * RFC3376, relevant sections:
 *  - 4.1.1. Maximum Response Code
 *  - 8.3. Query Response Interval
 *
 * After decode, MRC represents the Maximum Response Time (MRT) in
 * units of 0.1 seconds (100 ms).
 */
static inline unsigned long igmpv3_mrt(const struct igmpv3_query *ih3)
{
        return igmpv3_exp_field_decode(ih3->code);
}

/* Calculate Querier's Query Interval from Querier's Query Interval Code
 *
 * RFC3376, relevant sections:
 *  - 4.1.7. QQIC (Querier's Query Interval Code)
 *  - 8.2. Query Interval
 *  - 8.12. Older Version Querier Present Timeout
 *    (the [Query Interval] in the last Query received)
 *
 * After decode, QQIC represents the Querier's Query Interval in units
 * of seconds.
 */
static inline unsigned long igmpv3_qqi(const struct igmpv3_query *ih3)
{
        return igmpv3_exp_field_decode(ih3->qqic);
}

static inline int ip_mc_may_pull(struct sk_buff *skb, unsigned int len)
{
        if (skb_transport_offset(skb) + ip_transport_len(skb) < len)
                return 0;

        return pskb_may_pull(skb, len);
}

extern int ip_check_mc_rcu(struct in_device *dev, __be32 mc_addr, __be32 src_addr, u8 proto);
extern int igmp_rcv(struct sk_buff *);
extern int ip_mc_join_group(struct sock *sk, struct ip_mreqn *imr);
extern int ip_mc_join_group_ssm(struct sock *sk, struct ip_mreqn *imr,
                                unsigned int mode);
extern int ip_mc_leave_group(struct sock *sk, struct ip_mreqn *imr);
extern void ip_mc_drop_socket(struct sock *sk);
extern int ip_mc_source(int add, int omode, struct sock *sk,
                struct ip_mreq_source *mreqs, int ifindex);
extern int ip_mc_msfilter(struct sock *sk, struct ip_msfilter *msf,int ifindex);
extern int ip_mc_msfget(struct sock *sk, struct ip_msfilter *msf,
                        sockptr_t optval, sockptr_t optlen);
extern int ip_mc_gsfget(struct sock *sk, struct group_filter *gsf,
                        sockptr_t optval, size_t offset);
extern int ip_mc_sf_allow(const struct sock *sk, __be32 local, __be32 rmt,
                          int dif, int sdif);
extern void ip_mc_init_dev(struct in_device *);
extern void ip_mc_destroy_dev(struct in_device *);
extern void ip_mc_up(struct in_device *);
extern void ip_mc_down(struct in_device *);
extern void ip_mc_unmap(struct in_device *);
extern void ip_mc_remap(struct in_device *);
extern void __ip_mc_dec_group(struct in_device *in_dev, __be32 addr, gfp_t gfp);
static inline void ip_mc_dec_group(struct in_device *in_dev, __be32 addr)
{
        return __ip_mc_dec_group(in_dev, addr, GFP_KERNEL);
}
extern void __ip_mc_inc_group(struct in_device *in_dev, __be32 addr,
                              gfp_t gfp);
extern void ip_mc_inc_group(struct in_device *in_dev, __be32 addr);
int ip_mc_check_igmp(struct sk_buff *skb);

#endif