首页 后端开发 Golang 让我们构建一个我们可以实际使用的密码生成器

让我们构建一个我们可以实际使用的密码生成器

Nov 14, 2024 pm 08:13 PM

对于我们的下一个初学者项目,我们将构建一个密码生成器,它不仅可以生成密码,还可以加密并保存密码 - 以便它真正发挥作用。

我们将把代码分成不同的文件,这样我们就不会得到一个大的“main.go”文件。

首先,我们初始化一个 go 项目并创建一个“profile.go”文件,其中包含加密和解密密码的逻辑。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

package main

 

import (

    "crypto/aes"

    "crypto/cipher"

    "crypto/rand"

    "encoding/hex"

    "errors"

    "io"

)

 

// must be 32 characters

var key = "askdjasjdbreonfsdfibsdhfgsdfhboo"

 

var ErrMalformedEncryption = errors.New("malformed encryption")

 

// password in small letters so it is not stored

type profile struct {

    Enc, Platform, password string

}

 

func (p *profile) encrypt() error {

    block, err := aes.NewCipher([]byte(key))

    if err != nil {

        return err

    }

 

    gcm, err := cipher.NewGCM(block)

    if err != nil {

        return err

    }

 

    nonce := make([]byte, gcm.NonceSize())

 

    if _, err := io.ReadFull(rand.Reader, nonce); err != nil {

        return err

    }

 

    enc := gcm.Seal(nonce, nonce, []byte(p.password), nil)

    p.Enc = hex.EncodeToString(enc)

 

    return nil

}

 

func (p *profile) decrypt() error {

    block, err := aes.NewCipher([]byte(key))

    if err != nil {

        return err

    }

 

    gcm, err := cipher.NewGCM(block)

    if err != nil {

        return err

    }

 

    nsize := gcm.NonceSize()

 

    if len(p.Enc) < nsize {

        return ErrMalformedEncryption

    }

 

    enc, err := hex.DecodeString(p.Enc)

    if err != nil {

        return err

    }

 

    password, err := gcm.Open(nil, enc[:nsize], enc[nsize:], nil)

    if err != nil {

        return err

    }

 

    p.password = string(password)

 

    return nil

}

登录后复制

这里我们创建一个包含 3 个字段的配置文件结构 - Enc、平台和密码。 Enc 将保存加密的密码,我们为其生成密码的服务将存储在平台中,而密码将保存实际生成的密码。配置文件结构有两种方法“加密”和“解密”。我们使用 AES - 一种对称密钥加密算法来加密和解密我们的密码。

接下来我们创建一个“store.go”文件,其中包含存储和检索密码的逻辑。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

package main

 

import (

    "encoding/gob"

    "errors"

    "os"

    "sync"

)

 

const filename = "profile.bin"

 

var (

    ErrInvalidArgs = errors.New("invalid args")

    ErrNotFound    = errors.New("not found")

)

 

type store struct {

    sync.RWMutex

    data map[string]*profile

}

 

func newStore() (*store, error) {

    s := &store{

        data: make(map[string]*profile),

    }

 

    if err := s.load(); err != nil {

        return nil, err

    }

 

    return s, nil

}

 

func (s *store) load() error {

    flags := os.O_CREATE | os.O_RDONLY

    f, err := os.OpenFile(filename, flags, 0644)

    if err != nil {

        return err

    }

    defer f.Close()

 

    info, err := f.Stat()

    if err != nil {

        return err

    }

 

    if info.Size() == 0 {

        return nil

    }

 

    return gob.NewDecoder(f).Decode(&s.data)

}

 

func (s *store) save() error {

    f, err := os.OpenFile(filename, os.O_WRONLY, 0644)

    if err != nil {

        return err

    }

    defer f.Close()

 

    return gob.NewEncoder(f).Encode(s.data)

}

 

func (s *store) find(platform string) (string, error) {

    s.RLock()

    defer s.RUnlock()

 

    p, ok := s.data[platform]

    if !ok {

        return "", ErrNotFound

    }

 

    if err := p.decrypt(); err != nil {

        return "", err

    }

 

    return p.password, nil

}

 

func (s *store) add(platform, password string) error {

    if platform == "" {

        return ErrInvalidArgs

    }

 

    p := &profile{

        Platform: platform,

        password: password,

    }

 

    if err := p.encrypt(); err != nil {

        return err

    }

 

    s.Lock()

    defer s.Unlock()

 

    s.data[platform] = p

 

    return s.save()

}

登录后复制

我们选择 gob 文件进行存储,因为它们不完全是人类可读的。如果文件被泄露,您的密码是安全的,因为它们将被加密并且很难读取。 store 结构包含加载、查找和保存到 gob 文件的方法。我们将密码保存在字典中。我们还使用互斥体来确保字典并发安全。需要注意的重要一点是,我们不会存储纯生成的密码 - 我们将存储其加密值。

现在让我们编写几个实际生成密码的函数。创建一个“password.go”文件并输入以下内容

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

package main

 

import (

    "math"

    "math/rand"

    "slices"

    "strings"

)

 

const (

    half      = .5

    onethird  = .3

    onefourth = .25

)

 

var (

    randlowers  = randFromSeed(lowers())

    randuppers  = randFromSeed(uppers())

    randdigits  = randFromSeed(digits())

    randsymbols = randFromSeed(symbols())

)

 

var basicPassword = randlowers

 

func mediumPassword(n int) string {

    frac := math.Round(float64(n) * half)

    pwd := basicPassword(n)

    return pwd[:n-int(frac)] + randuppers(int(frac))

}

 

func hardPassword(n int) string {

    pwd := mediumPassword(n)

    frac := math.Round(float64(n) * onethird)

    return pwd[:n-int(frac)] + randdigits(int(frac))

}

 

func xhardPassword(n int) string {

    pwd := hardPassword(n)

    frac := math.Round(float64(n) * onefourth)

    return pwd[:n-int(frac)] + randsymbols(int(frac))

}

 

func randFromSeed(seed string) func(int) string {

    return func(n int) string {

        var b strings.Builder

        for range n {

            b.WriteByte(seed[rand.Intn(len(seed))])

        }

        return b.String()

    }

}

 

func lowers() string {

    var b strings.Builder

    for i := 'a'; i < 'a'+26; i++ {

        b.WriteRune(i)

    }

    return b.String()

}

 

func uppers() string {

    var b strings.Builder

    for i := 'A'; i < 'A'+26; i++ {

        b.WriteRune(i)

    }

    return b.String()

}

 

func symbols() string {

    var b strings.Builder

    for i := '!'; i < '!'+14; i++ {

        b.WriteRune(i)

    }

    for i := ':'; i < ':'+6; i++ {

        b.WriteRune(i)

    }

    for i := '['; i < '['+5; i++ {

        b.WriteRune(i)

    }

    for i := '{'; i < '{'+3; i++ {

        b.WriteRune(i)

    }

    return b.String()

}

 

func digits() string {

    var b strings.Builder

    for i := '0'; i < '0'+9; i++ {

        b.WriteRune(i)

    }

    return b.String()

}

 

func shuffle[T any](ts []T) []T {

    cloned := slices.Clone(ts)

    rand.Shuffle(len(cloned), func(i, j int) {

        cloned[i], cloned[j] = cloned[j], cloned[i]

    })

    return cloned

}

 

func shuffleStr(s string) string {

    return strings.Join(shuffle(strings.Split(s, "")), "")

}

登录后复制

在这里,我们编写了生成不同难度级别密码的函数。 basicPassword 函数生成随机小字母字符串。 mediumPassword 函数从 basicPassword 函数中获取一小部分字符,并向其中添加随机大写字母。 HardPassword 函数对mediumPassword 执行相同的操作,但添加了数字。 xhardPassword 执行相同操作并添加符号。 shuffle 函数完全符合您对切片的期望,而 shuffleStr 则对字符串进行洗牌。

现在让我们把所有东西放在一起。创建一个“main.go”文件并输入以下内容

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

package main

 

import (

    "errors"

    "flag"

    "fmt"

    "log"

    "regexp"

    "strconv"

    "strings"

)

 

var usage = `

Usage

-----

--get platform=[string] - Gets saved password for a platform

--set platform=[string] len=[int] level=(basic|medium|hard|xhard) - Creates and saves a password

`

 

var ErrUsage = errors.New(usage)

 

var pattern = regexp.MustCompile(`\S+=\S+`)

 

type level int

 

const (

    _ level = iota

    level_basic

    level_medium

    level_hard

    level_xhard

)

 

var level_key = map[string]level{

    "basic":  level_basic,

    "medium": level_medium,

    "hard":   level_hard,

    "xhard":  level_xhard,

}

 

type commands struct {

    get, set bool

}

 

func createCommands() (c commands) {

    flag.BoolVar(&c.get, "get", false, "get password for platform")

    flag.BoolVar(&c.set, "set", false, "set password for platform")

    flag.Parse()

    return

}

 

func (c commands) exec(store *store) (string, error) {

    switch {

    case c.get:

        return c.getPassword(store)

    case c.set:

        return c.setPassword(store)

    default:

        return "", ErrUsage

    }

}

 

func (c commands) getPassword(store *store) (string, error) {

    params, err := c.parse()

    if err != nil {

        return "", err

    }

 

    return store.find(params["platform"])

}

 

func (c commands) setPassword(store *store) (string, error) {

    params, err := c.parse()

    if err != nil {

        return "", err

    }

 

    var password string

 

    n, err := strconv.Atoi(params["len"])

    if err != nil {

        return "", err

    }

 

    if n < 8 {

        return "", fmt.Errorf("password len cannot be less than 8")

    }

 

    switch level_key[params["level"]] {

    case level_basic:

        password = basicPassword(n)

    case level_medium:

        password = mediumPassword(n)

    case level_hard:

        password = hardPassword(n)

    case level_xhard:

        password = xhardPassword(n)

    default:

        return "", ErrUsage

    }

 

    password = shuffleStr(password)

 

    if err := store.add(params["platform"], password); err != nil {

        return "", err

    }

 

    return password, nil

}

 

func (c commands) parse() (map[string]string, error) {

    args := flag.Args()

 

    if len(args) == 0 {

        return nil, ErrUsage

    }

 

    params := make(map[string]string)

 

    for i := range args {

        if !pattern.MatchString(args[i]) {

            return nil, ErrUsage

        }

 

        parts := strings.Split(args[i], "=")

        params[parts[0]] = parts[1]

    }

 

    return params, nil

}

 

func main() {

    store, err := newStore()

    if err != nil {

        log.Fatalf("could not initialize store: %v", err)

    }

 

    c := createCommands()

 

    password, err := c.exec(store)

    if err != nil {

        log.Fatalf("could not execute flag commands: %v", err)

    }

 

    fmt.Printf("password: %s\n", password)

}

登录后复制

我们使用标志来指定我们期望应用程序的行为方式。 “--get”获取密码,“--set”生成并保存密码。要设置密码,用户提供带有标志的参数,以指示应用程序生成和保存的密码类型。要获取密码,用户还提供参数来指定要检索的密码。

您现在可以运行“go build”来构建二进制文件并测试应用程序。

Let

以上是让我们构建一个我们可以实际使用的密码生成器的详细内容。更多信息请关注PHP中文网其他相关文章!

本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热AI工具

Undresser.AI Undress

Undresser.AI Undress

人工智能驱动的应用程序,用于创建逼真的裸体照片

AI Clothes Remover

AI Clothes Remover

用于从照片中去除衣服的在线人工智能工具。

Undress AI Tool

Undress AI Tool

免费脱衣服图片

Clothoff.io

Clothoff.io

AI脱衣机

Video Face Swap

Video Face Swap

使用我们完全免费的人工智能换脸工具轻松在任何视频中换脸!

热门文章

<🎜>:泡泡胶模拟器无穷大 - 如何获取和使用皇家钥匙
4 周前 By 尊渡假赌尊渡假赌尊渡假赌
北端:融合系统,解释
4 周前 By 尊渡假赌尊渡假赌尊渡假赌
Mandragora:巫婆树的耳语 - 如何解锁抓钩
3 周前 By 尊渡假赌尊渡假赌尊渡假赌

热工具

记事本++7.3.1

记事本++7.3.1

好用且免费的代码编辑器

SublimeText3汉化版

SublimeText3汉化版

中文版,非常好用

禅工作室 13.0.1

禅工作室 13.0.1

功能强大的PHP集成开发环境

Dreamweaver CS6

Dreamweaver CS6

视觉化网页开发工具

SublimeText3 Mac版

SublimeText3 Mac版

神级代码编辑软件(SublimeText3)

热门话题

Java教程
1673
14
CakePHP 教程
1429
52
Laravel 教程
1333
25
PHP教程
1278
29
C# 教程
1257
24
Golang vs. Python:性能和可伸缩性 Golang vs. Python:性能和可伸缩性 Apr 19, 2025 am 12:18 AM

Golang在性能和可扩展性方面优于Python。1)Golang的编译型特性和高效并发模型使其在高并发场景下表现出色。2)Python作为解释型语言,执行速度较慢,但通过工具如Cython可优化性能。

Golang和C:并发与原始速度 Golang和C:并发与原始速度 Apr 21, 2025 am 12:16 AM

Golang在并发性上优于C ,而C 在原始速度上优于Golang。1)Golang通过goroutine和channel实现高效并发,适合处理大量并发任务。2)C 通过编译器优化和标准库,提供接近硬件的高性能,适合需要极致优化的应用。

开始GO:初学者指南 开始GO:初学者指南 Apr 26, 2025 am 12:21 AM

goisidealforbeginnersandsubableforforcloudnetworkservicesduetoitssimplicity,效率和concurrencyFeatures.1)installgromtheofficialwebsitealwebsiteandverifywith'.2)

Golang vs.C:性能和速度比较 Golang vs.C:性能和速度比较 Apr 21, 2025 am 12:13 AM

Golang适合快速开发和并发场景,C 适用于需要极致性能和低级控制的场景。1)Golang通过垃圾回收和并发机制提升性能,适合高并发Web服务开发。2)C 通过手动内存管理和编译器优化达到极致性能,适用于嵌入式系统开发。

Golang vs. Python:主要差异和相似之处 Golang vs. Python:主要差异和相似之处 Apr 17, 2025 am 12:15 AM

Golang和Python各有优势:Golang适合高性能和并发编程,Python适用于数据科学和Web开发。 Golang以其并发模型和高效性能着称,Python则以简洁语法和丰富库生态系统着称。

Golang和C:性能的权衡 Golang和C:性能的权衡 Apr 17, 2025 am 12:18 AM

Golang和C 在性能上的差异主要体现在内存管理、编译优化和运行时效率等方面。1)Golang的垃圾回收机制方便但可能影响性能,2)C 的手动内存管理和编译器优化在递归计算中表现更为高效。

表演竞赛:Golang vs.C 表演竞赛:Golang vs.C Apr 16, 2025 am 12:07 AM

Golang和C 在性能竞赛中的表现各有优势:1)Golang适合高并发和快速开发,2)C 提供更高性能和细粒度控制。选择应基于项目需求和团队技术栈。

Golang vs. Python:利弊 Golang vs. Python:利弊 Apr 21, 2025 am 12:17 AM

Golangisidealforbuildingscalablesystemsduetoitsefficiencyandconcurrency,whilePythonexcelsinquickscriptinganddataanalysisduetoitssimplicityandvastecosystem.Golang'sdesignencouragesclean,readablecodeanditsgoroutinesenableefficientconcurrentoperations,t

See all articles