From 1bda968b226402a47dffb14a8f92207c7a15f8b8 Mon Sep 17 00:00:00 2001
From: Michael Jumper <zhangmaike@users.sourceforge.net>
Date: Sat, 4 Dec 2010 23:02:32 -0800
Subject: [PATCH] MD5 encoding, fixed build.xml

---
 guacamole/web-client/build.xml                |  2 +-
 .../basic/BasicUserMappingContentHandler.java | 81 ++++++++++++++++++-
 2 files changed, 79 insertions(+), 4 deletions(-)

diff --git a/guacamole/web-client/build.xml b/guacamole/web-client/build.xml
index 88685640e..c9b3ff621 100644
--- a/guacamole/web-client/build.xml
+++ b/guacamole/web-client/build.xml
@@ -66,7 +66,7 @@
 
         <copy todir="${tar.dir}">
             <fileset file="${doc.dir}/example/guacamole.xml"/>
-            <fileset file="${doc.dir}/example/guacamole-users.xml"/>
+            <fileset file="${doc.dir}/example/user-mapping.xml"/>
             <fileset file="${dist.dir}/guacamole.war"/>
             <fileset file="LICENSE.txt"/>
         </copy>
diff --git a/guacamole/web-client/src/net/sourceforge/guacamole/basic/BasicUserMappingContentHandler.java b/guacamole/web-client/src/net/sourceforge/guacamole/basic/BasicUserMappingContentHandler.java
index 6f71ebf30..8bb34240f 100644
--- a/guacamole/web-client/src/net/sourceforge/guacamole/basic/BasicUserMappingContentHandler.java
+++ b/guacamole/web-client/src/net/sourceforge/guacamole/basic/BasicUserMappingContentHandler.java
@@ -1,6 +1,8 @@
 
 package net.sourceforge.guacamole.basic;
 
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
@@ -16,19 +18,26 @@ public class BasicUserMappingContentHandler extends DefaultHandler {
         return Collections.unmodifiableMap(authMapping);
     }
 
-    public class AuthInfo {
+    public static class AuthInfo {
+
+        public static enum Encoding {
+            PLAIN_TEXT,
+            MD5
+        }
 
         private String auth_username;
         private String auth_password;
+        private Encoding auth_encoding;
 
         private String protocol;
         private String hostname;
         private int port;
         private String password;
 
-        public AuthInfo(String auth_username, String auth_password) {
+        public AuthInfo(String auth_username, String auth_password, Encoding auth_encoding) {
             this.auth_username = auth_username;
             this.auth_password = auth_password;
+            this.auth_encoding = auth_encoding;
         }
 
         public String getAuthorizedUsername() {
@@ -38,6 +47,59 @@ public class BasicUserMappingContentHandler extends DefaultHandler {
         public String getAuthorizedPassword() {
             return auth_password;
         }
+
+        private static final char HEX_CHARS[] = {
+            '0', '1', '2', '3', '4', '5', '6', '7',
+            '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
+        };
+
+        public static String getHexString(byte[] bytes) {
+
+            if (bytes == null)
+                return null;
+
+            StringBuilder hex = new StringBuilder(2 * bytes.length);
+            for (byte b : bytes) {
+                hex.append(HEX_CHARS[(b & 0xF0) >> 4])
+                   .append(HEX_CHARS[(b & 0x0F)     ]);
+            }
+
+            return hex.toString();
+
+        }
+
+
+        public boolean validate(String username, String password) {
+
+            // If username matches
+            if (username != null && password != null && username.equals(auth_username)) {
+
+                switch (auth_encoding) {
+
+                    case PLAIN_TEXT:
+
+                        // Compare plaintext
+                        return password.equals(auth_password);
+                    
+                    case MD5:
+
+                        // Compare hashed password
+                        try {
+                            MessageDigest digest = MessageDigest.getInstance("MD5");
+                            String hashedPassword = getHexString(digest.digest(password.getBytes()));
+                            return hashedPassword.equals(auth_password);
+                        }
+                        catch (NoSuchAlgorithmException e) {
+                            throw new UnsupportedOperationException("Unexpected lack of MD5 support.", e);
+                        }
+                    
+                }
+                
+            }
+
+            return false;
+            
+        }
         
         public String getHostname() {
             return hostname;
@@ -90,9 +152,22 @@ public class BasicUserMappingContentHandler extends DefaultHandler {
 
         if (localName.equals("authorize")) {
 
+            AuthInfo.Encoding encoding;
+            String encodingString = attributes.getValue("encoding");
+            if (encodingString == null)
+                encoding = AuthInfo.Encoding.PLAIN_TEXT;
+            else if (encodingString.equals("plain"))
+                encoding = AuthInfo.Encoding.PLAIN_TEXT;
+            else if (encodingString.equals("md5"))
+                encoding = AuthInfo.Encoding.MD5;
+            else
+                throw new SAXException("Invalid encoding type");
+
+
             current = new AuthInfo(
                 attributes.getValue("username"),
-                attributes.getValue("password")
+                attributes.getValue("password"),
+                encoding
             );
 
             infoState = null;