manage ipv6 localhost hostaddress

1 files changed, 11 insertions, 2 deletions

diff --git a/src/main/java/com/c2kernel/scripting/ScriptConsole.java b/src/main/java/com/c2kernel/scripting/ScriptConsole.java
index 83e210b..3aeda26 100644
--- a/src/main/java/com/c2kernel/scripting/ScriptConsole.java
+++ b/src/main/java/com/c2kernel/scripting/ScriptConsole.java
@@ -53,13 +53,20 @@ public class ScriptConsole implements SocketHandler {
         }

         if (hosts == null || hosts.equals("")) { // by default only allow localhost

             securityMode = ALLOW;

+            securityHosts.add("localhost");

             securityHosts.add("127.0.0.1");

+            securityHosts.add("0:0:0:0:0:0:0:1");//ipv6

         }

         else {

             StringTokenizer tok = new StringTokenizer(hosts, ",");

             while(tok.hasMoreTokens()) {

+            	String wHostName = tok.nextToken();

                 try {

-                    securityHosts.add(InetAddress.getByName(tok.nextToken()).getHostAddress());

+                    securityHosts.add(InetAddress.getByName(wHostName).getHostAddress());

+                    if ("localhost".equals(wHostName)){

+                        securityHosts.add("127.0.0.1");

+                        securityHosts.add("0:0:0:0:0:0:0:1");//ipv6

+                    }

                 } catch (UnknownHostException ex) {

                     Logger.error("Host not found "+ex.getMessage());

                 }

@@ -120,7 +127,9 @@ public class ScriptConsole implements SocketHandler {
         // check permission

         boolean allowed = true;

         if (securityMode!=NONE) {

-            if (securityHosts.contains(socket.getInetAddress().getHostAddress())) {

+        	//ogattaz

+        	String wHostAddress = socket.getInetAddress().getHostAddress();

+            if (securityHosts.contains(wHostAddress)) {

                 if (securityMode==DENY)

                     allowed = false;

             }