ccb158c99b
What: add monkit.Task to a bunch of functions that are missing it Why: this will significantly help our instrumentation, data collection, and tracing about what's going on in the network
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package vouchers
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/golang/protobuf/ptypes"
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
|
|
|
"storj.io/storj/pkg/auth/signing"
|
|
"storj.io/storj/pkg/identity"
|
|
"storj.io/storj/pkg/overlay"
|
|
"storj.io/storj/pkg/pb"
|
|
)
|
|
|
|
// Config contains voucher service configuration parameters
|
|
type Config struct {
|
|
Expiration int `help:"number of days before a voucher expires" default:"30"`
|
|
}
|
|
|
|
// Service for issuing signed vouchers
|
|
type Service struct {
|
|
log *zap.Logger
|
|
satellite signing.Signer
|
|
cache *overlay.Cache
|
|
expiration time.Duration
|
|
}
|
|
|
|
var (
|
|
// Error the default vouchers errs class
|
|
Error = errs.Class("vouchers error")
|
|
|
|
mon = monkit.Package()
|
|
)
|
|
|
|
// NewService creates a new service for issuing signed vouchers
|
|
func NewService(log *zap.Logger, satellite signing.Signer, cache *overlay.Cache, expiration time.Duration) *Service {
|
|
return &Service{
|
|
log: log,
|
|
satellite: satellite,
|
|
cache: cache,
|
|
expiration: expiration,
|
|
}
|
|
}
|
|
|
|
// Request receives a voucher request and returns a voucher and an error
|
|
func (service *Service) Request(ctx context.Context, req *pb.VoucherRequest) (_ *pb.Voucher, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
peer, err := identity.PeerIdentityFromContext(ctx)
|
|
if err != nil {
|
|
return &pb.Voucher{}, Error.Wrap(err)
|
|
}
|
|
|
|
reputable, err := service.cache.IsVetted(ctx, peer.ID)
|
|
if err != nil {
|
|
return &pb.Voucher{}, Error.Wrap(err)
|
|
}
|
|
|
|
service.log.Debug("Node reputation", zap.Bool("reputable", reputable))
|
|
|
|
if !reputable {
|
|
return &pb.Voucher{}, Error.New("Request rejected. Node not reputable")
|
|
}
|
|
|
|
expirationTime := time.Now().UTC().Add(service.expiration)
|
|
expiration, err := ptypes.TimestampProto(expirationTime)
|
|
if err != nil {
|
|
return &pb.Voucher{}, Error.Wrap(err)
|
|
}
|
|
|
|
unsigned := &pb.Voucher{
|
|
SatelliteId: service.satellite.ID(),
|
|
StorageNodeId: peer.ID,
|
|
Expiration: expiration,
|
|
}
|
|
|
|
return signing.SignVoucher(ctx, service.satellite, unsigned)
|
|
}
|