155 lines
3.1 KiB
Go
155 lines
3.1 KiB
Go
package overlay
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
|
|
"encoding/json"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
|
|
"github.com/Microsoft/hcsshim"
|
|
"github.com/docker/libnetwork/types"
|
|
)
|
|
|
|
const ovPeerTable = "overlay_peer_table"
|
|
|
|
func (d *driver) pushLocalDb() {
|
|
if !d.isSerfAlive() {
|
|
return
|
|
}
|
|
|
|
d.Lock()
|
|
networks := d.networks
|
|
d.Unlock()
|
|
|
|
for _, n := range networks {
|
|
n.Lock()
|
|
endpoints := n.endpoints
|
|
n.Unlock()
|
|
|
|
for _, ep := range endpoints {
|
|
if !ep.remote {
|
|
d.notifyCh <- ovNotify{
|
|
action: "join",
|
|
nw: n,
|
|
ep: ep,
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (d *driver) peerAdd(nid, eid string, peerIP net.IP, peerIPMask net.IPMask,
|
|
peerMac net.HardwareAddr, vtep net.IP, updateDb bool) error {
|
|
|
|
log.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.Info("WINOVERLAY: peerAdd: notifying HNS of the REMOTE endpoint")
|
|
|
|
hnsEndpoint := &hcsshim.HNSEndpoint{
|
|
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 a 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)
|
|
|
|
if err := d.writeEndpointToStore(ep); err != nil {
|
|
return fmt.Errorf("failed to update overlay endpoint %s to local store: %v", ep.id[0:7], err)
|
|
}
|
|
}
|
|
|
|
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.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)
|
|
|
|
if err := d.deleteEndpointFromStore(ep); err != nil {
|
|
log.Debugf("Failed to delete stale overlay endpoint (%s) from store", ep.id[0:7])
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|