#ifndef _PXE_NETWORK_H
#define _PXE_NETWORK_H
/*
 * Copyright 2006, Marcus Overhagen <marcus@overhagen.de. All rights reserved.
 * Copyright 2005-2007, Ingo Weinhold <bonefish@cs.tu-berlin.de>.
 * Distributed under the terms of the MIT License.
 */

#include <OS.h>
#include <boot/net/Ethernet.h>
#include <boot/net/NetStack.h>

struct PXE_STRUCT;


class PXEService {
protected:
								PXEService();
								~PXEService();

			status_t			Init();

public:
			mac_addr_t			MACAddress() const		{ return fMACAddress; }
			ip_addr_t			IPAddress() const		{ return fClientIP; }
			ip_addr_t			ServerIPAddress() const	{ return fServerIP; }
			const char*			RootPath() const		{ return fRootPath; }

protected:
			PXE_STRUCT*			fPxeData;
			ip_addr_t			fClientIP;
			ip_addr_t			fServerIP;
			mac_addr_t			fMACAddress;
			char*				fRootPath;
};


class UNDI : public EthernetInterface, public PXEService
{
public:
						UNDI();
	virtual 			~UNDI();

	status_t 			Init();

	virtual mac_addr_t	MACAddress() const;

	virtual	void *		AllocateSendReceiveBuffer(size_t size);
	virtual	void 		FreeSendReceiveBuffer(void *buffer);

	virtual ssize_t		Send(const void *buffer, size_t size);
	virtual ssize_t		Receive(void *buffer, size_t size);

private:
	bool				fRxFinished;
	PXE_STRUCT *		fPxeData;
};


class TFTP : public PXEService {
public:
								TFTP();
								~TFTP();

			status_t			Init();

			uint16				ServerPort() const;

			status_t			ReceiveFile(const char* fileName,
									uint8** data, size_t* size);

private:
			status_t			_Close();
};

#endif