func registerPod(p *Pod, ip net.IP) error {
uuid := p.UUID.String()
cmf, err := os.Open(common.PodManifestPath(p.Root))
if err != nil {
return fmt.Errorf("failed opening runtime manifest: %v", err)
}
defer cmf.Close()
pth := fmt.Sprintf("/pods/%v?ip=%v", uuid, ip.To4().String())
if err := httpRequest("PUT", pth, cmf); err != nil {
return fmt.Errorf("failed to register pod with metadata svc: %v", err)
}
for _, app := range p.Manifest.Apps {
ampath := common.ImageManifestPath(p.Root, app.Image.ID)
amf, err := os.Open(ampath)
if err != nil {
fmt.Errorf("failed reading app manifest %q: %v", ampath, err)
}
defer amf.Close()
if err := registerApp(uuid, app.Name.String(), amf); err != nil {
fmt.Errorf("failed to register app with metadata svc: %v", err)
}
}
return nil
}
// LoadPod loads a Pod Manifest (as prepared by stage0), the runtime data, and
// its associated Application Manifests, under $root/stage1/opt/stage1/$apphash
func LoadPod(root string, uuid *types.UUID, rp *RuntimePod) (*Pod, error) {
p := &Pod{
Root: root,
UUID: *uuid,
Images: make(map[string]*schema.ImageManifest),
UidRange: *user.NewBlankUidRange(),
}
// Unserialize runtime parameters
if rp != nil {
p.RuntimePod = *rp
} else {
buf, err := ioutil.ReadFile(filepath.Join(p.Root, RuntimeConfigPath))
if err != nil {
return nil, errwrap.Wrap(errors.New("failed reading runtime params"), err)
}
if err := json.Unmarshal(buf, &p.RuntimePod); err != nil {
return nil, errwrap.Wrap(errors.New("failed unmarshalling runtime params"), err)
}
}
buf, err := ioutil.ReadFile(common.PodManifestPath(p.Root))
if err != nil {
return nil, errwrap.Wrap(errors.New("failed reading pod manifest"), err)
}
pm := &schema.PodManifest{}
if err := json.Unmarshal(buf, pm); err != nil {
return nil, errwrap.Wrap(errors.New("failed unmarshalling pod manifest"), err)
}
p.Manifest = pm
for i, app := range p.Manifest.Apps {
impath := common.ImageManifestPath(p.Root, app.Name)
buf, err := ioutil.ReadFile(impath)
if err != nil {
return nil, errwrap.Wrap(fmt.Errorf("failed reading image manifest %q", impath), err)
}
im := &schema.ImageManifest{}
if err = json.Unmarshal(buf, im); err != nil {
return nil, errwrap.Wrap(fmt.Errorf("failed unmarshalling image manifest %q", impath), err)
}
if _, ok := p.Images[app.Name.String()]; ok {
return nil, fmt.Errorf("got multiple definitions for app: %v", app.Name)
}
if app.App == nil {
p.Manifest.Apps[i].App = im.App
}
p.Images[app.Name.String()] = im
}
if err := p.UidRange.Deserialize([]byte(p.PrivateUsers)); err != nil {
return nil, err
}
return p, nil
}
// registerPod registers pod with metadata service.
// Returns authentication token to be passed in the URL
func registerPod(root string, uuid *types.UUID, apps schema.AppList) (token string, rerr error) {
u := uuid.String()
var err error
token, err = generateMDSToken()
if err != nil {
rerr = errwrap.Wrap(errors.New("failed to generate MDS token"), err)
return
}
pmfPath := common.PodManifestPath(root)
pmf, err := os.Open(pmfPath)
if err != nil {
rerr = errwrap.Wrap(fmt.Errorf("failed to open runtime manifest (%v)", pmfPath), err)
return
}
pth := fmt.Sprintf("/pods/%v?token=%v", u, token)
err = httpRequest("PUT", pth, pmf)
pmf.Close()
if err != nil {
rerr = errwrap.Wrap(errors.New("failed to register pod with metadata svc"), err)
return
}
defer func() {
if rerr != nil {
unregisterPod(root, uuid)
}
}()
rf, err := os.Create(filepath.Join(root, mdsRegisteredFile))
if err != nil {
rerr = errwrap.Wrap(errors.New("failed to create mds-register file"), err)
return
}
rf.Close()
for _, app := range apps {
ampath := common.ImageManifestPath(root, app.Name)
amf, err := os.Open(ampath)
if err != nil {
rerr = errwrap.Wrap(fmt.Errorf("failed reading app manifest %q", ampath), err)
return
}
err = registerApp(u, app.Name.String(), amf)
amf.Close()
if err != nil {
rerr = errwrap.Wrap(errors.New("failed to register app with metadata svc"), err)
return
}
}
return
}
// registerPod registers pod with metadata service.
// Returns authentication token to be passed in the URL
func registerPod(root string, uuid *types.UUID, apps schema.AppList) (token string, rerr error) {
u := uuid.String()
var err error
token, err = generateMDSToken()
if err != nil {
rerr = fmt.Errorf("failed to generate MDS token: %v", err)
return
}
pmfPath := common.PodManifestPath(root)
pmf, err := os.Open(pmfPath)
if err != nil {
rerr = fmt.Errorf("failed to open runtime manifest (%v): %v", pmfPath, err)
return
}
pth := fmt.Sprintf("/pods/%v?token=%v", u, token)
err = httpRequest("PUT", pth, pmf)
pmf.Close()
if err != nil {
rerr = fmt.Errorf("failed to register pod with metadata svc: %v", err)
return
}
defer func() {
if rerr != nil {
unregisterPod(uuid)
}
}()
for _, app := range apps {
ampath := common.ImageManifestPath(root, app.Image.ID)
amf, err := os.Open(ampath)
if err != nil {
rerr = fmt.Errorf("failed reading app manifest %q: %v", ampath, err)
return
}
err = registerApp(u, app.Name.String(), amf)
amf.Close()
if err != nil {
rerr = fmt.Errorf("failed to register app with metadata svc: %v", err)
return
}
}
return
}
// LoadPod loads a Pod Manifest (as prepared by stage0) and
// its associated Application Manifests, under $root/stage1/opt/stage1/$apphash
func LoadPod(root string, uuid *types.UUID) (*Pod, error) {
p := &Pod{
Root: root,
UUID: *uuid,
Apps: make(map[string]*schema.ImageManifest),
}
buf, err := ioutil.ReadFile(common.PodManifestPath(p.Root))
if err != nil {
return nil, fmt.Errorf("failed reading pod manifest: %v", err)
}
pm := &schema.PodManifest{}
if err := json.Unmarshal(buf, pm); err != nil {
return nil, fmt.Errorf("failed unmarshalling pod manifest: %v", err)
}
p.Manifest = pm
for i, app := range p.Manifest.Apps {
ampath := common.ImageManifestPath(p.Root, app.Image.ID)
buf, err := ioutil.ReadFile(ampath)
if err != nil {
return nil, fmt.Errorf("failed reading app manifest %q: %v", ampath, err)
}
am := &schema.ImageManifest{}
if err = json.Unmarshal(buf, am); err != nil {
return nil, fmt.Errorf("failed unmarshalling app manifest %q: %v", ampath, err)
}
name := am.Name.String()
if _, ok := p.Apps[name]; ok {
return nil, fmt.Errorf("got multiple definitions for app: %s", name)
}
if app.App == nil {
p.Manifest.Apps[i].App = am.App
}
p.Apps[name] = am
}
return p, nil
}
func registerPod(p *Pod, token string) (rerr error) {
uuid := p.UUID.String()
cmf, err := os.Open(common.PodManifestPath(p.Root))
if err != nil {
rerr = fmt.Errorf("failed opening runtime manifest: %v", err)
return
}
pth := fmt.Sprintf("/pods/%v?token=%v", uuid, token)
err = httpRequest("PUT", pth, cmf)
cmf.Close()
if err != nil {
rerr = fmt.Errorf("failed to register pod with metadata svc: %v", err)
return
}
defer func() {
if rerr != nil {
unregisterPod(p)
}
}()
for _, app := range p.Manifest.Apps {
ampath := common.ImageManifestPath(p.Root, app.Image.ID)
amf, err := os.Open(ampath)
if err != nil {
rerr = fmt.Errorf("failed reading app manifest %q: %v", ampath, err)
return
}
err = registerApp(uuid, app.Name.String(), amf)
amf.Close()
if err != nil {
rerr = fmt.Errorf("failed to register app with metadata svc: %v", err)
return
}
}
return nil
}
// LoadPod loads a Pod Manifest (as prepared by stage0) and
// its associated Application Manifests, under $root/stage1/opt/stage1/$apphash
func LoadPod(root string, uuid *types.UUID) (*Pod, error) {
p := &Pod{
Root: root,
UUID: *uuid,
Images: make(map[string]*schema.ImageManifest),
}
buf, err := ioutil.ReadFile(common.PodManifestPath(p.Root))
if err != nil {
return nil, errwrap.Wrap(errors.New("failed reading pod manifest"), err)
}
pm := &schema.PodManifest{}
if err := json.Unmarshal(buf, pm); err != nil {
return nil, errwrap.Wrap(errors.New("failed unmarshalling pod manifest"), err)
}
p.Manifest = pm
for i, app := range p.Manifest.Apps {
impath := common.ImageManifestPath(p.Root, app.Name)
buf, err := ioutil.ReadFile(impath)
if err != nil {
return nil, errwrap.Wrap(fmt.Errorf("failed reading image manifest %q", impath), err)
}
im := &schema.ImageManifest{}
if err = json.Unmarshal(buf, im); err != nil {
return nil, errwrap.Wrap(fmt.Errorf("failed unmarshalling image manifest %q", impath), err)
}
if _, ok := p.Images[app.Name.String()]; ok {
return nil, fmt.Errorf("got multiple definitions for app: %v", app.Name)
}
if app.App == nil {
p.Manifest.Apps[i].App = im.App
}
p.Images[app.Name.String()] = im
}
return p, nil
}
