如何給selenium.chrome寫擴充攔截或轉發請求

Selenium.WebDriver

Selenium WebDriver 是一組開源 API，用于自動測試 Web 應用程式，利用它可以通過代碼來控制chrome浏覽器！

有時候我們需要mock接口的傳回，或者攔截和轉發請求，今天就來實作這個功能

代碼已開源： https://github.com/yuzd/OpenQA.Selenium.Chrome.Fiddler

nuget

OpenQA.Selenium.Chrome.Fiddler

開始coding

我們新建立一個功能：OpenQA.Selenium.Chrome.Fiddler

一個chrome擴充 最起碼有2個檔案

• manifest.json
• background.js

稍微解釋一下：

manifest.json 是來描述chrome擴充的

{
    "version": "1.0.0",
    "manifest_version": 2,
    "name": "Chrome Fiddler",
    "permissions": [
        "proxy",
        "tabs",
        "unlimitedStorage",
        "storage",
        "<all_urls>",
        "webRequest",
        "webRequestBlocking"
    ],
    "background": {
        "scripts": ["background.js"]
 },
    "minimum_chrome_version":"22.0.0"
}           

background.js 是邏輯處理子產品

因為攔截api 或者 轉發 需要用的chrome的api

chrome.webRequest.onBeforeRequest.addListener(
   function(details) {
      //邏輯處理
   },
   { urls: ['<all_urls>']},
   ['blocking', 'extraHeaders', 'requestBody']
);           

這個api的函數 接收的details參數

• details.url 是api的接口

函數的傳回

• {cancel:true} 攔截請求
• {redirectUrl:''} 轉發到指定url

寫selenium.chrome插件

建立一個netstand工程，然後引用

• Selenium.WebDriver

複制以下代碼

/// <summary>
/// Add Fiddler extention
/// </summary>
/// <param name="options">Chrome options</param>
/// <param name="fiddlerOption">Proxy host</param>
public static void AddFiddler(this ChromeOptions options, FiddlerOption fiddlerOption)
{
    var backgroundProxyJs = ReplaceTemplates(background_js, fiddlerOption);

    if (!Directory.Exists("Plugins"))
        Directory.CreateDirectory("Plugins");

    var guid = Guid.NewGuid().ToString();

    var manifestPath = $"Plugins/manifest_{guid}.json";
    var backgroundPath = $"Plugins/background_{guid}.js";
    var archiveFilePath = $"Plugins/proxy_auth_plugin_{guid}.zip";

    File.WriteAllText(manifestPath, manifest_json);
    File.WriteAllText(backgroundPath, backgroundProxyJs);

    using (var zip = ZipFile.Open(archiveFilePath, ZipArchiveMode.Create))
    {
        zip.CreateEntryFromFile(manifestPath, "manifest.json");
        zip.CreateEntryFromFile(backgroundPath, "background.js");
    }

    File.Delete(manifestPath);
    File.Delete(backgroundPath);

    options.AddExtension(archiveFilePath);
}

private static string ReplaceTemplates(string str, FiddlerOption fiddlerOption)
{
    if (fiddlerOption.OnBeforeRequestOptions != null)
    {
        var beforeConfigs = Newtonsoft.Json.JsonConvert.SerializeObject(fiddlerOption.OnBeforeRequestOptions);
        str = str.Replace("{before_configs}", beforeConfigs);
    }
    return str;
}           

上面的代碼主要是建立一個chrome擴充zip包

然後再selenium.chrome啟動的時候傳進去這個zip包的位址

使用方法

var driverBinary = @"D:\soft\chrome\chrome2\Chrome-bin\";

ChromeOptions options = new ChromeOptions
{
    BinaryLocation = Path.Combine(driverBinary, "chrome.exe")
};

Environment.SetEnvironmentVariable("webdriver.chrome.driver", driverBinary);
options.AddArgument("--disable-blink-features=AutomationControlled");
options.AddArguments("--disable-infobars");
List<string> ls = new List<string> { "enable-automation" };
options.AddExcludedArguments(ls);



#region Fillder

options.AddFiddler(new FiddlerOption
{
    OnBeforeRequestOptions = new List<FiddlerOnBeforeRequestOptions>
    {
        // 配置轉發
        new FiddlerOnBeforeRequestOptions
        {
            Match = "https://www.cnblogs.com/yudongdong/ajax/GetPostStat",//正則
            RedirectUrl = "http://localhost:5000/GetPostStat",//如果比對成功則将requestBody轉發到這個url中去
            Cancel = false//如果配置了cancel=true那麼轉發将無效，true的意思是直接攔截這次的請求,不去發送了
        },
        // 配置攔截
        new FiddlerOnBeforeRequestOptions
        {
            Match = "https://www.cnblogs.com/yudongdong/ajax/blogStats",
            Cancel = true//true的意思是直接攔截這次的請求,不去發送了
        },
    }
});


#endregion

var chrome = new ChromeDriver(driverBinary, options);           

實作效果

image

如果您覺得閱讀本文對您有幫助，請點一下“推薦”按鈕，您的“推薦”将是我最大的寫作動力！歡迎各位轉載，轉載文章之後須在文章頁面明顯位置給出作者和原文連接配接，謝謝。