summaryrefslogtreecommitdiff
path: root/playground/main.go
blob: b8574f05173fb57181563d95d23ff688864636aa (plain)
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
package main

import (
    "fmt"
    "os"
)

func runGeneration() {
    _, privateBytes, publicBytes, err := generateKeys()
    if err != nil {
        fmt.Printf("Произошла ошибка при генерации ключей: %s\n", err)
        os.Exit(1)
    }

    err = writeFile(os.Args[2], privateBytes)
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }
    err = writeFile(os.Args[3], publicBytes)
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }
}

func runSign() {
    keyFile := os.Args[2]
    messageFile := os.Args[3]
    resultFile := os.Args[4]

    key, err := loadPublicPEM(keyFile)

    message, err := readFile(messageFile)
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }
    signedMessage, err := signData(message, key)
    if err != nil {
        fmt.Printf("Произошла ошибка при подписи данных: %s\n", err)
        os.Exit(1)
    }
    err = writeFile(resultFile, []byte(signedMessage))
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }
}

func runCheck() {
    keyFile := os.Args[2]
    signedFile := os.Args[3]

    key, err := loadPrivatePEM(keyFile)
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }

    data, err := readFile(signedFile)
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }
    result, err := checkSignature(string(data), key)
    if err != nil {
        fmt.Printf("%s\n", err)
        os.Exit(1)
    }

    if result {
        fmt.Println("Подпись верна")
    } else {
        fmt.Println("Подпись не верна")
    }
}

func main() {
    if os.Args[1] == "generate" {
        runGeneration()
    } else if os.Args[1] == "sign" {
    	runSign()
    } else if os.Args[1] == "check" {
        runCheck()
    }
}