/* SPDX-License-Identifier: GPL-2.0 */
/* Copyright (c) 2020 Facebook */

#ifndef _TEST_TCP_HDR_OPTIONS_H
#define _TEST_TCP_HDR_OPTIONS_H

struct bpf_test_option {
        __u8 flags;
        __u8 max_delack_ms;
        __u8 rand;
} __attribute__((packed));

enum {
        OPTION_RESEND,
        OPTION_MAX_DELACK_MS,
        OPTION_RAND,
        __NR_OPTION_FLAGS,
};

#define OPTION_F_RESEND         (1 << OPTION_RESEND)
#define OPTION_F_MAX_DELACK_MS  (1 << OPTION_MAX_DELACK_MS)
#define OPTION_F_RAND           (1 << OPTION_RAND)
#define OPTION_MASK             ((1 << __NR_OPTION_FLAGS) - 1)

#define TEST_OPTION_FLAGS(flags, option) (1 & ((flags) >> (option)))
#define SET_OPTION_FLAGS(flags, option) ((flags) |= (1 << (option)))

/* Store in bpf_sk_storage */
struct hdr_stg {
        bool active;
        bool resend_syn; /* active side only */
        bool syncookie;  /* passive side only */
        bool fastopen;  /* passive side only */
};

struct linum_err {
        unsigned int linum;
        int err;
};

#define TCPHDR_FIN 0x01
#define TCPHDR_SYN 0x02
#define TCPHDR_RST 0x04
#define TCPHDR_PSH 0x08
#define TCPHDR_ACK 0x10
#define TCPHDR_URG 0x20
#define TCPHDR_ECE 0x40
#define TCPHDR_CWR 0x80
#define TCPHDR_SYNACK (TCPHDR_SYN | TCPHDR_ACK)

#define TCPOPT_EOL              0
#define TCPOPT_NOP              1
#define TCPOPT_MSS              2
#define TCPOPT_WINDOW           3
#define TCPOPT_EXP              254

#define TCP_BPF_EXPOPT_BASE_LEN 4
#define MAX_TCP_HDR_LEN         60
#define MAX_TCP_OPTION_SPACE    40

#ifdef BPF_PROG_TEST_TCP_HDR_OPTIONS

#define CG_OK   1
#define CG_ERR  0

#ifndef SOL_TCP
#define SOL_TCP 6
#endif

struct tcp_exprm_opt {
        __u8 kind;
        __u8 len;
        __u16 magic;
        union {
                __u8 data[4];
                __u32 data32;
        };
} __attribute__((packed));

struct tcp_opt {
        __u8 kind;
        __u8 len;
        union {
                __u8 data[4];
                __u32 data32;
        };
} __attribute__((packed));

struct {
        __uint(type, BPF_MAP_TYPE_HASH);
        __uint(max_entries, 2);
        __type(key, int);
        __type(value, struct linum_err);
} lport_linum_map SEC(".maps");

static inline unsigned int tcp_hdrlen(const struct tcphdr *th)
{
        return th->doff << 2;
}

static inline __u8 skops_tcp_flags(const struct bpf_sock_ops *skops)
{
        return skops->skb_tcp_flags;
}

static inline void clear_hdr_cb_flags(struct bpf_sock_ops *skops)
{
        bpf_sock_ops_cb_flags_set(skops,
                                  skops->bpf_sock_ops_cb_flags &
                                  ~(BPF_SOCK_OPS_PARSE_UNKNOWN_HDR_OPT_CB_FLAG |
                                    BPF_SOCK_OPS_WRITE_HDR_OPT_CB_FLAG));
}

static inline void set_hdr_cb_flags(struct bpf_sock_ops *skops, __u32 extra)
{
        bpf_sock_ops_cb_flags_set(skops,
                                  skops->bpf_sock_ops_cb_flags |
                                  BPF_SOCK_OPS_PARSE_UNKNOWN_HDR_OPT_CB_FLAG |
                                  BPF_SOCK_OPS_WRITE_HDR_OPT_CB_FLAG |
                                  extra);
}
static inline void
clear_parse_all_hdr_cb_flags(struct bpf_sock_ops *skops)
{
        bpf_sock_ops_cb_flags_set(skops,
                                  skops->bpf_sock_ops_cb_flags &
                                  ~BPF_SOCK_OPS_PARSE_ALL_HDR_OPT_CB_FLAG);
}

static inline void
set_parse_all_hdr_cb_flags(struct bpf_sock_ops *skops)
{
        bpf_sock_ops_cb_flags_set(skops,
                                  skops->bpf_sock_ops_cb_flags |
                                  BPF_SOCK_OPS_PARSE_ALL_HDR_OPT_CB_FLAG);
}

#define RET_CG_ERR(__err) ({                    \
        struct linum_err __linum_err;           \
        int __lport;                            \
                                                \
        __linum_err.linum = __LINE__;           \
        __linum_err.err = __err;                \
        __lport = skops->local_port;            \
        bpf_map_update_elem(&lport_linum_map, &__lport, &__linum_err, BPF_NOEXIST); \
        clear_hdr_cb_flags(skops);                                      \
        clear_parse_all_hdr_cb_flags(skops);                            \
        return CG_ERR;                                                  \
})

#endif /* BPF_PROG_TEST_TCP_HDR_OPTIONS */

#endif /* _TEST_TCP_HDR_OPTIONS_H */