Skip to content

SecureValue API

SecureValue 类型用于安全存储敏感数据,提供内存锁定、自动清零和掩码功能。

线程安全

SecureValue 的所有方法都是线程安全的,可在多个 goroutine 中并发使用:

  • 读取方法String()Bytes()Length()Masked())使用读锁,支持并发读取
  • 关闭方法Close()Release())使用写锁,确保安全清零
  • 状态检查IsClosed()IsMemoryLocked())使用原子操作
go
secret := env.GetSecure("API_KEY")
if secret != nil {
    defer secret.Release()

    // 并发读取安全
    go func() { fmt.Println(secret.Masked()) }()
    go func() { fmt.Println(secret.Length()) }()
}

注意

Close()Release() 只应调用一次。重复调用是安全的但无效。

创建

NewSecureValue

go
func NewSecureValue(value string) *SecureValue

创建安全值包装器。

参数:

  • value - 要保护的字符串值

返回:

  • *SecureValue - 安全值对象

行为:

  • 使用对象池减少分配
  • 设置 GC 终结器自动清零
  • 如果启用内存锁定,尝试锁定内存(失败时静默忽略)
go
secret := env.NewSecureValue("my-secret-password")
defer secret.Release()  // 或 Close()

NewSecureValueStrict

go
func NewSecureValueStrict(value string) (*SecureValue, error)

创建安全值,如果内存锁定失败则返回错误。

参数:

  • value - 要保护的字符串值

返回:

  • *SecureValue - 安全值对象
  • error - 内存锁定错误(仅严格模式)
go
env.SetMemoryLockEnabled(true)
env.SetMemoryLockStrict(true)

secret, err := env.NewSecureValueStrict("my-secret")
if err != nil {
    // 内存锁定失败
    log.Printf("Warning: %v", err)
}
if secret != nil {
    defer secret.Release()
}

GetSecure (Loader 方法)

go
func (l *Loader) GetSecure(key string) *SecureValue

从加载器获取安全值。

参数:

  • key - 键名

返回:

  • *SecureValue - 安全值的防御性副本,调用者负责释放;键不存在或加载器关闭时返回 nil
go
secret := loader.GetSecure("API_KEY")
if secret != nil {
    defer secret.Release()
    // 使用 secret
}

防御性副本

GetSecure 返回的是原始值的副本,独立于父 Loader。调用者负责调用 Release()Close() 释放。


方法

String

go
func (sv *SecureValue) String() string

返回掩码表示,安全用于日志和格式化。实现了 fmt.Stringer 接口,防止通过 fmt.Printflog.Println 或错误包装意外泄露密钥。

返回:

  • string - 掩码表示(如 [SECURE:32 bytes]),nil 时返回 [NIL]
go
secret := env.GetSecure("PASSWORD")
if secret != nil {
    log.Printf("Password: %s", secret)  // 安全,输出掩码表示
    // 等效于 log.Printf("Password: %s", secret.Masked())
}

注意

String() 返回的是掩码表示,不是明文值。如需获取明文值,请使用 Reveal()


Reveal

go
func (sv *SecureValue) Reveal() string

返回明文值。调用者负责安全处理返回的字符串 —— 避免日志记录、序列化或存储到持久化位置。仅在需要实际值用于加密操作、API 调用或类似安全处理时使用。

返回:

  • string - 明文值,已关闭或 nil 时返回空字符串
go
secret := env.GetSecure("API_KEY")
if secret != nil {
    defer secret.Release()
    plaintext := secret.Reveal()  // 获取明文值
    // 使用 plaintext 进行 API 调用等安全操作
    _ = plaintext
}

安全警告

Reveal() 返回的是明文字符串。Go 字符串不可变,无法手动清零。仅在必要时使用,并避免将返回值记录到日志或存储。


Bytes

go
func (sv *SecureValue) Bytes() []byte

返回值的字节切片副本。调用者负责使用 ClearBytes 清零。

返回:

  • []byte - 值的字节副本,已关闭返回 nil
go
secret := env.GetSecure("API_KEY")
if secret != nil {
    data := secret.Bytes()
    defer env.ClearBytes(data)  // 使用后清零
    // 使用 data
}

Length

go
func (sv *SecureValue) Length() int

返回值的长度,不暴露内容。

返回:

  • int - 值长度,已关闭返回 0
go
secret := env.GetSecure("API_KEY")
if secret != nil {
    fmt.Printf("API Key length: %d\n", secret.Length())
}

Masked

go
func (sv *SecureValue) Masked() string

返回掩码后的值,用于日志输出。

返回:

  • string - 掩码表示

输出格式:

  • 已关闭:[CLOSED]
  • 空值:[SECURE:0 bytes]
  • 正常:[SECURE:N bytes][SECURE:N bytes locked][SECURE:N bytes lock-failed][SECURE:N bytes unlocked]
go
secret := env.GetSecure("API_KEY")
if secret != nil {
    log.Printf("API Key: %s", secret.Masked())
    // 输出:API Key: [SECURE:32 bytes]
    // 注:仅当启用内存锁定(SetMemoryLockEnabled(true))且锁定成功时,
    // 掩码才追加 " locked" 后缀(另有 " lock-failed" / " unlocked")
}

Close

go
func (sv *SecureValue) Close() error

安全清零内存并关闭对象。

返回:

  • error - 总是返回 nil

行为:

  • 安全清零内部数据
  • 标记为已关闭
  • 归还到对象池
go
secret := env.GetSecure("TOKEN")
if secret != nil {
    defer secret.Close()
    // Close 后内存被清零
}

Release

go
func (sv *SecureValue) Release()

清零内存并归还到对象池。

行为:

  • 安全清零内部数据
  • 清除 GC 终结器
  • 归还到对象池供重用
go
secret := env.GetSecure("KEY")
if secret != nil {
    defer secret.Release()
    // Release 后内存被清零且对象归还池
}

Close vs Release

  • Close() - 仅清零,不归还池
  • Release() - 清零并归还池(推荐用于高频场景)

IsClosed

go
func (sv *SecureValue) IsClosed() bool

检查对象是否已关闭。

返回:

  • bool - 是否已关闭
go
if secret.IsClosed() {
    // 对象已关闭,不可使用
}

IsMemoryLocked

go
func (sv *SecureValue) IsMemoryLocked() bool

检查内存是否被锁定(防止交换到磁盘)。

返回:

  • bool - 是否已锁定
go
if secret.IsMemoryLocked() {
    fmt.Println("Memory is locked, protected from swapping")
}

MemoryLockError

go
func (sv *SecureValue) MemoryLockError() error

返回内存锁定尝试的错误(如果有)。

返回:

  • error - 锁定错误,成功或未尝试返回 nil
go
if err := secret.MemoryLockError(); err != nil {
    log.Printf("Memory lock failed: %v", err)
}

内存锁定配置

SetMemoryLockEnabled

go
func SetMemoryLockEnabled(enabled bool)

全局启用/禁用内存锁定。影响所有新创建的 SecureValue。

参数:

  • enabled - 是否启用
go
package main

import "github.com/cybergodev/env"

func main() {
    // 应用启动时启用
    env.SetMemoryLockEnabled(true)

    // 后续所有 SecureValue 都会尝试锁定
}

IsMemoryLockEnabled

go
func IsMemoryLockEnabled() bool

检查内存锁定是否启用。

返回:

  • bool - 是否启用
go
if env.IsMemoryLockEnabled() {
    // 内存锁定已启用
}

SetMemoryLockStrict

go
func SetMemoryLockStrict(strict bool)

设置严格模式。启用后,NewSecureValueStrict 在锁定失败时返回错误。

参数:

  • strict - 是否启用严格模式
go
env.SetMemoryLockEnabled(true)
env.SetMemoryLockStrict(true)

secret, err := env.NewSecureValueStrict("sensitive-data")
if err != nil {
    // 锁定失败
}

IsMemoryLockStrict

go
func IsMemoryLockStrict() bool

检查是否为严格模式。

返回:

  • bool - 是否启用
go
strict := env.IsMemoryLockStrict()

IsMemoryLockSupported

go
func IsMemoryLockSupported() bool

检查当前平台是否支持内存锁定。

返回:

  • bool - 是否支持
平台支持
Linux
macOS
Windows
FreeBSD
wasm

注意

返回 true 只表示平台支持,不表示进程有足够权限。Linux 需要 CAP_IPC_LOCK 或 root 权限。

go
if env.IsMemoryLockSupported() {
    env.SetMemoryLockEnabled(true)
}

安全工具函数

ClearBytes

go
func ClearBytes(b []byte)

安全清零字节切片。使用后立即清零敏感数据。

参数:

  • b - 要清零的字节切片
go
sensitive := []byte("secret-data")
// 使用...
env.ClearBytes(sensitive)
// sensitive 现在全是 0

IsSensitiveKey

go
func IsSensitiveKey(key string) bool

检查键名是否匹配敏感模式。

参数:

  • key - 键名

返回:

  • bool - 是否敏感
go
if env.IsSensitiveKey("DB_PASSWORD") {
    // 敏感键,使用安全方式处理
    secret := env.GetSecure("DB_PASSWORD")
    if secret != nil {
        defer secret.Release()
    }
}

敏感模式: password, secret, token, key, api_key, credential 等


MaskValue

go
func MaskValue(key, value string) string

根据键的敏感性返回掩码值。

参数:

  • key - 键名
  • value - 原始值

返回:

  • string - 掩码后的值
go
// 敏感键 - 返回 [MASKED:N chars] 格式
masked := env.MaskValue("API_KEY", "secret123")
// 返回:[MASKED:9 chars]

// 非敏感键 - 返回原值(超过 20 字符则截断)
masked := env.MaskValue("APP_NAME", "myapp")
// 返回:myapp

MaskKey

go
func MaskKey(key string) string

掩码键名用于日志。

参数:

  • key - 键名

返回:

  • string - 掩码后的键名
go
masked := env.MaskKey("DB_PASSWORD")
// 返回:DB***

SanitizeForLog

go
func SanitizeForLog(s string) string

清理字符串中的敏感键值对信息。自动检测并掩码 key=value 格式中的敏感值。

参数:

  • s - 原始字符串

返回:

  • string - 清理后的字符串
go
// 自动掩码敏感键值对
msg := "Connected with password=secret123 api_key=abc123"
clean := env.SanitizeForLog(msg)
// 返回:"Connected with password=[MASKED] api_key=[MASKED]"

MaskSensitiveInString

go
func MaskSensitiveInString(s string) string

掩码字符串中的潜在敏感内容。截断超过 50 字符的字符串。

参数:

  • s - 原始字符串

返回:

  • string - 掩码后的字符串
go
// 长字符串会被截断(保留前 47 个字符并追加 "...")
long := "This is a very long string that exceeds 50 characters"
clean := env.MaskSensitiveInString(long)
// 返回:"This is a very long string that exceeds 50 char..."

使用场景

用于截断可能包含敏感数据的长字符串。如需自动掩码敏感键值对,使用 SanitizeForLog


完整示例

go
package main

import (
    "fmt"
    "log"

    "github.com/cybergodev/env"
)

func main() {
    // 检查并启用内存锁定
    if env.IsMemoryLockSupported() {
        env.SetMemoryLockEnabled(true)
        fmt.Println("Memory locking enabled")
    }

    // 加载环境变量
    if err := env.Load(".env"); err != nil {
        log.Printf("Warning: %v", err)
    }

    // 安全获取敏感值
    apiKey := env.GetSecure("API_KEY")
    if apiKey == nil {
        log.Fatal("API_KEY not found")
    }
    defer apiKey.Release()

    // 安全使用
    fmt.Printf("API Key length: %d\n", apiKey.Length())
    fmt.Printf("API Key (masked): %s\n", apiKey.Masked())

    // 检查内存锁定状态
    if apiKey.IsMemoryLocked() {
        fmt.Println("Memory is locked")
    }

    // 检查锁定错误
    if err := apiKey.MemoryLockError(); err != nil {
        fmt.Printf("Memory lock warning: %v\n", err)
    }

    // 传递给其他函数
    connectAPI(apiKey.Reveal())

    // 使用安全工具函数
    logMessage := "Processing with API_KEY=secret"
    safeMessage := env.SanitizeForLog(logMessage)
    fmt.Println(safeMessage)  // Processing with API_KEY=[MASKED]
}

func connectAPI(key string) {
    // 使用密钥连接...
    fmt.Printf("Connecting with key of length %d\n", len(key))
}

内部实现

对象池

SecureValue 使用 sync.Pool 减少内存分配:

go
var secureValuePool = sync.Pool{
    New: func() interface{} {
        return &SecureValue{}
    },
}

GC 终结器

创建时设置 GC 终结器,确保垃圾回收时自动清零:

go
runtime.SetFinalizer(sv, (*SecureValue).finalize)

安全清零

使用 unsafe.Pointer 防止编译器优化:

go
func (sv *SecureValue) clearData() {
    dataPtr := unsafe.Pointer(&sv.data[0])
    for i := range sv.data {
        *(*byte)(unsafe.Pointer(uintptr(dataPtr) + uintptr(i))) = 0
    }
    runtime.KeepAlive(sv.data)
    sv.data = nil
}

相关文档