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.
27 lines
500 B
27 lines
500 B
2 years ago
|
package security
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
||
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
||
|
|
)
|
||
|
|
|
||
|
|
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(),
|
||
|
|
}
|
||
|
|
}
|