☰

StegLLM

此脚本已不再维护，最新项目详见https://github.com/Rin313/StegLLM。This script is no longer maintenance, please refer to the https://github.com/Rin313/StegLLM for the latest project

// ==UserScript==
// @name         StegLLM
// @namespace    https://github.com/Rin313
// @version      1.03
// @description  此脚本已不再维护，最新项目详见https://github.com/Rin313/StegLLM。This script is no longer maintenance, please refer to the https://github.com/Rin313/StegLLM for the latest project
// @author       Rin
// @match        *://*/*
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_registerMenuCommand
// @license      MIT
// @require      https://cdn.jsdelivr.net/npm/[email protected]/umd/xxhash-wasm.min.js
// ==/UserScript==
(function() {
    'use strict';
    const hostname=window.location.hostname;
    const isMobile = /Android|webOS|iPhone|iPad|iPod|BlackBerry|IEMobile|Opera Mini/i.test(navigator.userAgent);
    window.onerror = function(message, source, lineno, colno, error) {
      if(!source)return;
      if(source.includes("StegLLM"))
        alert(`Error:${error.message}`);
    }
    const createElement = (tag, props = {}, styles = {}) => {
        const el = Object.assign(document.createElement(tag), props);//创建元素
        Object.assign(el.style, styles);//配置styles
        return el;
    };
    let settings = {
        prompt: GM_getValue("prompt", '续写这段散文：'),//如果不存在则使用默认值
    };
    GM_registerMenuCommand('prompt setting', function() {
        let customPrompt = prompt("", settings.prompt);
        if (customPrompt) {
          GM_setValue("prompt", customPrompt);
          settings.prompt=customPrompt;
        }
    });
    const gbkDecoder = new TextDecoder('gb18030');//能解码gbk不支持的符号，比如欧元、表意文字
    const utf8Encoder= new TextEncoder();
    const ranges = [
      [0xA1, 0xA9,  0xA1, 0xFE],
      [0xB0, 0xF7,  0xA1, 0xFE],
      [0x81, 0xA0,  0x40, 0xFE],//从这里开始的三个扩展区，第二个字节要排除0x7F
      [0xAA, 0xFE,  0x40, 0xA0],
      [0xA8, 0xA9,  0x40, 0xA0],
    ];
    let codes,table;
    const punctuations=["？","?","！","!","。","）",")","……"];//,"\n"
    const logitBias=[["　",false],[" ",false],["   ",false],["\n\n",false],["  \n",false],[" \n",false],["�",false],[" �",false],["．",false],["【",false],["】",false],["〈",false],["〉",false]]
    const intercept=2;
    const tokens=1;// const tokens=Math.ceil(intercept/0.75);
    const probs=10;
    function shuffle(array) {
      for (let i = array.length - 1; i > 0; i--) {
          const j = Math.floor(Math.random() * (i + 1)); // 生成 0 到 i 之间的随机整数
          [array[i], array[j]] = [array[j], array[i]]; // 交换元素
      }
      return array;
    }
    function encodeToGBK(str) {
      if(!codes){
        codes=new Uint16Array(22046);//先把全部gbk字符都保存到一个16位整型数组里
        let i = 0,t;
        for (const [b1Begin, b1End, b2Begin, b2End] of ranges) {
          for (let b2 = b2Begin; b2 <= b2End; b2++) {
            if (b2 !== 0x7F) {//反过来遍历，减少判断0x7F的次数
              t = b2 << 8; //不能用16位的codes[i]
              for (let b1 = b1Begin; b1 <= b1End; b1++)
                codes[i++] = t | b1;
            }
          }
        }
      }
      if(!table){
        table = new Uint16Array(65509);//gbk包含¤164，将164左移到0也才省一点点空间
        const str = gbkDecoder.decode(codes);//解码为包含全部gbk字符的字符串
        for (let i = 0; i < str.length; i++){
          table[str.charCodeAt(i)] = codes[i];//unicode到gbk的映射
        }
      }
      const buf = new Uint8Array(str.length * 2);
      let n = 0;
      for (let i = 0; i < str.length; i++) {
        const code = str.charCodeAt(i);
        if (code < 128)
          buf[n++] = code;
        else{
              const gbk = table[code];
              if (gbk === 0)
                throw new Error("文本中存在不支持的符号");//有些编码器会用问号替换来避免报错，但这实际已经发生信息丢失了，不能容忍
              else {
                buf[n++] = gbk;
                buf[n++] = gbk >> 8;
              }
        }
      }
      return buf.subarray(0, n);
    }
    async function readStream(stream) {
      const reader = stream.getReader();
      const chunks = [];
      while (true) {
        const { done, value } = await reader.read();
        if (done) {
          break;
        }
        chunks.push(value);
      }
      const compressedData = new Uint8Array(chunks.reduce((acc, val) => acc + val.length, 0));
      let offset = 0;
      for (const chunk of chunks) {
        compressedData.set(chunk, offset);
        offset += chunk.length;
      }
      return compressedData;
    }
    async function decompress(stream) {
      const ds = new DecompressionStream("deflate-raw");
      const decompressedStream = stream.pipeThrough(ds);
      return readStream(decompressedStream).then(data => {
        return data; // 或者在这里进行一些额外的处理
      });
    }
    // async function passwordToAesCtrKey(password) {
    //   const passwordBuffer = utf8Encoder.encode(password);
    //   // 使用 PBKDF2 算法从密码派生密钥
    //   const keyMaterial = await crypto.subtle.importKey(
    //     "raw",
    //     passwordBuffer,
    //     { name: "PBKDF2" },
    //     false,
    //     ["deriveKey"]
    //   );
    //   // 使用 PBKDF2 派生 AES-CTR 密钥
    //   const aesCtrKey = await crypto.subtle.deriveKey(
    //     {
    //     name: "PBKDF2",
    //     salt: new Uint8Array(0), // 空盐值
    //     iterations: 1000, // 较低的迭代次数
    //     hash: "SHA-256",
    //     },
    //     keyMaterial,
    //     { name: "AES-CTR", length: 256 }, // 指定 AES-CTR 算法和密钥长度 (256位)
    //     true, // 密钥可导出
    //     ["encrypt", "decrypt"] // 密钥用途
    //   );
    //   return aesCtrKey;
    // }
    async function encryptAesCtr(data, str) {
      const buffer=await crypto.subtle.digest('SHA-256', utf8Encoder.encode(str));
      const iv=new Uint8Array(buffer).subarray(0, 16);
      const key= await crypto.subtle.importKey(
        "raw",
        buffer,
        { name: "AES-CTR", length: 256},
        false,
        ["encrypt", "decrypt"]
      );
      const encrypted = await crypto.subtle.encrypt(
        {
          name: "AES-CTR",
          counter: iv,
          length: 64, // 计数器块大小（以位为单位），通常为 64 或 128
        },
        key,data
      );
      return new Uint8Array(encrypted);
    }
    async function decryptAesCtr(data, str) {
      const buffer=await crypto.subtle.digest('SHA-256', utf8Encoder.encode(str));
      const iv=new Uint8Array(buffer).subarray(0, 16);
      const key= await crypto.subtle.importKey(
        "raw",
        buffer,
        { name: "AES-CTR", length: 256},
        false,
        ["encrypt", "decrypt"]
      );
      const decrypted = await crypto.subtle.decrypt(
        {
        name: "AES-CTR",
        counter: iv,
        length: 64,
        },
        key,
        data
      );
      return new Uint8Array(decrypted);
    }
    async function chat(str,complete=false) {
      const body={//有些参数不生效，响应格式也和llama.cpp的api略有不同//在api中设置system_prompt会导致性能严重下降
          // "stream": true,
          "n_predict": tokens,//生成的token数，-1-2048
          "temperature": 1.4,//影响文本的随机性，0-2//较高的温度会增加计算量，较低的温度会导致重复
          // "stop": punctuations,
          "repeat_last_n": 256,
          "repeat_penalty": 1.18,//重复惩罚，1.0为无惩罚
          // "top_p": 0.95,//默认0.95，增大后似乎能增加更多的选词可能性
          // "min_p": 0.05,
          // "tfs_z": 1,
          // "typical_p": 1,
          // "presence_penalty": 0,
          // "frequency_penalty": 0,
          // "mirostat": 0,//关闭mirostat
          // "mirostat_tau": 5,
          // "mirostat_eta": 0.1,
          // "grammar": "",
          // "min_keep": 0,
          // "image_data": [],
          "cache_prompt": true,//提示词复用
          "api_key": "",
          "slot_id": -1,
          "prompt": str,//支持输入多个prompt
          // "response_fields": ["content"],//不生效？
          "top_k": probs,//选词范围，默认40
          "n_probs": probs,//按概率排序的前10个选词，太大或太小都会降低隐写效果
          "logit_bias": logitBias//禁用一些不自然的字符，注意空白符有非常多种
      }
      if(complete){
        body["n_predict"]=9;
        body["stop"]=punctuations;//动态截断
        body["n_probs"]=0;
        body["top_k"]=40;
      }
      const response = await fetch('http://localhost:8080/completion', {
        method: 'POST',
        body: JSON.stringify(body)
      });
      if(!response.ok)
        throw new Error(`HTTP error! status: ${response.status}`);
      const json=(await response.json());
      if(complete)
        return json.content+json.stopping_word;
      const t=json.completion_probabilities[0];
      if(!t)return chat(str);
      else return shuffle(t.probs);
    }
    async function encrypt() {
        const plainText = (await createCustomPrompt("🔒"));
        if(plainText){
          const { h32 } = await xxhash();
          let bytes= encodeToGBK(plainText);
          console.log(bytes);
          const stream=new ReadableStream({
            start(controller) {
              controller.enqueue(bytes);
              controller.close();
            }
          });
          const compressedStream = stream.pipeThrough(new CompressionStream("deflate-raw"),);
          const result=await readStream(compressedStream);
          if(bytes.length>result.length)
            bytes=result;
          console.log(bytes);
          bytes=(await encryptAesCtr(bytes,hostname));
          console.log(bytes);
          let base2=[];
          for (let b of bytes) {
            for(let i=7;i>=0;i--){
              base2.push(b>>i & 0x01);
            }
          }
          console.log(base2);
          const counts=new Uint8Array(base2.length);
          let coverText='';
          for(let i=0;i<base2.length;i++){
            const list=(await chat(settings.prompt+coverText));
            let j=0;
            aaa:for(;j<list.length;j++){
              const t=list[j].tok_str;
              if(t.length===2&&!t.includes("\uFFFD")&&h32(t)%2===base2[i]){
                coverText+=t;
                break;
              }
              else if(t.length===1){
                const list2=(await chat(settings.prompt+coverText+t));
                for(let k=0;k<list2.length;k++){
                  const t2=list2[k].tok_str;
                  if(t2.length===1&&!t2.includes("\uFFFD")&&h32(t+t2)%2===base2[i]){
                    coverText+=t+t2;
                    break aaa;
                  }
                }
              }
            }
            if(j===list.length){
              alert("选词失败，请重新再试");
              return;
            }

          }
          console.log(coverText.length);
          if(!punctuations.includes(coverText[coverText.length-1])){
            for(;;){
              const t=(await chat(settings.prompt+coverText,true));
              if(punctuations.includes(t[t.length-1])){
                coverText+=t;
                break;
              }
            }
          }
          showCustomAlert(coverText);
        }
    }
    async function decrypt() {
      const userInput=(await createCustomPrompt("🗝️"));//粘贴到prompt会导致空白字符等丢失//粘贴到input会导致换行符丢失
      if(userInput){
        const { h32 } = await xxhash();
        let base2 = [];
        let t='';
        //console.log(userInput)
        console.log(userInput.length);
        for (let i = 0; i < userInput.length; i++) {
          t+=userInput[i];
          if(t.length===intercept){
            //console.log(t+" "+t.length)
            base2.push(h32(t)%2);
            t='';
          }
        }
        let bytes = new Uint8Array(base2.length/8);let k=0;
        console.log(base2);
        for(let i=0;i<base2.length;){
          bytes[k++]=base2[i]*128+base2[i+1]*64+base2[i+2]*32+base2[i+3]*16+base2[i+4]*8+base2[i+5]*4+base2[i+6]*2+base2[i+7];
          i+=8;
        }
        console.log(bytes)
        bytes=(await decryptAesCtr(bytes,hostname));
        console.log(bytes)
        const stream=new ReadableStream({
            start(controller) {
              controller.enqueue(bytes);
              controller.close();
            }
        });
        await decompress(stream)
          .then(data=>{bytes=data;})
          .catch(error=>{console.log(error)});
        console.log(bytes)
        alert(gbkDecoder.decode(bytes));
      }
    }
    function swapColors(){
      let t=sidebarButton1.style.backgroundColor;
      sidebarButton1.style.backgroundColor=sidebarButton2.style.backgroundColor;
      sidebarButton2.style.backgroundColor=t;
    }
    const buttonStyles1 = {
      position: 'fixed',
      right: '0', //固定右侧
      zIndex: '9999', // 确保不被覆盖
      cursor: 'pointer',//显示可点击光标
      backgroundColor:'#f56c73',
      border: 'none',
      top:   '42%',
      height: '25px',
      width: '25px',
      overflow: 'hidden',
    };
    const buttonStyles2 = {
        position: 'fixed',
        right: '0', //固定右侧
        zIndex: '9999', // 确保不被覆盖
        cursor: 'pointer',//显示可点击光标
        backgroundColor:'#d87b83',
        border: 'none',
        top:   '47%',
        height: '25px',
        width: '25px',
        overflow: 'hidden',
      };
    const sidebarButton1 = createElement('button', {}, buttonStyles1);
    const sidebarButton2 = createElement('button', {}, buttonStyles2);
    sidebarButton1.addEventListener('mouseenter', () => swapColors() );
    sidebarButton2.addEventListener('mouseenter', () => swapColors() );
    sidebarButton1.addEventListener('click', () => encrypt());
    sidebarButton2.addEventListener('click', () => decrypt());
    document.body.append(sidebarButton1, sidebarButton2);
const showCustomAlert = (text) => {
    // 创建遮罩层
    const overlay = createElement('div', {}, {
        position: 'fixed',
        top: 0,
        left: 0,
        width: '100%',
        height: '100%',
        backgroundColor: 'rgba(0, 0, 0, 0.5)',
        display: 'flex',
        justifyContent: 'center',
        alignItems: 'center',
        zIndex: 9999,
    });

    // 创建弹出框容器
    const alertBox = createElement('div', {}, {
        backgroundColor: '#fff',
        padding: '20px',
        borderRadius: '8px',
        boxShadow: '0 4px 8px rgba(0, 0, 0, 0.2)',
        textAlign: 'center',
        width: '300px',
    });

    // 创建显示的文本
    const message = createElement('p', { textContent: text }, {
        margin: '0 0 20px',
        fontSize: '16px',
        color: '#333',
        wordBreak: 'break-word',
    });

    // 创建按钮容器
    const buttonContainer = createElement('div', {}, {
        display: 'flex',
        justifyContent: 'space-between',
        marginTop: '20px',
    });

    // 创建复制按钮
    const copyButton = createElement('button', { textContent: 'Copy' }, {
        padding: '10px 20px',
        backgroundColor: '#007bff',
        color: '#fff',
        border: 'none',
        borderRadius: '4px',
        cursor: 'pointer',
        fontSize: '14px',
        flex: '1',
        marginRight: '10px',
    });

    // 按钮点击事件 - 复制文本
    copyButton.onclick = () => {
        navigator.clipboard.writeText(text).then(() => {
            alert('Copied to clipboard!');
            document.body.removeChild(overlay);
        });
    };

    // 创建关闭按钮
    const closeButton = createElement('button', { textContent: 'Close' }, {
        padding: '10px 20px',
        backgroundColor: '#dc3545',
        color: '#fff',
        border: 'none',
        borderRadius: '4px',
        cursor: 'pointer',
        fontSize: '14px',
        flex: '1',
    });

    // 关闭按钮点击事件
    closeButton.onclick = () => {
        document.body.removeChild(overlay);
    };

    // 组装按钮容器
    buttonContainer.appendChild(copyButton);
    buttonContainer.appendChild(closeButton);

    // 组装弹出框
    alertBox.appendChild(message);
    alertBox.appendChild(buttonContainer);
    overlay.appendChild(alertBox);
    document.body.appendChild(overlay);
};
const createCustomPrompt = (placeholder = "请输入内容...") => {
  return new Promise((resolve) => {
    // 创建一个覆盖整个页面的背景遮罩
    const overlay = createElement('div', {}, {
      position: 'fixed',
      top: '0',
      left: '0',
      width: '100vw',
      height: '100vh',
      backgroundColor: 'rgba(0, 0, 0, 0.5)',
      display: 'flex',
      justifyContent: 'center',
      alignItems: 'center',
      zIndex: '1000',
    });

    // 创建一个容器来放置textarea和按钮
    const promptContainer = createElement('div', {}, {
      backgroundColor: '#fff',
      padding: '20px',
      borderRadius: '8px',
      boxShadow: '0 4px 10px rgba(0, 0, 0, 0.3)',
      display: 'flex',
      flexDirection: 'column',
      alignItems: 'center',
      minWidth: '300px',
    });

    // 创建textarea
    const textarea = createElement(
      'textarea',
      {
        placeholder: placeholder,
      },
      {
        width: '100%',
        height: '100px',
        marginBottom: '10px',
        padding: '10px',
        fontSize: '16px',
        borderRadius: '4px',
        border: '1px solid #ddd',
        outline: 'none',
        resize: 'none',
      }
    );

    // 创建提交按钮
    const submitButton = createElement(
      'button',
      {
        innerText: '提交',
        onclick: () => {
          const value = textarea.value;
          resolve(value); // 当点击提交时，resolve Promise 并返回值
          document.body.removeChild(overlay); // 移除遮罩层
        },
      },
      {
        padding: '10px 20px',
        fontSize: '16px',
        backgroundColor: '#007BFF',
        color: '#fff',
        border: 'none',
        borderRadius: '4px',
        cursor: 'pointer',
      }
    );

    // 提交按钮 hover 样式
    submitButton.addEventListener('mouseover', () => {
      submitButton.style.backgroundColor = '#0056b3';
    });
    submitButton.addEventListener('mouseout', () => {
      submitButton.style.backgroundColor = '#007BFF';
    });
    overlay.addEventListener('click', (event) => {
      if (event.target === overlay) {
        resolve(null); // 用户取消操作时，返回 null
        document.body.removeChild(overlay); // 移除遮罩层
      }
    });
    // 把textarea和按钮添加到容器中
    promptContainer.appendChild(textarea);
    promptContainer.appendChild(submitButton);

    // 把容器添加到遮罩层中
    overlay.appendChild(promptContainer);

    // 把遮罩层添加到body中
    document.body.appendChild(overlay);
  });
};
})();