如何在 Go 命令行程序中使用 flag 控制加密或解密操作-Golang-PHP中文网

如何在 Go 命令行程序中使用 flag 控制加密或解密操作

本文介绍如何使用 Go 标准库的 flag 包实现命令行参数驱动的功能选择（如加密/解密），避免硬编码逻辑，提升 CLI 程序的灵活性与可维护性。

本文介绍如何使用 go 标准库的 `flag` 包实现命令行参数驱动的功能选择（如加密/解密），避免硬编码逻辑，提升 cli 程序的灵活性与可维护性。

在构建 Go 命令行工具时，若需根据用户输入动态决定执行加密还是解密流程（而非同时运行两者），直接依赖 os.Args 不仅脆弱、难以扩展，还缺乏参数校验和帮助提示能力。推荐使用标准库 flag 包——它提供类型安全的参数解析、自动 -h/--help 支持、错误提示及位置参数管理。

以下是一个结构清晰、生产就绪的改造示例（基于原代码优化）：

package main

import (
    "flag"
    "io/ioutil"
    "log"
    "os"

    "crypto/aes"
    "crypto/cipher"
    "crypto/rand"
    "errors"
    "io"
)

// encrypt 加密明文，返回带 IV 的密文（CFB 模式）
func encrypt(key, text []byte) ([]byte, error) {
    block, err := aes.NewCipher(key)
    if err != nil {
        return nil, err
    }

    ciphertext := make([]byte, aes.BlockSize+len(text))
    iv := ciphertext[:aes.BlockSize]
    if _, err := io.ReadFull(rand.Reader, iv); err != nil {
        return nil, err
    }

    cfb := cipher.NewCFBEncrypter(block, iv)
    cfb.XORKeyStream(ciphertext[aes.BlockSize:], text)
    return ciphertext, nil
}

// decrypt 解密密文（首 aes.BlockSize 字节为 IV）
func decrypt(key, ciphertext []byte) ([]byte, error) {
    if len(ciphertext) <p><strong>关键改进说明：</strong>  </p><div class="aritcle_card flexRow artxards">
											<div class="artcardd flexRow">
												<a class="aritcle_card_img" rel="nofollow" href="/xiazai/learn/7564" title="使用Go语言搭建家庭相册系统-相关课件"><img
														src="https://img.php.cn/upload/webcode/000/000/164/636a2b4d84031727.png" alt="使用Go语言搭建家庭相册系统-相关课件" onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
												<div class="aritcle_card_info flexColumn">
													<a rel="nofollow" href="/xiazai/learn/7564" title="使用Go语言搭建家庭相册系统-相关课件" class="overflowclass">使用Go语言搭建家庭相册系统-相关课件</a>
													<p class="overflowclass">使用Go语言搭建家庭相册系统-相关课件</p>
												</div>
												<a rel="nofollow" href="/xiazai/learn/7564" title="使用Go语言搭建家庭相册系统-相关课件" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span>
												</a>
											</div>
										</div>

✅ 标志驱动逻辑：通过 flag.Bool 显式声明 -encrypt / -decrypt 开关，语义清晰；
✅ 参数校验：检查是否至少启用一个模式、是否冲突、是否提供完整文件路径；
✅ 安全性提示：注释强调密钥硬编码的风险（生产中应使用 crypto/rand 生成密钥 + 安全存储）；
✅ 健壮 I/O：使用 0644 替代 0755（文件无需执行权限），并统一错误处理与日志；

✅ 用户体验：支持 yourtool -h 自动输出用法，例如：

$ go run main.go -h
Usage of main:
  -decrypt
      启用解密模式
  -encrypt
      启用加密模式

使用示例：

# 加密 test.txt → encrypted.bin
$ go run main.go -encrypt test.txt encrypted.bin

# 解密 encrypted.bin → decrypted.txt
$ go run main.go -decrypt encrypted.bin decrypted.txt