Quit confirmation fixes, phase 1

git-svn-id: svn://svn.code.sf.net/p/sc2/code/trunk@1210 8092fc87-c524-0410-9efc-e669fe64eaf9

sc2/src/sc2code/controls.h

@@ -64,6 +64,7 @@ void ResetKeyRepeat (void);
 BOOLEAN PauseGame (void);
 BOOLEAN DoConfirmExit (void);
 void TFB_Abort (void);
+BOOLEAN WaitAnyButtonOrQuit (BOOLEAN CheckSpecial);
 
 BATTLE_INPUT_STATE p1_combat_summary (void);
 BATTLE_INPUT_STATE p2_combat_summary (void);

sc2/src/sc2code/utils.c

@@ -214,3 +214,18 @@ PauseGame (void)
 	GLOBAL (CurrentActivity) &= ~CHECK_PAUSE;
 	return (TRUE);
 }
+
+// Waits for a new button to be pressed
+// and returns TRUE if Quit was selected
+BOOLEAN
+WaitAnyButtonOrQuit (BOOLEAN CheckSpecial)
+{
+	while (AnyButtonPress (TRUE))
+		TaskSwitch ();
+
+	while (!AnyButtonPress (TRUE) &&
+			!(GLOBAL (CurrentActivity) & CHECK_ABORT))
+		TaskSwitch ();
+
+	return (GLOBAL (CurrentActivity) & CHECK_ABORT) != 0;
+}