update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

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
package cert 
 
import ( 
    "context" 
    "crypto/x509" 
    "encoding/json" 
    "os" 
    "strings" 
    "testing" 
    "time" 
 
    "github.com/v2fly/v2ray-core/v5/common" 
    "github.com/v2fly/v2ray-core/v5/common/task" 
) 
 
func TestGenerate(t *testing.T) { 
    err := generate(nil, true, true, "ca") 
    if err != nil { 
        t.Fatal(err) 
    } 
} 
 
func generate(domainNames []string, isCA bool, jsonOutput bool, fileOutput string) error { 
    commonName := "V2Ray Root CA" 
    organization := "V2Ray Inc" 
 
    expire := time.Hour * 3 
 
    var opts []Option 
    if isCA { 
        opts = append(opts, Authority(isCA)) 
        opts = append(opts, KeyUsage(x509.KeyUsageCertSign|x509.KeyUsageKeyEncipherment|x509.KeyUsageDigitalSignature)) 
    } 
 
    opts = append(opts, NotAfter(time.Now().Add(expire))) 
    opts = append(opts, CommonName(commonName)) 
    if len(domainNames) > 0 { 
        opts = append(opts, DNSNames(domainNames...)) 
    } 
    opts = append(opts, Organization(organization)) 
 
    cert, err := Generate(nil, opts...) 
    if err != nil { 
        return newError("failed to generate TLS certificate").Base(err) 
    } 
 
    if jsonOutput { 
        printJSON(cert) 
    } 
 
    if len(fileOutput) > 0 { 
        if err := printFile(cert, fileOutput); err != nil { 
            return err 
        } 
    } 
 
    return nil 
} 
 
type jsonCert struct { 
    Certificate []string `json:"certificate"` 
    Key         []string `json:"key"` 
} 
 
func printJSON(certificate *Certificate) { 
    certPEM, keyPEM := certificate.ToPEM() 
    jCert := &jsonCert{ 
        Certificate: strings.Split(strings.TrimSpace(string(certPEM)), "\n"), 
        Key:         strings.Split(strings.TrimSpace(string(keyPEM)), "\n"), 
    } 
    content, err := json.MarshalIndent(jCert, "", "  ") 
    common.Must(err) 
    os.Stdout.Write(content) 
    os.Stdout.WriteString("\n") 
} 
 
func printFile(certificate *Certificate, name string) error { 
    certPEM, keyPEM := certificate.ToPEM() 
    return task.Run(context.Background(), func() error { 
        return writeFile(certPEM, name+"_cert.pem") 
    }, func() error { 
        return writeFile(keyPEM, name+"_key.pem") 
    }) 
} 
 
func writeFile(content []byte, name string) error { 
    f, err := os.Create(name) 
    if err != nil { 
        return err 
    } 
    defer f.Close() 
 
    return common.Error2(f.Write(content)) 
}