dotcloud/docker

package overlay

import (
    "context"
    "encoding/json"
    "fmt"
    "net"

    "github.com/Microsoft/hcsshim"
    "github.com/containerd/log"
    "github.com/docker/docker/libnetwork/types"
)

const ovPeerTable = "overlay_peer_table"

func (d *driver) peerAdd(nid, eid string, peerIP net.IP, peerIPMask net.IPMask, peerMac net.HardwareAddr, vtep net.IP, updateDb bool) error {
    log.G(context.TODO()).Debugf("WINOVERLAY: Enter peerAdd for ca ip %s with ca mac %s", peerIP.String(), peerMac.String())

    if err := validateID(nid, eid); err != nil {
        return err
    }

    n := d.network(nid)
    if n == nil {
        return nil
    }

    if updateDb {
        log.G(context.TODO()).Info("WINOVERLAY: peerAdd: notifying HNS of the REMOTE endpoint")

        hnsEndpoint := &hcsshim.HNSEndpoint{
            Name:             eid,
            VirtualNetwork:   n.hnsID,
            MacAddress:       peerMac.String(),
            IPAddress:        peerIP,
            IsRemoteEndpoint: true,
        }

        paPolicy, err := json.Marshal(hcsshim.PaPolicy{
            Type: "PA",
            PA:   vtep.String(),
        })
        if err != nil {
            return err
        }

        hnsEndpoint.Policies = append(hnsEndpoint.Policies, paPolicy)

        configurationb, err := json.Marshal(hnsEndpoint)
        if err != nil {
            return err
        }

        // Temp: We have to create an endpoint object to keep track of the HNS ID for
        // this endpoint so that we can retrieve it later when the endpoint is deleted.
        // This seems unnecessary when we already have dockers EID. See if we can pass
        // the global EID to HNS to use as it's ID, rather than having each HNS assign
        // it's own local ID for the endpoint

        addr, err := types.ParseCIDR(peerIP.String() + "/32")
        if err != nil {
            return err
        }

        n.removeEndpointWithAddress(addr)
        hnsresponse, err := hcsshim.HNSEndpointRequest("POST", "", string(configurationb))
        if err != nil {
            return err
        }

        ep := &endpoint{
            id:        eid,
            nid:       nid,
            addr:      addr,
            mac:       peerMac,
            profileID: hnsresponse.Id,
            remote:    true,
        }

        n.addEndpoint(ep)
    }

    return nil
}

func (d *driver) peerDelete(nid, eid string, peerIP net.IP, peerIPMask net.IPMask, peerMac net.HardwareAddr, vtep net.IP, updateDb bool) error {
    log.G(context.TODO()).Infof("WINOVERLAY: Enter peerDelete for endpoint %s and peer ip %s", eid, peerIP.String())

    if err := validateID(nid, eid); err != nil {
        return err
    }

    n := d.network(nid)
    if n == nil {
        return nil
    }

    ep := n.endpoint(eid)
    if ep == nil {
        return fmt.Errorf("could not find endpoint with id %s", eid)
    }

    if updateDb {
        _, err := hcsshim.HNSEndpointRequest("DELETE", ep.profileID, "")
        if err != nil {
            return err
        }

        n.deleteEndpoint(eid)
    }

    return nil
}