server.go 67 KB

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