#pragma once

#include <linux/if.h>
#include <linux/if_packet.h>
#include <net/ethernet.h>
#include <netinet/in.h>
#include <sys/ioctl.h>
#include <sys/socket.h>

#include <functional>
#include <string>

namespace net
{

class IFaceBase
{
  public:
    explicit IFaceBase(const std::string& name);
    virtual ~IFaceBase() = default;

    /** @brief Get the index of the network interface corresponding
     * to this object.
     */
    int get_index() const;

    /** @brief Set interface flags using provided socket.
     *
     *  @param[in] sockfd - Socket to use for SIOCSIFFLAGS ioctl call.
     *  @param[in] flags - Flags to set.
     */
    int set_sock_flags(int sockfd, short flags) const;

    /** @brief Clear interface flags using provided socket.
     *
     *  @param[in] sockfd - Socket to use for SIOCSIFFLAGS/SIOCGIFFLAGS
     *      ioctl call.
     *  @param[in] flags - Flags to clear.
     */
    int clear_sock_flags(int sockfd, short flags) const;

    /** @brief Bind given socket to this interface. Similar to bind
     *     syscall, except that it fills in sll_ifindex field
     *     of struct sockaddr_ll with the index of this interface.
     */
    virtual int bind_sock(int sockfd, struct sockaddr_ll* saddr) const = 0;

  protected:
    std::string name_;

  private:
    /** @brief Similar to ioctl syscall, but the socket is created inside
     *      the function and the interface name in struct ifreq is
     *      properly populated with the index of this interface.
     */
    virtual int ioctl(int request, struct ifreq* ifr) const = 0;

    /** @brief Similar to ioctl syscall. The interface index in
     *      struct ifreq is
     *      properly populated with the index of this interface.
     */
    virtual int ioctl_sock(int sockfd, int request,
                           struct ifreq* ifr) const = 0;

    /** @brief Modify interface flags, using the given socket for
     *      ioctl call.
     */
    int mod_sock_flags(int sockfd, short flags, bool set) const;
};

class IFace : public IFaceBase
{
  public:
    explicit IFace(const std::string& name) : IFaceBase(name)
    {}

    /** @brief Bind given socket to this interface. Similar to bind
     *     syscall, except that it fills in sll_ifindex field
     *     of struct sockaddr_ll with the index of this interface.
     */
    int bind_sock(int sockfd, struct sockaddr_ll* saddr) const override;

  private:
    /** @brief Similar to ioctl syscall, but the socket is created inside
     *      the function and the interface name in struct ifreq is
     *      properly populated with the index of this interface.
     */
    int ioctl(int request, struct ifreq* ifr) const override;
    /** @brief Similar to ioctl syscall. The interface index in
     *      struct ifreq is
     *      properly populated with the index of this interface.
     */
    int ioctl_sock(int sockfd, int request, struct ifreq* ifr) const override;
};

} // namespace net