#ifndef _PLY_IR_H
#define _PLY_IR_H

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>

#include <bits/wordsize.h>
#include <linux/bpf.h>

#define INSN(_code, _dst, _src, _off, _imm)	\
	((struct bpf_insn) {			\
		.code  = _code,			\
		.dst_reg = _dst,		\
		.src_reg = _src,		\
		.off   = _off,			\
		.imm   = _imm			\
	})

#define MOV(_dst, _src)     INSN(BPF_ALU64 | BPF_MOV | BPF_X, _dst, _src, 0, 0)
#define MOV_IMM(_dst, _imm) INSN(BPF_ALU64 | BPF_MOV | BPF_K, _dst, 0, 0, _imm)

#define EXIT INSN(BPF_JMP | BPF_EXIT, 0, 0, 0, 0)
#define CALL(_imm) INSN(BPF_JMP | BPF_CALL, 0, 0, 0, _imm)
#define JMP(_op, _dst, _src, _off)     INSN(BPF_JMP | BPF_OP((_op)) | BPF_X, _dst, _src, _off, 0)
#define JMP_IMM(_op, _dst, _imm, _off) INSN(BPF_JMP | BPF_OP((_op)) | BPF_K, _dst, 0, _off, _imm)

#define ALU32(_op, _dst, _src)     INSN(BPF_ALU | BPF_OP((_op)) | BPF_X, _dst, _src, 0, 0)
#define ALU32_IMM(_op, _dst, _imm) INSN(BPF_ALU | BPF_OP((_op)) | BPF_K, _dst, 0, 0, _imm)

#define ALU64(_op, _dst, _src)     INSN(BPF_ALU64 | BPF_OP((_op)) | BPF_X, _dst, _src, 0, 0)
#define ALU64_IMM(_op, _dst, _imm) INSN(BPF_ALU64 | BPF_OP((_op)) | BPF_K, _dst, 0, 0, _imm)

#define STW_IMM(_dst, _off, _imm) INSN(BPF_ST  | BPF_SIZE(BPF_W)  | BPF_MEM, _dst, 0, _off, _imm)
#define STXB(_dst, _off, _src)   INSN(BPF_STX | BPF_SIZE(BPF_B) | BPF_MEM, _dst, _src, _off, 0)
#define STXH(_dst, _off, _src)   INSN(BPF_STX | BPF_SIZE(BPF_H) | BPF_MEM, _dst, _src, _off, 0)
#define STXW(_dst, _off, _src)   INSN(BPF_STX | BPF_SIZE(BPF_W) | BPF_MEM, _dst, _src, _off, 0)
#define STXDW(_dst, _off, _src)   INSN(BPF_STX | BPF_SIZE(BPF_DW) | BPF_MEM, _dst, _src, _off, 0)

#define LDXB(_dst, _off, _src)  INSN(BPF_LDX | BPF_SIZE(BPF_B)  | BPF_MEM, _dst, _src, _off, 0)
#define LDXH(_dst, _off, _src)  INSN(BPF_LDX | BPF_SIZE(BPF_H)  | BPF_MEM, _dst, _src, _off, 0)
#define LDXW(_dst, _off, _src)  INSN(BPF_LDX | BPF_SIZE(BPF_W)  | BPF_MEM, _dst, _src, _off, 0)
#define LDXDW(_dst, _off, _src) INSN(BPF_LDX | BPF_SIZE(BPF_DW) | BPF_MEM, _dst, _src, _off, 0)

#if __WORDSIZE == 64
#  define ALU ALU64
#  define ALU_IMM ALU64_IMM
#  define STX STXDW
#  define LDX LDXDW
#else
#  define ALU ALU32
#  define ALU_IMM ALU32_IMM
#  define STX STXW
#  define LDX LDXW
#endif

#define BPF_REG_BP BPF_REG_10

struct sym;

enum vitype {
	VI_INSN,
	VI_LDMAP,
	VI_LABEL,
	VI_REG_GET,
	VI_REG_PUT,
};

struct vinsn {
	enum vitype vitype;

	union {
		struct {
			struct bpf_insn bpf;
			uint16_t dst;
			uint16_t src;
		} insn;

		int16_t label;
		uint16_t reg;

		struct {
			uint16_t reg;
			struct sym *sym;
		} map;
	};
};

struct ir {
	struct vinsn *vi;
	size_t len;

	int16_t next_label;
	uint16_t next_reg;

	ssize_t sp;
};

enum irloc {
	LOC_IMM   = (1 << 0),
	LOC_REG   = (1 << 1),
	LOC_STACK = (1 << 2),
};


struct irstate {
	int loc;

	uint16_t reg;
	int32_t  stack;
	int32_t  imm;

	struct {
		int dot:1;
		int stack:1;
	} hint;
};

void insn_dump(struct bpf_insn insn, FILE *fp);
void vinsn_dump(struct vinsn *vi, FILE *fp);
void ir_dump(struct ir *ir, FILE *fp);

int16_t  ir_alloc_label   (struct ir *ir);
uint16_t ir_alloc_register(struct ir *ir);
ssize_t  ir_alloc_stack   (struct ir *ir, size_t size);

void ir_emit_insn   (struct ir *ir, struct bpf_insn bpf, uint16_t dst, uint16_t src);
void ir_emit_ldmap  (struct ir *ir, uint16_t dst, struct sym *map);
void ir_emit_label  (struct ir *ir, int16_t label);
/* void ir_emit_reg_get(struct ir *ir, uint16_t reg); */
/* void ir_emit_reg_put(struct ir *ir, uint16_t reg); */
void ir_emit_mov_irs(struct ir *ir, uint16_t dst, struct irstate *src);

struct ir *ir_new(void);


static inline void irs_alloc_register(struct irstate *irs, struct ir *ir)
{
	irs->loc = LOC_REG;
	irs->reg = ir_alloc_register(ir);
}

static inline void irs_alloc_stack(struct irstate *irs, struct ir *ir,
				   size_t size)
{
	irs->loc = LOC_STACK;
	irs->stack = ir_alloc_stack(ir, size);
}

#endif	/* _PLY_IR_H */