JSR-356: fixing Session.getProtocolVersion()

1 files changed, 23 insertions, 13 deletions

diff --git a/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java b/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java
index 984c357736..99b532c9ec 100644
--- a/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java
+++ b/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java
@@ -76,19 +76,6 @@ public class UpgradeRequest
             extensions.add(ExtensionConfig.parse(config));
         }
     }
-    
-    /**
-     * Get the User Principal for this request.
-     * <p>
-     * Only applicable when using UpgradeRequest from server side.
-     * 
-     * @return the user principal
-     */
-    public Principal getUserPrincipal()
-    {
-        // Server side should override to implement
-        return null;
-    }
 
     public void clearHeaders()
     {
@@ -201,6 +188,16 @@ public class UpgradeRequest
         return Collections.unmodifiableMap(parameters);
     }
 
+    public String getProtocolVersion()
+    {
+        String version = getHeader("Sec-WebSocket-Version");
+        if (version == null)
+        {
+            return "13";
+        }
+        return version;
+    }
+
     public String getQueryString()
     {
         return requestURI.getQuery();
@@ -228,6 +225,19 @@ public class UpgradeRequest
         return subProtocols;
     }
 
+    /**
+     * Get the User Principal for this request.
+     * <p>
+     * Only applicable when using UpgradeRequest from server side.
+     * 
+     * @return the user principal
+     */
+    public Principal getUserPrincipal()
+    {
+        // Server side should override to implement
+        return null;
+    }
+
     public boolean hasSubProtocol(String test)
     {
         for (String protocol : subProtocols)