server.go 69 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489
  1. // DEPRECATION NOTICE. PLEASE DO NOT ADD ANYTHING TO THIS FILE.
  2. //
  3. // server/server.go is deprecated. We are working on breaking it up into smaller, cleaner
  4. // pieces which will be easier to find and test. This will help make the code less
  5. // redundant and more readable.
  6. //
  7. // Contributors, please don't add anything to server/server.go, unless it has the explicit
  8. // goal of helping the deprecation effort.
  9. //
  10. // Maintainers, please refuse patches which add code to server/server.go.
  11. //
  12. // Instead try the following files:
  13. // * For code related to local image management, try graph/
  14. // * For code related to image downloading, uploading, remote search etc, try registry/
  15. // * For code related to the docker daemon, try daemon/
  16. // * For small utilities which could potentially be useful outside of Docker, try pkg/
  17. // * For miscalleneous "util" functions which are docker-specific, try encapsulating them
  18. // inside one of the subsystems above. If you really think they should be more widely
  19. // available, are you sure you can't remove the docker dependencies and move them to
  20. // pkg? In last resort, you can add them to utils/ (but please try not to).
  21. package server
  22. import (
  23. "encoding/json"
  24. "fmt"
  25. "io"
  26. "io/ioutil"
  27. "log"
  28. "net"
  29. "net/http"
  30. "net/url"
  31. "os"
  32. "os/exec"
  33. gosignal "os/signal"
  34. "path"
  35. "path/filepath"
  36. "runtime"
  37. "strconv"
  38. "strings"
  39. "sync"
  40. "sync/atomic"
  41. "syscall"
  42. "time"
  43. "github.com/dotcloud/docker/archive"
  44. "github.com/dotcloud/docker/daemon"
  45. "github.com/dotcloud/docker/daemonconfig"
  46. "github.com/dotcloud/docker/dockerversion"
  47. "github.com/dotcloud/docker/engine"
  48. "github.com/dotcloud/docker/graph"
  49. "github.com/dotcloud/docker/image"
  50. "github.com/dotcloud/docker/pkg/graphdb"
  51. "github.com/dotcloud/docker/pkg/signal"
  52. "github.com/dotcloud/docker/registry"
  53. "github.com/dotcloud/docker/runconfig"
  54. "github.com/dotcloud/docker/utils"
  55. "github.com/dotcloud/docker/utils/filters"
  56. )
  57. func (srv *Server) handlerWrap(h engine.Handler) engine.Handler {
  58. return func(job *engine.Job) engine.Status {
  59. if !srv.IsRunning() {
  60. return job.Errorf("Server is not running")
  61. }
  62. srv.tasks.Add(1)
  63. defer srv.tasks.Done()
  64. return h(job)
  65. }
  66. }
  67. // jobInitApi runs the remote api server `srv` as a daemon,
  68. // Only one api server can run at the same time - this is enforced by a pidfile.
  69. // The signals SIGINT, SIGQUIT and SIGTERM are intercepted for cleanup.
  70. func InitServer(job *engine.Job) engine.Status {
  71. job.Logf("Creating server")
  72. srv, err := NewServer(job.Eng, daemonconfig.ConfigFromJob(job))
  73. if err != nil {
  74. return job.Error(err)
  75. }
  76. if srv.daemon.Config().Pidfile != "" {
  77. job.Logf("Creating pidfile")
  78. if err := utils.CreatePidFile(srv.daemon.Config().Pidfile); err != nil {
  79. // FIXME: do we need fatal here instead of returning a job error?
  80. log.Fatal(err)
  81. }
  82. }
  83. job.Logf("Setting up signal traps")
  84. c := make(chan os.Signal, 1)
  85. gosignal.Notify(c, os.Interrupt, syscall.SIGTERM, syscall.SIGQUIT)
  86. go func() {
  87. interruptCount := uint32(0)
  88. for sig := range c {
  89. go func(sig os.Signal) {
  90. log.Printf("Received signal '%v', starting shutdown of docker...\n", sig)
  91. switch sig {
  92. case os.Interrupt, syscall.SIGTERM:
  93. // If the user really wants to interrupt, let him do so.
  94. if atomic.LoadUint32(&interruptCount) < 3 {
  95. atomic.AddUint32(&interruptCount, 1)
  96. // Initiate the cleanup only once
  97. if atomic.LoadUint32(&interruptCount) == 1 {
  98. utils.RemovePidFile(srv.daemon.Config().Pidfile)
  99. srv.Close()
  100. } else {
  101. return
  102. }
  103. } else {
  104. log.Printf("Force shutdown of docker, interrupting cleanup\n")
  105. }
  106. case syscall.SIGQUIT:
  107. }
  108. os.Exit(128 + int(sig.(syscall.Signal)))
  109. }(sig)
  110. }
  111. }()
  112. job.Eng.Hack_SetGlobalVar("httpapi.server", srv)
  113. job.Eng.Hack_SetGlobalVar("httpapi.daemon", srv.daemon)
  114. // FIXME: 'insert' is deprecated and should be removed in a future version.
  115. for name, handler := range map[string]engine.Handler{
  116. "export": srv.ContainerExport,
  117. "create": srv.ContainerCreate,
  118. "stop": srv.ContainerStop,
  119. "restart": srv.ContainerRestart,
  120. "start": srv.ContainerStart,
  121. "kill": srv.ContainerKill,
  122. "wait": srv.ContainerWait,
  123. "tag": srv.ImageTag, // FIXME merge with "image_tag"
  124. "resize": srv.ContainerResize,
  125. "commit": srv.ContainerCommit,
  126. "info": srv.DockerInfo,
  127. "container_delete": srv.ContainerDestroy,
  128. "image_export": srv.ImageExport,
  129. "images": srv.Images,
  130. "history": srv.ImageHistory,
  131. "viz": srv.ImagesViz,
  132. "container_copy": srv.ContainerCopy,
  133. "insert": srv.ImageInsert,
  134. "attach": srv.ContainerAttach,
  135. "logs": srv.ContainerLogs,
  136. "changes": srv.ContainerChanges,
  137. "top": srv.ContainerTop,
  138. "load": srv.ImageLoad,
  139. "build": srv.Build,
  140. "pull": srv.ImagePull,
  141. "import": srv.ImageImport,
  142. "image_delete": srv.ImageDelete,
  143. "events": srv.Events,
  144. "push": srv.ImagePush,
  145. "containers": srv.Containers,
  146. } {
  147. if err := job.Eng.Register(name, srv.handlerWrap(handler)); err != nil {
  148. return job.Error(err)
  149. }
  150. }
  151. // Install image-related commands from the image subsystem.
  152. // See `graph/service.go`
  153. if err := srv.daemon.Repositories().Install(job.Eng); err != nil {
  154. return job.Error(err)
  155. }
  156. // Install daemon-related commands from the daemon subsystem.
  157. // See `daemon/`
  158. if err := srv.daemon.Install(job.Eng); err != nil {
  159. return job.Error(err)
  160. }
  161. srv.SetRunning(true)
  162. return engine.StatusOK
  163. }
  164. // ContainerKill send signal to the container
  165. // If no signal is given (sig 0), then Kill with SIGKILL and wait
  166. // for the container to exit.
  167. // If a signal is given, then just send it to the container and return.
  168. func (srv *Server) ContainerKill(job *engine.Job) engine.Status {
  169. if n := len(job.Args); n < 1 || n > 2 {
  170. return job.Errorf("Usage: %s CONTAINER [SIGNAL]", job.Name)
  171. }
  172. var (
  173. name = job.Args[0]
  174. sig uint64
  175. err error
  176. )
  177. // If we have a signal, look at it. Otherwise, do nothing
  178. if len(job.Args) == 2 && job.Args[1] != "" {
  179. // Check if we passed the signal as a number:
  180. // The largest legal signal is 31, so let's parse on 5 bits
  181. sig, err = strconv.ParseUint(job.Args[1], 10, 5)
  182. if err != nil {
  183. // The signal is not a number, treat it as a string (either like "KILL" or like "SIGKILL")
  184. sig = uint64(signal.SignalMap[strings.TrimPrefix(job.Args[1], "SIG")])
  185. if sig == 0 {
  186. return job.Errorf("Invalid signal: %s", job.Args[1])
  187. }
  188. }
  189. }
  190. if container := srv.daemon.Get(name); container != nil {
  191. // If no signal is passed, or SIGKILL, perform regular Kill (SIGKILL + wait())
  192. if sig == 0 || syscall.Signal(sig) == syscall.SIGKILL {
  193. if err := container.Kill(); err != nil {
  194. return job.Errorf("Cannot kill container %s: %s", name, err)
  195. }
  196. srv.LogEvent("kill", container.ID, srv.daemon.Repositories().ImageName(container.Image))
  197. } else {
  198. // Otherwise, just send the requested signal
  199. if err := container.KillSig(int(sig)); err != nil {
  200. return job.Errorf("Cannot kill container %s: %s", name, err)
  201. }
  202. // FIXME: Add event for signals
  203. }
  204. } else {
  205. return job.Errorf("No such container: %s", name)
  206. }
  207. return engine.StatusOK
  208. }
  209. func (srv *Server) EvictListener(from int64) {
  210. srv.Lock()
  211. if old, ok := srv.listeners[from]; ok {
  212. delete(srv.listeners, from)
  213. close(old)
  214. }
  215. srv.Unlock()
  216. }
  217. func (srv *Server) Events(job *engine.Job) engine.Status {
  218. if len(job.Args) != 0 {
  219. return job.Errorf("Usage: %s", job.Name)
  220. }
  221. var (
  222. from = time.Now().UTC().UnixNano()
  223. since = job.GetenvInt64("since")
  224. until = job.GetenvInt64("until")
  225. timeout = time.NewTimer(time.Unix(until, 0).Sub(time.Now()))
  226. )
  227. sendEvent := func(event *utils.JSONMessage) error {
  228. b, err := json.Marshal(event)
  229. if err != nil {
  230. return fmt.Errorf("JSON error")
  231. }
  232. _, err = job.Stdout.Write(b)
  233. return err
  234. }
  235. listener := make(chan utils.JSONMessage)
  236. srv.Lock()
  237. if old, ok := srv.listeners[from]; ok {
  238. delete(srv.listeners, from)
  239. close(old)
  240. }
  241. srv.listeners[from] = listener
  242. srv.Unlock()
  243. job.Stdout.Write(nil) // flush
  244. if since != 0 {
  245. // If since, send previous events that happened after the timestamp and until timestamp
  246. for _, event := range srv.GetEvents() {
  247. if event.Time >= since && (event.Time <= until || until == 0) {
  248. err := sendEvent(&event)
  249. if err != nil && err.Error() == "JSON error" {
  250. continue
  251. }
  252. if err != nil {
  253. // On error, evict the listener
  254. srv.EvictListener(from)
  255. return job.Error(err)
  256. }
  257. }
  258. }
  259. }
  260. // If no until, disable timeout
  261. if until == 0 {
  262. timeout.Stop()
  263. }
  264. for {
  265. select {
  266. case event, ok := <-listener:
  267. if !ok { // Channel is closed: listener was evicted
  268. return engine.StatusOK
  269. }
  270. err := sendEvent(&event)
  271. if err != nil && err.Error() == "JSON error" {
  272. continue
  273. }
  274. if err != nil {
  275. // On error, evict the listener
  276. srv.EvictListener(from)
  277. return job.Error(err)
  278. }
  279. case <-timeout.C:
  280. return engine.StatusOK
  281. }
  282. }
  283. return engine.StatusOK
  284. }
  285. func (srv *Server) ContainerExport(job *engine.Job) engine.Status {
  286. if len(job.Args) != 1 {
  287. return job.Errorf("Usage: %s container_id", job.Name)
  288. }
  289. name := job.Args[0]
  290. if container := srv.daemon.Get(name); container != nil {
  291. data, err := container.Export()
  292. if err != nil {
  293. return job.Errorf("%s: %s", name, err)
  294. }
  295. defer data.Close()
  296. // Stream the entire contents of the container (basically a volatile snapshot)
  297. if _, err := io.Copy(job.Stdout, data); err != nil {
  298. return job.Errorf("%s: %s", name, err)
  299. }
  300. // FIXME: factor job-specific LogEvent to engine.Job.Run()
  301. srv.LogEvent("export", container.ID, srv.daemon.Repositories().ImageName(container.Image))
  302. return engine.StatusOK
  303. }
  304. return job.Errorf("No such container: %s", name)
  305. }
  306. // ImageExport exports all images with the given tag. All versions
  307. // containing the same tag are exported. The resulting output is an
  308. // uncompressed tar ball.
  309. // name is the set of tags to export.
  310. // out is the writer where the images are written to.
  311. func (srv *Server) ImageExport(job *engine.Job) engine.Status {
  312. if len(job.Args) != 1 {
  313. return job.Errorf("Usage: %s IMAGE\n", job.Name)
  314. }
  315. name := job.Args[0]
  316. // get image json
  317. tempdir, err := ioutil.TempDir("", "docker-export-")
  318. if err != nil {
  319. return job.Error(err)
  320. }
  321. defer os.RemoveAll(tempdir)
  322. utils.Debugf("Serializing %s", name)
  323. rootRepo, err := srv.daemon.Repositories().Get(name)
  324. if err != nil {
  325. return job.Error(err)
  326. }
  327. if rootRepo != nil {
  328. for _, id := range rootRepo {
  329. if err := srv.exportImage(job.Eng, id, tempdir); err != nil {
  330. return job.Error(err)
  331. }
  332. }
  333. // write repositories
  334. rootRepoMap := map[string]graph.Repository{}
  335. rootRepoMap[name] = rootRepo
  336. rootRepoJson, _ := json.Marshal(rootRepoMap)
  337. if err := ioutil.WriteFile(path.Join(tempdir, "repositories"), rootRepoJson, os.FileMode(0644)); err != nil {
  338. return job.Error(err)
  339. }
  340. } else {
  341. if err := srv.exportImage(job.Eng, name, tempdir); err != nil {
  342. return job.Error(err)
  343. }
  344. }
  345. fs, err := archive.Tar(tempdir, archive.Uncompressed)
  346. if err != nil {
  347. return job.Error(err)
  348. }
  349. defer fs.Close()
  350. if _, err := io.Copy(job.Stdout, fs); err != nil {
  351. return job.Error(err)
  352. }
  353. utils.Debugf("End Serializing %s", name)
  354. return engine.StatusOK
  355. }
  356. func (srv *Server) exportImage(eng *engine.Engine, name, tempdir string) error {
  357. for n := name; n != ""; {
  358. // temporary directory
  359. tmpImageDir := path.Join(tempdir, n)
  360. if err := os.Mkdir(tmpImageDir, os.FileMode(0755)); err != nil {
  361. if os.IsExist(err) {
  362. return nil
  363. }
  364. return err
  365. }
  366. var version = "1.0"
  367. var versionBuf = []byte(version)
  368. if err := ioutil.WriteFile(path.Join(tmpImageDir, "VERSION"), versionBuf, os.FileMode(0644)); err != nil {
  369. return err
  370. }
  371. // serialize json
  372. json, err := os.Create(path.Join(tmpImageDir, "json"))
  373. if err != nil {
  374. return err
  375. }
  376. job := eng.Job("image_inspect", n)
  377. job.Stdout.Add(json)
  378. if err := job.Run(); err != nil {
  379. return err
  380. }
  381. // serialize filesystem
  382. fsTar, err := os.Create(path.Join(tmpImageDir, "layer.tar"))
  383. if err != nil {
  384. return err
  385. }
  386. job = eng.Job("image_tarlayer", n)
  387. job.Stdout.Add(fsTar)
  388. if err := job.Run(); err != nil {
  389. return err
  390. }
  391. // find parent
  392. job = eng.Job("image_get", n)
  393. info, _ := job.Stdout.AddEnv()
  394. if err := job.Run(); err != nil {
  395. return err
  396. }
  397. n = info.Get("Parent")
  398. }
  399. return nil
  400. }
  401. func (srv *Server) Build(job *engine.Job) engine.Status {
  402. if len(job.Args) != 0 {
  403. return job.Errorf("Usage: %s\n", job.Name)
  404. }
  405. var (
  406. remoteURL = job.Getenv("remote")
  407. repoName = job.Getenv("t")
  408. suppressOutput = job.GetenvBool("q")
  409. noCache = job.GetenvBool("nocache")
  410. rm = job.GetenvBool("rm")
  411. forceRm = job.GetenvBool("forcerm")
  412. authConfig = &registry.AuthConfig{}
  413. configFile = &registry.ConfigFile{}
  414. tag string
  415. context io.ReadCloser
  416. )
  417. job.GetenvJson("authConfig", authConfig)
  418. job.GetenvJson("configFile", configFile)
  419. repoName, tag = utils.ParseRepositoryTag(repoName)
  420. if remoteURL == "" {
  421. context = ioutil.NopCloser(job.Stdin)
  422. } else if utils.IsGIT(remoteURL) {
  423. if !strings.HasPrefix(remoteURL, "git://") {
  424. remoteURL = "https://" + remoteURL
  425. }
  426. root, err := ioutil.TempDir("", "docker-build-git")
  427. if err != nil {
  428. return job.Error(err)
  429. }
  430. defer os.RemoveAll(root)
  431. if output, err := exec.Command("git", "clone", "--recursive", remoteURL, root).CombinedOutput(); err != nil {
  432. return job.Errorf("Error trying to use git: %s (%s)", err, output)
  433. }
  434. c, err := archive.Tar(root, archive.Uncompressed)
  435. if err != nil {
  436. return job.Error(err)
  437. }
  438. context = c
  439. } else if utils.IsURL(remoteURL) {
  440. f, err := utils.Download(remoteURL)
  441. if err != nil {
  442. return job.Error(err)
  443. }
  444. defer f.Body.Close()
  445. dockerFile, err := ioutil.ReadAll(f.Body)
  446. if err != nil {
  447. return job.Error(err)
  448. }
  449. c, err := archive.Generate("Dockerfile", string(dockerFile))
  450. if err != nil {
  451. return job.Error(err)
  452. }
  453. context = c
  454. }
  455. defer context.Close()
  456. sf := utils.NewStreamFormatter(job.GetenvBool("json"))
  457. b := NewBuildFile(srv,
  458. &utils.StdoutFormater{
  459. Writer: job.Stdout,
  460. StreamFormatter: sf,
  461. },
  462. &utils.StderrFormater{
  463. Writer: job.Stdout,
  464. StreamFormatter: sf,
  465. },
  466. !suppressOutput, !noCache, rm, forceRm, job.Stdout, sf, authConfig, configFile)
  467. id, err := b.Build(context)
  468. if err != nil {
  469. return job.Error(err)
  470. }
  471. if repoName != "" {
  472. srv.daemon.Repositories().Set(repoName, tag, id, false)
  473. }
  474. return engine.StatusOK
  475. }
  476. // Loads a set of images into the repository. This is the complementary of ImageExport.
  477. // The input stream is an uncompressed tar ball containing images and metadata.
  478. func (srv *Server) ImageLoad(job *engine.Job) engine.Status {
  479. tmpImageDir, err := ioutil.TempDir("", "docker-import-")
  480. if err != nil {
  481. return job.Error(err)
  482. }
  483. defer os.RemoveAll(tmpImageDir)
  484. var (
  485. repoTarFile = path.Join(tmpImageDir, "repo.tar")
  486. repoDir = path.Join(tmpImageDir, "repo")
  487. )
  488. tarFile, err := os.Create(repoTarFile)
  489. if err != nil {
  490. return job.Error(err)
  491. }
  492. if _, err := io.Copy(tarFile, job.Stdin); err != nil {
  493. return job.Error(err)
  494. }
  495. tarFile.Close()
  496. repoFile, err := os.Open(repoTarFile)
  497. if err != nil {
  498. return job.Error(err)
  499. }
  500. if err := os.Mkdir(repoDir, os.ModeDir); err != nil {
  501. return job.Error(err)
  502. }
  503. if err := archive.Untar(repoFile, repoDir, nil); err != nil {
  504. return job.Error(err)
  505. }
  506. dirs, err := ioutil.ReadDir(repoDir)
  507. if err != nil {
  508. return job.Error(err)
  509. }
  510. for _, d := range dirs {
  511. if d.IsDir() {
  512. if err := srv.recursiveLoad(job.Eng, d.Name(), tmpImageDir); err != nil {
  513. return job.Error(err)
  514. }
  515. }
  516. }
  517. repositoriesJson, err := ioutil.ReadFile(path.Join(tmpImageDir, "repo", "repositories"))
  518. if err == nil {
  519. repositories := map[string]graph.Repository{}
  520. if err := json.Unmarshal(repositoriesJson, &repositories); err != nil {
  521. return job.Error(err)
  522. }
  523. for imageName, tagMap := range repositories {
  524. for tag, address := range tagMap {
  525. if err := srv.daemon.Repositories().Set(imageName, tag, address, true); err != nil {
  526. return job.Error(err)
  527. }
  528. }
  529. }
  530. } else if !os.IsNotExist(err) {
  531. return job.Error(err)
  532. }
  533. return engine.StatusOK
  534. }
  535. func (srv *Server) recursiveLoad(eng *engine.Engine, address, tmpImageDir string) error {
  536. if err := eng.Job("image_get", address).Run(); err != nil {
  537. utils.Debugf("Loading %s", address)
  538. imageJson, err := ioutil.ReadFile(path.Join(tmpImageDir, "repo", address, "json"))
  539. if err != nil {
  540. utils.Debugf("Error reading json", err)
  541. return err
  542. }
  543. layer, err := os.Open(path.Join(tmpImageDir, "repo", address, "layer.tar"))
  544. if err != nil {
  545. utils.Debugf("Error reading embedded tar", err)
  546. return err
  547. }
  548. img, err := image.NewImgJSON(imageJson)
  549. if err != nil {
  550. utils.Debugf("Error unmarshalling json", err)
  551. return err
  552. }
  553. if img.Parent != "" {
  554. if !srv.daemon.Graph().Exists(img.Parent) {
  555. if err := srv.recursiveLoad(eng, img.Parent, tmpImageDir); err != nil {
  556. return err
  557. }
  558. }
  559. }
  560. if err := srv.daemon.Graph().Register(imageJson, layer, img); err != nil {
  561. return err
  562. }
  563. }
  564. utils.Debugf("Completed processing %s", address)
  565. return nil
  566. }
  567. // FIXME: 'insert' is deprecated and should be removed in a future version.
  568. func (srv *Server) ImageInsert(job *engine.Job) engine.Status {
  569. fmt.Fprintf(job.Stderr, "Warning: '%s' is deprecated and will be removed in a future version. Please use 'build' and 'ADD' instead.\n", job.Name)
  570. if len(job.Args) != 3 {
  571. return job.Errorf("Usage: %s IMAGE URL PATH\n", job.Name)
  572. }
  573. var (
  574. name = job.Args[0]
  575. url = job.Args[1]
  576. path = job.Args[2]
  577. )
  578. sf := utils.NewStreamFormatter(job.GetenvBool("json"))
  579. out := utils.NewWriteFlusher(job.Stdout)
  580. img, err := srv.daemon.Repositories().LookupImage(name)
  581. if err != nil {
  582. return job.Error(err)
  583. }
  584. file, err := utils.Download(url)
  585. if err != nil {
  586. return job.Error(err)
  587. }
  588. defer file.Body.Close()
  589. config, _, _, err := runconfig.Parse([]string{img.ID, "echo", "insert", url, path}, srv.daemon.SystemConfig())
  590. if err != nil {
  591. return job.Error(err)
  592. }
  593. c, _, err := srv.daemon.Create(config, "")
  594. if err != nil {
  595. return job.Error(err)
  596. }
  597. if err := c.Inject(utils.ProgressReader(file.Body, int(file.ContentLength), out, sf, false, utils.TruncateID(img.ID), "Downloading"), path); err != nil {
  598. return job.Error(err)
  599. }
  600. // FIXME: Handle custom repo, tag comment, author
  601. img, err = srv.daemon.Commit(c, "", "", img.Comment, img.Author, nil)
  602. if err != nil {
  603. out.Write(sf.FormatError(err))
  604. return engine.StatusErr
  605. }
  606. out.Write(sf.FormatStatus("", img.ID))
  607. return engine.StatusOK
  608. }
  609. func (srv *Server) ImagesViz(job *engine.Job) engine.Status {
  610. images, _ := srv.daemon.Graph().Map()
  611. if images == nil {
  612. return engine.StatusOK
  613. }
  614. job.Stdout.Write([]byte("digraph docker {\n"))
  615. var (
  616. parentImage *image.Image
  617. err error
  618. )
  619. for _, image := range images {
  620. parentImage, err = image.GetParent()
  621. if err != nil {
  622. return job.Errorf("Error while getting parent image: %v", err)
  623. }
  624. if parentImage != nil {
  625. job.Stdout.Write([]byte(" \"" + parentImage.ID + "\" -> \"" + image.ID + "\"\n"))
  626. } else {
  627. job.Stdout.Write([]byte(" base -> \"" + image.ID + "\" [style=invis]\n"))
  628. }
  629. }
  630. for id, repos := range srv.daemon.Repositories().GetRepoRefs() {
  631. job.Stdout.Write([]byte(" \"" + id + "\" [label=\"" + id + "\\n" + strings.Join(repos, "\\n") + "\",shape=box,fillcolor=\"paleturquoise\",style=\"filled,rounded\"];\n"))
  632. }
  633. job.Stdout.Write([]byte(" base [style=invisible]\n}\n"))
  634. return engine.StatusOK
  635. }
  636. func (srv *Server) Images(job *engine.Job) engine.Status {
  637. var (
  638. allImages map[string]*image.Image
  639. err error
  640. filt_tagged = true
  641. )
  642. imageFilters, err := filters.FromParam(job.Getenv("filters"))
  643. if err != nil {
  644. return job.Error(err)
  645. }
  646. if i, ok := imageFilters["dangling"]; ok {
  647. for _, value := range i {
  648. if strings.ToLower(value) == "true" {
  649. filt_tagged = false
  650. }
  651. }
  652. }
  653. if job.GetenvBool("all") && !filt_tagged {
  654. allImages, err = srv.daemon.Graph().Map()
  655. } else {
  656. allImages, err = srv.daemon.Graph().Heads()
  657. }
  658. if err != nil {
  659. return job.Error(err)
  660. }
  661. lookup := make(map[string]*engine.Env)
  662. srv.daemon.Repositories().Lock()
  663. for name, repository := range srv.daemon.Repositories().Repositories {
  664. if job.Getenv("filter") != "" {
  665. if match, _ := path.Match(job.Getenv("filter"), name); !match {
  666. continue
  667. }
  668. }
  669. for tag, id := range repository {
  670. image, err := srv.daemon.Graph().Get(id)
  671. if err != nil {
  672. log.Printf("Warning: couldn't load %s from %s/%s: %s", id, name, tag, err)
  673. continue
  674. }
  675. if out, exists := lookup[id]; exists {
  676. if filt_tagged {
  677. out.SetList("RepoTags", append(out.GetList("RepoTags"), fmt.Sprintf("%s:%s", name, tag)))
  678. }
  679. } else {
  680. // get the boolean list for if only the untagged images are requested
  681. delete(allImages, id)
  682. if filt_tagged {
  683. out := &engine.Env{}
  684. out.Set("ParentId", image.Parent)
  685. out.SetList("RepoTags", []string{fmt.Sprintf("%s:%s", name, tag)})
  686. out.Set("Id", image.ID)
  687. out.SetInt64("Created", image.Created.Unix())
  688. out.SetInt64("Size", image.Size)
  689. out.SetInt64("VirtualSize", image.GetParentsSize(0)+image.Size)
  690. lookup[id] = out
  691. }
  692. }
  693. }
  694. }
  695. srv.daemon.Repositories().Unlock()
  696. outs := engine.NewTable("Created", len(lookup))
  697. for _, value := range lookup {
  698. outs.Add(value)
  699. }
  700. // Display images which aren't part of a repository/tag
  701. if job.Getenv("filter") == "" {
  702. for _, image := range allImages {
  703. out := &engine.Env{}
  704. out.Set("ParentId", image.Parent)
  705. out.SetList("RepoTags", []string{"<none>:<none>"})
  706. out.Set("Id", image.ID)
  707. out.SetInt64("Created", image.Created.Unix())
  708. out.SetInt64("Size", image.Size)
  709. out.SetInt64("VirtualSize", image.GetParentsSize(0)+image.Size)
  710. outs.Add(out)
  711. }
  712. }
  713. outs.ReverseSort()
  714. if _, err := outs.WriteListTo(job.Stdout); err != nil {
  715. return job.Error(err)
  716. }
  717. return engine.StatusOK
  718. }
  719. func (srv *Server) DockerInfo(job *engine.Job) engine.Status {
  720. images, _ := srv.daemon.Graph().Map()
  721. var imgcount int
  722. if images == nil {
  723. imgcount = 0
  724. } else {
  725. imgcount = len(images)
  726. }
  727. kernelVersion := "<unknown>"
  728. if kv, err := utils.GetKernelVersion(); err == nil {
  729. kernelVersion = kv.String()
  730. }
  731. // if we still have the original dockerinit binary from before we copied it locally, let's return the path to that, since that's more intuitive (the copied path is trivial to derive by hand given VERSION)
  732. initPath := utils.DockerInitPath("")
  733. if initPath == "" {
  734. // if that fails, we'll just return the path from the daemon
  735. initPath = srv.daemon.SystemInitPath()
  736. }
  737. v := &engine.Env{}
  738. v.SetInt("Containers", len(srv.daemon.List()))
  739. v.SetInt("Images", imgcount)
  740. v.Set("Driver", srv.daemon.GraphDriver().String())
  741. v.SetJson("DriverStatus", srv.daemon.GraphDriver().Status())
  742. v.SetBool("MemoryLimit", srv.daemon.SystemConfig().MemoryLimit)
  743. v.SetBool("SwapLimit", srv.daemon.SystemConfig().SwapLimit)
  744. v.SetBool("IPv4Forwarding", !srv.daemon.SystemConfig().IPv4ForwardingDisabled)
  745. v.SetBool("Debug", os.Getenv("DEBUG") != "")
  746. v.SetInt("NFd", utils.GetTotalUsedFds())
  747. v.SetInt("NGoroutines", runtime.NumGoroutine())
  748. v.Set("ExecutionDriver", srv.daemon.ExecutionDriver().Name())
  749. v.SetInt("NEventsListener", len(srv.listeners))
  750. v.Set("KernelVersion", kernelVersion)
  751. v.Set("IndexServerAddress", registry.IndexServerAddress())
  752. v.Set("InitSha1", dockerversion.INITSHA1)
  753. v.Set("InitPath", initPath)
  754. if _, err := v.WriteTo(job.Stdout); err != nil {
  755. return job.Error(err)
  756. }
  757. return engine.StatusOK
  758. }
  759. func (srv *Server) ImageHistory(job *engine.Job) engine.Status {
  760. if n := len(job.Args); n != 1 {
  761. return job.Errorf("Usage: %s IMAGE", job.Name)
  762. }
  763. name := job.Args[0]
  764. foundImage, err := srv.daemon.Repositories().LookupImage(name)
  765. if err != nil {
  766. return job.Error(err)
  767. }
  768. lookupMap := make(map[string][]string)
  769. for name, repository := range srv.daemon.Repositories().Repositories {
  770. for tag, id := range repository {
  771. // If the ID already has a reverse lookup, do not update it unless for "latest"
  772. if _, exists := lookupMap[id]; !exists {
  773. lookupMap[id] = []string{}
  774. }
  775. lookupMap[id] = append(lookupMap[id], name+":"+tag)
  776. }
  777. }
  778. outs := engine.NewTable("Created", 0)
  779. err = foundImage.WalkHistory(func(img *image.Image) error {
  780. out := &engine.Env{}
  781. out.Set("Id", img.ID)
  782. out.SetInt64("Created", img.Created.Unix())
  783. out.Set("CreatedBy", strings.Join(img.ContainerConfig.Cmd, " "))
  784. out.SetList("Tags", lookupMap[img.ID])
  785. out.SetInt64("Size", img.Size)
  786. outs.Add(out)
  787. return nil
  788. })
  789. outs.ReverseSort()
  790. if _, err := outs.WriteListTo(job.Stdout); err != nil {
  791. return job.Error(err)
  792. }
  793. return engine.StatusOK
  794. }
  795. func (srv *Server) ContainerTop(job *engine.Job) engine.Status {
  796. if len(job.Args) != 1 && len(job.Args) != 2 {
  797. return job.Errorf("Not enough arguments. Usage: %s CONTAINER [PS_ARGS]\n", job.Name)
  798. }
  799. var (
  800. name = job.Args[0]
  801. psArgs = "-ef"
  802. )
  803. if len(job.Args) == 2 && job.Args[1] != "" {
  804. psArgs = job.Args[1]
  805. }
  806. if container := srv.daemon.Get(name); container != nil {
  807. if !container.State.IsRunning() {
  808. return job.Errorf("Container %s is not running", name)
  809. }
  810. pids, err := srv.daemon.ExecutionDriver().GetPidsForContainer(container.ID)
  811. if err != nil {
  812. return job.Error(err)
  813. }
  814. output, err := exec.Command("ps", psArgs).Output()
  815. if err != nil {
  816. return job.Errorf("Error running ps: %s", err)
  817. }
  818. lines := strings.Split(string(output), "\n")
  819. header := strings.Fields(lines[0])
  820. out := &engine.Env{}
  821. out.SetList("Titles", header)
  822. pidIndex := -1
  823. for i, name := range header {
  824. if name == "PID" {
  825. pidIndex = i
  826. }
  827. }
  828. if pidIndex == -1 {
  829. return job.Errorf("Couldn't find PID field in ps output")
  830. }
  831. processes := [][]string{}
  832. for _, line := range lines[1:] {
  833. if len(line) == 0 {
  834. continue
  835. }
  836. fields := strings.Fields(line)
  837. p, err := strconv.Atoi(fields[pidIndex])
  838. if err != nil {
  839. return job.Errorf("Unexpected pid '%s': %s", fields[pidIndex], err)
  840. }
  841. for _, pid := range pids {
  842. if pid == p {
  843. // Make sure number of fields equals number of header titles
  844. // merging "overhanging" fields
  845. process := fields[:len(header)-1]
  846. process = append(process, strings.Join(fields[len(header)-1:], " "))
  847. processes = append(processes, process)
  848. }
  849. }
  850. }
  851. out.SetJson("Processes", processes)
  852. out.WriteTo(job.Stdout)
  853. return engine.StatusOK
  854. }
  855. return job.Errorf("No such container: %s", name)
  856. }
  857. func (srv *Server) ContainerChanges(job *engine.Job) engine.Status {
  858. if n := len(job.Args); n != 1 {
  859. return job.Errorf("Usage: %s CONTAINER", job.Name)
  860. }
  861. name := job.Args[0]
  862. if container := srv.daemon.Get(name); container != nil {
  863. outs := engine.NewTable("", 0)
  864. changes, err := container.Changes()
  865. if err != nil {
  866. return job.Error(err)
  867. }
  868. for _, change := range changes {
  869. out := &engine.Env{}
  870. if err := out.Import(change); err != nil {
  871. return job.Error(err)
  872. }
  873. outs.Add(out)
  874. }
  875. if _, err := outs.WriteListTo(job.Stdout); err != nil {
  876. return job.Error(err)
  877. }
  878. } else {
  879. return job.Errorf("No such container: %s", name)
  880. }
  881. return engine.StatusOK
  882. }
  883. func (srv *Server) Containers(job *engine.Job) engine.Status {
  884. var (
  885. foundBefore bool
  886. displayed int
  887. all = job.GetenvBool("all")
  888. since = job.Getenv("since")
  889. before = job.Getenv("before")
  890. n = job.GetenvInt("limit")
  891. size = job.GetenvBool("size")
  892. )
  893. outs := engine.NewTable("Created", 0)
  894. names := map[string][]string{}
  895. srv.daemon.ContainerGraph().Walk("/", func(p string, e *graphdb.Entity) error {
  896. names[e.ID()] = append(names[e.ID()], p)
  897. return nil
  898. }, -1)
  899. var beforeCont, sinceCont *daemon.Container
  900. if before != "" {
  901. beforeCont = srv.daemon.Get(before)
  902. if beforeCont == nil {
  903. return job.Error(fmt.Errorf("Could not find container with name or id %s", before))
  904. }
  905. }
  906. if since != "" {
  907. sinceCont = srv.daemon.Get(since)
  908. if sinceCont == nil {
  909. return job.Error(fmt.Errorf("Could not find container with name or id %s", since))
  910. }
  911. }
  912. for _, container := range srv.daemon.List() {
  913. if !container.State.IsRunning() && !all && n <= 0 && since == "" && before == "" {
  914. continue
  915. }
  916. if before != "" && !foundBefore {
  917. if container.ID == beforeCont.ID {
  918. foundBefore = true
  919. }
  920. continue
  921. }
  922. if n > 0 && displayed == n {
  923. break
  924. }
  925. if since != "" {
  926. if container.ID == sinceCont.ID {
  927. break
  928. }
  929. }
  930. displayed++
  931. out := &engine.Env{}
  932. out.Set("Id", container.ID)
  933. out.SetList("Names", names[container.ID])
  934. out.Set("Image", srv.daemon.Repositories().ImageName(container.Image))
  935. if len(container.Args) > 0 {
  936. args := []string{}
  937. for _, arg := range container.Args {
  938. if strings.Contains(arg, " ") {
  939. args = append(args, fmt.Sprintf("'%s'", arg))
  940. } else {
  941. args = append(args, arg)
  942. }
  943. }
  944. argsAsString := strings.Join(args, " ")
  945. out.Set("Command", fmt.Sprintf("\"%s %s\"", container.Path, argsAsString))
  946. } else {
  947. out.Set("Command", fmt.Sprintf("\"%s\"", container.Path))
  948. }
  949. out.SetInt64("Created", container.Created.Unix())
  950. out.Set("Status", container.State.String())
  951. str, err := container.NetworkSettings.PortMappingAPI().ToListString()
  952. if err != nil {
  953. return job.Error(err)
  954. }
  955. out.Set("Ports", str)
  956. if size {
  957. sizeRw, sizeRootFs := container.GetSize()
  958. out.SetInt64("SizeRw", sizeRw)
  959. out.SetInt64("SizeRootFs", sizeRootFs)
  960. }
  961. outs.Add(out)
  962. }
  963. outs.ReverseSort()
  964. if _, err := outs.WriteListTo(job.Stdout); err != nil {
  965. return job.Error(err)
  966. }
  967. return engine.StatusOK
  968. }
  969. func (srv *Server) ContainerCommit(job *engine.Job) engine.Status {
  970. if len(job.Args) != 1 {
  971. return job.Errorf("Not enough arguments. Usage: %s CONTAINER\n", job.Name)
  972. }
  973. name := job.Args[0]
  974. container := srv.daemon.Get(name)
  975. if container == nil {
  976. return job.Errorf("No such container: %s", name)
  977. }
  978. var (
  979. config = container.Config
  980. newConfig runconfig.Config
  981. )
  982. if err := job.GetenvJson("config", &newConfig); err != nil {
  983. return job.Error(err)
  984. }
  985. if err := runconfig.Merge(&newConfig, config); err != nil {
  986. return job.Error(err)
  987. }
  988. img, err := srv.daemon.Commit(container, job.Getenv("repo"), job.Getenv("tag"), job.Getenv("comment"), job.Getenv("author"), &newConfig)
  989. if err != nil {
  990. return job.Error(err)
  991. }
  992. job.Printf("%s\n", img.ID)
  993. return engine.StatusOK
  994. }
  995. func (srv *Server) ImageTag(job *engine.Job) engine.Status {
  996. if len(job.Args) != 2 && len(job.Args) != 3 {
  997. return job.Errorf("Usage: %s IMAGE REPOSITORY [TAG]\n", job.Name)
  998. }
  999. var tag string
  1000. if len(job.Args) == 3 {
  1001. tag = job.Args[2]
  1002. }
  1003. if err := srv.daemon.Repositories().Set(job.Args[1], tag, job.Args[0], job.GetenvBool("force")); err != nil {
  1004. return job.Error(err)
  1005. }
  1006. return engine.StatusOK
  1007. }
  1008. func (srv *Server) pullImage(r *registry.Registry, out io.Writer, imgID, endpoint string, token []string, sf *utils.StreamFormatter) error {
  1009. history, err := r.GetRemoteHistory(imgID, endpoint, token)
  1010. if err != nil {
  1011. return err
  1012. }
  1013. out.Write(sf.FormatProgress(utils.TruncateID(imgID), "Pulling dependent layers", nil))
  1014. // FIXME: Try to stream the images?
  1015. // FIXME: Launch the getRemoteImage() in goroutines
  1016. for i := len(history) - 1; i >= 0; i-- {
  1017. id := history[i]
  1018. // ensure no two downloads of the same layer happen at the same time
  1019. if c, err := srv.poolAdd("pull", "layer:"+id); err != nil {
  1020. utils.Errorf("Image (id: %s) pull is already running, skipping: %v", id, err)
  1021. <-c
  1022. }
  1023. defer srv.poolRemove("pull", "layer:"+id)
  1024. if !srv.daemon.Graph().Exists(id) {
  1025. out.Write(sf.FormatProgress(utils.TruncateID(id), "Pulling metadata", nil))
  1026. var (
  1027. imgJSON []byte
  1028. imgSize int
  1029. err error
  1030. img *image.Image
  1031. )
  1032. retries := 5
  1033. for j := 1; j <= retries; j++ {
  1034. imgJSON, imgSize, err = r.GetRemoteImageJSON(id, endpoint, token)
  1035. if err != nil && j == retries {
  1036. out.Write(sf.FormatProgress(utils.TruncateID(id), "Error pulling dependent layers", nil))
  1037. return err
  1038. } else if err != nil {
  1039. time.Sleep(time.Duration(j) * 500 * time.Millisecond)
  1040. continue
  1041. }
  1042. img, err = image.NewImgJSON(imgJSON)
  1043. if err != nil && j == retries {
  1044. out.Write(sf.FormatProgress(utils.TruncateID(id), "Error pulling dependent layers", nil))
  1045. return fmt.Errorf("Failed to parse json: %s", err)
  1046. } else if err != nil {
  1047. time.Sleep(time.Duration(j) * 500 * time.Millisecond)
  1048. continue
  1049. } else {
  1050. break
  1051. }
  1052. }
  1053. for j := 1; j <= retries; j++ {
  1054. // Get the layer
  1055. status := "Pulling fs layer"
  1056. if j > 1 {
  1057. status = fmt.Sprintf("Pulling fs layer [retries: %d]", j)
  1058. }
  1059. out.Write(sf.FormatProgress(utils.TruncateID(id), status, nil))
  1060. layer, err := r.GetRemoteImageLayer(img.ID, endpoint, token)
  1061. if uerr, ok := err.(*url.Error); ok {
  1062. err = uerr.Err
  1063. }
  1064. if terr, ok := err.(net.Error); ok && terr.Timeout() && j < retries {
  1065. time.Sleep(time.Duration(j) * 500 * time.Millisecond)
  1066. continue
  1067. } else if err != nil {
  1068. out.Write(sf.FormatProgress(utils.TruncateID(id), "Error pulling dependent layers", nil))
  1069. return err
  1070. }
  1071. defer layer.Close()
  1072. err = srv.daemon.Graph().Register(imgJSON,
  1073. utils.ProgressReader(layer, imgSize, out, sf, false, utils.TruncateID(id), "Downloading"),
  1074. img)
  1075. if terr, ok := err.(net.Error); ok && terr.Timeout() && j < retries {
  1076. time.Sleep(time.Duration(j) * 500 * time.Millisecond)
  1077. continue
  1078. } else if err != nil {
  1079. out.Write(sf.FormatProgress(utils.TruncateID(id), "Error downloading dependent layers", nil))
  1080. return err
  1081. } else {
  1082. break
  1083. }
  1084. }
  1085. }
  1086. out.Write(sf.FormatProgress(utils.TruncateID(id), "Download complete", nil))
  1087. }
  1088. return nil
  1089. }
  1090. func (srv *Server) pullRepository(r *registry.Registry, out io.Writer, localName, remoteName, askedTag string, sf *utils.StreamFormatter, parallel bool) error {
  1091. out.Write(sf.FormatStatus("", "Pulling repository %s", localName))
  1092. repoData, err := r.GetRepositoryData(remoteName)
  1093. if err != nil {
  1094. return err
  1095. }
  1096. utils.Debugf("Retrieving the tag list")
  1097. tagsList, err := r.GetRemoteTags(repoData.Endpoints, remoteName, repoData.Tokens)
  1098. if err != nil {
  1099. utils.Errorf("%v", err)
  1100. return err
  1101. }
  1102. for tag, id := range tagsList {
  1103. repoData.ImgList[id] = &registry.ImgData{
  1104. ID: id,
  1105. Tag: tag,
  1106. Checksum: "",
  1107. }
  1108. }
  1109. utils.Debugf("Registering tags")
  1110. // If no tag has been specified, pull them all
  1111. if askedTag == "" {
  1112. for tag, id := range tagsList {
  1113. repoData.ImgList[id].Tag = tag
  1114. }
  1115. } else {
  1116. // Otherwise, check that the tag exists and use only that one
  1117. id, exists := tagsList[askedTag]
  1118. if !exists {
  1119. return fmt.Errorf("Tag %s not found in repository %s", askedTag, localName)
  1120. }
  1121. repoData.ImgList[id].Tag = askedTag
  1122. }
  1123. errors := make(chan error)
  1124. for _, image := range repoData.ImgList {
  1125. downloadImage := func(img *registry.ImgData) {
  1126. if askedTag != "" && img.Tag != askedTag {
  1127. utils.Debugf("(%s) does not match %s (id: %s), skipping", img.Tag, askedTag, img.ID)
  1128. if parallel {
  1129. errors <- nil
  1130. }
  1131. return
  1132. }
  1133. if img.Tag == "" {
  1134. utils.Debugf("Image (id: %s) present in this repository but untagged, skipping", img.ID)
  1135. if parallel {
  1136. errors <- nil
  1137. }
  1138. return
  1139. }
  1140. // ensure no two downloads of the same image happen at the same time
  1141. if c, err := srv.poolAdd("pull", "img:"+img.ID); err != nil {
  1142. if c != nil {
  1143. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Layer already being pulled by another client. Waiting.", nil))
  1144. <-c
  1145. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Download complete", nil))
  1146. } else {
  1147. utils.Errorf("Image (id: %s) pull is already running, skipping: %v", img.ID, err)
  1148. }
  1149. if parallel {
  1150. errors <- nil
  1151. }
  1152. return
  1153. }
  1154. defer srv.poolRemove("pull", "img:"+img.ID)
  1155. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), fmt.Sprintf("Pulling image (%s) from %s", img.Tag, localName), nil))
  1156. success := false
  1157. var lastErr error
  1158. for _, ep := range repoData.Endpoints {
  1159. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), fmt.Sprintf("Pulling image (%s) from %s, endpoint: %s", img.Tag, localName, ep), nil))
  1160. if err := srv.pullImage(r, out, img.ID, ep, repoData.Tokens, sf); err != nil {
  1161. // It's not ideal that only the last error is returned, it would be better to concatenate the errors.
  1162. // As the error is also given to the output stream the user will see the error.
  1163. lastErr = err
  1164. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), fmt.Sprintf("Error pulling image (%s) from %s, endpoint: %s, %s", img.Tag, localName, ep, err), nil))
  1165. continue
  1166. }
  1167. success = true
  1168. break
  1169. }
  1170. if !success {
  1171. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), fmt.Sprintf("Error pulling image (%s) from %s, %s", img.Tag, localName, lastErr), nil))
  1172. if parallel {
  1173. errors <- fmt.Errorf("Could not find repository on any of the indexed registries.")
  1174. return
  1175. }
  1176. }
  1177. out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Download complete", nil))
  1178. if parallel {
  1179. errors <- nil
  1180. }
  1181. }
  1182. if parallel {
  1183. go downloadImage(image)
  1184. } else {
  1185. downloadImage(image)
  1186. }
  1187. }
  1188. if parallel {
  1189. var lastError error
  1190. for i := 0; i < len(repoData.ImgList); i++ {
  1191. if err := <-errors; err != nil {
  1192. lastError = err
  1193. }
  1194. }
  1195. if lastError != nil {
  1196. return lastError
  1197. }
  1198. }
  1199. for tag, id := range tagsList {
  1200. if askedTag != "" && tag != askedTag {
  1201. continue
  1202. }
  1203. if err := srv.daemon.Repositories().Set(localName, tag, id, true); err != nil {
  1204. return err
  1205. }
  1206. }
  1207. return nil
  1208. }
  1209. func (srv *Server) poolAdd(kind, key string) (chan struct{}, error) {
  1210. srv.Lock()
  1211. defer srv.Unlock()
  1212. if c, exists := srv.pullingPool[key]; exists {
  1213. return c, fmt.Errorf("pull %s is already in progress", key)
  1214. }
  1215. if c, exists := srv.pushingPool[key]; exists {
  1216. return c, fmt.Errorf("push %s is already in progress", key)
  1217. }
  1218. c := make(chan struct{})
  1219. switch kind {
  1220. case "pull":
  1221. srv.pullingPool[key] = c
  1222. case "push":
  1223. srv.pushingPool[key] = c
  1224. default:
  1225. return nil, fmt.Errorf("Unknown pool type")
  1226. }
  1227. return c, nil
  1228. }
  1229. func (srv *Server) poolRemove(kind, key string) error {
  1230. srv.Lock()
  1231. defer srv.Unlock()
  1232. switch kind {
  1233. case "pull":
  1234. if c, exists := srv.pullingPool[key]; exists {
  1235. close(c)
  1236. delete(srv.pullingPool, key)
  1237. }
  1238. case "push":
  1239. if c, exists := srv.pushingPool[key]; exists {
  1240. close(c)
  1241. delete(srv.pushingPool, key)
  1242. }
  1243. default:
  1244. return fmt.Errorf("Unknown pool type")
  1245. }
  1246. return nil
  1247. }
  1248. func (srv *Server) ImagePull(job *engine.Job) engine.Status {
  1249. if n := len(job.Args); n != 1 && n != 2 {
  1250. return job.Errorf("Usage: %s IMAGE [TAG]", job.Name)
  1251. }
  1252. var (
  1253. localName = job.Args[0]
  1254. tag string
  1255. sf = utils.NewStreamFormatter(job.GetenvBool("json"))
  1256. authConfig = &registry.AuthConfig{}
  1257. metaHeaders map[string][]string
  1258. )
  1259. if len(job.Args) > 1 {
  1260. tag = job.Args[1]
  1261. }
  1262. job.GetenvJson("authConfig", authConfig)
  1263. job.GetenvJson("metaHeaders", metaHeaders)
  1264. c, err := srv.poolAdd("pull", localName+":"+tag)
  1265. if err != nil {
  1266. if c != nil {
  1267. // Another pull of the same repository is already taking place; just wait for it to finish
  1268. job.Stdout.Write(sf.FormatStatus("", "Repository %s already being pulled by another client. Waiting.", localName))
  1269. <-c
  1270. return engine.StatusOK
  1271. }
  1272. return job.Error(err)
  1273. }
  1274. defer srv.poolRemove("pull", localName+":"+tag)
  1275. // Resolve the Repository name from fqn to endpoint + name
  1276. hostname, remoteName, err := registry.ResolveRepositoryName(localName)
  1277. if err != nil {
  1278. return job.Error(err)
  1279. }
  1280. endpoint, err := registry.ExpandAndVerifyRegistryUrl(hostname)
  1281. if err != nil {
  1282. return job.Error(err)
  1283. }
  1284. r, err := registry.NewRegistry(authConfig, registry.HTTPRequestFactory(metaHeaders), endpoint)
  1285. if err != nil {
  1286. return job.Error(err)
  1287. }
  1288. if endpoint == registry.IndexServerAddress() {
  1289. // If pull "index.docker.io/foo/bar", it's stored locally under "foo/bar"
  1290. localName = remoteName
  1291. }
  1292. if err = srv.pullRepository(r, job.Stdout, localName, remoteName, tag, sf, job.GetenvBool("parallel")); err != nil {
  1293. return job.Error(err)
  1294. }
  1295. return engine.StatusOK
  1296. }
  1297. // Retrieve the all the images to be uploaded in the correct order
  1298. func (srv *Server) getImageList(localRepo map[string]string, requestedTag string) ([]string, map[string][]string, error) {
  1299. var (
  1300. imageList []string
  1301. imagesSeen map[string]bool = make(map[string]bool)
  1302. tagsByImage map[string][]string = make(map[string][]string)
  1303. )
  1304. for tag, id := range localRepo {
  1305. if requestedTag != "" && requestedTag != tag {
  1306. continue
  1307. }
  1308. var imageListForThisTag []string
  1309. tagsByImage[id] = append(tagsByImage[id], tag)
  1310. for img, err := srv.daemon.Graph().Get(id); img != nil; img, err = img.GetParent() {
  1311. if err != nil {
  1312. return nil, nil, err
  1313. }
  1314. if imagesSeen[img.ID] {
  1315. // This image is already on the list, we can ignore it and all its parents
  1316. break
  1317. }
  1318. imagesSeen[img.ID] = true
  1319. imageListForThisTag = append(imageListForThisTag, img.ID)
  1320. }
  1321. // reverse the image list for this tag (so the "most"-parent image is first)
  1322. for i, j := 0, len(imageListForThisTag)-1; i < j; i, j = i+1, j-1 {
  1323. imageListForThisTag[i], imageListForThisTag[j] = imageListForThisTag[j], imageListForThisTag[i]
  1324. }
  1325. // append to main image list
  1326. imageList = append(imageList, imageListForThisTag...)
  1327. }
  1328. if len(imageList) == 0 {
  1329. return nil, nil, fmt.Errorf("No images found for the requested repository / tag")
  1330. }
  1331. utils.Debugf("Image list: %v", imageList)
  1332. utils.Debugf("Tags by image: %v", tagsByImage)
  1333. return imageList, tagsByImage, nil
  1334. }
  1335. func (srv *Server) pushRepository(r *registry.Registry, out io.Writer, localName, remoteName string, localRepo map[string]string, tag string, sf *utils.StreamFormatter) error {
  1336. out = utils.NewWriteFlusher(out)
  1337. utils.Debugf("Local repo: %s", localRepo)
  1338. imgList, tagsByImage, err := srv.getImageList(localRepo, tag)
  1339. if err != nil {
  1340. return err
  1341. }
  1342. out.Write(sf.FormatStatus("", "Sending image list"))
  1343. var (
  1344. repoData *registry.RepositoryData
  1345. imageIndex []*registry.ImgData
  1346. )
  1347. for _, imgId := range imgList {
  1348. if tags, exists := tagsByImage[imgId]; exists {
  1349. // If an image has tags you must add an entry in the image index
  1350. // for each tag
  1351. for _, tag := range tags {
  1352. imageIndex = append(imageIndex, &registry.ImgData{
  1353. ID: imgId,
  1354. Tag: tag,
  1355. })
  1356. }
  1357. } else {
  1358. // If the image does not have a tag it still needs to be sent to the
  1359. // registry with an empty tag so that it is accociated with the repository
  1360. imageIndex = append(imageIndex, &registry.ImgData{
  1361. ID: imgId,
  1362. Tag: "",
  1363. })
  1364. }
  1365. }
  1366. utils.Debugf("Preparing to push %s with the following images and tags\n", localRepo)
  1367. for _, data := range imageIndex {
  1368. utils.Debugf("Pushing ID: %s with Tag: %s\n", data.ID, data.Tag)
  1369. }
  1370. // Register all the images in a repository with the registry
  1371. // If an image is not in this list it will not be associated with the repository
  1372. repoData, err = r.PushImageJSONIndex(remoteName, imageIndex, false, nil)
  1373. if err != nil {
  1374. return err
  1375. }
  1376. nTag := 1
  1377. if tag == "" {
  1378. nTag = len(localRepo)
  1379. }
  1380. for _, ep := range repoData.Endpoints {
  1381. out.Write(sf.FormatStatus("", "Pushing repository %s (%d tags)", localName, nTag))
  1382. for _, imgId := range imgList {
  1383. if r.LookupRemoteImage(imgId, ep, repoData.Tokens) {
  1384. out.Write(sf.FormatStatus("", "Image %s already pushed, skipping", utils.TruncateID(imgId)))
  1385. } else {
  1386. if _, err := srv.pushImage(r, out, remoteName, imgId, ep, repoData.Tokens, sf); err != nil {
  1387. // FIXME: Continue on error?
  1388. return err
  1389. }
  1390. }
  1391. for _, tag := range tagsByImage[imgId] {
  1392. out.Write(sf.FormatStatus("", "Pushing tag for rev [%s] on {%s}", utils.TruncateID(imgId), ep+"repositories/"+remoteName+"/tags/"+tag))
  1393. if err := r.PushRegistryTag(remoteName, imgId, tag, ep, repoData.Tokens); err != nil {
  1394. return err
  1395. }
  1396. }
  1397. }
  1398. }
  1399. if _, err := r.PushImageJSONIndex(remoteName, imageIndex, true, repoData.Endpoints); err != nil {
  1400. return err
  1401. }
  1402. return nil
  1403. }
  1404. func (srv *Server) pushImage(r *registry.Registry, out io.Writer, remote, imgID, ep string, token []string, sf *utils.StreamFormatter) (checksum string, err error) {
  1405. out = utils.NewWriteFlusher(out)
  1406. jsonRaw, err := ioutil.ReadFile(path.Join(srv.daemon.Graph().Root, imgID, "json"))
  1407. if err != nil {
  1408. return "", fmt.Errorf("Cannot retrieve the path for {%s}: %s", imgID, err)
  1409. }
  1410. out.Write(sf.FormatProgress(utils.TruncateID(imgID), "Pushing", nil))
  1411. imgData := &registry.ImgData{
  1412. ID: imgID,
  1413. }
  1414. // Send the json
  1415. if err := r.PushImageJSONRegistry(imgData, jsonRaw, ep, token); err != nil {
  1416. if err == registry.ErrAlreadyExists {
  1417. out.Write(sf.FormatProgress(utils.TruncateID(imgData.ID), "Image already pushed, skipping", nil))
  1418. return "", nil
  1419. }
  1420. return "", err
  1421. }
  1422. layerData, err := srv.daemon.Graph().TempLayerArchive(imgID, archive.Uncompressed, sf, out)
  1423. if err != nil {
  1424. return "", fmt.Errorf("Failed to generate layer archive: %s", err)
  1425. }
  1426. defer os.RemoveAll(layerData.Name())
  1427. // Send the layer
  1428. utils.Debugf("rendered layer for %s of [%d] size", imgData.ID, layerData.Size)
  1429. checksum, checksumPayload, err := r.PushImageLayerRegistry(imgData.ID, utils.ProgressReader(layerData, int(layerData.Size), out, sf, false, utils.TruncateID(imgData.ID), "Pushing"), ep, token, jsonRaw)
  1430. if err != nil {
  1431. return "", err
  1432. }
  1433. imgData.Checksum = checksum
  1434. imgData.ChecksumPayload = checksumPayload
  1435. // Send the checksum
  1436. if err := r.PushImageChecksumRegistry(imgData, ep, token); err != nil {
  1437. return "", err
  1438. }
  1439. out.Write(sf.FormatProgress(utils.TruncateID(imgData.ID), "Image successfully pushed", nil))
  1440. return imgData.Checksum, nil
  1441. }
  1442. // FIXME: Allow to interrupt current push when new push of same image is done.
  1443. func (srv *Server) ImagePush(job *engine.Job) engine.Status {
  1444. if n := len(job.Args); n != 1 {
  1445. return job.Errorf("Usage: %s IMAGE", job.Name)
  1446. }
  1447. var (
  1448. localName = job.Args[0]
  1449. sf = utils.NewStreamFormatter(job.GetenvBool("json"))
  1450. authConfig = &registry.AuthConfig{}
  1451. metaHeaders map[string][]string
  1452. )
  1453. tag := job.Getenv("tag")
  1454. job.GetenvJson("authConfig", authConfig)
  1455. job.GetenvJson("metaHeaders", metaHeaders)
  1456. if _, err := srv.poolAdd("push", localName); err != nil {
  1457. return job.Error(err)
  1458. }
  1459. defer srv.poolRemove("push", localName)
  1460. // Resolve the Repository name from fqn to endpoint + name
  1461. hostname, remoteName, err := registry.ResolveRepositoryName(localName)
  1462. if err != nil {
  1463. return job.Error(err)
  1464. }
  1465. endpoint, err := registry.ExpandAndVerifyRegistryUrl(hostname)
  1466. if err != nil {
  1467. return job.Error(err)
  1468. }
  1469. img, err := srv.daemon.Graph().Get(localName)
  1470. r, err2 := registry.NewRegistry(authConfig, registry.HTTPRequestFactory(metaHeaders), endpoint)
  1471. if err2 != nil {
  1472. return job.Error(err2)
  1473. }
  1474. if err != nil {
  1475. reposLen := 1
  1476. if tag == "" {
  1477. reposLen = len(srv.daemon.Repositories().Repositories[localName])
  1478. }
  1479. job.Stdout.Write(sf.FormatStatus("", "The push refers to a repository [%s] (len: %d)", localName, reposLen))
  1480. // If it fails, try to get the repository
  1481. if localRepo, exists := srv.daemon.Repositories().Repositories[localName]; exists {
  1482. if err := srv.pushRepository(r, job.Stdout, localName, remoteName, localRepo, tag, sf); err != nil {
  1483. return job.Error(err)
  1484. }
  1485. return engine.StatusOK
  1486. }
  1487. return job.Error(err)
  1488. }
  1489. var token []string
  1490. job.Stdout.Write(sf.FormatStatus("", "The push refers to an image: [%s]", localName))
  1491. if _, err := srv.pushImage(r, job.Stdout, remoteName, img.ID, endpoint, token, sf); err != nil {
  1492. return job.Error(err)
  1493. }
  1494. return engine.StatusOK
  1495. }
  1496. func (srv *Server) ImageImport(job *engine.Job) engine.Status {
  1497. if n := len(job.Args); n != 2 && n != 3 {
  1498. return job.Errorf("Usage: %s SRC REPO [TAG]", job.Name)
  1499. }
  1500. var (
  1501. src = job.Args[0]
  1502. repo = job.Args[1]
  1503. tag string
  1504. sf = utils.NewStreamFormatter(job.GetenvBool("json"))
  1505. archive archive.ArchiveReader
  1506. resp *http.Response
  1507. )
  1508. if len(job.Args) > 2 {
  1509. tag = job.Args[2]
  1510. }
  1511. if src == "-" {
  1512. archive = job.Stdin
  1513. } else {
  1514. u, err := url.Parse(src)
  1515. if err != nil {
  1516. return job.Error(err)
  1517. }
  1518. if u.Scheme == "" {
  1519. u.Scheme = "http"
  1520. u.Host = src
  1521. u.Path = ""
  1522. }
  1523. job.Stdout.Write(sf.FormatStatus("", "Downloading from %s", u))
  1524. // Download with curl (pretty progress bar)
  1525. // If curl is not available, fallback to http.Get()
  1526. resp, err = utils.Download(u.String())
  1527. if err != nil {
  1528. return job.Error(err)
  1529. }
  1530. progressReader := utils.ProgressReader(resp.Body, int(resp.ContentLength), job.Stdout, sf, true, "", "Importing")
  1531. defer progressReader.Close()
  1532. archive = progressReader
  1533. }
  1534. img, err := srv.daemon.Graph().Create(archive, "", "", "Imported from "+src, "", nil, nil)
  1535. if err != nil {
  1536. return job.Error(err)
  1537. }
  1538. // Optionally register the image at REPO/TAG
  1539. if repo != "" {
  1540. if err := srv.daemon.Repositories().Set(repo, tag, img.ID, true); err != nil {
  1541. return job.Error(err)
  1542. }
  1543. }
  1544. job.Stdout.Write(sf.FormatStatus("", img.ID))
  1545. return engine.StatusOK
  1546. }
  1547. func (srv *Server) ContainerCreate(job *engine.Job) engine.Status {
  1548. var name string
  1549. if len(job.Args) == 1 {
  1550. name = job.Args[0]
  1551. } else if len(job.Args) > 1 {
  1552. return job.Errorf("Usage: %s", job.Name)
  1553. }
  1554. config := runconfig.ContainerConfigFromJob(job)
  1555. if config.Memory != 0 && config.Memory < 524288 {
  1556. return job.Errorf("Minimum memory limit allowed is 512k")
  1557. }
  1558. if config.Memory > 0 && !srv.daemon.SystemConfig().MemoryLimit {
  1559. job.Errorf("Your kernel does not support memory limit capabilities. Limitation discarded.\n")
  1560. config.Memory = 0
  1561. }
  1562. if config.Memory > 0 && !srv.daemon.SystemConfig().SwapLimit {
  1563. job.Errorf("Your kernel does not support swap limit capabilities. Limitation discarded.\n")
  1564. config.MemorySwap = -1
  1565. }
  1566. container, buildWarnings, err := srv.daemon.Create(config, name)
  1567. if err != nil {
  1568. if srv.daemon.Graph().IsNotExist(err) {
  1569. _, tag := utils.ParseRepositoryTag(config.Image)
  1570. if tag == "" {
  1571. tag = graph.DEFAULTTAG
  1572. }
  1573. return job.Errorf("No such image: %s (tag: %s)", config.Image, tag)
  1574. }
  1575. return job.Error(err)
  1576. }
  1577. if !container.Config.NetworkDisabled && srv.daemon.SystemConfig().IPv4ForwardingDisabled {
  1578. job.Errorf("IPv4 forwarding is disabled.\n")
  1579. }
  1580. srv.LogEvent("create", container.ID, srv.daemon.Repositories().ImageName(container.Image))
  1581. // FIXME: this is necessary because daemon.Create might return a nil container
  1582. // with a non-nil error. This should not happen! Once it's fixed we
  1583. // can remove this workaround.
  1584. if container != nil {
  1585. job.Printf("%s\n", container.ID)
  1586. }
  1587. for _, warning := range buildWarnings {
  1588. job.Errorf("%s\n", warning)
  1589. }
  1590. return engine.StatusOK
  1591. }
  1592. func (srv *Server) ContainerRestart(job *engine.Job) engine.Status {
  1593. if len(job.Args) != 1 {
  1594. return job.Errorf("Usage: %s CONTAINER\n", job.Name)
  1595. }
  1596. var (
  1597. name = job.Args[0]
  1598. t = 10
  1599. )
  1600. if job.EnvExists("t") {
  1601. t = job.GetenvInt("t")
  1602. }
  1603. if container := srv.daemon.Get(name); container != nil {
  1604. if err := container.Restart(int(t)); err != nil {
  1605. return job.Errorf("Cannot restart container %s: %s\n", name, err)
  1606. }
  1607. srv.LogEvent("restart", container.ID, srv.daemon.Repositories().ImageName(container.Image))
  1608. } else {
  1609. return job.Errorf("No such container: %s\n", name)
  1610. }
  1611. return engine.StatusOK
  1612. }
  1613. func (srv *Server) ContainerDestroy(job *engine.Job) engine.Status {
  1614. if len(job.Args) != 1 {
  1615. return job.Errorf("Not enough arguments. Usage: %s CONTAINER\n", job.Name)
  1616. }
  1617. name := job.Args[0]
  1618. removeVolume := job.GetenvBool("removeVolume")
  1619. removeLink := job.GetenvBool("removeLink")
  1620. forceRemove := job.GetenvBool("forceRemove")
  1621. container := srv.daemon.Get(name)
  1622. if removeLink {
  1623. if container == nil {
  1624. return job.Errorf("No such link: %s", name)
  1625. }
  1626. name, err := daemon.GetFullContainerName(name)
  1627. if err != nil {
  1628. job.Error(err)
  1629. }
  1630. parent, n := path.Split(name)
  1631. if parent == "/" {
  1632. return job.Errorf("Conflict, cannot remove the default name of the container")
  1633. }
  1634. pe := srv.daemon.ContainerGraph().Get(parent)
  1635. if pe == nil {
  1636. return job.Errorf("Cannot get parent %s for name %s", parent, name)
  1637. }
  1638. parentContainer := srv.daemon.Get(pe.ID())
  1639. if parentContainer != nil {
  1640. parentContainer.DisableLink(n)
  1641. }
  1642. if err := srv.daemon.ContainerGraph().Delete(name); err != nil {
  1643. return job.Error(err)
  1644. }
  1645. return engine.StatusOK
  1646. }
  1647. if container != nil {
  1648. if container.State.IsRunning() {
  1649. if forceRemove {
  1650. if err := container.Stop(5); err != nil {
  1651. return job.Errorf("Could not stop running container, cannot remove - %v", err)
  1652. }
  1653. } else {
  1654. return job.Errorf("Impossible to remove a running container, please stop it first or use -f")
  1655. }
  1656. }
  1657. if err := srv.daemon.Destroy(container); err != nil {
  1658. return job.Errorf("Cannot destroy container %s: %s", name, err)
  1659. }
  1660. srv.LogEvent("destroy", container.ID, srv.daemon.Repositories().ImageName(container.Image))
  1661. if removeVolume {
  1662. var (
  1663. volumes = make(map[string]struct{})
  1664. binds = make(map[string]struct{})
  1665. usedVolumes = make(map[string]*daemon.Container)
  1666. )
  1667. // the volume id is always the base of the path
  1668. getVolumeId := func(p string) string {
  1669. return filepath.Base(strings.TrimSuffix(p, "/layer"))
  1670. }
  1671. // populate bind map so that they can be skipped and not removed
  1672. for _, bind := range container.HostConfig().Binds {
  1673. source := strings.Split(bind, ":")[0]
  1674. // TODO: refactor all volume stuff, all of it
  1675. // it is very important that we eval the link or comparing the keys to container.Volumes will not work
  1676. //
  1677. // eval symlink can fail, ref #5244 if we receive an is not exist error we can ignore it
  1678. p, err := filepath.EvalSymlinks(source)
  1679. if err != nil && !os.IsNotExist(err) {
  1680. return job.Error(err)
  1681. }
  1682. if p != "" {
  1683. source = p
  1684. }
  1685. binds[source] = struct{}{}
  1686. }
  1687. // Store all the deleted containers volumes
  1688. for _, volumeId := range container.Volumes {
  1689. // Skip the volumes mounted from external
  1690. // bind mounts here will will be evaluated for a symlink
  1691. if _, exists := binds[volumeId]; exists {
  1692. continue
  1693. }
  1694. volumeId = getVolumeId(volumeId)
  1695. volumes[volumeId] = struct{}{}
  1696. }
  1697. // Retrieve all volumes from all remaining containers
  1698. for _, container := range srv.daemon.List() {
  1699. for _, containerVolumeId := range container.Volumes {
  1700. containerVolumeId = getVolumeId(containerVolumeId)
  1701. usedVolumes[containerVolumeId] = container
  1702. }
  1703. }
  1704. for volumeId := range volumes {
  1705. // If the requested volu
  1706. if c, exists := usedVolumes[volumeId]; exists {
  1707. log.Printf("The volume %s is used by the container %s. Impossible to remove it. Skipping.\n", volumeId, c.ID)
  1708. continue
  1709. }
  1710. if err := srv.daemon.Volumes().Delete(volumeId); err != nil {
  1711. return job.Errorf("Error calling volumes.Delete(%q): %v", volumeId, err)
  1712. }
  1713. }
  1714. }
  1715. } else {
  1716. return job.Errorf("No such container: %s", name)
  1717. }
  1718. return engine.StatusOK
  1719. }
  1720. func (srv *Server) DeleteImage(name string, imgs *engine.Table, first, force, noprune bool) error {
  1721. var (
  1722. repoName, tag string
  1723. tags = []string{}
  1724. tagDeleted bool
  1725. )
  1726. repoName, tag = utils.ParseRepositoryTag(name)
  1727. if tag == "" {
  1728. tag = graph.DEFAULTTAG
  1729. }
  1730. img, err := srv.daemon.Repositories().LookupImage(name)
  1731. if err != nil {
  1732. if r, _ := srv.daemon.Repositories().Get(repoName); r != nil {
  1733. return fmt.Errorf("No such image: %s:%s", repoName, tag)
  1734. }
  1735. return fmt.Errorf("No such image: %s", name)
  1736. }
  1737. if strings.Contains(img.ID, name) {
  1738. repoName = ""
  1739. tag = ""
  1740. }
  1741. byParents, err := srv.daemon.Graph().ByParent()
  1742. if err != nil {
  1743. return err
  1744. }
  1745. //If delete by id, see if the id belong only to one repository
  1746. if repoName == "" {
  1747. for _, repoAndTag := range srv.daemon.Repositories().ByID()[img.ID] {
  1748. parsedRepo, parsedTag := utils.ParseRepositoryTag(repoAndTag)
  1749. if repoName == "" || repoName == parsedRepo {
  1750. repoName = parsedRepo
  1751. if parsedTag != "" {
  1752. tags = append(tags, parsedTag)
  1753. }
  1754. } else if repoName != parsedRepo && !force {
  1755. // the id belongs to multiple repos, like base:latest and user:test,
  1756. // in that case return conflict
  1757. return fmt.Errorf("Conflict, cannot delete image %s because it is tagged in multiple repositories, use -f to force", name)
  1758. }
  1759. }
  1760. } else {
  1761. tags = append(tags, tag)
  1762. }
  1763. if !first && len(tags) > 0 {
  1764. return nil
  1765. }
  1766. //Untag the current image
  1767. for _, tag := range tags {
  1768. tagDeleted, err = srv.daemon.Repositories().Delete(repoName, tag)
  1769. if err != nil {
  1770. return err
  1771. }
  1772. if tagDeleted {
  1773. out := &engine.Env{}
  1774. out.Set("Untagged", repoName+":"+tag)
  1775. imgs.Add(out)
  1776. srv.LogEvent("untag", img.ID, "")
  1777. }
  1778. }
  1779. tags = srv.daemon.Repositories().ByID()[img.ID]
  1780. if (len(tags) <= 1 && repoName == "") || len(tags) == 0 {
  1781. if len(byParents[img.ID]) == 0 {
  1782. if err := srv.canDeleteImage(img.ID, force, tagDeleted); err != nil {
  1783. return err
  1784. }
  1785. if err := srv.daemon.Repositories().DeleteAll(img.ID); err != nil {
  1786. return err
  1787. }
  1788. if err := srv.daemon.Graph().Delete(img.ID); err != nil {
  1789. return err
  1790. }
  1791. out := &engine.Env{}
  1792. out.Set("Deleted", img.ID)
  1793. imgs.Add(out)
  1794. srv.LogEvent("delete", img.ID, "")
  1795. if img.Parent != "" && !noprune {
  1796. err := srv.DeleteImage(img.Parent, imgs, false, force, noprune)
  1797. if first {
  1798. return err
  1799. }
  1800. }
  1801. }
  1802. }
  1803. return nil
  1804. }
  1805. func (srv *Server) ImageDelete(job *engine.Job) engine.Status {
  1806. if n := len(job.Args); n != 1 {
  1807. return job.Errorf("Usage: %s IMAGE", job.Name)
  1808. }
  1809. imgs := engine.NewTable("", 0)
  1810. if err := srv.DeleteImage(job.Args[0], imgs, true, job.GetenvBool("force"), job.GetenvBool("noprune")); err != nil {
  1811. return job.Error(err)
  1812. }
  1813. if len(imgs.Data) == 0 {
  1814. return job.Errorf("Conflict, %s wasn't deleted", job.Args[0])
  1815. }
  1816. if _, err := imgs.WriteListTo(job.Stdout); err != nil {
  1817. return job.Error(err)
  1818. }
  1819. return engine.StatusOK
  1820. }
  1821. func (srv *Server) canDeleteImage(imgID string, force, untagged bool) error {
  1822. var message string
  1823. if untagged {
  1824. message = " (docker untagged the image)"
  1825. }
  1826. for _, container := range srv.daemon.List() {
  1827. parent, err := srv.daemon.Repositories().LookupImage(container.Image)
  1828. if err != nil {
  1829. return err
  1830. }
  1831. if err := parent.WalkHistory(func(p *image.Image) error {
  1832. if imgID == p.ID {
  1833. if container.State.IsRunning() {
  1834. if force {
  1835. return fmt.Errorf("Conflict, cannot force delete %s because the running container %s is using it%s, stop it and retry", utils.TruncateID(imgID), utils.TruncateID(container.ID), message)
  1836. }
  1837. return fmt.Errorf("Conflict, cannot delete %s because the running container %s is using it%s, stop it and use -f to force", utils.TruncateID(imgID), utils.TruncateID(container.ID), message)
  1838. } else if !force {
  1839. return fmt.Errorf("Conflict, cannot delete %s because the container %s is using it%s, use -f to force", utils.TruncateID(imgID), utils.TruncateID(container.ID), message)
  1840. }
  1841. }
  1842. return nil
  1843. }); err != nil {
  1844. return err
  1845. }
  1846. }
  1847. return nil
  1848. }
  1849. func (srv *Server) ImageGetCached(imgID string, config *runconfig.Config) (*image.Image, error) {
  1850. // Retrieve all images
  1851. images, err := srv.daemon.Graph().Map()
  1852. if err != nil {
  1853. return nil, err
  1854. }
  1855. // Store the tree in a map of map (map[parentId][childId])
  1856. imageMap := make(map[string]map[string]struct{})
  1857. for _, img := range images {
  1858. if _, exists := imageMap[img.Parent]; !exists {
  1859. imageMap[img.Parent] = make(map[string]struct{})
  1860. }
  1861. imageMap[img.Parent][img.ID] = struct{}{}
  1862. }
  1863. // Loop on the children of the given image and check the config
  1864. var match *image.Image
  1865. for elem := range imageMap[imgID] {
  1866. img, err := srv.daemon.Graph().Get(elem)
  1867. if err != nil {
  1868. return nil, err
  1869. }
  1870. if runconfig.Compare(&img.ContainerConfig, config) {
  1871. if match == nil || match.Created.Before(img.Created) {
  1872. match = img
  1873. }
  1874. }
  1875. }
  1876. return match, nil
  1877. }
  1878. func (srv *Server) ContainerStart(job *engine.Job) engine.Status {
  1879. if len(job.Args) < 1 {
  1880. return job.Errorf("Usage: %s container_id", job.Name)
  1881. }
  1882. var (
  1883. name = job.Args[0]
  1884. daemon = srv.daemon
  1885. container = daemon.Get(name)
  1886. )
  1887. if container == nil {
  1888. return job.Errorf("No such container: %s", name)
  1889. }
  1890. // If no environment was set, then no hostconfig was passed.
  1891. if len(job.Environ()) > 0 {
  1892. hostConfig := runconfig.ContainerHostConfigFromJob(job)
  1893. // Validate the HostConfig binds. Make sure that:
  1894. // 1) the source of a bind mount isn't /
  1895. // The bind mount "/:/foo" isn't allowed.
  1896. // 2) Check that the source exists
  1897. // The source to be bind mounted must exist.
  1898. for _, bind := range hostConfig.Binds {
  1899. splitBind := strings.Split(bind, ":")
  1900. source := splitBind[0]
  1901. // refuse to bind mount "/" to the container
  1902. if source == "/" {
  1903. return job.Errorf("Invalid bind mount '%s' : source can't be '/'", bind)
  1904. }
  1905. // ensure the source exists on the host
  1906. _, err := os.Stat(source)
  1907. if err != nil && os.IsNotExist(err) {
  1908. err = os.MkdirAll(source, 0755)
  1909. if err != nil {
  1910. return job.Errorf("Could not create local directory '%s' for bind mount: %s!", source, err.Error())
  1911. }
  1912. }
  1913. }
  1914. // Register any links from the host config before starting the container
  1915. if err := srv.daemon.RegisterLinks(container, hostConfig); err != nil {
  1916. return job.Error(err)
  1917. }
  1918. container.SetHostConfig(hostConfig)
  1919. container.ToDisk()
  1920. }
  1921. if err := container.Start(); err != nil {
  1922. return job.Errorf("Cannot start container %s: %s", name, err)
  1923. }
  1924. srv.LogEvent("start", container.ID, daemon.Repositories().ImageName(container.Image))
  1925. return engine.StatusOK
  1926. }
  1927. func (srv *Server) ContainerStop(job *engine.Job) engine.Status {
  1928. if len(job.Args) != 1 {
  1929. return job.Errorf("Usage: %s CONTAINER\n", job.Name)
  1930. }
  1931. var (
  1932. name = job.Args[0]
  1933. t = 10
  1934. )
  1935. if job.EnvExists("t") {
  1936. t = job.GetenvInt("t")
  1937. }
  1938. if container := srv.daemon.Get(name); container != nil {
  1939. if err := container.Stop(int(t)); err != nil {
  1940. return job.Errorf("Cannot stop container %s: %s\n", name, err)
  1941. }
  1942. srv.LogEvent("stop", container.ID, srv.daemon.Repositories().ImageName(container.Image))
  1943. } else {
  1944. return job.Errorf("No such container: %s\n", name)
  1945. }
  1946. return engine.StatusOK
  1947. }
  1948. func (srv *Server) ContainerWait(job *engine.Job) engine.Status {
  1949. if len(job.Args) != 1 {
  1950. return job.Errorf("Usage: %s", job.Name)
  1951. }
  1952. name := job.Args[0]
  1953. if container := srv.daemon.Get(name); container != nil {
  1954. status := container.Wait()
  1955. job.Printf("%d\n", status)
  1956. return engine.StatusOK
  1957. }
  1958. return job.Errorf("%s: no such container: %s", job.Name, name)
  1959. }
  1960. func (srv *Server) ContainerResize(job *engine.Job) engine.Status {
  1961. if len(job.Args) != 3 {
  1962. return job.Errorf("Not enough arguments. Usage: %s CONTAINER HEIGHT WIDTH\n", job.Name)
  1963. }
  1964. name := job.Args[0]
  1965. height, err := strconv.Atoi(job.Args[1])
  1966. if err != nil {
  1967. return job.Error(err)
  1968. }
  1969. width, err := strconv.Atoi(job.Args[2])
  1970. if err != nil {
  1971. return job.Error(err)
  1972. }
  1973. if container := srv.daemon.Get(name); container != nil {
  1974. if err := container.Resize(height, width); err != nil {
  1975. return job.Error(err)
  1976. }
  1977. return engine.StatusOK
  1978. }
  1979. return job.Errorf("No such container: %s", name)
  1980. }
  1981. func (srv *Server) ContainerLogs(job *engine.Job) engine.Status {
  1982. if len(job.Args) != 1 {
  1983. return job.Errorf("Usage: %s CONTAINER\n", job.Name)
  1984. }
  1985. var (
  1986. name = job.Args[0]
  1987. stdout = job.GetenvBool("stdout")
  1988. stderr = job.GetenvBool("stderr")
  1989. follow = job.GetenvBool("follow")
  1990. times = job.GetenvBool("timestamps")
  1991. format string
  1992. )
  1993. if !(stdout || stderr) {
  1994. return job.Errorf("You must choose at least one stream")
  1995. }
  1996. if times {
  1997. format = time.StampMilli
  1998. }
  1999. container := srv.daemon.Get(name)
  2000. if container == nil {
  2001. return job.Errorf("No such container: %s", name)
  2002. }
  2003. cLog, err := container.ReadLog("json")
  2004. if err != nil && os.IsNotExist(err) {
  2005. // Legacy logs
  2006. utils.Debugf("Old logs format")
  2007. if stdout {
  2008. cLog, err := container.ReadLog("stdout")
  2009. if err != nil {
  2010. utils.Errorf("Error reading logs (stdout): %s", err)
  2011. } else if _, err := io.Copy(job.Stdout, cLog); err != nil {
  2012. utils.Errorf("Error streaming logs (stdout): %s", err)
  2013. }
  2014. }
  2015. if stderr {
  2016. cLog, err := container.ReadLog("stderr")
  2017. if err != nil {
  2018. utils.Errorf("Error reading logs (stderr): %s", err)
  2019. } else if _, err := io.Copy(job.Stderr, cLog); err != nil {
  2020. utils.Errorf("Error streaming logs (stderr): %s", err)
  2021. }
  2022. }
  2023. } else if err != nil {
  2024. utils.Errorf("Error reading logs (json): %s", err)
  2025. } else {
  2026. dec := json.NewDecoder(cLog)
  2027. for {
  2028. l := &utils.JSONLog{}
  2029. if err := dec.Decode(l); err == io.EOF {
  2030. break
  2031. } else if err != nil {
  2032. utils.Errorf("Error streaming logs: %s", err)
  2033. break
  2034. }
  2035. logLine := l.Log
  2036. if times {
  2037. logLine = fmt.Sprintf("[%s] %s", l.Created.Format(format), logLine)
  2038. }
  2039. if l.Stream == "stdout" && stdout {
  2040. fmt.Fprintf(job.Stdout, "%s", logLine)
  2041. }
  2042. if l.Stream == "stderr" && stderr {
  2043. fmt.Fprintf(job.Stderr, "%s", logLine)
  2044. }
  2045. }
  2046. }
  2047. if follow {
  2048. errors := make(chan error, 2)
  2049. if stdout {
  2050. stdoutPipe := container.StdoutLogPipe()
  2051. go func() {
  2052. errors <- utils.WriteLog(stdoutPipe, job.Stdout, format)
  2053. }()
  2054. }
  2055. if stderr {
  2056. stderrPipe := container.StderrLogPipe()
  2057. go func() {
  2058. errors <- utils.WriteLog(stderrPipe, job.Stderr, format)
  2059. }()
  2060. }
  2061. err := <-errors
  2062. if err != nil {
  2063. utils.Errorf("%s", err)
  2064. }
  2065. }
  2066. return engine.StatusOK
  2067. }
  2068. func (srv *Server) ContainerAttach(job *engine.Job) engine.Status {
  2069. if len(job.Args) != 1 {
  2070. return job.Errorf("Usage: %s CONTAINER\n", job.Name)
  2071. }
  2072. var (
  2073. name = job.Args[0]
  2074. logs = job.GetenvBool("logs")
  2075. stream = job.GetenvBool("stream")
  2076. stdin = job.GetenvBool("stdin")
  2077. stdout = job.GetenvBool("stdout")
  2078. stderr = job.GetenvBool("stderr")
  2079. )
  2080. container := srv.daemon.Get(name)
  2081. if container == nil {
  2082. return job.Errorf("No such container: %s", name)
  2083. }
  2084. //logs
  2085. if logs {
  2086. cLog, err := container.ReadLog("json")
  2087. if err != nil && os.IsNotExist(err) {
  2088. // Legacy logs
  2089. utils.Debugf("Old logs format")
  2090. if stdout {
  2091. cLog, err := container.ReadLog("stdout")
  2092. if err != nil {
  2093. utils.Errorf("Error reading logs (stdout): %s", err)
  2094. } else if _, err := io.Copy(job.Stdout, cLog); err != nil {
  2095. utils.Errorf("Error streaming logs (stdout): %s", err)
  2096. }
  2097. }
  2098. if stderr {
  2099. cLog, err := container.ReadLog("stderr")
  2100. if err != nil {
  2101. utils.Errorf("Error reading logs (stderr): %s", err)
  2102. } else if _, err := io.Copy(job.Stderr, cLog); err != nil {
  2103. utils.Errorf("Error streaming logs (stderr): %s", err)
  2104. }
  2105. }
  2106. } else if err != nil {
  2107. utils.Errorf("Error reading logs (json): %s", err)
  2108. } else {
  2109. dec := json.NewDecoder(cLog)
  2110. for {
  2111. l := &utils.JSONLog{}
  2112. if err := dec.Decode(l); err == io.EOF {
  2113. break
  2114. } else if err != nil {
  2115. utils.Errorf("Error streaming logs: %s", err)
  2116. break
  2117. }
  2118. if l.Stream == "stdout" && stdout {
  2119. fmt.Fprintf(job.Stdout, "%s", l.Log)
  2120. }
  2121. if l.Stream == "stderr" && stderr {
  2122. fmt.Fprintf(job.Stderr, "%s", l.Log)
  2123. }
  2124. }
  2125. }
  2126. }
  2127. //stream
  2128. if stream {
  2129. var (
  2130. cStdin io.ReadCloser
  2131. cStdout, cStderr io.Writer
  2132. cStdinCloser io.Closer
  2133. )
  2134. if stdin {
  2135. r, w := io.Pipe()
  2136. go func() {
  2137. defer w.Close()
  2138. defer utils.Debugf("Closing buffered stdin pipe")
  2139. io.Copy(w, job.Stdin)
  2140. }()
  2141. cStdin = r
  2142. cStdinCloser = job.Stdin
  2143. }
  2144. if stdout {
  2145. cStdout = job.Stdout
  2146. }
  2147. if stderr {
  2148. cStderr = job.Stderr
  2149. }
  2150. <-srv.daemon.Attach(container, cStdin, cStdinCloser, cStdout, cStderr)
  2151. // If we are in stdinonce mode, wait for the process to end
  2152. // otherwise, simply return
  2153. if container.Config.StdinOnce && !container.Config.Tty {
  2154. container.Wait()
  2155. }
  2156. }
  2157. return engine.StatusOK
  2158. }
  2159. func (srv *Server) ContainerCopy(job *engine.Job) engine.Status {
  2160. if len(job.Args) != 2 {
  2161. return job.Errorf("Usage: %s CONTAINER RESOURCE\n", job.Name)
  2162. }
  2163. var (
  2164. name = job.Args[0]
  2165. resource = job.Args[1]
  2166. )
  2167. if container := srv.daemon.Get(name); container != nil {
  2168. data, err := container.Copy(resource)
  2169. if err != nil {
  2170. return job.Error(err)
  2171. }
  2172. defer data.Close()
  2173. if _, err := io.Copy(job.Stdout, data); err != nil {
  2174. return job.Error(err)
  2175. }
  2176. return engine.StatusOK
  2177. }
  2178. return job.Errorf("No such container: %s", name)
  2179. }
  2180. func NewServer(eng *engine.Engine, config *daemonconfig.Config) (*Server, error) {
  2181. daemon, err := daemon.NewDaemon(config, eng)
  2182. if err != nil {
  2183. return nil, err
  2184. }
  2185. srv := &Server{
  2186. Eng: eng,
  2187. daemon: daemon,
  2188. pullingPool: make(map[string]chan struct{}),
  2189. pushingPool: make(map[string]chan struct{}),
  2190. events: make([]utils.JSONMessage, 0, 64), //only keeps the 64 last events
  2191. listeners: make(map[int64]chan utils.JSONMessage),
  2192. }
  2193. daemon.SetServer(srv)
  2194. return srv, nil
  2195. }
  2196. func (srv *Server) LogEvent(action, id, from string) *utils.JSONMessage {
  2197. now := time.Now().UTC().Unix()
  2198. jm := utils.JSONMessage{Status: action, ID: id, From: from, Time: now}
  2199. srv.AddEvent(jm)
  2200. for _, c := range srv.listeners {
  2201. select { // non blocking channel
  2202. case c <- jm:
  2203. default:
  2204. }
  2205. }
  2206. return &jm
  2207. }
  2208. func (srv *Server) AddEvent(jm utils.JSONMessage) {
  2209. srv.Lock()
  2210. defer srv.Unlock()
  2211. srv.events = append(srv.events, jm)
  2212. }
  2213. func (srv *Server) GetEvents() []utils.JSONMessage {
  2214. srv.RLock()
  2215. defer srv.RUnlock()
  2216. return srv.events
  2217. }
  2218. func (srv *Server) SetRunning(status bool) {
  2219. srv.Lock()
  2220. defer srv.Unlock()
  2221. srv.running = status
  2222. }
  2223. func (srv *Server) IsRunning() bool {
  2224. srv.RLock()
  2225. defer srv.RUnlock()
  2226. return srv.running
  2227. }
  2228. func (srv *Server) Close() error {
  2229. if srv == nil {
  2230. return nil
  2231. }
  2232. srv.SetRunning(false)
  2233. done := make(chan struct{})
  2234. go func() {
  2235. srv.tasks.Wait()
  2236. close(done)
  2237. }()
  2238. select {
  2239. // Waiting server jobs for 15 seconds, shutdown immediately after that time
  2240. case <-time.After(time.Second * 15):
  2241. case <-done:
  2242. }
  2243. if srv.daemon == nil {
  2244. return nil
  2245. }
  2246. return srv.daemon.Close()
  2247. }
  2248. type Server struct {
  2249. sync.RWMutex
  2250. daemon *daemon.Daemon
  2251. pullingPool map[string]chan struct{}
  2252. pushingPool map[string]chan struct{}
  2253. events []utils.JSONMessage
  2254. listeners map[int64]chan utils.JSONMessage
  2255. Eng *engine.Engine
  2256. running bool
  2257. tasks sync.WaitGroup
  2258. }