Skip to content

Commit 4ddb680

Browse files
committed
fix
1 parent 04acb74 commit 4ddb680

File tree

2 files changed

+16
-28
lines changed

2 files changed

+16
-28
lines changed

pkg/vpcmodel/explainabilityInput.go

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -467,13 +467,11 @@ func (c *VPCConfig) getCidrExternalNodes(inputIPBlock *netset.IPBlock) (cidrNode
467467
if !block.IsSubset(inputIPBlock) {
468468
continue
469469
}
470-
externalType := publicInternetNodeName
471470
isPublicInternet := true
472471
if block.IsSubset(ip) {
473-
externalType = serviceNetworkNodeName
474472
isPublicInternet = false
475473
}
476-
node, err1 := newExternalNode(isPublicInternet, block, externalType)
474+
node, err1 := newExternalNode(isPublicInternet, block)
477475
if err1 != nil {
478476
return nil, fatalErr, err1 // Should never get here. If still does - severe bug, exit with err
479477
}

pkg/vpcmodel/externalNetwork.go

Lines changed: 15 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -38,16 +38,8 @@ var networkAddressList = &NetworkAddressLists{}
3838

3939
func InitNetworkAddressLists(publicInternetAddressList, serviceNetworkAddressList []string) {
4040
once.Do(func() {
41-
ipbListPI, _, _ := ipStringsToIPblocks(publicInternetAddressList)
42-
ipbListSN, _, _ := ipStringsToIPblocks(serviceNetworkAddressList)
43-
publicInternetRanges := netset.NewIPBlock()
44-
serviceNetworkRanges := netset.NewIPBlock()
45-
for _, ipRange := range ipbListPI {
46-
publicInternetRanges = publicInternetRanges.Union(ipRange)
47-
}
48-
for _, ipRange := range ipbListSN {
49-
serviceNetworkRanges = serviceNetworkRanges.Union(ipRange)
50-
}
41+
ipbListPI, publicInternetRanges, _ := ipStringsToIPblocks(publicInternetAddressList)
42+
ipbListSN, serviceNetworkRanges, _ := ipStringsToIPblocks(serviceNetworkAddressList)
5143
networkAddressList = &NetworkAddressLists{ipbListPI, publicInternetRanges, ipbListSN, serviceNetworkRanges}
5244
})
5345
}
@@ -173,11 +165,15 @@ func ipStringsToIPblocks(ipList []string) (ipbList []*netset.IPBlock, unionIPblo
173165
return ipbList, unionIPblock, nil
174166
}
175167

176-
func newExternalNode(isPublicInternet bool, ipb *netset.IPBlock, resourceType string) (Node, error) {
168+
func newExternalNode(isPublicInternet bool, ipb *netset.IPBlock) (Node, error) {
177169
cidrsList := ipb.ToCidrList()
178170
if len(cidrsList) > 1 {
179171
return nil, errors.New("newExternalNode: input ip-block should be of a single CIDR")
180172
}
173+
resourceType := serviceNetworkNodeName
174+
if isPublicInternet {
175+
resourceType = publicInternetNodeName
176+
}
181177
cidr := ipb.ToCidrList()[0]
182178
return &ExternalNetwork{
183179
ResourceType: resourceType,
@@ -201,22 +197,19 @@ func newExternalNodeForCidr(cidr, resourceType string) (Node, error) {
201197

202198
func GetExternalNetworkNodes(disjointRefExternalIPBlocks []*netset.IPBlock) ([]Node, error) {
203199
res := []Node{}
204-
internetIPblocks, allInternetRagnes, err := GetNetworkAddressList().GetPublicInternetIPblocksList()
200+
internetIPblocks, allInternetRanges, err := GetNetworkAddressList().GetPublicInternetIPblocksList()
205201
if err != nil {
206202
return nil, err
207203
}
208-
serviceNetworkIPblocks, serviceNetworkRagnes, err := GetNetworkAddressList().GetServiceNetworkIPblocksList()
204+
serviceNetworkIPblocks, serviceNetworkRanges, err := GetNetworkAddressList().GetServiceNetworkIPblocksList()
209205
if err != nil {
210206
return nil, err
211207
}
212208
disjointRefExternalIPBlocksPublicInternet := netset.DisjointIPBlocks(internetIPblocks, disjointRefExternalIPBlocks)
213209
disjointRefExternalIPBlocksServiceNetwork := netset.DisjointIPBlocks(serviceNetworkIPblocks, disjointRefExternalIPBlocks)
214210

215211
for _, ipb := range disjointRefExternalIPBlocksPublicInternet {
216-
var isPublicInternet bool
217-
if ipb.IsSubset(allInternetRagnes) {
218-
isPublicInternet = true
219-
} else {
212+
if !ipb.IsSubset(allInternetRanges) {
220213
continue
221214
}
222215
cidrs := ipb.ToCidrList()
@@ -225,18 +218,15 @@ func GetExternalNetworkNodes(disjointRefExternalIPBlocks []*netset.IPBlock) ([]N
225218
if err != nil {
226219
return nil, err
227220
}
228-
newNode, err := newExternalNode(isPublicInternet, nodeIPBlock, publicInternetNodeName)
221+
newNode, err := newExternalNode(true, nodeIPBlock)
229222
if err != nil {
230223
return nil, err
231224
}
232225
res = append(res, newNode)
233226
}
234227
}
235228
for _, ipb := range disjointRefExternalIPBlocksServiceNetwork {
236-
var isPublicInternet bool
237-
if ipb.IsSubset(serviceNetworkRagnes) {
238-
isPublicInternet = false
239-
} else {
229+
if !ipb.IsSubset(serviceNetworkRanges) {
240230
continue
241231
}
242232
cidrs := ipb.ToCidrList()
@@ -245,7 +235,7 @@ func GetExternalNetworkNodes(disjointRefExternalIPBlocks []*netset.IPBlock) ([]N
245235
if err != nil {
246236
return nil, err
247237
}
248-
newNode, err := newExternalNode(isPublicInternet, nodeIPBlock, serviceNetworkNodeName)
238+
newNode, err := newExternalNode(false, nodeIPBlock)
249239
if err != nil {
250240
return nil, err
251241
}
@@ -282,9 +272,9 @@ func isEntirePublicInternetRange(nodes []*ExternalNetwork) (bool, error) {
282272
if err != nil {
283273
return false, err
284274
}
285-
_, allInternetRagnes, err := GetNetworkAddressList().GetPublicInternetIPblocksList()
275+
_, allInternetRanges, err := GetNetworkAddressList().GetPublicInternetIPblocksList()
286276
if err != nil {
287277
return false, err
288278
}
289-
return nodesRanges.Equal(allInternetRagnes), nil
279+
return nodesRanges.Equal(allInternetRanges), nil
290280
}

0 commit comments

Comments
 (0)