hoteas
7f7b585ffb
- 在应用程序中新增对达梦数据库（DM）的配置和连接支持 - 实现 SetDmDB 函数以配置达梦数据库连接 - 更新数据库操作逻辑，支持达梦特有的 SQL 语法和功能 - 在相关文件中添加达梦数据库的处理逻辑，包括表创建、数据插入和查询 - 更新 go.mod 和 go.sum 文件以引入达梦数据库驱动 - 增强文档，详细说明达梦数据库的配置和使用方法
88 lines
1.5 KiB
Go
88 lines
1.5 KiB
Go
/*
|
|
* Copyright (c) 2000-2018, 达梦数据库有限公司.
|
|
* All rights reserved.
|
|
*/
|
|
|
|
package security
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"errors"
|
|
"io"
|
|
"math/big"
|
|
)
|
|
|
|
type dhGroup struct {
|
|
p *big.Int
|
|
g *big.Int
|
|
}
|
|
|
|
func newDhGroup(prime, generator *big.Int) *dhGroup {
|
|
return &dhGroup{
|
|
p: prime,
|
|
g: generator,
|
|
}
|
|
}
|
|
|
|
func (dg *dhGroup) P() *big.Int {
|
|
p := new(big.Int)
|
|
p.Set(dg.p)
|
|
return p
|
|
}
|
|
|
|
func (dg *dhGroup) G() *big.Int {
|
|
g := new(big.Int)
|
|
g.Set(dg.g)
|
|
return g
|
|
}
|
|
|
|
// 生成本地公私钥
|
|
func (dg *dhGroup) GeneratePrivateKey(randReader io.Reader) (key *DhKey, err error) {
|
|
if randReader == nil {
|
|
randReader = rand.Reader
|
|
}
|
|
// 0 < x < p
|
|
x, err := rand.Int(randReader, dg.p)
|
|
if err != nil {
|
|
return
|
|
}
|
|
zero := big.NewInt(0)
|
|
for x.Cmp(zero) == 0 {
|
|
x, err = rand.Int(randReader, dg.p)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
key = new(DhKey)
|
|
key.x = x
|
|
|
|
// y = g ^ x mod p
|
|
key.y = new(big.Int).Exp(dg.g, x, dg.p)
|
|
key.group = dg
|
|
return
|
|
}
|
|
|
|
func (dg *dhGroup) ComputeKey(pubkey *DhKey, privkey *DhKey) (kye *DhKey, err error) {
|
|
if dg.p == nil {
|
|
err = errors.New("DH: invalid group")
|
|
return
|
|
}
|
|
if pubkey.y == nil {
|
|
err = errors.New("DH: invalid public key")
|
|
return
|
|
}
|
|
if pubkey.y.Sign() <= 0 || pubkey.y.Cmp(dg.p) >= 0 {
|
|
err = errors.New("DH parameter out of bounds")
|
|
return
|
|
}
|
|
if privkey.x == nil {
|
|
err = errors.New("DH: invalid private key")
|
|
return
|
|
}
|
|
k := new(big.Int).Exp(pubkey.y, privkey.x, dg.p)
|
|
key := new(DhKey)
|
|
key.y = k
|
|
key.group = dg
|
|
return
|
|
}
|