#ifndef TRAFGEN_PROTO_H
#define TRAFGEN_PROTO_H

#include <stddef.h>
#include <stdint.h>
#include <stdbool.h>

enum proto_id {
	PROTO_NONE = 0,
	PROTO_ETH,
	PROTO_PAUSE,
	PROTO_PFC,
	PROTO_VLAN,
	PROTO_ARP,
	PROTO_MPLS,
	PROTO_IP4,
	PROTO_ICMP4,
	PROTO_IP6,
	PROTO_ICMP6,
	PROTO_UDP,
	PROTO_TCP,
	__PROTO_MAX,
};

enum proto_layer {
	PROTO_L0, /* invalid layer */
	PROTO_L2,
	PROTO_L3,
	PROTO_L4,
};

struct proto_field;
struct proto_hdr;

struct proto_ops {
	enum proto_id id;
	enum proto_layer layer;

	void (*header_init)(struct proto_hdr *hdr);
	void (*header_finish)(struct proto_hdr *hdr);
	void (*field_changed)(struct proto_field *field);
	void (*packet_finish)(struct proto_hdr *hdr);
	void (*packet_update)(struct proto_hdr *hdr);
	void (*set_next_proto)(struct proto_hdr *hdr, enum proto_id pid);
};

struct proto_hdr {
	const struct proto_ops *ops;
	uint16_t pkt_offset;
	uint32_t pkt_id;
	uint32_t index;
	struct proto_field *fields;
	size_t fields_count;
	bool is_csum_valid;
	size_t len;
};

enum proto_field_func_t {
	PROTO_FIELD_FUNC_INC = 1 << 0,
	PROTO_FIELD_FUNC_MIN = 1 << 1,
	PROTO_FIELD_FUNC_RND = 1 << 2,
};

struct proto_field_func {
	enum proto_field_func_t type;
	uint32_t min;
	uint32_t max;
	int32_t inc;
	uint32_t val;

	void (*update_field)(struct proto_field *field);
};

struct proto_field {
	uint32_t id;
	size_t len;
	uint32_t shift;
	uint32_t mask;
	/* might be negative (e.g. VLAN TPID field) */
	int16_t offset;

	struct proto_field_func func;
	bool is_set;
	uint16_t pkt_offset;
	struct proto_hdr *hdr;
};

extern void protos_init(const char *dev);
extern void proto_ops_register(const struct proto_ops *ops);

extern struct proto_hdr *proto_header_push(enum proto_id pid);
extern void proto_header_finish(struct proto_hdr *hdr);
extern void proto_packet_finish(void);
extern void proto_packet_update(uint32_t idx);

extern struct proto_hdr *proto_lower_default_add(struct proto_hdr *hdr,
						 enum proto_id pid);

extern struct proto_hdr *proto_lower_header(struct proto_hdr *hdr);
extern struct proto_hdr *proto_upper_header(struct proto_hdr *hdr);
extern uint8_t *proto_header_ptr(struct proto_hdr *hdr);

extern void proto_header_fields_add(struct proto_hdr *hdr,
				    const struct proto_field *fields,
				    size_t count);

extern bool proto_field_is_set(struct proto_hdr *hdr, uint32_t fid);
extern void proto_field_set_bytes(struct proto_hdr *hdr, uint32_t fid,
				  const uint8_t *bytes);
extern void proto_field_set_u8(struct proto_hdr *hdr, uint32_t fid, uint8_t val);
extern uint8_t proto_field_get_u8(struct proto_hdr *hdr, uint32_t fid);
extern void proto_field_set_u16(struct proto_hdr *hdr, uint32_t fid, uint16_t val);
extern uint16_t proto_field_get_u16(struct proto_hdr *hdr, uint32_t fid);
extern void proto_field_set_u32(struct proto_hdr *hdr, uint32_t fid, uint32_t val);
extern uint32_t proto_field_get_u32(struct proto_hdr *hdr, uint32_t fid);

extern void proto_field_set_default_bytes(struct proto_hdr *hdr, uint32_t fid,
					  const uint8_t *bytes);
extern void proto_field_set_default_u8(struct proto_hdr *hdr, uint32_t fid,
				       uint8_t val);
extern void proto_field_set_default_u16(struct proto_hdr *hdr, uint32_t fid,
				        uint16_t val);
extern void proto_field_set_default_u32(struct proto_hdr *hdr, uint32_t fid,
				        uint32_t val);

extern void proto_field_set_be16(struct proto_hdr *hdr, uint32_t fid, uint16_t val);
extern void proto_field_set_be32(struct proto_hdr *hdr, uint32_t fid, uint32_t val);

extern void proto_field_set_default_be16(struct proto_hdr *hdr, uint32_t fid,
					 uint16_t val);
extern void proto_field_set_default_be32(struct proto_hdr *hdr, uint32_t fid,
					 uint32_t val);

extern void proto_field_set_dev_mac(struct proto_hdr *hdr, uint32_t fid);
extern void proto_field_set_default_dev_mac(struct proto_hdr *hdr, uint32_t fid);

extern void proto_field_set_dev_ipv4(struct proto_hdr *hdr, uint32_t fid);
extern void proto_field_set_default_dev_ipv4(struct proto_hdr *hdr, uint32_t fid);

extern void proto_field_set_dev_ipv6(struct proto_hdr *hdr, uint32_t fid);
extern void proto_field_set_default_dev_ipv6(struct proto_hdr *hdr, uint32_t fid);

extern void proto_field_dyn_apply(struct proto_field *field);

extern void proto_field_func_add(struct proto_hdr *hdr, uint32_t fid,
				 struct proto_field_func *func);

extern struct proto_field *proto_field_by_id(struct proto_hdr *hdr, uint32_t fid);

#endif /* TRAFGEN_PROTO_H */
/rdma/cxgb3-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/cxgb3-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/cxgb4-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>cxgb4-abi.h</a></td><td class='ls-size'>2488</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/cxgb4-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/cxgb4-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>d---------</td><td><a class='ls-dir' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/hfi?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>hfi</a></td><td class='ls-size'>73</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/hfi?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/hfi?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/hns-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>hns-abi.h</a></td><td class='ls-size'>1795</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/hns-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/hns-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/ib_user_cm.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>ib_user_cm.h</a></td><td class='ls-size'>6526</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/ib_user_cm.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/ib_user_cm.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/ib_user_mad.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>ib_user_mad.h</a></td><td class='ls-size'>8791</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/ib_user_mad.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/ib_user_mad.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/ib_user_sa.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>ib_user_sa.h</a></td><td class='ls-size'>2222</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/ib_user_sa.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/ib_user_sa.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/ib_user_verbs.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>ib_user_verbs.h</a></td><td class='ls-size'>21586</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/ib_user_verbs.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/ib_user_verbs.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/mlx4-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>mlx4-abi.h</a></td><td class='ls-size'>2864</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/mlx4-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/mlx4-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/mlx5-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>mlx5-abi.h</a></td><td class='ls-size'>7480</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/mlx5-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/mlx5-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/mthca-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>mthca-abi.h</a></td><td class='ls-size'>2932</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/mthca-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/mthca-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/nes-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>nes-abi.h</a></td><td class='ls-size'>3380</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/nes-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/nes-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/ocrdma-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>ocrdma-abi.h</a></td><td class='ls-size'>3918</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/ocrdma-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/ocrdma-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/qedr-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>qedr-abi.h</a></td><td class='ls-size'>2559</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/qedr-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/qedr-abi.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>
<tr><td class='ls-mode'>-rw-r--r--</td><td><a class='ls-blob h' href='/cgit.cgi/linux/net-next.git/tree/include/uapi/rdma/rdma_netlink.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>rdma_netlink.h</a></td><td class='ls-size'>4980</td><td><a class='button' href='/cgit.cgi/linux/net-next.git/log/include/uapi/rdma/rdma_netlink.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>log</a><a class='button' href='/cgit.cgi/linux/net-next.git/plain/include/uapi/rdma/rdma_netlink.h?h=nds-private-remove&amp;id=bbf892849678ff9e25590cbf78d0202e35d7e41d'>plain</a></td></tr>