Add support-dhcpv6-in-aws flag and add method to get link-local address space in IPVersion

1 files changed, 26 insertions, 25 deletions

diff --git a/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/network/IPVersion.java b/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/network/IPVersion.java
index de80d4dca18..4cc825dacd6 100644
--- a/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/network/IPVersion.java
+++ b/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/network/IPVersion.java
@@ -15,52 +15,53 @@ import java.util.regex.Pattern;
  */
 public enum IPVersion {
 
-    IPv6(6, "ip6tables", "ip -6", "ipv6-icmp", "/128", "icmp6-port-unreachable", "ip6tables-restore"),
-    IPv4(4, "iptables", "ip", "icmp", "/32", "icmp-port-unreachable", "iptables-restore");
+    IPv6(6, "ip6tables", "ip -6", "ipv6-icmp", 128, "icmp6-port-unreachable", "ip6tables-restore", "fe80::/10"),
+    IPv4(4, "iptables", "ip", "icmp", 32, "icmp-port-unreachable", "iptables-restore", "169.254.0.0/16");
 
     private static final Pattern cidrNotationPattern = Pattern.compile("/\\d+$");
 
     IPVersion(int version, String iptablesCmd, String ipCmd,
-              String icmpProtocol, String singleHostCidr, String icmpPortUnreachable,
-              String iptablesRestore) {
+              String icmpProtocol, int size, String icmpPortUnreachable,
+              String iptablesRestore, String linkLocalCidr) {
         this.version = version;
         this.ipCmd = ipCmd;
         this.iptablesCmd = iptablesCmd;
         this.icmpProtocol = icmpProtocol;
-        this.singleHostCidr = singleHostCidr;
+        this.size = size;
         this.icmpPortUnreachable = icmpPortUnreachable;
         this.iptablesRestore = iptablesRestore;
+        this.linkLocalCidr = linkLocalCidr;
     }
 
     private final int version;
     private final String iptablesCmd;
     private final String ipCmd;
     private final String icmpProtocol;
-    private final String singleHostCidr;
+    private final int size;
     private final String icmpPortUnreachable;
     private final String iptablesRestore;
+    private final String linkLocalCidr;
 
-    public int version() {
-        return version;
-    }
-    public String versionString() {
-        return String.valueOf(version);
-    }
-    public String iptablesCmd() {
-        return iptablesCmd;
-    }
-    public String iptablesRestore() {
-        return iptablesRestore;
-    }
-    public String ipCmd() {
-        return ipCmd;
-    }
-    public String icmpProtocol() {
-        return icmpProtocol;
-    }
-    public String singleHostCidr() { return singleHostCidr; }
+    /** The ID of the IP version, either IPv4 or IPv6. */
+    public String id() { return "IPv" + version; }
+
+    /** The IP version, either 4 or 6 */
+    public int version() { return version; }
+
+    public String versionString() { return String.valueOf(version); }
+    public String iptablesCmd() { return iptablesCmd; }
+    public String iptablesRestore() { return iptablesRestore;}
+    public String ipCmd() { return ipCmd; }
+    public String icmpProtocol() { return icmpProtocol; }
+    public String singleHostCidr() { return "/" + size; }
     public String icmpPortUnreachable() { return icmpPortUnreachable; }
 
+    /** The address size (in bits) of the IP version: 32 or 128. */
+    public int addressSize() { return size; }
+
+    /** Both IPv4 and IPv6 have exactly one link-local address space: 169.254.0.0/16 or fe80::/10. */
+    public String linkLocalAddressCidr() { return linkLocalCidr; }
+
     public boolean match(InetAddress address) {
         return this == IPVersion.get(address);
     }