#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