From 79b870499beca4797e95f6ba5bc4d5bc4ece1f79 Mon Sep 17 00:00:00 2001
From: Yizhe Shen <brrr@live.ca>
Date: Mon, 10 Feb 2014 23:39:42 -0500
Subject: [PATCH] Added additional limits for "gamestats" parameters.

Added check to make sure the parameter is an integer between 4 and
var.MAX_PLAYERS.
---
 modules/wolfgame.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/modules/wolfgame.py b/modules/wolfgame.py
index 9e1c08a..d8f83f1 100644
--- a/modules/wolfgame.py
+++ b/modules/wolfgame.py
@@ -2891,13 +2891,14 @@ def game_stats(cli, nick, chan, rest):
         cli.msg(chan, var.get_game_totals())
         return
     
-    # Check that size is an integer.
-    if not rest.isdigit():
-        cli.notice(nick, "Please enter an integer.")
+    # Check for invalid input
+    rest = rest.strip()
+    if not rest.isdigit() or int(rest) > var.MAX_PLAYERS or int(rest) < 4:
+        cli.notice(nick, "Please enter an integer between {0} and {1}.".format(4, var.MAX_PLAYERS))
         return
     
     # Attempt to find game stats for the given game size.
-    size = int(rest.strip())
+    size = int(rest)
     msg = var.get_game_stats(size)
     if msg == "":
         cli.msg(chan, "No stats for \u0002{0}\u0002 player games.".format(size))