server.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959
  1. package server
  2. import (
  3. ".."
  4. "../fs"
  5. "../future"
  6. "../rcli"
  7. "bufio"
  8. "bytes"
  9. "encoding/json"
  10. "errors"
  11. "fmt"
  12. "io"
  13. "net/http"
  14. "net/url"
  15. "os"
  16. "path"
  17. "strconv"
  18. "strings"
  19. "sync"
  20. "text/tabwriter"
  21. "time"
  22. )
  23. const VERSION = "0.0.1"
  24. func (srv *Server) ListenAndServe() error {
  25. go rcli.ListenAndServeHTTP("127.0.0.1:8080", srv)
  26. // FIXME: we want to use unix sockets here, but net.UnixConn doesn't expose
  27. // CloseWrite(), which we need to cleanly signal that stdin is closed without
  28. // closing the connection.
  29. // See http://code.google.com/p/go/issues/detail?id=3345
  30. return rcli.ListenAndServe("tcp", "127.0.0.1:4242", srv)
  31. }
  32. func (srv *Server) Name() string {
  33. return "docker"
  34. }
  35. func (srv *Server) Help() string {
  36. help := "Usage: docker COMMAND [arg...]\n\nA self-sufficient runtime for linux containers.\n\nCommands:\n"
  37. for _, cmd := range [][]interface{}{
  38. {"run", "Run a command in a container"},
  39. {"ps", "Display a list of containers"},
  40. {"import", "Create a new filesystem image from the contents of a tarball"},
  41. {"port", "Lookup the public-facing port which is NAT-ed to PRIVATE_PORT"},
  42. {"rm", "Remove containers"},
  43. {"kill", "Kill a running container"},
  44. {"wait", "Wait for the state of a container to change"},
  45. {"stop", "Stop a running container"},
  46. {"start", "Start a stopped container"},
  47. {"restart", "Restart a running container"},
  48. {"logs", "Fetch the logs of a container"},
  49. {"diff", "Inspect changes on a container's filesystem"},
  50. {"commit", "Save the state of a container"},
  51. {"attach", "Attach to the standard inputs and outputs of a running container"},
  52. {"wait", "Block until a container exits, then print its exit code"},
  53. {"info", "Display system-wide information"},
  54. {"tar", "Stream the contents of a container as a tar archive"},
  55. {"web", "Generate a web UI"},
  56. {"images", "List images"},
  57. } {
  58. help += fmt.Sprintf(" %-10.10s%s\n", cmd...)
  59. }
  60. return help
  61. }
  62. // 'docker wait': block until a container stops
  63. func (srv *Server) CmdWait(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  64. cmd := rcli.Subcmd(stdout, "wait", "[OPTIONS] NAME", "Block until a container stops, then print its exit code.")
  65. if err := cmd.Parse(args); err != nil {
  66. cmd.Usage()
  67. return nil
  68. }
  69. if cmd.NArg() < 1 {
  70. cmd.Usage()
  71. return nil
  72. }
  73. for _, name := range cmd.Args() {
  74. if container := srv.containers.Get(name); container != nil {
  75. fmt.Fprintln(stdout, container.Wait())
  76. } else {
  77. return errors.New("No such container: " + name)
  78. }
  79. }
  80. return nil
  81. }
  82. // 'docker info': display system-wide information.
  83. func (srv *Server) CmdInfo(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  84. images, _ := srv.images.Images()
  85. var imgcount int
  86. if images == nil {
  87. imgcount = 0
  88. } else {
  89. imgcount = len(images)
  90. }
  91. fmt.Fprintf(stdout, "containers: %d\nversion: %s\nimages: %d\n",
  92. len(srv.containers.List()),
  93. VERSION,
  94. imgcount)
  95. return nil
  96. }
  97. func (srv *Server) CmdStop(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  98. cmd := rcli.Subcmd(stdout, "stop", "[OPTIONS] NAME", "Stop a running container")
  99. if err := cmd.Parse(args); err != nil {
  100. cmd.Usage()
  101. return nil
  102. }
  103. if cmd.NArg() < 1 {
  104. cmd.Usage()
  105. return nil
  106. }
  107. for _, name := range cmd.Args() {
  108. if container := srv.containers.Get(name); container != nil {
  109. if err := container.Stop(); err != nil {
  110. return err
  111. }
  112. fmt.Fprintln(stdout, container.Id)
  113. } else {
  114. return errors.New("No such container: " + name)
  115. }
  116. }
  117. return nil
  118. }
  119. func (srv *Server) CmdRestart(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  120. cmd := rcli.Subcmd(stdout, "restart", "[OPTIONS] NAME", "Restart a running container")
  121. if err := cmd.Parse(args); err != nil {
  122. cmd.Usage()
  123. return nil
  124. }
  125. if cmd.NArg() < 1 {
  126. cmd.Usage()
  127. return nil
  128. }
  129. for _, name := range cmd.Args() {
  130. if container := srv.containers.Get(name); container != nil {
  131. if err := container.Restart(); err != nil {
  132. return err
  133. }
  134. fmt.Fprintln(stdout, container.Id)
  135. } else {
  136. return errors.New("No such container: " + name)
  137. }
  138. }
  139. return nil
  140. }
  141. func (srv *Server) CmdStart(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  142. cmd := rcli.Subcmd(stdout, "start", "[OPTIONS] NAME", "Start a stopped container")
  143. if err := cmd.Parse(args); err != nil {
  144. cmd.Usage()
  145. return nil
  146. }
  147. if cmd.NArg() < 1 {
  148. cmd.Usage()
  149. return nil
  150. }
  151. for _, name := range cmd.Args() {
  152. if container := srv.containers.Get(name); container != nil {
  153. if err := container.Start(); err != nil {
  154. return err
  155. }
  156. fmt.Fprintln(stdout, container.Id)
  157. } else {
  158. return errors.New("No such container: " + name)
  159. }
  160. }
  161. return nil
  162. }
  163. func (srv *Server) CmdUmount(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  164. cmd := rcli.Subcmd(stdout, "umount", "[OPTIONS] NAME", "umount a container's filesystem (debug only)")
  165. if err := cmd.Parse(args); err != nil {
  166. cmd.Usage()
  167. return nil
  168. }
  169. if cmd.NArg() < 1 {
  170. cmd.Usage()
  171. return nil
  172. }
  173. for _, name := range cmd.Args() {
  174. if container := srv.containers.Get(name); container != nil {
  175. if err := container.Mountpoint.Umount(); err != nil {
  176. return err
  177. }
  178. fmt.Fprintln(stdout, container.Id)
  179. } else {
  180. return errors.New("No such container: " + name)
  181. }
  182. }
  183. return nil
  184. }
  185. func (srv *Server) CmdMount(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  186. cmd := rcli.Subcmd(stdout, "umount", "[OPTIONS] NAME", "mount a container's filesystem (debug only)")
  187. if err := cmd.Parse(args); err != nil {
  188. cmd.Usage()
  189. return nil
  190. }
  191. if cmd.NArg() < 1 {
  192. cmd.Usage()
  193. return nil
  194. }
  195. for _, name := range cmd.Args() {
  196. if container := srv.containers.Get(name); container != nil {
  197. if err := container.Mountpoint.EnsureMounted(); err != nil {
  198. return err
  199. }
  200. fmt.Fprintln(stdout, container.Id)
  201. } else {
  202. return errors.New("No such container: " + name)
  203. }
  204. }
  205. return nil
  206. }
  207. func (srv *Server) CmdCat(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  208. cmd := rcli.Subcmd(stdout, "cat", "[OPTIONS] CONTAINER PATH", "write the contents of a container's file to standard output")
  209. if err := cmd.Parse(args); err != nil {
  210. cmd.Usage()
  211. return nil
  212. }
  213. if cmd.NArg() < 2 {
  214. cmd.Usage()
  215. return nil
  216. }
  217. name, path := cmd.Arg(0), cmd.Arg(1)
  218. if container := srv.containers.Get(name); container != nil {
  219. if f, err := container.Mountpoint.OpenFile(path, os.O_RDONLY, 0); err != nil {
  220. return err
  221. } else if _, err := io.Copy(stdout, f); err != nil {
  222. return err
  223. }
  224. return nil
  225. }
  226. return errors.New("No such container: " + name)
  227. }
  228. func (srv *Server) CmdWrite(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  229. cmd := rcli.Subcmd(stdout, "write", "[OPTIONS] CONTAINER PATH", "write the contents of standard input to a container's file")
  230. if err := cmd.Parse(args); err != nil {
  231. cmd.Usage()
  232. return nil
  233. }
  234. if cmd.NArg() < 2 {
  235. cmd.Usage()
  236. return nil
  237. }
  238. name, path := cmd.Arg(0), cmd.Arg(1)
  239. if container := srv.containers.Get(name); container != nil {
  240. if f, err := container.Mountpoint.OpenFile(path, os.O_WRONLY|os.O_CREATE, 0600); err != nil {
  241. return err
  242. } else if _, err := io.Copy(f, stdin); err != nil {
  243. return err
  244. }
  245. return nil
  246. }
  247. return errors.New("No such container: " + name)
  248. }
  249. func (srv *Server) CmdLs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  250. cmd := rcli.Subcmd(stdout, "ls", "[OPTIONS] CONTAINER PATH", "List the contents of a container's directory")
  251. if err := cmd.Parse(args); err != nil {
  252. cmd.Usage()
  253. return nil
  254. }
  255. if cmd.NArg() < 2 {
  256. cmd.Usage()
  257. return nil
  258. }
  259. name, path := cmd.Arg(0), cmd.Arg(1)
  260. if container := srv.containers.Get(name); container != nil {
  261. if files, err := container.Mountpoint.ReadDir(path); err != nil {
  262. return err
  263. } else {
  264. for _, f := range files {
  265. fmt.Fprintln(stdout, f.Name())
  266. }
  267. }
  268. return nil
  269. }
  270. return errors.New("No such container: " + name)
  271. }
  272. func (srv *Server) CmdInspect(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  273. cmd := rcli.Subcmd(stdout, "inspect", "[OPTIONS] CONTAINER", "Return low-level information on a container")
  274. if err := cmd.Parse(args); err != nil {
  275. cmd.Usage()
  276. return nil
  277. }
  278. if cmd.NArg() < 1 {
  279. cmd.Usage()
  280. return nil
  281. }
  282. name := cmd.Arg(0)
  283. var obj interface{}
  284. if container := srv.containers.Get(name); container != nil {
  285. obj = container
  286. //} else if image, err := srv.images.List(name); image != nil {
  287. // obj = image
  288. } else {
  289. return errors.New("No such container or image: " + name)
  290. }
  291. data, err := json.Marshal(obj)
  292. if err != nil {
  293. return err
  294. }
  295. indented := new(bytes.Buffer)
  296. if err = json.Indent(indented, data, "", " "); err != nil {
  297. return err
  298. }
  299. if _, err := io.Copy(stdout, indented); err != nil {
  300. return err
  301. }
  302. return nil
  303. }
  304. func (srv *Server) CmdPort(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  305. cmd := rcli.Subcmd(stdout, "port", "[OPTIONS] CONTAINER PRIVATE_PORT", "Lookup the public-facing port which is NAT-ed to PRIVATE_PORT")
  306. if err := cmd.Parse(args); err != nil {
  307. cmd.Usage()
  308. return nil
  309. }
  310. if cmd.NArg() != 2 {
  311. cmd.Usage()
  312. return nil
  313. }
  314. name := cmd.Arg(0)
  315. privatePort := cmd.Arg(1)
  316. if container := srv.containers.Get(name); container == nil {
  317. return errors.New("No such container: " + name)
  318. } else {
  319. if frontend, exists := container.NetworkSettings.PortMapping[privatePort]; !exists {
  320. return fmt.Errorf("No private port '%s' allocated on %s", privatePort, name)
  321. } else {
  322. fmt.Fprintln(stdout, frontend)
  323. }
  324. }
  325. return nil
  326. }
  327. // 'docker rmi NAME' removes all images with the name NAME
  328. // func (srv *Server) CmdRmi(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  329. // cmd := rcli.Subcmd(stdout, "rmimage", "[OPTIONS] IMAGE", "Remove an image")
  330. // fl_regexp := cmd.Bool("r", false, "Use IMAGE as a regular expression instead of an exact name")
  331. // if err := cmd.Parse(args); err != nil {
  332. // cmd.Usage()
  333. // return nil
  334. // }
  335. // if cmd.NArg() < 1 {
  336. // cmd.Usage()
  337. // return nil
  338. // }
  339. // for _, name := range cmd.Args() {
  340. // var err error
  341. // if *fl_regexp {
  342. // err = srv.images.DeleteMatch(name)
  343. // } else {
  344. // image := srv.images.Find(name)
  345. // if image == nil {
  346. // return errors.New("No such image: " + name)
  347. // }
  348. // err = srv.images.Delete(name)
  349. // }
  350. // if err != nil {
  351. // return err
  352. // }
  353. // }
  354. // return nil
  355. // }
  356. func (srv *Server) CmdRm(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  357. cmd := rcli.Subcmd(stdout, "rm", "[OPTIONS] CONTAINER", "Remove a container")
  358. if err := cmd.Parse(args); err != nil {
  359. return nil
  360. }
  361. for _, name := range cmd.Args() {
  362. container := srv.containers.Get(name)
  363. if container == nil {
  364. return errors.New("No such container: " + name)
  365. }
  366. if err := srv.containers.Destroy(container); err != nil {
  367. fmt.Fprintln(stdout, "Error destroying container "+name+": "+err.Error())
  368. }
  369. }
  370. return nil
  371. }
  372. // 'docker kill NAME' kills a running container
  373. func (srv *Server) CmdKill(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  374. cmd := rcli.Subcmd(stdout, "kill", "[OPTIONS] CONTAINER [CONTAINER...]", "Kill a running container")
  375. if err := cmd.Parse(args); err != nil {
  376. return nil
  377. }
  378. for _, name := range cmd.Args() {
  379. container := srv.containers.Get(name)
  380. if container == nil {
  381. return errors.New("No such container: " + name)
  382. }
  383. if err := container.Kill(); err != nil {
  384. fmt.Fprintln(stdout, "Error killing container "+name+": "+err.Error())
  385. }
  386. }
  387. return nil
  388. }
  389. func (srv *Server) CmdImport(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  390. cmd := rcli.Subcmd(stdout, "import", "[OPTIONS] NAME", "Create a new filesystem image from the contents of a tarball")
  391. fl_stdin := cmd.Bool("stdin", false, "Read tarball from stdin")
  392. if err := cmd.Parse(args); err != nil {
  393. return nil
  394. }
  395. var archive io.Reader
  396. name := cmd.Arg(0)
  397. if name == "" {
  398. return errors.New("Not enough arguments")
  399. }
  400. if *fl_stdin {
  401. archive = stdin
  402. } else {
  403. u, err := url.Parse(name)
  404. if err != nil {
  405. return err
  406. }
  407. if u.Scheme == "" {
  408. u.Scheme = "http"
  409. }
  410. // FIXME: hardcode a mirror URL that does not depend on a single provider.
  411. if u.Host == "" {
  412. u.Host = "s3.amazonaws.com"
  413. u.Path = path.Join("/docker.io/images", u.Path)
  414. }
  415. fmt.Fprintf(stdout, "Downloading from %s\n", u.String())
  416. // Download with curl (pretty progress bar)
  417. // If curl is not available, fallback to http.Get()
  418. archive, err = future.Curl(u.String(), stdout)
  419. if err != nil {
  420. if resp, err := http.Get(u.String()); err != nil {
  421. return err
  422. } else {
  423. archive = resp.Body
  424. }
  425. }
  426. }
  427. fmt.Fprintf(stdout, "Unpacking to %s\n", name)
  428. img, err := srv.images.Create(archive, nil, name, "")
  429. if err != nil {
  430. return err
  431. }
  432. fmt.Fprintln(stdout, img.Id)
  433. return nil
  434. }
  435. func (srv *Server) CmdImages(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  436. cmd := rcli.Subcmd(stdout, "images", "[OPTIONS] [NAME]", "List images")
  437. limit := cmd.Int("l", 0, "Only show the N most recent versions of each image")
  438. quiet := cmd.Bool("q", false, "only show numeric IDs")
  439. cmd.Parse(args)
  440. if cmd.NArg() > 1 {
  441. cmd.Usage()
  442. return nil
  443. }
  444. var nameFilter string
  445. if cmd.NArg() == 1 {
  446. nameFilter = cmd.Arg(0)
  447. }
  448. w := tabwriter.NewWriter(stdout, 20, 1, 3, ' ', 0)
  449. if !*quiet {
  450. fmt.Fprintf(w, "NAME\tID\tCREATED\tPARENT\n")
  451. }
  452. paths, err := srv.images.Paths()
  453. if err != nil {
  454. return err
  455. }
  456. for _, name := range paths {
  457. if nameFilter != "" && nameFilter != name {
  458. continue
  459. }
  460. ids, err := srv.images.List(name)
  461. if err != nil {
  462. return err
  463. }
  464. for idx, img := range ids {
  465. if *limit > 0 && idx >= *limit {
  466. break
  467. }
  468. if !*quiet {
  469. for idx, field := range []string{
  470. /* NAME */ name,
  471. /* ID */ img.Id,
  472. /* CREATED */ future.HumanDuration(time.Now().Sub(time.Unix(img.Created, 0))) + " ago",
  473. /* PARENT */ img.Parent,
  474. } {
  475. if idx == 0 {
  476. w.Write([]byte(field))
  477. } else {
  478. w.Write([]byte("\t" + field))
  479. }
  480. }
  481. w.Write([]byte{'\n'})
  482. } else {
  483. stdout.Write([]byte(img.Id + "\n"))
  484. }
  485. }
  486. }
  487. if !*quiet {
  488. w.Flush()
  489. }
  490. return nil
  491. }
  492. func (srv *Server) CmdPs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  493. cmd := rcli.Subcmd(stdout,
  494. "ps", "[OPTIONS]", "List containers")
  495. quiet := cmd.Bool("q", false, "Only display numeric IDs")
  496. fl_all := cmd.Bool("a", false, "Show all containers. Only running containers are shown by default.")
  497. fl_full := cmd.Bool("notrunc", false, "Don't truncate output")
  498. if err := cmd.Parse(args); err != nil {
  499. return nil
  500. }
  501. w := tabwriter.NewWriter(stdout, 12, 1, 3, ' ', 0)
  502. if !*quiet {
  503. fmt.Fprintf(w, "ID\tIMAGE\tCOMMAND\tCREATED\tSTATUS\tCOMMENT\n")
  504. }
  505. for _, container := range srv.containers.List() {
  506. comment := container.GetUserData("comment")
  507. if !container.State.Running && !*fl_all {
  508. continue
  509. }
  510. if !*quiet {
  511. command := fmt.Sprintf("%s %s", container.Path, strings.Join(container.Args, " "))
  512. if !*fl_full {
  513. command = docker.Trunc(command, 20)
  514. }
  515. for idx, field := range []string{
  516. /* ID */ container.Id,
  517. /* IMAGE */ container.GetUserData("image"),
  518. /* COMMAND */ command,
  519. /* CREATED */ future.HumanDuration(time.Now().Sub(container.Created)) + " ago",
  520. /* STATUS */ container.State.String(),
  521. /* COMMENT */ comment,
  522. } {
  523. if idx == 0 {
  524. w.Write([]byte(field))
  525. } else {
  526. w.Write([]byte("\t" + field))
  527. }
  528. }
  529. w.Write([]byte{'\n'})
  530. } else {
  531. stdout.Write([]byte(container.Id + "\n"))
  532. }
  533. }
  534. if !*quiet {
  535. w.Flush()
  536. }
  537. return nil
  538. }
  539. func (srv *Server) CmdLayers(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  540. cmd := rcli.Subcmd(stdout,
  541. "layers", "[OPTIONS]",
  542. "List filesystem layers (debug only)")
  543. if err := cmd.Parse(args); err != nil {
  544. return nil
  545. }
  546. for _, layer := range srv.images.Layers() {
  547. fmt.Fprintln(stdout, layer)
  548. }
  549. return nil
  550. }
  551. func (srv *Server) CmdCp(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  552. cmd := rcli.Subcmd(stdout,
  553. "cp", "[OPTIONS] IMAGE NAME",
  554. "Create a copy of IMAGE and call it NAME")
  555. if err := cmd.Parse(args); err != nil {
  556. return nil
  557. }
  558. if image, err := srv.images.Get(cmd.Arg(0)); err != nil {
  559. return err
  560. } else if image == nil {
  561. return errors.New("Image " + cmd.Arg(0) + " does not exist")
  562. } else {
  563. if img, err := image.Copy(cmd.Arg(1)); err != nil {
  564. return err
  565. } else {
  566. fmt.Fprintln(stdout, img.Id)
  567. }
  568. }
  569. return nil
  570. }
  571. func (srv *Server) CmdCommit(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  572. cmd := rcli.Subcmd(stdout,
  573. "commit", "[OPTIONS] CONTAINER [DEST]",
  574. "Create a new image from a container's changes")
  575. if err := cmd.Parse(args); err != nil {
  576. return nil
  577. }
  578. containerName, imgName := cmd.Arg(0), cmd.Arg(1)
  579. if containerName == "" || imgName == "" {
  580. cmd.Usage()
  581. return nil
  582. }
  583. if container := srv.containers.Get(containerName); container != nil {
  584. // FIXME: freeze the container before copying it to avoid data corruption?
  585. rwTar, err := fs.Tar(container.Mountpoint.Rw, fs.Uncompressed)
  586. if err != nil {
  587. return err
  588. }
  589. // Create a new image from the container's base layers + a new layer from container changes
  590. parentImg, err := srv.images.Get(container.Image)
  591. if err != nil {
  592. return err
  593. }
  594. img, err := srv.images.Create(rwTar, parentImg, imgName, "")
  595. if err != nil {
  596. return err
  597. }
  598. fmt.Fprintln(stdout, img.Id)
  599. return nil
  600. }
  601. return errors.New("No such container: " + containerName)
  602. }
  603. func (srv *Server) CmdTar(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  604. cmd := rcli.Subcmd(stdout,
  605. "tar", "CONTAINER",
  606. "Stream the contents of a container as a tar archive")
  607. fl_sparse := cmd.Bool("s", false, "Generate a sparse tar stream (top layer + reference to bottom layers)")
  608. if err := cmd.Parse(args); err != nil {
  609. return nil
  610. }
  611. if *fl_sparse {
  612. return errors.New("Sparse mode not yet implemented") // FIXME
  613. }
  614. name := cmd.Arg(0)
  615. if container := srv.containers.Get(name); container != nil {
  616. if err := container.Mountpoint.EnsureMounted(); err != nil {
  617. return err
  618. }
  619. data, err := fs.Tar(container.Mountpoint.Root, fs.Uncompressed)
  620. if err != nil {
  621. return err
  622. }
  623. // Stream the entire contents of the container (basically a volatile snapshot)
  624. if _, err := io.Copy(stdout, data); err != nil {
  625. return err
  626. }
  627. return nil
  628. }
  629. return errors.New("No such container: " + name)
  630. }
  631. func (srv *Server) CmdDiff(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  632. cmd := rcli.Subcmd(stdout,
  633. "diff", "CONTAINER [OPTIONS]",
  634. "Inspect changes on a container's filesystem")
  635. if err := cmd.Parse(args); err != nil {
  636. return nil
  637. }
  638. if cmd.NArg() < 1 {
  639. return errors.New("Not enough arguments")
  640. }
  641. if container := srv.containers.Get(cmd.Arg(0)); container == nil {
  642. return errors.New("No such container")
  643. } else {
  644. changes, err := srv.images.Changes(container.Mountpoint)
  645. if err != nil {
  646. return err
  647. }
  648. for _, change := range changes {
  649. fmt.Fprintln(stdout, change.String())
  650. }
  651. }
  652. return nil
  653. }
  654. func (srv *Server) CmdReset(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  655. cmd := rcli.Subcmd(stdout,
  656. "reset", "CONTAINER [OPTIONS]",
  657. "Reset changes to a container's filesystem")
  658. if err := cmd.Parse(args); err != nil {
  659. return nil
  660. }
  661. if cmd.NArg() < 1 {
  662. return errors.New("Not enough arguments")
  663. }
  664. for _, name := range cmd.Args() {
  665. if container := srv.containers.Get(name); container != nil {
  666. if err := container.Mountpoint.Reset(); err != nil {
  667. return errors.New("Reset " + container.Id + ": " + err.Error())
  668. }
  669. }
  670. }
  671. return nil
  672. }
  673. func (srv *Server) CmdLogs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  674. cmd := rcli.Subcmd(stdout, "logs", "[OPTIONS] CONTAINER", "Fetch the logs of a container")
  675. if err := cmd.Parse(args); err != nil {
  676. return nil
  677. }
  678. if cmd.NArg() != 1 {
  679. cmd.Usage()
  680. return nil
  681. }
  682. name := cmd.Arg(0)
  683. if container := srv.containers.Get(name); container != nil {
  684. if _, err := io.Copy(stdout, container.StdoutLog()); err != nil {
  685. return err
  686. }
  687. if _, err := io.Copy(stdout, container.StderrLog()); err != nil {
  688. return err
  689. }
  690. return nil
  691. }
  692. return errors.New("No such container: " + cmd.Arg(0))
  693. }
  694. func (srv *Server) CreateContainer(img *fs.Image, ports []int, user string, tty bool, openStdin bool, comment string, cmd string, args ...string) (*docker.Container, error) {
  695. id := future.RandomId()[:8]
  696. container, err := srv.containers.Create(id, cmd, args, img,
  697. &docker.Config{Hostname: id, Ports: ports, User: user, Tty: tty, OpenStdin: openStdin})
  698. if err != nil {
  699. return nil, err
  700. }
  701. if err := container.SetUserData("image", img.Id); err != nil {
  702. srv.containers.Destroy(container)
  703. return nil, errors.New("Error setting container userdata: " + err.Error())
  704. }
  705. if err := container.SetUserData("comment", comment); err != nil {
  706. srv.containers.Destroy(container)
  707. return nil, errors.New("Error setting container userdata: " + err.Error())
  708. }
  709. return container, nil
  710. }
  711. func (srv *Server) CmdAttach(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  712. cmd := rcli.Subcmd(stdout, "attach", "[OPTIONS]", "Attach to a running container")
  713. fl_i := cmd.Bool("i", false, "Attach to stdin")
  714. fl_o := cmd.Bool("o", true, "Attach to stdout")
  715. fl_e := cmd.Bool("e", true, "Attach to stderr")
  716. if err := cmd.Parse(args); err != nil {
  717. return nil
  718. }
  719. if cmd.NArg() != 1 {
  720. cmd.Usage()
  721. return nil
  722. }
  723. name := cmd.Arg(0)
  724. container := srv.containers.Get(name)
  725. if container == nil {
  726. return errors.New("No such container: " + name)
  727. }
  728. var wg sync.WaitGroup
  729. if *fl_i {
  730. c_stdin, err := container.StdinPipe()
  731. if err != nil {
  732. return err
  733. }
  734. wg.Add(1)
  735. go func() { io.Copy(c_stdin, stdin); wg.Add(-1) }()
  736. }
  737. if *fl_o {
  738. c_stdout, err := container.StdoutPipe()
  739. if err != nil {
  740. return err
  741. }
  742. wg.Add(1)
  743. go func() { io.Copy(stdout, c_stdout); wg.Add(-1) }()
  744. }
  745. if *fl_e {
  746. c_stderr, err := container.StderrPipe()
  747. if err != nil {
  748. return err
  749. }
  750. wg.Add(1)
  751. go func() { io.Copy(stdout, c_stderr); wg.Add(-1) }()
  752. }
  753. wg.Wait()
  754. return nil
  755. }
  756. // Ports type - Used to parse multiple -p flags
  757. type ports []int
  758. func (p *ports) String() string {
  759. return fmt.Sprint(*p)
  760. }
  761. func (p *ports) Set(value string) error {
  762. port, err := strconv.Atoi(value)
  763. if err != nil {
  764. return fmt.Errorf("Invalid port: %v", value)
  765. }
  766. *p = append(*p, port)
  767. return nil
  768. }
  769. func (srv *Server) CmdRun(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  770. cmd := rcli.Subcmd(stdout, "run", "[OPTIONS] IMAGE COMMAND [ARG...]", "Run a command in a new container")
  771. fl_user := cmd.String("u", "", "Username or UID")
  772. fl_attach := cmd.Bool("a", false, "Attach stdin and stdout")
  773. fl_stdin := cmd.Bool("i", false, "Keep stdin open even if not attached")
  774. fl_tty := cmd.Bool("t", false, "Allocate a pseudo-tty")
  775. fl_comment := cmd.String("c", "", "Comment")
  776. var fl_ports ports
  777. cmd.Var(&fl_ports, "p", "Map a network port to the container")
  778. if err := cmd.Parse(args); err != nil {
  779. return nil
  780. }
  781. name := cmd.Arg(0)
  782. var cmdline []string
  783. if len(cmd.Args()) >= 2 {
  784. cmdline = cmd.Args()[1:]
  785. }
  786. // Choose a default image if needed
  787. if name == "" {
  788. name = "base"
  789. }
  790. // Choose a default command if needed
  791. if len(cmdline) == 0 {
  792. *fl_stdin = true
  793. *fl_tty = true
  794. *fl_attach = true
  795. cmdline = []string{"/bin/bash", "-i"}
  796. }
  797. // Find the image
  798. img, err := srv.images.Find(name)
  799. if err != nil {
  800. return err
  801. } else if img == nil {
  802. return errors.New("No such image: " + name)
  803. }
  804. // Create new container
  805. container, err := srv.CreateContainer(img, fl_ports, *fl_user, *fl_tty, *fl_stdin, *fl_comment, cmdline[0], cmdline[1:]...)
  806. if err != nil {
  807. return errors.New("Error creating container: " + err.Error())
  808. }
  809. if *fl_stdin {
  810. cmd_stdin, err := container.StdinPipe()
  811. if err != nil {
  812. return err
  813. }
  814. if *fl_attach {
  815. future.Go(func() error {
  816. _, err := io.Copy(cmd_stdin, stdin)
  817. cmd_stdin.Close()
  818. return err
  819. })
  820. }
  821. }
  822. // Run the container
  823. if *fl_attach {
  824. cmd_stderr, err := container.StderrPipe()
  825. if err != nil {
  826. return err
  827. }
  828. cmd_stdout, err := container.StdoutPipe()
  829. if err != nil {
  830. return err
  831. }
  832. if err := container.Start(); err != nil {
  833. return err
  834. }
  835. sending_stdout := future.Go(func() error {
  836. _, err := io.Copy(stdout, cmd_stdout)
  837. return err
  838. })
  839. sending_stderr := future.Go(func() error {
  840. _, err := io.Copy(stdout, cmd_stderr)
  841. return err
  842. })
  843. err_sending_stdout := <-sending_stdout
  844. err_sending_stderr := <-sending_stderr
  845. if err_sending_stdout != nil {
  846. return err_sending_stdout
  847. }
  848. if err_sending_stderr != nil {
  849. return err_sending_stderr
  850. }
  851. container.Wait()
  852. } else {
  853. if err := container.Start(); err != nil {
  854. return err
  855. }
  856. fmt.Fprintln(stdout, container.Id)
  857. }
  858. return nil
  859. }
  860. func New() (*Server, error) {
  861. future.Seed()
  862. // if err != nil {
  863. // return nil, err
  864. // }
  865. containers, err := docker.New()
  866. if err != nil {
  867. return nil, err
  868. }
  869. srv := &Server{
  870. images: containers.Store,
  871. containers: containers,
  872. }
  873. return srv, nil
  874. }
  875. func (srv *Server) CmdMirror(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  876. _, err := io.Copy(stdout, stdin)
  877. return err
  878. }
  879. func (srv *Server) CmdDebug(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  880. for {
  881. if line, err := bufio.NewReader(stdin).ReadString('\n'); err == nil {
  882. fmt.Printf("--- %s", line)
  883. } else if err == io.EOF {
  884. if len(line) > 0 {
  885. fmt.Printf("--- %s\n", line)
  886. }
  887. break
  888. } else {
  889. return err
  890. }
  891. }
  892. return nil
  893. }
  894. func (srv *Server) CmdWeb(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  895. cmd := rcli.Subcmd(stdout, "web", "[OPTIONS]", "A web UI for docker")
  896. showurl := cmd.Bool("u", false, "Return the URL of the web UI")
  897. if err := cmd.Parse(args); err != nil {
  898. return nil
  899. }
  900. if *showurl {
  901. fmt.Fprintln(stdout, "http://localhost:4242/web")
  902. } else {
  903. if file, err := os.Open("dockerweb.html"); err != nil {
  904. return err
  905. } else if _, err := io.Copy(stdout, file); err != nil {
  906. return err
  907. }
  908. }
  909. return nil
  910. }
  911. type Server struct {
  912. containers *docker.Docker
  913. images *fs.Store
  914. }