#ifndef __NATIVE_CODE_H
#define __NATIVE_CODE_H

#include "Windas.h"
#include "XedWrap.h"
#include "Code.h"

struct _NATIVE_CODE_BLOCK;

typedef struct _NATIVE_CODE_LINK
{
	_NATIVE_CODE_LINK*	Next;
	_NATIVE_CODE_LINK*	Prev;
	_NATIVE_CODE_BLOCK* Block;
	ULONG				Flags;
	ULONG				Label;
	PUCHAR				RawData;
	ULONG				RawDataSize;
	XED_DECODED_INST	XedInst;
	_NATIVE_CODE_LINK();
	_NATIVE_CODE_LINK(ULONG LabelId, _NATIVE_CODE_BLOCK* B);
	~_NATIVE_CODE_LINK();
}NATIVE_CODE_LINK, *PNATIVE_CODE_LINK;

typedef struct _NATIVE_CODE_BLOCK
{
	PNATIVE_CODE_LINK Start;
	PNATIVE_CODE_LINK End;
	BOOL HasRelativeJumps;
	_NATIVE_CODE_BLOCK();
}NATIVE_CODE_BLOCK, *PNATIVE_CODE_BLOCK;

VOID NcInsertLinkAfter(PNATIVE_CODE_LINK Link1, PNATIVE_CODE_LINK Link2);

VOID NcInsertLinkBefore(PNATIVE_CODE_LINK Link1, PNATIVE_CODE_LINK Link2);

VOID NcUnlink(PNATIVE_CODE_LINK Link);

VOID NcConcat(PNATIVE_CODE_BLOCK Block1, PNATIVE_CODE_BLOCK Block2);

BOOL NcInsertBlockAfter(PNATIVE_CODE_LINK Link, PNATIVE_CODE_BLOCK Block);

BOOL NcInsertBlockBefore(PNATIVE_CODE_LINK Link, PNATIVE_CODE_BLOCK Block);

BOOL NcCreateLabels(PNATIVE_CODE_BLOCK Block);

PNATIVE_CODE_LINK NcValidateJmp(PNATIVE_CODE_LINK Link, INT32 Delta);

BOOL NcFromBuffer(PNATIVE_CODE_BLOCK Block, PVOID Buffer, ULONG BufferSize);

VOID NcDelete(PNATIVE_CODE_BLOCK Block);

VOID NcDebugPrint(PNATIVE_CODE_BLOCK Block);

VOID NcPrintBlockCode(PNATIVE_CODE_BLOCK Block);

#endif