From b85f2c625c3683e18f371fa1590c8c6109028fc4 Mon Sep 17 00:00:00 2001
From: David Anderson <dvander@alliedmods.net>
Date: Fri, 24 Aug 2007 23:56:57 +0000
Subject: [PATCH] fixed amb832

--HG--
extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%401374
---
 core/smn_menus.cpp | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/core/smn_menus.cpp b/core/smn_menus.cpp
index 59d2590b..80a87630 100644
--- a/core/smn_menus.cpp
+++ b/core/smn_menus.cpp
@@ -891,10 +891,18 @@ static cell_t SetMenuExitButton(IPluginContext *pContext, const cell_t *params)
 	}
 
 	unsigned int flags = menu->GetMenuOptionFlags();
-	flags |= MENUFLAG_BUTTON_EXIT;
+
+	if (params[2])
+	{
+		flags |= MENUFLAG_BUTTON_EXIT;
+	} else {
+		flags &= ~MENUFLAG_BUTTON_EXIT;
+	}
+
 	menu->SetMenuOptionFlags(flags);
-	flags = menu->GetMenuOptionFlags();
-	return ((flags & MENUFLAG_BUTTON_EXIT) == MENUFLAG_BUTTON_EXIT) ? 1 : 0;
+	unsigned int new_flags = menu->GetMenuOptionFlags();
+
+	return (flags == new_flags);
 }
 
 static cell_t SetMenuExitBackButton(IPluginContext *pContext, const cell_t *params)
@@ -909,7 +917,14 @@ static cell_t SetMenuExitBackButton(IPluginContext *pContext, const cell_t *para
 	}
 
 	unsigned int flags = menu->GetMenuOptionFlags();
-	flags |= MENUFLAG_BUTTON_EXITBACK;
+
+	if (params[2])
+	{
+		flags |= MENUFLAG_BUTTON_EXITBACK;
+	} else {
+		flags &= ~MENUFLAG_BUTTON_EXITBACK;
+	}
+
 	menu->SetMenuOptionFlags(flags);
 
 	return 1;