clock-test.c: do doesclrscrafterexit() handling the canonical way.

diff --git a/testcode/lib/clock-test.c b/testcode/lib/clock-test.c
index 0267668aa439bb9ed8d3dc1185838b5dfa100bec..9d420a5b0884248b2b4850180917b4ed74713a40 100644 (file)
--- a/testcode/lib/clock-test.c
+++ b/testcode/lib/clock-test.c
@@ -12,11 +12,6 @@
 #ifdef __CC65__
 #include <conio.h>
 #include <cc65.h>
-
-static void exitfn(void)
-{
-    if (doesclrscrafterexit()) cgetc();
-}
 #endif /* #ifdef __CC65__ */
 
 static void print_time(void)
@@ -42,7 +37,9 @@ int main(int argc, char **argv)
     static struct timespec new_time;
 
 #ifdef __CC65__
-    atexit(exitfn);
+    /* if DOS will automatically clear the screen after the program exits, wait for a keypress... */
+    if (doesclrscrafterexit())
+        atexit((void (*)(void))cgetc);
 #endif
 
     if (argc <= 1) {