agent.go 28 KB

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