Initial implementation of netconf monitoring module according to http://tools.ietf...

[controller.git] / opendaylight / netconf / netconf-impl / src / main / java / org / opendaylight / controller / netconf / impl / NetconfServerSessionNegotiator.java

diff --git a/opendaylight/netconf/netconf-impl/src/main/java/org/opendaylight/controller/netconf/impl/NetconfServerSessionNegotiator.java b/opendaylight/netconf/netconf-impl/src/main/java/org/opendaylight/controller/netconf/impl/NetconfServerSessionNegotiator.java
index e14ae3e4dca088cbd7d016ac119e6003bb8525a9..01ac018b3eadc3ee85d6ffd7a5e12fb8cbaf2daa 100644 (file)
--- a/opendaylight/netconf/netconf-impl/src/main/java/org/opendaylight/controller/netconf/impl/NetconfServerSessionNegotiator.java
+++ b/opendaylight/netconf/netconf-impl/src/main/java/org/opendaylight/controller/netconf/impl/NetconfServerSessionNegotiator.java
@@ -8,26 +8,91 @@
 
 package org.opendaylight.controller.netconf.impl;
 
+import com.google.common.base.Optional;
+import com.google.common.base.Preconditions;
+import io.netty.channel.Channel;
+import io.netty.util.Timer;
+import io.netty.util.concurrent.Promise;
+import org.opendaylight.controller.netconf.api.NetconfMessage;
 import org.opendaylight.controller.netconf.api.NetconfServerSessionPreferences;
 import org.opendaylight.controller.netconf.util.AbstractNetconfSessionNegotiator;
 import org.opendaylight.protocol.framework.SessionListener;
-import org.w3c.dom.Document;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
-import io.netty.channel.Channel;
-import io.netty.util.Timer;
-import io.netty.util.concurrent.Promise;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 public class NetconfServerSessionNegotiator extends
         AbstractNetconfSessionNegotiator<NetconfServerSessionPreferences, NetconfServerSession> {
 
+    static final Logger logger = LoggerFactory.getLogger(NetconfServerSessionNegotiator.class);
+
+    private static final AdditionalHeader DEFAULT_HEADER = new AdditionalHeader();
+
     protected NetconfServerSessionNegotiator(NetconfServerSessionPreferences sessionPreferences,
             Promise<NetconfServerSession> promise, Channel channel, Timer timer, SessionListener sessionListener) {
         super(sessionPreferences, promise, channel, timer, sessionListener);
     }
 
     @Override
-    protected NetconfServerSession getSession(SessionListener sessionListener, Channel channel, Document doc) {
-        return new NetconfServerSession(sessionListener, channel, sessionPreferences.getSessionId());
+    protected NetconfServerSession getSession(SessionListener sessionListener, Channel channel, NetconfMessage message) {
+        Optional<String> additionalHeader = message.getAdditionalHeader();
+
+        AdditionalHeader parsedHeader;
+        if (additionalHeader.isPresent()) {
+            parsedHeader = new AdditionalHeader(additionalHeader.get());
+        } else {
+            parsedHeader = DEFAULT_HEADER;
+        }
+        logger.debug("Additional header from hello parsed as {} from {}", parsedHeader, additionalHeader);
+
+        return new NetconfServerSession(sessionListener, channel, sessionPreferences.getSessionId(), parsedHeader);
     }
 
+    static class AdditionalHeader {
+
+        private static final Pattern pattern = Pattern
+                .compile("\\[(?<username>[^;]+);(?<address>[0-9\\.]+)[:/](?<port>[0-9]+);(?<transport>[a-z]+)[^\\]]+\\]");
+        private final String username;
+        private final String address;
+        private final String transport;
+
+        public AdditionalHeader(String addHeaderAsString) {
+            addHeaderAsString = addHeaderAsString.trim();
+            Matcher matcher = pattern.matcher(addHeaderAsString);
+            Preconditions.checkArgument(matcher.matches(), "Additional header in wrong format %s, expected %s",
+                    addHeaderAsString, pattern);
+            this.username = matcher.group("username");
+            this.address = matcher.group("address");
+            this.transport = matcher.group("transport");
+        }
+
+        private AdditionalHeader() {
+            this.username = this.address = "unknown";
+            this.transport = "ssh";
+        }
+
+        String getUsername() {
+            return username;
+        }
+
+        String getAddress() {
+            return address;
+        }
+
+        String getTransport() {
+            return transport;
+        }
+
+        @Override
+        public String toString() {
+            final StringBuffer sb = new StringBuffer("AdditionalHeader{");
+            sb.append("username='").append(username).append('\'');
+            sb.append(", address='").append(address).append('\'');
+            sb.append(", transport='").append(transport).append('\'');
+            sb.append('}');
+            return sb.toString();
+        }
+    }
 }