114 lines
2 KiB
Go
114 lines
2 KiB
Go
package crypto
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"lux/proto"
|
|
"os"
|
|
)
|
|
|
|
type LuxKey struct {
|
|
Type proto.LuxType // 2
|
|
Id proto.LuxID // 16
|
|
Key []byte // 32
|
|
}
|
|
|
|
const LUX_KEY_AES_SIZE = 32
|
|
const LUX_PROTO_KEY_SIZE = proto.LUX_PROTO_TYPE_SIZE + proto.LUX_PROTO_ID_SIZE + LUX_KEY_AES_SIZE
|
|
|
|
func NewLuxKey(keyType proto.LuxType) (LuxKey, error) {
|
|
key := LuxKey{
|
|
Type: keyType,
|
|
Id: proto.NewLuxID(),
|
|
Key: make([]byte, LUX_KEY_AES_SIZE),
|
|
}
|
|
|
|
if _, err := rand.Read(key.Key); err != nil {
|
|
return key, err
|
|
} else {
|
|
return key, nil
|
|
}
|
|
}
|
|
|
|
func (key *LuxKey) Read(rd *proto.LuxBuffer) error {
|
|
if err := key.Type.Read(rd); err != nil {
|
|
return err
|
|
}
|
|
if err := key.Id.Read(rd); err != nil {
|
|
return err
|
|
}
|
|
|
|
_key, err := rd.ReadNext(32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
key.Key = _key
|
|
return nil
|
|
}
|
|
|
|
func (key *LuxKey) Write(wd *proto.LuxBuffer) {
|
|
key.Type.Write(wd)
|
|
key.Id.Write(wd)
|
|
wd.WriteBytes(key.Key)
|
|
}
|
|
|
|
const LUX_KEYSTORE_FILEMODE = os.FileMode(int(0600))
|
|
|
|
type LuxKeyStore struct {
|
|
filePath string
|
|
keys map[proto.LuxID]LuxKey
|
|
}
|
|
|
|
func NewLuxKeyStore(filePath string) LuxKeyStore {
|
|
return LuxKeyStore{
|
|
filePath: filePath,
|
|
keys: make(map[proto.LuxID]LuxKey),
|
|
}
|
|
}
|
|
|
|
func (ks *LuxKeyStore) Load() error {
|
|
bytes, err := os.ReadFile(ks.filePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rd := proto.FromSlice(bytes)
|
|
keyNum := rd.Remaining() / LUX_PROTO_KEY_SIZE
|
|
for i := 0; i < keyNum; i++ {
|
|
var key LuxKey
|
|
if err := key.Read(&rd); err != nil {
|
|
return err
|
|
}
|
|
|
|
// add new key
|
|
ks.keys[key.Id] = key
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (ks *LuxKeyStore) Save() error {
|
|
wd := proto.AllocLuxBuffer(len(ks.keys))
|
|
for _, key := range ks.keys {
|
|
key.Write(&wd)
|
|
}
|
|
|
|
return os.WriteFile(ks.filePath, wd.AllBytes(), LUX_KEYSTORE_FILEMODE)
|
|
}
|
|
|
|
func (ks *LuxKeyStore) Get(id proto.LuxID) *LuxKey {
|
|
if key, ok := ks.keys[id]; ok {
|
|
return &key
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (ks *LuxKeyStore) Put(key LuxKey) error {
|
|
ks.keys[key.Id] = key
|
|
if err := ks.Save(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|