| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648 |
- // Copyright 2019 Yunion
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package models
- import (
- "context"
- "fmt"
- "path"
- "strconv"
- "gopkg.in/fatih/set.v0"
- "yunion.io/x/jsonutils"
- "yunion.io/x/log"
- "yunion.io/x/pkg/errors"
- "yunion.io/x/pkg/utils"
- "yunion.io/x/sqlchemy"
- api "yunion.io/x/onecloud/pkg/apis/compute"
- "yunion.io/x/onecloud/pkg/cloudcommon/db"
- "yunion.io/x/onecloud/pkg/cloudcommon/validators"
- "yunion.io/x/onecloud/pkg/httperrors"
- "yunion.io/x/onecloud/pkg/mcclient"
- "yunion.io/x/onecloud/pkg/util/logclient"
- "yunion.io/x/onecloud/pkg/util/stringutils2"
- )
- // +onecloud:swagger-gen-model-singular=guestnetworksecgroup
- // +onecloud:swagger-gen-model-plural=guestnetworksecgroups
- type SGuestnetworksecgroupManager struct {
- db.SResourceBaseManager
- SGuestResourceBaseManager
- SSecurityGroupResourceBaseManager
- SCloudregionResourceBaseManager
- SManagedResourceBaseManager
- SVpcResourceBaseManager
- }
- var GuestnetworksecgroupManager *SGuestnetworksecgroupManager
- func init() {
- db.InitManager(func() {
- GuestnetworksecgroupManager = &SGuestnetworksecgroupManager{
- SResourceBaseManager: db.NewResourceBaseManager(
- SGuestnetworksecgroup{},
- "guestnetworksecgroups_tbl",
- "guestnetworksecgroup",
- "guestnetworksecgroups",
- ),
- }
- GuestnetworksecgroupManager.SetVirtualObject(GuestnetworksecgroupManager)
- })
- }
- // +onecloud:model-api-gen
- type SGuestnetworksecgroup struct {
- db.SResourceBase
- RowId int64 `primary:"true" auto_increment:"true" list:"user"`
- GuestId string `width:"36" charset:"ascii" nullable:"false" list:"user" create:"required" index:"true"`
- SSecurityGroupResourceBase `width:"36" charset:"ascii" nullable:"false" list:"user" create:"required"`
- NetworkIndex int `nullable:"false" list:"user" update:"user"`
- Admin bool `nullable:"false" default:"false" list:"user" create:"optional"`
- }
- func (manager *SGuestnetworksecgroupManager) GetSlaveFieldName() string {
- return "secgroup_id"
- }
- func (self *SGuestnetworksecgroup) Delete(ctx context.Context, userCred mcclient.TokenCredential) error {
- return db.DeleteModel(ctx, userCred, self)
- }
- func (manager *SGuestnetworksecgroupManager) ListItemFilter(
- ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- query api.GuestnetworksecgroupListInput,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SGuestResourceBaseManager.ListItemFilter(ctx, q, userCred, query.ServerFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SGuestJointsManager.ListItemFilter")
- }
- q, err = manager.SSecurityGroupResourceBaseManager.ListItemFilter(ctx, q, userCred, query.SecgroupFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SSecurityGroupResourceBaseManager.ListItemFilter")
- }
- if query.NetworkIndex != nil {
- q = q.Equals("network_index", *query.NetworkIndex)
- }
- if query.IsAdmin {
- q = q.IsFalse("admin")
- }
- return q, nil
- }
- func (manager *SGuestnetworksecgroupManager) OrderByExtraFields(
- ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- query api.GuestnetworksecgroupListInput,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SGuestResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.ServerFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SGuestJointsManager.OrderByExtraFields")
- }
- q, err = manager.SSecurityGroupResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.SecgroupFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SSecurityGroupResourceBaseManager.OrderByExtraFields")
- }
- return q, nil
- }
- func (manager *SGuestnetworksecgroupManager) ListItemExportKeys(ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- keys stringutils2.SSortedStrings,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SGuestResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
- if err != nil {
- return nil, errors.Wrap(err, "SGuestJointsManager.ListItemExportKeys")
- }
- if keys.ContainsAny(manager.SSecurityGroupResourceBaseManager.GetExportKeys()...) {
- q, err = manager.SSecurityGroupResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
- if err != nil {
- return nil, errors.Wrap(err, "SSecurityGroupResourceBaseManager.ListItemExportKeys")
- }
- }
- return q, nil
- }
- func (manager *SGuestnetworksecgroupManager) FetchCustomizeColumns(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- objs []interface{},
- fields stringutils2.SSortedStrings,
- isList bool,
- ) []api.GuestnetworksecgroupDetails {
- rows := make([]api.GuestnetworksecgroupDetails, len(objs))
- guestList := make([]interface{}, len(objs))
- guestIds := make([]string, len(objs))
- guestNetworkIdx := make([]int, len(objs))
- secgrpIds := make([]string, len(objs))
- for i := range objs {
- secgrpIds[i] = objs[i].(*SGuestnetworksecgroup).SecgroupId
- guestList[i] = &SGuestResourceBase{objs[i].(*SGuestnetworksecgroup).GuestId}
- guestIds[i] = objs[i].(*SGuestnetworksecgroup).GuestId
- guestNetworkIdx[i] = objs[i].(*SGuestnetworksecgroup).NetworkIndex
- }
- groups := make(map[string]SSecurityGroup)
- err := db.FetchStandaloneObjectsByIds(SecurityGroupManager, secgrpIds, groups)
- if err != nil {
- log.Errorf("FetchStandaloneObjectsByIds fail %s", err)
- return nil
- }
- guestRows := manager.SGuestResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, guestList, fields, isList)
- gns, err := GuestnetworkManager.fetchGuestnetworksByGuestnetworkIndex(guestIds, guestNetworkIdx)
- if err != nil {
- log.Errorf("failed fetchGuestnetworksByGuestnetworkIndex")
- return nil
- }
- regionList := make([]interface{}, len(objs))
- managerList := make([]interface{}, len(objs))
- vpcList := make([]interface{}, len(objs))
- projectList := make([]interface{}, len(objs))
- for i := range rows {
- secgroupId := objs[i].(*SGuestnetworksecgroup).SecgroupId
- guestId := objs[i].(*SGuestnetworksecgroup).GuestId
- networkIndex := objs[i].(*SGuestnetworksecgroup).NetworkIndex
- rows[i].GuestResourceInfo = guestRows[i]
- rows[i].NetworkIndex = networkIndex
- rows[i].Admin = objs[i].(*SGuestnetworksecgroup).Admin
- rows[i].GuestNetwork = path.Join(guestId, secgroupId, strconv.Itoa(rows[i].NetworkIndex))
- if group, ok := groups[secgrpIds[i]]; ok {
- rows[i].Secgroup = group.Name
- rows[i].CloudregionId = group.CloudregionId
- rows[i].ManagerId = group.ManagerId
- rows[i].VpcId = group.VpcId
- rows[i].SecgroupStatus = group.Status
- rows[i].ProjectId = group.ProjectId
- secgroup := group
- projectList[i] = &secgroup
- }
- regionList[i] = &SCloudregionResourceBase{rows[i].CloudregionId}
- managerList[i] = &SManagedResourceBase{rows[i].ManagerId}
- vpcList[i] = &SVpcResourceBase{rows[i].VpcId}
- key := fmt.Sprintf("%s/%d", guestId, networkIndex)
- if gn, ok := gns[key]; ok {
- rows[i].MacAddr = gn.MacAddr
- rows[i].IpAddr = gn.IpAddr
- rows[i].Ip6Addr = gn.Ip6Addr
- rows[i].Ifname = gn.Ifname
- if network, _ := gn.GetNetwork(); network != nil {
- rows[i].NetworkId = gn.NetworkId
- rows[i].NetworkName = network.Name
- }
- }
- }
- projRows := SecurityGroupManager.SProjectizedResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, projectList, fields, isList)
- regionRows := manager.SCloudregionResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, regionList, fields, isList)
- managerRows := manager.SManagedResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, managerList, fields, isList)
- vpcRows := manager.SVpcResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, vpcList, fields, isList)
- for i := range rows {
- rows[i].ProjectizedResourceInfo = projRows[i]
- rows[i].CloudregionResourceInfo = regionRows[i]
- rows[i].ManagedResourceInfo = managerRows[i]
- rows[i].Vpc = vpcRows[i].Vpc
- }
- return rows
- }
- func (manager *SGuestnetworkManager) fetchGuestnetworksByGuestnetworkIndex(guestIds []string, networkIndex []int) (map[string]SGuestnetwork, error) {
- q := manager.Query()
- q = q.In("guest_id", guestIds)
- q = q.In("index", networkIndex)
- gns := make([]SGuestnetwork, 0)
- err := q.All(&gns)
- if err != nil {
- return nil, err
- }
- res := map[string]SGuestnetwork{}
- for i := range gns {
- key := fmt.Sprintf("%s/%d", gns[i].GuestId, gns[i].Index)
- if _, ok := res[key]; !ok {
- res[key] = gns[i]
- }
- }
- return res, nil
- }
- func (manager *SGuestnetworksecgroupManager) QueryDistinctExtraField(q *sqlchemy.SQuery, field string) (*sqlchemy.SQuery, error) {
- q, err := manager.SResourceBaseManager.QueryDistinctExtraField(q, field)
- if err == nil {
- return q, nil
- }
- q, err = manager.SGuestResourceBaseManager.QueryDistinctExtraField(q, field)
- if err == nil {
- return q, nil
- }
- q, err = manager.SSecurityGroupResourceBaseManager.QueryDistinctExtraField(q, field)
- if err == nil {
- return q, nil
- }
- return q, httperrors.ErrNotFound
- }
- func (manager *SGuestnetworksecgroupManager) QueryDistinctExtraFields(q *sqlchemy.SQuery, resource string, fields []string) (*sqlchemy.SQuery, error) {
- q, err := manager.SResourceBaseManager.QueryDistinctExtraFields(q, resource, fields)
- if err == nil {
- return q, nil
- }
- q, err = manager.SGuestResourceBaseManager.QueryDistinctExtraFields(q, resource, fields)
- if err == nil {
- return q, nil
- }
- q, err = manager.SSecurityGroupResourceBaseManager.QueryDistinctExtraFields(q, resource, fields)
- if err == nil {
- return q, nil
- }
- return q, httperrors.ErrNotFound
- }
- func (manager *SGuestnetworksecgroupManager) GetGuestnetworksecgroups(guestId string, networkIndex int) ([]SSecurityGroup, error) {
- q := GuestnetworksecgroupManager.Query("secgroup_id").Equals("guest_id", guestId)
- if networkIndex >= 0 {
- q = q.Equals("network_index", networkIndex)
- }
- subQ := q.SubQuery()
- secgrpQuery := SecurityGroupManager.Query()
- secgrpQuery = secgrpQuery.In("id", subQ)
- secgroups := []SSecurityGroup{}
- err := db.FetchModelObjects(SecurityGroupManager, secgrpQuery, &secgroups)
- if err != nil {
- return nil, errors.Wrapf(err, "db.FetchModelObjects")
- }
- return secgroups, nil
- }
- // guest network attach secgroup
- func (self *SGuest) PerformAddNetworkSecgroup(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- input api.GuestNetworkAddSecgroupInput,
- ) (jsonutils.JSONObject, error) {
- if !utils.IsInStringArray(self.Status, []string{api.VM_READY, api.VM_RUNNING, api.VM_SUSPEND}) {
- return nil, httperrors.NewInputParameterError("Cannot add security groups in status %s", self.Status)
- }
- if input.NetworkIndex == nil || *input.NetworkIndex < 0 {
- return nil, httperrors.NewBadRequestError("input network index %#v is invalid", input.NetworkIndex)
- }
- driver, _ := self.GetDriver()
- maxCount := driver.GetMaxSecurityGroupCount()
- if maxCount == 0 {
- return nil, httperrors.NewUnsupportOperationError("Cannot add security groups for hypervisor %s", self.Hypervisor)
- }
- if len(input.SecgroupIds) == 0 {
- return nil, httperrors.NewMissingParameterError("secgroup_ids")
- }
- guestnetwork, err := self.getGuestnetworkByIndex(*input.NetworkIndex)
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "getGuestnetworkByIndex"))
- }
- secgroups, err := GuestnetworksecgroupManager.GetGuestnetworksecgroups(self.Id, *input.NetworkIndex)
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "GetGuestnetworksecgroups"))
- }
- if len(secgroups)+len(input.SecgroupIds) > maxCount {
- return nil, httperrors.NewUnsupportOperationError("guest %s band to up to %d security groups", self.Name, maxCount)
- }
- network, err := guestnetwork.GetNetwork()
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "GetNetwork"))
- }
- vpc, err := network.GetVpc()
- if err != nil {
- return nil, errors.Wrap(err, "GetVpc")
- }
- secgroupIds := []string{}
- for _, secgroup := range secgroups {
- secgroupIds = append(secgroupIds, secgroup.Id)
- }
- secgroupNames := []string{}
- for i := range input.SecgroupIds {
- secObj, err := validators.ValidateModel(ctx, userCred, SecurityGroupManager, &input.SecgroupIds[i])
- if err != nil {
- return nil, err
- }
- secgroup := secObj.(*SSecurityGroup)
- if utils.IsInStringArray(secObj.GetId(), secgroupIds) {
- return nil, httperrors.NewInputParameterError(
- "security group %s has already been assigned to guest %s network %d",
- secObj.GetName(), self.GetName(), input.NetworkIndex)
- }
- err = vpc.CheckSecurityGroupConsistent(secgroup)
- if err != nil {
- return nil, err
- }
- secgroupIds = append(secgroupIds, secgroup.GetId())
- secgroupNames = append(secgroupNames, secgroup.Name)
- }
- err = self.SaveNetworkSecgroups(ctx, userCred, secgroupIds, *input.NetworkIndex)
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "SaveNetworkSecgroups"))
- }
- notes := map[string][]string{"secgroups": secgroupNames}
- logclient.AddActionLogWithContext(ctx, self, logclient.ACT_VM_ASSIGNSECGROUP, notes, userCred, true)
- return nil, self.StartSyncTask(ctx, userCred, true, "")
- }
- func (self *SGuest) PerformRevokeNetworkSecgroup(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- input api.GuestRevokeNetworkSecgroupInput,
- ) (jsonutils.JSONObject, error) {
- if !utils.IsInStringArray(self.Status, []string{api.VM_READY, api.VM_RUNNING, api.VM_SUSPEND}) {
- return nil, httperrors.NewInputParameterError("Cannot revoke security groups in status %s", self.Status)
- }
- if len(input.SecgroupIds) == 0 {
- return nil, nil
- }
- var guestnetwork *SGuestnetwork
- var err error
- if input.MacAddr != "" {
- guestnetwork, err = self.GetGuestnetworkByMac(input.MacAddr)
- } else if input.NetworkIndex != nil {
- guestnetwork, err = self.getGuestnetworkByIndex(*input.NetworkIndex)
- } else {
- return nil, httperrors.NewBadRequestError("no valid network index or mac addr")
- }
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "get guest network"))
- }
- secgroups, err := GuestnetworksecgroupManager.GetGuestnetworksecgroups(self.Id, guestnetwork.Index)
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "GetGuestnetworksecgroups"))
- }
- secgroupMaps := map[string]string{}
- for _, secgroup := range secgroups {
- secgroupMaps[secgroup.Id] = secgroup.Name
- }
- secgroupNames := []string{}
- for i := range input.SecgroupIds {
- secObj, err := validators.ValidateModel(ctx, userCred, SecurityGroupManager, &input.SecgroupIds[i])
- if err != nil {
- return nil, err
- }
- secgrp := secObj.(*SSecurityGroup)
- _, ok := secgroupMaps[secgrp.GetId()]
- if !ok {
- return nil, httperrors.NewInputParameterError("security group %s network index %d not assigned to guest %s",
- secgrp.GetName(), guestnetwork.Index, self.GetName())
- }
- delete(secgroupMaps, secgrp.GetId())
- secgroupNames = append(secgroupNames, secgrp.GetName())
- }
- secgrpIds := []string{}
- for secgroupId := range secgroupMaps {
- secgrpIds = append(secgrpIds, secgroupId)
- }
- err = self.SaveNetworkSecgroups(ctx, userCred, secgrpIds, guestnetwork.Index)
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "SaveNetworkSecgroups"))
- }
- notes := map[string][]string{"secgroups": secgroupNames}
- logclient.AddActionLogWithContext(ctx, self, logclient.ACT_VM_REVOKESECGROUP, notes, userCred, true)
- return nil, self.StartSyncTask(ctx, userCred, true, "")
- }
- func (self *SGuest) PerformSetNetworkSecgroup(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- input api.GuestSetNetworkSecgroupInput,
- ) (jsonutils.JSONObject, error) {
- if !utils.IsInStringArray(self.Status, []string{api.VM_READY, api.VM_RUNNING, api.VM_SUSPEND}) {
- return nil, httperrors.NewInputParameterError("Cannot set security rules in status %s", self.Status)
- }
- driver, _ := self.GetDriver()
- maxCount := driver.GetMaxSecurityGroupCount()
- if maxCount == 0 {
- return nil, httperrors.NewUnsupportOperationError("Cannot set security group for this guest %s", self.Name)
- }
- if len(input.SecgroupIds) > maxCount {
- return nil, httperrors.NewUnsupportOperationError("guest %s band to up to %d security groups", self.Name, maxCount)
- }
- var guestnetwork *SGuestnetwork
- var err error
- if input.MacAddr != "" {
- guestnetwork, err = self.GetGuestnetworkByMac(input.MacAddr)
- } else if input.NetworkIndex != nil {
- guestnetwork, err = self.getGuestnetworkByIndex(*input.NetworkIndex)
- } else {
- return nil, httperrors.NewBadRequestError("no valid network index or mac addr")
- }
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "get guest network"))
- }
- network, err := guestnetwork.GetNetwork()
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrap(err, "GetNetwork"))
- }
- vpc, err := network.GetVpc()
- if err != nil {
- return nil, errors.Wrap(err, "GetVpc")
- }
- secgroupIds := []string{}
- secgroupNames := []string{}
- for i := range input.SecgroupIds {
- secObj, err := validators.ValidateModel(ctx, userCred, SecurityGroupManager, &input.SecgroupIds[i])
- if err != nil {
- return nil, err
- }
- secgrp := secObj.(*SSecurityGroup)
- err = vpc.CheckSecurityGroupConsistent(secgrp)
- if err != nil {
- return nil, err
- }
- if !utils.IsInStringArray(secgrp.GetId(), secgroupIds) {
- secgroupIds = append(secgroupIds, secgrp.GetId())
- secgroupNames = append(secgroupNames, secgrp.GetName())
- }
- }
- err = self.SaveNetworkSecgroups(ctx, userCred, secgroupIds, guestnetwork.Index)
- if err != nil {
- return nil, httperrors.NewGeneralError(errors.Wrapf(err, "SaveNetworkSecgroups"))
- }
- notes := map[string][]string{"secgroups": secgroupNames}
- logclient.AddActionLogWithContext(ctx, self, logclient.ACT_VM_SETSECGROUP, notes, userCred, true)
- return nil, self.StartSyncTask(ctx, userCred, true, "")
- }
- func (self *SGuest) SaveNetworkSecgroups(
- ctx context.Context, userCred mcclient.TokenCredential, secgroupIds []string, networkIndex int,
- ) error {
- if len(secgroupIds) == 0 {
- return self.RevokeNetworkAllSecgroups(ctx, userCred, networkIndex)
- }
- oldIds := set.New(set.ThreadSafe)
- newIds := set.New(set.ThreadSafe)
- gnss, err := self.GetGuestNetworkSecgroups(networkIndex)
- if err != nil {
- return errors.Wrap(err, "GetGuestNetworkSecgroups")
- }
- secgroupMaps := map[string]SGuestnetworksecgroup{}
- for i := range gnss {
- oldIds.Add(gnss[i].SecgroupId)
- secgroupMaps[gnss[i].SecgroupId] = gnss[i]
- }
- for i := range secgroupIds {
- newIds.Add(secgroupIds[i])
- }
- for _, removed := range set.Difference(oldIds, newIds).List() {
- id := removed.(string)
- gns, ok := secgroupMaps[id]
- if ok {
- err = gns.Delete(ctx, userCred)
- if err != nil {
- return errors.Wrapf(err,
- "Delete guest network secgroup for guest %s network index %d secgroup %s",
- self.GetName(), networkIndex, id)
- }
- }
- }
- for _, added := range set.Difference(newIds, oldIds).List() {
- id := added.(string)
- err = self.newGuestNetworkSecgroup(ctx, id, networkIndex, false)
- if err != nil {
- return errors.Wrapf(err,
- "New guest network secgroup for guest %s network index %d with secgroup %s",
- self.GetName(), networkIndex, id)
- }
- }
- return nil
- }
- func (self *SGuest) newGuestNetworkSecgroup(ctx context.Context, secgroupId string, networkIndex int, isAdmin bool) error {
- gns := &SGuestnetworksecgroup{}
- gns.SetModelManager(GuestnetworksecgroupManager, gns)
- gns.GuestId = self.Id
- gns.SecgroupId = secgroupId
- gns.NetworkIndex = networkIndex
- gns.Admin = isAdmin
- return GuestnetworksecgroupManager.TableSpec().Insert(ctx, gns)
- }
- func (self *SGuest) GetGuestNetworkSecgroups(networkIndex int) ([]SGuestnetworksecgroup, error) {
- gss := []SGuestnetworksecgroup{}
- q := GuestnetworksecgroupManager.Query().Equals("guest_id", self.Id).Equals("network_index", networkIndex)
- err := db.FetchModelObjects(GuestnetworksecgroupManager, q, &gss)
- if err != nil {
- return nil, errors.Wrapf(err, "db.FetchModelObjects")
- }
- return gss, nil
- }
- func (self *SGuest) RevokeNetworkAllSecgroups(ctx context.Context, userCred mcclient.TokenCredential, networkIndex int) error {
- gss, err := self.GetGuestNetworkSecgroups(networkIndex)
- if err != nil {
- return errors.Wrapf(err, "GetGuestNetworkSecgroups")
- }
- for i := range gss {
- err = gss[i].Delete(ctx, userCred)
- if err != nil {
- return errors.Wrap(err, "Delete")
- }
- }
- //return self.newGuestNetworkSecgroup(ctx, options.Options.DefaultSecurityGroupId, networkIndex, false)
- return nil
- }
- func (self *SGuestnetworksecgroupManager) getNetworkSecgroupJson(guestId string, networkIndex int) ([]*api.SecgroupJsonDesc, error) {
- ret := []*api.SecgroupJsonDesc{}
- secgroups, err := GuestnetworksecgroupManager.GetGuestnetworksecgroups(guestId, networkIndex)
- if err != nil {
- return nil, errors.Wrap(err, "GetSecgroups")
- }
- for _, secGrp := range secgroups {
- ret = append(ret, secGrp.getDesc())
- }
- return ret, nil
- }
- func (self *SGuest) RevokeAllNetworkSecgroups(ctx context.Context, userCred mcclient.TokenCredential) error {
- gns, err := self.GetNetworks("")
- if err != nil {
- return errors.Wrap(err, "GetNetworks")
- }
- for i := range gns {
- gnss, err := self.GetGuestNetworkSecgroups(gns[i].Index)
- if err != nil {
- return errors.Wrap(err, "GetGuestNetworkSecgroups")
- }
- for j := range gnss {
- err = gnss[j].Delete(ctx, userCred)
- if err != nil {
- return errors.Wrap(err, "Delete guestnetworksecgroup")
- }
- }
- }
- return nil
- }
|