agent.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978
  1. package libnetwork
  2. //go:generate protoc -I.:Godeps/_workspace/src/github.com/gogo/protobuf --gogo_out=import_path=github.com/docker/libnetwork,Mgogoproto/gogo.proto=github.com/gogo/protobuf/gogoproto:. agent.proto
  3. import (
  4. "encoding/json"
  5. "fmt"
  6. "net"
  7. "sort"
  8. "sync"
  9. "github.com/docker/go-events"
  10. "github.com/docker/libnetwork/cluster"
  11. "github.com/docker/libnetwork/datastore"
  12. "github.com/docker/libnetwork/discoverapi"
  13. "github.com/docker/libnetwork/driverapi"
  14. "github.com/docker/libnetwork/networkdb"
  15. "github.com/docker/libnetwork/types"
  16. "github.com/gogo/protobuf/proto"
  17. "github.com/sirupsen/logrus"
  18. )
  19. const (
  20. subsysGossip = "networking:gossip"
  21. subsysIPSec = "networking:ipsec"
  22. keyringSize = 3
  23. )
  24. // ByTime implements sort.Interface for []*types.EncryptionKey based on
  25. // the LamportTime field.
  26. type ByTime []*types.EncryptionKey
  27. func (b ByTime) Len() int { return len(b) }
  28. func (b ByTime) Swap(i, j int) { b[i], b[j] = b[j], b[i] }
  29. func (b ByTime) Less(i, j int) bool { return b[i].LamportTime < b[j].LamportTime }
  30. type agent struct {
  31. networkDB *networkdb.NetworkDB
  32. bindAddr string
  33. advertiseAddr string
  34. dataPathAddr string
  35. coreCancelFuncs []func()
  36. driverCancelFuncs map[string][]func()
  37. sync.Mutex
  38. }
  39. func (a *agent) dataPathAddress() string {
  40. a.Lock()
  41. defer a.Unlock()
  42. if a.dataPathAddr != "" {
  43. return a.dataPathAddr
  44. }
  45. return a.advertiseAddr
  46. }
  47. const libnetworkEPTable = "endpoint_table"
  48. func getBindAddr(ifaceName string) (string, error) {
  49. iface, err := net.InterfaceByName(ifaceName)
  50. if err != nil {
  51. return "", fmt.Errorf("failed to find interface %s: %v", ifaceName, err)
  52. }
  53. addrs, err := iface.Addrs()
  54. if err != nil {
  55. return "", fmt.Errorf("failed to get interface addresses: %v", err)
  56. }
  57. for _, a := range addrs {
  58. addr, ok := a.(*net.IPNet)
  59. if !ok {
  60. continue
  61. }
  62. addrIP := addr.IP
  63. if addrIP.IsLinkLocalUnicast() {
  64. continue
  65. }
  66. return addrIP.String(), nil
  67. }
  68. return "", fmt.Errorf("failed to get bind address")
  69. }
  70. func resolveAddr(addrOrInterface string) (string, error) {
  71. // Try and see if this is a valid IP address
  72. if net.ParseIP(addrOrInterface) != nil {
  73. return addrOrInterface, nil
  74. }
  75. addr, err := net.ResolveIPAddr("ip", addrOrInterface)
  76. if err != nil {
  77. // If not a valid IP address, it should be a valid interface
  78. return getBindAddr(addrOrInterface)
  79. }
  80. return addr.String(), nil
  81. }
  82. func (c *controller) handleKeyChange(keys []*types.EncryptionKey) error {
  83. drvEnc := discoverapi.DriverEncryptionUpdate{}
  84. a := c.getAgent()
  85. if a == nil {
  86. logrus.Debug("Skipping key change as agent is nil")
  87. return nil
  88. }
  89. // Find the deleted key. If the deleted key was the primary key,
  90. // a new primary key should be set before removing if from keyring.
  91. c.Lock()
  92. added := []byte{}
  93. deleted := []byte{}
  94. j := len(c.keys)
  95. for i := 0; i < j; {
  96. same := false
  97. for _, key := range keys {
  98. if same = key.LamportTime == c.keys[i].LamportTime; same {
  99. break
  100. }
  101. }
  102. if !same {
  103. cKey := c.keys[i]
  104. if cKey.Subsystem == subsysGossip {
  105. deleted = cKey.Key
  106. }
  107. if cKey.Subsystem == subsysIPSec {
  108. drvEnc.Prune = cKey.Key
  109. drvEnc.PruneTag = cKey.LamportTime
  110. }
  111. c.keys[i], c.keys[j-1] = c.keys[j-1], c.keys[i]
  112. c.keys[j-1] = nil
  113. j--
  114. }
  115. i++
  116. }
  117. c.keys = c.keys[:j]
  118. // Find the new key and add it to the key ring
  119. for _, key := range keys {
  120. same := false
  121. for _, cKey := range c.keys {
  122. if same = cKey.LamportTime == key.LamportTime; same {
  123. break
  124. }
  125. }
  126. if !same {
  127. c.keys = append(c.keys, key)
  128. if key.Subsystem == subsysGossip {
  129. added = key.Key
  130. }
  131. if key.Subsystem == subsysIPSec {
  132. drvEnc.Key = key.Key
  133. drvEnc.Tag = key.LamportTime
  134. }
  135. }
  136. }
  137. c.Unlock()
  138. if len(added) > 0 {
  139. a.networkDB.SetKey(added)
  140. }
  141. key, _, err := c.getPrimaryKeyTag(subsysGossip)
  142. if err != nil {
  143. return err
  144. }
  145. a.networkDB.SetPrimaryKey(key)
  146. key, tag, err := c.getPrimaryKeyTag(subsysIPSec)
  147. if err != nil {
  148. return err
  149. }
  150. drvEnc.Primary = key
  151. drvEnc.PrimaryTag = tag
  152. if len(deleted) > 0 {
  153. a.networkDB.RemoveKey(deleted)
  154. }
  155. c.drvRegistry.WalkDrivers(func(name string, driver driverapi.Driver, capability driverapi.Capability) bool {
  156. err := driver.DiscoverNew(discoverapi.EncryptionKeysUpdate, drvEnc)
  157. if err != nil {
  158. logrus.Warnf("Failed to update datapath keys in driver %s: %v", name, err)
  159. }
  160. return false
  161. })
  162. return nil
  163. }
  164. func (c *controller) agentSetup(clusterProvider cluster.Provider) error {
  165. agent := c.getAgent()
  166. // If the agent is already present there is no need to try to initialize it again
  167. if agent != nil {
  168. return nil
  169. }
  170. bindAddr := clusterProvider.GetLocalAddress()
  171. advAddr := clusterProvider.GetAdvertiseAddress()
  172. dataAddr := clusterProvider.GetDataPathAddress()
  173. remoteList := clusterProvider.GetRemoteAddressList()
  174. remoteAddrList := make([]string, 0, len(remoteList))
  175. for _, remote := range remoteList {
  176. addr, _, _ := net.SplitHostPort(remote)
  177. remoteAddrList = append(remoteAddrList, addr)
  178. }
  179. listen := clusterProvider.GetListenAddress()
  180. listenAddr, _, _ := net.SplitHostPort(listen)
  181. logrus.Infof("Initializing Libnetwork Agent Listen-Addr=%s Local-addr=%s Adv-addr=%s Data-addr=%s Remote-addr-list=%v MTU=%d",
  182. listenAddr, bindAddr, advAddr, dataAddr, remoteAddrList, c.Config().Daemon.NetworkControlPlaneMTU)
  183. if advAddr != "" && agent == nil {
  184. if err := c.agentInit(listenAddr, bindAddr, advAddr, dataAddr); err != nil {
  185. logrus.Errorf("error in agentInit: %v", err)
  186. return err
  187. }
  188. c.drvRegistry.WalkDrivers(func(name string, driver driverapi.Driver, capability driverapi.Capability) bool {
  189. if capability.ConnectivityScope == datastore.GlobalScope {
  190. c.agentDriverNotify(driver)
  191. }
  192. return false
  193. })
  194. }
  195. if len(remoteAddrList) > 0 {
  196. if err := c.agentJoin(remoteAddrList); err != nil {
  197. logrus.Errorf("Error in joining gossip cluster : %v(join will be retried in background)", err)
  198. }
  199. }
  200. return nil
  201. }
  202. // For a given subsystem getKeys sorts the keys by lamport time and returns
  203. // slice of keys and lamport time which can used as a unique tag for the keys
  204. func (c *controller) getKeys(subsys string) ([][]byte, []uint64) {
  205. c.Lock()
  206. defer c.Unlock()
  207. sort.Sort(ByTime(c.keys))
  208. keys := [][]byte{}
  209. tags := []uint64{}
  210. for _, key := range c.keys {
  211. if key.Subsystem == subsys {
  212. keys = append(keys, key.Key)
  213. tags = append(tags, key.LamportTime)
  214. }
  215. }
  216. keys[0], keys[1] = keys[1], keys[0]
  217. tags[0], tags[1] = tags[1], tags[0]
  218. return keys, tags
  219. }
  220. // getPrimaryKeyTag returns the primary key for a given subsystem from the
  221. // list of sorted key and the associated tag
  222. func (c *controller) getPrimaryKeyTag(subsys string) ([]byte, uint64, error) {
  223. c.Lock()
  224. defer c.Unlock()
  225. sort.Sort(ByTime(c.keys))
  226. keys := []*types.EncryptionKey{}
  227. for _, key := range c.keys {
  228. if key.Subsystem == subsys {
  229. keys = append(keys, key)
  230. }
  231. }
  232. return keys[1].Key, keys[1].LamportTime, nil
  233. }
  234. func (c *controller) agentInit(listenAddr, bindAddrOrInterface, advertiseAddr, dataPathAddr string) error {
  235. bindAddr, err := resolveAddr(bindAddrOrInterface)
  236. if err != nil {
  237. return err
  238. }
  239. keys, _ := c.getKeys(subsysGossip)
  240. netDBConf := networkdb.DefaultConfig()
  241. netDBConf.BindAddr = listenAddr
  242. netDBConf.AdvertiseAddr = advertiseAddr
  243. netDBConf.Keys = keys
  244. if c.Config().Daemon.NetworkControlPlaneMTU != 0 {
  245. // Consider the MTU remove the IP hdr (IPv4 or IPv6) and the TCP/UDP hdr.
  246. // To be on the safe side let's cut 100 bytes
  247. netDBConf.PacketBufferSize = (c.Config().Daemon.NetworkControlPlaneMTU - 100)
  248. logrus.Debugf("Control plane MTU: %d will initialize NetworkDB with: %d",
  249. c.Config().Daemon.NetworkControlPlaneMTU, netDBConf.PacketBufferSize)
  250. }
  251. nDB, err := networkdb.New(netDBConf)
  252. if err != nil {
  253. return err
  254. }
  255. // Register the diagnostic handlers
  256. c.DiagnosticServer.RegisterHandler(nDB, networkdb.NetDbPaths2Func)
  257. var cancelList []func()
  258. ch, cancel := nDB.Watch(libnetworkEPTable, "", "")
  259. cancelList = append(cancelList, cancel)
  260. nodeCh, cancel := nDB.Watch(networkdb.NodeTable, "", "")
  261. cancelList = append(cancelList, cancel)
  262. c.Lock()
  263. c.agent = &agent{
  264. networkDB: nDB,
  265. bindAddr: bindAddr,
  266. advertiseAddr: advertiseAddr,
  267. dataPathAddr: dataPathAddr,
  268. coreCancelFuncs: cancelList,
  269. driverCancelFuncs: make(map[string][]func()),
  270. }
  271. c.Unlock()
  272. go c.handleTableEvents(ch, c.handleEpTableEvent)
  273. go c.handleTableEvents(nodeCh, c.handleNodeTableEvent)
  274. drvEnc := discoverapi.DriverEncryptionConfig{}
  275. keys, tags := c.getKeys(subsysIPSec)
  276. drvEnc.Keys = keys
  277. drvEnc.Tags = tags
  278. c.drvRegistry.WalkDrivers(func(name string, driver driverapi.Driver, capability driverapi.Capability) bool {
  279. err := driver.DiscoverNew(discoverapi.EncryptionKeysConfig, drvEnc)
  280. if err != nil {
  281. logrus.Warnf("Failed to set datapath keys in driver %s: %v", name, err)
  282. }
  283. return false
  284. })
  285. c.WalkNetworks(joinCluster)
  286. return nil
  287. }
  288. func (c *controller) agentJoin(remoteAddrList []string) error {
  289. agent := c.getAgent()
  290. if agent == nil {
  291. return nil
  292. }
  293. return agent.networkDB.Join(remoteAddrList)
  294. }
  295. func (c *controller) agentDriverNotify(d driverapi.Driver) {
  296. agent := c.getAgent()
  297. if agent == nil {
  298. return
  299. }
  300. if err := d.DiscoverNew(discoverapi.NodeDiscovery, discoverapi.NodeDiscoveryData{
  301. Address: agent.dataPathAddress(),
  302. BindAddress: agent.bindAddr,
  303. Self: true,
  304. }); err != nil {
  305. logrus.Warnf("Failed the node discovery in driver: %v", err)
  306. }
  307. drvEnc := discoverapi.DriverEncryptionConfig{}
  308. keys, tags := c.getKeys(subsysIPSec)
  309. drvEnc.Keys = keys
  310. drvEnc.Tags = tags
  311. if err := d.DiscoverNew(discoverapi.EncryptionKeysConfig, drvEnc); err != nil {
  312. logrus.Warnf("Failed to set datapath keys in driver: %v", err)
  313. }
  314. }
  315. func (c *controller) agentClose() {
  316. // Acquire current agent instance and reset its pointer
  317. // then run closing functions
  318. c.Lock()
  319. agent := c.agent
  320. c.agent = nil
  321. c.Unlock()
  322. // when the agent is closed the cluster provider should be cleaned up
  323. c.SetClusterProvider(nil)
  324. if agent == nil {
  325. return
  326. }
  327. var cancelList []func()
  328. agent.Lock()
  329. for _, cancelFuncs := range agent.driverCancelFuncs {
  330. cancelList = append(cancelList, cancelFuncs...)
  331. }
  332. // Add also the cancel functions for the network db
  333. cancelList = append(cancelList, agent.coreCancelFuncs...)
  334. agent.Unlock()
  335. for _, cancel := range cancelList {
  336. cancel()
  337. }
  338. agent.networkDB.Close()
  339. }
  340. // Task has the backend container details
  341. type Task struct {
  342. Name string
  343. EndpointID string
  344. EndpointIP string
  345. Info map[string]string
  346. }
  347. // ServiceInfo has service specific details along with the list of backend tasks
  348. type ServiceInfo struct {
  349. VIP string
  350. LocalLBIndex int
  351. Tasks []Task
  352. Ports []string
  353. }
  354. type epRecord struct {
  355. ep EndpointRecord
  356. info map[string]string
  357. lbIndex int
  358. }
  359. func (n *network) Services() map[string]ServiceInfo {
  360. eps := make(map[string]epRecord)
  361. if !n.isClusterEligible() {
  362. return nil
  363. }
  364. agent := n.getController().getAgent()
  365. if agent == nil {
  366. return nil
  367. }
  368. // Walk through libnetworkEPTable and fetch the driver agnostic endpoint info
  369. entries := agent.networkDB.GetTableByNetwork(libnetworkEPTable, n.id)
  370. for eid, value := range entries {
  371. var epRec EndpointRecord
  372. nid := n.ID()
  373. if err := proto.Unmarshal(value.Value, &epRec); err != nil {
  374. logrus.Errorf("Unmarshal of libnetworkEPTable failed for endpoint %s in network %s, %v", eid, nid, err)
  375. continue
  376. }
  377. i := n.getController().getLBIndex(epRec.ServiceID, nid, epRec.IngressPorts)
  378. eps[eid] = epRecord{
  379. ep: epRec,
  380. lbIndex: i,
  381. }
  382. }
  383. // Walk through the driver's tables, have the driver decode the entries
  384. // and return the tuple {ep ID, value}. value is a string that coveys
  385. // relevant info about the endpoint.
  386. d, err := n.driver(true)
  387. if err != nil {
  388. logrus.Errorf("Could not resolve driver for network %s/%s while fetching services: %v", n.networkType, n.ID(), err)
  389. return nil
  390. }
  391. for _, table := range n.driverTables {
  392. if table.objType != driverapi.EndpointObject {
  393. continue
  394. }
  395. entries := agent.networkDB.GetTableByNetwork(table.name, n.id)
  396. for key, value := range entries {
  397. epID, info := d.DecodeTableEntry(table.name, key, value.Value)
  398. if ep, ok := eps[epID]; !ok {
  399. logrus.Errorf("Inconsistent driver and libnetwork state for endpoint %s", epID)
  400. } else {
  401. ep.info = info
  402. eps[epID] = ep
  403. }
  404. }
  405. }
  406. // group the endpoints into a map keyed by the service name
  407. sinfo := make(map[string]ServiceInfo)
  408. for ep, epr := range eps {
  409. var (
  410. s ServiceInfo
  411. ok bool
  412. )
  413. if s, ok = sinfo[epr.ep.ServiceName]; !ok {
  414. s = ServiceInfo{
  415. VIP: epr.ep.VirtualIP,
  416. LocalLBIndex: epr.lbIndex,
  417. }
  418. }
  419. ports := []string{}
  420. if s.Ports == nil {
  421. for _, port := range epr.ep.IngressPorts {
  422. p := fmt.Sprintf("Target: %d, Publish: %d", port.TargetPort, port.PublishedPort)
  423. ports = append(ports, p)
  424. }
  425. s.Ports = ports
  426. }
  427. s.Tasks = append(s.Tasks, Task{
  428. Name: epr.ep.Name,
  429. EndpointID: ep,
  430. EndpointIP: epr.ep.EndpointIP,
  431. Info: epr.info,
  432. })
  433. sinfo[epr.ep.ServiceName] = s
  434. }
  435. return sinfo
  436. }
  437. func (n *network) isClusterEligible() bool {
  438. if n.scope != datastore.SwarmScope || !n.driverIsMultihost() {
  439. return false
  440. }
  441. return n.getController().getAgent() != nil
  442. }
  443. func (n *network) joinCluster() error {
  444. if !n.isClusterEligible() {
  445. return nil
  446. }
  447. agent := n.getController().getAgent()
  448. if agent == nil {
  449. return nil
  450. }
  451. return agent.networkDB.JoinNetwork(n.ID())
  452. }
  453. func (n *network) leaveCluster() error {
  454. if !n.isClusterEligible() {
  455. return nil
  456. }
  457. agent := n.getController().getAgent()
  458. if agent == nil {
  459. return nil
  460. }
  461. return agent.networkDB.LeaveNetwork(n.ID())
  462. }
  463. func (ep *endpoint) addDriverInfoToCluster() error {
  464. n := ep.getNetwork()
  465. if !n.isClusterEligible() {
  466. return nil
  467. }
  468. if ep.joinInfo == nil {
  469. return nil
  470. }
  471. agent := n.getController().getAgent()
  472. if agent == nil {
  473. return nil
  474. }
  475. for _, te := range ep.joinInfo.driverTableEntries {
  476. if err := agent.networkDB.CreateEntry(te.tableName, n.ID(), te.key, te.value); err != nil {
  477. return err
  478. }
  479. }
  480. return nil
  481. }
  482. func (ep *endpoint) deleteDriverInfoFromCluster() error {
  483. n := ep.getNetwork()
  484. if !n.isClusterEligible() {
  485. return nil
  486. }
  487. if ep.joinInfo == nil {
  488. return nil
  489. }
  490. agent := n.getController().getAgent()
  491. if agent == nil {
  492. return nil
  493. }
  494. for _, te := range ep.joinInfo.driverTableEntries {
  495. if err := agent.networkDB.DeleteEntry(te.tableName, n.ID(), te.key); err != nil {
  496. return err
  497. }
  498. }
  499. return nil
  500. }
  501. func (ep *endpoint) addServiceInfoToCluster(sb *sandbox) error {
  502. if ep.isAnonymous() && len(ep.myAliases) == 0 || ep.Iface().Address() == nil {
  503. return nil
  504. }
  505. n := ep.getNetwork()
  506. if !n.isClusterEligible() {
  507. return nil
  508. }
  509. sb.Service.Lock()
  510. defer sb.Service.Unlock()
  511. logrus.Debugf("addServiceInfoToCluster START for %s %s", ep.svcName, ep.ID())
  512. // Check that the endpoint is still present on the sandbox before adding it to the service discovery.
  513. // This is to handle a race between the EnableService and the sbLeave
  514. // It is possible that the EnableService starts, fetches the list of the endpoints and
  515. // by the time the addServiceInfoToCluster is called the endpoint got removed from the sandbox
  516. // The risk is that the deleteServiceInfoToCluster happens before the addServiceInfoToCluster.
  517. // This check under the Service lock of the sandbox ensure the correct behavior.
  518. // If the addServiceInfoToCluster arrives first may find or not the endpoint and will proceed or exit
  519. // but in any case the deleteServiceInfoToCluster will follow doing the cleanup if needed.
  520. // In case the deleteServiceInfoToCluster arrives first, this one is happening after the endpoint is
  521. // removed from the list, in this situation the delete will bail out not finding any data to cleanup
  522. // and the add will bail out not finding the endpoint on the sandbox.
  523. if e := sb.getEndpoint(ep.ID()); e == nil {
  524. logrus.Warnf("addServiceInfoToCluster suppressing service resolution ep is not anymore in the sandbox %s", ep.ID())
  525. return nil
  526. }
  527. c := n.getController()
  528. agent := c.getAgent()
  529. name := ep.Name()
  530. if ep.isAnonymous() {
  531. name = ep.MyAliases()[0]
  532. }
  533. var ingressPorts []*PortConfig
  534. if ep.svcID != "" {
  535. // This is a task part of a service
  536. // Gossip ingress ports only in ingress network.
  537. if n.ingress {
  538. ingressPorts = ep.ingressPorts
  539. }
  540. if err := c.addServiceBinding(ep.svcName, ep.svcID, n.ID(), ep.ID(), name, ep.virtualIP, ingressPorts, ep.svcAliases, ep.myAliases, ep.Iface().Address().IP, "addServiceInfoToCluster"); err != nil {
  541. return err
  542. }
  543. } else {
  544. // This is a container simply attached to an attachable network
  545. if err := c.addContainerNameResolution(n.ID(), ep.ID(), name, ep.myAliases, ep.Iface().Address().IP, "addServiceInfoToCluster"); err != nil {
  546. return err
  547. }
  548. }
  549. buf, err := proto.Marshal(&EndpointRecord{
  550. Name: name,
  551. ServiceName: ep.svcName,
  552. ServiceID: ep.svcID,
  553. VirtualIP: ep.virtualIP.String(),
  554. IngressPorts: ingressPorts,
  555. Aliases: ep.svcAliases,
  556. TaskAliases: ep.myAliases,
  557. EndpointIP: ep.Iface().Address().IP.String(),
  558. ServiceDisabled: false,
  559. })
  560. if err != nil {
  561. return err
  562. }
  563. if agent != nil {
  564. if err := agent.networkDB.CreateEntry(libnetworkEPTable, n.ID(), ep.ID(), buf); err != nil {
  565. logrus.Warnf("addServiceInfoToCluster NetworkDB CreateEntry failed for %s %s err:%s", ep.id, n.id, err)
  566. return err
  567. }
  568. }
  569. logrus.Debugf("addServiceInfoToCluster END for %s %s", ep.svcName, ep.ID())
  570. return nil
  571. }
  572. func (ep *endpoint) deleteServiceInfoFromCluster(sb *sandbox, fullRemove bool, method string) error {
  573. if ep.isAnonymous() && len(ep.myAliases) == 0 {
  574. return nil
  575. }
  576. n := ep.getNetwork()
  577. if !n.isClusterEligible() {
  578. return nil
  579. }
  580. sb.Service.Lock()
  581. defer sb.Service.Unlock()
  582. logrus.Debugf("deleteServiceInfoFromCluster from %s START for %s %s", method, ep.svcName, ep.ID())
  583. // Avoid a race w/ with a container that aborts preemptively. This would
  584. // get caught in disableServceInNetworkDB, but we check here to make the
  585. // nature of the condition more clear.
  586. // See comment in addServiceInfoToCluster()
  587. if e := sb.getEndpoint(ep.ID()); e == nil {
  588. logrus.Warnf("deleteServiceInfoFromCluster suppressing service resolution ep is not anymore in the sandbox %s", ep.ID())
  589. return nil
  590. }
  591. c := n.getController()
  592. agent := c.getAgent()
  593. name := ep.Name()
  594. if ep.isAnonymous() {
  595. name = ep.MyAliases()[0]
  596. }
  597. if agent != nil {
  598. // First update the networkDB then locally
  599. if fullRemove {
  600. if err := agent.networkDB.DeleteEntry(libnetworkEPTable, n.ID(), ep.ID()); err != nil {
  601. logrus.Warnf("deleteServiceInfoFromCluster NetworkDB DeleteEntry failed for %s %s err:%s", ep.id, n.id, err)
  602. }
  603. } else {
  604. disableServiceInNetworkDB(agent, n, ep)
  605. }
  606. }
  607. if ep.Iface().Address() != nil {
  608. if ep.svcID != "" {
  609. // This is a task part of a service
  610. var ingressPorts []*PortConfig
  611. if n.ingress {
  612. ingressPorts = ep.ingressPorts
  613. }
  614. if err := c.rmServiceBinding(ep.svcName, ep.svcID, n.ID(), ep.ID(), name, ep.virtualIP, ingressPorts, ep.svcAliases, ep.myAliases, ep.Iface().Address().IP, "deleteServiceInfoFromCluster", true, fullRemove); err != nil {
  615. return err
  616. }
  617. } else {
  618. // This is a container simply attached to an attachable network
  619. if err := c.delContainerNameResolution(n.ID(), ep.ID(), name, ep.myAliases, ep.Iface().Address().IP, "deleteServiceInfoFromCluster"); err != nil {
  620. return err
  621. }
  622. }
  623. }
  624. logrus.Debugf("deleteServiceInfoFromCluster from %s END for %s %s", method, ep.svcName, ep.ID())
  625. return nil
  626. }
  627. func disableServiceInNetworkDB(a *agent, n *network, ep *endpoint) {
  628. var epRec EndpointRecord
  629. logrus.Debugf("disableServiceInNetworkDB for %s %s", ep.svcName, ep.ID())
  630. // Update existing record to indicate that the service is disabled
  631. inBuf, err := a.networkDB.GetEntry(libnetworkEPTable, n.ID(), ep.ID())
  632. if err != nil {
  633. logrus.Warnf("disableServiceInNetworkDB GetEntry failed for %s %s err:%s", ep.id, n.id, err)
  634. return
  635. }
  636. // Should never fail
  637. if err := proto.Unmarshal(inBuf, &epRec); err != nil {
  638. logrus.Errorf("disableServiceInNetworkDB unmarshal failed for %s %s err:%s", ep.id, n.id, err)
  639. return
  640. }
  641. epRec.ServiceDisabled = true
  642. // Should never fail
  643. outBuf, err := proto.Marshal(&epRec)
  644. if err != nil {
  645. logrus.Errorf("disableServiceInNetworkDB marshalling failed for %s %s err:%s", ep.id, n.id, err)
  646. return
  647. }
  648. // Send update to the whole cluster
  649. if err := a.networkDB.UpdateEntry(libnetworkEPTable, n.ID(), ep.ID(), outBuf); err != nil {
  650. logrus.Warnf("disableServiceInNetworkDB UpdateEntry failed for %s %s err:%s", ep.id, n.id, err)
  651. }
  652. }
  653. func (n *network) addDriverWatches() {
  654. if !n.isClusterEligible() {
  655. return
  656. }
  657. c := n.getController()
  658. agent := c.getAgent()
  659. if agent == nil {
  660. return
  661. }
  662. for _, table := range n.driverTables {
  663. ch, cancel := agent.networkDB.Watch(table.name, n.ID(), "")
  664. agent.Lock()
  665. agent.driverCancelFuncs[n.ID()] = append(agent.driverCancelFuncs[n.ID()], cancel)
  666. agent.Unlock()
  667. go c.handleTableEvents(ch, n.handleDriverTableEvent)
  668. d, err := n.driver(false)
  669. if err != nil {
  670. logrus.Errorf("Could not resolve driver %s while walking driver tabl: %v", n.networkType, err)
  671. return
  672. }
  673. agent.networkDB.WalkTable(table.name, func(nid, key string, value []byte, deleted bool) bool {
  674. // skip the entries that are mark for deletion, this is safe because this function is
  675. // called at initialization time so there is no state to delete
  676. if nid == n.ID() && !deleted {
  677. d.EventNotify(driverapi.Create, nid, table.name, key, value)
  678. }
  679. return false
  680. })
  681. }
  682. }
  683. func (n *network) cancelDriverWatches() {
  684. if !n.isClusterEligible() {
  685. return
  686. }
  687. agent := n.getController().getAgent()
  688. if agent == nil {
  689. return
  690. }
  691. agent.Lock()
  692. cancelFuncs := agent.driverCancelFuncs[n.ID()]
  693. delete(agent.driverCancelFuncs, n.ID())
  694. agent.Unlock()
  695. for _, cancel := range cancelFuncs {
  696. cancel()
  697. }
  698. }
  699. func (c *controller) handleTableEvents(ch *events.Channel, fn func(events.Event)) {
  700. for {
  701. select {
  702. case ev := <-ch.C:
  703. fn(ev)
  704. case <-ch.Done():
  705. return
  706. }
  707. }
  708. }
  709. func (n *network) handleDriverTableEvent(ev events.Event) {
  710. d, err := n.driver(false)
  711. if err != nil {
  712. logrus.Errorf("Could not resolve driver %s while handling driver table event: %v", n.networkType, err)
  713. return
  714. }
  715. var (
  716. etype driverapi.EventType
  717. tname string
  718. key string
  719. value []byte
  720. )
  721. switch event := ev.(type) {
  722. case networkdb.CreateEvent:
  723. tname = event.Table
  724. key = event.Key
  725. value = event.Value
  726. etype = driverapi.Create
  727. case networkdb.DeleteEvent:
  728. tname = event.Table
  729. key = event.Key
  730. value = event.Value
  731. etype = driverapi.Delete
  732. case networkdb.UpdateEvent:
  733. tname = event.Table
  734. key = event.Key
  735. value = event.Value
  736. etype = driverapi.Delete
  737. }
  738. d.EventNotify(etype, n.ID(), tname, key, value)
  739. }
  740. func (c *controller) handleNodeTableEvent(ev events.Event) {
  741. var (
  742. value []byte
  743. isAdd bool
  744. nodeAddr networkdb.NodeAddr
  745. )
  746. switch event := ev.(type) {
  747. case networkdb.CreateEvent:
  748. value = event.Value
  749. isAdd = true
  750. case networkdb.DeleteEvent:
  751. value = event.Value
  752. case networkdb.UpdateEvent:
  753. logrus.Errorf("Unexpected update node table event = %#v", event)
  754. }
  755. err := json.Unmarshal(value, &nodeAddr)
  756. if err != nil {
  757. logrus.Errorf("Error unmarshalling node table event %v", err)
  758. return
  759. }
  760. c.processNodeDiscovery([]net.IP{nodeAddr.Addr}, isAdd)
  761. }
  762. func (c *controller) handleEpTableEvent(ev events.Event) {
  763. var (
  764. nid string
  765. eid string
  766. value []byte
  767. epRec EndpointRecord
  768. )
  769. switch event := ev.(type) {
  770. case networkdb.CreateEvent:
  771. nid = event.NetworkID
  772. eid = event.Key
  773. value = event.Value
  774. case networkdb.DeleteEvent:
  775. nid = event.NetworkID
  776. eid = event.Key
  777. value = event.Value
  778. case networkdb.UpdateEvent:
  779. nid = event.NetworkID
  780. eid = event.Key
  781. value = event.Value
  782. default:
  783. logrus.Errorf("Unexpected update service table event = %#v", event)
  784. return
  785. }
  786. err := proto.Unmarshal(value, &epRec)
  787. if err != nil {
  788. logrus.Errorf("Failed to unmarshal service table value: %v", err)
  789. return
  790. }
  791. containerName := epRec.Name
  792. svcName := epRec.ServiceName
  793. svcID := epRec.ServiceID
  794. vip := net.ParseIP(epRec.VirtualIP)
  795. ip := net.ParseIP(epRec.EndpointIP)
  796. ingressPorts := epRec.IngressPorts
  797. serviceAliases := epRec.Aliases
  798. taskAliases := epRec.TaskAliases
  799. if containerName == "" || ip == nil {
  800. logrus.Errorf("Invalid endpoint name/ip received while handling service table event %s", value)
  801. return
  802. }
  803. switch ev.(type) {
  804. case networkdb.CreateEvent:
  805. logrus.Debugf("handleEpTableEvent ADD %s R:%v", eid, epRec)
  806. if svcID != "" {
  807. // This is a remote task part of a service
  808. if err := c.addServiceBinding(svcName, svcID, nid, eid, containerName, vip, ingressPorts, serviceAliases, taskAliases, ip, "handleEpTableEvent"); err != nil {
  809. logrus.Errorf("failed adding service binding for %s epRec:%v err:%v", eid, epRec, err)
  810. return
  811. }
  812. } else {
  813. // This is a remote container simply attached to an attachable network
  814. if err := c.addContainerNameResolution(nid, eid, containerName, taskAliases, ip, "handleEpTableEvent"); err != nil {
  815. logrus.Errorf("failed adding container name resolution for %s epRec:%v err:%v", eid, epRec, err)
  816. }
  817. }
  818. case networkdb.DeleteEvent:
  819. logrus.Debugf("handleEpTableEvent DEL %s R:%v", eid, epRec)
  820. if svcID != "" {
  821. // This is a remote task part of a service
  822. if err := c.rmServiceBinding(svcName, svcID, nid, eid, containerName, vip, ingressPorts, serviceAliases, taskAliases, ip, "handleEpTableEvent", true, true); err != nil {
  823. logrus.Errorf("failed removing service binding for %s epRec:%v err:%v", eid, epRec, err)
  824. return
  825. }
  826. } else {
  827. // This is a remote container simply attached to an attachable network
  828. if err := c.delContainerNameResolution(nid, eid, containerName, taskAliases, ip, "handleEpTableEvent"); err != nil {
  829. logrus.Errorf("failed removing container name resolution for %s epRec:%v err:%v", eid, epRec, err)
  830. }
  831. }
  832. case networkdb.UpdateEvent:
  833. logrus.Debugf("handleEpTableEvent UPD %s R:%v", eid, epRec)
  834. // We currently should only get these to inform us that an endpoint
  835. // is disabled. Report if otherwise.
  836. if svcID == "" || !epRec.ServiceDisabled {
  837. logrus.Errorf("Unexpected update table event for %s epRec:%v", eid, epRec)
  838. return
  839. }
  840. // This is a remote task that is part of a service that is now disabled
  841. if err := c.rmServiceBinding(svcName, svcID, nid, eid, containerName, vip, ingressPorts, serviceAliases, taskAliases, ip, "handleEpTableEvent", true, false); err != nil {
  842. logrus.Errorf("failed disabling service binding for %s epRec:%v err:%v", eid, epRec, err)
  843. return
  844. }
  845. }
  846. }