session_auth.js 1.1 KB

12345678910111213141516171819202122232425262728293031323334
  1. 'use strict';
  2. // 加密类
  3. const crypto = require('crypto');
  4. module.exports = options => {
  5. /**
  6. * session判断中间件
  7. *
  8. * @param {function} next - 中间件继续执行的方法
  9. * @return {void}
  10. */
  11. return function* sessionAuth(next) {
  12. try {
  13. // 判断session
  14. const managerSession = this.session.managerSession;
  15. if (managerSession === undefined) {
  16. throw '不存在session';
  17. }
  18. // 校验session
  19. if (managerSession.username === undefined || managerSession.loginTime === undefined) {
  20. throw '用户数据不完整';
  21. }
  22. // 校验session
  23. const sessionToken = crypto.createHmac('sha1', managerSession.loginTime + '')
  24. .update(managerSession.username).digest().toString('base64');
  25. if (sessionToken !== managerSession.sessionToken) {
  26. throw 'session数据错误';
  27. }
  28. } catch (error) {
  29. return this.redirect('/');
  30. }
  31. yield next;
  32. };
  33. };