cluster.go 49 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898
  1. package cluster
  2. import (
  3. "crypto/x509"
  4. "encoding/base64"
  5. "encoding/json"
  6. "fmt"
  7. "io"
  8. "io/ioutil"
  9. "net"
  10. "os"
  11. "path/filepath"
  12. "runtime"
  13. "strings"
  14. "sync"
  15. "time"
  16. "github.com/Sirupsen/logrus"
  17. distreference "github.com/docker/distribution/reference"
  18. apierrors "github.com/docker/docker/api/errors"
  19. apitypes "github.com/docker/docker/api/types"
  20. "github.com/docker/docker/api/types/backend"
  21. "github.com/docker/docker/api/types/filters"
  22. "github.com/docker/docker/api/types/network"
  23. types "github.com/docker/docker/api/types/swarm"
  24. "github.com/docker/docker/daemon/cluster/convert"
  25. executorpkg "github.com/docker/docker/daemon/cluster/executor"
  26. "github.com/docker/docker/daemon/cluster/executor/container"
  27. "github.com/docker/docker/daemon/logger"
  28. "github.com/docker/docker/opts"
  29. "github.com/docker/docker/pkg/ioutils"
  30. "github.com/docker/docker/pkg/signal"
  31. "github.com/docker/docker/pkg/stdcopy"
  32. "github.com/docker/docker/reference"
  33. "github.com/docker/docker/runconfig"
  34. swarmapi "github.com/docker/swarmkit/api"
  35. "github.com/docker/swarmkit/manager/encryption"
  36. swarmnode "github.com/docker/swarmkit/node"
  37. "github.com/docker/swarmkit/protobuf/ptypes"
  38. "github.com/pkg/errors"
  39. "golang.org/x/net/context"
  40. "google.golang.org/grpc"
  41. )
  42. const swarmDirName = "swarm"
  43. const controlSocket = "control.sock"
  44. const swarmConnectTimeout = 20 * time.Second
  45. const swarmRequestTimeout = 20 * time.Second
  46. const stateFile = "docker-state.json"
  47. const defaultAddr = "0.0.0.0:2377"
  48. const (
  49. initialReconnectDelay = 100 * time.Millisecond
  50. maxReconnectDelay = 30 * time.Second
  51. contextPrefix = "com.docker.swarm"
  52. )
  53. // ErrNoSwarm is returned on leaving a cluster that was never initialized
  54. var ErrNoSwarm = fmt.Errorf("This node is not part of a swarm")
  55. // ErrSwarmExists is returned on initialize or join request for a cluster that has already been activated
  56. var ErrSwarmExists = fmt.Errorf("This node is already part of a swarm. Use \"docker swarm leave\" to leave this swarm and join another one.")
  57. // ErrPendingSwarmExists is returned on initialize or join request for a cluster that is already processing a similar request but has not succeeded yet.
  58. var ErrPendingSwarmExists = fmt.Errorf("This node is processing an existing join request that has not succeeded yet. Use \"docker swarm leave\" to cancel the current request.")
  59. // ErrSwarmJoinTimeoutReached is returned when cluster join could not complete before timeout was reached.
  60. var ErrSwarmJoinTimeoutReached = fmt.Errorf("Timeout was reached before node was joined. The attempt to join the swarm will continue in the background. Use the \"docker info\" command to see the current swarm status of your node.")
  61. // ErrSwarmLocked is returned if the swarm is encrypted and needs a key to unlock it.
  62. var ErrSwarmLocked = fmt.Errorf("Swarm is encrypted and needs to be unlocked before it can be used. Please use \"docker swarm unlock\" to unlock it.")
  63. // ErrSwarmCertificatesExpired is returned if docker was not started for the whole validity period and they had no chance to renew automatically.
  64. var ErrSwarmCertificatesExpired = errors.New("Swarm certificates have expired. To replace them, leave the swarm and join again.")
  65. // NetworkSubnetsProvider exposes functions for retrieving the subnets
  66. // of networks managed by Docker, so they can be filtered.
  67. type NetworkSubnetsProvider interface {
  68. V4Subnets() []net.IPNet
  69. V6Subnets() []net.IPNet
  70. }
  71. // Config provides values for Cluster.
  72. type Config struct {
  73. Root string
  74. Name string
  75. Backend executorpkg.Backend
  76. NetworkSubnetsProvider NetworkSubnetsProvider
  77. // DefaultAdvertiseAddr is the default host/IP or network interface to use
  78. // if no AdvertiseAddr value is specified.
  79. DefaultAdvertiseAddr string
  80. // path to store runtime state, such as the swarm control socket
  81. RuntimeRoot string
  82. }
  83. // Cluster provides capabilities to participate in a cluster as a worker or a
  84. // manager.
  85. type Cluster struct {
  86. sync.RWMutex
  87. *node
  88. root string
  89. runtimeRoot string
  90. config Config
  91. configEvent chan struct{} // todo: make this array and goroutine safe
  92. actualLocalAddr string // after resolution, not persisted
  93. stop bool
  94. err error
  95. cancelDelay func()
  96. attachers map[string]*attacher
  97. locked bool
  98. lastNodeConfig *nodeStartConfig
  99. }
  100. // attacher manages the in-memory attachment state of a container
  101. // attachment to a global scope network managed by swarm manager. It
  102. // helps in identifying the attachment ID via the taskID and the
  103. // corresponding attachment configuration obtained from the manager.
  104. type attacher struct {
  105. taskID string
  106. config *network.NetworkingConfig
  107. attachWaitCh chan *network.NetworkingConfig
  108. attachCompleteCh chan struct{}
  109. detachWaitCh chan struct{}
  110. }
  111. type node struct {
  112. *swarmnode.Node
  113. done chan struct{}
  114. ready bool
  115. conn *grpc.ClientConn
  116. client swarmapi.ControlClient
  117. logs swarmapi.LogsClient
  118. reconnectDelay time.Duration
  119. config nodeStartConfig
  120. }
  121. // nodeStartConfig holds configuration needed to start a new node. Exported
  122. // fields of this structure are saved to disk in json. Unexported fields
  123. // contain data that shouldn't be persisted between daemon reloads.
  124. type nodeStartConfig struct {
  125. // LocalAddr is this machine's local IP or hostname, if specified.
  126. LocalAddr string
  127. // RemoteAddr is the address that was given to "swarm join". It is used
  128. // to find LocalAddr if necessary.
  129. RemoteAddr string
  130. // ListenAddr is the address we bind to, including a port.
  131. ListenAddr string
  132. // AdvertiseAddr is the address other nodes should connect to,
  133. // including a port.
  134. AdvertiseAddr string
  135. joinAddr string
  136. forceNewCluster bool
  137. joinToken string
  138. lockKey []byte
  139. autolock bool
  140. }
  141. // New creates a new Cluster instance using provided config.
  142. func New(config Config) (*Cluster, error) {
  143. root := filepath.Join(config.Root, swarmDirName)
  144. if err := os.MkdirAll(root, 0700); err != nil {
  145. return nil, err
  146. }
  147. if config.RuntimeRoot == "" {
  148. config.RuntimeRoot = root
  149. }
  150. if err := os.MkdirAll(config.RuntimeRoot, 0700); err != nil {
  151. return nil, err
  152. }
  153. c := &Cluster{
  154. root: root,
  155. config: config,
  156. configEvent: make(chan struct{}, 10),
  157. runtimeRoot: config.RuntimeRoot,
  158. attachers: make(map[string]*attacher),
  159. }
  160. nodeConfig, err := c.loadState()
  161. if err != nil {
  162. if os.IsNotExist(err) {
  163. return c, nil
  164. }
  165. return nil, err
  166. }
  167. n, err := c.startNewNode(*nodeConfig)
  168. if err != nil {
  169. return nil, err
  170. }
  171. select {
  172. case <-time.After(swarmConnectTimeout):
  173. logrus.Error("swarm component could not be started before timeout was reached")
  174. case <-n.Ready():
  175. case <-n.done:
  176. if errors.Cause(c.err) == ErrSwarmLocked {
  177. return c, nil
  178. }
  179. if err, ok := errors.Cause(c.err).(x509.CertificateInvalidError); ok && err.Reason == x509.Expired {
  180. c.err = ErrSwarmCertificatesExpired
  181. return c, nil
  182. }
  183. return nil, fmt.Errorf("swarm component could not be started: %v", c.err)
  184. }
  185. go c.reconnectOnFailure(n)
  186. return c, nil
  187. }
  188. func (c *Cluster) loadState() (*nodeStartConfig, error) {
  189. dt, err := ioutil.ReadFile(filepath.Join(c.root, stateFile))
  190. if err != nil {
  191. return nil, err
  192. }
  193. // missing certificate means no actual state to restore from
  194. if _, err := os.Stat(filepath.Join(c.root, "certificates/swarm-node.crt")); err != nil {
  195. if os.IsNotExist(err) {
  196. c.clearState()
  197. }
  198. return nil, err
  199. }
  200. var st nodeStartConfig
  201. if err := json.Unmarshal(dt, &st); err != nil {
  202. return nil, err
  203. }
  204. return &st, nil
  205. }
  206. func (c *Cluster) saveState(config nodeStartConfig) error {
  207. dt, err := json.Marshal(config)
  208. if err != nil {
  209. return err
  210. }
  211. return ioutils.AtomicWriteFile(filepath.Join(c.root, stateFile), dt, 0600)
  212. }
  213. func (c *Cluster) reconnectOnFailure(n *node) {
  214. for {
  215. <-n.done
  216. c.Lock()
  217. if c.stop || c.node != nil {
  218. c.Unlock()
  219. return
  220. }
  221. n.reconnectDelay *= 2
  222. if n.reconnectDelay > maxReconnectDelay {
  223. n.reconnectDelay = maxReconnectDelay
  224. }
  225. logrus.Warnf("Restarting swarm in %.2f seconds", n.reconnectDelay.Seconds())
  226. delayCtx, cancel := context.WithTimeout(context.Background(), n.reconnectDelay)
  227. c.cancelDelay = cancel
  228. c.Unlock()
  229. <-delayCtx.Done()
  230. if delayCtx.Err() != context.DeadlineExceeded {
  231. return
  232. }
  233. c.Lock()
  234. if c.node != nil {
  235. c.Unlock()
  236. return
  237. }
  238. var err error
  239. config := n.config
  240. config.RemoteAddr = c.getRemoteAddress()
  241. config.joinAddr = config.RemoteAddr
  242. n, err = c.startNewNode(config)
  243. if err != nil {
  244. c.err = err
  245. close(n.done)
  246. }
  247. c.Unlock()
  248. }
  249. }
  250. func (c *Cluster) startNewNode(conf nodeStartConfig) (*node, error) {
  251. if err := c.config.Backend.IsSwarmCompatible(); err != nil {
  252. return nil, err
  253. }
  254. actualLocalAddr := conf.LocalAddr
  255. if actualLocalAddr == "" {
  256. // If localAddr was not specified, resolve it automatically
  257. // based on the route to joinAddr. localAddr can only be left
  258. // empty on "join".
  259. listenHost, _, err := net.SplitHostPort(conf.ListenAddr)
  260. if err != nil {
  261. return nil, fmt.Errorf("could not parse listen address: %v", err)
  262. }
  263. listenAddrIP := net.ParseIP(listenHost)
  264. if listenAddrIP == nil || !listenAddrIP.IsUnspecified() {
  265. actualLocalAddr = listenHost
  266. } else {
  267. if conf.RemoteAddr == "" {
  268. // Should never happen except using swarms created by
  269. // old versions that didn't save remoteAddr.
  270. conf.RemoteAddr = "8.8.8.8:53"
  271. }
  272. conn, err := net.Dial("udp", conf.RemoteAddr)
  273. if err != nil {
  274. return nil, fmt.Errorf("could not find local IP address: %v", err)
  275. }
  276. localHostPort := conn.LocalAddr().String()
  277. actualLocalAddr, _, _ = net.SplitHostPort(localHostPort)
  278. conn.Close()
  279. }
  280. }
  281. var control string
  282. if runtime.GOOS == "windows" {
  283. control = `\\.\pipe\` + controlSocket
  284. } else {
  285. control = filepath.Join(c.runtimeRoot, controlSocket)
  286. }
  287. c.node = nil
  288. c.cancelDelay = nil
  289. c.stop = false
  290. n, err := swarmnode.New(&swarmnode.Config{
  291. Hostname: c.config.Name,
  292. ForceNewCluster: conf.forceNewCluster,
  293. ListenControlAPI: control,
  294. ListenRemoteAPI: conf.ListenAddr,
  295. AdvertiseRemoteAPI: conf.AdvertiseAddr,
  296. JoinAddr: conf.joinAddr,
  297. StateDir: c.root,
  298. JoinToken: conf.joinToken,
  299. Executor: container.NewExecutor(c.config.Backend),
  300. HeartbeatTick: 1,
  301. ElectionTick: 3,
  302. UnlockKey: conf.lockKey,
  303. AutoLockManagers: conf.autolock,
  304. })
  305. if err != nil {
  306. return nil, err
  307. }
  308. ctx := context.Background()
  309. if err := n.Start(ctx); err != nil {
  310. return nil, err
  311. }
  312. node := &node{
  313. Node: n,
  314. done: make(chan struct{}),
  315. reconnectDelay: initialReconnectDelay,
  316. config: conf,
  317. }
  318. c.node = node
  319. c.actualLocalAddr = actualLocalAddr // not saved
  320. c.saveState(conf)
  321. c.config.Backend.SetClusterProvider(c)
  322. go func() {
  323. err := detectLockedError(n.Err(ctx))
  324. if err != nil {
  325. logrus.Errorf("cluster exited with error: %v", err)
  326. }
  327. c.Lock()
  328. c.node = nil
  329. c.err = err
  330. if errors.Cause(err) == ErrSwarmLocked {
  331. c.locked = true
  332. confClone := conf
  333. c.lastNodeConfig = &confClone
  334. }
  335. c.Unlock()
  336. close(node.done)
  337. }()
  338. go func() {
  339. select {
  340. case <-n.Ready():
  341. c.Lock()
  342. node.ready = true
  343. c.err = nil
  344. c.Unlock()
  345. case <-ctx.Done():
  346. }
  347. c.configEvent <- struct{}{}
  348. }()
  349. go func() {
  350. for conn := range n.ListenControlSocket(ctx) {
  351. c.Lock()
  352. if node.conn != conn {
  353. if conn == nil {
  354. node.client = nil
  355. node.logs = nil
  356. } else {
  357. node.client = swarmapi.NewControlClient(conn)
  358. node.logs = swarmapi.NewLogsClient(conn)
  359. }
  360. }
  361. node.conn = conn
  362. c.Unlock()
  363. c.configEvent <- struct{}{}
  364. }
  365. }()
  366. return node, nil
  367. }
  368. // Init initializes new cluster from user provided request.
  369. func (c *Cluster) Init(req types.InitRequest) (string, error) {
  370. c.Lock()
  371. if c.swarmExists() {
  372. if !req.ForceNewCluster {
  373. c.Unlock()
  374. return "", ErrSwarmExists
  375. }
  376. if err := c.stopNode(); err != nil {
  377. c.Unlock()
  378. return "", err
  379. }
  380. }
  381. if err := validateAndSanitizeInitRequest(&req); err != nil {
  382. c.Unlock()
  383. return "", err
  384. }
  385. listenHost, listenPort, err := resolveListenAddr(req.ListenAddr)
  386. if err != nil {
  387. c.Unlock()
  388. return "", err
  389. }
  390. advertiseHost, advertisePort, err := c.resolveAdvertiseAddr(req.AdvertiseAddr, listenPort)
  391. if err != nil {
  392. c.Unlock()
  393. return "", err
  394. }
  395. localAddr := listenHost
  396. // If the local address is undetermined, the advertise address
  397. // will be used as local address, if it belongs to this system.
  398. // If the advertise address is not local, then we try to find
  399. // a system address to use as local address. If this fails,
  400. // we give up and ask user to pass the listen address.
  401. if net.ParseIP(localAddr).IsUnspecified() {
  402. advertiseIP := net.ParseIP(advertiseHost)
  403. found := false
  404. for _, systemIP := range listSystemIPs() {
  405. if systemIP.Equal(advertiseIP) {
  406. localAddr = advertiseIP.String()
  407. found = true
  408. break
  409. }
  410. }
  411. if !found {
  412. ip, err := c.resolveSystemAddr()
  413. if err != nil {
  414. c.Unlock()
  415. logrus.Warnf("Could not find a local address: %v", err)
  416. return "", errMustSpecifyListenAddr
  417. }
  418. localAddr = ip.String()
  419. }
  420. }
  421. // todo: check current state existing
  422. n, err := c.startNewNode(nodeStartConfig{
  423. forceNewCluster: req.ForceNewCluster,
  424. autolock: req.AutoLockManagers,
  425. LocalAddr: localAddr,
  426. ListenAddr: net.JoinHostPort(listenHost, listenPort),
  427. AdvertiseAddr: net.JoinHostPort(advertiseHost, advertisePort),
  428. })
  429. if err != nil {
  430. c.Unlock()
  431. return "", err
  432. }
  433. c.Unlock()
  434. select {
  435. case <-n.Ready():
  436. if err := initClusterSpec(n, req.Spec); err != nil {
  437. return "", err
  438. }
  439. go c.reconnectOnFailure(n)
  440. return n.NodeID(), nil
  441. case <-n.done:
  442. c.RLock()
  443. defer c.RUnlock()
  444. if !req.ForceNewCluster { // if failure on first attempt don't keep state
  445. if err := c.clearState(); err != nil {
  446. return "", err
  447. }
  448. }
  449. return "", c.err
  450. }
  451. }
  452. // Join makes current Cluster part of an existing swarm cluster.
  453. func (c *Cluster) Join(req types.JoinRequest) error {
  454. c.Lock()
  455. if c.swarmExists() {
  456. c.Unlock()
  457. return ErrSwarmExists
  458. }
  459. if err := validateAndSanitizeJoinRequest(&req); err != nil {
  460. c.Unlock()
  461. return err
  462. }
  463. listenHost, listenPort, err := resolveListenAddr(req.ListenAddr)
  464. if err != nil {
  465. c.Unlock()
  466. return err
  467. }
  468. var advertiseAddr string
  469. if req.AdvertiseAddr != "" {
  470. advertiseHost, advertisePort, err := c.resolveAdvertiseAddr(req.AdvertiseAddr, listenPort)
  471. // For joining, we don't need to provide an advertise address,
  472. // since the remote side can detect it.
  473. if err == nil {
  474. advertiseAddr = net.JoinHostPort(advertiseHost, advertisePort)
  475. }
  476. }
  477. // todo: check current state existing
  478. n, err := c.startNewNode(nodeStartConfig{
  479. RemoteAddr: req.RemoteAddrs[0],
  480. ListenAddr: net.JoinHostPort(listenHost, listenPort),
  481. AdvertiseAddr: advertiseAddr,
  482. joinAddr: req.RemoteAddrs[0],
  483. joinToken: req.JoinToken,
  484. })
  485. if err != nil {
  486. c.Unlock()
  487. return err
  488. }
  489. c.Unlock()
  490. select {
  491. case <-time.After(swarmConnectTimeout):
  492. // attempt to connect will continue in background, but reconnect only if it didn't fail
  493. go func() {
  494. select {
  495. case <-n.Ready():
  496. c.reconnectOnFailure(n)
  497. case <-n.done:
  498. logrus.Errorf("failed to join the cluster: %+v", c.err)
  499. }
  500. }()
  501. return ErrSwarmJoinTimeoutReached
  502. case <-n.Ready():
  503. go c.reconnectOnFailure(n)
  504. return nil
  505. case <-n.done:
  506. c.RLock()
  507. defer c.RUnlock()
  508. return c.err
  509. }
  510. }
  511. // GetUnlockKey returns the unlock key for the swarm.
  512. func (c *Cluster) GetUnlockKey() (string, error) {
  513. c.RLock()
  514. defer c.RUnlock()
  515. if !c.isActiveManager() {
  516. return "", c.errNoManager()
  517. }
  518. ctx, cancel := c.getRequestContext()
  519. defer cancel()
  520. client := swarmapi.NewCAClient(c.conn)
  521. r, err := client.GetUnlockKey(ctx, &swarmapi.GetUnlockKeyRequest{})
  522. if err != nil {
  523. return "", err
  524. }
  525. if len(r.UnlockKey) == 0 {
  526. // no key
  527. return "", nil
  528. }
  529. return encryption.HumanReadableKey(r.UnlockKey), nil
  530. }
  531. // UnlockSwarm provides a key to decrypt data that is encrypted at rest.
  532. func (c *Cluster) UnlockSwarm(req types.UnlockRequest) error {
  533. key, err := encryption.ParseHumanReadableKey(req.UnlockKey)
  534. if err != nil {
  535. return err
  536. }
  537. c.Lock()
  538. if c.node != nil || c.locked != true {
  539. c.Unlock()
  540. return errors.New("swarm is not locked")
  541. }
  542. config := *c.lastNodeConfig
  543. config.lockKey = key
  544. n, err := c.startNewNode(config)
  545. if err != nil {
  546. c.Unlock()
  547. return err
  548. }
  549. c.Unlock()
  550. select {
  551. case <-n.Ready():
  552. case <-n.done:
  553. if errors.Cause(c.err) == ErrSwarmLocked {
  554. return errors.New("swarm could not be unlocked: invalid key provided")
  555. }
  556. return fmt.Errorf("swarm component could not be started: %v", c.err)
  557. }
  558. go c.reconnectOnFailure(n)
  559. return nil
  560. }
  561. // stopNode is a helper that stops the active c.node and waits until it has
  562. // shut down. Call while keeping the cluster lock.
  563. func (c *Cluster) stopNode() error {
  564. if c.node == nil {
  565. return nil
  566. }
  567. c.stop = true
  568. if c.cancelDelay != nil {
  569. c.cancelDelay()
  570. c.cancelDelay = nil
  571. }
  572. node := c.node
  573. ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
  574. defer cancel()
  575. // TODO: can't hold lock on stop because it calls back to network
  576. c.Unlock()
  577. defer c.Lock()
  578. if err := node.Stop(ctx); err != nil && !strings.Contains(err.Error(), "context canceled") {
  579. return err
  580. }
  581. <-node.done
  582. return nil
  583. }
  584. func removingManagerCausesLossOfQuorum(reachable, unreachable int) bool {
  585. return reachable-2 <= unreachable
  586. }
  587. func isLastManager(reachable, unreachable int) bool {
  588. return reachable == 1 && unreachable == 0
  589. }
  590. // Leave shuts down Cluster and removes current state.
  591. func (c *Cluster) Leave(force bool) error {
  592. c.Lock()
  593. node := c.node
  594. if node == nil {
  595. if c.locked {
  596. c.locked = false
  597. c.lastNodeConfig = nil
  598. c.Unlock()
  599. } else if c.err == ErrSwarmCertificatesExpired {
  600. c.err = nil
  601. c.Unlock()
  602. } else {
  603. c.Unlock()
  604. return ErrNoSwarm
  605. }
  606. } else {
  607. if node.Manager() != nil && !force {
  608. msg := "You are attempting to leave the swarm on a node that is participating as a manager. "
  609. if c.isActiveManager() {
  610. active, reachable, unreachable, err := c.managerStats()
  611. if err == nil {
  612. if active && removingManagerCausesLossOfQuorum(reachable, unreachable) {
  613. if isLastManager(reachable, unreachable) {
  614. msg += "Removing the last manager erases all current state of the swarm. Use `--force` to ignore this message. "
  615. c.Unlock()
  616. return fmt.Errorf(msg)
  617. }
  618. msg += fmt.Sprintf("Removing this node leaves %v managers out of %v. Without a Raft quorum your swarm will be inaccessible. ", reachable-1, reachable+unreachable)
  619. }
  620. }
  621. } else {
  622. msg += "Doing so may lose the consensus of your cluster. "
  623. }
  624. msg += "The only way to restore a swarm that has lost consensus is to reinitialize it with `--force-new-cluster`. Use `--force` to suppress this message."
  625. c.Unlock()
  626. return fmt.Errorf(msg)
  627. }
  628. if err := c.stopNode(); err != nil {
  629. logrus.Errorf("failed to shut down cluster node: %v", err)
  630. signal.DumpStacks("")
  631. c.Unlock()
  632. return err
  633. }
  634. c.Unlock()
  635. if nodeID := node.NodeID(); nodeID != "" {
  636. nodeContainers, err := c.listContainerForNode(nodeID)
  637. if err != nil {
  638. return err
  639. }
  640. for _, id := range nodeContainers {
  641. if err := c.config.Backend.ContainerRm(id, &apitypes.ContainerRmConfig{ForceRemove: true}); err != nil {
  642. logrus.Errorf("error removing %v: %v", id, err)
  643. }
  644. }
  645. }
  646. }
  647. c.configEvent <- struct{}{}
  648. // todo: cleanup optional?
  649. if err := c.clearState(); err != nil {
  650. return err
  651. }
  652. return nil
  653. }
  654. func (c *Cluster) listContainerForNode(nodeID string) ([]string, error) {
  655. var ids []string
  656. filters := filters.NewArgs()
  657. filters.Add("label", fmt.Sprintf("com.docker.swarm.node.id=%s", nodeID))
  658. containers, err := c.config.Backend.Containers(&apitypes.ContainerListOptions{
  659. Filters: filters,
  660. })
  661. if err != nil {
  662. return []string{}, err
  663. }
  664. for _, c := range containers {
  665. ids = append(ids, c.ID)
  666. }
  667. return ids, nil
  668. }
  669. func (c *Cluster) clearState() error {
  670. // todo: backup this data instead of removing?
  671. if err := os.RemoveAll(c.root); err != nil {
  672. return err
  673. }
  674. if err := os.MkdirAll(c.root, 0700); err != nil {
  675. return err
  676. }
  677. c.config.Backend.SetClusterProvider(nil)
  678. return nil
  679. }
  680. func (c *Cluster) getRequestContext() (context.Context, func()) { // TODO: not needed when requests don't block on qourum lost
  681. return context.WithTimeout(context.Background(), swarmRequestTimeout)
  682. }
  683. // Inspect retrieves the configuration properties of a managed swarm cluster.
  684. func (c *Cluster) Inspect() (types.Swarm, error) {
  685. c.RLock()
  686. defer c.RUnlock()
  687. if !c.isActiveManager() {
  688. return types.Swarm{}, c.errNoManager()
  689. }
  690. ctx, cancel := c.getRequestContext()
  691. defer cancel()
  692. swarm, err := getSwarm(ctx, c.client)
  693. if err != nil {
  694. return types.Swarm{}, err
  695. }
  696. return convert.SwarmFromGRPC(*swarm), nil
  697. }
  698. // Update updates configuration of a managed swarm cluster.
  699. func (c *Cluster) Update(version uint64, spec types.Spec, flags types.UpdateFlags) error {
  700. c.RLock()
  701. defer c.RUnlock()
  702. if !c.isActiveManager() {
  703. return c.errNoManager()
  704. }
  705. ctx, cancel := c.getRequestContext()
  706. defer cancel()
  707. swarm, err := getSwarm(ctx, c.client)
  708. if err != nil {
  709. return err
  710. }
  711. // In update, client should provide the complete spec of the swarm, including
  712. // Name and Labels. If a field is specified with 0 or nil, then the default value
  713. // will be used to swarmkit.
  714. clusterSpec, err := convert.SwarmSpecToGRPC(spec)
  715. if err != nil {
  716. return err
  717. }
  718. _, err = c.client.UpdateCluster(
  719. ctx,
  720. &swarmapi.UpdateClusterRequest{
  721. ClusterID: swarm.ID,
  722. Spec: &clusterSpec,
  723. ClusterVersion: &swarmapi.Version{
  724. Index: version,
  725. },
  726. Rotation: swarmapi.KeyRotation{
  727. WorkerJoinToken: flags.RotateWorkerToken,
  728. ManagerJoinToken: flags.RotateManagerToken,
  729. ManagerUnlockKey: flags.RotateManagerUnlockKey,
  730. },
  731. },
  732. )
  733. return err
  734. }
  735. // IsManager returns true if Cluster is participating as a manager.
  736. func (c *Cluster) IsManager() bool {
  737. c.RLock()
  738. defer c.RUnlock()
  739. return c.isActiveManager()
  740. }
  741. // IsAgent returns true if Cluster is participating as a worker/agent.
  742. func (c *Cluster) IsAgent() bool {
  743. c.RLock()
  744. defer c.RUnlock()
  745. return c.node != nil && c.ready
  746. }
  747. // GetLocalAddress returns the local address.
  748. func (c *Cluster) GetLocalAddress() string {
  749. c.RLock()
  750. defer c.RUnlock()
  751. return c.actualLocalAddr
  752. }
  753. // GetListenAddress returns the listen address.
  754. func (c *Cluster) GetListenAddress() string {
  755. c.RLock()
  756. defer c.RUnlock()
  757. if c.node != nil {
  758. return c.node.config.ListenAddr
  759. }
  760. return ""
  761. }
  762. // GetAdvertiseAddress returns the remotely reachable address of this node.
  763. func (c *Cluster) GetAdvertiseAddress() string {
  764. c.RLock()
  765. defer c.RUnlock()
  766. if c.node != nil && c.node.config.AdvertiseAddr != "" {
  767. advertiseHost, _, _ := net.SplitHostPort(c.node.config.AdvertiseAddr)
  768. return advertiseHost
  769. }
  770. return c.actualLocalAddr
  771. }
  772. // GetRemoteAddress returns a known advertise address of a remote manager if
  773. // available.
  774. // todo: change to array/connect with info
  775. func (c *Cluster) GetRemoteAddress() string {
  776. c.RLock()
  777. defer c.RUnlock()
  778. return c.getRemoteAddress()
  779. }
  780. func (c *Cluster) getRemoteAddress() string {
  781. if c.node == nil {
  782. return ""
  783. }
  784. nodeID := c.node.NodeID()
  785. for _, r := range c.node.Remotes() {
  786. if r.NodeID != nodeID {
  787. return r.Addr
  788. }
  789. }
  790. return ""
  791. }
  792. // ListenClusterEvents returns a channel that receives messages on cluster
  793. // participation changes.
  794. // todo: make cancelable and accessible to multiple callers
  795. func (c *Cluster) ListenClusterEvents() <-chan struct{} {
  796. return c.configEvent
  797. }
  798. // Info returns information about the current cluster state.
  799. func (c *Cluster) Info() types.Info {
  800. info := types.Info{
  801. NodeAddr: c.GetAdvertiseAddress(),
  802. }
  803. c.RLock()
  804. defer c.RUnlock()
  805. if c.node == nil {
  806. info.LocalNodeState = types.LocalNodeStateInactive
  807. if c.cancelDelay != nil {
  808. info.LocalNodeState = types.LocalNodeStateError
  809. }
  810. if c.locked {
  811. info.LocalNodeState = types.LocalNodeStateLocked
  812. } else if c.err == ErrSwarmCertificatesExpired {
  813. info.LocalNodeState = types.LocalNodeStateError
  814. }
  815. } else {
  816. info.LocalNodeState = types.LocalNodeStatePending
  817. if c.ready == true {
  818. info.LocalNodeState = types.LocalNodeStateActive
  819. } else if c.locked {
  820. info.LocalNodeState = types.LocalNodeStateLocked
  821. }
  822. }
  823. if c.err != nil {
  824. info.Error = c.err.Error()
  825. }
  826. ctx, cancel := c.getRequestContext()
  827. defer cancel()
  828. if c.isActiveManager() {
  829. info.ControlAvailable = true
  830. swarm, err := c.Inspect()
  831. if err != nil {
  832. info.Error = err.Error()
  833. }
  834. // Strip JoinTokens
  835. info.Cluster = swarm.ClusterInfo
  836. if r, err := c.client.ListNodes(ctx, &swarmapi.ListNodesRequest{}); err == nil {
  837. info.Nodes = len(r.Nodes)
  838. for _, n := range r.Nodes {
  839. if n.ManagerStatus != nil {
  840. info.Managers = info.Managers + 1
  841. }
  842. }
  843. }
  844. }
  845. if c.node != nil {
  846. for _, r := range c.node.Remotes() {
  847. info.RemoteManagers = append(info.RemoteManagers, types.Peer{NodeID: r.NodeID, Addr: r.Addr})
  848. }
  849. info.NodeID = c.node.NodeID()
  850. }
  851. return info
  852. }
  853. // isActiveManager should not be called without a read lock
  854. func (c *Cluster) isActiveManager() bool {
  855. return c.node != nil && c.conn != nil
  856. }
  857. // swarmExists should not be called without a read lock
  858. func (c *Cluster) swarmExists() bool {
  859. return c.node != nil || c.locked || c.err == ErrSwarmCertificatesExpired
  860. }
  861. // errNoManager returns error describing why manager commands can't be used.
  862. // Call with read lock.
  863. func (c *Cluster) errNoManager() error {
  864. if c.node == nil {
  865. if c.locked {
  866. return ErrSwarmLocked
  867. }
  868. if c.err == ErrSwarmCertificatesExpired {
  869. return ErrSwarmCertificatesExpired
  870. }
  871. return fmt.Errorf("This node is not a swarm manager. Use \"docker swarm init\" or \"docker swarm join\" to connect this node to swarm and try again.")
  872. }
  873. if c.node.Manager() != nil {
  874. return fmt.Errorf("This node is not a swarm manager. Manager is being prepared or has trouble connecting to the cluster.")
  875. }
  876. return fmt.Errorf("This node is not a swarm manager. Worker nodes can't be used to view or modify cluster state. Please run this command on a manager node or promote the current node to a manager.")
  877. }
  878. // GetServices returns all services of a managed swarm cluster.
  879. func (c *Cluster) GetServices(options apitypes.ServiceListOptions) ([]types.Service, error) {
  880. c.RLock()
  881. defer c.RUnlock()
  882. if !c.isActiveManager() {
  883. return nil, c.errNoManager()
  884. }
  885. filters, err := newListServicesFilters(options.Filters)
  886. if err != nil {
  887. return nil, err
  888. }
  889. ctx, cancel := c.getRequestContext()
  890. defer cancel()
  891. r, err := c.client.ListServices(
  892. ctx,
  893. &swarmapi.ListServicesRequest{Filters: filters})
  894. if err != nil {
  895. return nil, err
  896. }
  897. services := []types.Service{}
  898. for _, service := range r.Services {
  899. services = append(services, convert.ServiceFromGRPC(*service))
  900. }
  901. return services, nil
  902. }
  903. // imageWithDigestString takes an image such as name or name:tag
  904. // and returns the image pinned to a digest, such as name@sha256:34234...
  905. // Due to the difference between the docker/docker/reference, and the
  906. // docker/distribution/reference packages, we're parsing the image twice.
  907. // As the two packages converge, this function should be simplified.
  908. // TODO(nishanttotla): After the packages converge, the function must
  909. // convert distreference.Named -> distreference.Canonical, and the logic simplified.
  910. func (c *Cluster) imageWithDigestString(ctx context.Context, image string, authConfig *apitypes.AuthConfig) (string, error) {
  911. ref, err := distreference.ParseNamed(image)
  912. if err != nil {
  913. return "", err
  914. }
  915. // only query registry if not a canonical reference (i.e. with digest)
  916. if _, ok := ref.(distreference.Canonical); !ok {
  917. // create a docker/docker/reference Named object because GetRepository needs it
  918. dockerRef, err := reference.ParseNamed(image)
  919. if err != nil {
  920. return "", err
  921. }
  922. dockerRef = reference.WithDefaultTag(dockerRef)
  923. namedTaggedRef, ok := dockerRef.(reference.NamedTagged)
  924. if !ok {
  925. return "", fmt.Errorf("unable to cast image to NamedTagged reference object")
  926. }
  927. repo, _, err := c.config.Backend.GetRepository(ctx, namedTaggedRef, authConfig)
  928. if err != nil {
  929. return "", err
  930. }
  931. dscrptr, err := repo.Tags(ctx).Get(ctx, namedTaggedRef.Tag())
  932. if err != nil {
  933. return "", err
  934. }
  935. namedDigestedRef, err := distreference.WithDigest(distreference.EnsureTagged(ref), dscrptr.Digest)
  936. if err != nil {
  937. return "", err
  938. }
  939. return namedDigestedRef.String(), nil
  940. } else {
  941. // reference already contains a digest, so just return it
  942. return ref.String(), nil
  943. }
  944. }
  945. // CreateService creates a new service in a managed swarm cluster.
  946. func (c *Cluster) CreateService(s types.ServiceSpec, encodedAuth string) (string, error) {
  947. c.RLock()
  948. defer c.RUnlock()
  949. if !c.isActiveManager() {
  950. return "", c.errNoManager()
  951. }
  952. ctx, cancel := c.getRequestContext()
  953. defer cancel()
  954. err := c.populateNetworkID(ctx, c.client, &s)
  955. if err != nil {
  956. return "", err
  957. }
  958. serviceSpec, err := convert.ServiceSpecToGRPC(s)
  959. if err != nil {
  960. return "", err
  961. }
  962. ctnr := serviceSpec.Task.GetContainer()
  963. if ctnr == nil {
  964. return "", fmt.Errorf("service does not use container tasks")
  965. }
  966. if encodedAuth != "" {
  967. ctnr.PullOptions = &swarmapi.ContainerSpec_PullOptions{RegistryAuth: encodedAuth}
  968. }
  969. // retrieve auth config from encoded auth
  970. authConfig := &apitypes.AuthConfig{}
  971. if encodedAuth != "" {
  972. if err := json.NewDecoder(base64.NewDecoder(base64.URLEncoding, strings.NewReader(encodedAuth))).Decode(authConfig); err != nil {
  973. logrus.Warnf("invalid authconfig: %v", err)
  974. }
  975. }
  976. // pin image by digest
  977. if os.Getenv("DOCKER_SERVICE_PREFER_OFFLINE_IMAGE") != "1" {
  978. digestImage, err := c.imageWithDigestString(ctx, ctnr.Image, authConfig)
  979. if err != nil {
  980. logrus.Warnf("unable to pin image %s to digest: %s", ctnr.Image, err.Error())
  981. } else {
  982. logrus.Debugf("pinning image %s by digest: %s", ctnr.Image, digestImage)
  983. ctnr.Image = digestImage
  984. }
  985. }
  986. r, err := c.client.CreateService(ctx, &swarmapi.CreateServiceRequest{Spec: &serviceSpec})
  987. if err != nil {
  988. return "", err
  989. }
  990. return r.Service.ID, nil
  991. }
  992. // GetService returns a service based on an ID or name.
  993. func (c *Cluster) GetService(input string) (types.Service, error) {
  994. c.RLock()
  995. defer c.RUnlock()
  996. if !c.isActiveManager() {
  997. return types.Service{}, c.errNoManager()
  998. }
  999. ctx, cancel := c.getRequestContext()
  1000. defer cancel()
  1001. service, err := getService(ctx, c.client, input)
  1002. if err != nil {
  1003. return types.Service{}, err
  1004. }
  1005. return convert.ServiceFromGRPC(*service), nil
  1006. }
  1007. // UpdateService updates existing service to match new properties.
  1008. func (c *Cluster) UpdateService(serviceIDOrName string, version uint64, spec types.ServiceSpec, encodedAuth string, registryAuthFrom string) error {
  1009. c.RLock()
  1010. defer c.RUnlock()
  1011. if !c.isActiveManager() {
  1012. return c.errNoManager()
  1013. }
  1014. ctx, cancel := c.getRequestContext()
  1015. defer cancel()
  1016. err := c.populateNetworkID(ctx, c.client, &spec)
  1017. if err != nil {
  1018. return err
  1019. }
  1020. serviceSpec, err := convert.ServiceSpecToGRPC(spec)
  1021. if err != nil {
  1022. return err
  1023. }
  1024. currentService, err := getService(ctx, c.client, serviceIDOrName)
  1025. if err != nil {
  1026. return err
  1027. }
  1028. newCtnr := serviceSpec.Task.GetContainer()
  1029. if newCtnr == nil {
  1030. return fmt.Errorf("service does not use container tasks")
  1031. }
  1032. if encodedAuth != "" {
  1033. newCtnr.PullOptions = &swarmapi.ContainerSpec_PullOptions{RegistryAuth: encodedAuth}
  1034. } else {
  1035. // this is needed because if the encodedAuth isn't being updated then we
  1036. // shouldn't lose it, and continue to use the one that was already present
  1037. var ctnr *swarmapi.ContainerSpec
  1038. switch registryAuthFrom {
  1039. case apitypes.RegistryAuthFromSpec, "":
  1040. ctnr = currentService.Spec.Task.GetContainer()
  1041. case apitypes.RegistryAuthFromPreviousSpec:
  1042. if currentService.PreviousSpec == nil {
  1043. return fmt.Errorf("service does not have a previous spec")
  1044. }
  1045. ctnr = currentService.PreviousSpec.Task.GetContainer()
  1046. default:
  1047. return fmt.Errorf("unsupported registryAuthFromValue")
  1048. }
  1049. if ctnr == nil {
  1050. return fmt.Errorf("service does not use container tasks")
  1051. }
  1052. newCtnr.PullOptions = ctnr.PullOptions
  1053. // update encodedAuth so it can be used to pin image by digest
  1054. if ctnr.PullOptions != nil {
  1055. encodedAuth = ctnr.PullOptions.RegistryAuth
  1056. }
  1057. }
  1058. // retrieve auth config from encoded auth
  1059. authConfig := &apitypes.AuthConfig{}
  1060. if encodedAuth != "" {
  1061. if err := json.NewDecoder(base64.NewDecoder(base64.URLEncoding, strings.NewReader(encodedAuth))).Decode(authConfig); err != nil {
  1062. logrus.Warnf("invalid authconfig: %v", err)
  1063. }
  1064. }
  1065. // pin image by digest
  1066. if os.Getenv("DOCKER_SERVICE_PREFER_OFFLINE_IMAGE") != "1" {
  1067. digestImage, err := c.imageWithDigestString(ctx, newCtnr.Image, authConfig)
  1068. if err != nil {
  1069. logrus.Warnf("unable to pin image %s to digest: %s", newCtnr.Image, err.Error())
  1070. } else if newCtnr.Image != digestImage {
  1071. logrus.Debugf("pinning image %s by digest: %s", newCtnr.Image, digestImage)
  1072. newCtnr.Image = digestImage
  1073. }
  1074. }
  1075. _, err = c.client.UpdateService(
  1076. ctx,
  1077. &swarmapi.UpdateServiceRequest{
  1078. ServiceID: currentService.ID,
  1079. Spec: &serviceSpec,
  1080. ServiceVersion: &swarmapi.Version{
  1081. Index: version,
  1082. },
  1083. },
  1084. )
  1085. return err
  1086. }
  1087. // RemoveService removes a service from a managed swarm cluster.
  1088. func (c *Cluster) RemoveService(input string) error {
  1089. c.RLock()
  1090. defer c.RUnlock()
  1091. if !c.isActiveManager() {
  1092. return c.errNoManager()
  1093. }
  1094. ctx, cancel := c.getRequestContext()
  1095. defer cancel()
  1096. service, err := getService(ctx, c.client, input)
  1097. if err != nil {
  1098. return err
  1099. }
  1100. if _, err := c.client.RemoveService(ctx, &swarmapi.RemoveServiceRequest{ServiceID: service.ID}); err != nil {
  1101. return err
  1102. }
  1103. return nil
  1104. }
  1105. // ServiceLogs collects service logs and writes them back to `config.OutStream`
  1106. func (c *Cluster) ServiceLogs(ctx context.Context, input string, config *backend.ContainerLogsConfig, started chan struct{}) error {
  1107. c.RLock()
  1108. if !c.isActiveManager() {
  1109. c.RUnlock()
  1110. return c.errNoManager()
  1111. }
  1112. service, err := getService(ctx, c.client, input)
  1113. if err != nil {
  1114. c.RUnlock()
  1115. return err
  1116. }
  1117. stream, err := c.logs.SubscribeLogs(ctx, &swarmapi.SubscribeLogsRequest{
  1118. Selector: &swarmapi.LogSelector{
  1119. ServiceIDs: []string{service.ID},
  1120. },
  1121. Options: &swarmapi.LogSubscriptionOptions{
  1122. Follow: true,
  1123. },
  1124. })
  1125. if err != nil {
  1126. c.RUnlock()
  1127. return err
  1128. }
  1129. wf := ioutils.NewWriteFlusher(config.OutStream)
  1130. defer wf.Close()
  1131. close(started)
  1132. wf.Flush()
  1133. outStream := stdcopy.NewStdWriter(wf, stdcopy.Stdout)
  1134. errStream := stdcopy.NewStdWriter(wf, stdcopy.Stderr)
  1135. // Release the lock before starting the stream.
  1136. c.RUnlock()
  1137. for {
  1138. // Check the context before doing anything.
  1139. select {
  1140. case <-ctx.Done():
  1141. return ctx.Err()
  1142. default:
  1143. }
  1144. subscribeMsg, err := stream.Recv()
  1145. if err == io.EOF {
  1146. return nil
  1147. }
  1148. if err != nil {
  1149. return err
  1150. }
  1151. for _, msg := range subscribeMsg.Messages {
  1152. data := []byte{}
  1153. if config.Timestamps {
  1154. ts, err := ptypes.Timestamp(msg.Timestamp)
  1155. if err != nil {
  1156. return err
  1157. }
  1158. data = append(data, []byte(ts.Format(logger.TimeFormat)+" ")...)
  1159. }
  1160. data = append(data, []byte(fmt.Sprintf("%s.node.id=%s,%s.service.id=%s,%s.task.id=%s ",
  1161. contextPrefix, msg.Context.NodeID,
  1162. contextPrefix, msg.Context.ServiceID,
  1163. contextPrefix, msg.Context.TaskID,
  1164. ))...)
  1165. data = append(data, msg.Data...)
  1166. switch msg.Stream {
  1167. case swarmapi.LogStreamStdout:
  1168. outStream.Write(data)
  1169. case swarmapi.LogStreamStderr:
  1170. errStream.Write(data)
  1171. }
  1172. }
  1173. }
  1174. }
  1175. // GetNodes returns a list of all nodes known to a cluster.
  1176. func (c *Cluster) GetNodes(options apitypes.NodeListOptions) ([]types.Node, error) {
  1177. c.RLock()
  1178. defer c.RUnlock()
  1179. if !c.isActiveManager() {
  1180. return nil, c.errNoManager()
  1181. }
  1182. filters, err := newListNodesFilters(options.Filters)
  1183. if err != nil {
  1184. return nil, err
  1185. }
  1186. ctx, cancel := c.getRequestContext()
  1187. defer cancel()
  1188. r, err := c.client.ListNodes(
  1189. ctx,
  1190. &swarmapi.ListNodesRequest{Filters: filters})
  1191. if err != nil {
  1192. return nil, err
  1193. }
  1194. nodes := []types.Node{}
  1195. for _, node := range r.Nodes {
  1196. nodes = append(nodes, convert.NodeFromGRPC(*node))
  1197. }
  1198. return nodes, nil
  1199. }
  1200. // GetNode returns a node based on an ID or name.
  1201. func (c *Cluster) GetNode(input string) (types.Node, error) {
  1202. c.RLock()
  1203. defer c.RUnlock()
  1204. if !c.isActiveManager() {
  1205. return types.Node{}, c.errNoManager()
  1206. }
  1207. ctx, cancel := c.getRequestContext()
  1208. defer cancel()
  1209. node, err := getNode(ctx, c.client, input)
  1210. if err != nil {
  1211. return types.Node{}, err
  1212. }
  1213. return convert.NodeFromGRPC(*node), nil
  1214. }
  1215. // UpdateNode updates existing nodes properties.
  1216. func (c *Cluster) UpdateNode(input string, version uint64, spec types.NodeSpec) error {
  1217. c.RLock()
  1218. defer c.RUnlock()
  1219. if !c.isActiveManager() {
  1220. return c.errNoManager()
  1221. }
  1222. nodeSpec, err := convert.NodeSpecToGRPC(spec)
  1223. if err != nil {
  1224. return err
  1225. }
  1226. ctx, cancel := c.getRequestContext()
  1227. defer cancel()
  1228. currentNode, err := getNode(ctx, c.client, input)
  1229. if err != nil {
  1230. return err
  1231. }
  1232. _, err = c.client.UpdateNode(
  1233. ctx,
  1234. &swarmapi.UpdateNodeRequest{
  1235. NodeID: currentNode.ID,
  1236. Spec: &nodeSpec,
  1237. NodeVersion: &swarmapi.Version{
  1238. Index: version,
  1239. },
  1240. },
  1241. )
  1242. return err
  1243. }
  1244. // RemoveNode removes a node from a cluster
  1245. func (c *Cluster) RemoveNode(input string, force bool) error {
  1246. c.RLock()
  1247. defer c.RUnlock()
  1248. if !c.isActiveManager() {
  1249. return c.errNoManager()
  1250. }
  1251. ctx, cancel := c.getRequestContext()
  1252. defer cancel()
  1253. node, err := getNode(ctx, c.client, input)
  1254. if err != nil {
  1255. return err
  1256. }
  1257. if _, err := c.client.RemoveNode(ctx, &swarmapi.RemoveNodeRequest{NodeID: node.ID, Force: force}); err != nil {
  1258. return err
  1259. }
  1260. return nil
  1261. }
  1262. // GetTasks returns a list of tasks matching the filter options.
  1263. func (c *Cluster) GetTasks(options apitypes.TaskListOptions) ([]types.Task, error) {
  1264. c.RLock()
  1265. defer c.RUnlock()
  1266. if !c.isActiveManager() {
  1267. return nil, c.errNoManager()
  1268. }
  1269. byName := func(filter filters.Args) error {
  1270. if filter.Include("service") {
  1271. serviceFilters := filter.Get("service")
  1272. for _, serviceFilter := range serviceFilters {
  1273. service, err := c.GetService(serviceFilter)
  1274. if err != nil {
  1275. return err
  1276. }
  1277. filter.Del("service", serviceFilter)
  1278. filter.Add("service", service.ID)
  1279. }
  1280. }
  1281. if filter.Include("node") {
  1282. nodeFilters := filter.Get("node")
  1283. for _, nodeFilter := range nodeFilters {
  1284. node, err := c.GetNode(nodeFilter)
  1285. if err != nil {
  1286. return err
  1287. }
  1288. filter.Del("node", nodeFilter)
  1289. filter.Add("node", node.ID)
  1290. }
  1291. }
  1292. return nil
  1293. }
  1294. filters, err := newListTasksFilters(options.Filters, byName)
  1295. if err != nil {
  1296. return nil, err
  1297. }
  1298. ctx, cancel := c.getRequestContext()
  1299. defer cancel()
  1300. r, err := c.client.ListTasks(
  1301. ctx,
  1302. &swarmapi.ListTasksRequest{Filters: filters})
  1303. if err != nil {
  1304. return nil, err
  1305. }
  1306. tasks := []types.Task{}
  1307. for _, task := range r.Tasks {
  1308. if task.Spec.GetContainer() != nil {
  1309. tasks = append(tasks, convert.TaskFromGRPC(*task))
  1310. }
  1311. }
  1312. return tasks, nil
  1313. }
  1314. // GetTask returns a task by an ID.
  1315. func (c *Cluster) GetTask(input string) (types.Task, error) {
  1316. c.RLock()
  1317. defer c.RUnlock()
  1318. if !c.isActiveManager() {
  1319. return types.Task{}, c.errNoManager()
  1320. }
  1321. ctx, cancel := c.getRequestContext()
  1322. defer cancel()
  1323. task, err := getTask(ctx, c.client, input)
  1324. if err != nil {
  1325. return types.Task{}, err
  1326. }
  1327. return convert.TaskFromGRPC(*task), nil
  1328. }
  1329. // GetNetwork returns a cluster network by an ID.
  1330. func (c *Cluster) GetNetwork(input string) (apitypes.NetworkResource, error) {
  1331. c.RLock()
  1332. defer c.RUnlock()
  1333. if !c.isActiveManager() {
  1334. return apitypes.NetworkResource{}, c.errNoManager()
  1335. }
  1336. ctx, cancel := c.getRequestContext()
  1337. defer cancel()
  1338. network, err := getNetwork(ctx, c.client, input)
  1339. if err != nil {
  1340. return apitypes.NetworkResource{}, err
  1341. }
  1342. return convert.BasicNetworkFromGRPC(*network), nil
  1343. }
  1344. // GetNetworks returns all current cluster managed networks.
  1345. func (c *Cluster) GetNetworks() ([]apitypes.NetworkResource, error) {
  1346. c.RLock()
  1347. defer c.RUnlock()
  1348. if !c.isActiveManager() {
  1349. return nil, c.errNoManager()
  1350. }
  1351. ctx, cancel := c.getRequestContext()
  1352. defer cancel()
  1353. r, err := c.client.ListNetworks(ctx, &swarmapi.ListNetworksRequest{})
  1354. if err != nil {
  1355. return nil, err
  1356. }
  1357. var networks []apitypes.NetworkResource
  1358. for _, network := range r.Networks {
  1359. networks = append(networks, convert.BasicNetworkFromGRPC(*network))
  1360. }
  1361. return networks, nil
  1362. }
  1363. func attacherKey(target, containerID string) string {
  1364. return containerID + ":" + target
  1365. }
  1366. // UpdateAttachment signals the attachment config to the attachment
  1367. // waiter who is trying to start or attach the container to the
  1368. // network.
  1369. func (c *Cluster) UpdateAttachment(target, containerID string, config *network.NetworkingConfig) error {
  1370. c.RLock()
  1371. attacher, ok := c.attachers[attacherKey(target, containerID)]
  1372. c.RUnlock()
  1373. if !ok || attacher == nil {
  1374. return fmt.Errorf("could not find attacher for container %s to network %s", containerID, target)
  1375. }
  1376. attacher.attachWaitCh <- config
  1377. close(attacher.attachWaitCh)
  1378. return nil
  1379. }
  1380. // WaitForDetachment waits for the container to stop or detach from
  1381. // the network.
  1382. func (c *Cluster) WaitForDetachment(ctx context.Context, networkName, networkID, taskID, containerID string) error {
  1383. c.RLock()
  1384. attacher, ok := c.attachers[attacherKey(networkName, containerID)]
  1385. if !ok {
  1386. attacher, ok = c.attachers[attacherKey(networkID, containerID)]
  1387. }
  1388. if c.node == nil || c.node.Agent() == nil {
  1389. c.RUnlock()
  1390. return fmt.Errorf("invalid cluster node while waiting for detachment")
  1391. }
  1392. agent := c.node.Agent()
  1393. c.RUnlock()
  1394. if ok && attacher != nil &&
  1395. attacher.detachWaitCh != nil &&
  1396. attacher.attachCompleteCh != nil {
  1397. // Attachment may be in progress still so wait for
  1398. // attachment to complete.
  1399. select {
  1400. case <-attacher.attachCompleteCh:
  1401. case <-ctx.Done():
  1402. return ctx.Err()
  1403. }
  1404. if attacher.taskID == taskID {
  1405. select {
  1406. case <-attacher.detachWaitCh:
  1407. case <-ctx.Done():
  1408. return ctx.Err()
  1409. }
  1410. }
  1411. }
  1412. return agent.ResourceAllocator().DetachNetwork(ctx, taskID)
  1413. }
  1414. // AttachNetwork generates an attachment request towards the manager.
  1415. func (c *Cluster) AttachNetwork(target string, containerID string, addresses []string) (*network.NetworkingConfig, error) {
  1416. aKey := attacherKey(target, containerID)
  1417. c.Lock()
  1418. if c.node == nil || c.node.Agent() == nil {
  1419. c.Unlock()
  1420. return nil, fmt.Errorf("invalid cluster node while attaching to network")
  1421. }
  1422. if attacher, ok := c.attachers[aKey]; ok {
  1423. c.Unlock()
  1424. return attacher.config, nil
  1425. }
  1426. agent := c.node.Agent()
  1427. attachWaitCh := make(chan *network.NetworkingConfig)
  1428. detachWaitCh := make(chan struct{})
  1429. attachCompleteCh := make(chan struct{})
  1430. c.attachers[aKey] = &attacher{
  1431. attachWaitCh: attachWaitCh,
  1432. attachCompleteCh: attachCompleteCh,
  1433. detachWaitCh: detachWaitCh,
  1434. }
  1435. c.Unlock()
  1436. ctx, cancel := c.getRequestContext()
  1437. defer cancel()
  1438. taskID, err := agent.ResourceAllocator().AttachNetwork(ctx, containerID, target, addresses)
  1439. if err != nil {
  1440. c.Lock()
  1441. delete(c.attachers, aKey)
  1442. c.Unlock()
  1443. return nil, fmt.Errorf("Could not attach to network %s: %v", target, err)
  1444. }
  1445. c.Lock()
  1446. c.attachers[aKey].taskID = taskID
  1447. close(attachCompleteCh)
  1448. c.Unlock()
  1449. logrus.Debugf("Successfully attached to network %s with tid %s", target, taskID)
  1450. var config *network.NetworkingConfig
  1451. select {
  1452. case config = <-attachWaitCh:
  1453. case <-ctx.Done():
  1454. return nil, fmt.Errorf("attaching to network failed, make sure your network options are correct and check manager logs: %v", ctx.Err())
  1455. }
  1456. c.Lock()
  1457. c.attachers[aKey].config = config
  1458. c.Unlock()
  1459. return config, nil
  1460. }
  1461. // DetachNetwork unblocks the waiters waiting on WaitForDetachment so
  1462. // that a request to detach can be generated towards the manager.
  1463. func (c *Cluster) DetachNetwork(target string, containerID string) error {
  1464. aKey := attacherKey(target, containerID)
  1465. c.Lock()
  1466. attacher, ok := c.attachers[aKey]
  1467. delete(c.attachers, aKey)
  1468. c.Unlock()
  1469. if !ok {
  1470. return fmt.Errorf("could not find network attachment for container %s to network %s", containerID, target)
  1471. }
  1472. close(attacher.detachWaitCh)
  1473. return nil
  1474. }
  1475. // CreateNetwork creates a new cluster managed network.
  1476. func (c *Cluster) CreateNetwork(s apitypes.NetworkCreateRequest) (string, error) {
  1477. c.RLock()
  1478. defer c.RUnlock()
  1479. if !c.isActiveManager() {
  1480. return "", c.errNoManager()
  1481. }
  1482. if runconfig.IsPreDefinedNetwork(s.Name) {
  1483. err := fmt.Errorf("%s is a pre-defined network and cannot be created", s.Name)
  1484. return "", apierrors.NewRequestForbiddenError(err)
  1485. }
  1486. ctx, cancel := c.getRequestContext()
  1487. defer cancel()
  1488. networkSpec := convert.BasicNetworkCreateToGRPC(s)
  1489. r, err := c.client.CreateNetwork(ctx, &swarmapi.CreateNetworkRequest{Spec: &networkSpec})
  1490. if err != nil {
  1491. return "", err
  1492. }
  1493. return r.Network.ID, nil
  1494. }
  1495. // RemoveNetwork removes a cluster network.
  1496. func (c *Cluster) RemoveNetwork(input string) error {
  1497. c.RLock()
  1498. defer c.RUnlock()
  1499. if !c.isActiveManager() {
  1500. return c.errNoManager()
  1501. }
  1502. ctx, cancel := c.getRequestContext()
  1503. defer cancel()
  1504. network, err := getNetwork(ctx, c.client, input)
  1505. if err != nil {
  1506. return err
  1507. }
  1508. if _, err := c.client.RemoveNetwork(ctx, &swarmapi.RemoveNetworkRequest{NetworkID: network.ID}); err != nil {
  1509. return err
  1510. }
  1511. return nil
  1512. }
  1513. func (c *Cluster) populateNetworkID(ctx context.Context, client swarmapi.ControlClient, s *types.ServiceSpec) error {
  1514. // Always prefer NetworkAttachmentConfigs from TaskTemplate
  1515. // but fallback to service spec for backward compatibility
  1516. networks := s.TaskTemplate.Networks
  1517. if len(networks) == 0 {
  1518. networks = s.Networks
  1519. }
  1520. for i, n := range networks {
  1521. apiNetwork, err := getNetwork(ctx, client, n.Target)
  1522. if err != nil {
  1523. if ln, _ := c.config.Backend.FindNetwork(n.Target); ln != nil && !ln.Info().Dynamic() {
  1524. err = fmt.Errorf("network %s is not eligible for docker services", ln.Name())
  1525. return apierrors.NewRequestForbiddenError(err)
  1526. }
  1527. return err
  1528. }
  1529. networks[i].Target = apiNetwork.ID
  1530. }
  1531. return nil
  1532. }
  1533. // Cleanup stops active swarm node. This is run before daemon shutdown.
  1534. func (c *Cluster) Cleanup() {
  1535. c.Lock()
  1536. node := c.node
  1537. if node == nil {
  1538. c.Unlock()
  1539. return
  1540. }
  1541. defer c.Unlock()
  1542. if c.isActiveManager() {
  1543. active, reachable, unreachable, err := c.managerStats()
  1544. if err == nil {
  1545. singlenode := active && isLastManager(reachable, unreachable)
  1546. if active && !singlenode && removingManagerCausesLossOfQuorum(reachable, unreachable) {
  1547. logrus.Errorf("Leaving cluster with %v managers left out of %v. Raft quorum will be lost.", reachable-1, reachable+unreachable)
  1548. }
  1549. }
  1550. }
  1551. c.stopNode()
  1552. }
  1553. func (c *Cluster) managerStats() (current bool, reachable int, unreachable int, err error) {
  1554. ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
  1555. defer cancel()
  1556. nodes, err := c.client.ListNodes(ctx, &swarmapi.ListNodesRequest{})
  1557. if err != nil {
  1558. return false, 0, 0, err
  1559. }
  1560. for _, n := range nodes.Nodes {
  1561. if n.ManagerStatus != nil {
  1562. if n.ManagerStatus.Reachability == swarmapi.RaftMemberStatus_REACHABLE {
  1563. reachable++
  1564. if n.ID == c.node.NodeID() {
  1565. current = true
  1566. }
  1567. }
  1568. if n.ManagerStatus.Reachability == swarmapi.RaftMemberStatus_UNREACHABLE {
  1569. unreachable++
  1570. }
  1571. }
  1572. }
  1573. return
  1574. }
  1575. func validateAndSanitizeInitRequest(req *types.InitRequest) error {
  1576. var err error
  1577. req.ListenAddr, err = validateAddr(req.ListenAddr)
  1578. if err != nil {
  1579. return fmt.Errorf("invalid ListenAddr %q: %v", req.ListenAddr, err)
  1580. }
  1581. return nil
  1582. }
  1583. func validateAndSanitizeJoinRequest(req *types.JoinRequest) error {
  1584. var err error
  1585. req.ListenAddr, err = validateAddr(req.ListenAddr)
  1586. if err != nil {
  1587. return fmt.Errorf("invalid ListenAddr %q: %v", req.ListenAddr, err)
  1588. }
  1589. if len(req.RemoteAddrs) == 0 {
  1590. return fmt.Errorf("at least 1 RemoteAddr is required to join")
  1591. }
  1592. for i := range req.RemoteAddrs {
  1593. req.RemoteAddrs[i], err = validateAddr(req.RemoteAddrs[i])
  1594. if err != nil {
  1595. return fmt.Errorf("invalid remoteAddr %q: %v", req.RemoteAddrs[i], err)
  1596. }
  1597. }
  1598. return nil
  1599. }
  1600. func validateAddr(addr string) (string, error) {
  1601. if addr == "" {
  1602. return addr, fmt.Errorf("invalid empty address")
  1603. }
  1604. newaddr, err := opts.ParseTCPAddr(addr, defaultAddr)
  1605. if err != nil {
  1606. return addr, nil
  1607. }
  1608. return strings.TrimPrefix(newaddr, "tcp://"), nil
  1609. }
  1610. func initClusterSpec(node *node, spec types.Spec) error {
  1611. ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
  1612. for conn := range node.ListenControlSocket(ctx) {
  1613. if ctx.Err() != nil {
  1614. return ctx.Err()
  1615. }
  1616. if conn != nil {
  1617. client := swarmapi.NewControlClient(conn)
  1618. var cluster *swarmapi.Cluster
  1619. for i := 0; ; i++ {
  1620. lcr, err := client.ListClusters(ctx, &swarmapi.ListClustersRequest{})
  1621. if err != nil {
  1622. return fmt.Errorf("error on listing clusters: %v", err)
  1623. }
  1624. if len(lcr.Clusters) == 0 {
  1625. if i < 10 {
  1626. time.Sleep(200 * time.Millisecond)
  1627. continue
  1628. }
  1629. return fmt.Errorf("empty list of clusters was returned")
  1630. }
  1631. cluster = lcr.Clusters[0]
  1632. break
  1633. }
  1634. // In init, we take the initial default values from swarmkit, and merge
  1635. // any non nil or 0 value from spec to GRPC spec. This will leave the
  1636. // default value alone.
  1637. // Note that this is different from Update(), as in Update() we expect
  1638. // user to specify the complete spec of the cluster (as they already know
  1639. // the existing one and knows which field to update)
  1640. clusterSpec, err := convert.MergeSwarmSpecToGRPC(spec, cluster.Spec)
  1641. if err != nil {
  1642. return fmt.Errorf("error updating cluster settings: %v", err)
  1643. }
  1644. _, err = client.UpdateCluster(ctx, &swarmapi.UpdateClusterRequest{
  1645. ClusterID: cluster.ID,
  1646. ClusterVersion: &cluster.Meta.Version,
  1647. Spec: &clusterSpec,
  1648. })
  1649. if err != nil {
  1650. return fmt.Errorf("error updating cluster settings: %v", err)
  1651. }
  1652. return nil
  1653. }
  1654. }
  1655. return ctx.Err()
  1656. }
  1657. func detectLockedError(err error) error {
  1658. if err == swarmnode.ErrInvalidUnlockKey {
  1659. return errors.WithStack(ErrSwarmLocked)
  1660. }
  1661. return err
  1662. }