LHHDZ
2 years ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 1233 additions and 0 deletions
-
12weed/s3api/s3_constants/acp_grantee_group.go
-
8weed/s3api/s3_constants/header.go
-
505weed/s3api/s3acl/acl_helper.go
-
708weed/s3api/s3acl/acl_helper_test.go
@ -0,0 +1,505 @@ |
|||||
|
package s3acl |
||||
|
|
||||
|
import ( |
||||
|
"encoding/json" |
||||
|
"encoding/xml" |
||||
|
"github.com/aws/aws-sdk-go/private/protocol/xml/xmlutil" |
||||
|
"github.com/aws/aws-sdk-go/service/s3" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/glog" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3account" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3err" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/util" |
||||
|
"net/http" |
||||
|
"strings" |
||||
|
) |
||||
|
|
||||
|
// GetAccountId get AccountId from request headers, AccountAnonymousId will be return if not presen
|
||||
|
func GetAccountId(r *http.Request) string { |
||||
|
id := r.Header.Get(s3_constants.AmzAccountId) |
||||
|
if len(id) == 0 { |
||||
|
return s3account.AccountAnonymous.Id |
||||
|
} else { |
||||
|
return id |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// ExtractAcl extracts the acl from the request body, or from the header if request body is empty
|
||||
|
func ExtractAcl(r *http.Request, accountManager *s3account.AccountManager, ownership, bucketOwnerId, ownerId, accountId string) (grants []*s3.Grant, errCode s3err.ErrorCode) { |
||||
|
if r.Body != nil && r.Body != http.NoBody { |
||||
|
defer util.CloseRequest(r) |
||||
|
|
||||
|
var acp s3.AccessControlPolicy |
||||
|
err := xmlutil.UnmarshalXML(&acp, xml.NewDecoder(r.Body), "") |
||||
|
if err != nil || acp.Owner == nil || acp.Owner.ID == nil { |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
|
||||
|
//owner should present && owner is immutable
|
||||
|
if *acp.Owner.ID != ownerId { |
||||
|
glog.V(3).Infof("set acl denied! owner account is not consistent, request account id: %s, expect account id: %s", accountId, ownerId) |
||||
|
return nil, s3err.ErrAccessDenied |
||||
|
} |
||||
|
|
||||
|
return ValidateAndTransferGrants(accountManager, acp.Grants) |
||||
|
} else { |
||||
|
_, grants, errCode = ParseAndValidateAclHeadersOrElseDefault(r, accountManager, ownership, bucketOwnerId, accountId, true) |
||||
|
return grants, errCode |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// ParseAndValidateAclHeadersOrElseDefault will callParseAndValidateAclHeaders to get Grants, if empty, it will return Grant that grant `accountId` with `FullControl` permission
|
||||
|
func ParseAndValidateAclHeadersOrElseDefault(r *http.Request, accountManager *s3account.AccountManager, ownership, bucketOwnerId, accountId string, putAcl bool) (ownerId string, grants []*s3.Grant, errCode s3err.ErrorCode) { |
||||
|
ownerId, grants, errCode = ParseAndValidateAclHeaders(r, accountManager, ownership, bucketOwnerId, accountId, putAcl) |
||||
|
if errCode != s3err.ErrNone { |
||||
|
return |
||||
|
} |
||||
|
if len(grants) == 0 { |
||||
|
//if no acl(both customAcl and cannedAcl) specified, grant accountId(object writer) with full control permission
|
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}) |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// ParseAndValidateAclHeaders parse and validate acl from header
|
||||
|
func ParseAndValidateAclHeaders(r *http.Request, accountManager *s3account.AccountManager, ownership, bucketOwnerId, accountId string, putAcl bool) (ownerId string, grants []*s3.Grant, errCode s3err.ErrorCode) { |
||||
|
ownerId, grants, errCode = ParseAclHeaders(r, ownership, bucketOwnerId, accountId, putAcl) |
||||
|
if errCode != s3err.ErrNone { |
||||
|
return |
||||
|
} |
||||
|
if len(grants) > 0 { |
||||
|
grants, errCode = ValidateAndTransferGrants(accountManager, grants) |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// ParseAclHeaders parse acl headers
|
||||
|
// When `putAcl` is true, only `CannedAcl` is parsed, such as `PutBucketAcl` or `PutObjectAcl`
|
||||
|
// is requested, `CustomAcl` is parsed from the request body not from headers, and only if the
|
||||
|
// request body is empty, `CannedAcl` is parsed from the header, and will not parse `CustomAcl` from the header
|
||||
|
//
|
||||
|
// Since `CustomAcl` has higher priority, it will be parsed first; if `CustomAcl` does not exist, `CannedAcl` will be parsed
|
||||
|
func ParseAclHeaders(r *http.Request, ownership, bucketOwnerId, accountId string, putAcl bool) (ownerId string, grants []*s3.Grant, errCode s3err.ErrorCode) { |
||||
|
if !putAcl { |
||||
|
errCode = ParseCustomAclHeaders(r, &grants) |
||||
|
if errCode != s3err.ErrNone { |
||||
|
return "", nil, errCode |
||||
|
} |
||||
|
} |
||||
|
if len(grants) > 0 { |
||||
|
return accountId, grants, s3err.ErrNone |
||||
|
} |
||||
|
|
||||
|
cannedAcl := r.Header.Get(s3_constants.AmzCannedAcl) |
||||
|
if len(cannedAcl) == 0 { |
||||
|
return accountId, grants, s3err.ErrNone |
||||
|
} |
||||
|
|
||||
|
//if canned acl specified, parse cannedAcl (lower priority to custom acl)
|
||||
|
ownerId, grants, errCode = ParseCannedAclHeader(ownership, bucketOwnerId, accountId, cannedAcl, putAcl) |
||||
|
if errCode != s3err.ErrNone { |
||||
|
return "", nil, errCode |
||||
|
} |
||||
|
return ownerId, grants, errCode |
||||
|
} |
||||
|
|
||||
|
func ParseCustomAclHeaders(r *http.Request, grants *[]*s3.Grant) s3err.ErrorCode { |
||||
|
customAclHeaders := []string{s3_constants.AmzAclFullControl, s3_constants.AmzAclRead, s3_constants.AmzAclReadAcp, s3_constants.AmzAclWrite, s3_constants.AmzAclWriteAcp} |
||||
|
var errCode s3err.ErrorCode |
||||
|
for _, customAclHeader := range customAclHeaders { |
||||
|
headerValue := r.Header.Get(customAclHeader) |
||||
|
switch customAclHeader { |
||||
|
case s3_constants.AmzAclRead: |
||||
|
errCode = ParseCustomAclHeader(headerValue, s3_constants.PermissionRead, grants) |
||||
|
case s3_constants.AmzAclWrite: |
||||
|
errCode = ParseCustomAclHeader(headerValue, s3_constants.PermissionWrite, grants) |
||||
|
case s3_constants.AmzAclReadAcp: |
||||
|
errCode = ParseCustomAclHeader(headerValue, s3_constants.PermissionReadAcp, grants) |
||||
|
case s3_constants.AmzAclWriteAcp: |
||||
|
errCode = ParseCustomAclHeader(headerValue, s3_constants.PermissionWriteAcp, grants) |
||||
|
case s3_constants.AmzAclFullControl: |
||||
|
errCode = ParseCustomAclHeader(headerValue, s3_constants.PermissionFullControl, grants) |
||||
|
} |
||||
|
if errCode != s3err.ErrNone { |
||||
|
return errCode |
||||
|
} |
||||
|
} |
||||
|
return s3err.ErrNone |
||||
|
} |
||||
|
|
||||
|
func ParseCustomAclHeader(headerValue, permission string, grants *[]*s3.Grant) s3err.ErrorCode { |
||||
|
if len(headerValue) > 0 { |
||||
|
split := strings.Split(headerValue, ", ") |
||||
|
for _, grantStr := range split { |
||||
|
kv := strings.Split(grantStr, "=") |
||||
|
if len(kv) != 2 { |
||||
|
return s3err.ErrInvalidRequest |
||||
|
} |
||||
|
|
||||
|
switch kv[0] { |
||||
|
case "id": |
||||
|
var accountId string |
||||
|
_ = json.Unmarshal([]byte(kv[1]), &accountId) |
||||
|
*grants = append(*grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &permission, |
||||
|
}) |
||||
|
case "emailAddress": |
||||
|
var emailAddress string |
||||
|
_ = json.Unmarshal([]byte(kv[1]), &emailAddress) |
||||
|
*grants = append(*grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeAmazonCustomerByEmail, |
||||
|
EmailAddress: &emailAddress, |
||||
|
}, |
||||
|
Permission: &permission, |
||||
|
}) |
||||
|
case "uri": |
||||
|
var groupName string |
||||
|
_ = json.Unmarshal([]byte(kv[1]), &groupName) |
||||
|
*grants = append(*grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &groupName, |
||||
|
}, |
||||
|
Permission: &permission, |
||||
|
}) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return s3err.ErrNone |
||||
|
|
||||
|
} |
||||
|
|
||||
|
func ParseCannedAclHeader(bucketOwnership, bucketOwnerId, accountId, cannedAcl string, putAcl bool) (ownerId string, grants []*s3.Grant, err s3err.ErrorCode) { |
||||
|
err = s3err.ErrNone |
||||
|
ownerId = accountId |
||||
|
|
||||
|
//objectWrite automatically has full control on current object
|
||||
|
objectWriterFullControl := &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
ID: &accountId, |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
} |
||||
|
|
||||
|
switch cannedAcl { |
||||
|
case s3_constants.CannedAclPrivate: |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
case s3_constants.CannedAclPublicRead: |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
grants = append(grants, s3_constants.PublicRead...) |
||||
|
case s3_constants.CannedAclPublicReadWrite: |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
grants = append(grants, s3_constants.PublicReadWrite...) |
||||
|
case s3_constants.CannedAclAuthenticatedRead: |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
grants = append(grants, s3_constants.AuthenticatedRead...) |
||||
|
case s3_constants.CannedAclLogDeliveryWrite: |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
grants = append(grants, s3_constants.LogDeliveryWrite...) |
||||
|
case s3_constants.CannedAclBucketOwnerRead: |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
if bucketOwnerId != "" && bucketOwnerId != accountId { |
||||
|
grants = append(grants, |
||||
|
&s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &bucketOwnerId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
}) |
||||
|
} |
||||
|
case s3_constants.CannedAclBucketOwnerFullControl: |
||||
|
if bucketOwnerId != "" { |
||||
|
// if set ownership to 'BucketOwnerPreferred' when upload object, the bucket owner will be the object owner
|
||||
|
if !putAcl && bucketOwnership == s3_constants.OwnershipBucketOwnerPreferred { |
||||
|
ownerId = bucketOwnerId |
||||
|
grants = append(grants, |
||||
|
&s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &bucketOwnerId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}) |
||||
|
} else { |
||||
|
grants = append(grants, objectWriterFullControl) |
||||
|
if accountId != bucketOwnerId { |
||||
|
grants = append(grants, |
||||
|
&s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &bucketOwnerId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
case s3_constants.CannedAclAwsExecRead: |
||||
|
err = s3err.ErrNotImplemented |
||||
|
default: |
||||
|
err = s3err.ErrInvalidRequest |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// ValidateAndTransferGrants validate grant & transfer Email-Grant to Id-Grant
|
||||
|
func ValidateAndTransferGrants(accountManager *s3account.AccountManager, grants []*s3.Grant) ([]*s3.Grant, s3err.ErrorCode) { |
||||
|
var result []*s3.Grant |
||||
|
for _, grant := range grants { |
||||
|
grantee := grant.Grantee |
||||
|
if grantee == nil || grantee.Type == nil { |
||||
|
glog.Warning("invalid grantee! grantee or granteeType is nil") |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
|
||||
|
switch *grantee.Type { |
||||
|
case s3_constants.GrantTypeGroup: |
||||
|
if grantee.URI == nil { |
||||
|
glog.Warning("invalid group grantee! group URI is nil") |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
ok := s3_constants.ValidateGroup(*grantee.URI) |
||||
|
if !ok { |
||||
|
glog.Warningf("invalid group grantee! group name[%s] is not valid", *grantee.URI) |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
result = append(result, grant) |
||||
|
case s3_constants.GrantTypeCanonicalUser: |
||||
|
if grantee.ID == nil { |
||||
|
glog.Warning("invalid canonical grantee! account id is nil") |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
_, ok := accountManager.IdNameMapping[*grantee.ID] |
||||
|
if !ok { |
||||
|
glog.Warningf("invalid canonical grantee! account id[%s] is not exists", *grantee.ID) |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
result = append(result, grant) |
||||
|
case s3_constants.GrantTypeAmazonCustomerByEmail: |
||||
|
if grantee.EmailAddress == nil { |
||||
|
glog.Warning("invalid email grantee! email address is nil") |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
accountId, ok := accountManager.EmailIdMapping[*grantee.EmailAddress] |
||||
|
if !ok { |
||||
|
glog.Warningf("invalid email grantee! email address[%s] is not exists", *grantee.EmailAddress) |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
result = append(result, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: grant.Permission, |
||||
|
}) |
||||
|
default: |
||||
|
return nil, s3err.ErrInvalidRequest |
||||
|
} |
||||
|
} |
||||
|
return result, s3err.ErrNone |
||||
|
} |
||||
|
|
||||
|
// DetermineReqGrants generates the grant set (Grants) according to accountId and reqPermission.
|
||||
|
func DetermineReqGrants(accountId, aclAction string) (grants []*s3.Grant) { |
||||
|
// group grantee (AllUsers)
|
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &aclAction, |
||||
|
}) |
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}) |
||||
|
|
||||
|
// canonical grantee (accountId)
|
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &aclAction, |
||||
|
}) |
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}) |
||||
|
|
||||
|
// group grantee (AuthenticateUsers)
|
||||
|
if accountId != s3account.AccountAnonymous.Id { |
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAuthenticatedUsers, |
||||
|
}, |
||||
|
Permission: &aclAction, |
||||
|
}) |
||||
|
grants = append(grants, &s3.Grant{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAuthenticatedUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}) |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
func SetAcpOwnerHeader(r *http.Request, acpOwnerId string) { |
||||
|
r.Header.Set(s3_constants.ExtAmzOwnerKey, acpOwnerId) |
||||
|
} |
||||
|
|
||||
|
func GetAcpOwner(entryExtended map[string][]byte, defaultOwner string) string { |
||||
|
ownerIdBytes, ok := entryExtended[s3_constants.ExtAmzOwnerKey] |
||||
|
if ok && len(ownerIdBytes) > 0 { |
||||
|
return string(ownerIdBytes) |
||||
|
} |
||||
|
return defaultOwner |
||||
|
} |
||||
|
|
||||
|
func SetAcpGrantsHeader(r *http.Request, acpGrants []*s3.Grant) { |
||||
|
if len(acpGrants) > 0 { |
||||
|
a, err := json.Marshal(acpGrants) |
||||
|
if err == nil { |
||||
|
r.Header.Set(s3_constants.ExtAmzAclKey, string(a)) |
||||
|
} else { |
||||
|
glog.Warning("Marshal acp grants err", err) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// GetAcpGrants return grants parsed from entry
|
||||
|
func GetAcpGrants(entryExtended map[string][]byte) []*s3.Grant { |
||||
|
acpBytes, ok := entryExtended[s3_constants.ExtAmzAclKey] |
||||
|
if ok && len(acpBytes) > 0 { |
||||
|
var grants []*s3.Grant |
||||
|
err := json.Unmarshal(acpBytes, &grants) |
||||
|
if err == nil { |
||||
|
return grants |
||||
|
} |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
// AssembleEntryWithAcp fill entry with owner and grants
|
||||
|
func AssembleEntryWithAcp(objectEntry *filer_pb.Entry, objectOwner string, grants []*s3.Grant) s3err.ErrorCode { |
||||
|
if objectEntry.Extended == nil { |
||||
|
objectEntry.Extended = make(map[string][]byte) |
||||
|
} |
||||
|
|
||||
|
if len(objectOwner) > 0 { |
||||
|
objectEntry.Extended[s3_constants.ExtAmzOwnerKey] = []byte(objectOwner) |
||||
|
} |
||||
|
|
||||
|
if len(grants) > 0 { |
||||
|
grantsBytes, err := json.Marshal(grants) |
||||
|
if err != nil { |
||||
|
glog.Warning("assemble acp to entry:", err) |
||||
|
return s3err.ErrInvalidRequest |
||||
|
} |
||||
|
objectEntry.Extended[s3_constants.ExtAmzAclKey] = grantsBytes |
||||
|
} |
||||
|
|
||||
|
return s3err.ErrNone |
||||
|
} |
||||
|
|
||||
|
// GrantEquals Compare whether two Grants are equal in meaning, not completely
|
||||
|
// equal (compare Grantee.Type and the corresponding Value for equality, other
|
||||
|
// fields of Grantee are ignored)
|
||||
|
func GrantEquals(a, b *s3.Grant) bool { |
||||
|
// grant
|
||||
|
if a == b { |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
if a == nil || b == nil { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// grant.Permission
|
||||
|
if a.Permission != b.Permission { |
||||
|
if a.Permission == nil || b.Permission == nil { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if *a.Permission != *b.Permission { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// grant.Grantee
|
||||
|
ag := a.Grantee |
||||
|
bg := b.Grantee |
||||
|
if ag != bg { |
||||
|
if ag == nil || bg == nil { |
||||
|
return false |
||||
|
} |
||||
|
// grantee.Type
|
||||
|
if ag.Type != bg.Type { |
||||
|
if ag.Type == nil || bg.Type == nil { |
||||
|
return false |
||||
|
} |
||||
|
if *ag.Type != *bg.Type { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
// value corresponding to granteeType
|
||||
|
if ag.Type != nil { |
||||
|
switch *ag.Type { |
||||
|
case s3_constants.GrantTypeGroup: |
||||
|
if ag.URI != bg.URI { |
||||
|
if ag.URI == nil || bg.URI == nil { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if *ag.URI != *bg.URI { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
case s3_constants.GrantTypeCanonicalUser: |
||||
|
if ag.ID != bg.ID { |
||||
|
if ag.ID == nil || bg.ID == nil { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if *ag.ID != *bg.ID { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
case s3_constants.GrantTypeAmazonCustomerByEmail: |
||||
|
if ag.EmailAddress != bg.EmailAddress { |
||||
|
if ag.EmailAddress == nil || bg.EmailAddress == nil { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if *ag.EmailAddress != *bg.EmailAddress { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return true |
||||
|
} |
@ -0,0 +1,708 @@ |
|||||
|
package s3acl |
||||
|
|
||||
|
import ( |
||||
|
"bytes" |
||||
|
"encoding/json" |
||||
|
"github.com/aws/aws-sdk-go/service/s3" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3account" |
||||
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3err" |
||||
|
"io" |
||||
|
"net/http" |
||||
|
"testing" |
||||
|
) |
||||
|
|
||||
|
var ( |
||||
|
accountManager = &s3account.AccountManager{ |
||||
|
IdNameMapping: map[string]string{ |
||||
|
s3account.AccountAdmin.Id: s3account.AccountAdmin.Name, |
||||
|
s3account.AccountAnonymous.Id: s3account.AccountAnonymous.Name, |
||||
|
"accountA": "accountA", |
||||
|
"accountB": "accountB", |
||||
|
}, |
||||
|
EmailIdMapping: map[string]string{ |
||||
|
s3account.AccountAdmin.EmailAddress: s3account.AccountAdmin.Id, |
||||
|
s3account.AccountAnonymous.EmailAddress: s3account.AccountAnonymous.Id, |
||||
|
"accountA@example.com": "accountA", |
||||
|
"accountBexample.com": "accountB", |
||||
|
}, |
||||
|
} |
||||
|
) |
||||
|
|
||||
|
func TestGetAccountId(t *testing.T) { |
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
//case1
|
||||
|
//accountId: "admin"
|
||||
|
req.Header.Set(s3_constants.AmzAccountId, s3account.AccountAdmin.Id) |
||||
|
if GetAccountId(req) != s3account.AccountAdmin.Id { |
||||
|
t.Fatal("expect accountId: admin") |
||||
|
} |
||||
|
|
||||
|
//case2
|
||||
|
//accountId: "anoymous"
|
||||
|
req.Header.Set(s3_constants.AmzAccountId, s3account.AccountAnonymous.Id) |
||||
|
if GetAccountId(req) != s3account.AccountAnonymous.Id { |
||||
|
t.Fatal("expect accountId: anonymous") |
||||
|
} |
||||
|
|
||||
|
//case3
|
||||
|
//accountId is nil => "anonymous"
|
||||
|
req.Header.Del(s3_constants.AmzAccountId) |
||||
|
if GetAccountId(req) != s3account.AccountAnonymous.Id { |
||||
|
t.Fatal("expect accountId: anonymous") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func TestExtractAcl(t *testing.T) { |
||||
|
type Case struct { |
||||
|
id int |
||||
|
resultErrCode, expectErrCode s3err.ErrorCode |
||||
|
resultGrants, expectGrants []*s3.Grant |
||||
|
} |
||||
|
testCases := make([]*Case, 0) |
||||
|
accountAdminId := "admin" |
||||
|
|
||||
|
{ |
||||
|
//case1 (good case)
|
||||
|
//parse acp from request body
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
req.Body = io.NopCloser(bytes.NewReader([]byte(` |
||||
|
<AccessControlPolicy xmlns="http://s3.amazonaws.com/doc/2006-03-01/"> |
||||
|
<Owner> |
||||
|
<ID>admin</ID> |
||||
|
<DisplayName>admin</DisplayName> |
||||
|
</Owner> |
||||
|
<AccessControlList> |
||||
|
<Grant> |
||||
|
<Grantee xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:type="CanonicalUser"> |
||||
|
<ID>admin</ID> |
||||
|
</Grantee> |
||||
|
<Permission>FULL_CONTROL</Permission> |
||||
|
</Grant> |
||||
|
<Grant> |
||||
|
<Grantee xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:type="Group"> |
||||
|
<URI>http://acs.amazonaws.com/groups/global/AllUsers</URI>
|
||||
|
</Grantee> |
||||
|
<Permission>FULL_CONTROL</Permission> |
||||
|
</Grant> |
||||
|
</AccessControlList> |
||||
|
</AccessControlPolicy> |
||||
|
`))) |
||||
|
objectWriter := "accountA" |
||||
|
grants, errCode := ExtractAcl(req, accountManager, s3_constants.OwnershipObjectWriter, accountAdminId, accountAdminId, objectWriter) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
1, |
||||
|
errCode, s3err.ErrNone, |
||||
|
grants, []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountAdminId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
}, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
{ |
||||
|
//case2 (good case)
|
||||
|
//parse acp from header (cannedAcl)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
req.Body = nil |
||||
|
req.Header.Set(s3_constants.AmzCannedAcl, s3_constants.CannedAclPrivate) |
||||
|
objectWriter := "accountA" |
||||
|
grants, errCode := ExtractAcl(req, accountManager, s3_constants.OwnershipObjectWriter, accountAdminId, accountAdminId, objectWriter) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
2, |
||||
|
errCode, s3err.ErrNone, |
||||
|
grants, []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &objectWriter, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
}, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
{ |
||||
|
//case3 (bad case)
|
||||
|
//parse acp from request body (content is invalid)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
req.Body = io.NopCloser(bytes.NewReader([]byte("zdfsaf"))) |
||||
|
req.Header.Set(s3_constants.AmzCannedAcl, s3_constants.CannedAclPrivate) |
||||
|
objectWriter := "accountA" |
||||
|
_, errCode := ExtractAcl(req, accountManager, s3_constants.OwnershipObjectWriter, accountAdminId, accountAdminId, objectWriter) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
id: 3, |
||||
|
resultErrCode: errCode, expectErrCode: s3err.ErrInvalidRequest, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
//case4 (bad case)
|
||||
|
//parse acp from header (cannedAcl is invalid)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
req.Body = nil |
||||
|
req.Header.Set(s3_constants.AmzCannedAcl, "dfaksjfk") |
||||
|
objectWriter := "accountA" |
||||
|
_, errCode := ExtractAcl(req, accountManager, s3_constants.OwnershipObjectWriter, accountAdminId, "", objectWriter) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
id: 4, |
||||
|
resultErrCode: errCode, expectErrCode: s3err.ErrInvalidRequest, |
||||
|
}) |
||||
|
|
||||
|
{ |
||||
|
//case5 (bad case)
|
||||
|
//parse acp from request body: owner is inconsistent
|
||||
|
req.Body = io.NopCloser(bytes.NewReader([]byte(` |
||||
|
<AccessControlPolicy xmlns="http://s3.amazonaws.com/doc/2006-03-01/"> |
||||
|
<Owner> |
||||
|
<ID>admin</ID> |
||||
|
<DisplayName>admin</DisplayName> |
||||
|
</Owner> |
||||
|
<AccessControlList> |
||||
|
<Grant> |
||||
|
<Grantee xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:type="CanonicalUser"> |
||||
|
<ID>admin</ID> |
||||
|
</Grantee> |
||||
|
<Permission>FULL_CONTROL</Permission> |
||||
|
</Grant> |
||||
|
<Grant> |
||||
|
<Grantee xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:type="Group"> |
||||
|
<URI>http://acs.amazonaws.com/groups/global/AllUsers</URI>
|
||||
|
</Grantee> |
||||
|
<Permission>FULL_CONTROL</Permission> |
||||
|
</Grant> |
||||
|
</AccessControlList> |
||||
|
</AccessControlPolicy> |
||||
|
`))) |
||||
|
objectWriter = "accountA" |
||||
|
_, errCode := ExtractAcl(req, accountManager, s3_constants.OwnershipObjectWriter, accountAdminId, objectWriter, objectWriter) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
id: 5, |
||||
|
resultErrCode: errCode, expectErrCode: s3err.ErrAccessDenied, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
for _, tc := range testCases { |
||||
|
if tc.resultErrCode != tc.expectErrCode { |
||||
|
t.Fatalf("case[%d]: errorCode not expect", tc.id) |
||||
|
} |
||||
|
if !grantsEquals(tc.resultGrants, tc.expectGrants) { |
||||
|
t.Fatalf("case[%d]: grants not expect", tc.id) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func TestParseAndValidateAclHeaders(t *testing.T) { |
||||
|
type Case struct { |
||||
|
id int |
||||
|
resultOwner, expectOwner string |
||||
|
resultErrCode, expectErrCode s3err.ErrorCode |
||||
|
resultGrants, expectGrants []*s3.Grant |
||||
|
} |
||||
|
testCases := make([]*Case, 0) |
||||
|
bucketOwner := "admin" |
||||
|
|
||||
|
{ |
||||
|
//case1 (good case)
|
||||
|
//parse custom acl
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
objectWriter := "accountA" |
||||
|
req.Header.Set(s3_constants.AmzAclFullControl, `uri="http://acs.amazonaws.com/groups/global/AllUsers", id="anonymous", emailAddress="admin@example.com"`) |
||||
|
ownerId, grants, errCode := ParseAndValidateAclHeaders(req, accountManager, s3_constants.OwnershipObjectWriter, bucketOwner, objectWriter, false) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
1, |
||||
|
ownerId, objectWriter, |
||||
|
errCode, s3err.ErrNone, |
||||
|
grants, []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &s3account.AccountAnonymous.Id, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
}, |
||||
|
}) |
||||
|
} |
||||
|
{ |
||||
|
//case2 (good case)
|
||||
|
//parse canned acl (ownership=ObjectWriter)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
objectWriter := "accountA" |
||||
|
req.Header.Set(s3_constants.AmzCannedAcl, s3_constants.CannedAclBucketOwnerFullControl) |
||||
|
ownerId, grants, errCode := ParseAndValidateAclHeaders(req, accountManager, s3_constants.OwnershipObjectWriter, bucketOwner, objectWriter, false) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
2, |
||||
|
ownerId, objectWriter, |
||||
|
errCode, s3err.ErrNone, |
||||
|
grants, []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &objectWriter, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &bucketOwner, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
}, |
||||
|
}) |
||||
|
} |
||||
|
{ |
||||
|
//case3 (good case)
|
||||
|
//parse canned acl (ownership=OwnershipBucketOwnerPreferred)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
objectWriter := "accountA" |
||||
|
req.Header.Set(s3_constants.AmzCannedAcl, s3_constants.CannedAclBucketOwnerFullControl) |
||||
|
ownerId, grants, errCode := ParseAndValidateAclHeaders(req, accountManager, s3_constants.OwnershipBucketOwnerPreferred, bucketOwner, objectWriter, false) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
3, |
||||
|
ownerId, bucketOwner, |
||||
|
errCode, s3err.ErrNone, |
||||
|
grants, []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &bucketOwner, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
}, |
||||
|
}) |
||||
|
} |
||||
|
{ |
||||
|
//case4 (bad case)
|
||||
|
//parse custom acl (grantee id not exists)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
objectWriter := "accountA" |
||||
|
req.Header.Set(s3_constants.AmzAclFullControl, `uri="http://acs.amazonaws.com/groups/global/AllUsers", id="notExistsAccount", emailAddress="admin@example.com"`) |
||||
|
_, _, errCode := ParseAndValidateAclHeaders(req, accountManager, s3_constants.OwnershipObjectWriter, bucketOwner, objectWriter, false) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
id: 4, |
||||
|
resultErrCode: errCode, expectErrCode: s3err.ErrInvalidRequest, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
{ |
||||
|
//case5 (bad case)
|
||||
|
//parse custom acl (invalid format)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
objectWriter := "accountA" |
||||
|
req.Header.Set(s3_constants.AmzAclFullControl, `uri="http:sfasf"`) |
||||
|
_, _, errCode := ParseAndValidateAclHeaders(req, accountManager, s3_constants.OwnershipObjectWriter, bucketOwner, objectWriter, false) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
id: 5, |
||||
|
resultErrCode: errCode, expectErrCode: s3err.ErrInvalidRequest, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
{ |
||||
|
//case6 (bad case)
|
||||
|
//parse canned acl (invalid value)
|
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
objectWriter := "accountA" |
||||
|
req.Header.Set(s3_constants.AmzCannedAcl, `uri="http:sfasf"`) |
||||
|
_, _, errCode := ParseAndValidateAclHeaders(req, accountManager, s3_constants.OwnershipObjectWriter, bucketOwner, objectWriter, false) |
||||
|
testCases = append(testCases, &Case{ |
||||
|
id: 5, |
||||
|
resultErrCode: errCode, expectErrCode: s3err.ErrInvalidRequest, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
for _, tc := range testCases { |
||||
|
if tc.expectErrCode != tc.resultErrCode { |
||||
|
t.Errorf("case[%d]: errCode unexpect", tc.id) |
||||
|
} |
||||
|
if tc.resultOwner != tc.expectOwner { |
||||
|
t.Errorf("case[%d]: ownerId unexpect", tc.id) |
||||
|
} |
||||
|
if !grantsEquals(tc.resultGrants, tc.expectGrants) { |
||||
|
t.Fatalf("case[%d]: grants not expect", tc.id) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func grantsEquals(a, b []*s3.Grant) bool { |
||||
|
if len(a) != len(b) { |
||||
|
return false |
||||
|
} |
||||
|
for i, grant := range a { |
||||
|
if !GrantEquals(grant, b[i]) { |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
func TestDetermineReqGrants(t *testing.T) { |
||||
|
{ |
||||
|
//case1: request account is anonymous
|
||||
|
accountId := s3account.AccountAnonymous.Id |
||||
|
reqPermission := s3_constants.PermissionRead |
||||
|
|
||||
|
resultGrants := DetermineReqGrants(accountId, reqPermission) |
||||
|
expectGrants := []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &reqPermission, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &reqPermission, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
} |
||||
|
if !grantsEquals(resultGrants, expectGrants) { |
||||
|
t.Fatalf("grants not expect") |
||||
|
} |
||||
|
} |
||||
|
{ |
||||
|
//case2: request account is not anonymous (Iam authed)
|
||||
|
accountId := "accountX" |
||||
|
reqPermission := s3_constants.PermissionRead |
||||
|
|
||||
|
resultGrants := DetermineReqGrants(accountId, reqPermission) |
||||
|
expectGrants := []*s3.Grant{ |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &reqPermission, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &reqPermission, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeCanonicalUser, |
||||
|
ID: &accountId, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAuthenticatedUsers, |
||||
|
}, |
||||
|
Permission: &reqPermission, |
||||
|
}, |
||||
|
{ |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAuthenticatedUsers, |
||||
|
}, |
||||
|
Permission: &s3_constants.PermissionFullControl, |
||||
|
}, |
||||
|
} |
||||
|
if !grantsEquals(resultGrants, expectGrants) { |
||||
|
t.Fatalf("grants not expect") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func TestAssembleEntryWithAcp(t *testing.T) { |
||||
|
defaultOwner := "admin" |
||||
|
{ |
||||
|
//case1
|
||||
|
expectOwner := "accountS" |
||||
|
expectGrants := []*s3.Grant{ |
||||
|
{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}, |
||||
|
} |
||||
|
entry := &filer_pb.Entry{} |
||||
|
AssembleEntryWithAcp(entry, expectOwner, expectGrants) |
||||
|
|
||||
|
resultOwner := GetAcpOwner(entry.Extended, defaultOwner) |
||||
|
if resultOwner != expectOwner { |
||||
|
t.Fatalf("owner not expect") |
||||
|
} |
||||
|
|
||||
|
resultGrants := GetAcpGrants(entry.Extended) |
||||
|
if !grantsEquals(resultGrants, expectGrants) { |
||||
|
t.Fatal("grants not expect") |
||||
|
} |
||||
|
} |
||||
|
{ |
||||
|
//case2
|
||||
|
entry := &filer_pb.Entry{} |
||||
|
AssembleEntryWithAcp(entry, "", nil) |
||||
|
|
||||
|
resultOwner := GetAcpOwner(entry.Extended, defaultOwner) |
||||
|
if resultOwner != defaultOwner { |
||||
|
t.Fatalf("owner not expect") |
||||
|
} |
||||
|
|
||||
|
resultGrants := GetAcpGrants(entry.Extended) |
||||
|
if len(resultGrants) != 0 { |
||||
|
t.Fatal("grants not expect") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
func TestGrantEquals(t *testing.T) { |
||||
|
testCases := map[bool]bool{ |
||||
|
GrantEquals(nil, nil): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{}, nil): false, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{}, &s3.Grant{}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
}, &s3.Grant{}): false, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{}, |
||||
|
}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{}, |
||||
|
}): false, |
||||
|
|
||||
|
//type not present, compare other fields of grant is meaningless
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
EmailAddress: &s3account.AccountAdmin.EmailAddress, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
}, |
||||
|
}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
}, |
||||
|
}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionWrite, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}): false, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
}, |
||||
|
}): true, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
}, |
||||
|
}): false, |
||||
|
|
||||
|
GrantEquals(&s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}, &s3.Grant{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}): true, |
||||
|
} |
||||
|
|
||||
|
for tc, expect := range testCases { |
||||
|
if tc != expect { |
||||
|
t.Fatal("TestGrantEquals not expect!") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func TestSetAcpOwnerHeader(t *testing.T) { |
||||
|
ownerId := "accountZ" |
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
SetAcpOwnerHeader(req, ownerId) |
||||
|
|
||||
|
if req.Header.Get(s3_constants.ExtAmzOwnerKey) != ownerId { |
||||
|
t.Fatalf("owner unexpect") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func TestSetAcpGrantsHeader(t *testing.T) { |
||||
|
req := &http.Request{ |
||||
|
Header: make(map[string][]string), |
||||
|
} |
||||
|
grants := []*s3.Grant{ |
||||
|
{ |
||||
|
Permission: &s3_constants.PermissionRead, |
||||
|
Grantee: &s3.Grantee{ |
||||
|
Type: &s3_constants.GrantTypeGroup, |
||||
|
ID: &s3account.AccountAdmin.Id, |
||||
|
URI: &s3_constants.GranteeGroupAllUsers, |
||||
|
}, |
||||
|
}, |
||||
|
} |
||||
|
SetAcpGrantsHeader(req, grants) |
||||
|
|
||||
|
grantsJson, _ := json.Marshal(grants) |
||||
|
if req.Header.Get(s3_constants.ExtAmzAclKey) != string(grantsJson) { |
||||
|
t.Fatalf("owner unexpect") |
||||
|
} |
||||
|
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue