☰

Mark CVE

Mark the current page CVE

// ==UserScript==
// @name         Mark CVE
// @namespace    http://tampermonkey.net/
// @version      0.1.3
// @description  Mark the current page CVE
// @author       Kali-Team
// @match        *://*/*
// @exclude      https://scap.kali-team.cn/*
// @icon         https://avatars.githubusercontent.com/u/99640169?s=200&v=4
// @grant        none
// @run-at       document-idle
// @homepage     https://github.com/cn-kali-team/mark-cve
// @license      GPL-3.0-only
// ==/UserScript==

(function () {
    'use strict';
    let DefaultBaseUrl = 'https://scap.kali-team.cn/cve/';

    function GetBaseURL() {
        const KeyName = 'base_url';
        if (typeof browser === "undefined") {
            return DefaultBaseUrl;
        }
        try {
            browser.storage.local.get(KeyName).then((item) => {
                DefaultBaseUrl = item.base_url || DefaultBaseUrl;
            });
        } catch (error) {
            console.error(error);
        }
        return DefaultBaseUrl;
    }

    function Mark() {
        const userSelection = window.getSelection();
        const id = userSelection.toString().toLocaleUpperCase();
        const selectedTextRange = userSelection.getRangeAt(0);
        let cve = selectedTextRange.startContainer.parentNode;
        if (cve.querySelector(".Marked") != null || cve.getAttribute("class") === "Marked") {
            return;
        }
        if (["A", "CODE"].includes(cve.tagName)) {
            return;
        }
        const spanElement = document.createElement("span");
        spanElement.setAttribute("class", "Marked");
        selectedTextRange.surroundContents(spanElement);
        const icon = document.createElement("a");
        icon.href = DefaultBaseUrl + id;
        icon.target = "_blank";
        const svg = document.createElement('img');
        svg.setAttribute("style", "background-color: rgb(154, 205, 50);");
        svg.src = "data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIGNsYXNzPSJpY29uIGljb24tdGFibGVyIGljb24tdGFibGVyLWJ1ZyIgd2lkdGg9IjI0IiBoZWlnaHQ9IjI0IiB2aWV3Qm94PSIwIDAgMjQgMjQiIHN0cm9rZS13aWR0aD0iMiIgc3Ryb2tlPSJjdXJyZW50Q29sb3IiIGZpbGw9Im5vbmUiIHN0cm9rZS1saW5lY2FwPSJyb3VuZCIgc3Ryb2tlLWxpbmVqb2luPSJyb3VuZCI+PHBhdGggc3Ryb2tlPSJub25lIiBkPSJNMCAwaDI0djI0SDB6IiBmaWxsPSJub25lIi8+PHBhdGggZD0iTTkgOXYtMWEzIDMgMCAwIDEgNiAwdjEiIC8+PHBhdGggZD0iTTggOWg4YTYgNiAwIDAgMSAxIDN2M2E1IDUgMCAwIDEgLTEwIDB2LTNhNiA2IDAgMCAxIDEgLTMiIC8+PHBhdGggZD0iTTMgMTNsNCAwIiAvPjxwYXRoIGQ9Ik0xNyAxM2w0IDAiIC8+PHBhdGggZD0iTTEyIDIwbDAgLTYiIC8+PHBhdGggZD0iTTQgMTlsMy4zNSAtMiIgLz48cGF0aCBkPSJNMjAgMTlsLTMuMzUgLTIiIC8+PHBhdGggZD0iTTQgN2wzLjc1IDIuNCIgLz48cGF0aCBkPSJNMjAgN2wtMy43NSAyLjQiIC8+PC9zdmc+";
        icon.appendChild(svg);
        spanElement.appendChild(icon);
    }

    function FindCVE() {
        GetBaseURL();
        if (DefaultBaseUrl.startsWith(location.hostname)) {
            return;
        }
        const regex = new RegExp('\\bCVE-\\d{4}-\\d{4,7}\\b', 'gmi');
        let m;
        let cve_arr = [];
        while (m = regex.exec(document.body.innerText)) {
            cve_arr.push(m.toString().toLocaleUpperCase());
        }
        document.designMode = "on";
        Array.from(new Set(cve_arr)).forEach(cve => {
            const sel = window.getSelection();
            sel.collapse(document.body, 0);
            while (window.find(cve)) {
                Mark();
                sel.collapseToEnd();
            }
            Mark();
            sel.collapseToEnd();
        });
        window.scrollTo(0, 0);
        document.designMode = "off";
    }

    FindCVE();
})();