mirror of
https://github.com/cloudreve/cloudreve.git
synced 2025-10-31 08:39:10 +08:00
117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package oss
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto"
|
|
"crypto/md5"
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/base64"
|
|
"encoding/pem"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/HFO4/cloudreve/pkg/cache"
|
|
"github.com/HFO4/cloudreve/pkg/request"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
// GetPublicKey 从回调请求或缓存中获取OSS的回调签名公钥
|
|
func GetPublicKey(r *http.Request) ([]byte, error) {
|
|
var pubKey []byte
|
|
|
|
// 尝试从缓存中获取
|
|
pub, exist := cache.Get("oss_public_key")
|
|
if exist {
|
|
return pub.([]byte), nil
|
|
}
|
|
|
|
// 从请求中获取
|
|
pubURL, err := base64.StdEncoding.DecodeString(r.Header.Get("x-oss-pub-key-url"))
|
|
if err != nil {
|
|
return pubKey, err
|
|
}
|
|
|
|
// 确保这个 public key 是由 OSS 颁发的
|
|
if !strings.HasPrefix(string(pubURL), "http://gosspublic.alicdn.com/") &&
|
|
!strings.HasPrefix(string(pubURL), "https://gosspublic.alicdn.com/") {
|
|
return pubKey, errors.New("公钥URL无效")
|
|
}
|
|
|
|
// 获取公钥
|
|
client := request.HTTPClient{}
|
|
body, err := client.Request("GET", string(pubURL), nil).
|
|
CheckHTTPResponse(200).
|
|
GetResponse()
|
|
if err != nil {
|
|
return pubKey, err
|
|
}
|
|
|
|
// 写入缓存
|
|
_ = cache.Set("oss_public_key", []byte(body), 86400*7)
|
|
|
|
return []byte(body), nil
|
|
}
|
|
|
|
func getRequestMD5(r *http.Request) ([]byte, error) {
|
|
var byteMD5 []byte
|
|
|
|
// 获取请求正文
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
r.Body.Close()
|
|
if err != nil {
|
|
return byteMD5, err
|
|
}
|
|
r.Body = ioutil.NopCloser(bytes.NewReader(body))
|
|
|
|
strURLPathDecode, err := url.PathUnescape(r.URL.Path)
|
|
if err != nil {
|
|
return byteMD5, err
|
|
}
|
|
|
|
strAuth := fmt.Sprintf("%s\n%s", strURLPathDecode, string(body))
|
|
md5Ctx := md5.New()
|
|
md5Ctx.Write([]byte(strAuth))
|
|
byteMD5 = md5Ctx.Sum(nil)
|
|
|
|
return byteMD5, nil
|
|
}
|
|
|
|
// VerifyCallbackSignature 验证OSS回调请求
|
|
func VerifyCallbackSignature(r *http.Request) error {
|
|
bytePublicKey, err := GetPublicKey(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
byteMD5, err := getRequestMD5(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
strAuthorizationBase64 := r.Header.Get("authorization")
|
|
if strAuthorizationBase64 == "" {
|
|
return errors.New("no authorization field in Request header")
|
|
}
|
|
authorization, _ := base64.StdEncoding.DecodeString(strAuthorizationBase64)
|
|
|
|
pubBlock, _ := pem.Decode(bytePublicKey)
|
|
if pubBlock == nil {
|
|
return errors.New("pubBlock not exist")
|
|
}
|
|
pubInterface, err := x509.ParsePKIXPublicKey(pubBlock.Bytes)
|
|
if (pubInterface == nil) || (err != nil) {
|
|
return err
|
|
}
|
|
pub := pubInterface.(*rsa.PublicKey)
|
|
|
|
errorVerifyPKCS1v15 := rsa.VerifyPKCS1v15(pub, crypto.MD5, byteMD5, authorization)
|
|
if errorVerifyPKCS1v15 != nil {
|
|
return errorVerifyPKCS1v15
|
|
}
|
|
|
|
return nil
|
|
}
|