86 lines
2.4 KiB
JavaScript
86 lines
2.4 KiB
JavaScript
const crypto = require('crypto')
|
||
const createConfig = require('uni-config-center')
|
||
const { verifyHttpInfo } = require('uni-cloud-s2s')
|
||
|
||
const { ERROR } = require('../common/error')
|
||
const s2sConfig = createConfig({
|
||
pluginId: 'uni-cloud-s2s'
|
||
})
|
||
const needSignFunctions = new Set([
|
||
'externalRegister',
|
||
'externalLogin',
|
||
'updateUserInfoByExternal'
|
||
])
|
||
|
||
module.exports = function () {
|
||
const methodName = this.getMethodName()
|
||
const { source } = this.getUniversalClientInfo()
|
||
|
||
// 指定接口需要鉴权
|
||
if (!needSignFunctions.has(methodName)) return
|
||
|
||
// 非 HTTP 方式请求拒绝访问
|
||
if (source !== 'http') {
|
||
throw {
|
||
errCode: ERROR.ILLEGAL_REQUEST
|
||
}
|
||
}
|
||
|
||
// 支持 uni-cloud-s2s 验证请求
|
||
if (s2sConfig.hasFile('config.json')) {
|
||
try {
|
||
if (!verifyHttpInfo(this.getHttpInfo())) {
|
||
throw {
|
||
errCode: ERROR.ILLEGAL_REQUEST
|
||
}
|
||
}
|
||
} catch (e) {
|
||
if (e.errSubject === 'uni-cloud-s2s') {
|
||
throw {
|
||
errCode: ERROR.ILLEGAL_REQUEST,
|
||
errMsg: e.errMsg
|
||
}
|
||
}
|
||
throw e
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
if (!this.config.requestAuthSecret || typeof this.config.requestAuthSecret !== 'string') {
|
||
throw {
|
||
errCode: ERROR.CONFIG_FIELD_REQUIRED,
|
||
errMsgValue: {
|
||
field: 'requestAuthSecret'
|
||
}
|
||
}
|
||
}
|
||
|
||
const timeout = 20 * 1000 // 请求超过20秒不能再请求，防止重放攻击
|
||
const { headers, body: _body } = this.getHttpInfo()
|
||
const { 'uni-id-nonce': nonce, 'uni-id-timestamp': timestamp, 'uni-id-signature': signature } = headers
|
||
const body = JSON.parse(_body).params || {}
|
||
const bodyStr = Object.keys(body)
|
||
.sort()
|
||
.filter(item => typeof body[item] !== 'object')
|
||
.map(item => `${item}=${body[item]}`)
|
||
.join('&')
|
||
|
||
if (isNaN(Number(timestamp)) || (Number(timestamp) + timeout) < Date.now()) {
|
||
console.error('[timestamp error], timestamp:', timestamp, 'timeout:', timeout)
|
||
|
||
throw {
|
||
errCode: ERROR.ILLEGAL_REQUEST
|
||
}
|
||
}
|
||
|
||
const reSignature = crypto.createHmac('sha256', `${this.config.requestAuthSecret + nonce}`).update(`${timestamp}${bodyStr}`).digest('hex')
|
||
|
||
if (signature !== reSignature.toUpperCase()) {
|
||
console.error('[signature error], signature:', signature, 'reSignature:', reSignature.toUpperCase(), 'requestAuthSecret:', this.config.requestAuthSecret)
|
||
throw {
|
||
errCode: ERROR.ILLEGAL_REQUEST
|
||
}
|
||
}
|
||
}
|