Node.js废弃加密方法如何用Golang安全地重构？-Golang-PHP中文网

Node.js废弃加密方法如何用Golang安全地重构？

花韻仙語

发布： 2025-03-09 10:44:18

原创

387人浏览过

Golang安全重构Node.js废弃加密方法

本文介绍如何使用golang安全地重构一个使用了node.js废弃加密方法crypto.createcipher('aes256', key)的项目。该node.js代码存在安全漏洞，因为它缺少盐值和明确的加密模式。我们将分析原代码并提供安全的golang实现。

Node.js废弃加密方法如何用Golang安全地重构？

原Node.js代码:

加密函数:

createtoken(src, timestamp, key) {
    const msg = src + '|' + timestamp;
    const cipher = crypto.createCipher('aes256', key);
    let enc = cipher.update(msg, 'utf8', 'hex');
    enc += cipher.final('hex');
    return enc;
}

登录后复制

解密函数:

立即学习“go语言免费学习笔记（深入）”；

parsetoken(token, key) {
    const decipher = crypto.createDecipher('aes256', key);
    let dec;
    try {
        dec = decipher.update(token, 'hex', 'utf8');
        dec += decipher.final('utf8');
    } catch (err) {
        return null;
    }
    const ts = dec.split('|');
    if (ts.length !== 2) {
        return null;
    }
    return {src: ts[0], timestamp: Number(ts[1])};
}

登录后复制

这段代码使用crypto.createCipher('aes256', key)进行AES-256加密，但未指定加密模式和填充方式，存在安全风险。

安全的Golang重构:

由于原代码未指定加密模式，我们假设其使用了CBC模式（一种常见的块密码模式）。为了提高安全性，我们将添加随机生成的初始化向量（IV）和PKCS7填充。

package main

import (
    "crypto/aes"
    "crypto/cipher"
    "crypto/rand"
    "encoding/hex"
    "fmt"
    "io"
)

// PKCS7Padding adds PKCS7 padding to a byte array.
func PKCS7Padding(ciphertext []byte, blockSize int) []byte {
    padding := blockSize - len(ciphertext)%blockSize
    padtext := bytes.Repeat([]byte{byte(padding)}, padding)
    return append(ciphertext, padtext...)
}

// PKCS7UnPadding removes PKCS7 padding from a byte array.
func PKCS7UnPadding(origData []byte) []byte {
    length := len(origData)
    unpadding := int(origData[length-1])
    return origData[:(length - unpadding)]
}


func encrypt(src, timestamp string, key []byte) (string, error) {
    msg := []byte(src + "|" + timestamp)
    block, err := aes.NewCipher(key)
    if err != nil {
        return "", err
    }
    blockSize := block.BlockSize()
    // Generate a random IV
    iv := make([]byte, blockSize)
    if _, err := io.ReadFull(rand.Reader, iv); err != nil {
        return "", err
    }

    paddedMsg := PKCS7Padding(msg, blockSize)
    blockMode := cipher.NewCBCEncrypter(block, iv)
    crypted := make([]byte, len(paddedMsg))
    blockMode.CryptBlocks(crypted, paddedMsg)

    return hex.EncodeToString(append(iv, crypted...)), nil
}

func decrypt(token string, key []byte) (string, error) {
    tokenBytes, err := hex.DecodeString(token)
    if err != nil {
        return "", err
    }
    iv := tokenBytes[:aes.BlockSize]
    crypted := tokenBytes[aes.BlockSize:]

    block, err := aes.NewCipher(key)
    if err != nil {
        return "", err
    }
    blockMode := cipher.NewCBCDecrypter(block, iv)
    origData := make([]byte, len(crypted))
    blockMode.CryptBlocks(origData, crypted)
    unpaddedData := PKCS7UnPadding(origData)

    return string(unpaddedData), nil
}

func main() {
    key := []byte("your_32_byte_secret_key") // Replace with a 32-byte key
    src := "your_source_data"
    timestamp := "1678886400" // Example timestamp

    encrypted, err := encrypt(src, timestamp, key)
    if err != nil {
        fmt.Println("Encryption error:", err)
        return
    }
    fmt.Println("Encrypted:", encrypted)

    decrypted, err := decrypt(encrypted, key)
    if err != nil {
        fmt.Println("Decryption error:", err)
        return
    }
    fmt.Println("Decrypted:", decrypted)
}

登录后复制

关键改进: