mirror of https://github.com/rocboss/paopao-ce
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.
31 lines
573 B
31 lines
573 B
package security
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
)
|
|
|
|
var (
|
|
_ core.AttachmentCheckService = (*attachmentCheckServant)(nil)
|
|
)
|
|
|
|
type attachmentCheckServant struct {
|
|
domain string
|
|
}
|
|
|
|
func (s *attachmentCheckServant) CheckAttachment(uri string) error {
|
|
if strings.Index(uri, s.domain) != 0 {
|
|
return fmt.Errorf("附件非本站资源")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func NewAttachmentCheckService() core.AttachmentCheckService {
|
|
return &attachmentCheckServant{
|
|
domain: conf.GetOssDomain(),
|
|
}
|
|
}
|