dockerd.go 21 KB

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