| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600 |
- // 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"
- "database/sql"
- "fmt"
- "regexp"
- "strings"
- "time"
- "yunion.io/x/jsonutils"
- "yunion.io/x/log"
- "yunion.io/x/pkg/errors"
- "yunion.io/x/pkg/gotypes"
- "yunion.io/x/pkg/util/netutils"
- randutil "yunion.io/x/pkg/util/rand"
- "yunion.io/x/pkg/util/rbacscope"
- "yunion.io/x/pkg/util/regutils"
- "yunion.io/x/sqlchemy"
- billing_api "yunion.io/x/onecloud/pkg/apis/billing"
- api "yunion.io/x/onecloud/pkg/apis/compute"
- "yunion.io/x/onecloud/pkg/cloudcommon/db"
- "yunion.io/x/onecloud/pkg/cloudcommon/db/lockman"
- "yunion.io/x/onecloud/pkg/compute/options"
- "yunion.io/x/onecloud/pkg/httperrors"
- "yunion.io/x/onecloud/pkg/mcclient"
- "yunion.io/x/onecloud/pkg/util/stringutils2"
- )
- const (
- MAX_IFNAME_SIZE = 13
- MAX_HINT_LEN = MAX_IFNAME_SIZE - 4 // 9
- HINT_BASE_LEN = 6 // 6
- HINT_RAND_LEN = MAX_HINT_LEN - HINT_BASE_LEN // 3
- MAX_GUESTNIC_TO_SAME_NETWORK = 2
- )
- // +onecloud:swagger-gen-ignore
- type SGuestnetworkManager struct {
- SGuestJointsManager
- SNetworkResourceBaseManager
- }
- var GuestnetworkManager *SGuestnetworkManager
- func init() {
- db.InitManager(func() {
- GuestnetworkManager = &SGuestnetworkManager{
- SGuestJointsManager: NewGuestJointsManager(
- SGuestnetwork{},
- "guestnetworks_tbl",
- "guestnetwork",
- "guestnetworks",
- NetworkManager,
- ),
- }
- GuestnetworkManager.SetVirtualObject(GuestnetworkManager)
- GuestnetworkManager.TableSpec().AddIndex(true, "ip_addr", "guest_id", "deleted")
- GuestnetworkManager.TableSpec().AddIndex(true, "ip6_addr", "guest_id", "deleted")
- GuestnetworkManager.TableSpec().AddIndex(false, "mac_addr", "deleted")
- })
- }
- // +onecloud:model-api-gen
- type SGuestnetwork struct {
- SGuestJointsBase
- NetworkId string `width:"36" charset:"ascii" nullable:"false" list:"user" `
- // MAC地址
- MacAddr string `width:"32" charset:"ascii" nullable:"false" list:"user" index:"true"`
- // IPv4地址
- IpAddr string `width:"16" charset:"ascii" nullable:"true" list:"user"`
- // IPv6地址
- Ip6Addr string `width:"64" charset:"ascii" nullable:"true" list:"user"`
- // 虚拟网卡驱动
- Driver string `width:"16" charset:"ascii" nullable:"true" list:"user" update:"user"`
- // 网卡队列数
- NumQueues int `nullable:"true" default:"1" list:"user" update:"user"`
- // 带宽限制,单位mbps
- BwLimit int `nullable:"false" default:"0" list:"user"`
- // 下行流量限制,单位 bytes
- RxTrafficLimit int64 `nullable:"false" default:"0" list:"user"`
- RxTrafficUsed int64 `nullable:"false" default:"0" list:"user"`
- // 上行流量限制,单位 bytes
- TxTrafficLimit int64 `nullable:"false" default:"0" list:"user"`
- TxTrafficUsed int64 `nullable:"false" default:"0" list:"user"`
- // 网卡序号
- Index int `nullable:"false" default:"0" list:"user" update:"user"`
- // 是否为虚拟接口(无IP)
- Virtual bool `default:"false" list:"user"`
- // 虚拟网卡设备名称
- Ifname string `width:"16" charset:"ascii" nullable:"true" list:"user" update:"user"`
- // bind配对网卡MAC地址
- TeamWith string `width:"32" charset:"ascii" nullable:"false" list:"user"`
- // IPv4映射地址,当子网属于私有云vpc的时候分配,用于访问外网
- MappedIpAddr string `width:"16" charset:"ascii" nullable:"true" list:"user"`
- // IPv6映射地址,当子网属于私有云vpc的时候分配,用于访问外网
- MappedIp6Addr string `width:"64" charset:"ascii" nullable:"true" list:"user"`
- // 网卡关联的Eip实例
- EipId string `width:"36" charset:"ascii" nullable:"true" list:"user"`
- // 是否为缺省路由
- IsDefault bool `default:"false" list:"user"`
- // 端口映射
- PortMappings api.GuestPortMappings `length:"long" list:"user" update:"user"`
- SBillingTypeBase `billing_type->default:"prepaid"`
- SBillingChargeTypeBase `charge_type->default:"bandwidth"`
- }
- func (gn SGuestnetwork) GetIP() string {
- return gn.IpAddr
- }
- func (gn SGuestnetwork) GetMAC() string {
- return gn.MacAddr
- }
- func (manager *SGuestnetworkManager) GetSlaveFieldName() string {
- return "network_id"
- }
- func (manager *SGuestnetworkManager) FetchCustomizeColumns(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- objs []interface{},
- fields stringutils2.SSortedStrings,
- isList bool,
- ) []api.GuestnetworkDetails {
- rows := make([]api.GuestnetworkDetails, len(objs))
- guestRows := manager.SGuestJointsManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
- netIds := make([]string, len(rows))
- eipIds := make([]string, len(rows))
- for i := range rows {
- rows[i] = api.GuestnetworkDetails{
- GuestJointResourceDetails: guestRows[i],
- }
- netIds[i] = objs[i].(*SGuestnetwork).NetworkId
- eipIds[i] = objs[i].(*SGuestnetwork).EipId
- ipnets, err := NetworkAddressManager.fetchAddressesByGuestnetworkId(objs[i].(*SGuestnetwork).RowId)
- if err != nil {
- log.Errorf("NetworkAddressManager.fetchAddressesByGuestnetworkId %s", err)
- } else if len(ipnets) > 0 {
- rows[i].NetworkAddresses = ipnets
- }
- }
- netMap := make(map[string]SNetwork)
- err := db.FetchModelObjectsByIds(NetworkManager, "id", netIds, netMap)
- if err != nil {
- log.Errorf("FetchIdNameMap2 fail %s", err)
- return rows
- }
- for i := range rows {
- if net, ok := netMap[netIds[i]]; ok {
- rows[i].Network = net.Name
- rows[i].WireId = net.WireId
- rows[i].GuestIpMask = net.GuestIpMask
- rows[i].GuestGateway = net.GuestGateway
- rows[i].GuestIp6Mask = net.GuestIp6Mask
- rows[i].GuestGateway6 = net.GuestGateway6
- }
- }
- eipIdMaps, err := db.FetchIdFieldMap2(ElasticipManager, "ip_addr", eipIds)
- if err != nil {
- log.Errorf("FetchIdFieldMap2 fail %s", err)
- return rows
- }
- for i := range rows {
- if ip, ok := eipIdMaps[eipIds[i]]; ok {
- rows[i].EipAddr = ip
- }
- }
- return rows
- }
- func (manager *SGuestnetworkManager) fetchByRowIds(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- rowIds []int64,
- ) ([]SGuestnetwork, error) {
- var gns []SGuestnetwork
- q := manager.Query().In("row_id", rowIds)
- if err := db.FetchModelObjects(manager, q, &gns); err != nil {
- return nil, errors.Wrapf(err, "fetch guestnetworks by row_id list")
- }
- idmap := map[int64]int{}
- for i := range gns {
- idmap[gns[i].RowId] = i
- }
- ret := make([]SGuestnetwork, len(rowIds))
- for i, rowId := range rowIds {
- if j, ok := idmap[rowId]; ok {
- ret[i] = gns[j]
- } else {
- return nil, errors.Wrapf(errors.ErrNotFound, "guestnetwork row %d", rowId)
- }
- }
- return ret, nil
- }
- func (manager *SGuestnetworkManager) fetchByRowId(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- rowId int64,
- ) (*SGuestnetwork, error) {
- gns, err := manager.fetchByRowIds(ctx, userCred, []int64{rowId})
- if err != nil {
- return nil, err
- }
- if len(gns) != 1 {
- return nil, errors.Errorf("row_id %d: got %d guesnetwork entries", rowId, len(gns))
- }
- return &gns[0], nil
- }
- func (manager *SGuestnetworkManager) GenerateMac(suggestion string) (string, error) {
- return generateMac(suggestion)
- }
- func (manager *SGuestnetworkManager) FilterByMac(mac string) *sqlchemy.SQuery {
- return manager.Query().Equals("mac_addr", mac)
- }
- type newGuestNetworkArgs struct {
- guest *SGuest
- network *SNetwork
- index int
- ipAddr string
- allocDir api.IPAllocationDirection
- tryReserved bool
- requireDesignatedIP bool
- useDesignatedIP bool
- ip6Addr string
- requireIPv6 bool
- strictIPv6 bool
- // 是否为缺省路由
- isDefault bool
- ifname string
- macAddr string
- bwLimit int
- nicDriver string
- numQueues int
- teamWithMac string
- rxTrafficLimit int64
- txTrafficLimit int64
- virtual bool
- portMappings api.GuestPortMappings
- billingType billing_api.TBillingType
- chargeType billing_api.TNetChargeType
- }
- func (manager *SGuestnetworkManager) newGuestNetwork(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- args newGuestNetworkArgs,
- ) (*SGuestnetwork, error) {
- gn := SGuestnetwork{}
- gn.SetModelManager(GuestnetworkManager, &gn)
- var (
- guest = args.guest
- network = args.network
- index = args.index
- address = args.ipAddr
- mac = args.macAddr
- driver = args.nicDriver
- numQueues = args.numQueues
- bwLimit = args.bwLimit
- virtual = args.virtual
- reserved = args.tryReserved
- allocDir = args.allocDir
- requiredDesignatedIp = args.requireDesignatedIP
- reUseAddr = args.useDesignatedIP
- ifname = args.ifname
- teamWithMac = args.teamWithMac
- isDefault = args.isDefault
- address6 = args.ip6Addr
- requireIPv6 = args.requireIPv6
- strictIPv6 = args.strictIPv6
- )
- gn.GuestId = guest.Id
- gn.NetworkId = network.Id
- gn.Index = index
- gn.Virtual = virtual
- if len(driver) == 0 {
- driver = api.NETWORK_DRIVER_VIRTIO
- }
- gn.Driver = driver
- gn.NumQueues = numQueues
- gn.RxTrafficLimit = args.rxTrafficLimit
- gn.TxTrafficLimit = args.txTrafficLimit
- if bwLimit >= 0 {
- gn.BwLimit = bwLimit
- }
- gn.PortMappings = args.portMappings
- gn.BillingType = args.billingType
- gn.ChargeType = args.chargeType
- lockman.LockObject(ctx, network)
- defer lockman.ReleaseObject(ctx, network)
- vpc, _ := network.GetVpc()
- if vpc == nil {
- return nil, fmt.Errorf("cannot find vpc of network %s(%s)", network.Id, network.Name)
- }
- provider := vpc.GetProviderName()
- if !virtual && !strictIPv6 && network.IsSupportIPv4() {
- if len(address) > 0 && reUseAddr {
- ipAddr, err := netutils.NewIPV4Addr(address)
- if err != nil {
- return nil, errors.Wrapf(err, "Reuse invalid address %s", address)
- }
- if !network.IsAddressInRange(ipAddr) {
- return nil, errors.Wrapf(httperrors.ErrOutOfRange, "%s not in network address range", address)
- }
- // if reuse Ip address, no need to check address availability
- // assign it anyway
- gn.IpAddr = address
- } else {
- // 如果是不具备IPAM能力的平台(主要是OneCloud和VMware,也就是VPC为ONECLOUD的平台 provider == api.CLOUD_PROVIDER_ONECLOUD ),则需要分配IP地址
- // 如果是其他云平台(具体IPAM能力的平台),则
- // * 开启options.Options.EnablePreAllocateIpAddr,也就是把IPAM的任务交给平台的,则需要分配IP地址
- // * IP地址为空并且 !options.Options.EnablePreAllocateIpAddr 时,不需要分配IP,等创建后自动同步过来
- // * 否则,还是需要先分配了
- if provider == api.CLOUD_PROVIDER_ONECLOUD || options.Options.EnablePreAllocateIpAddr || (!options.Options.EnablePreAllocateIpAddr && len(address) > 0) {
- addrTable := network.GetUsedAddresses(ctx)
- recentAddrTable := manager.getRecentlyReleasedIPAddresses(network.Id, network.getAllocTimoutDuration())
- ipAddr, err := network.GetFreeIP(ctx, userCred, addrTable, recentAddrTable, address, allocDir, reserved, api.AddressTypeIPv4)
- if err != nil {
- return nil, errors.Wrap(err, "GetFreeIPv4")
- }
- if len(address) > 0 && ipAddr != address && requiredDesignatedIp {
- usedAddr, _ := network.GetUsedAddressDetails(ctx, address)
- return nil, errors.Wrapf(httperrors.ErrConflict, "candidate ip %s is occupied with %#v!", address, usedAddr)
- }
- gn.IpAddr = ipAddr
- }
- }
- }
- var err error
- if ipBindMac := NetworkIpMacManager.GetMacFromIp(network.Id, gn.IpAddr); ipBindMac != "" {
- gn.MacAddr = ipBindMac
- } else {
- gn.MacAddr, err = manager.GenerateMac(mac)
- if err != nil {
- return nil, err
- }
- }
- if len(gn.MacAddr) == 0 {
- log.Errorf("Mac address generate fails")
- return nil, fmt.Errorf("mac address generate fails")
- }
- // assign ipv6 address
- if !virtual {
- if len(address6) > 0 || (requireIPv6 || len(gn.IpAddr) == 0) {
- if provider == api.CLOUD_PROVIDER_ONECLOUD || options.Options.EnablePreAllocateIpAddr || (!options.Options.EnablePreAllocateIpAddr && len(address6) > 0) {
- addrTable := network.GetUsedAddresses6(ctx)
- recentAddrTable := manager.getRecentlyReleasedIPAddresses6(network.Id, network.getAllocTimoutDuration())
- derived := false
- if len(address6) == 0 {
- // try to derive ipv6 address from mac and ipv4 address
- deriveAddr6 := netutils.DeriveIPv6AddrFromIPv4AddrMac(gn.IpAddr, gn.MacAddr, network.GuestIp6Start, network.GuestIp6End, network.GuestIp6Mask)
- if !isIpUsed(deriveAddr6, addrTable, recentAddrTable) {
- address6 = deriveAddr6
- derived = true
- }
- }
- ip6Addr, err := network.GetFreeIP(ctx, userCred, addrTable, recentAddrTable, address6, allocDir, reserved, api.AddressTypeIPv6)
- if err != nil {
- return nil, errors.Wrap(err, "GetFreeIPv6")
- }
- if len(address6) > 0 && ip6Addr != address6 && !derived && requiredDesignatedIp {
- usedAddr, _ := network.GetUsedAddressDetails(ctx, address6)
- return nil, errors.Wrapf(httperrors.ErrConflict, "candidate v6 ip %s is occupied with %#v!", address6, usedAddr)
- }
- gn.Ip6Addr = ip6Addr
- }
- }
- }
- if vpc.Id != api.DEFAULT_VPC_ID && provider == api.CLOUD_PROVIDER_ONECLOUD && (len(gn.IpAddr) > 0 || len(gn.Ip6Addr) > 0) {
- var err error
- GuestnetworkManager.lockAllocMappedAddr(ctx)
- defer GuestnetworkManager.unlockAllocMappedAddr(ctx)
- gn.MappedIpAddr, err = GuestnetworkManager.allocMappedIpAddr(ctx)
- if err != nil {
- return nil, errors.Wrap(err, "GuestnetworkManager.allocMappedIpAddr")
- }
- gn.MappedIp6Addr = api.GenVpcMappedIP6(gn.MappedIpAddr)
- }
- ifname, err = gn.checkOrAllocateIfname(network, ifname)
- if err != nil {
- return nil, err
- }
- gn.Ifname = ifname
- gn.TeamWith = teamWithMac
- if isDefault && ((len(gn.IpAddr) > 0 && len(network.GuestGateway) > 0) || (len(gn.Ip6Addr) > 0 && len(network.GuestGateway6) > 0)) {
- gn.IsDefault = isDefault
- }
- if len(gn.Ip6Addr) > 0 && len(gn.IpAddr) == 0 {
- gn.NumQueues = 1
- }
- err = manager.TableSpec().Insert(ctx, &gn)
- if err != nil {
- return nil, err
- }
- return &gn, nil
- }
- func (gn *SGuestnetwork) generateIfname(network *SNetwork, virtual bool, randomized bool) string {
- // It may happen that external networks when synced can miss ifname hint
- network.ensureIfnameHint()
- pattern := regexp.MustCompile(`\W+`)
- nName := pattern.ReplaceAllString(network.IfnameHint, "")
- if len(nName) > MAX_IFNAME_SIZE-4 {
- nName = nName[:(MAX_IFNAME_SIZE - 4)]
- }
- if virtual {
- return fmt.Sprintf("%s-%s", nName, randutil.String(3))
- } else {
- ip, _ := netutils.NewIPV4Addr(gn.IpAddr)
- cliaddr := ip.CliAddr(network.GuestIpMask)
- return fmt.Sprintf("%s-%d", nName, uint32(cliaddr))
- }
- }
- func (man *SGuestnetworkManager) ifnameUsed(ifname string) bool {
- // inviable names are always used
- if ifname == "" {
- return true
- }
- if len(ifname) > MAX_IFNAME_SIZE {
- return true
- }
- isa := func(c byte) bool {
- return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')
- }
- if !isa(ifname[0]) {
- return true
- }
- for i := range ifname[1:] {
- c := ifname[i]
- if isa(c) || c >= '0' || c <= '9' || c == '_' || c == '-' {
- continue
- }
- return true
- }
- count, err := GuestnetworkManager.Query().Equals("ifname", ifname).CountWithError()
- if err != nil {
- panic(errors.Wrap(err, "query if ifname is used"))
- }
- return count > 0
- }
- func (gn *SGuestnetwork) checkOrAllocateIfname(network *SNetwork, preferIfname string) (string, error) {
- man := GuestnetworkManager
- if !man.ifnameUsed(preferIfname) {
- return preferIfname, nil
- }
- ifname := gn.generateIfname(network, gn.Virtual, false)
- if !man.ifnameUsed(ifname) {
- return ifname, nil
- }
- if !gn.Virtual {
- ifname = gn.generateIfname(network, true, false)
- }
- found := false
- for i := 0; i < 5; i++ {
- if !man.ifnameUsed(ifname) {
- found = true
- break
- }
- ifname = gn.generateIfname(network, true, true)
- }
- if !found {
- return "", httperrors.NewConflictError("cannot allocate ifname")
- }
- return ifname, nil
- }
- func (gn *SGuestnetwork) GetGuest() *SGuest {
- guest, _ := GuestManager.FetchById(gn.GuestId)
- if guest != nil {
- return guest.(*SGuest)
- }
- return nil
- }
- func (gn *SGuestnetwork) GetNetwork() (*SNetwork, error) {
- net, err := NetworkManager.FetchById(gn.NetworkId)
- if err != nil {
- return nil, errors.Wrapf(err, "FetchById %s", gn.NetworkId)
- }
- return net.(*SNetwork), nil
- }
- func (gn *SGuestnetwork) GetTeamGuestnetwork() (*SGuestnetwork, error) {
- if len(gn.TeamWith) > 0 {
- return GuestnetworkManager.FetchByIdsAndIpMac(gn.GuestId, gn.NetworkId, "", gn.TeamWith)
- }
- return nil, nil
- }
- func (gn *SGuestnetwork) getJsonDescAtBaremetal(host *SHost) *api.GuestnetworkJsonDesc {
- net, _ := gn.GetNetwork()
- netif := guestGetHostNetifFromNetwork(host, net)
- if netif == nil {
- log.Errorf("fail to find a valid net interface on baremetal %s for network %s", host.String(), net.String())
- }
- return gn.getJsonDescHostwire(netif)
- }
- func guestGetHostNetifFromNetwork(host *SHost, network *SNetwork) *SNetInterface {
- netifs := host.getNetifsOnWire(network.WireId)
- var netif *SNetInterface
- for i := range netifs {
- if netifs[i].NicType != api.NIC_TYPE_IPMI {
- if netif == nil || netif.NicType != api.NIC_TYPE_ADMIN {
- netif = &netifs[i]
- }
- }
- }
- return netif
- }
- func (gn *SGuestnetwork) getJsonDescAtHost(ctx context.Context, host *SHost) *api.GuestnetworkJsonDesc {
- var (
- ret *api.GuestnetworkJsonDesc = nil
- network, _ = gn.GetNetwork()
- )
- if network.isOneCloudVpcNetwork() {
- // onecloud vpc nic
- ret = gn.getJsonDescOneCloudVpc(network)
- } else if network.WireId == api.DEFAULT_HOST_LOCAL_WIRE_ID {
- // host local nic
- ret = gn.getJsonDescHostLocal(network)
- } else {
- // classic nic
- netifs := host.getNetifsOnWire(network.WireId)
- var netif *SNetInterface
- for i := range netifs {
- if len(netifs[i].Bridge) > 0 {
- netif = &netifs[i]
- break
- }
- }
- if netif == nil && len(netifs) > 0 {
- log.Errorf("fail to find a bridged net_interface on host %s for network %s?????", host.String(), network.String())
- netif = &netifs[0]
- }
- ret = gn.getJsonDescHostwire(netif)
- }
- {
- ipnets, err := NetworkAddressManager.fetchAddressesByGuestnetworkId(gn.RowId)
- if err != nil {
- log.Errorln(err)
- }
- if len(ipnets) > 0 {
- ret.Networkaddresses = jsonutils.Marshal(ipnets)
- }
- }
- return ret
- }
- func (gn *SGuestnetwork) getJsonDescHostwire(netif *SNetInterface) *api.GuestnetworkJsonDesc {
- desc := gn.getJsonDesc()
- if netif != nil {
- desc.Bridge = netif.Bridge
- desc.WireId = netif.WireId
- desc.Interface = netif.Interface
- }
- return desc
- }
- func (gn *SGuestnetwork) getJsonDescOneCloudVpc(network *SNetwork) *api.GuestnetworkJsonDesc {
- if gn.MappedIpAddr == "" {
- var (
- err error
- addr string
- )
- addr, err = GuestnetworkManager.allocMappedIpAddr(context.TODO())
- if err != nil {
- log.Errorf("getJsonDescOneCloudVpc: row %d: alloc mapped ipaddr: %v", gn.RowId, err)
- } else {
- if _, err := db.Update(gn, func() error {
- gn.MappedIpAddr = addr
- gn.MappedIp6Addr = api.GenVpcMappedIP6(addr)
- return nil
- }); err != nil {
- log.Errorf("getJsonDescOneCloudVpc: row %d: db update mapped addr: %v", gn.RowId, err)
- gn.MappedIpAddr = ""
- gn.MappedIp6Addr = ""
- }
- }
- } else if gn.MappedIp6Addr == "" {
- if _, err := db.Update(gn, func() error {
- gn.MappedIp6Addr = api.GenVpcMappedIP6(gn.MappedIpAddr)
- return nil
- }); err != nil {
- log.Errorf("getJsonDescOneCloudVpc for v6: row %d: db update mapped addr6: %v", gn.RowId, err)
- gn.MappedIp6Addr = ""
- }
- }
- desc := gn.getJsonDesc()
- vpc, _ := network.GetVpc()
- desc.Vpc.Id = vpc.Id
- desc.Vpc.Provider = api.VPC_PROVIDER_OVN
- desc.Vpc.MappedIpAddr = gn.MappedIpAddr
- desc.Vpc.MappedIp6Addr = gn.MappedIp6Addr
- return desc
- }
- func (gn *SGuestnetwork) getJsonDescHostLocal(network *SNetwork) *api.GuestnetworkJsonDesc {
- desc := gn.getJsonDesc()
- desc.Bridge = api.HostLocalBridge
- desc.WireId = api.DEFAULT_HOST_LOCAL_WIRE_ID
- return desc
- }
- func (gn *SGuestnetwork) getJsonDesc() *api.GuestnetworkJsonDesc {
- net, _ := gn.GetNetwork()
- var wire *SWire
- if net != nil {
- wire, _ = net.GetWire()
- }
- desc := &api.GuestnetworkJsonDesc{
- GuestnetworkBaseDesc: api.GuestnetworkBaseDesc{
- Net: net.Name,
- NetId: gn.NetworkId,
- Mac: gn.MacAddr,
- Virtual: gn.Virtual,
- IsDefault: gn.IsDefault,
- PortMappings: gn.PortMappings,
- },
- }
- if gn.Virtual {
- if len(gn.TeamWith) > 0 {
- teamGN, _ := gn.GetTeamGuestnetwork()
- if teamGN != nil {
- desc.Ip = teamGN.IpAddr
- }
- } else {
- desc.Ip = net.GetNetAddr().String()
- }
- } else {
- desc.Ip = gn.IpAddr
- }
- desc.Gateway = net.GuestGateway
- desc.Dns = net.GetDNS("")
- desc.Domain = net.GetDomain()
- desc.Ntp = net.GetNTP()
- desc.Ip6 = gn.Ip6Addr
- desc.Masklen6 = net.GuestIp6Mask
- desc.Gateway6 = net.GuestGateway6
- routes := net.GetRoutes()
- if routes != nil && len(routes) > 0 {
- desc.Routes = jsonutils.Marshal(routes)
- }
- desc.Masklen = net.GuestIpMask
- desc.Driver = gn.Driver
- desc.NumQueues = gn.NumQueues
- desc.RxTrafficLimit = gn.RxTrafficLimit
- desc.TxTrafficLimit = gn.TxTrafficLimit
- desc.Vlan = net.VlanId
- desc.Bw = gn.getBandwidth(net, wire)
- desc.Mtu = gn.getMtu(net, wire)
- desc.Index = gn.Index
- desc.VirtualIps = gn.GetVirtualIPs()
- desc.ExternalId = net.ExternalId
- desc.TeamWith = gn.TeamWith
- desc.BillingType = gn.BillingType
- desc.ChargeType = gn.ChargeType
- guest := gn.getGuest()
- if ifname, ok := gn.OvsOffloadIfname(); ok {
- desc.Ifname = ifname
- } else {
- desc.Ifname = gn.Ifname
- }
- if guest.GetHypervisor() != api.HYPERVISOR_KVM {
- manual := true
- desc.Manual = &manual
- } else {
- if gn.Driver == api.NETWORK_DRIVER_VFIO {
- dev, _ := gn.GetIsolatedDevice()
- if dev != nil {
- if dev.OvsOffloadInterface == "" {
- manual := true
- desc.Manual = &manual
- }
- if dev.IsInfinibandNic {
- desc.NicType = api.NIC_TYPE_INFINIBAND
- }
- }
- }
- }
- if options.Options.NetworkAlwaysManualConfig {
- manual := true
- desc.Manual = &manual
- }
- return desc
- }
- func (gn *SGuestnetwork) getSecgroupDesc() *api.GuestnetworkSecgroupDesc {
- secgroupJson, _ := GuestnetworksecgroupManager.getNetworkSecgroupJson(gn.GuestId, gn.Index)
- if len(secgroupJson) == 0 {
- return nil
- }
- guest := gn.GetGuest()
- securityRules := guest.getNetworkSecurityGroupsRules(gn.Index)
- return &api.GuestnetworkSecgroupDesc{
- Secgroups: secgroupJson,
- SecurityRules: securityRules,
- Index: gn.Index,
- Mac: gn.MacAddr,
- }
- }
- func (gn *SGuestnetwork) IsSriovWithoutOffload() bool {
- if gn.Driver != api.NETWORK_DRIVER_VFIO {
- return false
- }
- if dev, _ := gn.GetIsolatedDevice(); dev != nil && dev.OvsOffloadInterface != "" {
- return false
- }
- return true
- }
- func (gn *SGuestnetwork) OvsOffloadIfname() (string, bool) {
- if gn.Driver != api.NETWORK_DRIVER_VFIO {
- return "", false
- }
- if dev, _ := gn.GetIsolatedDevice(); dev != nil && dev.OvsOffloadInterface != "" {
- return dev.OvsOffloadInterface, true
- }
- return "", false
- }
- func (gn *SGuestnetwork) UpdateNicTrafficUsed(ctx context.Context, guest *SGuest, matrics *api.SNicTrafficRecord, tm time.Time, isReset bool) error {
- if gn.BillingType == billing_api.BILLING_TYPE_POSTPAID && gn.ChargeType == billing_api.NET_CHARGE_TYPE_BY_TRAFFIC {
- err := GuestNetworkTrafficLogManager.logTraffic(ctx, guest, gn, matrics, tm, isReset)
- if err != nil {
- return errors.Wrap(err, "log traffic")
- }
- }
- _, err := db.Update(gn, func() error {
- gn.RxTrafficUsed = matrics.RxTraffic
- gn.TxTrafficUsed = matrics.TxTraffic
- return nil
- })
- return errors.Wrap(err, "update guestnetwork traffic used")
- }
- func (gn *SGuestnetwork) UpdateBillingMode(ctx context.Context, userCred mcclient.TokenCredential, input api.ServerNicTrafficLimit) error {
- billingChange := false
- if len(input.BillingType) > 0 && input.BillingType != gn.BillingType {
- billingChange = true
- }
- if len(input.ChargeType) > 0 && input.ChargeType != gn.ChargeType {
- billingChange = true
- }
- var oldDesc *jsonutils.JSONDict
- if billingChange {
- oldDesc = gn.GetShortDesc(ctx)
- }
- _, err := db.Update(gn, func() error {
- if input.RxTrafficLimit != nil {
- gn.RxTrafficLimit = *input.RxTrafficLimit
- }
- if input.TxTrafficLimit != nil {
- gn.TxTrafficLimit = *input.TxTrafficLimit
- }
- if len(input.BillingType) > 0 && input.BillingType != gn.BillingType {
- gn.BillingType = input.BillingType
- }
- if len(input.ChargeType) > 0 && input.ChargeType != gn.ChargeType {
- gn.ChargeType = input.ChargeType
- }
- return nil
- })
- if err != nil {
- return errors.Wrap(err, "update guestnetwork billing mode")
- }
- if billingChange {
- guest := gn.GetGuest()
- net, _ := gn.GetNetwork()
- db.OpsLog.LogDetachEvent(ctx, guest, net, userCred, oldDesc)
- db.OpsLog.LogAttachEvent(ctx, guest, net, userCred, gn.GetShortDesc(ctx))
- }
- return nil
- }
- func (gn *SGuestnetwork) UpdatePortMappings(pms api.GuestPortMappings) error {
- _, err := db.Update(gn, func() error {
- gn.PortMappings = pms
- return nil
- })
- return err
- }
- func (manager *SGuestnetworkManager) GetGuestByAddress(address string, projectId string) *SGuest {
- gnQ := manager.Query()
- ipField := "ip_addr"
- if regutils.MatchIP6Addr(address) {
- ipField = "ip6_addr"
- }
- gnQ = gnQ.Equals(ipField, address)
- gnSubQ := gnQ.SubQuery()
- q := GuestManager.Query("hostname", "tenant_id")
- if len(projectId) > 0 {
- q = q.Equals("tenant_id", projectId)
- }
- q = q.Join(gnSubQ, sqlchemy.Equals(gnSubQ.Field("guest_id"), q.Field("id")))
- guests := make([]SGuest, 0)
- err := q.All(&guests)
- if err != nil {
- log.Errorf("GetGuestByAddress %s fail %s", address, err)
- return nil
- }
- if len(guests) == 0 {
- return nil
- }
- return &guests[0]
- }
- func (gn *SGuestnetwork) GetDetailedString() string {
- network, err := gn.GetNetwork()
- if err != nil {
- return ""
- }
- naCount, _ := NetworkAddressManager.fetchAddressCountByGuestnetworkId(gn.RowId)
- parts := []string{
- gn.IpAddr, fmt.Sprintf("%d", network.GuestIpMask),
- }
- if len(gn.Ip6Addr) > 0 {
- parts = append(parts, gn.Ip6Addr, fmt.Sprintf("%d", network.GuestIp6Mask))
- }
- parts = append(parts,
- gn.MacAddr,
- fmt.Sprintf("%d", network.VlanId),
- network.Name,
- gn.Driver,
- fmt.Sprintf("%d", gn.getBandwidth(network, nil)),
- fmt.Sprintf("%d", naCount),
- )
- return fmt.Sprintf("eth%d:%s", gn.Index, strings.Join(parts, "/"))
- }
- func (gn *SGuestnetwork) ValidateUpdateData(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- input api.GuestnetworkUpdateInput,
- ) (api.GuestnetworkUpdateInput, error) {
- if input.Index != nil {
- index := *input.Index
- q := GuestnetworkManager.Query().SubQuery()
- count, err := q.Query().Filter(sqlchemy.Equals(q.Field("guest_id"), gn.GuestId)).
- Filter(sqlchemy.NotEquals(q.Field("network_id"), gn.NetworkId)).
- Filter(sqlchemy.Equals(q.Field("index"), index)).CountWithError()
- if err != nil {
- return input, httperrors.NewInternalServerError("checkout nic index uniqueness fail %s", err)
- }
- if count > 0 {
- return input, httperrors.NewDuplicateResourceError("NIC Index %d has been occupied", index)
- }
- }
- if input.IsDefault != nil && *input.IsDefault {
- if gn.Virtual || len(gn.TeamWith) > 0 {
- return input, errors.Wrap(httperrors.ErrInvalidStatus, "cannot set virtual/slave interface as default")
- }
- net, err := gn.GetNetwork()
- if err != nil {
- return input, errors.Wrapf(err, "GetNetwork")
- }
- if len(net.GuestGateway) == 0 && len(net.GuestGateway6) == 0 {
- return input, errors.Wrap(httperrors.ErrInvalidStatus, "network of default gateway has no gateway")
- }
- if (len(gn.IpAddr) == 0 || (len(gn.IpAddr) > 0 && len(net.GuestGateway) == 0)) && (len(gn.Ip6Addr) == 0 || (len(gn.Ip6Addr) > 0 && len(net.GuestGateway6) == 0)) {
- return input, errors.Wrap(httperrors.ErrInvalidStatus, "nic of default gateway has no ip")
- }
- }
- for _, pm := range input.PortMappings {
- if err := validatePortMapping(pm); err != nil {
- return input, err
- }
- }
- var err error
- input.GuestJointBaseUpdateInput, err = gn.SGuestJointsBase.ValidateUpdateData(ctx, userCred, query, input.GuestJointBaseUpdateInput)
- if err != nil {
- return input, errors.Wrap(err, "SGuestJointsBase.ValidateUpdateData")
- }
- return input, nil
- }
- func (gn *SGuestnetwork) PostUpdate(ctx context.Context, userCred mcclient.TokenCredential, query, data jsonutils.JSONObject) {
- gn.SGuestJointsBase.PostUpdate(ctx, userCred, query, data)
- input := api.GuestnetworkUpdateInput{}
- err := data.Unmarshal(&input)
- if err != nil {
- log.Errorf("GuestnetworkUpdateInput unmarshal fail %s", err)
- return
- }
- if input.IsDefault != nil && *input.IsDefault {
- // make this nic as default, unset others
- guest := gn.GetGuest()
- err := guest.setDefaultGateway(ctx, userCred, gn.MacAddr)
- if err != nil {
- log.Errorf("fail to setDefaultGateway: %s", err)
- }
- } else {
- // try fix default gateway
- guest := gn.GetGuest()
- err := guest.fixDefaultGateway(ctx, userCred)
- if err != nil {
- log.Errorf("fail to fixDefaultGateway %s", err)
- }
- }
- }
- func (manager *SGuestnetworkManager) DeleteGuestNics(ctx context.Context, userCred mcclient.TokenCredential, gns []SGuestnetwork, reserve bool) error {
- for i := range gns {
- gn := gns[i]
- if len(gn.EipId) > 0 {
- return errors.Wrapf(httperrors.ErrInvalidStatus, "eip associate with %s", gn.IpAddr)
- }
- guest := gn.GetGuest()
- dev, err := guest.GetIsolatedDeviceByNetworkIndex(gn.Index)
- if err != nil {
- return errors.Wrap(err, "GetIsolatedDeviceByNetworkIndex")
- }
- if dev != nil {
- err = guest.detachIsolateDevice(ctx, userCred, dev)
- if err != nil {
- return errors.Wrapf(err, "detachIsolateDevice %s", dev.GetName())
- }
- }
- net, _ := gn.GetNetwork()
- if !gotypes.IsNil(net) && (regutils.MatchIP4Addr(gn.IpAddr) || regutils.MatchIP6Addr(gn.Ip6Addr)) {
- net.updateDnsRecord(&gn, false)
- if regutils.MatchIP4Addr(gn.IpAddr) {
- // ??
- // netman.get_manager().netmap_remove_node(gn.ip_addr)
- }
- }
- err = gn.Delete(ctx, userCred)
- if err != nil {
- log.Errorf("guest network %s delete fail %s", gn.GetDetailedString(), err)
- return errors.Wrapf(err, "Delete %s", gn.GetDetailedString())
- }
- gn.LogDetachEvent(ctx, userCred, guest, net)
- if !gotypes.IsNil(net) {
- if reserve && regutils.MatchIP4Addr(gn.IpAddr) {
- ReservedipManager.ReserveIP(ctx, userCred, net, gn.IpAddr, "Delete to reserve", api.AddressTypeIPv4)
- }
- if reserve && regutils.MatchIP6Addr(gn.Ip6Addr) {
- ReservedipManager.ReserveIP(ctx, userCred, net, gn.Ip6Addr, "Delete to reserve", api.AddressTypeIPv6)
- }
- }
- }
- return nil
- }
- func (manager *SGuestnetworkManager) getGuestNicByIP(ip string, networkId string, addrType api.TAddressType) (*SGuestnetwork, error) {
- gn := SGuestnetwork{}
- q := manager.Query()
- field := "ip_addr"
- if addrType == api.AddressTypeIPv6 {
- field = "ip6_addr"
- }
- q = q.Equals(field, ip).Equals("network_id", networkId)
- err := q.First(&gn)
- if err != nil {
- if err != sql.ErrNoRows {
- log.Errorf("getGuestNicByIP fail %s", err)
- return nil, err
- }
- return nil, nil
- }
- gn.SetModelManager(manager, &gn)
- return &gn, nil
- }
- func (gn *SGuestnetwork) LogDetachEvent(ctx context.Context, userCred mcclient.TokenCredential, guest *SGuest, network *SNetwork) {
- if network == nil {
- netTmp, err := NetworkManager.FetchById(gn.NetworkId)
- if err != nil {
- return
- }
- network = netTmp.(*SNetwork)
- }
- db.OpsLog.LogDetachEvent(ctx, guest, network, userCred, gn.GetShortDesc(ctx))
- }
- func (gn *SGuestnetwork) Delete(ctx context.Context, userCred mcclient.TokenCredential) error {
- if err := NetworkAddressManager.deleteByGuestnetworkId(ctx, userCred, gn.RowId); err != nil {
- return errors.Wrap(err, "delete attached network addresses")
- }
- return db.DeleteModel(ctx, userCred, gn)
- }
- func (gn *SGuestnetwork) Detach(ctx context.Context, userCred mcclient.TokenCredential) error {
- return db.DetachJoint(ctx, userCred, gn)
- }
- func totalGuestNicCount(
- scope rbacscope.TRbacScope,
- ownerId mcclient.IIdentityProvider,
- rangeObjs []db.IStandaloneModel,
- includeSystem bool,
- providers []string,
- brands []string,
- cloudEnv string,
- ) GuestnicsCount {
- guests := GuestManager.Query().SubQuery()
- hosts := HostManager.Query().SubQuery()
- guestnics := GuestnetworkManager.Query().SubQuery()
- q := guestnics.Query()
- q = q.Join(guests, sqlchemy.Equals(guests.Field("id"), guestnics.Field("guest_id")))
- q = q.Join(hosts, sqlchemy.Equals(guests.Field("host_id"), hosts.Field("id")))
- q = q.Filter(sqlchemy.IsFalse(guests.Field("pending_deleted")))
- q = CloudProviderFilter(q, hosts.Field("manager_id"), providers, brands, cloudEnv)
- q = RangeObjectsFilter(q, rangeObjs, nil, hosts.Field("zone_id"), hosts.Field("manager_id"), hosts.Field("id"), nil)
- switch scope {
- case rbacscope.ScopeSystem:
- // do nothing
- case rbacscope.ScopeDomain:
- q = q.Filter(sqlchemy.Equals(guests.Field("domain_id"), ownerId.GetProjectDomainId()))
- case rbacscope.ScopeProject:
- q = q.Filter(sqlchemy.Equals(guests.Field("tenant_id"), ownerId.GetProjectId()))
- }
- if !includeSystem {
- q = q.Filter(sqlchemy.OR(sqlchemy.IsNull(guests.Field("is_system")),
- sqlchemy.IsFalse(guests.Field("is_system"))))
- }
- return calculateNics(q)
- }
- type GuestnicsCount struct {
- InternalNicCount int
- InternalVirtualNicCount int
- ExternalNicCount int
- ExternalVirtualNicCount int
- InternalBandwidth int
- ExternalBandwidth int
- }
- func calculateNics(q *sqlchemy.SQuery) GuestnicsCount {
- cnt := GuestnicsCount{}
- gns := make([]SGuestnetwork, 0)
- err := db.FetchModelObjects(GuestnetworkManager, q, &gns)
- if err != nil {
- log.Errorf("guestnics total count query error %s", err)
- }
- for _, gn := range gns {
- if gn.IsExit(nil) {
- if gn.Virtual {
- cnt.ExternalVirtualNicCount += 1
- } else {
- cnt.ExternalNicCount += 1
- }
- cnt.ExternalBandwidth += gn.BwLimit
- } else {
- if gn.Virtual {
- cnt.InternalVirtualNicCount += 1
- } else {
- cnt.InternalNicCount += 1
- }
- cnt.InternalBandwidth += gn.BwLimit
- }
- }
- return cnt
- }
- func (gn *SGuestnetwork) IsExit(net *SNetwork) bool {
- if gn.IpAddr != "" {
- addr, err := netutils.NewIPV4Addr(gn.IpAddr)
- if err == nil {
- return netutils.IsExitAddress(addr)
- }
- }
- if net == nil {
- net, _ = gn.GetNetwork()
- }
- if net != nil {
- return net.IsExitNetwork()
- }
- return false
- }
- func (gn *SGuestnetwork) getBandwidth(net *SNetwork, wire *SWire) int {
- if gn.BwLimit == 0 {
- return 0
- }
- if gn.BwLimit > 0 && gn.BwLimit <= api.MAX_BANDWIDTH {
- return gn.BwLimit
- } else {
- if wire == nil {
- if net == nil {
- net, _ = gn.GetNetwork()
- }
- if net != nil {
- wire, _ = net.GetWire()
- }
- }
- if wire != nil {
- return wire.Bandwidth
- }
- return options.Options.DefaultBandwidth
- }
- }
- func (gn *SGuestnetwork) getMtu(net *SNetwork, wire *SWire) int16 {
- return net.getMtu(wire)
- }
- func (gn *SGuestnetwork) IsAllocated() bool {
- region, _ := gn.GetGuest().getRegion()
- provider := region.Provider
- if regutils.MatchMacAddr(gn.MacAddr) && (gn.Virtual || regutils.MatchIP4Addr(gn.IpAddr) || regutils.MatchIP6Addr(gn.Ip6Addr) || (provider != api.CLOUD_PROVIDER_ONECLOUD && !options.Options.EnablePreAllocateIpAddr)) {
- return true
- }
- return false
- }
- /* func GetIPTenantIdPairs() {
- from guests import Guests
- from hosts import Hosts
- from sqlalchemy.sql.expression import bindparam
- q = Guestnics.query(Guestnics.ip_addr, Guestnics.mac_addr,
- bindparam('tunnel_key', None), # XXX: tunnel_key for VPC
- Guests.tenant_id, Guests.name,
- bindparam('tunnel_ip', None), Hosts.access_ip) \
- .join(Guests, and_(Guests.id==Guestnics.guest_id,
- Guests.deleted==False)) \
- .join(Hosts, and_(Hosts.id==Guests.host_id,
- Hosts.deleted==False))
- return q.all()
- } */
- func (gn *SGuestnetwork) GetVirtualIPs() []string {
- ips := make([]string, 0)
- guest := gn.GetGuest()
- net, _ := gn.GetNetwork()
- for _, guestgroup := range guest.GetGroups() {
- group := guestgroup.GetGroup()
- groupnets, err := group.GetNetworks()
- if err != nil {
- continue
- }
- for _, groupnetwork := range groupnets {
- gnet := groupnetwork.GetNetwork()
- if net != nil && gnet.WireId == net.WireId {
- ips = append(ips, groupnetwork.IpAddr)
- }
- }
- }
- return ips
- }
- func (gn *SGuestnetwork) GetIsolatedDevice() (*SIsolatedDevice, error) {
- dev := SIsolatedDevice{}
- q := IsolatedDeviceManager.Query().Equals("guest_id", gn.GuestId).Equals("network_index", gn.Index)
- if cnt, err := q.CountWithError(); err != nil {
- return nil, err
- } else if cnt == 0 {
- return nil, nil
- }
- err := q.First(&dev)
- if err != nil {
- return nil, err
- }
- dev.SetModelManager(IsolatedDeviceManager, &dev)
- return &dev, nil
- }
- func (manager *SGuestnetworkManager) getRecentlyReleasedIPAddresses(networkId string, recentDuration time.Duration) map[string]bool {
- return manager.getRecentlyReleasedIPAddressesInternal(networkId, recentDuration, api.AddressTypeIPv4)
- }
- func (manager *SGuestnetworkManager) getRecentlyReleasedIPAddresses6(networkId string, recentDuration time.Duration) map[string]bool {
- return manager.getRecentlyReleasedIPAddressesInternal(networkId, recentDuration, api.AddressTypeIPv6)
- }
- func (manager *SGuestnetworkManager) getRecentlyReleasedIPAddressesInternal(networkId string, recentDuration time.Duration, addrType api.TAddressType) map[string]bool {
- if recentDuration == 0 {
- return nil
- }
- field := "ip_addr"
- if addrType == api.AddressTypeIPv6 {
- field = "ip6_addr"
- }
- since := time.Now().UTC().Add(-recentDuration)
- q := manager.RawQuery(field)
- q = q.Equals("network_id", networkId).IsTrue("deleted")
- q = q.IsNotEmpty(field)
- q = q.GT("deleted_at", since).Distinct()
- rows, err := q.Rows()
- if err != nil {
- log.Errorf("GetRecentlyReleasedIPAddresses fail %s", err)
- return nil
- }
- defer rows.Close()
- ret := make(map[string]bool)
- for rows.Next() {
- var ip string
- err = rows.Scan(&ip)
- if err != nil {
- log.Errorf("scan error %s", err)
- } else {
- ret[ip] = true
- }
- }
- return ret
- }
- func (manager *SGuestnetworkManager) FilterByParams(q *sqlchemy.SQuery, params jsonutils.JSONObject) *sqlchemy.SQuery {
- macStr := jsonutils.GetAnyString(params, []string{"mac", "mac_addr"})
- if len(macStr) > 0 {
- q = q.Filter(sqlchemy.Equals(q.Field("mac_addr"), macStr))
- }
- ipStr := jsonutils.GetAnyString(params, []string{"ipaddr", "ip_addr", "ip"})
- if len(ipStr) > 0 {
- q = q.Filter(sqlchemy.Equals(q.Field("ip_addr"), ipStr))
- }
- ip6Str := jsonutils.GetAnyString(params, []string{"ip6addr", "ip6_addr", "ip6"})
- if len(ip6Str) > 0 {
- q = q.Filter(sqlchemy.Equals(q.Field("ip6_addr"), ip6Str))
- }
- return q
- }
- func (manager *SGuestnetworkManager) FetchByIdsAndIpMac(guestId string, netId string, ipAddr string, mac string) (*SGuestnetwork, error) {
- query := jsonutils.NewDict()
- if len(mac) > 0 {
- query.Add(jsonutils.NewString(mac), "mac_addr")
- }
- if len(ipAddr) > 0 {
- query.Add(jsonutils.NewString(ipAddr), "ip_addr")
- }
- ign, err := db.FetchJointByIds(manager, guestId, netId, query)
- if err != nil {
- return nil, err
- }
- return ign.(*SGuestnetwork), nil
- }
- func (manager *SGuestnetworkManager) FetchByGuestId(guestId string) ([]SGuestnetwork, error) {
- return manager.fetchGuestNetworks(func(q *sqlchemy.SQuery) *sqlchemy.SQuery {
- q = q.Equals("guest_id", guestId).Asc(q.Field("index"))
- return q
- })
- }
- func (manager *SGuestnetworkManager) fetchGuestNetworks(filter func(q *sqlchemy.SQuery) *sqlchemy.SQuery) ([]SGuestnetwork, error) {
- q := manager.Query()
- q = filter(q)
- var rets []SGuestnetwork
- if err := db.FetchModelObjects(manager, q, &rets); err != nil {
- return nil, errors.Wrap(err, "FetchModelObjects")
- }
- return rets, nil
- }
- func (manager *SGuestnetworkManager) FetchByGuestIdIndex(guestId string, index int8) (*SGuestnetwork, error) {
- rets, err := manager.FetchByGuestId(guestId)
- if err != nil {
- return nil, errors.Wrap(err, "FetchByGuestId")
- }
- if index >= 0 && int(index) < len(rets) {
- return &rets[index], nil
- }
- return nil, errors.ErrNotFound
- }
- func (gn *SGuestnetwork) GetShortDesc(ctx context.Context) *jsonutils.JSONDict {
- var net *SNetwork
- var wire *SWire
- if net == nil {
- net, _ = gn.GetNetwork()
- }
- if wire == nil {
- if net != nil {
- wire, _ = net.GetWire()
- }
- }
- desc := api.GuestnetworkShortDesc{}
- if net != nil {
- desc.BgpType = net.BgpType
- }
- if wire != nil {
- desc.VpcId = wire.VpcId
- }
- if len(gn.IpAddr) > 0 {
- desc.IpAddr = gn.IpAddr
- desc.IsExit = gn.IsExit(net)
- if desc.IsExit {
- desc.NicType = "exit"
- } else {
- desc.NicType = "internal"
- }
- } else {
- desc.NicType = "unused"
- }
- if len(gn.Ip6Addr) > 0 {
- desc.Ip6Addr = gn.Ip6Addr
- }
- desc.Mac = gn.MacAddr
- if len(gn.TeamWith) > 0 {
- desc.TeamWith = gn.TeamWith
- }
- desc.BwLimitMbps = gn.getBandwidth(net, wire)
- desc.Ifname = gn.Ifname
- desc.IsDefault = gn.IsDefault
- desc.BillingType = gn.BillingType
- desc.ChargeType = gn.ChargeType
- desc.GuestId = gn.GuestId
- desc.NetworkId = gn.NetworkId
- desc.PortMappings = gn.PortMappings
- desc.SubIps = gn.GetSubIps()
- return jsonutils.Marshal(desc).(*jsonutils.JSONDict)
- }
- func (gn *SGuestnetwork) GetWire() (*SWire, error) {
- net, err := gn.GetNetwork()
- if err != nil {
- return nil, errors.Wrap(err, "GetNetwork")
- }
- return net.GetWire()
- }
- func (gn *SGuestnetwork) GetVpc() (*SVpc, error) {
- net, err := gn.GetNetwork()
- if err != nil {
- return nil, errors.Wrap(err, "GetNetwork")
- }
- return net.GetVpc()
- }
- func (gn *SGuestnetwork) ToNetworkConfig() *api.NetworkConfig {
- net, err := gn.GetNetwork()
- if err != nil {
- return nil
- }
- wire, _ := net.GetWire()
- ret := &api.NetworkConfig{
- Index: int(gn.Index),
- Network: net.Id,
- Wire: wire.Id,
- Mac: gn.MacAddr,
- Address: gn.IpAddr,
- Address6: gn.Ip6Addr,
- Driver: gn.Driver,
- BwLimit: gn.BwLimit,
- Project: net.ProjectId,
- Domain: net.DomainId,
- Ifname: gn.Ifname,
- NetType: net.ServerType,
- Exit: net.IsExitNetwork(),
- }
- return ret
- }
- func (manager *SGuestnetworkManager) ListItemFilter(
- ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- query api.GuestnetworkListInput,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SGuestJointsManager.ListItemFilter(ctx, q, userCred, query.GuestJointsListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SGuestJointsManager.ListItemFilter")
- }
- q, err = manager.SNetworkResourceBaseManager.ListItemFilter(ctx, q, userCred, query.NetworkFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SNetworkResourceBaseManager.ListItemFilter")
- }
- if len(query.MacAddr) > 0 {
- q = q.In("mac_addr", query.MacAddr)
- }
- if len(query.IpAddr) > 0 {
- naSubQ := NetworkAddressManager.Query("parent_id").Equals("type", api.NetworkAddressTypeSubIP).Equals("parent_type", api.NetworkAddressParentTypeGuestnetwork).In("ip_addr", query.IpAddr).SubQuery()
- q = q.Filter(sqlchemy.OR(
- sqlchemy.In(q.Field("ip_addr"), query.IpAddr),
- sqlchemy.In(q.Field("row_id"), naSubQ),
- ))
- }
- if len(query.Ip6Addr) > 0 {
- q = q.In("ip6_addr", query.Ip6Addr)
- }
- if len(query.Driver) > 0 {
- q = q.In("driver", query.Driver)
- }
- if len(query.Ifname) > 0 {
- q = q.In("ifname", query.Ifname)
- }
- if len(query.TeamWith) > 0 {
- q = q.In("team_with", query.TeamWith)
- }
- return q, nil
- }
- func (manager *SGuestnetworkManager) OrderByExtraFields(
- ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- query api.GuestnetworkListInput,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SGuestJointsManager.OrderByExtraFields(ctx, q, userCred, query.GuestJointsListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SGuestJointsManager.OrderByExtraFields")
- }
- q, err = manager.SNetworkResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.NetworkFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SNetworkResourceBaseManager.OrderByExtraFields")
- }
- return q, nil
- }
- func (manager *SGuestnetworkManager) ListItemExportKeys(ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- keys stringutils2.SSortedStrings,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SGuestJointsManager.ListItemExportKeys(ctx, q, userCred, keys)
- if err != nil {
- return nil, errors.Wrap(err, "SGuestJointsManager.ListItemExportKeys")
- }
- if keys.ContainsAny(manager.SNetworkResourceBaseManager.GetExportKeys()...) {
- q, err = manager.SNetworkResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
- if err != nil {
- return nil, errors.Wrap(err, "SNetworkResourceBaseManager.ListItemExportKeys")
- }
- }
- return q, nil
- }
- func (manager *SGuestnetworkManager) InitializeData() error {
- err := manager.initOvnMappedIps()
- if err != nil {
- return errors.Wrap(err, "initOvnMappedIps")
- }
- return nil
- }
- func (manager *SGuestnetworkManager) initOvnMappedIps() error {
- q := manager.Query()
- networksQ := NetworkManager.Query().SubQuery()
- wiresQ := WireManager.Query().SubQuery()
- vpcQ := VpcManager.Query().SubQuery()
- regionQ := CloudregionManager.Query().SubQuery()
- q = q.Join(networksQ, sqlchemy.Equals(q.Field("network_id"), networksQ.Field("id")))
- q = q.Join(wiresQ, sqlchemy.Equals(networksQ.Field("wire_id"), wiresQ.Field("id")))
- q = q.Join(vpcQ, sqlchemy.Equals(wiresQ.Field("vpc_id"), vpcQ.Field("id")))
- q = q.Join(regionQ, sqlchemy.Equals(vpcQ.Field("cloudregion_id"), regionQ.Field("id")))
- q = q.Filter(sqlchemy.Equals(regionQ.Field("provider"), api.CLOUD_PROVIDER_ONECLOUD))
- q = q.Filter(sqlchemy.NotEquals(vpcQ.Field("id"), api.DEFAULT_VPC_ID))
- q = q.Filter(sqlchemy.OR(
- sqlchemy.IsEmpty(q.Field("mapped_ip_addr")),
- sqlchemy.IsEmpty(q.Field("mapped_ip6_addr")),
- ))
- gns := make([]SGuestnetwork, 0)
- err := db.FetchModelObjects(manager, q, &gns)
- if err != nil {
- return errors.Wrap(err, "FetchModelObjects")
- }
- for i := range gns {
- gn := &gns[i]
- var v4addr string
- if gn.MappedIpAddr == "" {
- addr, err := manager.allocMappedIpAddr(context.Background())
- if err != nil {
- return errors.Wrap(err, "allocMappedIpAddr")
- }
- v4addr = addr
- } else {
- v4addr = gn.MappedIpAddr
- }
- if _, err := db.Update(gn, func() error {
- gn.MappedIpAddr = v4addr
- gn.MappedIp6Addr = api.GenVpcMappedIP6(v4addr)
- return nil
- }); err != nil {
- return errors.Wrap(err, "db.Update")
- }
- }
- return nil
- }
- func (guest *SGuest) IsStrictIpv6() (bool, error) {
- q := GuestnetworkManager.Query().Equals("guest_id", guest.Id)
- q = q.IsNotEmpty("ip6_addr").IsNullOrEmpty("ip_addr").IsFalse("virtual")
- cnt, err := q.CountWithError()
- if err != nil {
- return false, errors.Wrap(err, "CountWithError")
- }
- return cnt > 0, nil
- }
- func (gn *SGuestnetwork) GetSubIps() string {
- subIPQ := NetworkAddressManager.fetchSubIpsQuery(api.NetworkAddressParentTypeGuestnetwork)
- subIPQ = subIPQ.Equals("parent_id", gn.RowId)
- result := struct {
- SubIps string `json:"sub_ips"`
- }{}
- err := subIPQ.First(&result)
- if err != nil {
- log.Errorf("Guestnetwork %s/%s/%s GetSubIps fail %s", gn.GuestId, gn.NetworkId, gn.MacAddr, err)
- return ""
- }
- return result.SubIps
- }
|