mirror of
https://github.com/fofolee/uTools-quickcommand.git
synced 2025-09-24 13:03:30 +08:00
拆分数据处理分类,新增编码加密、数学计算分类
This commit is contained in:
261
plugin/lib/quickcomposer/coding/crypto.js
Normal file
261
plugin/lib/quickcomposer/coding/crypto.js
Normal file
@@ -0,0 +1,261 @@
|
||||
const sm2 = require("sm-crypto").sm2;
|
||||
const sm4 = require("sm-crypto").sm4;
|
||||
const CryptoJS = require("crypto-js");
|
||||
const NodeForge = require("node-forge");
|
||||
const { processSecret, dataConv } = require("./utils");
|
||||
|
||||
const crypto = {
|
||||
// 非对称加解密
|
||||
asymmetricCrypto: function (config) {
|
||||
const {
|
||||
text,
|
||||
algorithm,
|
||||
operation,
|
||||
format = "Base64",
|
||||
publicKey = { key: "", codec: "Pem" },
|
||||
privateKey = { key: "", codec: "Pem" },
|
||||
padding = "RSAES-PKCS1-V1_5",
|
||||
cipherMode = 1,
|
||||
} = config;
|
||||
|
||||
if (algorithm === "SM2") {
|
||||
if (operation === "encrypt") {
|
||||
if (!publicKey.key) throw "缺少公钥";
|
||||
// 转换公钥格式
|
||||
const hexPubKey =
|
||||
publicKey.codec === "Hex"
|
||||
? publicKey.key
|
||||
: dataConv(publicKey.key, publicKey.codec, "Hex");
|
||||
// 加密
|
||||
const cipher = sm2.doEncrypt(text, hexPubKey, cipherMode);
|
||||
// 转换输出格式
|
||||
return format === "Base64" ? dataConv(cipher, "Hex", "Base64") : cipher;
|
||||
} else {
|
||||
if (!privateKey.key) throw "缺少私钥";
|
||||
// 转换私钥格式
|
||||
const hexPriKey =
|
||||
privateKey.codec === "Hex"
|
||||
? privateKey.key
|
||||
: dataConv(privateKey.key, privateKey.codec, "Hex");
|
||||
// 转换输入格式
|
||||
const hexCipher =
|
||||
format === "Base64" ? dataConv(text, "Base64", "Hex") : text;
|
||||
// 解密
|
||||
const msg = sm2.doDecrypt(hexCipher, hexPriKey, cipherMode, {
|
||||
output: "utf8",
|
||||
});
|
||||
if (!msg) throw "解密失败";
|
||||
return msg;
|
||||
}
|
||||
} else if (algorithm === "RSA") {
|
||||
if (operation === "encrypt") {
|
||||
if (!publicKey.key) throw "缺少公钥";
|
||||
// 转换公钥格式
|
||||
let formattedPubKey = publicKey.key;
|
||||
if (publicKey.codec !== "Pem") {
|
||||
formattedPubKey =
|
||||
"-----BEGIN RSA PUBLIC KEY-----\n" +
|
||||
dataConv(publicKey.key, publicKey.codec, "Base64") +
|
||||
"\n-----END RSA PUBLIC KEY-----";
|
||||
}
|
||||
formattedPubKey = formattedPubKey.replace(/\\n/g, "\n");
|
||||
|
||||
// 创建 RSA 公钥对象
|
||||
const publicKeyObj = NodeForge.pki.publicKeyFromPem(formattedPubKey);
|
||||
// 将文本转换为二进制数据
|
||||
const binaryData = NodeForge.util.encodeUtf8(text);
|
||||
// 使用指定的填充方式加密
|
||||
const encrypted = publicKeyObj.encrypt(binaryData, padding);
|
||||
// 转换输出格式
|
||||
return format === "Base64"
|
||||
? dataConv(encrypted, "binary", "Base64")
|
||||
: dataConv(encrypted, "binary", "Hex");
|
||||
} else {
|
||||
if (!privateKey.key) throw "缺少私钥";
|
||||
// 转换私钥格式
|
||||
let formattedPriKey = privateKey.key;
|
||||
if (privateKey.codec !== "Pem") {
|
||||
formattedPriKey =
|
||||
"-----BEGIN RSA PRIVATE KEY-----\n" +
|
||||
dataConv(privateKey.key, privateKey.codec, "Base64") +
|
||||
"\n-----END RSA PRIVATE KEY-----";
|
||||
}
|
||||
formattedPriKey = formattedPriKey.replace(/\\n/g, "\n");
|
||||
|
||||
// 创建 RSA 私钥对象
|
||||
const privateKeyObj = NodeForge.pki.privateKeyFromPem(formattedPriKey);
|
||||
// 转换输入格式
|
||||
const binary =
|
||||
format === "Base64"
|
||||
? dataConv(text, "Base64", "binary")
|
||||
: dataConv(text, "Hex", "binary");
|
||||
// 解密
|
||||
try {
|
||||
const decrypted = privateKeyObj.decrypt(binary, padding);
|
||||
// 将二进制数据转换回文本
|
||||
return NodeForge.util.decodeUtf8(decrypted);
|
||||
} catch (e) {
|
||||
console.error(e);
|
||||
throw "解密失败";
|
||||
}
|
||||
}
|
||||
}
|
||||
throw "不支持的算法";
|
||||
},
|
||||
|
||||
// 对称加解密
|
||||
symmetricCrypto: function (config) {
|
||||
const {
|
||||
text,
|
||||
algorithm,
|
||||
mode = "CBC",
|
||||
padding = "Pkcs7",
|
||||
key = { value: "", codec: "Utf8" },
|
||||
keyLength = 128,
|
||||
operation = "encrypt",
|
||||
format = "Base64",
|
||||
iv = { value: "", codec: "Utf8" },
|
||||
} = config;
|
||||
|
||||
// 处理密钥和IV
|
||||
const processedKey = processSecret(key.value, key.codec, keyLength / 8);
|
||||
const processedIV =
|
||||
mode === "ECB"
|
||||
? undefined
|
||||
: processSecret(iv?.value || key.value, iv?.codec || key.codec, 16);
|
||||
|
||||
// SM4 使用专门的库
|
||||
if (algorithm === "SM4") {
|
||||
const hexKey = processedKey.toString();
|
||||
// 处理输入文本格式
|
||||
let inputText = text;
|
||||
if (operation === "decrypt") {
|
||||
inputText =
|
||||
format === "Base64"
|
||||
? CryptoJS.enc.Base64.parse(text).toString(CryptoJS.enc.Hex)
|
||||
: text;
|
||||
}
|
||||
|
||||
if (mode === "CBC") {
|
||||
const hexIv = processedIV ? processedIV.toString() : "0".repeat(32);
|
||||
const result =
|
||||
operation === "encrypt"
|
||||
? sm4.encrypt(inputText, hexKey, {
|
||||
mode: "cbc",
|
||||
iv: hexIv,
|
||||
padding: padding.toLowerCase(),
|
||||
})
|
||||
: sm4.decrypt(inputText, hexKey, {
|
||||
mode: "cbc",
|
||||
iv: hexIv,
|
||||
padding: padding.toLowerCase(),
|
||||
});
|
||||
// 处理输出格式
|
||||
return operation === "encrypt" && format === "Base64"
|
||||
? CryptoJS.enc.Base64.stringify(CryptoJS.enc.Hex.parse(result))
|
||||
: result;
|
||||
}
|
||||
|
||||
const result =
|
||||
operation === "encrypt"
|
||||
? sm4.encrypt(inputText, hexKey, {
|
||||
padding: padding.toLowerCase(),
|
||||
})
|
||||
: sm4.decrypt(inputText, hexKey, {
|
||||
padding: padding.toLowerCase(),
|
||||
});
|
||||
// 处理输出格式
|
||||
return operation === "encrypt" && format === "Base64"
|
||||
? CryptoJS.enc.Base64.stringify(CryptoJS.enc.Hex.parse(result))
|
||||
: result;
|
||||
}
|
||||
|
||||
// 准备加密配置
|
||||
const cryptoConfig = {
|
||||
mode: CryptoJS.mode[mode],
|
||||
padding: CryptoJS.pad[padding],
|
||||
};
|
||||
|
||||
// 添加 IV(如果需要)
|
||||
if (mode !== "ECB" && processedIV) {
|
||||
cryptoConfig.iv = processedIV;
|
||||
}
|
||||
|
||||
// 加密/解密操作
|
||||
if (operation === "encrypt") {
|
||||
let encrypted;
|
||||
const inputText = CryptoJS.enc.Utf8.parse(text);
|
||||
|
||||
switch (algorithm) {
|
||||
case "AES":
|
||||
if (mode === "GCM") {
|
||||
const cipher = NodeForge.cipher.createCipher(
|
||||
"AES-GCM",
|
||||
processedKey.toString()
|
||||
);
|
||||
cipher.start({
|
||||
iv: cryptoConfig.iv ? cryptoConfig.iv.toString() : "",
|
||||
tagLength: 128,
|
||||
});
|
||||
cipher.update(NodeForge.util.createBuffer(text, "utf8"));
|
||||
cipher.finish();
|
||||
return {
|
||||
enc: cipher.output.toHex(),
|
||||
tag: cipher.mode.tag.toHex(),
|
||||
};
|
||||
}
|
||||
encrypted = CryptoJS.AES.encrypt(
|
||||
inputText,
|
||||
processedKey,
|
||||
cryptoConfig
|
||||
);
|
||||
break;
|
||||
default:
|
||||
throw "不支持的算法";
|
||||
}
|
||||
return encrypted.ciphertext.toString(CryptoJS.enc[format]);
|
||||
} else {
|
||||
// 解密
|
||||
if (algorithm === "AES" && mode === "GCM") {
|
||||
try {
|
||||
const { enc, tag } = JSON.parse(text);
|
||||
const decipher = NodeForge.cipher.createDecipher(
|
||||
"AES-GCM",
|
||||
processedKey.toString()
|
||||
);
|
||||
decipher.start({
|
||||
iv: cryptoConfig.iv ? cryptoConfig.iv.toString() : "",
|
||||
tag: NodeForge.util.createBuffer(tag, "hex"),
|
||||
});
|
||||
decipher.update(NodeForge.util.createBuffer(enc, "hex"));
|
||||
const pass = decipher.finish();
|
||||
return pass ? decipher.output.toString() : null;
|
||||
} catch (e) {
|
||||
throw "解密失败";
|
||||
}
|
||||
}
|
||||
|
||||
// 将输入转换为 CipherParams 格式
|
||||
const ciphertext = CryptoJS.enc[format].parse(text);
|
||||
const cipherParams = CryptoJS.lib.CipherParams.create({
|
||||
ciphertext: ciphertext,
|
||||
});
|
||||
|
||||
let decrypted;
|
||||
switch (algorithm) {
|
||||
case "AES":
|
||||
decrypted = CryptoJS.AES.decrypt(
|
||||
cipherParams,
|
||||
processedKey,
|
||||
cryptoConfig
|
||||
);
|
||||
break;
|
||||
default:
|
||||
throw "不支持的算法";
|
||||
}
|
||||
return CryptoJS.enc.Utf8.stringify(decrypted);
|
||||
}
|
||||
},
|
||||
};
|
||||
|
||||
module.exports = crypto;
|
48
plugin/lib/quickcomposer/coding/encoder.js
Normal file
48
plugin/lib/quickcomposer/coding/encoder.js
Normal file
@@ -0,0 +1,48 @@
|
||||
const { dataConv } = require("./utils");
|
||||
|
||||
const encoder = {
|
||||
// base64 编码
|
||||
base64Encode: function (text) {
|
||||
return dataConv(text, "utf8", "base64");
|
||||
},
|
||||
// base64 解码
|
||||
base64Decode: function (text) {
|
||||
return dataConv(text, "base64", "utf8");
|
||||
},
|
||||
// URL 编码
|
||||
urlEncode: function (text) {
|
||||
return encodeURIComponent(text);
|
||||
},
|
||||
// URL 解码
|
||||
urlDecode: function (text) {
|
||||
return decodeURIComponent(text);
|
||||
},
|
||||
// html 编码
|
||||
htmlEncode: function (text) {
|
||||
return text
|
||||
.replace(/&/g, "&")
|
||||
.replace(/</g, "<")
|
||||
.replace(/>/g, ">")
|
||||
.replace(/"/g, """)
|
||||
.replace(/'/g, "'");
|
||||
},
|
||||
// html 解码
|
||||
htmlDecode: function (text) {
|
||||
return text
|
||||
.replace(/&/g, "&")
|
||||
.replace(/</g, "<")
|
||||
.replace(/>/g, ">")
|
||||
.replace(/"/g, '"')
|
||||
.replace(/'/g, "'");
|
||||
},
|
||||
// 十六进制
|
||||
hexEncode: function (text) {
|
||||
return dataConv(text, "utf8", "hex");
|
||||
},
|
||||
// 十六进制解码
|
||||
hexDecode: function (text) {
|
||||
return dataConv(text, "hex", "utf8");
|
||||
},
|
||||
};
|
||||
|
||||
module.exports = encoder;
|
27
plugin/lib/quickcomposer/coding/hash.js
Normal file
27
plugin/lib/quickcomposer/coding/hash.js
Normal file
@@ -0,0 +1,27 @@
|
||||
const NodeForge = require("node-forge");
|
||||
const sm3 = require("sm-crypto").sm3;
|
||||
|
||||
const hash = {
|
||||
// MD5 哈希
|
||||
md5Hash: function (text) {
|
||||
return NodeForge.md.md5.create().update(text).digest().toHex();
|
||||
},
|
||||
// SHA1 哈希
|
||||
sha1Hash: function (text) {
|
||||
return NodeForge.md.sha1.create().update(text).digest().toHex();
|
||||
},
|
||||
// SHA256 哈希
|
||||
sha256Hash: function (text) {
|
||||
return NodeForge.md.sha256.create().update(text).digest().toHex();
|
||||
},
|
||||
// SHA512 哈希
|
||||
sha512Hash: function (text) {
|
||||
return NodeForge.md.sha512.create().update(text).digest().toHex();
|
||||
},
|
||||
// SM3 哈希
|
||||
sm3Hash: function (text) {
|
||||
return sm3(text);
|
||||
},
|
||||
};
|
||||
|
||||
module.exports = hash;
|
9
plugin/lib/quickcomposer/coding/index.js
Normal file
9
plugin/lib/quickcomposer/coding/index.js
Normal file
@@ -0,0 +1,9 @@
|
||||
const crypto = require("./crypto");
|
||||
const hash = require("./hash");
|
||||
const encoder = require("./encoder");
|
||||
|
||||
module.exports = {
|
||||
...crypto,
|
||||
...hash,
|
||||
...encoder,
|
||||
};
|
30
plugin/lib/quickcomposer/coding/utils.js
Normal file
30
plugin/lib/quickcomposer/coding/utils.js
Normal file
@@ -0,0 +1,30 @@
|
||||
const CryptoJS = require("crypto-js");
|
||||
|
||||
// 数据编码转换
|
||||
const dataConv = (str, fromCodec, toCodec) => {
|
||||
// 特殊处理 PEM 格式
|
||||
if (fromCodec.toLowerCase() === "pem") {
|
||||
const pemContent = str
|
||||
.replace(/-----(BEGIN|END)[^-]+-----/g, "")
|
||||
.replace(/[\r\n]/g, "");
|
||||
return Buffer.from(pemContent, "base64").toString(toCodec.toLowerCase());
|
||||
}
|
||||
// 其他格式直接转换
|
||||
return Buffer.from(str, fromCodec.toLowerCase()).toString(
|
||||
toCodec.toLowerCase()
|
||||
);
|
||||
};
|
||||
|
||||
// 处理密钥和IV
|
||||
const processSecret = (key, codec, len) => {
|
||||
// 转换成 hex 并填充到指定长度
|
||||
const hexStr = dataConv(key, codec, "hex")
|
||||
.padEnd(len * 2, "0")
|
||||
.slice(0, len * 2);
|
||||
return CryptoJS.enc.Hex.parse(hexStr);
|
||||
};
|
||||
|
||||
module.exports = {
|
||||
dataConv,
|
||||
processSecret,
|
||||
};
|
Reference in New Issue
Block a user