杭州明婳科技
126 lines
3.4 KiB
Go
126 lines
3.4 KiB
Go
package utils
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/md5" // #nosec G501 -- exported checksum helper for non-password/non-security use; callers needing security should use SHA256/HMAC.
|
||
"crypto/sha1" // #nosec G505 -- exported checksum helper for legacy/non-security use; callers needing security should use SHA256/HMAC.
|
||
"crypto/sha256"
|
||
"encoding/base64"
|
||
"encoding/hex"
|
||
"hash"
|
||
"io"
|
||
"os"
|
||
)
|
||
|
||
// MD5 计算字符串的 MD5 哈希值
|
||
// 注意: 不应用于密码存储
|
||
func MD5(s string) string {
|
||
return MD5Bytes([]byte(s))
|
||
}
|
||
|
||
// MD5Bytes 计算字节数组的 MD5 哈希值
|
||
func MD5Bytes(data []byte) string {
|
||
// #nosec G401 -- MD5 helper is retained for legacy checksums, not password storage or security decisions.
|
||
h := md5.New()
|
||
h.Write(data)
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
// SHA1 计算字符串的 SHA1 哈希值
|
||
// 注意: 不应用于密码存储
|
||
func SHA1(s string) string {
|
||
// #nosec G401 -- SHA1 helper is retained for legacy checksums, not password storage or security decisions.
|
||
h := sha1.New()
|
||
h.Write([]byte(s))
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
// SHA256 计算字符串的 SHA256 哈希值
|
||
func SHA256(s string) string {
|
||
h := sha256.New()
|
||
h.Write([]byte(s))
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
// SHA256Bytes 计算字节数组的 SHA256 哈希值
|
||
func SHA256Bytes(data []byte) string {
|
||
h := sha256.New()
|
||
h.Write(data)
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
// HashFile 计算文件的哈希值。
|
||
//
|
||
// M-F 修复：改为流式 io.Copy(h, f)，不再经 ReadFile 把整文件读入内存——原实现大文件 OOM。
|
||
// 流式哈希内存占用恒定、可处理任意大小文件，与 hash.Hash 的 io.Writer 接口契合。
|
||
// 调用方负责选择哈希算法（如 sha256.New）。
|
||
func HashFile(path string, newHash func() hash.Hash) (string, error) {
|
||
// #nosec G304 -- generic hashing helper intentionally opens caller-provided trusted local paths.
|
||
f, err := os.Open(path)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
defer f.Close()
|
||
h := newHash()
|
||
if _, err := io.Copy(h, f); err != nil {
|
||
return "", err
|
||
}
|
||
return hex.EncodeToString(h.Sum(nil)), nil
|
||
}
|
||
|
||
// Base64Encode Base64 编码
|
||
func Base64Encode(data []byte) string {
|
||
return base64.StdEncoding.EncodeToString(data)
|
||
}
|
||
|
||
// Base64Decode Base64 解码
|
||
func Base64Decode(s string) ([]byte, error) {
|
||
return base64.StdEncoding.DecodeString(s)
|
||
}
|
||
|
||
// Base64URLEncode URL 安全的 Base64 编码
|
||
func Base64URLEncode(data []byte) string {
|
||
return base64.URLEncoding.EncodeToString(data)
|
||
}
|
||
|
||
// Base64URLDecode URL 安全的 Base64 解码
|
||
func Base64URLDecode(s string) ([]byte, error) {
|
||
return base64.URLEncoding.DecodeString(s)
|
||
}
|
||
|
||
// Nl2br 将换行符替换为 <br> 标签
|
||
func Nl2br(s string, isXhtml bool) string {
|
||
var br string
|
||
if isXhtml {
|
||
br = "<br />"
|
||
} else {
|
||
br = "<br>"
|
||
}
|
||
|
||
var buf bytes.Buffer
|
||
runes := []rune(s)
|
||
length := len(runes)
|
||
|
||
for i, r := range runes {
|
||
switch r {
|
||
case '\n':
|
||
// \n\r（罕见顺序）作为一个换行处理；普通 \n 单独处理。
|
||
// （N4：原条件含 r == '\r' 半，但本 case 内 r 恒为 '\n'，该半恒假，已清理。）
|
||
if i+1 < length && runes[i+1] == '\r' {
|
||
buf.WriteString(br)
|
||
continue
|
||
}
|
||
buf.WriteString(br)
|
||
case '\r':
|
||
// \r\n 由上面的 \n 分支处理（\r 在此跳过）；单独 \r 作为一个换行。
|
||
if i+1 < length && runes[i+1] == '\n' {
|
||
continue
|
||
}
|
||
buf.WriteString(br)
|
||
default:
|
||
buf.WriteRune(r)
|
||
}
|
||
}
|
||
return buf.String()
|
||
}
|