commands.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812
  1. package docker
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "github.com/dotcloud/docker/auth"
  8. "github.com/dotcloud/docker/rcli"
  9. "io"
  10. "log"
  11. "math/rand"
  12. "net/http"
  13. "net/url"
  14. "path"
  15. "runtime"
  16. "strconv"
  17. "strings"
  18. "sync"
  19. "text/tabwriter"
  20. "time"
  21. )
  22. const VERSION = "0.0.3"
  23. func (srv *Server) Name() string {
  24. return "docker"
  25. }
  26. // FIXME: Stop violating DRY by repeating usage here and in Subcmd declarations
  27. func (srv *Server) Help() string {
  28. help := "Usage: docker COMMAND [arg...]\n\nA self-sufficient runtime for linux containers.\n\nCommands:\n"
  29. for _, cmd := range [][]interface{}{
  30. {"run", "Run a command in a container"},
  31. {"ps", "Display a list of containers"},
  32. {"import", "Create a new filesystem image from the contents of a tarball"},
  33. {"attach", "Attach to a running container"},
  34. {"commit", "Create a new image from a container's changes"},
  35. {"diff", "Inspect changes on a container's filesystem"},
  36. {"images", "List images"},
  37. {"info", "Display system-wide information"},
  38. {"inspect", "Return low-level information on a container"},
  39. {"kill", "Kill a running container"},
  40. {"login", "Register or Login to the docker registry server"},
  41. {"logs", "Fetch the logs of a container"},
  42. {"port", "Lookup the public-facing port which is NAT-ed to PRIVATE_PORT"},
  43. {"ps", "List containers"},
  44. {"restart", "Restart a running container"},
  45. {"rm", "Remove a container"},
  46. {"rmi", "Remove an image"},
  47. {"run", "Run a command in a new container"},
  48. {"start", "Start a stopped container"},
  49. {"stop", "Stop a running container"},
  50. {"export", "Stream the contents of a container as a tar archive"},
  51. {"version", "Show the docker version information"},
  52. {"wait", "Block until a container stops, then print its exit code"},
  53. } {
  54. help += fmt.Sprintf(" %-10.10s%s\n", cmd[0], cmd[1])
  55. }
  56. return help
  57. }
  58. // 'docker login': login / register a user to registry service.
  59. func (srv *Server) CmdLogin(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  60. cmd := rcli.Subcmd(stdout, "login", "", "Register or Login to the docker registry server")
  61. if err := cmd.Parse(args); err != nil {
  62. return nil
  63. }
  64. var username string
  65. var password string
  66. var email string
  67. authConfig, err := auth.LoadConfig()
  68. if err != nil {
  69. fmt.Fprintf(stdout, "Error : %s\n", err)
  70. }
  71. fmt.Fprint(stdout, "Username (", authConfig.Username, "): ")
  72. fmt.Fscanf(stdin, "%s", &username)
  73. if username == "" {
  74. username = authConfig.Username
  75. }
  76. if username != authConfig.Username {
  77. fmt.Fprint(stdout, "Password: ")
  78. fmt.Fscanf(stdin, "%s", &password)
  79. if password == "" {
  80. return errors.New("Error : Password Required\n")
  81. }
  82. fmt.Fprint(stdout, "Email (", authConfig.Email, "): ")
  83. fmt.Fscanf(stdin, "%s", &email)
  84. if email == "" {
  85. email = authConfig.Email
  86. }
  87. } else {
  88. password = authConfig.Password
  89. email = authConfig.Email
  90. }
  91. newAuthConfig := auth.AuthConfig{Username: username, Password: password, Email: email}
  92. status, err := auth.Login(newAuthConfig)
  93. if err != nil {
  94. fmt.Fprintf(stdout, "Error : %s\n", err)
  95. }
  96. if status != "" {
  97. fmt.Fprintf(stdout, status)
  98. }
  99. return nil
  100. }
  101. // 'docker wait': block until a container stops
  102. func (srv *Server) CmdWait(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  103. cmd := rcli.Subcmd(stdout, "wait", "[OPTIONS] NAME", "Block until a container stops, then print its exit code.")
  104. if err := cmd.Parse(args); err != nil {
  105. return nil
  106. }
  107. if cmd.NArg() < 1 {
  108. cmd.Usage()
  109. return nil
  110. }
  111. for _, name := range cmd.Args() {
  112. if container := srv.runtime.Get(name); container != nil {
  113. fmt.Fprintln(stdout, container.Wait())
  114. } else {
  115. return errors.New("No such container: " + name)
  116. }
  117. }
  118. return nil
  119. }
  120. // 'docker version': show version information
  121. func (srv *Server) CmdVersion(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  122. fmt.Fprintf(stdout, "Version:%s\n", VERSION)
  123. return nil
  124. }
  125. // 'docker info': display system-wide information.
  126. func (srv *Server) CmdInfo(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  127. images, _ := srv.runtime.graph.All()
  128. var imgcount int
  129. if images == nil {
  130. imgcount = 0
  131. } else {
  132. imgcount = len(images)
  133. }
  134. cmd := rcli.Subcmd(stdout, "info", "", "Display system-wide information.")
  135. if err := cmd.Parse(args); err != nil {
  136. return nil
  137. }
  138. if cmd.NArg() > 0 {
  139. cmd.Usage()
  140. return nil
  141. }
  142. fmt.Fprintf(stdout, "containers: %d\nversion: %s\nimages: %d\n",
  143. len(srv.runtime.List()),
  144. VERSION,
  145. imgcount)
  146. return nil
  147. }
  148. func (srv *Server) CmdStop(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  149. cmd := rcli.Subcmd(stdout, "stop", "[OPTIONS] NAME", "Stop a running container")
  150. if err := cmd.Parse(args); err != nil {
  151. return nil
  152. }
  153. if cmd.NArg() < 1 {
  154. cmd.Usage()
  155. return nil
  156. }
  157. for _, name := range cmd.Args() {
  158. if container := srv.runtime.Get(name); container != nil {
  159. if err := container.Stop(); err != nil {
  160. return err
  161. }
  162. fmt.Fprintln(stdout, container.Id)
  163. } else {
  164. return errors.New("No such container: " + name)
  165. }
  166. }
  167. return nil
  168. }
  169. func (srv *Server) CmdRestart(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  170. cmd := rcli.Subcmd(stdout, "restart", "[OPTIONS] NAME", "Restart a running container")
  171. if err := cmd.Parse(args); err != nil {
  172. return nil
  173. }
  174. if cmd.NArg() < 1 {
  175. cmd.Usage()
  176. return nil
  177. }
  178. for _, name := range cmd.Args() {
  179. if container := srv.runtime.Get(name); container != nil {
  180. if err := container.Restart(); err != nil {
  181. return err
  182. }
  183. fmt.Fprintln(stdout, container.Id)
  184. } else {
  185. return errors.New("No such container: " + name)
  186. }
  187. }
  188. return nil
  189. }
  190. func (srv *Server) CmdStart(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  191. cmd := rcli.Subcmd(stdout, "start", "[OPTIONS] NAME", "Start a stopped container")
  192. if err := cmd.Parse(args); err != nil {
  193. return nil
  194. }
  195. if cmd.NArg() < 1 {
  196. cmd.Usage()
  197. return nil
  198. }
  199. for _, name := range cmd.Args() {
  200. if container := srv.runtime.Get(name); container != nil {
  201. if err := container.Start(); err != nil {
  202. return err
  203. }
  204. fmt.Fprintln(stdout, container.Id)
  205. } else {
  206. return errors.New("No such container: " + name)
  207. }
  208. }
  209. return nil
  210. }
  211. func (srv *Server) CmdMount(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  212. cmd := rcli.Subcmd(stdout, "umount", "[OPTIONS] NAME", "mount a container's filesystem (debug only)")
  213. if err := cmd.Parse(args); err != nil {
  214. return nil
  215. }
  216. if cmd.NArg() < 1 {
  217. cmd.Usage()
  218. return nil
  219. }
  220. for _, name := range cmd.Args() {
  221. if container := srv.runtime.Get(name); container != nil {
  222. if err := container.EnsureMounted(); err != nil {
  223. return err
  224. }
  225. fmt.Fprintln(stdout, container.Id)
  226. } else {
  227. return errors.New("No such container: " + name)
  228. }
  229. }
  230. return nil
  231. }
  232. func (srv *Server) CmdInspect(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  233. cmd := rcli.Subcmd(stdout, "inspect", "[OPTIONS] CONTAINER", "Return low-level information on a container")
  234. if err := cmd.Parse(args); err != nil {
  235. return nil
  236. }
  237. if cmd.NArg() < 1 {
  238. cmd.Usage()
  239. return nil
  240. }
  241. name := cmd.Arg(0)
  242. var obj interface{}
  243. if container := srv.runtime.Get(name); container != nil {
  244. obj = container
  245. } else if image, err := srv.runtime.graph.Get(name); err != nil {
  246. return err
  247. } else if image != nil {
  248. obj = image
  249. } else {
  250. // No output means the object does not exist
  251. // (easier to script since stdout and stderr are not differentiated atm)
  252. return nil
  253. }
  254. data, err := json.Marshal(obj)
  255. if err != nil {
  256. return err
  257. }
  258. indented := new(bytes.Buffer)
  259. if err = json.Indent(indented, data, "", " "); err != nil {
  260. return err
  261. }
  262. if _, err := io.Copy(stdout, indented); err != nil {
  263. return err
  264. }
  265. stdout.Write([]byte{'\n'})
  266. return nil
  267. }
  268. func (srv *Server) CmdPort(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  269. cmd := rcli.Subcmd(stdout, "port", "[OPTIONS] CONTAINER PRIVATE_PORT", "Lookup the public-facing port which is NAT-ed to PRIVATE_PORT")
  270. if err := cmd.Parse(args); err != nil {
  271. return nil
  272. }
  273. if cmd.NArg() != 2 {
  274. cmd.Usage()
  275. return nil
  276. }
  277. name := cmd.Arg(0)
  278. privatePort := cmd.Arg(1)
  279. if container := srv.runtime.Get(name); container == nil {
  280. return errors.New("No such container: " + name)
  281. } else {
  282. if frontend, exists := container.NetworkSettings.PortMapping[privatePort]; !exists {
  283. return fmt.Errorf("No private port '%s' allocated on %s", privatePort, name)
  284. } else {
  285. fmt.Fprintln(stdout, frontend)
  286. }
  287. }
  288. return nil
  289. }
  290. // 'docker rmi NAME' removes all images with the name NAME
  291. func (srv *Server) CmdRmi(stdin io.ReadCloser, stdout io.Writer, args ...string) (err error) {
  292. cmd := rcli.Subcmd(stdout, "rmimage", "[OPTIONS] IMAGE", "Remove an image")
  293. if cmd.Parse(args) != nil || cmd.NArg() < 1 {
  294. cmd.Usage()
  295. return nil
  296. }
  297. for _, name := range cmd.Args() {
  298. if err := srv.runtime.graph.Delete(name); err != nil {
  299. return err
  300. }
  301. }
  302. return nil
  303. }
  304. func (srv *Server) CmdRm(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  305. cmd := rcli.Subcmd(stdout, "rm", "[OPTIONS] CONTAINER", "Remove a container")
  306. if err := cmd.Parse(args); err != nil {
  307. return nil
  308. }
  309. for _, name := range cmd.Args() {
  310. container := srv.runtime.Get(name)
  311. if container == nil {
  312. return errors.New("No such container: " + name)
  313. }
  314. if err := srv.runtime.Destroy(container); err != nil {
  315. fmt.Fprintln(stdout, "Error destroying container "+name+": "+err.Error())
  316. }
  317. }
  318. return nil
  319. }
  320. // 'docker kill NAME' kills a running container
  321. func (srv *Server) CmdKill(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  322. cmd := rcli.Subcmd(stdout, "kill", "[OPTIONS] CONTAINER [CONTAINER...]", "Kill a running container")
  323. if err := cmd.Parse(args); err != nil {
  324. return nil
  325. }
  326. for _, name := range cmd.Args() {
  327. container := srv.runtime.Get(name)
  328. if container == nil {
  329. return errors.New("No such container: " + name)
  330. }
  331. if err := container.Kill(); err != nil {
  332. fmt.Fprintln(stdout, "Error killing container "+name+": "+err.Error())
  333. }
  334. }
  335. return nil
  336. }
  337. func (srv *Server) CmdImport(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  338. cmd := rcli.Subcmd(stdout, "import", "[OPTIONS] NAME", "Create a new filesystem image from the contents of a tarball")
  339. fl_stdin := cmd.Bool("stdin", false, "Read tarball from stdin")
  340. var archive io.Reader
  341. var resp *http.Response
  342. if err := cmd.Parse(args); err != nil {
  343. return nil
  344. }
  345. name := cmd.Arg(0)
  346. if name == "" {
  347. return errors.New("Not enough arguments")
  348. }
  349. if *fl_stdin {
  350. archive = stdin
  351. } else {
  352. u, err := url.Parse(name)
  353. if err != nil {
  354. return err
  355. }
  356. if u.Scheme == "" {
  357. u.Scheme = "http"
  358. }
  359. if u.Host == "" {
  360. u.Host = "get.docker.io"
  361. u.Path = path.Join("/images", u.Path)
  362. }
  363. fmt.Fprintf(stdout, "Downloading from %s\n", u.String())
  364. // Download with curl (pretty progress bar)
  365. // If curl is not available, fallback to http.Get()
  366. resp, err = Download(u.String(), stdout)
  367. if err != nil {
  368. return err
  369. }
  370. archive = ProgressReader(resp.Body, int(resp.ContentLength), stdout)
  371. }
  372. fmt.Fprintf(stdout, "Unpacking to %s\n", name)
  373. img, err := srv.runtime.graph.Create(archive, "", "")
  374. if err != nil {
  375. return err
  376. }
  377. fmt.Fprintln(stdout, img.Id)
  378. return nil
  379. }
  380. func (srv *Server) CmdImages(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  381. cmd := rcli.Subcmd(stdout, "images", "[OPTIONS] [NAME]", "List images")
  382. //limit := cmd.Int("l", 0, "Only show the N most recent versions of each image")
  383. quiet := cmd.Bool("q", false, "only show numeric IDs")
  384. if err := cmd.Parse(args); err != nil {
  385. return nil
  386. }
  387. if cmd.NArg() > 1 {
  388. cmd.Usage()
  389. return nil
  390. }
  391. /*
  392. var nameFilter string
  393. if cmd.NArg() == 1 {
  394. nameFilter = cmd.Arg(0)
  395. }
  396. */
  397. w := tabwriter.NewWriter(stdout, 20, 1, 3, ' ', 0)
  398. if !*quiet {
  399. fmt.Fprintf(w, "NAME\tID\tCREATED\tPARENT\n")
  400. }
  401. if *quiet {
  402. images, err := srv.runtime.graph.All()
  403. if err != nil {
  404. return err
  405. }
  406. for _, image := range images {
  407. fmt.Fprintln(stdout, image.Id)
  408. }
  409. } else {
  410. // FIXME:
  411. // paths, err := srv.images.Paths()
  412. // if err != nil {
  413. // return err
  414. // }
  415. // for _, name := range paths {
  416. // if nameFilter != "" && nameFilter != name {
  417. // continue
  418. // }
  419. // ids, err := srv.images.List(name)
  420. // if err != nil {
  421. // return err
  422. // }
  423. // for idx, img := range ids {
  424. // if *limit > 0 && idx >= *limit {
  425. // break
  426. // }
  427. // if !*quiet {
  428. // for idx, field := range []string{
  429. // /* NAME */ name,
  430. // /* ID */ img.Id,
  431. // /* CREATED */ HumanDuration(time.Now().Sub(time.Unix(img.Created, 0))) + " ago",
  432. // /* PARENT */ img.Parent,
  433. // } {
  434. // if idx == 0 {
  435. // w.Write([]byte(field))
  436. // } else {
  437. // w.Write([]byte("\t" + field))
  438. // }
  439. // }
  440. // w.Write([]byte{'\n'})
  441. // } else {
  442. // stdout.Write([]byte(img.Id + "\n"))
  443. // }
  444. // }
  445. // }
  446. // if !*quiet {
  447. // w.Flush()
  448. // }
  449. //
  450. }
  451. return nil
  452. }
  453. func (srv *Server) CmdPs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  454. cmd := rcli.Subcmd(stdout,
  455. "ps", "[OPTIONS]", "List containers")
  456. quiet := cmd.Bool("q", false, "Only display numeric IDs")
  457. fl_all := cmd.Bool("a", false, "Show all containers. Only running containers are shown by default.")
  458. fl_full := cmd.Bool("notrunc", false, "Don't truncate output")
  459. if err := cmd.Parse(args); err != nil {
  460. return nil
  461. }
  462. w := tabwriter.NewWriter(stdout, 12, 1, 3, ' ', 0)
  463. if !*quiet {
  464. fmt.Fprintf(w, "ID\tIMAGE\tCOMMAND\tCREATED\tSTATUS\tCOMMENT\n")
  465. }
  466. for _, container := range srv.runtime.List() {
  467. if !container.State.Running && !*fl_all {
  468. continue
  469. }
  470. if !*quiet {
  471. command := fmt.Sprintf("%s %s", container.Path, strings.Join(container.Args, " "))
  472. if !*fl_full {
  473. command = Trunc(command, 20)
  474. }
  475. for idx, field := range []string{
  476. /* ID */ container.Id,
  477. /* IMAGE */ container.Image,
  478. /* COMMAND */ command,
  479. /* CREATED */ HumanDuration(time.Now().Sub(container.Created)) + " ago",
  480. /* STATUS */ container.State.String(),
  481. /* COMMENT */ "",
  482. } {
  483. if idx == 0 {
  484. w.Write([]byte(field))
  485. } else {
  486. w.Write([]byte("\t" + field))
  487. }
  488. }
  489. w.Write([]byte{'\n'})
  490. } else {
  491. stdout.Write([]byte(container.Id + "\n"))
  492. }
  493. }
  494. if !*quiet {
  495. w.Flush()
  496. }
  497. return nil
  498. }
  499. func (srv *Server) CmdCommit(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  500. cmd := rcli.Subcmd(stdout,
  501. "commit", "[OPTIONS] CONTAINER [DEST]",
  502. "Create a new image from a container's changes")
  503. if err := cmd.Parse(args); err != nil {
  504. return nil
  505. }
  506. containerName, imgName := cmd.Arg(0), cmd.Arg(1)
  507. if containerName == "" || imgName == "" {
  508. cmd.Usage()
  509. return nil
  510. }
  511. if container := srv.runtime.Get(containerName); container != nil {
  512. // FIXME: freeze the container before copying it to avoid data corruption?
  513. // FIXME: this shouldn't be in commands.
  514. rwTar, err := container.ExportRw()
  515. if err != nil {
  516. return err
  517. }
  518. // Create a new image from the container's base layers + a new layer from container changes
  519. img, err := srv.runtime.graph.Create(rwTar, container.Image, "")
  520. if err != nil {
  521. return err
  522. }
  523. fmt.Fprintln(stdout, img.Id)
  524. return nil
  525. }
  526. return errors.New("No such container: " + containerName)
  527. }
  528. func (srv *Server) CmdExport(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  529. cmd := rcli.Subcmd(stdout,
  530. "export", "CONTAINER",
  531. "Export the contents of a filesystem as a tar archive")
  532. if err := cmd.Parse(args); err != nil {
  533. return nil
  534. }
  535. name := cmd.Arg(0)
  536. if container := srv.runtime.Get(name); container != nil {
  537. data, err := container.Export()
  538. if err != nil {
  539. return err
  540. }
  541. // Stream the entire contents of the container (basically a volatile snapshot)
  542. if _, err := io.Copy(stdout, data); err != nil {
  543. return err
  544. }
  545. return nil
  546. }
  547. return errors.New("No such container: " + name)
  548. }
  549. func (srv *Server) CmdDiff(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  550. cmd := rcli.Subcmd(stdout,
  551. "diff", "CONTAINER [OPTIONS]",
  552. "Inspect changes on a container's filesystem")
  553. if err := cmd.Parse(args); err != nil {
  554. return nil
  555. }
  556. if cmd.NArg() < 1 {
  557. return errors.New("Not enough arguments")
  558. }
  559. if container := srv.runtime.Get(cmd.Arg(0)); container == nil {
  560. return errors.New("No such container")
  561. } else {
  562. changes, err := container.Changes()
  563. if err != nil {
  564. return err
  565. }
  566. for _, change := range changes {
  567. fmt.Fprintln(stdout, change.String())
  568. }
  569. }
  570. return nil
  571. }
  572. func (srv *Server) CmdLogs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  573. cmd := rcli.Subcmd(stdout, "logs", "[OPTIONS] CONTAINER", "Fetch the logs of a container")
  574. if err := cmd.Parse(args); err != nil {
  575. return nil
  576. }
  577. if cmd.NArg() != 1 {
  578. cmd.Usage()
  579. return nil
  580. }
  581. name := cmd.Arg(0)
  582. if container := srv.runtime.Get(name); container != nil {
  583. log_stdout, err := container.ReadLog("stdout")
  584. if err != nil {
  585. return err
  586. }
  587. log_stderr, err := container.ReadLog("stderr")
  588. if err != nil {
  589. return err
  590. }
  591. // FIXME: Interpolate stdout and stderr instead of concatenating them
  592. // FIXME: Differentiate stdout and stderr in the remote protocol
  593. if _, err := io.Copy(stdout, log_stdout); err != nil {
  594. return err
  595. }
  596. if _, err := io.Copy(stdout, log_stderr); err != nil {
  597. return err
  598. }
  599. return nil
  600. }
  601. return errors.New("No such container: " + cmd.Arg(0))
  602. }
  603. func (srv *Server) CmdAttach(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  604. cmd := rcli.Subcmd(stdout, "attach", "[OPTIONS]", "Attach to a running container")
  605. fl_i := cmd.Bool("i", false, "Attach to stdin")
  606. fl_o := cmd.Bool("o", true, "Attach to stdout")
  607. fl_e := cmd.Bool("e", true, "Attach to stderr")
  608. if err := cmd.Parse(args); err != nil {
  609. return nil
  610. }
  611. if cmd.NArg() != 1 {
  612. cmd.Usage()
  613. return nil
  614. }
  615. name := cmd.Arg(0)
  616. container := srv.runtime.Get(name)
  617. if container == nil {
  618. return errors.New("No such container: " + name)
  619. }
  620. var wg sync.WaitGroup
  621. if *fl_i {
  622. c_stdin, err := container.StdinPipe()
  623. if err != nil {
  624. return err
  625. }
  626. wg.Add(1)
  627. go func() { io.Copy(c_stdin, stdin); wg.Add(-1) }()
  628. }
  629. if *fl_o {
  630. c_stdout, err := container.StdoutPipe()
  631. if err != nil {
  632. return err
  633. }
  634. wg.Add(1)
  635. go func() { io.Copy(stdout, c_stdout); wg.Add(-1) }()
  636. }
  637. if *fl_e {
  638. c_stderr, err := container.StderrPipe()
  639. if err != nil {
  640. return err
  641. }
  642. wg.Add(1)
  643. go func() { io.Copy(stdout, c_stderr); wg.Add(-1) }()
  644. }
  645. wg.Wait()
  646. return nil
  647. }
  648. // Ports type - Used to parse multiple -p flags
  649. type ports []int
  650. func (p *ports) String() string {
  651. return fmt.Sprint(*p)
  652. }
  653. func (p *ports) Set(value string) error {
  654. port, err := strconv.Atoi(value)
  655. if err != nil {
  656. return fmt.Errorf("Invalid port: %v", value)
  657. }
  658. *p = append(*p, port)
  659. return nil
  660. }
  661. func (srv *Server) CmdRun(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  662. cmd := rcli.Subcmd(stdout, "run", "[OPTIONS] IMAGE COMMAND [ARG...]", "Run a command in a new container")
  663. fl_user := cmd.String("u", "", "Username or UID")
  664. fl_attach := cmd.Bool("a", false, "Attach stdin and stdout")
  665. fl_stdin := cmd.Bool("i", false, "Keep stdin open even if not attached")
  666. fl_tty := cmd.Bool("t", false, "Allocate a pseudo-tty")
  667. fl_memory := cmd.Int64("m", 0, "Memory limit (in bytes)")
  668. var fl_ports ports
  669. cmd.Var(&fl_ports, "p", "Map a network port to the container")
  670. if err := cmd.Parse(args); err != nil {
  671. return nil
  672. }
  673. name := cmd.Arg(0)
  674. var cmdline []string
  675. if len(cmd.Args()) >= 2 {
  676. cmdline = cmd.Args()[1:]
  677. }
  678. // Choose a default image if needed
  679. if name == "" {
  680. name = "base"
  681. }
  682. // Choose a default command if needed
  683. if len(cmdline) == 0 {
  684. *fl_stdin = true
  685. *fl_tty = true
  686. *fl_attach = true
  687. cmdline = []string{"/bin/bash", "-i"}
  688. }
  689. // Create new container
  690. container, err := srv.runtime.Create(cmdline[0], cmdline[1:], name,
  691. &Config{
  692. Ports: fl_ports,
  693. User: *fl_user,
  694. Tty: *fl_tty,
  695. OpenStdin: *fl_stdin,
  696. Memory: *fl_memory,
  697. })
  698. if err != nil {
  699. return errors.New("Error creating container: " + err.Error())
  700. }
  701. if *fl_stdin {
  702. cmd_stdin, err := container.StdinPipe()
  703. if err != nil {
  704. return err
  705. }
  706. if *fl_attach {
  707. Go(func() error {
  708. _, err := io.Copy(cmd_stdin, stdin)
  709. cmd_stdin.Close()
  710. return err
  711. })
  712. }
  713. }
  714. // Run the container
  715. if *fl_attach {
  716. cmd_stderr, err := container.StderrPipe()
  717. if err != nil {
  718. return err
  719. }
  720. cmd_stdout, err := container.StdoutPipe()
  721. if err != nil {
  722. return err
  723. }
  724. if err := container.Start(); err != nil {
  725. return err
  726. }
  727. sending_stdout := Go(func() error {
  728. _, err := io.Copy(stdout, cmd_stdout)
  729. return err
  730. })
  731. sending_stderr := Go(func() error {
  732. _, err := io.Copy(stdout, cmd_stderr)
  733. return err
  734. })
  735. err_sending_stdout := <-sending_stdout
  736. err_sending_stderr := <-sending_stderr
  737. if err_sending_stdout != nil {
  738. return err_sending_stdout
  739. }
  740. if err_sending_stderr != nil {
  741. return err_sending_stderr
  742. }
  743. container.Wait()
  744. } else {
  745. if err := container.Start(); err != nil {
  746. return err
  747. }
  748. fmt.Fprintln(stdout, container.Id)
  749. }
  750. return nil
  751. }
  752. func NewServer() (*Server, error) {
  753. rand.Seed(time.Now().UTC().UnixNano())
  754. if runtime.GOARCH != "amd64" {
  755. log.Fatalf("The docker runtime currently only supports amd64 (not %s). This will change in the future. Aborting.", runtime.GOARCH)
  756. }
  757. // if err != nil {
  758. // return nil, err
  759. // }
  760. runtime, err := New()
  761. if err != nil {
  762. return nil, err
  763. }
  764. srv := &Server{
  765. runtime: runtime,
  766. }
  767. return srv, nil
  768. }
  769. type Server struct {
  770. runtime *Runtime
  771. }