93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
|
package oss
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"crypto/hmac"
|
|||
|
|
"crypto/sha1"
|
|||
|
|
"encoding/base64"
|
|||
|
|
"hash"
|
|||
|
|
"io"
|
|||
|
|
"net/http"
|
|||
|
|
"sort"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 用于signHeader的字典排序存放容器。
|
|||
|
|
type headerSorter struct {
|
|||
|
|
Keys []string
|
|||
|
|
Vals []string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 生成签名方法（直接设置请求的Header）。
|
|||
|
|
func (conn Conn) signHeader(req *http.Request, canonicalizedResource string) {
|
|||
|
|
// Find out the "x-oss-"'s address in this request'header
|
|||
|
|
temp := make(map[string]string)
|
|||
|
|
|
|||
|
|
for k, v := range req.Header {
|
|||
|
|
if strings.HasPrefix(strings.ToLower(k), "x-oss-") {
|
|||
|
|
temp[strings.ToLower(k)] = v[0]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
hs := newHeaderSorter(temp)
|
|||
|
|
|
|||
|
|
// Sort the temp by the Ascending Order
|
|||
|
|
hs.Sort()
|
|||
|
|
|
|||
|
|
// Get the CanonicalizedOSSHeaders
|
|||
|
|
canonicalizedOSSHeaders := ""
|
|||
|
|
for i := range hs.Keys {
|
|||
|
|
canonicalizedOSSHeaders += hs.Keys[i] + ":" + hs.Vals[i] + "\n"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Give other parameters values
|
|||
|
|
date := req.Header.Get(HTTPHeaderDate)
|
|||
|
|
contentType := req.Header.Get(HTTPHeaderContentType)
|
|||
|
|
contentMd5 := req.Header.Get(HTTPHeaderContentMD5)
|
|||
|
|
|
|||
|
|
signStr := req.Method + "\n" + contentMd5 + "\n" + contentType + "\n" + date + "\n" + canonicalizedOSSHeaders + canonicalizedResource
|
|||
|
|
h := hmac.New(func() hash.Hash { return sha1.New() }, []byte(conn.config.AccessKeySecret))
|
|||
|
|
io.WriteString(h, signStr)
|
|||
|
|
signedStr := base64.StdEncoding.EncodeToString(h.Sum(nil))
|
|||
|
|
|
|||
|
|
// Get the final Authorization' string
|
|||
|
|
authorizationStr := "OSS " + conn.config.AccessKeyID + ":" + signedStr
|
|||
|
|
|
|||
|
|
// Give the parameter "Authorization" value
|
|||
|
|
req.Header.Set(HTTPHeaderAuthorization, authorizationStr)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Additional function for function SignHeader.
|
|||
|
|
func newHeaderSorter(m map[string]string) *headerSorter {
|
|||
|
|
hs := &headerSorter{
|
|||
|
|
Keys: make([]string, 0, len(m)),
|
|||
|
|
Vals: make([]string, 0, len(m)),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for k, v := range m {
|
|||
|
|
hs.Keys = append(hs.Keys, k)
|
|||
|
|
hs.Vals = append(hs.Vals, v)
|
|||
|
|
}
|
|||
|
|
return hs
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Additional function for function SignHeader.
|
|||
|
|
func (hs *headerSorter) Sort() {
|
|||
|
|
sort.Sort(hs)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Additional function for function SignHeader.
|
|||
|
|
func (hs *headerSorter) Len() int {
|
|||
|
|
return len(hs.Vals)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Additional function for function SignHeader.
|
|||
|
|
func (hs *headerSorter) Less(i, j int) bool {
|
|||
|
|
return bytes.Compare([]byte(hs.Keys[i]), []byte(hs.Keys[j])) < 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Additional function for function SignHeader.
|
|||
|
|
func (hs *headerSorter) Swap(i, j int) {
|
|||
|
|
hs.Vals[i], hs.Vals[j] = hs.Vals[j], hs.Vals[i]
|
|||
|
|
hs.Keys[i], hs.Keys[j] = hs.Keys[j], hs.Keys[i]
|
|||
|
|
}
|