From 0d92b713e1c775ba5fa02211bb1e9898c88c9168 Mon Sep 17 00:00:00 2001
From: David Timms <davidtimms@btinternet.com>
Date: Fri, 21 Jun 2013 01:53:02 +0100
Subject: [PATCH] Encryption Enabled

---
 static/chat.js    | 18 +++++++++++++++---
 static/index.html | 12 +++++++++++-
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/static/chat.js b/static/chat.js
index d20b113..10d506c 100644
--- a/static/chat.js
+++ b/static/chat.js
@@ -2,12 +2,13 @@ function Chat($scope) {
 	if (!$scope.server) {
 		window.ng_scope = $scope;
 		$scope.chat_url = $scope.chat_url || location.pathname.substring(1);
+		$scope.key = location.hash.substring(1);
+		console.log($scope.key);
 	}
 	var i;
 	$scope.chat_name = 	$scope.chat_name || $scope.chat_url.split('_').join(' ') || 'Chat';
 	$scope.my_username = undefined;
 	$scope.locked = false;
-
 	$scope.chatters = [];
 	$scope.chatters.get = function (name) {
 		for (var i = 0; i<this.length; i++) {
@@ -140,6 +141,7 @@ function Chat($scope) {
 				$scope.chatters.pop();
 			}
 			for (i = 0; i < data.messages.length; i++) {
+				data.messages[i].text = $scope.decrypt(data.messages[i].text);
 				new $scope.Message(data.messages[i]);
 			}
 			for (i = 0; i < data.chatters.length; i++) {
@@ -151,7 +153,7 @@ function Chat($scope) {
 		});
 		socket.on('new message', function (data) {
 			if (data.sender !== $scope.my_username || !data.sender) {
-				console.log('message: ' + data.text);
+				data.text = $scope.decrypt(data.text);
 				new $scope.Message(data);
 			}
 			$scope.$apply();
@@ -159,9 +161,12 @@ function Chat($scope) {
 		});
 		$scope.sendMessage = function() {
 			if ($scope.message_text !== '') {
-				var message = new $scope.Message({text: $scope.message_text, sender: $scope.my_username});
+				var original_text = $scope.message_text;
+				var enc_text =  'ENCRYPTED: ' + GibberishAES.enc(original_text, $scope.key);
+				var message = new $scope.Message({text: enc_text, sender: $scope.my_username});
 			    $scope.message_text = '';
 			    socket.emit('new message', message);
+			    message.text = original_text;
 		    }
 		};
 		// Send message on enter key
@@ -289,6 +294,13 @@ function Chat($scope) {
 			$scope.$apply();
 		});
 
+		$scope.decrypt = function (text) {		
+			if (text.indexOf('ENCRYPTED:') === 0) {
+				text = GibberishAES.dec(text.substring(11), $scope.key);
+			}
+			return text;
+		}
+
 		// Sidebar sliding
 		var slide_speed = 300;
 		var sidebar = $('.left-column');
diff --git a/static/index.html b/static/index.html
index 509691d..c781db8 100644
--- a/static/index.html
+++ b/static/index.html
@@ -36,6 +36,16 @@
 			}
 		</style>
 		<script>
+			function generateKey () {
+				var key = '';
+				for (var i = 0; i < 64; i += 1) {
+					var code = 55 + Math.floor(Math.random()*36);
+					if (code < 65) code -= 7;
+					key += String.fromCharCode(code);
+				}
+				return key;
+			};
+
 			$(document).ready(function () {
 				$('.alert').hide();
 				$('#new_chat_form').submit(function (e) {
@@ -52,7 +62,7 @@
 							$('.alert').html(res.error).fadeIn(250);
 						}
 						else {
-							window.location = res.redirect;
+							window.location = res.redirect + '#' + generateKey();
 						}
 					});
 				});