Additional test for color output.

Color output is only enabled if stderr is a tty, and the
environment variable TERM is not empty or dumb.
Bug 18004

3 files changed, 33 insertions, 26 deletions

diff --git a/driver/Commandline.ml b/driver/Commandline.ml
index 0a2c8fca..c5c2b82c 100644
--- a/driver/Commandline.ml
+++ b/driver/Commandline.ml
@@ -31,6 +31,8 @@ type action =
   | Self of (string -> unit)
   | String of (string -> unit)
   | Integer of (int -> unit)
+  | Ignore
+  | Unit of (unit -> unit)
 
 let match_pattern text = function
   | Exact s ->
@@ -95,6 +97,8 @@ let parse_array spec argv first last =
           end else begin
             eprintf "Option `%s' expects an argument\n" s; exit 2
           end
+      | Some (Ignore) -> parse (i+1)
+      | Some (Unit f) -> f (); parse (i+1)
     end
   in parse first
 
diff --git a/driver/Commandline.mli b/driver/Commandline.mli
index 79786678..5f9d8704 100644
--- a/driver/Commandline.mli
+++ b/driver/Commandline.mli
@@ -33,6 +33,8 @@ type action =
   | Self of (string -> unit)    (** call the function with the matched string *)
   | String of (string -> unit)  (** read next arg as a string, call function *)
   | Integer of (int -> unit)    (** read next arg as an int, call function *)
+  | Ignore                      (** ignore the arg *)
+  | Unit of (unit -> unit)      (** call the function with unit as argument *)
 
 val parse_cmdline: (pattern * action) list -> unit
 
diff --git a/driver/Driver.ml b/driver/Driver.ml
index 038bd423..78baed47 100644
--- a/driver/Driver.ml
+++ b/driver/Driver.ml
@@ -348,10 +348,10 @@ General options:\n\
 \  -random        Randomize execution order\n\
 \  -all           Simulate all possible execution orders\n"
 
-let print_usage_and_exit _ =
+let print_usage_and_exit () =
   printf "%s" usage_string; exit 0
 
-let print_version_and_exit _ =
+let print_version_and_exit () =
   printf "%s" version_string; exit 0
 
 let language_support_options = [
@@ -364,8 +364,8 @@ let optimization_options = [
   option_ftailcalls; option_fconstprop; option_fcse; option_fredundancy
 ]
 
-let set_all opts = List.iter (fun r -> r := true) opts
-let unset_all opts = List.iter (fun r -> r := false) opts
+let set_all opts () = List.iter (fun r -> r := true) opts
+let unset_all opts () = List.iter (fun r -> r := false) opts
 
 let num_source_files = ref 0
 
@@ -374,13 +374,16 @@ let num_input_files = ref 0
 let cmdline_actions =
   let f_opt name ref =
     [Exact("-f" ^ name), Set ref; Exact("-fno-" ^ name), Unset ref] in
+  let dwarf_version version () =
+    option_g:=true;
+    option_gdwarf := version in
   [
 (* Getting help *)
-  Exact "-help", Self print_usage_and_exit;
-  Exact "--help", Self print_usage_and_exit;
+  Exact "-help", Unit print_usage_and_exit;
+  Exact "--help", Unit print_usage_and_exit;
 (* Getting version info *)
-  Exact "-version", Self print_version_and_exit;
-  Exact "--version", Self print_version_and_exit;
+  Exact "-version", Unit print_version_and_exit;
+  Exact "--version", Unit print_version_and_exit;
 (* Processing options *)
   Exact "-c", Set option_c;
   Exact "-E", Set option_E;
@@ -391,17 +394,15 @@ let cmdline_actions =
   (* Preprocessing options *)
     @ prepro_actions @
 (* Language support options -- more below *)
- [ Exact "-fall", Self (fun _ -> set_all language_support_options);
-  Exact "-fnone", Self (fun _ -> unset_all language_support_options);
+ [ Exact "-fall", Unit (set_all language_support_options);
+  Exact "-fnone", Unit (unset_all language_support_options);
 (* Debugging options *)
-  Exact "-g", Self (fun s -> option_g := true;
-    option_gdwarf := 3);] @
+  Exact "-g", Unit (dwarf_version 3);] @
   (if gnu_system then
- [ Exact "-gdwarf-2", Self (fun s -> option_g:=true;
-    option_gdwarf := 2);
-  Exact "-gdwarf-3", Self (fun s -> option_g := true;
-    option_gdwarf := 3);] else []) @
- [ Exact "-frename-static", Self (fun s -> option_rename_static:= true);
+    [ Exact "-gdwarf-2", Unit (dwarf_version 2);
+      Exact "-gdwarf-3", Unit (dwarf_version 3);]
+  else []) @
+ [ Exact "-frename-static", Set option_rename_static;
    Exact "-gdepth", Integer (fun n -> if n = 0 || n <0 then begin
      option_g := false
    end else begin
@@ -409,9 +410,9 @@ let cmdline_actions =
      option_gdepth := if n > 3 then 3 else n
    end);
 (* Code generation options -- more below *)
-  Exact "-O0", Self (fun _ -> unset_all optimization_options);
-  Exact "-O", Self (fun _ -> set_all optimization_options);
-  _Regexp "-O[123]$", Self (fun _ -> set_all optimization_options);
+  Exact "-O0", Unit (unset_all optimization_options);
+  Exact "-O", Unit (set_all optimization_options);
+  _Regexp "-O[123]$", Unit (set_all optimization_options);
   Exact "-Os", Set option_Osize;
   Exact "-fsmall-data", Integer(fun n -> option_small_data := n);
   Exact "-fsmall-const", Integer(fun n -> option_small_const := n);
@@ -420,8 +421,8 @@ let cmdline_actions =
   Exact "-falign-branch-targets", Integer(fun n -> option_falignbranchtargets := n);
   Exact "-falign-cond-branches", Integer(fun n -> option_faligncondbranchs := n);
 (* Target processor options *)
-  Exact "-conf", String (fun _ -> ()); (* Ignore option since it is already handled *)
-  Exact "-target", String (fun _ -> ());] @ (* Ignore option since it is already handled *)
+  Exact "-conf", Ignore; (* Ignore option since it is already handled *)
+  Exact "-target", Ignore;] @ (* Ignore option since it is already handled *)
   (if Configuration.arch = "arm" then
     [ Exact "-mthumb", Set option_mthumb;
       Exact "-marm", Unset option_mthumb; ]
@@ -452,10 +453,10 @@ let cmdline_actions =
   Cerrors.warning_options @
 (* Interpreter mode *)
  [ Exact "-interp", Set option_interp;
-  Exact "-quiet", Self (fun _ -> Interp.trace := 0);
-  Exact "-trace", Self (fun _ -> Interp.trace := 2);
-  Exact "-random", Self (fun _ -> Interp.mode := Interp.Random);
-  Exact "-all", Self (fun _ -> Interp.mode := Interp.All)
+  Exact "-quiet", Unit (fun () -> Interp.trace := 0);
+  Exact "-trace", Unit (fun () -> Interp.trace := 2);
+  Exact "-random", Unit (fun () -> Interp.mode := Interp.Random);
+  Exact "-all", Unit (fun () -> Interp.mode := Interp.All)
  ]
 (* -f options: come in -f and -fno- variants *)
 (* Language support options *)