mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
113 lines
2.6 KiB
Go
113 lines
2.6 KiB
Go
![]() |
package daemon
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/docker/docker/container"
|
||
|
"github.com/docker/docker/daemon/names"
|
||
|
"github.com/docker/docker/pkg/namesgenerator"
|
||
|
"github.com/docker/docker/pkg/stringid"
|
||
|
"github.com/pkg/errors"
|
||
|
"github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
validContainerNameChars = names.RestrictedNameChars
|
||
|
validContainerNamePattern = names.RestrictedNamePattern
|
||
|
)
|
||
|
|
||
|
func (daemon *Daemon) registerName(container *container.Container) error {
|
||
|
if daemon.Exists(container.ID) {
|
||
|
return fmt.Errorf("Container is already loaded")
|
||
|
}
|
||
|
if err := validateID(container.ID); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if container.Name == "" {
|
||
|
name, err := daemon.generateNewName(container.ID)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
container.Name = name
|
||
|
}
|
||
|
return daemon.containersReplica.ReserveName(container.Name, container.ID)
|
||
|
}
|
||
|
|
||
|
func (daemon *Daemon) generateIDAndName(name string) (string, string, error) {
|
||
|
var (
|
||
|
err error
|
||
|
id = stringid.GenerateNonCryptoID()
|
||
|
)
|
||
|
|
||
|
if name == "" {
|
||
|
if name, err = daemon.generateNewName(id); err != nil {
|
||
|
return "", "", err
|
||
|
}
|
||
|
return id, name, nil
|
||
|
}
|
||
|
|
||
|
if name, err = daemon.reserveName(id, name); err != nil {
|
||
|
return "", "", err
|
||
|
}
|
||
|
|
||
|
return id, name, nil
|
||
|
}
|
||
|
|
||
|
func (daemon *Daemon) reserveName(id, name string) (string, error) {
|
||
|
if !validContainerNamePattern.MatchString(strings.TrimPrefix(name, "/")) {
|
||
|
return "", validationError{errors.Errorf("Invalid container name (%s), only %s are allowed", name, validContainerNameChars)}
|
||
|
}
|
||
|
if name[0] != '/' {
|
||
|
name = "/" + name
|
||
|
}
|
||
|
|
||
|
if err := daemon.containersReplica.ReserveName(name, id); err != nil {
|
||
|
if err == container.ErrNameReserved {
|
||
|
id, err := daemon.containersReplica.Snapshot().GetID(name)
|
||
|
if err != nil {
|
||
|
logrus.Errorf("got unexpected error while looking up reserved name: %v", err)
|
||
|
return "", err
|
||
|
}
|
||
|
return "", nameConflictError{id: id, name: name}
|
||
|
}
|
||
|
return "", errors.Wrapf(err, "error reserving name: %q", name)
|
||
|
}
|
||
|
return name, nil
|
||
|
}
|
||
|
|
||
|
func (daemon *Daemon) releaseName(name string) {
|
||
|
daemon.containersReplica.ReleaseName(name)
|
||
|
}
|
||
|
|
||
|
func (daemon *Daemon) generateNewName(id string) (string, error) {
|
||
|
var name string
|
||
|
for i := 0; i < 6; i++ {
|
||
|
name = namesgenerator.GetRandomName(i)
|
||
|
if name[0] != '/' {
|
||
|
name = "/" + name
|
||
|
}
|
||
|
|
||
|
if err := daemon.containersReplica.ReserveName(name, id); err != nil {
|
||
|
if err == container.ErrNameReserved {
|
||
|
continue
|
||
|
}
|
||
|
return "", err
|
||
|
}
|
||
|
return name, nil
|
||
|
}
|
||
|
|
||
|
name = "/" + stringid.TruncateID(id)
|
||
|
if err := daemon.containersReplica.ReserveName(name, id); err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
return name, nil
|
||
|
}
|
||
|
|
||
|
func validateID(id string) error {
|
||
|
if id == "" {
|
||
|
return fmt.Errorf("Invalid empty id")
|
||
|
}
|
||
|
return nil
|
||
|
}
|