mm_os 3.2.9 → 3.3.1

package/middleware/rate_limit/index.js

@@ -1,112 +0,0 @@
-/**
- * API速率限制中间件
- * 用于防止DoS攻击，限制客户端在一定时间内的请求频率
- * @param {Object} server 服务实例
- * @param {Object} config 配置参数
- */
-module.exports = function(server, config) {
-    // 初始化速率限制配置
-    const cg = {
-        // 默认配置
-        windowMs: 60,      // 时间窗口，调整为1分钟（更精细的限制）
-        maxRequests: 5000,         // 每个时间窗口内的最大请求数，从1000提高到5000
-        message: '请求过于频繁，请稍后再试', // 超过限制时的提示信息
-        statusCode: 429,          // 超过限制时的HTTP状态码
-        // 白名单路径，这些路径不受速率限制
-        whitelistPaths: [],
-        // 合并用户配置
-        ...(config && config.rate_limit ? config.rate_limit : {})
-    };
-    
-    // 生成基于IP的唯一标识符
-    function getClientKey(ctx) {
-        // 优先使用X-Forwarded-For头部（如果有代理的话）
-        const forwardedFor = ctx.headers['x-forwarded-for'];
-        if (forwardedFor) {
-            // 通常格式为：X-Forwarded-For: client, proxy1, proxy2
-            return forwardedFor.split(',')[0].trim();
-        }
-        // 直接使用IP地址
-        return ctx.ip;
-    }
-    
-    // 使用Redis存储请求计数
-    async function incrementRequestWithRedis(clientKey) {
-        try {
-            // 检查Redis是否可用
-            if ($.cache && typeof $.cache.addInt === 'function') {
-                const key = `rate_limit:${clientKey}`;
-                
-                // 使用addInt方法增加计数（mm_redis包提供的方法）
-                const count = await $.cache.addInt(key, 1);
-                
-                // 设置过期时间
-                await $.cache.ttl(key, Math.ceil(cg.windowMs));
-                
-                return count || 0;
-            }
-        } catch (error) {
-            $.log.error('Redis速率限制失败:', error);
-        }
-        
-        // Redis不可用时返回null，将使用内存存储
-        return null;
-    }
-
-    // 中间件主逻辑
-    server.use(async (ctx, next) => {
-        try {
-            // 跳过静态文件和favicon请求
-            if (ctx.path === '/favicon.ico' || ctx.path.startsWith('/static/')) {
-                await next();
-                return;
-            }
-            
-            // 检查是否是白名单路径
-            if (cg.whitelistPaths && cg.whitelistPaths.some(path => 
-                ctx.path === path || ctx.path.startsWith(path + '/'))) {
-                await next();
-                return;
-            }
-            
-            // 获取客户端唯一标识符
-            const clientKey = getClientKey(ctx);
-            
-            let requestCount = await incrementRequestWithRedis(clientKey);
-            
-            // 设置响应头部，告知客户端当前的限制状态
-            ctx.set('X-RateLimit-Limit', cg.maxRequests);
-            ctx.set('X-RateLimit-Remaining', Math.max(0, cg.maxRequests - requestCount));
-            
-            // 检查是否超过限制
-            if (requestCount > cg.maxRequests) {
-                $.log.warn(`API速率限制触发: ${clientKey} 请求 ${ctx.path} 次数过多`);
-                
-                ctx.status = cg.statusCode;
-                ctx.body = {
-                    code: cg.statusCode,
-                    msg: cg.message
-                };
-                
-                // 记录超过限制的请求
-                if ($.log && $.log.warn) {
-                    $.log.warn(`速率限制触发: IP=${clientKey}, Path=${ctx.path}, Method=${ctx.method}`);
-                }
-                
-                return;
-            }
-            
-            // 继续处理请求
-            await next();
-        } catch (error) {
-            $.log.error('速率限制中间件错误:', error);
-            // 确保请求可以继续处理
-            await next();
-        }
-    });
-    
-    // 记录中间件初始化信息
-    $.log.info(`速率限制中间件已加载: ${cg.maxRequests}请求/${cg.windowMs}秒`);
-    
-    return server;
-};

package/middleware/rate_limit/middleware.json

@@ -1,10 +0,0 @@
-{
-  "name": "rate_limit",
-  "title": "API速率限制",
-  "description": "限制客户端在一定时间内的请求频率，防止DoS攻击",
-  "version": "1.0",
-  "type": "web",
-  "process_type": "common_before",
-  "sort": 50,
-  "script": "index.js"
-}

package/middleware/waf_ip/index.js

@@ -1,168 +0,0 @@
-const {
-	exec
-} = require('child_process');
-const platform = require('os').platform();
-
-/**
- * 获取客户端IP
- * @param {Object} req 请求对象
- * @returns {String} 返回真实IP
- */
-function getClientIP(req) {
-	var ip = req.headers['x-forwarded-for'] || req.headers['X-Forwarded-For'] || req.headers['x-real-ip'] ||
-		req.connection.remoteAddress ||
-		req.socket.remoteAddress ||
-		req.connection.socket.remoteAddress;
-	if (ip && ip.split(',').length > 0) {
-		ip = ip.split(',')[0]; // 取第一个IP地址
-	}
-	return ip;
-};
-
-/**
- * 设置黑名单
- * @param {String} ip IP地址
- */
-function setting_blacklist(ip) {
-	var cmd;
-	if (platform == "win32") {
-		// window 系统
-		cmd =
-			`netsh advfirewall firewall add rule name="Blacklist ${ip}" dir=in action=block remoteip="${ip}" protocol=any`
-	} else {
-		// linux 系统
-		cmd = `sudo iptables -A INPUT -s ${ip} -j DROP`;
-	}
-
-	exec(cmd, (error, stdout, stderr) => {
-		if (error) {
-			console.error(`执行的错误: ${error}`);
-			return;
-		}
-		$.log.info(`加入黑名单: ${ip}`);
-		if (stderr) {
-			console.error(`标准错误输出: ${stderr}`);
-		}
-	});
-}
-
-/**
- * IP防火墙
- * @param {Object} server 服务
- * @param {Object} config 配置参数
- */
-module.exports = function(server, config) {
-	var limit = config.request_limit || 0;
-	var duration = config.request_duration || 0;
-	var block = config.request_block || false;
-
-	if (limit && duration) {
-		/* WAF（web防火墙） */
-		server.use(async (ctx, next) => {
-			try {
-				var pass = true;
-				// 获取IP
-				var ip = getClientIP(ctx.req);
-				var num = 1;
-				var now = new Date();
-				var date = now.toStr('yyyy-MM-dd');
-				var time;
-				var json;
-				try {
-					var str = await $.cache.get("ip_" + ip);
-					if (str) {
-						if (typeof(str) === "string") {
-							try {
-								json = JSON.parse(str);
-							} catch (jsonError) {
-								$.log.error('WAF IP中间件JSON解析错误:', jsonError);
-								json = null;
-							}
-						} else {
-							json = str;
-						}
-						if (json) {
-							try {
-								if (json.date !== date) {
-									num = 1;
-								} else {
-									// 判断时间间隔是否在范围外
-									// 安全地处理json.time，无论它是字符串还是对象
-									if (typeof json.time === 'string') {
-										// 如果json.time是字符串，尝试解析它
-										const savedTime = new Date(json.time);
-										if (!isNaN(savedTime.getTime()) && (now - savedTime) > duration) {
-											num = 1;
-										} else {
-											// 如果是在周期内，访问次数+1，并判断是否超出上限
-											num = json.num + 1;
-											if (num > limit) {
-												// 超出上限禁止访问，并加入黑名单
-												pass = false;
-												if (block) {
-													setting_blacklist(ip);
-												}
-											}
-										}
-									} else if (json.time && json.time.toTime && typeof json.time.toTime().interval === 'function') {
-										// 原有逻辑，处理对象类型的时间
-										if (json.time.toTime().interval(now) > duration) {
-											num = 1;
-										} else {
-											// 如果是在周期内，访问次数+1，并判断是否超出上限
-											num = json.num + 1;
-											if (num > limit) {
-												// 超出上限禁止访问，并加入黑名单
-												pass = false;
-												if (block) {
-													setting_blacklist(ip);
-												}
-											}
-										}
-									} else {
-										// 默认情况：如果时间格式不正确，重置计数
-										num = 1;
-									}
-								}
-							} catch (timeError) {
-								$.log.error('WAF IP中间件时间处理错误:', timeError);
-								// 出错时重置计数以保证安全
-								num = 1;
-							}
-						}
-					}
-				} catch (cacheError) {
-					$.log.error('WAF IP中间件缓存操作错误:', cacheError);
-					// 缓存出错时，默认允许请求通过
-				}
-				if (!time) {
-					time = now.toStr('yyyy-MM-dd hh:mm:ss');
-				}
-				if (pass) {
-					try {
-						await $.cache.set("ip_" + ip, JSON.stringify({
-							date,
-							time,
-							num
-						}), duration);
-					} catch (setCacheError) {
-						$.log.error('WAF IP中间件缓存设置错误:', setCacheError);
-						// 缓存设置失败不影响请求处理
-					}
-					ctx.request.ip = ip;
-					ctx.ip = ip;
-					await next();
-				} else {
-					ctx.status = 429;
-					ctx.body = '请求频率过高，请稍后再试。';
-				}
-			} catch (error) {
-				$.log.error('WAF IP中间件错误:', error);
-				// 出错时默认允许请求通过
-				ctx.ip = getClientIP(ctx.req);
-				await next();
-			}
-		});
-	}
-	return server;
-};

package/middleware/waf_ip/middleware.json

@@ -1,10 +0,0 @@
-{
-	"name": "web_waf_ip",
-	"title": "IP防火墙",
-	"description": "用于防止DOS攻击",
-	"version": "1.0",
-	"type": "web",
-	"process_type": "common_before",
-	"sort": 10,
-	"state": 0
-}

package/nodemon.json

@@ -1,31 +0,0 @@
-{
-	"restartable": "rs",
-	"ignore": [
-		".git",
-		".svn",
-		"cache",
-		"log",
-		"node_modules/**/node_modules",
-		"**/cache/**",
-		"game/data",
-		"**/conf.json",
-		"*/static/*/*.json",
-		"*/static/*.js"
-	],
-	"verbose": true,
-	"execMap": {
-		"": "node",
-		"js": "node --harmony"
-	},
-	"watch": [
-		"demo",
-		"lib",
-		"core",
-		"test"
-	],
-	"env": {
-		"NODE_ENV": "development"
-	},
-	"ext": "js ts json",
-	"legacy-watch": false
-}

package/package.txt

@@ -1 +0,0 @@
-mm_check mm_excel mm_expand mm_html mm_koa_proxy mm_logs mm_machine mm_mongodb mm_mqtt mm_mysql mm_redis mm_ret mm_session mm_statics mm_tpl mm_xml 

package/rps.bat

@@ -1,3 +0,0 @@
-@echo off
-loadtest http://localhost:5000/test/helloworld -t 20 -c 10 --keepalive --rps 1000
-pause

package/test.js

@@ -1,10 +0,0 @@
-var OS = require("./index.js");
-
-$.sql = $.mysql_admin('sys', __dirname);
-$.sql.config.database = "face";
-$.sql.config.password = "Asd159357";
-$.sql.open();
-
-var os = new OS();
-
-os.run();

package/tps.bat

@@ -1,3 +0,0 @@
-@echo off
-loadtest -n 10000 -s TLSv1_method http://localhost:5000/ws/tt
-pause

package/update.bat

@@ -1 +0,0 @@
-npm i mm_check mm_excel mm_expand mm_html mm_koa_proxy mm_logs mm_machine mm_mongodb mm_mqtt mm_mysql mm_redis mm_ret mm_session mm_statics mm_tpl mm_xml