From b4153a68e9a5586ef3338ab2773f4870fd9b1603 Mon Sep 17 00:00:00 2001
From: satyr <murky.satyr@gmail.com>
Date: Mon, 29 Mar 2010 04:30:44 +0900
Subject: [PATCH] Utils: Made escapeHtml()/unescapeHtml() more efficient.

---
 ubiquity/modules/utils.js  | 28 ++++++++++++++++------------
 ubiquity/tests/test_all.js | 10 ++++++++++
 2 files changed, 26 insertions(+), 12 deletions(-)

diff --git a/ubiquity/modules/utils.js b/ubiquity/modules/utils.js
index e0d0e1ba..c8143ab9 100755
--- a/ubiquity/modules/utils.js
+++ b/ubiquity/modules/utils.js
@@ -826,26 +826,30 @@ function signHMAC(algo, key, str) {
 // Useful when you just want to concatenate a bunch of strings into
 // an HTML fragment and ensure that everything's escaped properly.
 
-function escapeHtml(string) (
-  String(string)
-  .replace(/&/g, "&amp;")
-  .replace(/</g, "&lt;")
-  .replace(/>/g, "&gt;")
-  .replace(/\"/g, "&quot;")
-  .replace(/\'/g, "&#39;"));
+function escapeHtml(s) String(s).replace(escapeHtml.re, escapeHtml.fn);
+escapeHtml.re = /[&<>\"\']/g;
+escapeHtml.fn = function escapeHtml_sub($) {
+  switch ($) {
+    case "&": return "&amp;";
+    case "<": return "&lt;";
+    case ">": return "&gt;";
+    case '"': return "&quot;";
+    case "'": return "&#39;";
+  }
+};
 
 // === {{{ Utils.unescapeHtml(string) }}} ===
 // Returns a version of the {{{string}}} with all occurrences of HTML character
-// references (&spades; &#x2665; &#9827; etc.) in it decoded.
+// references (e.g. &spades; &#x2665; &#9827; etc.) in it decoded.
 
-function unescapeHtml(string) (
-  String.replace(string, /&#?\w+;/g, unescapeHtml.parse));
-unescapeHtml.parse = function unescapeHtml_parse(ref) {
+function unescapeHtml(s) String(s).replace(unescapeHtml.re, unescapeHtml.fn);
+unescapeHtml.re = /(?:&#?\w+;)+/g;
+unescapeHtml.fn = function unescapeHtml_parse(ref) {
   var {div} = unescapeHtml_parse;
   div.innerHTML = ref;
   return div.textContent;
 };
-defineLazyProperty(unescapeHtml.parse, function div() (
+defineLazyProperty(unescapeHtml.fn, function div() (
   Utils.hiddenWindow.document.createElementNS(
     "http://www.w3.org/1999/xhtml", "div")));
 
diff --git a/ubiquity/tests/test_all.js b/ubiquity/tests/test_all.js
index b60ac96a..8a4821c0 100755
--- a/ubiquity/tests/test_all.js
+++ b/ubiquity/tests/test_all.js
@@ -611,6 +611,16 @@ function testUtilsPrefs() {
   }
 }
 
+function testUtilsEscapeUnescapeHtml() {
+  this.skipIfXPCShell();
+
+  var {escapeHtml, unescapeHtml} = Utils;
+  var html = Utils.hiddenWindow.document.documentElement.innerHTML;
+  this.assertEquals(
+    html += "&spades;'&#x2665;&#9827;",
+    unescapeHtml(unescapeHtml(escapeHtml(escapeHtml(html)))));
+}
+
 function testL10nUtilsPropertySelector() {
   var ps = LocalizationUtils.propertySelector("data:," + encodeURI(<![CDATA[
     foo=%S %S