DhcpClient.h

Go to the documentation of this file.

/*
 * Copyright (C) 2020 Niek Linnenbank
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef __BIN_DHCPC_DHCPCLIENT_H
#define __BIN_DHCPC_DHCPCLIENT_H
 
#include <NetworkClient.h>
#include <POSIXApplication.h>
 
class DhcpClient : public POSIXApplication
{
  private:
 
    static const u16 ServerPort = 67;
 
    static const u16 ClientPort = 68;
 
    static const Size MaximumRetries = 25;
 
    static const u32 MagicValue = 0x63825363;
 
    static const Size ReceiveTimeoutMs = 500;
 
    struct Header
    {
        u8 operation;
        u8 hardwareType;
        u8 hardwareLength;
        u8 hopCount;
        u32 transactionId;
        u16 seconds;
        u16 flags;
        u32 clientAddress;
        u32 yourAddress;
        u32 serverAddress;
        u32 gatewayAddress;
        u8 clientHardware[16];
        u8 padding[192];
        u32 magic;
    };
 
    enum Operation
    {
        BootRequest  = 1,
        BootResponse = 2
    };
 
    enum MessageType
    {
        Discover = 1,
        Offer    = 2,
        Request  = 3,
        Decline  = 4,
        Ack      = 5,
        Nak      = 6,
        Release  = 7
    };
 
    enum Options
    {
        SubnetMask           = 1,
        Router               = 3,
        DomainNameServer     = 6,
        RequestedIP          = 50,
        DhcpMessageType      = 53,
        ServerIdentifier     = 54,
        ParameterRequestList = 55,
        EndMark              = 255
    };
 
  public:
 
    DhcpClient(int argc, char **argv);
 
    virtual ~DhcpClient();
 
    virtual Result initialize();
 
    virtual Result exec();
 
  private:
 
    Result setIpAddress(const char *device,
                        const IPV4::Address ipAddr) const;
 
    Result discover(const IPV4::Address &ipAddr,
                    const IPV4::Address &ipServer,
                    const IPV4::Address &ipGateway) const;
 
    Result offer(IPV4::Address &ipAddr,
                 IPV4::Address &ipServer,
                 IPV4::Address &ipGateway) const;
 
    Result request(const IPV4::Address &ipAddr,
                   const IPV4::Address &ipServer,
                   const IPV4::Address &ipGateway) const;
 
    Result acknowledge(IPV4::Address &ipAddr,
                       IPV4::Address &ipServer,
                       IPV4::Address &ipGateway) const;
 
    Result sendBootRequest(const DhcpClient::MessageType messageType,
                           const IPV4::Address & ipAddr,
                           const IPV4::Address & ipServer,
                           const IPV4::Address & ipGateway) const;
 
    Result receiveBootResponse(const DhcpClient::MessageType messageType,
                               IPV4::Address &ipAddr,
                               IPV4::Address &ipServer,
                               IPV4::Address &ipGateway) const;
 
    Result udpSend(const void *packet,
                   const Size size) const;
 
    Result udpReceive(void *packet,
                      Size & size) const;
 
  private:
 
    NetworkClient *m_client;
 
    int m_socket;
 
    Ethernet::Address m_etherAddress;
 
    u32 m_transactionId;
};
 
#endif /* __BIN_DHCPC_DHCPCLIENT_H */