嘗試在Firefox擴展中獲取HTTP POST請求

Question

我正在構建一個擴展來獲取Firefox中的POST請求。 我閱讀了文檔以了解攔截頁面加載和HTTP觀察者，但仍無法在頁面加載時獲取特定的POST數據（例如：data1 = 50＆sdata2 = 0＆data3 = 50）。

我查看了TamperData的代碼，發現他們使用了stream.available（）和stream.read（1）。 但是，我無法使用這些命令來處理我的代碼。

目前我的代碼如下所示：

var ObserverTest = {


    observe: function(subject, topic, data) {
        if (topic == 'http-on-modify-request') {
            var httpChannel = subject.QueryInterface(Components.interfaces.nsIHttpChannel);
        }
        if (topic == "http-on-examine-response") {
            var newListener = new TracingListener();
            subject.QueryInterface(Ci.nsITraceableChannel);
            newListener.originalListener = subject.setNewListener(newListener);
        }
    },

    register: function() {
        var observerService = Components.classes["@mozilla.org/observer-service;1"].getService(Components.interfaces.nsIObserverService);
        observerService.addObserver(ObserverTest, "http-on-modify-request", false);
        observerService.addObserver(ObserverTest, "http-on-examine-response", false);
    },
    unregister: function() {
        var observerService = Components.classes["@mozilla.org/observer-service;1"].getService(Components.interfaces.nsIObserverService);
        observerService.removeObserver(ObserverTest, "http-on-modify-request");
        observerService.removeObserver(ObserverTest,"http-on-examine-response");
    }
}

window.addEventListener("load", ObserverTest.register, false);
window.addEventListener("unload", ObserverTest.unregister, false);


//Helper function for XPCOM instanciation (from Firebug)
function CCIN(cName, ifaceName) {
    return Cc[cName].createInstance(Ci[ifaceName]);
}
// Copy response listener implementation.
function TracingListener() {
    this.originalListener = null;
    this.receivedData = [];   // array for incoming data.
}

TracingListener.prototype = {
    onDataAvailable: function(request, context, inputStream, offset, count) {
        var binaryInputStream = CCIN("@mozilla.org/binaryinputstream;1", "nsIBinaryInputStream");
        var storageStream = CCIN("@mozilla.org/storagestream;1", "nsIStorageStream");
        var binaryOutputStream = CCIN("@mozilla.org/binaryoutputstream;1", "nsIBinaryOutputStream");


        var stream = Components.classes["@mozilla.org/scriptableinputstream;1"].createInstance(Components.interfaces.nsIScriptableInputStream);
        stream.init(binaryInputStream);

        binaryInputStream.setInputStream(inputStream);
        storageStream.init(8192, count, null);
        binaryOutputStream.setOutputStream(storageStream.getOutputStream(0));

        // Copy received data as they come.
        var data = binaryInputStream.readBytes(count);
        this.receivedData.push(data);

        binaryOutputStream.writeBytes(data, count);

        this.originalListener.onDataAvailable(request, context, storageStream.newInputStream(0), offset, count);
    },

    onStartRequest: function(request, context) {
        this.originalListener.onStartRequest(request, context);
    },

    onStopRequest: function(request, context, statusCode)
    {
        // Get entire response
        var responseSource = this.receivedData.join();
        this.originalListener.onStopRequest(request, context, statusCode);
    },

    QueryInterface: function (aIID) {
        if (aIID.equals(Ci.nsIStreamListener) || aIID.equals(Ci.nsISupports)) {
            return this;
        }
        throw Components.results.NS_NOINTERFACE;
    }
}

Answer 1

首先，根本不需要"http-on-examine-response"和TracingListener 。 如果您想對響應執行某些操作，那么這些內容將具有優點，但您在請求中的數據之后，因此topic == 'http-on-modify-request'就是這樣。

以下函數（未經測試，但從我的一個擴展中復制並清理了一下）演示了如何獲取發布數據。 假定從http-on-modify-request調用該函數。

const ScriptableInputStream = Components.Constructor(
  "@mozilla.org/scriptableinputstream;1",
  "nsIScriptableInputStream",
  "init");

function observeRequest(channel, topic, data) {
  let post = null;

  if (!(channel instanceof Ci.nsIHttpChannel) ||
    !(channel instanceof Ci.nsIUploadChannel)) {
    return post;
  }
  if (channel.requestMethod !== 'POST') {
    return post;
  }

  try {
    let us = channel.uploadStream;
    if (!us) {
      return post;
    }
    if (us instanceof Ci.nsIMultiplexInputStream) {
      // Seeking in a nsIMultiplexInputStream effectively breaks the stream.
      return post;
    }
    if (!(us instanceof Ci.nsISeekableStream)) {
      // Cannot seek within the stream :(
      return post;
    }

    let oldpos = us.tell();
    us.seek(0, 0);

    try {
      let is = new ScriptableInputStream(us);

      // we'll read max 64k
      let available = Math.min(is.available(), 1 << 16);
      if (available) {
        post = is.read(available);
      }
    }
    finally {
      // Always restore the stream position!
      us.seek(0, oldpos);
    }
  }
  catch (ex) {
    Cu.reportError(ex);
  }
  return post;
}

根據您的使用情況，您可能需要檢查us instanceof例如nsIMIMEInputStream或nsIStringInputStream是否用於特殊處理或快速路徑...

你可以從你的觀察者那里稱它為：

  observe: function(subject, topic, data) {
    if (topic == 'http-on-modify-request') {
      observeRequest(subject, topic, data);
    }
  },