forked from TrueCloudLab/frostfs-api-go
83 lines
1.5 KiB
Go
83 lines
1.5 KiB
Go
package session
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"crypto/elliptic"
|
|
"crypto/rand"
|
|
"sync"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/refs"
|
|
crypto "github.com/nspcc-dev/neofs-crypto"
|
|
)
|
|
|
|
type simpleStore struct {
|
|
*sync.RWMutex
|
|
|
|
tokens map[TokenID]*PToken
|
|
}
|
|
|
|
// TODO get curve from neofs-crypto
|
|
func defaultCurve() elliptic.Curve {
|
|
return elliptic.P256()
|
|
}
|
|
|
|
// NewSimpleStore creates simple token storage
|
|
func NewSimpleStore() TokenStore {
|
|
return &simpleStore{
|
|
RWMutex: new(sync.RWMutex),
|
|
tokens: make(map[TokenID]*PToken),
|
|
}
|
|
}
|
|
|
|
// New returns new token with specified parameters.
|
|
func (s *simpleStore) New(p TokenParams) *PToken {
|
|
tid, err := refs.NewUUID()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
key, err := ecdsa.GenerateKey(defaultCurve(), rand.Reader)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
if p.FirstEpoch > p.LastEpoch || p.OwnerID.Empty() {
|
|
return nil
|
|
}
|
|
|
|
token := new(Token)
|
|
token.SetID(tid)
|
|
token.SetOwnerID(p.OwnerID)
|
|
token.SetVerb(p.Verb)
|
|
token.SetAddress(p.Address)
|
|
token.SetCreationEpoch(p.FirstEpoch)
|
|
token.SetExpirationEpoch(p.LastEpoch)
|
|
token.SetSessionKey(crypto.MarshalPublicKey(&key.PublicKey))
|
|
|
|
t := &PToken{
|
|
mtx: new(sync.Mutex),
|
|
Token: *token,
|
|
PrivateKey: key,
|
|
}
|
|
|
|
s.Lock()
|
|
s.tokens[tid] = t
|
|
s.Unlock()
|
|
|
|
return t
|
|
}
|
|
|
|
// Fetch tries to fetch a token with specified id.
|
|
func (s *simpleStore) Fetch(id TokenID) *PToken {
|
|
s.RLock()
|
|
defer s.RUnlock()
|
|
|
|
return s.tokens[id]
|
|
}
|
|
|
|
// Remove removes token with id from store.
|
|
func (s *simpleStore) Remove(id TokenID) {
|
|
s.Lock()
|
|
delete(s.tokens, id)
|
|
s.Unlock()
|
|
}
|