michael@0: // 2-byte charsets:
michael@0: const charsets = [ "Big5", "EUC-KR", "x-euc-tw", "x-johab" ]
michael@0: const ScriptableUnicodeConverter =
michael@0: 	Components.Constructor("@mozilla.org/intl/scriptableunicodeconverter",
michael@0: 			       "nsIScriptableUnicodeConverter");
michael@0: var gConverter;
michael@0: 
michael@0: function error(inString, outString, msg) {
michael@0:     var dispIn = "";
michael@0:     var dispOut = "";
michael@0:     var i;
michael@0:     for (i = 0; i < inString.length; ++i) {
michael@0: 	dispIn += " x" + inString.charCodeAt(i).toString(16);
michael@0:     }
michael@0:     if (outString.length == 0) {
michael@0: 	dispOut = "<empty>";
michael@0:     } else {
michael@0: 	for (i = 0; i < outString.length; ++i) {
michael@0: 	    dispOut += " x" + outString.charCodeAt(i).toString(16);
michael@0: 	}
michael@0:     }
michael@0:     dump("\"" + dispIn + "\" ==> \"" + dispOut + "\"\n");
michael@0:     do_throw("security risk: " + msg);
michael@0: }
michael@0: 
michael@0: function IsASCII(charCode) {
michael@0:     return (charCode <= 0x7e);
michael@0: }
michael@0: 
michael@0: function test(inString) {
michael@0:     var outString = gConverter.ConvertToUnicode(inString) +
michael@0: 	            gConverter.Finish();
michael@0: 
michael@0:     var outLen = outString.length;
michael@0: 
michael@0:     if (IsASCII(inString.charCodeAt(1)) && 
michael@0: 	(outLen < 4 || outString.charCodeAt(outLen - 4) == 0xFFFD)) {
michael@0: 	error(inString, outString, "ASCII input eaten in " + gConverter.charset);
michael@0:     }
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:     gConverter = new ScriptableUnicodeConverter();
michael@0:     for (var i = 0; i < charsets.length; ++i) {
michael@0: 	gConverter.charset = charsets[i];
michael@0: 
michael@0: 	var byte1, byte2;
michael@0: 	for (byte1 = 1; byte1 < 0x100; ++byte1) {
michael@0: 	    for (byte2 = 1; byte2 < 0x100; ++byte2) {
michael@0: 		test(String.fromCharCode(byte1, byte2) + "foo");
michael@0: 	    }
michael@0: 	}
michael@0:     }
michael@0: }