1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633 |
- package docker
- import (
- "bufio"
- "encoding/json"
- "errors"
- "fmt"
- "github.com/dotcloud/docker/archive"
- "github.com/dotcloud/docker/auth"
- "github.com/dotcloud/docker/engine"
- "github.com/dotcloud/docker/gograph"
- "github.com/dotcloud/docker/registry"
- "github.com/dotcloud/docker/utils"
- "io"
- "io/ioutil"
- "log"
- "net/http"
- "net/url"
- "os"
- "os/exec"
- "os/signal"
- "path"
- "path/filepath"
- "runtime"
- "strings"
- "sync"
- "syscall"
- "time"
- )
- func (srv *Server) Close() error {
- return srv.runtime.Close()
- }
- func init() {
- engine.Register("initapi", jobInitApi)
- }
- // jobInitApi runs the remote api server `srv` as a daemon,
- // Only one api server can run at the same time - this is enforced by a pidfile.
- // The signals SIGINT, SIGKILL and SIGTERM are intercepted for cleanup.
- func jobInitApi(job *engine.Job) string {
- job.Logf("Creating server")
- srv, err := NewServer(job.Eng, ConfigFromJob(job))
- if err != nil {
- return err.Error()
- }
- if srv.runtime.config.Pidfile != "" {
- job.Logf("Creating pidfile")
- if err := utils.CreatePidFile(srv.runtime.config.Pidfile); err != nil {
- log.Fatal(err)
- }
- }
- job.Logf("Setting up signal traps")
- c := make(chan os.Signal, 1)
- signal.Notify(c, os.Interrupt, os.Kill, os.Signal(syscall.SIGTERM))
- go func() {
- sig := <-c
- log.Printf("Received signal '%v', exiting\n", sig)
- utils.RemovePidFile(srv.runtime.config.Pidfile)
- srv.Close()
- os.Exit(0)
- }()
- job.Eng.Hack_SetGlobalVar("httpapi.server", srv)
- job.Eng.Hack_SetGlobalVar("httpapi.runtime", srv.runtime)
- job.Eng.Hack_SetGlobalVar("httpapi.bridgeIP", srv.runtime.networkManager.bridgeNetwork.IP)
- if err := job.Eng.Register("create", srv.ContainerCreate); err != nil {
- return err.Error()
- }
- if err := job.Eng.Register("start", srv.ContainerStart); err != nil {
- return err.Error()
- }
- if err := job.Eng.Register("serveapi", srv.ListenAndServe); err != nil {
- return err.Error()
- }
- return "0"
- }
- func (srv *Server) ListenAndServe(job *engine.Job) string {
- protoAddrs := job.Args
- chErrors := make(chan error, len(protoAddrs))
- for _, protoAddr := range protoAddrs {
- protoAddrParts := strings.SplitN(protoAddr, "://", 2)
- switch protoAddrParts[0] {
- case "unix":
- if err := syscall.Unlink(protoAddrParts[1]); err != nil && !os.IsNotExist(err) {
- log.Fatal(err)
- }
- case "tcp":
- if !strings.HasPrefix(protoAddrParts[1], "127.0.0.1") {
- log.Println("/!\\ DON'T BIND ON ANOTHER IP ADDRESS THAN 127.0.0.1 IF YOU DON'T KNOW WHAT YOU'RE DOING /!\\")
- }
- default:
- return "Invalid protocol format."
- }
- go func() {
- // FIXME: merge Server.ListenAndServe with ListenAndServe
- chErrors <- ListenAndServe(protoAddrParts[0], protoAddrParts[1], srv, job.GetenvBool("Logging"))
- }()
- }
- for i := 0; i < len(protoAddrs); i += 1 {
- err := <-chErrors
- if err != nil {
- return err.Error()
- }
- }
- return "0"
- }
- func (srv *Server) DockerVersion() APIVersion {
- return APIVersion{
- Version: VERSION,
- GitCommit: GITCOMMIT,
- GoVersion: runtime.Version(),
- }
- }
- // simpleVersionInfo is a simple implementation of
- // the interface VersionInfo, which is used
- // to provide version information for some product,
- // component, etc. It stores the product name and the version
- // in string and returns them on calls to Name() and Version().
- type simpleVersionInfo struct {
- name string
- version string
- }
- func (v *simpleVersionInfo) Name() string {
- return v.name
- }
- func (v *simpleVersionInfo) Version() string {
- return v.version
- }
- // versionCheckers() returns version informations of:
- // docker, go, git-commit (of the docker) and the host's kernel.
- //
- // Such information will be used on call to NewRegistry().
- func (srv *Server) versionInfos() []utils.VersionInfo {
- v := srv.DockerVersion()
- ret := append(make([]utils.VersionInfo, 0, 4), &simpleVersionInfo{"docker", v.Version})
- if len(v.GoVersion) > 0 {
- ret = append(ret, &simpleVersionInfo{"go", v.GoVersion})
- }
- if len(v.GitCommit) > 0 {
- ret = append(ret, &simpleVersionInfo{"git-commit", v.GitCommit})
- }
- if kernelVersion, err := utils.GetKernelVersion(); err == nil {
- ret = append(ret, &simpleVersionInfo{"kernel", kernelVersion.String()})
- }
- return ret
- }
- // ContainerKill send signal to the container
- // If no signal is given (sig 0), then Kill with SIGKILL and wait
- // for the container to exit.
- // If a signal is given, then just send it to the container and return.
- func (srv *Server) ContainerKill(name string, sig int) error {
- if container := srv.runtime.Get(name); container != nil {
- // If no signal is passed, perform regular Kill (SIGKILL + wait())
- if sig == 0 {
- if err := container.Kill(); err != nil {
- return fmt.Errorf("Cannot kill container %s: %s", name, err)
- }
- srv.LogEvent("kill", container.ID, srv.runtime.repositories.ImageName(container.Image))
- } else {
- // Otherwise, just send the requested signal
- if err := container.kill(sig); err != nil {
- return fmt.Errorf("Cannot kill container %s: %s", name, err)
- }
- // FIXME: Add event for signals
- }
- } else {
- return fmt.Errorf("No such container: %s", name)
- }
- return nil
- }
- func (srv *Server) ContainerExport(name string, out io.Writer) error {
- if container := srv.runtime.Get(name); container != nil {
- data, err := container.Export()
- if err != nil {
- return err
- }
- // Stream the entire contents of the container (basically a volatile snapshot)
- if _, err := io.Copy(out, data); err != nil {
- return err
- }
- srv.LogEvent("export", container.ID, srv.runtime.repositories.ImageName(container.Image))
- return nil
- }
- return fmt.Errorf("No such container: %s", name)
- }
- func (srv *Server) ImagesSearch(term string) ([]registry.SearchResult, error) {
- r, err := registry.NewRegistry(srv.runtime.config.Root, nil, srv.HTTPRequestFactory(nil))
- if err != nil {
- return nil, err
- }
- results, err := r.SearchRepositories(term)
- if err != nil {
- return nil, err
- }
- return results.Results, nil
- }
- func (srv *Server) ImageInsert(name, url, path string, out io.Writer, sf *utils.StreamFormatter) error {
- out = utils.NewWriteFlusher(out)
- img, err := srv.runtime.repositories.LookupImage(name)
- if err != nil {
- return err
- }
- file, err := utils.Download(url, out)
- if err != nil {
- return err
- }
- defer file.Body.Close()
- config, _, _, err := ParseRun([]string{img.ID, "echo", "insert", url, path}, srv.runtime.capabilities)
- if err != nil {
- return err
- }
- c, _, err := srv.runtime.Create(config, "")
- if err != nil {
- return err
- }
- if err := c.Inject(utils.ProgressReader(file.Body, int(file.ContentLength), out, sf.FormatProgress("", "Downloading", "%8v/%v (%v)"), sf, false), path); err != nil {
- return err
- }
- // FIXME: Handle custom repo, tag comment, author
- img, err = srv.runtime.Commit(c, "", "", img.Comment, img.Author, nil)
- if err != nil {
- return err
- }
- out.Write(sf.FormatStatus(img.ID, ""))
- return nil
- }
- func (srv *Server) ImagesViz(out io.Writer) error {
- images, _ := srv.runtime.graph.Map()
- if images == nil {
- return nil
- }
- out.Write([]byte("digraph docker {\n"))
- var (
- parentImage *Image
- err error
- )
- for _, image := range images {
- parentImage, err = image.GetParent()
- if err != nil {
- return fmt.Errorf("Error while getting parent image: %v", err)
- }
- if parentImage != nil {
- out.Write([]byte(" \"" + parentImage.ID + "\" -> \"" + image.ID + "\"\n"))
- } else {
- out.Write([]byte(" base -> \"" + image.ID + "\" [style=invis]\n"))
- }
- }
- reporefs := make(map[string][]string)
- for name, repository := range srv.runtime.repositories.Repositories {
- for tag, id := range repository {
- reporefs[utils.TruncateID(id)] = append(reporefs[utils.TruncateID(id)], fmt.Sprintf("%s:%s", name, tag))
- }
- }
- for id, repos := range reporefs {
- out.Write([]byte(" \"" + id + "\" [label=\"" + id + "\\n" + strings.Join(repos, "\\n") + "\",shape=box,fillcolor=\"paleturquoise\",style=\"filled,rounded\"];\n"))
- }
- out.Write([]byte(" base [style=invisible]\n}\n"))
- return nil
- }
- func (srv *Server) Images(all bool, filter string) ([]APIImages, error) {
- var (
- allImages map[string]*Image
- err error
- )
- if all {
- allImages, err = srv.runtime.graph.Map()
- } else {
- allImages, err = srv.runtime.graph.Heads()
- }
- if err != nil {
- return nil, err
- }
- lookup := make(map[string]APIImages)
- for name, repository := range srv.runtime.repositories.Repositories {
- if filter != "" {
- if match, _ := path.Match(filter, name); !match {
- continue
- }
- }
- for tag, id := range repository {
- image, err := srv.runtime.graph.Get(id)
- if err != nil {
- log.Printf("Warning: couldn't load %s from %s/%s: %s", id, name, tag, err)
- continue
- }
- if out, exists := lookup[id]; exists {
- out.RepoTags = append(out.RepoTags, fmt.Sprintf("%s:%s", name, tag))
- lookup[id] = out
- } else {
- var out APIImages
- delete(allImages, id)
- out.ParentId = image.Parent
- out.RepoTags = []string{fmt.Sprintf("%s:%s", name, tag)}
- out.ID = image.ID
- out.Created = image.Created.Unix()
- out.Size = image.Size
- out.VirtualSize = image.getParentsSize(0) + image.Size
- lookup[id] = out
- }
- }
- }
- outs := make([]APIImages, 0, len(lookup))
- for _, value := range lookup {
- outs = append(outs, value)
- }
- // Display images which aren't part of a repository/tag
- if filter == "" {
- for _, image := range allImages {
- var out APIImages
- out.ID = image.ID
- out.ParentId = image.Parent
- out.RepoTags = []string{"<none>:<none>"}
- out.Created = image.Created.Unix()
- out.Size = image.Size
- out.VirtualSize = image.getParentsSize(0) + image.Size
- outs = append(outs, out)
- }
- }
- sortImagesByCreationAndTag(outs)
- return outs, nil
- }
- func (srv *Server) DockerInfo() *APIInfo {
- images, _ := srv.runtime.graph.Map()
- var imgcount int
- if images == nil {
- imgcount = 0
- } else {
- imgcount = len(images)
- }
- lxcVersion := ""
- if output, err := exec.Command("lxc-version").CombinedOutput(); err == nil {
- outputStr := string(output)
- if len(strings.SplitN(outputStr, ":", 2)) == 2 {
- lxcVersion = strings.TrimSpace(strings.SplitN(string(output), ":", 2)[1])
- }
- }
- kernelVersion := "<unknown>"
- if kv, err := utils.GetKernelVersion(); err == nil {
- kernelVersion = kv.String()
- }
- return &APIInfo{
- Containers: len(srv.runtime.List()),
- Images: imgcount,
- MemoryLimit: srv.runtime.capabilities.MemoryLimit,
- SwapLimit: srv.runtime.capabilities.SwapLimit,
- IPv4Forwarding: !srv.runtime.capabilities.IPv4ForwardingDisabled,
- Debug: os.Getenv("DEBUG") != "",
- NFd: utils.GetTotalUsedFds(),
- NGoroutines: runtime.NumGoroutine(),
- LXCVersion: lxcVersion,
- NEventsListener: len(srv.events),
- KernelVersion: kernelVersion,
- IndexServerAddress: auth.IndexServerAddress(),
- }
- }
- func (srv *Server) ImageHistory(name string) ([]APIHistory, error) {
- image, err := srv.runtime.repositories.LookupImage(name)
- if err != nil {
- return nil, err
- }
- lookupMap := make(map[string][]string)
- for name, repository := range srv.runtime.repositories.Repositories {
- for tag, id := range repository {
- // If the ID already has a reverse lookup, do not update it unless for "latest"
- if _, exists := lookupMap[id]; !exists {
- lookupMap[id] = []string{}
- }
- lookupMap[id] = append(lookupMap[id], name+":"+tag)
- }
- }
- outs := []APIHistory{} //produce [] when empty instead of 'null'
- err = image.WalkHistory(func(img *Image) error {
- var out APIHistory
- out.ID = img.ID
- out.Created = img.Created.Unix()
- out.CreatedBy = strings.Join(img.ContainerConfig.Cmd, " ")
- out.Tags = lookupMap[img.ID]
- out.Size = img.Size
- outs = append(outs, out)
- return nil
- })
- return outs, nil
- }
- func (srv *Server) ContainerTop(name, ps_args string) (*APITop, error) {
- if container := srv.runtime.Get(name); container != nil {
- output, err := exec.Command("lxc-ps", "--name", container.ID, "--", ps_args).CombinedOutput()
- if err != nil {
- return nil, fmt.Errorf("lxc-ps: %s (%s)", err, output)
- }
- procs := APITop{}
- for i, line := range strings.Split(string(output), "\n") {
- if len(line) == 0 {
- continue
- }
- words := []string{}
- scanner := bufio.NewScanner(strings.NewReader(line))
- scanner.Split(bufio.ScanWords)
- if !scanner.Scan() {
- return nil, fmt.Errorf("Wrong output using lxc-ps")
- }
- // no scanner.Text because we skip container id
- for scanner.Scan() {
- if i != 0 && len(words) == len(procs.Titles) {
- words[len(words)-1] = fmt.Sprintf("%s %s", words[len(words)-1], scanner.Text())
- } else {
- words = append(words, scanner.Text())
- }
- }
- if i == 0 {
- procs.Titles = words
- } else {
- procs.Processes = append(procs.Processes, words)
- }
- }
- return &procs, nil
- }
- return nil, fmt.Errorf("No such container: %s", name)
- }
- func (srv *Server) ContainerChanges(name string) ([]Change, error) {
- if container := srv.runtime.Get(name); container != nil {
- return container.Changes()
- }
- return nil, fmt.Errorf("No such container: %s", name)
- }
- func (srv *Server) Containers(all, size bool, n int, since, before string) []APIContainers {
- var foundBefore bool
- var displayed int
- out := []APIContainers{}
- for _, container := range srv.runtime.List() {
- if !container.State.Running && !all && n == -1 && since == "" && before == "" {
- continue
- }
- if before != "" {
- if container.ID == before || utils.TruncateID(container.ID) == before {
- foundBefore = true
- continue
- }
- if !foundBefore {
- continue
- }
- }
- if displayed == n {
- break
- }
- if container.ID == since || utils.TruncateID(container.ID) == since {
- break
- }
- displayed++
- c := createAPIContainer(container, size, srv.runtime)
- out = append(out, c)
- }
- return out
- }
- func createAPIContainer(container *Container, size bool, runtime *Runtime) APIContainers {
- c := APIContainers{
- ID: container.ID,
- }
- names := []string{}
- runtime.containerGraph.Walk("/", func(p string, e *gograph.Entity) error {
- if e.ID() == container.ID {
- names = append(names, p)
- }
- return nil
- }, -1)
- c.Names = names
- c.Image = runtime.repositories.ImageName(container.Image)
- c.Command = fmt.Sprintf("%s %s", container.Path, strings.Join(container.Args, " "))
- c.Created = container.Created.Unix()
- c.Status = container.State.String()
- c.Ports = container.NetworkSettings.PortMappingAPI()
- if size {
- c.SizeRw, c.SizeRootFs = container.GetSize()
- }
- return c
- }
- func (srv *Server) ContainerCommit(name, repo, tag, author, comment string, config *Config) (string, error) {
- container := srv.runtime.Get(name)
- if container == nil {
- return "", fmt.Errorf("No such container: %s", name)
- }
- img, err := srv.runtime.Commit(container, repo, tag, comment, author, config)
- if err != nil {
- return "", err
- }
- return img.ID, err
- }
- // FIXME: this should be called ImageTag
- func (srv *Server) ContainerTag(name, repo, tag string, force bool) error {
- if err := srv.runtime.repositories.Set(repo, tag, name, force); err != nil {
- return err
- }
- return nil
- }
- func (srv *Server) pullImage(r *registry.Registry, out io.Writer, imgID, endpoint string, token []string, sf *utils.StreamFormatter) error {
- history, err := r.GetRemoteHistory(imgID, endpoint, token)
- if err != nil {
- return err
- }
- out.Write(sf.FormatProgress(utils.TruncateID(imgID), "Pulling", "dependend layers"))
- // FIXME: Try to stream the images?
- // FIXME: Launch the getRemoteImage() in goroutines
- for _, id := range history {
- // ensure no two downloads of the same layer happen at the same time
- if err := srv.poolAdd("pull", "layer:"+id); err != nil {
- utils.Errorf("Image (id: %s) pull is already running, skipping: %v", id, err)
- return nil
- }
- defer srv.poolRemove("pull", "layer:"+id)
- if !srv.runtime.graph.Exists(id) {
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Pulling", "metadata"))
- imgJSON, imgSize, err := r.GetRemoteImageJSON(id, endpoint, token)
- if err != nil {
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Error", "pulling dependend layers"))
- // FIXME: Keep going in case of error?
- return err
- }
- img, err := NewImgJSON(imgJSON)
- if err != nil {
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Error", "pulling dependend layers"))
- return fmt.Errorf("Failed to parse json: %s", err)
- }
- // Get the layer
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Pulling", "fs layer"))
- layer, err := r.GetRemoteImageLayer(img.ID, endpoint, token)
- if err != nil {
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Error", "pulling dependend layers"))
- return err
- }
- defer layer.Close()
- if err := srv.runtime.graph.Register(imgJSON, utils.ProgressReader(layer, imgSize, out, sf.FormatProgress(utils.TruncateID(id), "Downloading", "%8v/%v (%v)"), sf, false), img); err != nil {
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Error", "downloading dependend layers"))
- return err
- }
- }
- out.Write(sf.FormatProgress(utils.TruncateID(id), "Download", "complete"))
- }
- return nil
- }
- func (srv *Server) pullRepository(r *registry.Registry, out io.Writer, localName, remoteName, askedTag, indexEp string, sf *utils.StreamFormatter, parallel bool) error {
- out.Write(sf.FormatStatus("", "Pulling repository %s", localName))
- repoData, err := r.GetRepositoryData(indexEp, remoteName)
- if err != nil {
- return err
- }
- utils.Debugf("Retrieving the tag list")
- tagsList, err := r.GetRemoteTags(repoData.Endpoints, remoteName, repoData.Tokens)
- if err != nil {
- utils.Errorf("%v", err)
- return err
- }
- for tag, id := range tagsList {
- repoData.ImgList[id] = ®istry.ImgData{
- ID: id,
- Tag: tag,
- Checksum: "",
- }
- }
- utils.Debugf("Registering tags")
- // If no tag has been specified, pull them all
- if askedTag == "" {
- for tag, id := range tagsList {
- repoData.ImgList[id].Tag = tag
- }
- } else {
- // Otherwise, check that the tag exists and use only that one
- id, exists := tagsList[askedTag]
- if !exists {
- return fmt.Errorf("Tag %s not found in repository %s", askedTag, localName)
- }
- repoData.ImgList[id].Tag = askedTag
- }
- errors := make(chan error)
- for _, image := range repoData.ImgList {
- downloadImage := func(img *registry.ImgData) {
- if askedTag != "" && img.Tag != askedTag {
- utils.Debugf("(%s) does not match %s (id: %s), skipping", img.Tag, askedTag, img.ID)
- if parallel {
- errors <- nil
- }
- return
- }
- if img.Tag == "" {
- utils.Debugf("Image (id: %s) present in this repository but untagged, skipping", img.ID)
- if parallel {
- errors <- nil
- }
- return
- }
- // ensure no two downloads of the same image happen at the same time
- if err := srv.poolAdd("pull", "img:"+img.ID); err != nil {
- utils.Errorf("Image (id: %s) pull is already running, skipping: %v", img.ID, err)
- if parallel {
- errors <- nil
- }
- return
- }
- defer srv.poolRemove("pull", "img:"+img.ID)
- out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Pulling", fmt.Sprintf("image (%s) from %s", img.Tag, localName)))
- success := false
- var lastErr error
- for _, ep := range repoData.Endpoints {
- out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Pulling", fmt.Sprintf("image (%s) from %s, endpoint: %s", img.Tag, localName, ep)))
- if err := srv.pullImage(r, out, img.ID, ep, repoData.Tokens, sf); err != nil {
- // Its not ideal that only the last error is returned, it would be better to concatenate the errors.
- // As the error is also given to the output stream the user will see the error.
- lastErr = err
- out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Error pulling", fmt.Sprintf("image (%s) from %s, endpoint: %s, %s", img.Tag, localName, ep, err)))
- continue
- }
- success = true
- break
- }
- if !success {
- out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Error pulling", fmt.Sprintf("image (%s) from %s, %s", img.Tag, localName, lastErr)))
- if parallel {
- errors <- fmt.Errorf("Could not find repository on any of the indexed registries.")
- return
- }
- }
- out.Write(sf.FormatProgress(utils.TruncateID(img.ID), "Download", "complete"))
- if parallel {
- errors <- nil
- }
- }
- if parallel {
- go downloadImage(image)
- } else {
- downloadImage(image)
- }
- }
- if parallel {
- var lastError error
- for i := 0; i < len(repoData.ImgList); i++ {
- if err := <-errors; err != nil {
- lastError = err
- }
- }
- if lastError != nil {
- return lastError
- }
- }
- for tag, id := range tagsList {
- if askedTag != "" && tag != askedTag {
- continue
- }
- if err := srv.runtime.repositories.Set(localName, tag, id, true); err != nil {
- return err
- }
- }
- if err := srv.runtime.repositories.Save(); err != nil {
- return err
- }
- return nil
- }
- func (srv *Server) poolAdd(kind, key string) error {
- srv.Lock()
- defer srv.Unlock()
- if _, exists := srv.pullingPool[key]; exists {
- return fmt.Errorf("pull %s is already in progress", key)
- }
- if _, exists := srv.pushingPool[key]; exists {
- return fmt.Errorf("push %s is already in progress", key)
- }
- switch kind {
- case "pull":
- srv.pullingPool[key] = struct{}{}
- break
- case "push":
- srv.pushingPool[key] = struct{}{}
- break
- default:
- return fmt.Errorf("Unknown pool type")
- }
- return nil
- }
- func (srv *Server) poolRemove(kind, key string) error {
- switch kind {
- case "pull":
- delete(srv.pullingPool, key)
- break
- case "push":
- delete(srv.pushingPool, key)
- break
- default:
- return fmt.Errorf("Unknown pool type")
- }
- return nil
- }
- func (srv *Server) ImagePull(localName string, tag string, out io.Writer, sf *utils.StreamFormatter, authConfig *auth.AuthConfig, metaHeaders map[string][]string, parallel bool) error {
- r, err := registry.NewRegistry(srv.runtime.config.Root, authConfig, srv.HTTPRequestFactory(metaHeaders))
- if err != nil {
- return err
- }
- if err := srv.poolAdd("pull", localName+":"+tag); err != nil {
- return err
- }
- defer srv.poolRemove("pull", localName+":"+tag)
- // Resolve the Repository name from fqn to endpoint + name
- endpoint, remoteName, err := registry.ResolveRepositoryName(localName)
- if err != nil {
- return err
- }
- if endpoint == auth.IndexServerAddress() {
- // If pull "index.docker.io/foo/bar", it's stored locally under "foo/bar"
- localName = remoteName
- }
- out = utils.NewWriteFlusher(out)
- err = srv.pullRepository(r, out, localName, remoteName, tag, endpoint, sf, parallel)
- if err == registry.ErrLoginRequired {
- return err
- }
- if err != nil {
- if err := srv.pullImage(r, out, remoteName, endpoint, nil, sf); err != nil {
- return err
- }
- return nil
- }
- return nil
- }
- // Retrieve the all the images to be uploaded in the correct order
- // Note: we can't use a map as it is not ordered
- func (srv *Server) getImageList(localRepo map[string]string) ([][]*registry.ImgData, error) {
- imgList := map[string]*registry.ImgData{}
- depGraph := utils.NewDependencyGraph()
- for tag, id := range localRepo {
- img, err := srv.runtime.graph.Get(id)
- if err != nil {
- return nil, err
- }
- depGraph.NewNode(img.ID)
- img.WalkHistory(func(current *Image) error {
- imgList[current.ID] = ®istry.ImgData{
- ID: current.ID,
- Tag: tag,
- }
- parent, err := current.GetParent()
- if err != nil {
- return err
- }
- if parent == nil {
- return nil
- }
- depGraph.NewNode(parent.ID)
- depGraph.AddDependency(current.ID, parent.ID)
- return nil
- })
- }
- traversalMap, err := depGraph.GenerateTraversalMap()
- if err != nil {
- return nil, err
- }
- utils.Debugf("Traversal map: %v", traversalMap)
- result := [][]*registry.ImgData{}
- for _, round := range traversalMap {
- dataRound := []*registry.ImgData{}
- for _, imgID := range round {
- dataRound = append(dataRound, imgList[imgID])
- }
- result = append(result, dataRound)
- }
- return result, nil
- }
- func flatten(slc [][]*registry.ImgData) []*registry.ImgData {
- result := []*registry.ImgData{}
- for _, x := range slc {
- result = append(result, x...)
- }
- return result
- }
- func (srv *Server) pushRepository(r *registry.Registry, out io.Writer, localName, remoteName string, localRepo map[string]string, indexEp string, sf *utils.StreamFormatter) error {
- out = utils.NewWriteFlusher(out)
- imgList, err := srv.getImageList(localRepo)
- if err != nil {
- return err
- }
- flattenedImgList := flatten(imgList)
- out.Write(sf.FormatStatus("", "Sending image list"))
- var repoData *registry.RepositoryData
- repoData, err = r.PushImageJSONIndex(indexEp, remoteName, flattenedImgList, false, nil)
- if err != nil {
- return err
- }
- for _, ep := range repoData.Endpoints {
- out.Write(sf.FormatStatus("", "Pushing repository %s (%d tags)", localName, len(localRepo)))
- // This section can not be parallelized (each round depends on the previous one)
- for _, round := range imgList {
- // FIXME: This section can be parallelized
- for _, elem := range round {
- var pushTags func() error
- pushTags = func() error {
- out.Write(sf.FormatStatus("", "Pushing tags for rev [%s] on {%s}", elem.ID, ep+"repositories/"+remoteName+"/tags/"+elem.Tag))
- if err := r.PushRegistryTag(remoteName, elem.ID, elem.Tag, ep, repoData.Tokens); err != nil {
- return err
- }
- return nil
- }
- if _, exists := repoData.ImgList[elem.ID]; exists {
- if err := pushTags(); err != nil {
- return err
- }
- out.Write(sf.FormatStatus("", "Image %s already pushed, skipping", elem.ID))
- continue
- } else if r.LookupRemoteImage(elem.ID, ep, repoData.Tokens) {
- if err := pushTags(); err != nil {
- return err
- }
- out.Write(sf.FormatStatus("", "Image %s already pushed, skipping", elem.ID))
- continue
- }
- if checksum, err := srv.pushImage(r, out, remoteName, elem.ID, ep, repoData.Tokens, sf); err != nil {
- // FIXME: Continue on error?
- return err
- } else {
- elem.Checksum = checksum
- }
- if err := pushTags(); err != nil {
- return err
- }
- }
- }
- }
- if _, err := r.PushImageJSONIndex(indexEp, remoteName, flattenedImgList, true, repoData.Endpoints); err != nil {
- return err
- }
- return nil
- }
- func (srv *Server) pushImage(r *registry.Registry, out io.Writer, remote, imgID, ep string, token []string, sf *utils.StreamFormatter) (checksum string, err error) {
- out = utils.NewWriteFlusher(out)
- jsonRaw, err := ioutil.ReadFile(path.Join(srv.runtime.graph.Root, imgID, "json"))
- if err != nil {
- return "", fmt.Errorf("Cannot retrieve the path for {%s}: %s", imgID, err)
- }
- out.Write(sf.FormatStatus("", "Pushing %s", imgID))
- imgData := ®istry.ImgData{
- ID: imgID,
- }
- // Send the json
- if err := r.PushImageJSONRegistry(imgData, jsonRaw, ep, token); err != nil {
- if err == registry.ErrAlreadyExists {
- out.Write(sf.FormatStatus("", "Image %s already pushed, skipping", imgData.ID))
- return "", nil
- }
- return "", err
- }
- layerData, err := srv.runtime.graph.TempLayerArchive(imgID, archive.Uncompressed, sf, out)
- if err != nil {
- return "", fmt.Errorf("Failed to generate layer archive: %s", err)
- }
- // Send the layer
- if checksum, err := r.PushImageLayerRegistry(imgData.ID, utils.ProgressReader(layerData, int(layerData.Size), out, sf.FormatProgress("", "Pushing", "%8v/%v (%v)"), sf, false), ep, token, jsonRaw); err != nil {
- return "", err
- } else {
- imgData.Checksum = checksum
- }
- out.Write(sf.FormatStatus("", ""))
- // Send the checksum
- if err := r.PushImageChecksumRegistry(imgData, ep, token); err != nil {
- return "", err
- }
- return imgData.Checksum, nil
- }
- // FIXME: Allow to interrupt current push when new push of same image is done.
- func (srv *Server) ImagePush(localName string, out io.Writer, sf *utils.StreamFormatter, authConfig *auth.AuthConfig, metaHeaders map[string][]string) error {
- if err := srv.poolAdd("push", localName); err != nil {
- return err
- }
- defer srv.poolRemove("push", localName)
- // Resolve the Repository name from fqn to endpoint + name
- endpoint, remoteName, err := registry.ResolveRepositoryName(localName)
- if err != nil {
- return err
- }
- out = utils.NewWriteFlusher(out)
- img, err := srv.runtime.graph.Get(localName)
- r, err2 := registry.NewRegistry(srv.runtime.config.Root, authConfig, srv.HTTPRequestFactory(metaHeaders))
- if err2 != nil {
- return err2
- }
- if err != nil {
- reposLen := len(srv.runtime.repositories.Repositories[localName])
- out.Write(sf.FormatStatus("", "The push refers to a repository [%s] (len: %d)", localName, reposLen))
- // If it fails, try to get the repository
- if localRepo, exists := srv.runtime.repositories.Repositories[localName]; exists {
- if err := srv.pushRepository(r, out, localName, remoteName, localRepo, endpoint, sf); err != nil {
- return err
- }
- return nil
- }
- return err
- }
- var token []string
- out.Write(sf.FormatStatus("", "The push refers to an image: [%s]", localName))
- if _, err := srv.pushImage(r, out, remoteName, img.ID, endpoint, token, sf); err != nil {
- return err
- }
- return nil
- }
- func (srv *Server) ImageImport(src, repo, tag string, in io.Reader, out io.Writer, sf *utils.StreamFormatter) error {
- var archive io.Reader
- var resp *http.Response
- if src == "-" {
- archive = in
- } else {
- u, err := url.Parse(src)
- if err != nil {
- return err
- }
- if u.Scheme == "" {
- u.Scheme = "http"
- u.Host = src
- u.Path = ""
- }
- out.Write(sf.FormatStatus("", "Downloading from %s", u))
- // Download with curl (pretty progress bar)
- // If curl is not available, fallback to http.Get()
- resp, err = utils.Download(u.String(), out)
- if err != nil {
- return err
- }
- archive = utils.ProgressReader(resp.Body, int(resp.ContentLength), out, sf.FormatProgress("", "Importing", "%8v/%v (%v)"), sf, true)
- }
- img, err := srv.runtime.graph.Create(archive, nil, "Imported from "+src, "", nil)
- if err != nil {
- return err
- }
- // Optionally register the image at REPO/TAG
- if repo != "" {
- if err := srv.runtime.repositories.Set(repo, tag, img.ID, true); err != nil {
- return err
- }
- }
- out.Write(sf.FormatStatus("", img.ID))
- return nil
- }
- func (srv *Server) ContainerCreate(job *engine.Job) string {
- var name string
- if len(job.Args) == 1 {
- name = job.Args[0]
- } else if len(job.Args) > 1 {
- return fmt.Sprintf("Usage: %s ", job.Name)
- }
- var config Config
- if err := job.ExportEnv(&config); err != nil {
- return err.Error()
- }
- if config.Memory != 0 && config.Memory < 524288 {
- return "Minimum memory limit allowed is 512k"
- }
- if config.Memory > 0 && !srv.runtime.capabilities.MemoryLimit {
- config.Memory = 0
- }
- if config.Memory > 0 && !srv.runtime.capabilities.SwapLimit {
- config.MemorySwap = -1
- }
- container, buildWarnings, err := srv.runtime.Create(&config, name)
- if err != nil {
- if srv.runtime.graph.IsNotExist(err) {
- _, tag := utils.ParseRepositoryTag(config.Image)
- if tag == "" {
- tag = DEFAULTTAG
- }
- return fmt.Sprintf("No such image: %s (tag: %s)", config.Image, tag)
- }
- return err.Error()
- }
- srv.LogEvent("create", container.ID, srv.runtime.repositories.ImageName(container.Image))
- // FIXME: this is necessary because runtime.Create might return a nil container
- // with a non-nil error. This should not happen! Once it's fixed we
- // can remove this workaround.
- if container != nil {
- job.Printf("%s\n", container.ID)
- }
- for _, warning := range buildWarnings {
- job.Errorf("%s\n", warning)
- }
- return "0"
- }
- func (srv *Server) ContainerRestart(name string, t int) error {
- if container := srv.runtime.Get(name); container != nil {
- if err := container.Restart(t); err != nil {
- return fmt.Errorf("Cannot restart container %s: %s", name, err)
- }
- srv.LogEvent("restart", container.ID, srv.runtime.repositories.ImageName(container.Image))
- } else {
- return fmt.Errorf("No such container: %s", name)
- }
- return nil
- }
- func (srv *Server) ContainerDestroy(name string, removeVolume, removeLink bool) error {
- container := srv.runtime.Get(name)
- if removeLink {
- if container == nil {
- return fmt.Errorf("No such link: %s", name)
- }
- name, err := srv.runtime.getFullName(name)
- if err != nil {
- return err
- }
- parent, n := path.Split(name)
- if parent == "/" {
- return fmt.Errorf("Conflict, cannot remove the default name of the container")
- }
- pe := srv.runtime.containerGraph.Get(parent)
- if pe == nil {
- return fmt.Errorf("Cannot get parent %s for name %s", parent, name)
- }
- parentContainer := srv.runtime.Get(pe.ID())
- if parentContainer != nil && parentContainer.activeLinks != nil {
- if link, exists := parentContainer.activeLinks[n]; exists {
- link.Disable()
- } else {
- utils.Debugf("Could not find active link for %s", name)
- }
- }
- if err := srv.runtime.containerGraph.Delete(name); err != nil {
- return err
- }
- return nil
- }
- if container != nil {
- if container.State.Running {
- return fmt.Errorf("Impossible to remove a running container, please stop it first")
- }
- volumes := make(map[string]struct{})
- // Store all the deleted containers volumes
- for _, volumeId := range container.Volumes {
- volumeId = strings.TrimRight(volumeId, "/layer")
- volumeId = filepath.Base(volumeId)
- volumes[volumeId] = struct{}{}
- }
- if err := srv.runtime.Destroy(container); err != nil {
- return fmt.Errorf("Cannot destroy container %s: %s", name, err)
- }
- srv.LogEvent("destroy", container.ID, srv.runtime.repositories.ImageName(container.Image))
- if removeVolume {
- // Retrieve all volumes from all remaining containers
- usedVolumes := make(map[string]*Container)
- for _, container := range srv.runtime.List() {
- for _, containerVolumeId := range container.Volumes {
- usedVolumes[containerVolumeId] = container
- }
- }
- for volumeId := range volumes {
- // If the requested volu
- if c, exists := usedVolumes[volumeId]; exists {
- log.Printf("The volume %s is used by the container %s. Impossible to remove it. Skipping.\n", volumeId, c.ID)
- continue
- }
- if err := srv.runtime.volumes.Delete(volumeId); err != nil {
- return err
- }
- }
- }
- } else {
- return fmt.Errorf("No such container: %s", name)
- }
- return nil
- }
- var ErrImageReferenced = errors.New("Image referenced by a repository")
- func (srv *Server) deleteImageAndChildren(id string, imgs *[]APIRmi) error {
- // If the image is referenced by a repo, do not delete
- if len(srv.runtime.repositories.ByID()[id]) != 0 {
- return ErrImageReferenced
- }
- // If the image is not referenced but has children, go recursive
- referenced := false
- byParents, err := srv.runtime.graph.ByParent()
- if err != nil {
- return err
- }
- for _, img := range byParents[id] {
- if err := srv.deleteImageAndChildren(img.ID, imgs); err != nil {
- if err != ErrImageReferenced {
- return err
- }
- referenced = true
- }
- }
- if referenced {
- return ErrImageReferenced
- }
- // If the image is not referenced and has no children, remove it
- byParents, err = srv.runtime.graph.ByParent()
- if err != nil {
- return err
- }
- if len(byParents[id]) == 0 {
- if err := srv.runtime.repositories.DeleteAll(id); err != nil {
- return err
- }
- err := srv.runtime.graph.Delete(id)
- if err != nil {
- return err
- }
- *imgs = append(*imgs, APIRmi{Deleted: utils.TruncateID(id)})
- srv.LogEvent("delete", utils.TruncateID(id), "")
- return nil
- }
- return nil
- }
- func (srv *Server) deleteImageParents(img *Image, imgs *[]APIRmi) error {
- if img.Parent != "" {
- parent, err := srv.runtime.graph.Get(img.Parent)
- if err != nil {
- return err
- }
- // Remove all children images
- if err := srv.deleteImageAndChildren(img.Parent, imgs); err != nil {
- return err
- }
- return srv.deleteImageParents(parent, imgs)
- }
- return nil
- }
- func (srv *Server) deleteImage(img *Image, repoName, tag string) ([]APIRmi, error) {
- imgs := []APIRmi{}
- tags := []string{}
- //If delete by id, see if the id belong only to one repository
- if repoName == "" {
- for _, repoAndTag := range srv.runtime.repositories.ByID()[img.ID] {
- parsedRepo, parsedTag := utils.ParseRepositoryTag(repoAndTag)
- if repoName == "" || repoName == parsedRepo {
- repoName = parsedRepo
- if parsedTag != "" {
- tags = append(tags, parsedTag)
- }
- } else if repoName != parsedRepo {
- // the id belongs to multiple repos, like base:latest and user:test,
- // in that case return conflict
- return imgs, nil
- }
- }
- } else {
- tags = append(tags, tag)
- }
- //Untag the current image
- for _, tag := range tags {
- tagDeleted, err := srv.runtime.repositories.Delete(repoName, tag)
- if err != nil {
- return nil, err
- }
- if tagDeleted {
- imgs = append(imgs, APIRmi{Untagged: img.ID})
- srv.LogEvent("untag", img.ID, "")
- }
- }
- if len(srv.runtime.repositories.ByID()[img.ID]) == 0 {
- if err := srv.deleteImageAndChildren(img.ID, &imgs); err != nil {
- if err != ErrImageReferenced {
- return imgs, err
- }
- } else if err := srv.deleteImageParents(img, &imgs); err != nil {
- if err != ErrImageReferenced {
- return imgs, err
- }
- }
- }
- return imgs, nil
- }
- func (srv *Server) ImageDelete(name string, autoPrune bool) ([]APIRmi, error) {
- img, err := srv.runtime.repositories.LookupImage(name)
- if err != nil {
- return nil, fmt.Errorf("No such image: %s", name)
- }
- if !autoPrune {
- if err := srv.runtime.graph.Delete(img.ID); err != nil {
- return nil, fmt.Errorf("Cannot delete image %s: %s", name, err)
- }
- return nil, nil
- }
- // Prevent deletion if image is used by a running container
- for _, container := range srv.runtime.List() {
- if container.State.Running {
- parent, err := srv.runtime.repositories.LookupImage(container.Image)
- if err != nil {
- return nil, err
- }
- if err := parent.WalkHistory(func(p *Image) error {
- if img.ID == p.ID {
- return fmt.Errorf("Conflict, cannot delete %s because the running container %s is using it", name, container.ID)
- }
- return nil
- }); err != nil {
- return nil, err
- }
- }
- }
- if strings.Contains(img.ID, name) {
- //delete via ID
- return srv.deleteImage(img, "", "")
- }
- name, tag := utils.ParseRepositoryTag(name)
- return srv.deleteImage(img, name, tag)
- }
- func (srv *Server) ImageGetCached(imgID string, config *Config) (*Image, error) {
- // Retrieve all images
- images, err := srv.runtime.graph.Map()
- if err != nil {
- return nil, err
- }
- // Store the tree in a map of map (map[parentId][childId])
- imageMap := make(map[string]map[string]struct{})
- for _, img := range images {
- if _, exists := imageMap[img.Parent]; !exists {
- imageMap[img.Parent] = make(map[string]struct{})
- }
- imageMap[img.Parent][img.ID] = struct{}{}
- }
- // Loop on the children of the given image and check the config
- for elem := range imageMap[imgID] {
- img, err := srv.runtime.graph.Get(elem)
- if err != nil {
- return nil, err
- }
- if CompareConfig(&img.ContainerConfig, config) {
- return img, nil
- }
- }
- return nil, nil
- }
- func (srv *Server) RegisterLinks(name string, hostConfig *HostConfig) error {
- runtime := srv.runtime
- container := runtime.Get(name)
- if container == nil {
- return fmt.Errorf("No such container: %s", name)
- }
- if hostConfig != nil && hostConfig.Links != nil {
- for _, l := range hostConfig.Links {
- parts, err := parseLink(l)
- if err != nil {
- return err
- }
- child, err := srv.runtime.GetByName(parts["name"])
- if err != nil {
- return err
- }
- if child == nil {
- return fmt.Errorf("Could not get container for %s", parts["name"])
- }
- if err := runtime.RegisterLink(container, child, parts["alias"]); err != nil {
- return err
- }
- }
- // After we load all the links into the runtime
- // set them to nil on the hostconfig
- hostConfig.Links = nil
- if err := container.writeHostConfig(); err != nil {
- return err
- }
- }
- return nil
- }
- func (srv *Server) ContainerStart(job *engine.Job) string {
- if len(job.Args) < 1 {
- return fmt.Sprintf("Usage: %s container_id", job.Name)
- }
- name := job.Args[0]
- runtime := srv.runtime
- container := runtime.Get(name)
- if container == nil {
- return fmt.Sprintf("No such container: %s", name)
- }
- // If no environment was set, then no hostconfig was passed.
- if len(job.Environ()) > 0 {
- var hostConfig HostConfig
- if err := job.ExportEnv(&hostConfig); err != nil {
- return err.Error()
- }
- // Validate the HostConfig binds. Make sure that:
- // 1) the source of a bind mount isn't /
- // The bind mount "/:/foo" isn't allowed.
- // 2) Check that the source exists
- // The source to be bind mounted must exist.
- for _, bind := range hostConfig.Binds {
- splitBind := strings.Split(bind, ":")
- source := splitBind[0]
- // refuse to bind mount "/" to the container
- if source == "/" {
- return fmt.Sprintf("Invalid bind mount '%s' : source can't be '/'", bind)
- }
- // ensure the source exists on the host
- _, err := os.Stat(source)
- if err != nil && os.IsNotExist(err) {
- return fmt.Sprintf("Invalid bind mount '%s' : source doesn't exist", bind)
- }
- }
- // Register any links from the host config before starting the container
- // FIXME: we could just pass the container here, no need to lookup by name again.
- if err := srv.RegisterLinks(name, &hostConfig); err != nil {
- return err.Error()
- }
- container.hostConfig = &hostConfig
- container.ToDisk()
- }
- if err := container.Start(); err != nil {
- return fmt.Sprintf("Cannot start container %s: %s", name, err)
- }
- srv.LogEvent("start", container.ID, runtime.repositories.ImageName(container.Image))
- return "0"
- }
- func (srv *Server) ContainerStop(name string, t int) error {
- if container := srv.runtime.Get(name); container != nil {
- if err := container.Stop(t); err != nil {
- return fmt.Errorf("Cannot stop container %s: %s", name, err)
- }
- srv.LogEvent("stop", container.ID, srv.runtime.repositories.ImageName(container.Image))
- } else {
- return fmt.Errorf("No such container: %s", name)
- }
- return nil
- }
- func (srv *Server) ContainerWait(name string) (int, error) {
- if container := srv.runtime.Get(name); container != nil {
- return container.Wait(), nil
- }
- return 0, fmt.Errorf("No such container: %s", name)
- }
- func (srv *Server) ContainerResize(name string, h, w int) error {
- if container := srv.runtime.Get(name); container != nil {
- return container.Resize(h, w)
- }
- return fmt.Errorf("No such container: %s", name)
- }
- func (srv *Server) ContainerAttach(name string, logs, stream, stdin, stdout, stderr bool, inStream io.ReadCloser, outStream, errStream io.Writer) error {
- container := srv.runtime.Get(name)
- if container == nil {
- return fmt.Errorf("No such container: %s", name)
- }
- //logs
- if logs {
- cLog, err := container.ReadLog("json")
- if err != nil && os.IsNotExist(err) {
- // Legacy logs
- utils.Errorf("Old logs format")
- if stdout {
- cLog, err := container.ReadLog("stdout")
- if err != nil {
- utils.Errorf("Error reading logs (stdout): %s", err)
- } else if _, err := io.Copy(outStream, cLog); err != nil {
- utils.Errorf("Error streaming logs (stdout): %s", err)
- }
- }
- if stderr {
- cLog, err := container.ReadLog("stderr")
- if err != nil {
- utils.Errorf("Error reading logs (stderr): %s", err)
- } else if _, err := io.Copy(errStream, cLog); err != nil {
- utils.Errorf("Error streaming logs (stderr): %s", err)
- }
- }
- } else if err != nil {
- utils.Errorf("Error reading logs (json): %s", err)
- } else {
- dec := json.NewDecoder(cLog)
- for {
- l := &utils.JSONLog{}
- if err := dec.Decode(l); err == io.EOF {
- break
- } else if err != nil {
- utils.Errorf("Error streaming logs: %s", err)
- break
- }
- if l.Stream == "stdout" && stdout {
- fmt.Fprintf(outStream, "%s", l.Log)
- }
- if l.Stream == "stderr" && stderr {
- fmt.Fprintf(errStream, "%s", l.Log)
- }
- }
- }
- }
- //stream
- if stream {
- if container.State.Ghost {
- return fmt.Errorf("Impossible to attach to a ghost container")
- }
- var (
- cStdin io.ReadCloser
- cStdout, cStderr io.Writer
- cStdinCloser io.Closer
- )
- if stdin {
- r, w := io.Pipe()
- go func() {
- defer w.Close()
- defer utils.Debugf("Closing buffered stdin pipe")
- io.Copy(w, inStream)
- }()
- cStdin = r
- cStdinCloser = inStream
- }
- if stdout {
- cStdout = outStream
- }
- if stderr {
- cStderr = errStream
- }
- <-container.Attach(cStdin, cStdinCloser, cStdout, cStderr)
- // If we are in stdinonce mode, wait for the process to end
- // otherwise, simply return
- if container.Config.StdinOnce && !container.Config.Tty {
- container.Wait()
- }
- }
- return nil
- }
- func (srv *Server) ContainerInspect(name string) (*Container, error) {
- if container := srv.runtime.Get(name); container != nil {
- return container, nil
- }
- return nil, fmt.Errorf("No such container: %s", name)
- }
- func (srv *Server) ImageInspect(name string) (*Image, error) {
- if image, err := srv.runtime.repositories.LookupImage(name); err == nil && image != nil {
- return image, nil
- }
- return nil, fmt.Errorf("No such image: %s", name)
- }
- func (srv *Server) ContainerCopy(name string, resource string, out io.Writer) error {
- if container := srv.runtime.Get(name); container != nil {
- data, err := container.Copy(resource)
- if err != nil {
- return err
- }
- if _, err := io.Copy(out, data); err != nil {
- return err
- }
- return nil
- }
- return fmt.Errorf("No such container: %s", name)
- }
- func NewServer(eng *engine.Engine, config *DaemonConfig) (*Server, error) {
- runtime, err := NewRuntime(config)
- if err != nil {
- return nil, err
- }
- srv := &Server{
- Eng: eng,
- runtime: runtime,
- pullingPool: make(map[string]struct{}),
- pushingPool: make(map[string]struct{}),
- events: make([]utils.JSONMessage, 0, 64), //only keeps the 64 last events
- listeners: make(map[string]chan utils.JSONMessage),
- reqFactory: nil,
- }
- runtime.srv = srv
- return srv, nil
- }
- func (srv *Server) HTTPRequestFactory(metaHeaders map[string][]string) *utils.HTTPRequestFactory {
- if srv.reqFactory == nil {
- ud := utils.NewHTTPUserAgentDecorator(srv.versionInfos()...)
- md := &utils.HTTPMetaHeadersDecorator{
- Headers: metaHeaders,
- }
- factory := utils.NewHTTPRequestFactory(ud, md)
- srv.reqFactory = factory
- }
- return srv.reqFactory
- }
- func (srv *Server) LogEvent(action, id, from string) *utils.JSONMessage {
- now := time.Now().Unix()
- jm := utils.JSONMessage{Status: action, ID: id, From: from, Time: now}
- srv.events = append(srv.events, jm)
- for _, c := range srv.listeners {
- select { // non blocking channel
- case c <- jm:
- default:
- }
- }
- return &jm
- }
- type Server struct {
- sync.Mutex
- runtime *Runtime
- pullingPool map[string]struct{}
- pushingPool map[string]struct{}
- events []utils.JSONMessage
- listeners map[string]chan utils.JSONMessage
- reqFactory *utils.HTTPRequestFactory
- Eng *engine.Engine
- }
|