复现双验签
This commit is contained in:
parent
b7239af0e0
commit
43c9890f06
|
|
@ -1,6 +1,10 @@
|
|||
package model
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
|
|
@ -9,3 +13,144 @@ type KillaraCustomerTokenModel struct {
|
|||
db *gorm.DB
|
||||
TableName string // 表名
|
||||
}
|
||||
|
||||
// InsertToken inserts a new token record
|
||||
// func (m *KillaraCustomerTokenModel) InsertToken(data map[string]interface{}) error {
|
||||
// token, ok := data["token"].(string)
|
||||
// if !ok {
|
||||
// return ErrInvalidTokenValue
|
||||
// }
|
||||
|
||||
// m.DeleteToken(token)
|
||||
|
||||
// customerID, _ := data["customer_id"].(uint64)
|
||||
// platform, _ := data["platform"].(int64)
|
||||
// if customerID != 0 && platform != 0 {
|
||||
// m.db.Where("customer_id = ? AND platform = ?", customerID, platform).Delete(&KillaraCustomerToken{})
|
||||
// }
|
||||
|
||||
// return m.db.Create(&KillaraCustomerToken{
|
||||
// Token: &token,
|
||||
// CustomerId: &customerID,
|
||||
// Data: nilStringPtr(data["data"].(string)),
|
||||
// Expiry: nilTimePtr(time.Unix(int64(data["expiry"].(float64)), 0)),
|
||||
// Platform: platform,
|
||||
// }).Error
|
||||
// }
|
||||
|
||||
// UpdateTokenData updates the data field of a token record
|
||||
func (m *KillaraCustomerTokenModel) UpdateTokenData(token string, data map[string]interface{}) error {
|
||||
dataStr, _ := json.Marshal(data)
|
||||
return m.db.Model(&KillaraCustomerToken{}).Where("token = ?", token).Update("data", string(dataStr)).Error
|
||||
}
|
||||
|
||||
// UpdateTokenExpiry updates the expiry field of a token record
|
||||
func (m *KillaraCustomerTokenModel) UpdateTokenExpiry(token string, expiry int64) error {
|
||||
return m.db.Model(&KillaraCustomerToken{}).Where("token = ?", token).Update("expiry", time.Unix(expiry, 0)).Error
|
||||
}
|
||||
|
||||
// UpdateTokenCustomerID updates the customer_id field of a token record
|
||||
func (m *KillaraCustomerTokenModel) UpdateTokenCustomerID(token string, customerID uint64) error {
|
||||
err := m.db.Model(&KillaraCustomerToken{}).Where("token = ?", token).Update("customer_id", customerID).Error
|
||||
if err == gorm.ErrRecordNotFound {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// DeleteToken deletes a token record
|
||||
func (m *KillaraCustomerTokenModel) DeleteToken(token string) error {
|
||||
err := m.db.Where("token = ?", token).Delete(&KillaraCustomerToken{}).Error
|
||||
if err == gorm.ErrRecordNotFound {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// GetToken retrieves a token record by token value
|
||||
func (m *KillaraCustomerTokenModel) GetToken(token string) (*KillaraCustomerToken, error) {
|
||||
var record KillaraCustomerToken
|
||||
err := m.db.Where("token = ?", token).First(&record).Error
|
||||
if err == gorm.ErrRecordNotFound {
|
||||
return nil, nil
|
||||
}
|
||||
return &record, err
|
||||
}
|
||||
|
||||
// CheckToken checks if a token is valid
|
||||
// func (m *KillaraCustomerTokenModel) CheckToken(token string) (*KillaraCustomerToken, error) {
|
||||
// var record KillaraCustomerToken
|
||||
// err := m.db.Where("token = ?", token).First(&record).Error
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
|
||||
// if record.Expiry == nil || record.Expiry.Before(time.Now()) {
|
||||
// m.DeleteToken(token)
|
||||
// return nil, ErrTokenExpired
|
||||
// }
|
||||
|
||||
// return &record, nil
|
||||
// }
|
||||
|
||||
// ClearDuplicateToken clears duplicate tokens for a customer_id
|
||||
func (m *KillaraCustomerTokenModel) ClearDuplicateToken(customerID uint64, currentToken string, platform int64) error {
|
||||
tokens := []KillaraCustomerToken{}
|
||||
err := m.db.Where("customer_id = ? AND platform = ?", customerID, platform).Find(&tokens).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, token := range tokens {
|
||||
if *token.Token != currentToken {
|
||||
m.UpdateTokenCustomerID(*token.Token, 0)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetTokenByCustomerID retrieves the token value for a customer_id
|
||||
func (m *KillaraCustomerTokenModel) GetTokenByCustomerID(customerID uint64) (string, error) {
|
||||
var record KillaraCustomerToken
|
||||
err := m.db.Select("token").Where("customer_id = ?", customerID).First(&record).Error
|
||||
return *record.Token, err
|
||||
}
|
||||
|
||||
// GetTokenDataByCustomerID retrieves the token data for a customer_id
|
||||
func (m *KillaraCustomerTokenModel) GetTokenDataByCustomerID(customerID uint64) (map[string]interface{}, error) {
|
||||
var record KillaraCustomerToken
|
||||
err := m.db.Select("data").Where("customer_id = ?", customerID).First(&record).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var data map[string]interface{}
|
||||
if record.Data != nil {
|
||||
err = json.Unmarshal([]byte(*record.Data), &data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return data, nil
|
||||
}
|
||||
|
||||
func nilStringPtr(s string) *string {
|
||||
if s == "" {
|
||||
return nil
|
||||
}
|
||||
return &s
|
||||
}
|
||||
|
||||
func nilTimePtr(t time.Time) *time.Time {
|
||||
if t.IsZero() {
|
||||
return nil
|
||||
}
|
||||
return &t
|
||||
}
|
||||
|
||||
var (
|
||||
ErrInvalidTokenValue = errors.New("invalid token value")
|
||||
ErrTokenExpired = errors.New("token has expired")
|
||||
)
|
||||
|
|
|
|||
197
php_model/Token.php
Normal file
197
php_model/Token.php
Normal file
|
|
@ -0,0 +1,197 @@
|
|||
<?php
|
||||
|
||||
/**
|
||||
* Token类存储APP和微信小程序用户登录产生的数据，并负责较验用户的状态。
|
||||
*
|
||||
* @author Evan
|
||||
*/
|
||||
|
||||
class Model_Customer_Token extends \application\Model\Base\BaseModel
|
||||
{
|
||||
private $_customerTokenTable;
|
||||
private $_redisKey = 'token_';
|
||||
|
||||
public $db;
|
||||
public $redis;
|
||||
|
||||
public function __construct()
|
||||
{
|
||||
$this->db = \system\engine\Registry::get('db');
|
||||
|
||||
// $this->redis = \system\engine\Registry::get('redis');
|
||||
|
||||
$this->redis = false;
|
||||
|
||||
$config = \system\engine\Registry::get('config');
|
||||
|
||||
$this->tableFullName = $this->_customerTokenTable = $config->database->prefix
|
||||
. 'customer_token';
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除指定的登录数据
|
||||
*
|
||||
* @param Array $data <p>登录认证数据</p>
|
||||
*
|
||||
*/
|
||||
public function insertToken($data)
|
||||
{
|
||||
|
||||
$this->deleteToken($data['token']);
|
||||
|
||||
if (!empty($data['customer_id']) && !empty($data['platform'])) {
|
||||
$this->db->query('DELETE FROM ' . $this->_customerTokenTable . ' WHERE customer_id = '
|
||||
. $this->db->quote($data['customer_id']) . ' AND platform = ' . $this->db->quote($data['platform']));
|
||||
}
|
||||
|
||||
$this->db->insert($this->_customerTokenTable, $data);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新指定的登录数据
|
||||
*
|
||||
* @param String $token <p>用户登录时生成的token字符串</p>
|
||||
* @param Array $data <p>其它需要记录的数据</p>
|
||||
*
|
||||
*/
|
||||
public function updateTokenData($token, $data = array())
|
||||
{
|
||||
|
||||
$where = $this->db->quoteInto('token = ?', $token);
|
||||
|
||||
$this->db->update($this->_customerTokenTable, array(
|
||||
'data' => json_encode($data)
|
||||
), $where);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新登录数据的生命期
|
||||
*
|
||||
* @param String $token <p>用户登录时生成的token字符串</p>
|
||||
* @param Int $expiry <p>过期时间，Unix时间戳</p>
|
||||
*
|
||||
*/
|
||||
public function updateTokenExpiry($token, $expiry)
|
||||
{
|
||||
|
||||
$where = $this->db->quoteInto('token = ?', $token);
|
||||
|
||||
$this->db->update($this->_customerTokenTable, array(
|
||||
'expiry' => $expiry
|
||||
), $where);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新登录数据的会员ID
|
||||
*
|
||||
* @param String $token <p>用户登录时生成的token字符串</p>
|
||||
* @param Int $customer_id <p>已登录会员的ID</p>
|
||||
*
|
||||
*/
|
||||
public function updateTokenCustomerId($token, $customer_id)
|
||||
{
|
||||
|
||||
$where = $this->db->quoteInto('token = ?', $token);
|
||||
|
||||
$this->db->update($this->_customerTokenTable, array(
|
||||
'customer_id' => $customer_id
|
||||
), $where);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除指定的登录数据
|
||||
*
|
||||
* @param String $token <p>用户登录时生成的token字符串</p>
|
||||
*
|
||||
*/
|
||||
public function deleteToken($token)
|
||||
{
|
||||
|
||||
$where = $this->db->quoteInto('token = ?', $token);
|
||||
|
||||
$this->db->delete($this->_customerTokenTable, $where);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
*获取用户的登录数据
|
||||
*
|
||||
* @param String $token <p>用户登录时生成的token字符串</p>
|
||||
*
|
||||
* @return Array 一个包含登录数据的数组
|
||||
*/
|
||||
public function getToken($token)
|
||||
{
|
||||
|
||||
return $this->db->fetchRow('SELECT * FROM ' . $this->_customerTokenTable
|
||||
. ' WHERE token = ' . $this->db->quote($token));
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 较验Token是否有效
|
||||
*
|
||||
* @param String $token <p>用户登录时生成的token字符串</p>
|
||||
*
|
||||
* @return Mix 返回一组会员数据代表验证通过，<b>FALSE</b>代表验证不通过。
|
||||
*/
|
||||
public function checkToken($token)
|
||||
{
|
||||
|
||||
$result = $this->db->fetchRow('SELECT * FROM ' . $this->_customerTokenTable
|
||||
. ' WHERE token = ' . $this->db->quote($token));
|
||||
|
||||
if ($result) {
|
||||
if ($result['expiry'] > time()) {
|
||||
return $result;
|
||||
} else {
|
||||
$this->deleteToken($token);
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
} else {
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
*获取用户的登录数据
|
||||
*
|
||||
* @param String $customer_id <p>用户ID</p>
|
||||
* @param String $current_token <p>用户当前有效的token字符串</p>
|
||||
* @param String $platform <p>登录平台ID</p>
|
||||
* @return Boolean 一个包含登录数据的数组
|
||||
*/
|
||||
public function clearDuplicateToken($customer_id, $current_token, $platform = 1)
|
||||
{
|
||||
$results = $this->db->fetchAll('SELECT * FROM ' . $this->_customerTokenTable
|
||||
. ' WHERE customer_id = ' . $this->db->quote($customer_id) . 'AND platform = ' . $this->db->quote($platform));
|
||||
|
||||
foreach ($results as $result) {
|
||||
if ($result['token'] != $current_token) {
|
||||
$this->updateTokenCustomerId($result['token'], 0);
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
public function getTokenByCustomerId($customer_id)
|
||||
{
|
||||
return $this->db->fetchOne('SELECT token FROM ' . $this->_customerTokenTable
|
||||
. ' WHERE customer_id = ' . $this->db->quote($customer_id));
|
||||
}
|
||||
|
||||
public function getTokenDataByCustomerId($customer_id)
|
||||
{
|
||||
$token_data = $this->db->fetchOne('SELECT data FROM ' . $this->_customerTokenTable
|
||||
. ' WHERE customer_id = ' . $this->db->quote($customer_id));
|
||||
|
||||
return json_decode($token_data, true);
|
||||
}
|
||||
}
|
||||
30
test/password_test.go
Normal file
30
test/password_test.go
Normal file
|
|
@ -0,0 +1,30 @@
|
|||
package vertmore_test
|
||||
|
||||
import (
|
||||
"crypto/sha1"
|
||||
"fmt"
|
||||
"log"
|
||||
"testing"
|
||||
|
||||
"github.com/iapologizewhenimwrong/Vestmore_GO/utils/auth"
|
||||
)
|
||||
|
||||
func TestPassword(t *testing.T) {
|
||||
|
||||
// customer := map[string]string{
|
||||
// "password": "486790f71aea661645b38e94c2734e454dd73a4d",
|
||||
// "salt": "b83192106fa45fef0d914c579848ebea",
|
||||
// "random_password": "709889",
|
||||
// }
|
||||
// password := "6601502.."
|
||||
password := "709889"
|
||||
pwdhash := fmt.Sprintf("%x", sha1.Sum([]byte(password)))
|
||||
|
||||
log.Println()
|
||||
|
||||
if !auth.CheckPassword("486790f71aea661645b38e94c2734e454dd73a4d", "b83192106fa45fef0d914c579848ebea", "709889", pwdhash) {
|
||||
fmt.Println("密码验证失败")
|
||||
} else {
|
||||
fmt.Println("密码验证成功")
|
||||
}
|
||||
}
|
||||
39
utils/auth/check_customer_password.go
Normal file
39
utils/auth/check_customer_password.go
Normal file
|
|
@ -0,0 +1,39 @@
|
|||
package auth
|
||||
|
||||
import (
|
||||
"crypto/md5"
|
||||
"crypto/sha1"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// sha1($customer['salt'] . md5($customer['salt'] . $password))
|
||||
func HashPassword(salt, password string) string {
|
||||
saltBytes := []byte(salt)
|
||||
passwordBytes := []byte(password)
|
||||
|
||||
// 计算 md5($customer['salt'] . $password)
|
||||
md5Hash := md5.Sum(append(saltBytes, passwordBytes...))
|
||||
|
||||
// 计算 $customer['salt'] . md5($customer['salt'] . $password)
|
||||
combined := append(saltBytes, []byte(fmt.Sprintf("%x", md5Hash))...)
|
||||
|
||||
// 计算 sha1($customer['salt'] . md5($customer['salt'] . $password))
|
||||
sha1Hash := sha1.Sum(combined)
|
||||
|
||||
return fmt.Sprintf("%x", sha1Hash)
|
||||
}
|
||||
|
||||
func CheckPassword(custPassword, salt, randomPassword string, password string) bool {
|
||||
|
||||
if custPassword == HashPassword(salt, password) {
|
||||
return true
|
||||
}
|
||||
|
||||
// 验证密码规则2
|
||||
if password == fmt.Sprintf("%x", sha1.Sum([]byte(randomPassword))) {
|
||||
return true
|
||||
}
|
||||
|
||||
// 如果两个条件都不满足，则返回false
|
||||
return false
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user