The Tor Browser: netwerk/test/TestCacheCollisions.js@a63d609f5ebe (annotated)

netwerk/test/TestCacheCollisions.js@a63d609f5ebe (annotated)

netwerk/test/TestCacheCollisions.js

Thu, 15 Jan 2015 15:55:04 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 15 Jan 2015 15:55:04 +0100
branch: TOR_BUG_9701
changeset 9: a63d609f5ebe
permissions: -rw-r--r--

Back out 97036ab72558 which inappropriately compared turds to third parties.

 var DEBUG = true;
 var clientID = "HTTP";
 var nsICache = Components.interfaces.nsICache;
 function getEventQueue()
 {
     var nsIEventQueueService = Components.interfaces.nsIEventQueueService;
     var CID = Components.classes["@mozilla.org/event-queue-service;1"];
     var service = CID.getService(nsIEventQueueService);
     return service.getSpecialEventQueue(nsIEventQueueService.CURRENT_THREAD_EVENT_QUEUE);
 }
 var eventQueue = getEventQueue();
 function getCacheService()
 {
     var nsCacheService = Components.classes["@mozilla.org/network/cache-service;1"];
     var service = nsCacheService.getService(Components.interfaces.nsICacheService);
     return service;
 }
 function createCacheSession(clientID, storagePolicy, streamable)
 {
     var service = getCacheService();
     var session = service.createSession(clientID, storagePolicy, streamable);
     return session;
 }
 function openCacheEntry(url, mode)
 {
     var session = createCacheSession(clientID, nsICache.STORE_ON_DISK, true);
     var entry = session.openCacheEntry(url, mode);
     return entry;
 }
 function wrapInputStream(input)
 {
     var nsIScriptableInputStream = Components.interfaces.nsIScriptableInputStream;
     var factory = Components.classes["@mozilla.org/scriptableinputstream;1"];
     var wrapper = factory.createInstance(nsIScriptableInputStream);
     wrapper.init(input);
     return wrapper;
 }
 function download(key)
 {
     var url = new java.net.URL(key);
     var data = "";
     var buffer = java.lang.reflect.Array.newInstance(java.lang.Byte.TYPE, 65536);
     var stream = url.openStream();
     while (true) {
         var count = stream.read(buffer);
         if (count <= 0)
             break;
         var str = new java.lang.String(buffer, 0, count);
         data += str;
     }
     stream.close();
     return data;
 }
 function write(url, data)
 {
     var key = url.toString();
     var outputEntry = openCacheEntry(key, nsICache.ACCESS_WRITE);
     var output = outputEntry.transport.openOutputStream(0, -1, 0);
     var count = output.write(data, data.length);
     // store some metadata.
     outputEntry.setMetaDataElement("size", data.length);
     output.close();
     outputEntry.markValid();
     outputEntry.close();
     return count;
 }
 function CacheListener()
 {
     this.done = false;
 }
 CacheListener.prototype = {
     QueryInterface : function(iid)
     {
         if (iid.equals(Components.interfaces.nsICacheListener))
             return this;
         throw Components.results.NS_NOINTERFACE;
     },
     onCacheEntryAvailable : function(/* in nsICacheEntryDescriptor */   descriptor,
                                      /* in nsCacheAccessMode */         accessGranted,
                                      /* in nsresult */                  status)
     {
         this.descriptor = descriptor;
         this.status = status;
         this.done = true;
     }
 };
 function asyncOpenCacheEntry(url, mode)
 {
     var session = createCacheSession(clientID, nsICache.STORE_ON_DISK, true);
     var listener = new CacheListener();
     session.asyncOpenCacheEntry(url, mode, listener);
     while (!listener.done)
         eventQueue.processPendingEvents();
     return listener.descriptor;
 }
 function asyncWrite(key, data)
 {
     var outputEntry = asyncOpenCacheEntry(key, nsICache.ACCESS_WRITE);
     var output = outputEntry.transport.openOutputStream(0, -1, 0);
     var count = output.write(data, data.length);
     // store some metadata.
     outputEntry.setMetaDataElement("size", data.length);
     output.close();
     outputEntry.markValid();
     outputEntry.close();
     return count;
 }
 function StreamListener()
 {
     this.done = false;
     this.data = "";
     this.wrapper = null;
 }
 StreamListener.prototype = {
     QueryInterface : function(iid)
     {
         if (iid.equals(Components.interfaces.nsIStreamListener) ||
             iid.equals(Components.interfaces.nsIStreamObserver))
             return this;
         throw Components.results.NS_NOINTERFACE;
     },
     onStartRequest : function(request, context)
     {
     },
     onStopRequest : function(request, context, statusCode, statusText)
     {
         this.statusCode = statusCode;
         this.done = true;
     },
     onDataAvailable : function(request, context, input, offset, count)
     {
         if (this.wrapper == null)
             this.wrapper = wrapInputStream(input);
         input = this.wrapper;
         this.data += input.read(count);
     }
 };
 function asyncRead(key)
 {
     var inputEntry = asyncOpenCacheEntry(key, nsICache.ACCESS_READ);
     var listener = new StreamListener();
     inputEntry.transport.asyncRead(listener, null, 0, inputEntry.dataSize, 0);
     while (!listener.done)
         eventQueue.processPendingEvents();
     inputEntry.close();
     return listener.data;
 }
 function read(key)
 {
     var inputEntry = openCacheEntry(key, nsICache.ACCESS_READ);
     var input = wrapInputStream(inputEntry.transport.openInputStream(0, -1, 0));
     var data = input.read(input.available());
     input.close();
     inputEntry.close();
     return data;
 }
 function readMetaData(key, element)
 {
     var inputEntry = openCacheEntry(key, nsICache.ACCESS_READ);
     var metadata = inputEntry.getMetaDataElement(element);
     inputEntry.close();
     return metadata;
 }
 function doom(url)
 {
     var key = url.toString();
     var doomedEntry = openCacheEntry(key, nsICache.ACCESS_READ_WRITE);
     doomedEntry.doom();
     doomedEntry.close();
 }
 // two keys which are known to collide right now.
 var key1 = "http://a772.g.akamai.net/7/772/51/7648437e551b56/www.apple.com/t/2001/us/en/i/2.gif";
 var key2 = "http://a772.g.akamai.net/7/772/51/70601300d0bde6/www.apple.com/t/2001/us/en/i/1right.gif";
 function test()
 {
     // 1. generate a collision, and close the colliding entry first.
     var entry1 = asyncOpenCacheEntry(key1, nsICache.ACCESS_READ_WRITE);
     var data1 = key1;
     entry1.setMetaDataElement("size", data1.length);
     entry1.markValid();
     var output1 = entry1.transport.openOutputStream(0, -1, 0);
     output1.write(data1, data1.length);
     var entry2 = asyncOpenCacheEntry(key2, nsICache.ACCESS_READ_WRITE);
     var data2 = key2;
     entry2.setMetaDataElement("size", data2.length);
     entry2.markValid();
     var output2 = entry2.transport.openOutputStream(0, -1, 0);
     output2.write(data2, data2.length);
     output1.close();
     output2.close();
     entry1.close();
     entry2.close();
 }
 function median(array)
 {
     var cmp = function(x, y) { return x - y; }
     array.sort(cmp);
     var middle = Math.floor(array.length / 2);
     return array[middle];
 }
 function sum(array)
 {
     var s = 0;
     var len = array.length;
     for (var i = 0; i < len; ++i)
         s += array[i];
     return s;
 }
 function time()
 {
     var N = 50;
     var System = java.lang.System;
     var url = new java.net.URL("http://www.mozilla.org");
     var key = url.toString();
     var downloadTimes = new Array();
     for (var i = 0; i < N; ++i) {
         var begin = System.currentTimeMillis();
         download(url);
         var end = System.currentTimeMillis();
         downloadTimes.push(end - begin);
     }
     var downloadTotal = sum(downloadTimes);
     var downloadMean = downloadTotal / N;
     var downloadMedian = median(downloadTimes);
     print("" + N + " downloads took " + downloadTotal + " milliseconds.");
     print("mean = " + downloadMean + " milliseconds.");
     print("median = " + downloadMedian + " milliseconds.");
     var readTimes = new Array();
     for (var i = 0; i < N; ++i) {
         var begin = System.currentTimeMillis();
         asyncRead(key);
         var end = System.currentTimeMillis();
         readTimes.push(end - begin);
     }
     var readTotal = sum(readTimes);
     var readMean = readTotal / N;
     var readMedian = median(readTimes);
     print("" + N + " reads took " + readTotal + " milliseconds.");
     print("mean = " + readMean + " milliseconds.");
     print("median = " + readMedian + " milliseconds.");
 }
 // load the cache service before doing anything with Java...
 getCacheService();
 if (DEBUG) {
     print("cache service loaded.");
 } else {
     print("running disk cache test.");
     test();
     print("disk cache test complete.");
 }