mm_session 1.5.9 → 1.6.1

package/lib/helper.js

@@ -47,28 +47,25 @@ Helper.prototype.aesDecode = function (text, key, iv) {
 
 
 /**
- * 生成session ID的密钥
- * @param {string} user_agent 用户代理字符串
- * @returns {string} 密钥
+ * 从key派生IV（固定派生，不依赖user_agent）
+ * @param {string} key 密钥
+ * @returns {string} IV（16字节）
  */
-Helper.prototype._getSecret = function (user_agent = 'mm') {
-  // md5方法由mm_expand模块提供
-  return user_agent.md5().substring(0, 16);
+Helper.prototype._getIvFromKey = function (key) {
+  return key.md5().substring(0, 16);
 };
 
 /**
  * 生成session ID
  * @param {string} key 密钥
  * @param {string} ip 客户端IP
- * @param {string} user_agent 用户代理字符串
+ * @param {string} user_agent 用户代理字符串（不再用于IV，仅保留参数兼容）
  * @param {number} uuid_expire 过期时间（秒）
  * @returns {string} session ID
  */
 Helper.prototype.genId = function (key, ip, user_agent = 'mm', uuid_expire = 604800) {
-  // 生成session ID的初始化向量
-  var iv = this._getSecret(user_agent);
+  var iv = this._getIvFromKey(key);
   var end_time = new Date().addSeconds(uuid_expire).stamp();
-  // 加密
   var uuid = this.aesEncode(ip + '_' + end_time, key, iv);
   return uuid;
 };
@@ -77,20 +74,19 @@ Helper.prototype.genId = function (key, ip, user_agent = 'mm', uuid_expire = 604
  * 解析session ID
  * @param {string} key 密钥
  * @param {string} uuid session ID
- * @param {string} user_agent 用户代理字符串
+ * @param {string} user_agent 用户代理字符串（不再用于IV，仅保留参数兼容）
  * @returns {object} 解析结果
  */
 Helper.prototype.parseId = function (key, uuid, user_agent = 'mm') {
-  // 生成session ID的初始化向量
-  var iv = this._getSecret(user_agent);
-  // 解密
+  var iv = this._getIvFromKey(key);
   var dec = this.aesDecode(uuid, key, iv);
-  // 解析IP和过期时间
-  var ip = dec.split('_')[0];
-  var end_time = dec.split('_')[1];
+  var parts = dec.split('_');
+  if (parts.length !== 2) {
+    return null;
+  }
   return {
-    ip,
-    end_time
+    ip: parts[0],
+    end_time: Number(parts[1])
   };
 };
 

package/lib/session.js

@@ -18,20 +18,14 @@ class Session {
    * 默认配置
    */
   static config = {
-    // 密钥
     encrypt_key: $.config?.encrypt_key ? $.config.encrypt_key : 'mm'.md5().substring(0, 16),
-    // session存储在缓存中的key前缀，默认'mm:uuid'
     key_prefix: $.dict.session_id || 'mm:uuid',
-    // session ID存储在cookie中的字段名，默认'mm:uuid'
     cookie_token: $.dict.cookie_token || 'mm:uuid',
-    // HTTP请求头中用于传递session ID的字段名，默认'x-auth-token'
     header_token: $.dict.header_token || 'x-auth-token',
-    // session ID过期时间，单位秒，默认7天
     uuid_expire: 60 * 60 * 24 * 7,
-    // session过期时间，单位秒，默认2小时
     max_age: 60 * 60 * 2,
-    // 过期token处理选项：'reject'（拒绝访问）或'allow'（允许访问缓存）
     expired_handling: 'reject',
+    options: { path: '/' },
   };
 
   /**
@@ -89,30 +83,23 @@ Session.prototype.middleware = function () {
  * @returns {object} session对象
  */
 Session.prototype.get = async function (ctx) {
-  // 获取协议头信息
   let { ip, user_agent } = this._getHeaderInfo(ctx);
-  // 获取session ID
   let session_id = this.getId(ctx);
 
-  // 验证session ID标识的有效性
   if (session_id) {
     let is_valid = await this._checkSessionId(session_id, ip, user_agent);
     if (!is_valid) {
-      // token验证失败，返回null创建新session
       session_id = null;
     }
   }
 
-  // 获取session对象
   let session = await this._get(session_id);
 
   if (!session.uuid) {
     session.uuid = this._genId(ip, user_agent);
   }
-  // 创建session代理对象
+  this._bindSessionMethods(session, ctx);
   ctx.session = this._create(session);
-  // 追加方法
-  this._addMethod(ctx);
   return ctx.session;
 };
 
@@ -122,9 +109,13 @@ Session.prototype.get = async function (ctx) {
  * @param {Function} next 下一个中间件
  */
 Session.prototype._handle = async function (ctx, next) {
-  let session_id = await this.get(ctx);
+  let session_id = this.getId(ctx);
+  await this.get(ctx);
   await next();
   await this._save(ctx, session_id);
+  if (ctx.session) {
+    this._setCookie(ctx, ctx.session.uuid);
+  }
 };
 
 /**
@@ -133,24 +124,21 @@ Session.prototype._handle = async function (ctx, next) {
  * @returns {string} 客户端IP地址
  */
 Session.prototype._getClientIP = function (ctx) {
-  // 优先从ctx.ip获取
+  var ip;
   if (ctx.ip) {
-    return ctx.ip;
-  }
-  
-  // 从X-Forwarded-For头部获取（代理服务器场景）
-  if (ctx.headers['x-forwarded-for']) {
-    const ips = ctx.headers['x-forwarded-for'].split(',');
-    return ips[0].trim();
+    ip = ctx.ip;
+  } else if (ctx.headers['x-forwarded-for']) {
+    ip = ctx.headers['x-forwarded-for'].split(',')[0].trim();
+  } else if (ctx.headers['x-real-ip']) {
+    ip = ctx.headers['x-real-ip'];
+  } else {
+    ip = '127.0.0.1';
   }
-  
-  // 从X-Real-IP头部获取
-  if (ctx.headers['x-real-ip']) {
-    return ctx.headers['x-real-ip'];
+  // 归一化本地回环地址，避免 IPv6/IPv4 格式差异导致校验失败
+  if (ip === '::1' || ip === '::ffff:127.0.0.1' || ip === 'localhost') {
+    ip = '127.0.0.1';
   }
-  
-  // 默认返回本地IP
-  return '127.0.0.1';
+  return ip;
 };
 
 /**
@@ -255,27 +243,29 @@ Session.prototype._create = function (session) {
 };
 
 /**
- * 添加session方法
+ * 给session目标对象绑定方法（在创建Proxy前调用，避免触发set陷阱）
+ * @param {object} session session目标对象
  * @param {object} ctx HTTP上下文
  */
-Session.prototype._addMethod = function (ctx) {
+Session.prototype._bindSessionMethods = function (session, ctx) {
+  var self = this;
+
   /**
    * 刷新session
    */
-  ctx.session.refresh = () => {
-    // 刷新标记在外部处理
-    ctx.session._modified = true;
+  session.refresh = function () {
+    this._modified = true;
   };
 
   /**
    * 获取session ID
    * @returns {string} session ID
    */
-  ctx.session.getId = () => {
-    if (!ctx.session.uuid) {
-      ctx.session.uuid = this.genId(ctx);
+  session.getId = function () {
+    if (!this.uuid) {
+      this.uuid = self.genId(ctx);
     }
-    return ctx.session.uuid;
+    return this.uuid;
   };
 };
 
@@ -340,31 +330,18 @@ Session.prototype._handleCookie = async function (ctx, session_id) {
  * @returns {Promise<void>} 保存结果
  */
 Session.prototype._save = async function (ctx, session_id) {
-  // 如果session不存在，直接删除session
   if (!ctx.session) {
     await this._delSession(ctx, session_id);
     return;
   }
 
-  // 如果session被修改过，需要保存
-  if (ctx.session._modified) {
+  if (ctx.session._is_new || ctx.session._modified) {
     await this._handleCookie(ctx, session_id);
 
-    // 提取session数据
     const session_to_save = this._extractSessionData(ctx);
 
-    // 保存内部属性（_is_new和_modified）
-    if (ctx.session._is_new !== undefined) {
-      session_to_save._is_new = ctx.session._is_new;
-    }
-    if (ctx.session._modified !== undefined) {
-      session_to_save._modified = ctx.session._modified;
-    }
-
-    // 清理session
     this._cleanup(ctx);
 
-    // 保存session到存储
     return await this._store.set(ctx.session.uuid, session_to_save, this.config.max_age || 7200);
   }
 };
@@ -403,7 +380,7 @@ Session.prototype.save = async function (ctx, session_id_or_obj) {
  */
 Session.prototype._delSession = async function (ctx, session_id) {
   ctx.cookies.set(this.config.cookie_token, null);
-  this._store.del(session_id).catch();
+  await this._store.del(session_id);
 };
 
 /**
@@ -422,10 +399,9 @@ Session.prototype._shouldSetCookie = function (session_id, sid) {
  * @param {string} sid session ID
  */
 Session.prototype._setCookie = function (ctx, sid) {
-  let cg = { ...this.config.config };
-  if (cg.max_age) {
-    cg.maxAge = cg.max_age * 1000;
-    delete cg.max_age;
+  let cg = { ...this.config.options };
+  if (this.config.max_age && !cg.maxAge) {
+    cg.maxAge = this.config.max_age * 1000;
   }
   ctx.cookies.set(this.config.cookie_token, sid, cg);
 };
@@ -435,7 +411,7 @@ Session.prototype._setCookie = function (ctx, sid) {
  * @param {number} end_time token过期时间戳
  * @returns {boolean} 是否过期
  */
-Session.prototype._verifyExpiration = async function (end_time) {
+Session.prototype._verifyExpiration = function (end_time) {
   var current_time = Date.parse(new Date()) / 1000;
   return current_time > end_time;
 };
@@ -449,22 +425,18 @@ Session.prototype._verifyExpiration = async function (end_time) {
  */
 Session.prototype._checkSessionId = async function (session_id, client_ip, client_ua) {
   try {
-    // 解码session_id
     var data = this._helper.parseId(this.config.encrypt_key, session_id, client_ua);
     if (!data) return false;
     let { ip, end_time } = data;
-    // 验证IP
     if (ip !== client_ip) return false;
 
-    // 验证时效性
-    var is_expired = await this._verifyExpiration(end_time);
+    var is_expired = this._verifyExpiration(end_time);
     if (is_expired) {
       if (this.config.expired_handling !== 'allow') {
         return false;
       }
     }
 
-    // 所有验证通过
     return true;
   } catch (err) {
     $.log.debug('Session ID验证错误:', err);

package/lib/store.js

@@ -27,7 +27,7 @@ class Store {
  * @returns {object|null} session对象或null
  */
 Store.prototype.get = async function (uuid) {
-  if (!$.cache.has(this.key + uuid)) return undefined;
+  if (!$.cache.has(this.key + uuid)) return null;
   // 我们正在解码数据来自我们的srote, 我们假定它是在储存之前
   var val = await $.cache.get(this.key + uuid);
   if (val) {
@@ -65,8 +65,8 @@ Store.prototype.set = async function (uuid, session, max_age = 7200) {
  * 删除缓存
  * @param {string} uuid 客户端唯一ID
  */
-Store.prototype.del = function (uuid) {
-  $.cache.del(this.key + uuid);
+Store.prototype.del = async function (uuid) {
+  await $.cache.del(this.key + uuid);
 };
 
 module.exports = {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "mm_session",
-    "version": "1.5.9",
+    "version": "1.6.1",
     "description": "这是超级美眉session函数模块，用于web服务端session缓存",
     "main": "index.js",
     "scripts": {
@@ -30,10 +30,10 @@
     },
     "homepage": "https://gitee.com/qiuwenwu91/mm_session#readme",
     "dependencies": {
-        "mm_cache": "^1.4.9"
+        "mm_cache": "^1.5.0"
     },
     "devDependencies": {
-        "eslint-plugin-jsdoc": "^63.0.8",
+        "eslint-plugin-jsdoc": "^63.0.10",
         "mm_eslint": "^1.7.5"
     }
 }