Skip to content

Commit d9f5406

Browse files
zac-nixonniclask25
authored and
niclask25
committed
refactor duplicated icmp values
1 parent b8aa4b6 commit d9f5406

File tree

4 files changed

+30
-40
lines changed

4 files changed

+30
-40
lines changed

pkg/gateway/model/model_build_loadbalancer.go

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -15,15 +15,7 @@ import (
1515
var invalidLoadBalancerNamePattern = regexp.MustCompile("[[:^alnum:]]")
1616

1717
const (
18-
lbAttrsAccessLogsS3Enabled = "access_logs.s3.enabled"
19-
lbAttrsAccessLogsS3Bucket = "access_logs.s3.bucket"
20-
lbAttrsAccessLogsS3Prefix = "access_logs.s3.prefix"
21-
lbAttrsLoadBalancingCrossZoneEnabled = "load_balancing.cross_zone.enabled"
22-
lbAttrsLoadBalancingDnsClientRoutingPolicy = "dns_record.client_routing_policy"
23-
availabilityZoneAffinity = "availability_zone_affinity"
24-
partialAvailabilityZoneAffinity = "partial_availability_zone_affinity"
25-
anyAvailabilityZone = "any_availability_zone"
26-
resourceIDLoadBalancer = "LoadBalancer"
18+
resourceIDLoadBalancer = "LoadBalancer"
2719
)
2820

2921
type loadBalancerBuilder interface {

pkg/gateway/model/model_build_security_group.go

Lines changed: 10 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ import (
1919
ec2model "sigs.k8s.io/aws-load-balancer-controller/pkg/model/ec2"
2020
elbv2model "sigs.k8s.io/aws-load-balancer-controller/pkg/model/elbv2"
2121
"sigs.k8s.io/aws-load-balancer-controller/pkg/networking"
22+
"sigs.k8s.io/aws-load-balancer-controller/pkg/shared_constants"
2223
gwv1 "sigs.k8s.io/gateway-api/apis/v1"
2324
"strings"
2425
)
@@ -28,16 +29,9 @@ var (
2829
)
2930

3031
const (
31-
icmpv4Protocol = "icmp"
32-
icmpv6Protocol = "icmpv6"
33-
34-
icmpv4TypeForPathMtu = 3 // https://www.iana.org/assignments/icmp-parameters/icmp-parameters.xhtml#icmp-parameters-codes-3
35-
icmpv4CodeForPathMtu = 4
36-
37-
icmpv6TypeForPathMtu = 2 // https://www.iana.org/assignments/icmpv6-parameters/icmpv6-parameters.xhtml#icmpv6-parameters-codes-2
38-
icmpv6CodeForPathMtu = 0
39-
4032
resourceIDManagedSecurityGroup = "ManagedLBSecurityGroup"
33+
34+
managedSGDescription = "[k8s] Managed SecurityGroup for LoadBalancer"
4135
)
4236

4337
type securityGroupOutput struct {
@@ -161,7 +155,7 @@ func (builder *securityGroupBuilderImpl) buildManagedSecurityGroup(stack core.St
161155
ingressPermissions := builder.buildManagedSecurityGroupIngressPermissions(lbConf, routes, ipAddressType)
162156
return ec2model.NewSecurityGroup(stack, resourceIDManagedSecurityGroup, ec2model.SecurityGroupSpec{
163157
GroupName: name,
164-
Description: "[k8s] Managed SecurityGroup for LoadBalancer",
158+
Description: managedSGDescription,
165159
Tags: tags,
166160
Ingress: ingressPermissions,
167161
}), nil
@@ -223,9 +217,9 @@ func (builder *securityGroupBuilderImpl) buildManagedSecurityGroupIngressPermiss
223217

224218
if enableICMP {
225219
permissions = append(permissions, ec2model.IPPermission{
226-
IPProtocol: icmpv4Protocol,
227-
FromPort: awssdk.Int32(icmpv4TypeForPathMtu),
228-
ToPort: awssdk.Int32(icmpv4CodeForPathMtu),
220+
IPProtocol: shared_constants.ICMPV4Protocol,
221+
FromPort: awssdk.Int32(shared_constants.ICMPV4TypeForPathMtu),
222+
ToPort: awssdk.Int32(shared_constants.ICMPV4CodeForPathMtu),
229223
IPRanges: []ec2model.IPRange{
230224
{
231225
CIDRIP: cidr,
@@ -248,9 +242,9 @@ func (builder *securityGroupBuilderImpl) buildManagedSecurityGroupIngressPermiss
248242

249243
if enableICMP {
250244
permissions = append(permissions, ec2model.IPPermission{
251-
IPProtocol: icmpv6Protocol,
252-
FromPort: awssdk.Int32(icmpv6TypeForPathMtu),
253-
ToPort: awssdk.Int32(icmpv6CodeForPathMtu),
245+
IPProtocol: shared_constants.ICMPV6Protocol,
246+
FromPort: awssdk.Int32(shared_constants.ICMPV6TypeForPathMtu),
247+
ToPort: awssdk.Int32(shared_constants.ICMPV6CodeForPathMtu),
254248
IPv6Range: []ec2model.IPv6Range{
255249
{
256250
CIDRIPv6: cidr,

pkg/service/model_build_managed_sg.go

Lines changed: 7 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"encoding/hex"
77
"fmt"
88
"regexp"
9+
"sigs.k8s.io/aws-load-balancer-controller/pkg/shared_constants"
910
"strings"
1011

1112
awssdk "github.com/aws/aws-sdk-go-v2/aws"
@@ -17,15 +18,6 @@ import (
1718
)
1819

1920
const (
20-
icmpv4Protocol = "icmp"
21-
icmpv6Protocol = "icmpv6"
22-
23-
icmpv4TypeForPathMtu = 3 // https://www.iana.org/assignments/icmp-parameters/icmp-parameters.xhtml#icmp-parameters-codes-3
24-
icmpv4CodeForPathMtu = 4
25-
26-
icmpv6TypeForPathMtu = 2 // https://www.iana.org/assignments/icmpv6-parameters/icmpv6-parameters.xhtml#icmpv6-parameters-codes-2
27-
icmpv6CodeForPathMtu = 0
28-
2921
resourceIDManagedSecurityGroup = "ManagedLBSecurityGroup"
3022
)
3123

@@ -99,9 +91,9 @@ func (t *defaultModelBuildTask) buildManagedSecurityGroupIngressPermissions(ctx
9991
})
10092
if icmpForPathMtuConfigured && icmpForPathMtuConfiguredFlag == "on" {
10193
permissions = append(permissions, ec2model.IPPermission{
102-
IPProtocol: string(icmpv4Protocol),
103-
FromPort: awssdk.Int32(icmpv4TypeForPathMtu),
104-
ToPort: awssdk.Int32(icmpv4CodeForPathMtu),
94+
IPProtocol: shared_constants.ICMPV4Protocol,
95+
FromPort: awssdk.Int32(shared_constants.ICMPV4TypeForPathMtu),
96+
ToPort: awssdk.Int32(shared_constants.ICMPV4CodeForPathMtu),
10597
IPRanges: []ec2model.IPRange{
10698
{
10799
CIDRIP: cidr,
@@ -122,9 +114,9 @@ func (t *defaultModelBuildTask) buildManagedSecurityGroupIngressPermissions(ctx
122114
})
123115
if icmpForPathMtuConfigured && icmpForPathMtuConfiguredFlag == "on" {
124116
permissions = append(permissions, ec2model.IPPermission{
125-
IPProtocol: string(icmpv6Protocol),
126-
FromPort: awssdk.Int32(icmpv6TypeForPathMtu),
127-
ToPort: awssdk.Int32(icmpv6CodeForPathMtu),
117+
IPProtocol: shared_constants.ICMPV6Protocol,
118+
FromPort: awssdk.Int32(shared_constants.ICMPV6TypeForPathMtu),
119+
ToPort: awssdk.Int32(shared_constants.ICMPV6CodeForPathMtu),
128120
IPv6Range: []ec2model.IPv6Range{
129121
{
130122
CIDRIPv6: cidr,

pkg/shared_constants/icmp.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
package shared_constants
2+
3+
const (
4+
ICMPV4Protocol = "icmp"
5+
ICMPV6Protocol = "icmpv6"
6+
7+
ICMPV4CodeForPathMtu = 3 // https://www.iana.org/assignments/icmp-parameters/icmp-parameters.xhtml#icmp-parameters-codes-3
8+
ICMPV6CodeForPathMtu = 4
9+
10+
ICMPV4TypeForPathMtu = 2 // https://www.iana.org/assignments/icmpv6-parameters/icmpv6-parameters.xhtml#icmpv6-parameters-codes-2
11+
ICMPV6TypeForPathMtu = 0
12+
)

0 commit comments

Comments
 (0)