/*
 * Copyright 2006, Marcus Overhagen <marcus@overhagen.de. All rights reserved.
 * Distributed under the terms of the MIT License.
 */

#ifndef _PXE_UNDI_H
#define _PXE_UNDI_H

extern "C" uint16 call_pxe_bios(void *pxe, uint16 opcode, void *param);

#define UNDI_OPEN				0x0006
#define UNDI_TRANSMIT			0x0008
#define UNDI_GET_INFORMATION	0x000C
#define UNDI_ISR				0x0014
#define UNDI_GET_STATE			0x0015

#define	TFTP_OPEN				0x0020
#define	TFTP_CLOSE				0x0021
#define	TFTP_READ				0x0022
#define	TFTP_GET_FILE_SIZE		0x0025

#define GET_CACHED_INFO			0x0071

#define SEG(ptr)	((uint16)(((uint32)(ptr)) >> 4))
#define OFS(ptr)	((uint16)(((uint32)(ptr)) & 15))

#define MAC_ADDR_LEN 16
typedef uint8 MAC_ADDR[MAC_ADDR_LEN];

struct SEGSEL
{
	uint16 a;
};

struct SEGOFF16
{
	uint16 ofs;
	uint16 seg;
};

struct SEGDESC
{
	uint64 a;
};

struct PXE_STRUCT
{
	uint32	Signature;
	uint8	StructLength;
	uint8	StructCksum;
	uint8	StructRev;
	uint8	reserved_1;
  	SEGOFF16 UNDIROMID;
	SEGOFF16 BaseROMID;
	SEGOFF16 EntryPointSP;
	SEGOFF16 EntryPointESP;
	SEGOFF16 StatusCallout;
	uint8	reserved_2;
	uint8	SegDescCnt;
	SEGSEL	FirstSelector;
	SEGDESC	Stack;
	SEGDESC	UNDIData;
	SEGDESC	UNDICode;
	SEGDESC	UNDICodeWrite;
	SEGDESC	BC_Data;
	SEGDESC	BC_Code;
	SEGDESC	BC_CodeWrite;
};

struct PXENV_GET_CACHED_INFO
{
	uint16	Status;
	uint16	PacketType;
	#define PXENV_PACKET_TYPE_DHCP_DISCOVER 1
	#define PXENV_PACKET_TYPE_DHCP_ACK      2
	#define PXENV_PACKET_TYPE_CACHED_REPLY  3
	uint16	BufferSize;
	SEGOFF16 Buffer;
	uint16	BufferLimit;
};

struct PXENV_UNDI_MCAST_ADDRESS
{
	#define MAXNUM_MCADDR       8
	uint16	MCastAddrCount;
	MAC_ADDR McastAddr[MAXNUM_MCADDR];
};

struct PXENV_UNDI_OPEN
{
	uint16	Status;
	uint16	OpenFlag;
	uint16	PktFilter;
	#define FLTR_DIRECTED  0x0001
	#define FLTR_BRDCST    0x0002
	#define FLTR_PRMSCS    0x0004
	#define FLTR_SRC_RTG   0x0008
	PXENV_UNDI_MCAST_ADDRESS R_Mcast_Buf;
};

struct PXENV_UNDI_GET_STATE
{
	#define PXE_UNDI_GET_STATE_STARTED     1
	#define PXE_UNDI_GET_STATE_INITIALIZED 2
	#define PXE_UNDI_GET_STATE_OPENED      3
	uint16	Status;
	uint8	UNDIstate;
};

struct PXENV_UNDI_GET_INFORMATION
{
	uint16	Status;
	uint16	BaseIo;
	uint16	IntNumber;
	uint16	MaxTranUnit;
	uint16	HwType;
	#define ETHER_TYPE      1
	#define EXP_ETHER_TYPE	2
	#define IEEE_TYPE       6
	#define ARCNET_TYPE     7
	uint16	HwAddrLen;
	MAC_ADDR CurrentNodeAddress;
	MAC_ADDR PermNodeAddress;
	SEGSEL	ROMAddress;
	uint16	RxBufCt;
	uint16	TxBufCt;
};

struct PXENV_UNDI_TRANSMIT
{
	uint16	Status;
	uint8	Protocol;
	#define P_UNKNOWN	0
	#define P_IP		1
	#define P_ARP		2
	#define P_RARP		3
	uint8	XmitFlag;
	#define XMT_DESTADDR   0x0000
	#define XMT_BROADCAST  0x0001
	SEGOFF16 DestAddr;
	SEGOFF16 TBD;
	uint32 Reserved[2];
};

#define MAX_DATA_BLKS   8
struct PXENV_UNDI_TBD
{
	uint16	ImmedLength;
	SEGOFF16 Xmit;
	uint16	DataBlkCount;
	struct DataBlk
	{
		uint8	TDPtrType;
		uint8	TDRsvdByte;
		uint16	TDDataLen;
    	SEGOFF16 TDDataPtr;
	} DataBlock[MAX_DATA_BLKS];
};

struct PXENV_UNDI_ISR
{
	uint16	Status;
	uint16	FuncFlag;
	uint16	BufferLength;
	uint16	FrameLength;
	uint16	FrameHeaderLength;
	SEGOFF16 Frame;
	uint8	ProtType;
	uint8	PktType;
};

#define PXENV_UNDI_ISR_IN_START		1
#define PXENV_UNDI_ISR_IN_PROCESS	2
#define PXENV_UNDI_ISR_IN_GET_NEXT	3

/* One of these will be returned for
	PXENV_UNDI_ISR_IN_START
*/
#define PXENV_UNDI_ISR_OUT_OURS		0
#define PXENV_UNDI_USR_OUT_NOT_OURS	1

/* One of these will be returned for
	PXENV_UNDI_ISR_IN_PROCESS and
	PXENV_UNDI_ISR_IN_GET_NEXT
*/
#define PXENV_UNDI_ISR_OUT_DONE      0
#define PXENV_UNDI_ISR_OUT_TRANSMIT  2
#define PXENV_UNDI_ISR_OUT_RECEIVE   3
#define PXENV_UNDI_ISR_OUT_BUSY      4


typedef union {
	uint32	num;
	uint8	array[4];
} pxenv_ip4;

struct pxenv_tftp_open {
	uint16		status;
	pxenv_ip4	server_ip;
	pxenv_ip4	gateway_ip;
	char		file_name[128];
	uint16		port;
	uint16		packet_size;
} __attribute__((packed));

struct pxenv_tftp_close {
	uint16		status;
} __attribute__((packed));

struct pxenv_tftp_read {
	uint16		status;
	uint16		packet_number;
	uint16		buffer_size;
	SEGOFF16	buffer;
} __attribute__((packed));

struct pxenv_tftp_get_fsize {
	uint16		status;
	pxenv_ip4	server_ip;
	pxenv_ip4	gateway_ip;
	char		file_name[128];
	uint32		file_size;
} __attribute__((packed));


PXE_STRUCT * pxe_undi_find_data();

#endif