func (s *apiSuite) mockSnap(c *C, yamlText string) *snap.Info {
if s.d == nil {
panic("call s.daemon(c) in your test first")
}
st := s.d.overlord.State()
st.Lock()
defer st.Unlock()
// Parse the yaml
snapInfo, err := snap.InfoFromSnapYaml([]byte(yamlText))
c.Assert(err, IsNil)
snap.AddImplicitSlots(snapInfo)
// Create on-disk yaml file (it is read by snapstate)
dname := filepath.Join(dirs.SnapSnapsDir, snapInfo.Name(),
strconv.Itoa(snapInfo.Revision), "meta")
fname := filepath.Join(dname, "snap.yaml")
err = os.MkdirAll(dname, 0755)
c.Assert(err, IsNil)
err = ioutil.WriteFile(fname, []byte(yamlText), 0644)
c.Assert(err, IsNil)
// Put a side info into the state
snapstate.Set(st, snapInfo.Name(), &snapstate.SnapState{
Active: true,
Sequence: []*snap.SideInfo{{Revision: snapInfo.Revision}},
})
// Put the snap into the interface repository
repo := s.d.overlord.InterfaceManager().Repository()
err = repo.AddSnap(snapInfo)
c.Assert(err, IsNil)
return snapInfo
}
func (s *interfaceManagerSuite) mockSnap(c *C, yamlText string) *snap.Info {
s.state.Lock()
defer s.state.Unlock()
// Parse the yaml
snapInfo, err := snap.InfoFromSnapYaml([]byte(yamlText))
c.Assert(err, IsNil)
snap.AddImplicitSlots(snapInfo)
// Create on-disk yaml file (it is read by snapstate)
dname := filepath.Join(dirs.SnapSnapsDir, snapInfo.Name(),
strconv.Itoa(snapInfo.Revision), "meta")
fname := filepath.Join(dname, "snap.yaml")
err = os.MkdirAll(dname, 0755)
c.Assert(err, IsNil)
err = ioutil.WriteFile(fname, []byte(yamlText), 0644)
c.Assert(err, IsNil)
// Put a side info into the state
snapstate.Set(s.state, snapInfo.Name(), &snapstate.SnapState{
Active: true,
Sequence: []*snap.SideInfo{{Revision: snapInfo.Revision}},
})
return snapInfo
}
func (s *InfoSnapYamlTestSuite) TestAddImplicitSlots(c *C) {
osYaml := []byte("name: ubuntu-core\ntype: os\n")
info, err := snap.InfoFromSnapYaml(osYaml)
c.Assert(err, IsNil)
snap.AddImplicitSlots(info)
c.Assert(info.Slots["network"].Interface, Equals, "network")
c.Assert(info.Slots["network"].Name, Equals, "network")
c.Assert(info.Slots["network"].Snap, Equals, info)
c.Assert(info.Slots, HasLen, 16)
}
func (m *InterfaceManager) addSnaps() error {
snaps, err := snapstate.ActiveInfos(m.state)
if err != nil {
return err
}
for _, snapInfo := range snaps {
snap.AddImplicitSlots(snapInfo)
if err := m.repo.AddSnap(snapInfo); err != nil {
logger.Noticef("%s", err)
}
}
return nil
}
func (m *InterfaceManager) doSetupProfiles(task *state.Task, _ *tomb.Tomb) error {
task.State().Lock()
defer task.State().Unlock()
// Get snap.Info from bits handed by the snap manager.
ss, err := snapstate.TaskSnapSetup(task)
if err != nil {
return err
}
snapInfo, err := snapstate.Info(task.State(), ss.Name, ss.Revision)
if err != nil {
return err
}
snap.AddImplicitSlots(snapInfo)
snapName := snapInfo.Name()
var snapState snapstate.SnapState
if err := snapstate.Get(task.State(), snapName, &snapState); err != nil {
task.Errorf("cannot get state of snap %q: %s", snapName, err)
return err
}
// Set DevMode flag if SnapSetup.Flags indicates it should be done
// but remember the old value in the task in case we undo.
task.Set("old-devmode", snapState.DevMode())
if ss.DevMode() {
snapState.Flags |= snapstate.DevMode
} else {
snapState.Flags &= ^snapstate.DevMode
}
snapstate.Set(task.State(), snapName, &snapState)
// The snap may have been updated so perform the following operation to
// ensure that we are always working on the correct state:
//
// - disconnect all connections to/from the given snap
// - remembering the snaps that were affected by this operation
// - remove the (old) snap from the interfaces repository
// - add the (new) snap to the interfaces repository
// - restore connections based on what is kept in the state
// - if a connection cannot be restored then remove it from the state
// - setup the security of all the affected snaps
blacklist := m.repo.AutoConnectBlacklist(snapName)
affectedSnaps, err := m.repo.DisconnectSnap(snapName)
if err != nil {
return err
}
// XXX: what about snap renames? We should remove the old name (or switch
// to IDs in the interfaces repository)
if err := m.repo.RemoveSnap(snapName); err != nil {
return err
}
if err := m.repo.AddSnap(snapInfo); err != nil {
if _, ok := err.(*interfaces.BadInterfacesError); ok {
logger.Noticef("%s", err)
} else {
return err
}
}
if err := m.reloadConnections(snapName); err != nil {
return err
}
if err := m.autoConnect(task, snapName, blacklist); err != nil {
return err
}
if len(affectedSnaps) == 0 {
affectedSnaps = append(affectedSnaps, snapInfo)
}
for _, snapInfo := range affectedSnaps {
if err := setupSnapSecurity(task, snapInfo, m.repo); err != nil {
return state.Retry
}
}
return nil
}