fix comments

Author: paulyufan2

cni/network/network.go

@@ -781,7 +781,8 @@ func (plugin *NetPlugin) createEndpointInternal(opt *createEndpointInternalOpt)
 		NATInfo:            opt.natInfo,
 	}
 
-	epPolicies, err := getPoliciesFromRuntimeCfg(opt.nwCfg)
+	isIPv6Enabled := opt.resultV6 != nil
+	epPolicies, err := getPoliciesFromRuntimeCfg(opt.nwCfg, isIPv6Enabled)
 	if err != nil {
 		logger.Error("failed to get policies from runtime configurations", zap.Error(err))
 		return epInfo, err

cni/network/network_linux.go

@@ -123,7 +123,7 @@ func getEndpointPolicies(PolicyArgs) ([]policy.Policy, error) {
 
 // getPoliciesFromRuntimeCfg returns network policies from network config.
 // getPoliciesFromRuntimeCfg is a dummy function for Linux platform.
-func getPoliciesFromRuntimeCfg(_ *cni.NetworkConfig) ([]policy.Policy, error) {
+func getPoliciesFromRuntimeCfg(_ *cni.NetworkConfig, _ bool) ([]policy.Policy, error) {
 	return nil, nil
 }
 

cni/network/network_windows.go

@@ -248,7 +248,7 @@ func getEndpointDNSSettings(nwCfg *cni.NetworkConfig, result *cniTypesCurr.Resul
 }
 
 // getPoliciesFromRuntimeCfg returns network policies from network config.
-func getPoliciesFromRuntimeCfg(nwCfg *cni.NetworkConfig) ([]policy.Policy, error) {
+func getPoliciesFromRuntimeCfg(nwCfg *cni.NetworkConfig, isIPv6Enabled bool) ([]policy.Policy, error) {
 	logger.Info("Runtime Info", zap.Any("config", nwCfg.RuntimeConfig))
 	var policies []policy.Policy
 	var protocol uint32
@@ -265,7 +265,7 @@ func getPoliciesFromRuntimeCfg(nwCfg *cni.NetworkConfig) ([]policy.Policy, error
 
 		hostIP, err := netip.ParseAddr(mapping.HostIp)
 		if err != nil {
-			return nil, errors.Wrap(err, "failed to parse hostIP")
+			return nil, errors.Wrapf(err, "failed to parse hostIP %v", hostIP)
 		}
 
 		// To support hostport policy mapping
@@ -276,7 +276,7 @@ func getPoliciesFromRuntimeCfg(nwCfg *cni.NetworkConfig) ([]policy.Policy, error
 		switch {
 		case hostIP.Is4():
 			flag = hnsv2.NatFlagsLocalRoutedVip
-		case hostIP.Is6():
+		case hostIP.Is6() && isIPv6Enabled:
 			flag = hnsv2.NatFlagsIPv6
 		}
 

cni/network/network_windows_test.go

@@ -253,7 +253,8 @@ func TestSetPoliciesFromNwCfg(t *testing.T) {
 	for _, tt := range tests {
 		tt := tt
 		t.Run(tt.name, func(t *testing.T) {
-			policies, err := getPoliciesFromRuntimeCfg(&tt.nwCfg)
+			isIPv6Enabled := false
+			policies, err := getPoliciesFromRuntimeCfg(&tt.nwCfg, isIPv6Enabled)
 			require.NoError(t, err)
 			require.Condition(t, assert.Comparison(func() bool {
 				return len(policies) > 0 && policies[0].Type == policy.EndpointPolicy