|
|
package middleware
|
|
|
|
|
|
import (
|
|
|
"bytes"
|
|
|
"context"
|
|
|
"crypto/md5"
|
|
|
"fmt"
|
|
|
"github.com/HFO4/cloudreve/models"
|
|
|
"github.com/HFO4/cloudreve/pkg/auth"
|
|
|
"github.com/HFO4/cloudreve/pkg/cache"
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/driver/onedrive"
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/driver/oss"
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/driver/upyun"
|
|
|
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
|
"github.com/HFO4/cloudreve/pkg/util"
|
|
|
"github.com/gin-contrib/sessions"
|
|
|
"github.com/gin-gonic/gin"
|
|
|
"github.com/qiniu/api.v7/v7/auth/qbox"
|
|
|
"io/ioutil"
|
|
|
"net/http"
|
|
|
)
|
|
|
|
|
|
// SignRequired 验证请求签名
|
|
|
func SignRequired() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
var err error
|
|
|
switch c.Request.Method {
|
|
|
case "PUT", "POST":
|
|
|
err = auth.CheckRequest(auth.General, c.Request)
|
|
|
// TODO 生产环境去掉下一行
|
|
|
//err = nil
|
|
|
default:
|
|
|
err = auth.CheckURI(auth.General, c.Request.URL)
|
|
|
}
|
|
|
|
|
|
if err != nil {
|
|
|
c.JSON(200, serializer.Err(serializer.CodeCheckLogin, err.Error(), err))
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// CurrentUser 获取登录用户
|
|
|
func CurrentUser() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
session := sessions.Default(c)
|
|
|
uid := session.Get("user_id")
|
|
|
if uid != nil {
|
|
|
user, err := model.GetActiveUserByID(uid)
|
|
|
if err == nil {
|
|
|
c.Set("user", &user)
|
|
|
}
|
|
|
}
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// AuthRequired 需要登录
|
|
|
func AuthRequired() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
if user, _ := c.Get("user"); user != nil {
|
|
|
if _, ok := user.(*model.User); ok {
|
|
|
c.Next()
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
|
|
|
c.JSON(200, serializer.CheckLogin())
|
|
|
c.Abort()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// WebDAVAuth 验证WebDAV登录及权限
|
|
|
func WebDAVAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// OPTIONS 请求不需要鉴权，否则Windows10下无法保存文档
|
|
|
if c.Request.Method == "OPTIONS" {
|
|
|
c.Next()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
username, password, ok := c.Request.BasicAuth()
|
|
|
if !ok {
|
|
|
c.Writer.Header()["WWW-Authenticate"] = []string{`Basic realm="cloudreve"`}
|
|
|
c.Status(http.StatusUnauthorized)
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
expectedUser, err := model.GetUserByEmail(username)
|
|
|
if err != nil {
|
|
|
c.Status(http.StatusUnauthorized)
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 密码正确？
|
|
|
webdav, err := model.GetWebdavByPassword(password, expectedUser.ID)
|
|
|
if err != nil {
|
|
|
c.Status(http.StatusUnauthorized)
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 用户组已启用WebDAV？
|
|
|
if !expectedUser.Group.WebDAVEnabled {
|
|
|
c.Status(http.StatusForbidden)
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Set("user", &expectedUser)
|
|
|
c.Set("webdav", webdav)
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// uploadCallbackCheck 对上传回调请求的 callback key 进行验证，如果成功则返回上传用户
|
|
|
func uploadCallbackCheck(c *gin.Context) (serializer.Response, *model.User) {
|
|
|
// 验证 Callback Key
|
|
|
callbackKey := c.Param("key")
|
|
|
if callbackKey == "" {
|
|
|
return serializer.ParamErr("Callback Key 不能为空", nil), nil
|
|
|
}
|
|
|
callbackSessionRaw, exist := cache.Get("callback_" + callbackKey)
|
|
|
if !exist {
|
|
|
return serializer.ParamErr("回调会话不存在或已过期", nil), nil
|
|
|
}
|
|
|
callbackSession := callbackSessionRaw.(serializer.UploadSession)
|
|
|
c.Set("callbackSession", &callbackSession)
|
|
|
|
|
|
// 清理回调会话
|
|
|
_ = cache.Deletes([]string{callbackKey}, "callback_")
|
|
|
|
|
|
// 查找用户
|
|
|
user, err := model.GetActiveUserByID(callbackSession.UID)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCheckLogin, "找不到用户", err), nil
|
|
|
}
|
|
|
c.Set("user", &user)
|
|
|
|
|
|
return serializer.Response{}, &user
|
|
|
}
|
|
|
|
|
|
// RemoteCallbackAuth 远程回调签名验证
|
|
|
func RemoteCallbackAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// 验证key并查找用户
|
|
|
resp, user := uploadCallbackCheck(c)
|
|
|
if resp.Code != 0 {
|
|
|
c.JSON(200, resp)
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 验证签名
|
|
|
authInstance := auth.HMACAuth{SecretKey: []byte(user.Policy.SecretKey)}
|
|
|
if err := auth.CheckRequest(authInstance, c.Request); err != nil {
|
|
|
c.JSON(200, serializer.Err(serializer.CodeCheckLogin, err.Error(), err))
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Next()
|
|
|
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// QiniuCallbackAuth 七牛回调签名验证
|
|
|
func QiniuCallbackAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// 验证key并查找用户
|
|
|
resp, user := uploadCallbackCheck(c)
|
|
|
if resp.Code != 0 {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: resp.Msg})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 验证回调是否来自qiniu
|
|
|
mac := qbox.NewMac(user.Policy.AccessKey, user.Policy.SecretKey)
|
|
|
ok, err := mac.VerifyCallback(c.Request)
|
|
|
if err != nil {
|
|
|
util.Log().Debug("无法验证回调请求，%s", err)
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: "无法验证回调请求"})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
if !ok {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: "回调签名无效"})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// OSSCallbackAuth 阿里云OSS回调签名验证
|
|
|
func OSSCallbackAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// 验证key并查找用户
|
|
|
resp, _ := uploadCallbackCheck(c)
|
|
|
if resp.Code != 0 {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: resp.Msg})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
err := oss.VerifyCallbackSignature(c.Request)
|
|
|
if err != nil {
|
|
|
util.Log().Debug("回调签名验证失败，%s", err)
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: "回调签名验证失败"})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// UpyunCallbackAuth 又拍云回调签名验证
|
|
|
func UpyunCallbackAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// 验证key并查找用户
|
|
|
resp, user := uploadCallbackCheck(c)
|
|
|
if resp.Code != 0 {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: resp.Msg})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 获取请求正文
|
|
|
body, err := ioutil.ReadAll(c.Request.Body)
|
|
|
c.Request.Body.Close()
|
|
|
if err != nil {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: err.Error()})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Request.Body = ioutil.NopCloser(bytes.NewReader(body))
|
|
|
|
|
|
// 准备验证Upyun回调签名
|
|
|
handler := upyun.Driver{Policy: &user.Policy}
|
|
|
contentMD5 := c.Request.Header.Get("Content-Md5")
|
|
|
date := c.Request.Header.Get("Date")
|
|
|
actualSignature := c.Request.Header.Get("Authorization")
|
|
|
|
|
|
// 计算正文MD5
|
|
|
actualContentMD5 := fmt.Sprintf("%x", md5.Sum(body))
|
|
|
if actualContentMD5 != contentMD5 {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: "MD5不一致"})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 计算理论签名
|
|
|
signature := handler.Sign(context.Background(), []string{
|
|
|
"POST",
|
|
|
c.Request.URL.Path,
|
|
|
date,
|
|
|
contentMD5,
|
|
|
})
|
|
|
|
|
|
// 对比签名
|
|
|
if signature != actualSignature {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: "鉴权失败"})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// OneDriveCallbackAuth OneDrive回调签名验证
|
|
|
// TODO 解耦
|
|
|
func OneDriveCallbackAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// 验证key并查找用户
|
|
|
resp, _ := uploadCallbackCheck(c)
|
|
|
if resp.Code != 0 {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: resp.Msg})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 发送回调结束信号
|
|
|
onedrive.FinishCallback(c.Param("key"))
|
|
|
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// COSCallbackAuth 腾讯云COS回调签名验证
|
|
|
// TODO 解耦 测试
|
|
|
func COSCallbackAuth() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
// 验证key并查找用户
|
|
|
resp, _ := uploadCallbackCheck(c)
|
|
|
if resp.Code != 0 {
|
|
|
c.JSON(401, serializer.QiniuCallbackFailed{Error: resp.Msg})
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// IsAdmin 必须为管理员用户组
|
|
|
func IsAdmin() gin.HandlerFunc {
|
|
|
return func(c *gin.Context) {
|
|
|
user, _ := c.Get("user")
|
|
|
if user.(*model.User).Group.ID != 1 && user.(*model.User).ID != 1 {
|
|
|
c.JSON(200, serializer.Err(serializer.CodeAdminRequired, "您不是管理组成员", nil))
|
|
|
c.Abort()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
c.Next()
|
|
|
}
|
|
|
}
|
