You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.2 KiB
78 lines
2.2 KiB
// Copyright © 2023 OpenIM. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package third
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/mcontext"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/tokenverify"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/third"
|
|
"strings"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
func toPbMapArray(m map[string][]string) []*third.KeyValues {
|
|
res := make([]*third.KeyValues, 0, len(m))
|
|
for key := range m {
|
|
res = append(res, &third.KeyValues{
|
|
Key: key,
|
|
Values: m[key],
|
|
})
|
|
}
|
|
return res
|
|
}
|
|
|
|
func checkUploadName(ctx context.Context, name string) error {
|
|
if name == "" {
|
|
return errs.ErrArgs.Wrap("name is empty")
|
|
}
|
|
if name[0] == '/' {
|
|
return errs.ErrArgs.Wrap("name cannot start with `/`")
|
|
}
|
|
if err := checkValidObjectName(name); err != nil {
|
|
return errs.ErrArgs.Wrap(err.Error())
|
|
}
|
|
opUserID := mcontext.GetOpUserID(ctx)
|
|
if opUserID == "" {
|
|
return errs.ErrNoPermission.Wrap("opUserID is empty")
|
|
}
|
|
if !tokenverify.IsManagerUserID(opUserID) {
|
|
if !strings.HasPrefix(name, opUserID+"/") {
|
|
return errs.ErrNoPermission.Wrap(fmt.Sprintf("name must start with `%s/`", opUserID))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func checkValidObjectNamePrefix(objectName string) error {
|
|
if len(objectName) > 1024 {
|
|
return errors.New("object name cannot be longer than 1024 characters")
|
|
}
|
|
if !utf8.ValidString(objectName) {
|
|
return errors.New("object name with non UTF-8 strings are not supported")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func checkValidObjectName(objectName string) error {
|
|
if strings.TrimSpace(objectName) == "" {
|
|
return errors.New("object name cannot be empty")
|
|
}
|
|
return checkValidObjectNamePrefix(objectName)
|
|
}
|