summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Templates/TestDriver.cxx.in6
1 files changed, 5 insertions, 1 deletions
diff --git a/Templates/TestDriver.cxx.in b/Templates/TestDriver.cxx.in
index 41fccae..8c9f54e 100644
--- a/Templates/TestDriver.cxx.in
+++ b/Templates/TestDriver.cxx.in
@@ -70,7 +70,11 @@ int main(int ac, char *av[])
printf("To run a test, enter the test number: ");
fflush(stdout);
testNum = 0;
- scanf("%d", &testNum);
+ if( scanf("%d", &testNum) != 1 )
+ {
+ printf("Couldn't parse that input as a number\n");
+ return -1;
+ }
if (testNum >= NumTests)
{
printf("%3d is an invalid test number.\n", testNum);