Go 语言实现复杂密码生成器算法（附源码） | 极客日志

Go 语言实现复杂密码生成器算法（附源码） | 极客日志

// ==========================================
// 文件名：password_generator.go
// 功能：复杂密码生成器（Password Generator）
// ==========================================
package main

import (
	"crypto/rand"
	"fmt"
	"math/big"
)

// PasswordOptions 密码生成配置
type PasswordOptions struct {
	Length     int // 密码长度
	UseLowercase bool // 是否包含小写字母
	UseUppercase bool // 是否包含大写字母
	UseNumbers   bool // 是否包含数字
	UseSymbols   bool // 是否包含特殊字符
}

// GeneratePassword 根据配置生成复杂密码
func GeneratePassword(opt PasswordOptions) (string, error) {
	if opt.Length <= 0 {
		return "", fmt.Errorf("password length must be greater than 0")
	}

	// 各类字符集
	lowercase := "abcdefghijklmnopqrstuvwxyz"
	uppercase := "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
	numbers := "0123456789"
	symbols := "!@#$%^&*()-_=+[]{}<>?"

	// 构建字符池
	charPool := ""
	if opt.UseLowercase {
		charPool += lowercase
	}
	if opt.UseUppercase {
		charPool += uppercase
	}
	if opt.UseNumbers {
		charPool += numbers
	}
	if opt.UseSymbols {
		charPool += symbols
	}

	if len(charPool) == 0 {
		return "", fmt.Errorf("no character set selected")
	}

	password := make([]byte, opt.Length)

	// 使用 crypto/rand 生成安全随机字符
	for i := 0; i < opt.Length; i++ {
		index, err := rand.Int(rand.Reader, big.NewInt(int64(len(charPool))))
		if err != nil {
			return "", err
		}
		password[i] = charPool[index.Int64()]
	}

	return string(password), nil
}

func main() {
	opt := PasswordOptions{
		Length:       16,
		UseLowercase: true,
		UseUppercase: true,
		UseNumbers:   true,
		UseSymbols:   true,
	}

	password, err := GeneratePassword(opt)
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	fmt.Println("Generated Password:", password)
}