networkgateway.h

/*
 * Copyright (C) 2016-2017 Pelagicore AB
 *
 * Permission to use, copy, modify, and/or distribute this software for
 * any purpose with or without fee is hereby granted, provided that the
 * above copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
 * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR
 * BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES
 * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
 * WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
 * ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
 * SOFTWARE.
 *
 * For further information see LICENSE
 */

#pragma once

#include "gateway/gateway.h"
#include "jansson.h"
#include "netlink.h"
#include "iptableentry.h"
#include "softwarecontainererror.h"
#include "networkgatewayfunctions.h"

namespace softwarecontainer {

class NetworkGateway :
        public Gateway
{
    LOG_DECLARE_CLASS_CONTEXT("NETG", "Network gateway");

public:
    static constexpr const char *ID = "network";

    // Exit codes used when running inside container
    static const constexpr int FAILURE = -1;
    static const constexpr int SUCCESS = 0;
    static const constexpr int NO_LINK = 1; // The desired link was not found
    static const constexpr int BAD_LINKUP = 2; // Could not bring link up
    static const constexpr int BAD_LINKDOWN = 3; // Could not bring link up

    NetworkGateway(const int32_t id,
                   const std::string bridgeDevice,
                   const std::string gateway,
                   const uint8_t maskBits,
                   std::shared_ptr<ContainerAbstractInterface> container);

    virtual ~NetworkGateway();

    bool readConfigElement(const json_t *element) override;

    bool activateGateway() override;

    bool teardownGateway() override;
private:
    bool setDefaultGateway();

    bool up();

    bool down();

    virtual bool isBridgeAvailable();

    struct in_addr m_ip;
    uint32_t m_netmask;
    std::string m_gateway;
    std::string m_bridgeDevice;

    std::vector<IPTableEntry> m_entries;

    bool m_interfaceInitialized;

    int32_t m_containerID;
    Netlink m_netlinkHost;

    NetworkGatewayFunctions m_functions;
};

class NetworkGatewayError : public SoftwareContainerError
{
public:
    NetworkGatewayError():
        m_message("NetworkGateway exception")
    {
    }

    NetworkGatewayError(const std::string &message):
        m_message(message)
    {
    }

    virtual const char *what() const throw()
    {
        return m_message.c_str();
    }

protected:
    std::string m_message;
};

class IPAllocationError : public NetworkGatewayError
{
public:
    IPAllocationError():
        NetworkGatewayError("An error is occurred when trying to generate IP address")
    {
    }

    IPAllocationError(const std::string &message):
        NetworkGatewayError(message)
    {
    }
};

} // namespace softwarecontainer