seaweedfs/weed/s3api/auth_credentials.go

195 lines
4.3 KiB
Go
Raw Normal View History

2020-02-09 22:30:02 +00:00
package s3api
import (
"bytes"
"fmt"
"io/ioutil"
"net/http"
"github.com/golang/protobuf/jsonpb"
2020-02-23 05:34:18 +00:00
"github.com/gorilla/mux"
2020-02-09 22:30:02 +00:00
"github.com/chrislusf/seaweedfs/weed/glog"
"github.com/chrislusf/seaweedfs/weed/pb/iam_pb"
)
type Action string
const (
ACTION_READ = "Read"
ACTION_WRITE = "Write"
ACTION_ADMIN = "Admin"
)
type Iam interface {
Check(f http.HandlerFunc, actions ...Action) http.HandlerFunc
}
type IdentityAccessManagement struct {
identities []*Identity
2020-02-10 00:02:05 +00:00
domain string
2020-02-09 22:30:02 +00:00
}
type Identity struct {
Name string
Credentials []*Credential
Actions []Action
}
type Credential struct {
AccessKey string
SecretKey string
}
2020-02-10 00:02:05 +00:00
func NewIdentityAccessManagement(fileName string, domain string) *IdentityAccessManagement {
iam := &IdentityAccessManagement{
domain: domain,
}
2020-02-09 22:30:02 +00:00
if fileName == "" {
return iam
}
2020-02-17 20:31:59 +00:00
if err := iam.loadS3ApiConfiguration(fileName); err != nil {
2020-02-09 22:30:02 +00:00
glog.Fatalf("fail to load config file %s: %v", fileName, err)
}
return iam
}
2020-02-17 20:31:59 +00:00
func (iam *IdentityAccessManagement) loadS3ApiConfiguration(fileName string) error {
2020-02-09 22:30:02 +00:00
2020-02-17 20:31:59 +00:00
s3ApiConfiguration := &iam_pb.S3ApiConfiguration{}
2020-02-09 22:30:02 +00:00
rawData, readErr := ioutil.ReadFile(fileName)
if readErr != nil {
glog.Warningf("fail to read %s : %v", fileName, readErr)
return fmt.Errorf("fail to read %s : %v", fileName, readErr)
}
glog.V(1).Infof("maybeLoadVolumeInfo Unmarshal volume info %v", fileName)
2020-02-17 20:31:59 +00:00
if err := jsonpb.Unmarshal(bytes.NewReader(rawData), s3ApiConfiguration); err != nil {
2020-02-09 22:30:02 +00:00
glog.Warningf("unmarshal error: %v", err)
return fmt.Errorf("unmarshal %s error: %v", fileName, err)
}
2020-02-17 20:31:59 +00:00
for _, ident := range s3ApiConfiguration.Identities {
2020-02-09 22:30:02 +00:00
t := &Identity{
Name: ident.Name,
Credentials: nil,
Actions: nil,
}
for _, action := range ident.Actions {
t.Actions = append(t.Actions, Action(action))
}
for _, cred := range ident.Credentials {
t.Credentials = append(t.Credentials, &Credential{
AccessKey: cred.AccessKey,
SecretKey: cred.SecretKey,
})
}
iam.identities = append(iam.identities, t)
}
return nil
}
2020-07-11 16:11:15 +00:00
func (iam *IdentityAccessManagement) isEnabled() bool {
return len(iam.identities) > 0
}
2020-02-09 22:30:02 +00:00
func (iam *IdentityAccessManagement) lookupByAccessKey(accessKey string) (identity *Identity, cred *Credential, found bool) {
2020-07-11 16:11:15 +00:00
2020-02-09 22:30:02 +00:00
for _, ident := range iam.identities {
for _, cred := range ident.Credentials {
if cred.AccessKey == accessKey {
return ident, cred, true
}
}
}
return nil, nil, false
}
2020-02-23 05:34:18 +00:00
func (iam *IdentityAccessManagement) Auth(f http.HandlerFunc, action Action) http.HandlerFunc {
2020-02-09 22:30:02 +00:00
2020-07-11 16:11:15 +00:00
if iam.isEnabled() {
2020-02-09 22:30:02 +00:00
return f
}
return func(w http.ResponseWriter, r *http.Request) {
2020-02-23 05:34:18 +00:00
errCode := iam.authRequest(r, action)
2020-02-09 22:30:02 +00:00
if errCode == ErrNone {
f(w, r)
return
}
writeErrorResponse(w, errCode, r.URL)
}
}
// check whether the request has valid access keys
2020-02-23 05:34:18 +00:00
func (iam *IdentityAccessManagement) authRequest(r *http.Request, action Action) ErrorCode {
2020-02-09 22:30:02 +00:00
var identity *Identity
var s3Err ErrorCode
switch getRequestAuthType(r) {
2020-02-10 00:02:05 +00:00
case authTypeStreamingSigned:
return ErrNone
case authTypeUnknown:
glog.V(3).Infof("unknown auth type")
2020-02-09 22:30:02 +00:00
return ErrAccessDenied
case authTypePresignedV2, authTypeSignedV2:
2020-02-10 00:02:05 +00:00
glog.V(3).Infof("v2 auth type")
identity, s3Err = iam.isReqAuthenticatedV2(r)
2020-02-09 22:30:02 +00:00
case authTypeSigned, authTypePresigned:
2020-02-10 00:02:05 +00:00
glog.V(3).Infof("v4 auth type")
2020-02-09 22:30:02 +00:00
identity, s3Err = iam.reqSignatureV4Verify(r)
case authTypePostPolicy:
2020-02-22 22:01:04 +00:00
glog.V(3).Infof("post policy auth type")
return ErrNotImplemented
case authTypeJWT:
2020-02-22 22:01:04 +00:00
glog.V(3).Infof("jwt auth type")
return ErrNotImplemented
case authTypeAnonymous:
2020-02-22 22:01:04 +00:00
return ErrAccessDenied
default:
return ErrNotImplemented
2020-02-09 22:30:02 +00:00
}
2020-02-10 00:02:05 +00:00
glog.V(3).Infof("auth error: %v", s3Err)
if s3Err != ErrNone {
return s3Err
}
glog.V(3).Infof("user name: %v actions: %v", identity.Name, identity.Actions)
2020-02-23 05:34:18 +00:00
vars := mux.Vars(r)
bucket := vars["bucket"]
if !identity.canDo(action, bucket) {
2020-02-09 22:30:02 +00:00
return ErrAccessDenied
}
return ErrNone
}
2020-02-23 05:34:18 +00:00
func (identity *Identity) canDo(action Action, bucket string) bool {
2020-02-09 22:30:02 +00:00
for _, a := range identity.Actions {
2020-02-23 05:34:18 +00:00
if a == "Admin" {
return true
}
}
for _, a := range identity.Actions {
if a == action {
return true
}
}
if bucket == "" {
return false
}
limitedByBucket := string(action) + ":" + bucket
for _, a := range identity.Actions {
if string(a) == limitedByBucket {
return true
2020-02-09 22:30:02 +00:00
}
}
return false
}