fix a secure issue
Signed-off-by: yxxhero <aiopsclub@163.com>
This commit is contained in:
parent
4b047497bb
commit
7ab0d0e8a1
|
|
@ -10,6 +10,7 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"d7y.io/dragonfly/v2/manager/model"
|
"d7y.io/dragonfly/v2/manager/model"
|
||||||
|
"d7y.io/dragonfly/v2/pkg/util/stringutils"
|
||||||
"golang.org/x/crypto/bcrypt"
|
"golang.org/x/crypto/bcrypt"
|
||||||
"golang.org/x/oauth2"
|
"golang.org/x/oauth2"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
|
@ -68,7 +69,7 @@ func (oa *baseOauth2) GetRediectURL(db *gorm.DB) (string, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (oa *baseOauth2) AuthCodeURL() string {
|
func (oa *baseOauth2) AuthCodeURL() string {
|
||||||
return oa.Config.AuthCodeURL("state")
|
return oa.Config.AuthCodeURL(stringutils.RandString(5))
|
||||||
}
|
}
|
||||||
|
|
||||||
func (oa *baseOauth2) GetOauthUserInfo(token string) (*oauth2User, error) {
|
func (oa *baseOauth2) GetOauthUserInfo(token string) (*oauth2User, error) {
|
||||||
|
|
|
||||||
|
|
@ -18,7 +18,9 @@
|
||||||
package stringutils
|
package stringutils
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"math/rand"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
"unicode"
|
"unicode"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -65,3 +67,13 @@ func Contains(slice []string, ele string) bool {
|
||||||
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func RandString(len int) string {
|
||||||
|
r := rand.New(rand.NewSource(time.Now().Unix()))
|
||||||
|
bytes := make([]byte, len)
|
||||||
|
for i := 0; i < len; i++ {
|
||||||
|
b := r.Intn(26) + 65
|
||||||
|
bytes[i] = byte(b)
|
||||||
|
}
|
||||||
|
return string(bytes)
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue