Change order of return values from getyx - fixes display of historical CPU.
[virt-top.git] / virt-top / virt_top.ml
index e2a7435..3990b55 100644 (file)
@@ -1,15 +1,35 @@
 (* 'top'-like tool for libvirt domains.
- * $Id: virt_top.ml,v 1.5 2007/08/30 13:52:40 rjones Exp $
- *)
+   (C) Copyright 2007-2009 Richard W.M. Jones, Red Hat Inc.
+   http://libvirt.org/
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*)
 
 open Printf
 open ExtList
 open Curses
 
+open Virt_top_gettext.Gettext
+open Virt_top_utils
+
 module C = Libvirt.Connect
 module D = Libvirt.Domain
 module N = Libvirt.Network
 
+let rcfile = ".virt-toprc"
+
 (* Hook for XML support (see virt_top_xml.ml). *)
 let parse_device_xml : (int -> [>`R] D.t -> string list * string list) ref =
   ref (
@@ -19,19 +39,21 @@ let parse_device_xml : (int -> [>`R] D.t -> string list * string list) ref =
 (* Hooks for CSV support (see virt_top_csv.ml). *)
 let csv_start : (string -> unit) ref =
   ref (
-    fun _ -> failwith "virt-top was compiled without support for CSV"
+    fun _ -> failwith (s_"virt-top was compiled without support for CSV files")
   )
 let csv_write : (string list -> unit) ref =
   ref (
     fun _ -> ()
   )
 
-(* Int64 operators for convenience. *)
-let (+^) = Int64.add
-let (-^) = Int64.sub
-let ( *^ ) = Int64.mul
-let (/^) = Int64.div
+(* Hook for calendar support (see virt_top_calendar.ml). *)
+let parse_date_time : (string -> float) ref =
+  ref (
+    fun _ ->
+      failwith (s_"virt-top was compiled without support for dates and times")
+  )
 
+(* Sort order. *)
 type sort_order =
   | DomainID | DomainName | Processor | Memory | Time
   | NetRX | NetTX | BlockRdRq | BlockWrRq
@@ -40,24 +62,62 @@ let all_sort_fields = [
   NetRX; NetTX; BlockRdRq; BlockWrRq
 ]
 let printable_sort_order = function
-  | Processor -> "%CPU"
-  | Memory -> "%MEM"
-  | Time -> "TIME (CPU time)"
-  | DomainID -> "Domain ID"
-  | DomainName -> "Domain name"
-  | NetRX -> "Net RX bytes"
-  | NetTX -> "Net TX bytes"
-  | BlockRdRq -> "Block read reqs"
-  | BlockWrRq -> "Block write reqs"
+  | Processor -> s_"%CPU"
+  | Memory -> s_"%MEM"
+  | Time -> s_"TIME (CPU time)"
+  | DomainID -> s_"Domain ID"
+  | DomainName -> s_"Domain name"
+  | NetRX -> s_"Net RX bytes"
+  | NetTX -> s_"Net TX bytes"
+  | BlockRdRq -> s_"Block read reqs"
+  | BlockWrRq -> s_"Block write reqs"
+let sort_order_of_cli = function
+  | "cpu" | "processor" -> Processor
+  | "mem" | "memory" -> Memory
+  | "time" -> Time
+  | "id" -> DomainID
+  | "name" -> DomainName
+  | "netrx" -> NetRX | "nettx" -> NetTX
+  | "blockrdrq" -> BlockRdRq | "blockwrrq" -> BlockWrRq
+  | str ->
+      failwithf (f_"%s: sort order should be: %s")
+       str "cpu|mem|time|id|name|netrx|nettx|blockrdrq|blockwrrq"
+let cli_of_sort_order = function
+  | Processor -> "cpu"
+  | Memory -> "mem"
+  | Time -> "time"
+  | DomainID -> "id"
+  | DomainName -> "name"
+  | NetRX -> "netrx"
+  | NetTX -> "nettx"
+  | BlockRdRq -> "blockrdrq"
+  | BlockWrRq -> "blockwrrq"
 
 (* Current major display mode: TaskDisplay is the normal display. *)
 type display = TaskDisplay | PCPUDisplay | BlockDisplay | NetDisplay
 
+let display_of_cli = function
+  | "task" -> TaskDisplay
+  | "pcpu" -> PCPUDisplay
+  | "block" -> BlockDisplay
+  | "net" -> NetDisplay
+  | str ->
+      failwithf (f_"%s: display should be %s") str "task|pcpu|block|net"
+let cli_of_display = function
+  | TaskDisplay -> "task"
+  | PCPUDisplay -> "pcpu"
+  | BlockDisplay -> "block"
+  | NetDisplay -> "net"
+
+(* Init file. *)
+type init_file = NoInitFile | DefaultInitFile | InitFile of string
+
 (* Settings. *)
 let quit = ref false
 let delay = ref 3000 (* milliseconds *)
 let historical_cpu_delay = ref 20 (* secs *)
 let iterations = ref (-1)
+let end_time = ref None
 let batch_mode = ref false
 let secure_mode = ref false
 let sort_order = ref Processor
@@ -65,49 +125,87 @@ let display_mode = ref TaskDisplay
 let uri = ref None
 let debug_file = ref ""
 let csv_enabled = ref false
+let csv_cpu = ref true
+let csv_block = ref true
+let csv_net = ref true
+let init_file = ref DefaultInitFile
+let script_mode = ref false
+
+(* Tuple of never-changing data returned by start_up function. *)
+type setup =
+    Libvirt.ro C.t * bool * bool * bool * C.node_info * string *
+      (int * int * int)
 
 (* Function to read command line arguments and go into curses mode. *)
 let start_up () =
   (* Read command line arguments. *)
   let rec set_delay newdelay =
     if newdelay <= 0. then
-      failwith "-d: cannot set a negative delay";
+      failwith (s_"-d: cannot set a negative delay");
     delay := int_of_float (newdelay *. 1000.)
   and set_uri = function "" -> uri := None | u -> uri := Some u
-  and set_sort = function
-    | "cpu" | "processor" -> sort_order := Processor
-    | "mem" | "memory" -> sort_order := Memory
-    | "time" -> sort_order := Time
-    | "id" -> sort_order := DomainID
-    | "name" -> sort_order := DomainName
-    | "netrx" -> sort_order := NetRX | "nettx" -> sort_order := NetTX
-    | "blockrdrq" -> sort_order := BlockRdRq
-    | "blockwrrq" -> sort_order := BlockWrRq
-    | str -> failwith (str ^ ": sort order should be: cpu|mem|time|id|name|netrx|nettx|blockrdrq|blockwrrq")
+  and set_sort order = sort_order := sort_order_of_cli order
   and set_pcpu_mode () = display_mode := PCPUDisplay
   and set_net_mode () = display_mode := NetDisplay
   and set_block_mode () = display_mode := BlockDisplay
   and set_csv filename =
     (!csv_start) filename;
     csv_enabled := true
+  and no_init_file () = init_file := NoInitFile
+  and set_init_file filename = init_file := InitFile filename
+  and set_end_time time = end_time := Some ((!parse_date_time) time)
+  and display_version () =
+    printf "virt-top %s ocaml-libvirt %s\n"
+      Virt_top_version.version Libvirt_version.version;
+    exit 0
   in
   let argspec = Arg.align [
-    "-1", Arg.Unit set_pcpu_mode, " Start by displaying pCPUs (default: tasks)";
-    "-2", Arg.Unit set_net_mode, " Start by displaying network interfaces";
-    "-3", Arg.Unit set_block_mode, " Start by displaying block devices";
-    "-b", Arg.Set batch_mode, " Batch mode";
-    "-c", Arg.String set_uri, "uri Connect to URI (default: Xen)";
-    "--connect", Arg.String set_uri, "uri Connect to URI (default: Xen)";
-    "--csv", Arg.String set_csv, "file Log statistics to CSV file";
-    "-d", Arg.Float set_delay, "delay Delay time interval (seconds)";
-    "--debug", Arg.Set_string debug_file, "file Send debug messages to file";
-    "--hist-cpu", Arg.Set_int historical_cpu_delay, "secs Historical CPU delay";
-    "-n", Arg.Set_int iterations, "iterations Number of iterations to run";
-    "-o", Arg.String set_sort, "sort Set sort order (cpu|mem|time|id|name)";
-    "-s", Arg.Set secure_mode, " Secure (\"kiosk\") mode";
+    "-1", Arg.Unit set_pcpu_mode,
+      " " ^ s_"Start by displaying pCPUs (default: tasks)";
+    "-2", Arg.Unit set_net_mode,
+      " " ^ s_"Start by displaying network interfaces";
+    "-3", Arg.Unit set_block_mode,
+      " " ^ s_"Start by displaying block devices";
+    "-b", Arg.Set batch_mode,
+      " " ^ s_"Batch mode";
+    "-c", Arg.String set_uri,
+      "uri " ^ s_"Connect to URI (default: Xen)";
+    "--connect", Arg.String set_uri,
+      "uri " ^ s_"Connect to URI (default: Xen)";
+    "--csv", Arg.String set_csv,
+      "file " ^ s_"Log statistics to CSV file";
+    "--no-csv-cpu", Arg.Clear csv_cpu,
+      " " ^ s_"Disable CPU stats in CSV";
+    "--no-csv-block", Arg.Clear csv_block,
+      " " ^ s_"Disable block device stats in CSV";
+    "--no-csv-net", Arg.Clear csv_net,
+      " " ^ s_"Disable net stats in CSV";
+    "-d", Arg.Float set_delay,
+      "delay " ^ s_"Delay time interval (seconds)";
+    "--debug", Arg.Set_string debug_file,
+      "file " ^ s_"Send debug messages to file";
+    "--end-time", Arg.String set_end_time,
+      "time " ^ s_"Exit at given time";
+    "--hist-cpu", Arg.Set_int historical_cpu_delay,
+      "secs " ^ s_"Historical CPU delay";
+    "--init-file", Arg.String set_init_file,
+      "file " ^ s_"Set name of init file";
+    "--no-init-file", Arg.Unit no_init_file,
+      " " ^ s_"Do not read init file";
+    "-n", Arg.Set_int iterations,
+      "iterations " ^ s_"Number of iterations to run";
+    "-o", Arg.String set_sort,
+      "sort " ^ sprintf (f_"Set sort order (%s)") "cpu|mem|time|id|name";
+    "-s", Arg.Set secure_mode,
+      " " ^ s_"Secure (\"kiosk\") mode";
+    "--script", Arg.Set script_mode,
+      " " ^ s_"Run from a script (no user interface)";
+    "--version", Arg.Unit display_version,
+      " " ^ s_"Display version number and exit";
   ] in
-  let anon_fun str = raise (Arg.Bad (str ^ ": unknown parameter")) in
-  let usage_msg = "virt-top : a 'top'-like utility for virtualization
+  let anon_fun str =
+    raise (Arg.Bad (sprintf (f_"%s: unknown parameter") str)) in
+  let usage_msg = s_"virt-top : a 'top'-like utility for virtualization
 
 SUMMARY
   virt-top [-options]
@@ -115,6 +213,42 @@ SUMMARY
 OPTIONS" in
   Arg.parse argspec anon_fun usage_msg;
 
+  (* Read the init file. *)
+  let try_to_read_init_file filename =
+    let config = read_config_file filename in
+    List.iter (
+      function
+      | _, "display", mode -> display_mode := display_of_cli mode
+      | _, "delay", secs -> set_delay (float_of_string secs)
+      | _, "hist-cpu", secs -> historical_cpu_delay := int_of_string secs
+      | _, "iterations", n -> iterations := int_of_string n
+      | _, "sort", order -> set_sort order
+      | _, "connect", uri -> set_uri uri
+      | _, "debug", filename -> debug_file := filename
+      | _, "csv", filename -> set_csv filename
+      | _, "csv-cpu", b -> csv_cpu := bool_of_string b
+      | _, "csv-block", b -> csv_block := bool_of_string b
+      | _, "csv-net", b -> csv_net := bool_of_string b
+      | _, "batch", b -> batch_mode := bool_of_string b
+      | _, "secure", b -> secure_mode := bool_of_string b
+      | _, "script", b -> script_mode := bool_of_string b
+      | _, "end-time", t -> set_end_time t
+      | _, "overwrite-init-file", "false" -> no_init_file ()
+      | lineno, key, _ ->
+         eprintf (f_"%s:%d: configuration item ``%s'' ignored\n%!")
+           filename lineno key
+    ) config
+  in
+  (match !init_file with
+   | NoInitFile -> ()
+   | DefaultInitFile ->
+       let home = try Sys.getenv "HOME" with Not_found -> "/" in
+       let filename = home // rcfile in
+       try_to_read_init_file filename
+   | InitFile filename ->
+       try_to_read_init_file filename
+  );
+
   (* Connect to the hypervisor before going into curses mode, since
    * this is the most likely thing to fail.
    *)
@@ -126,7 +260,7 @@ OPTIONS" in
        prerr_endline (Libvirt.Virterror.to_string err);
        (* If non-root and no explicit connection URI, print a warning. *)
        if Unix.geteuid () <> 0 && name = None then (
-         print_endline "NB: If you want to monitor a local Xen hypervisor, you usually need to be root";
+         print_endline (s_"NB: If you want to monitor a local Xen hypervisor, you usually need to be root");
        );
        exit 1 in
 
@@ -137,7 +271,10 @@ OPTIONS" in
   let hostname =
     try C.get_hostname conn
     with
-    | Invalid_argument "virConnectGetHostname not supported" -> "unknown" in
+    (* qemu:/// and other URIs didn't support virConnectGetHostname until
+     * libvirt 0.3.3.  Before that they'd throw a virterror. *)
+    | Libvirt.Virterror _
+    | Libvirt.Not_supported "virConnectGetHostname" -> "unknown" in
 
   let libvirt_version =
     let v, _ = Libvirt.get_version () in
@@ -147,13 +284,13 @@ OPTIONS" in
    * NB: Do this just before jumping into curses mode.
    *)
   (match !debug_file with
-   | "" -> (* No debug file specified, send stderr to /dev/null. *)
-       (try
-         let fd = Unix.openfile "/dev/null" [Unix.O_WRONLY] 0o644 in
-         Unix.dup2 fd Unix.stderr;
-         Unix.close fd
-       with
-         Unix.Unix_error _ -> ()
+   | "" -> (* No debug file specified, send stderr to /dev/null unless
+           * we're in script mode.
+           *)
+       if not !script_mode then (
+        let fd = Unix.openfile "/dev/null" [Unix.O_WRONLY] 0o644 in
+        Unix.dup2 fd Unix.stderr;
+        Unix.close fd
        )
    | filename -> (* Send stderr to the named file. *)
        let fd =
@@ -164,68 +301,24 @@ OPTIONS" in
   );
 
   (* Curses voodoo (see ncurses(3)). *)
-  let stdscr =
-    initscr ();
-    cbreak ();
-    noecho ();
+  if not !script_mode then (
+    ignore (initscr ());
+    ignore (cbreak ());
+    ignore (noecho ());
     nonl ();
     let stdscr = stdscr () in
-    intrflush stdscr false;
-    keypad stdscr true;
-    stdscr in
+    ignore (intrflush stdscr false);
+    ignore (keypad stdscr true);
+    ()
+  );
 
-  (* This tuple of static information is called 'state' in other parts
+  (* This tuple of static information is called 'setup' in other parts
    * of this program, and is passed to other functions such as redraw and
-   * main_loop.  See virt_top_main.ml.  It's not really "state" though.
+   * main_loop.  See virt_top_main.ml.
    *)
-  conn, stdscr, node_info, hostname, libvirt_version
-
-(* Show a percentage in 4 chars. *)
-let show_percent percent =
-  if percent <= 0. then " 0.0"
-  else if percent <= 9.9 then sprintf " %1.1f" percent
-  else if percent <= 99.9 then sprintf "%2.1f" percent
-  else "100 "
-
-(* Show an int64 option in 4 chars. *)
-let rec show_int64_option = function
-  | None -> "    "
-  | Some n -> show_int64 n
-(* Show an int64 in 4 chars. *)
-and show_int64 = function
-  | n when n < 0L -> "-!!!"
-  | n when n <= 9999L ->
-      sprintf "%4Ld" n
-  | n when n /^ 1024L <= 999L ->
-      sprintf "%3LdK" (n /^ 1024L)
-  | n when n /^ 1_048_576L <= 999L ->
-      sprintf "%3LdM" (n /^ 1_048_576L)
-  | n when n /^ 1_073_741_824L <= 999L ->
-      sprintf "%3LdG" (n /^ 1_073_741_824L)
-  | _ -> ">!!!"
-
-(* Format the total time (may be large!) in 9 chars. *)
-let show_time ns =
-  let secs_in_ns = 1_000_000_000L in
-  let mins_in_ns = 60_000_000_000L in
-  let hours_in_ns = 3_600_000_000_000L in
-
-  let hours = ns /^ hours_in_ns in
-  let ns = ns -^ (hours *^ hours_in_ns) in
-  let mins = ns /^ mins_in_ns in
-  let ns = ns -^ (mins *^ mins_in_ns) in
-  let secs = ns /^ secs_in_ns in
-  let ns = ns -^ (secs *^ secs_in_ns) in
-  let pennies = ns /^ 10_000_000L in
-
-  if hours < 12L then
-    sprintf "%3Ld:%02Ld.%02Ld" (hours *^ 60L +^ mins) secs pennies
-  else if hours <= 999L then
-    sprintf "%3Ld:%02Ld:%02Ld" hours mins secs
-  else (
-    let days = hours /^ 24L in
-    let hours = hours -^ (days *^ 24L) in
-    sprintf "%3Ldd%02Ld:%02Ld" days hours mins
+  (conn,
+   !batch_mode, !script_mode, !csv_enabled, (* immutable modes *)
+   node_info, hostname, libvirt_version (* info that doesn't change *)
   )
 
 (* Show a domain state (the 'S' column). *)
@@ -238,76 +331,21 @@ let show_state = function
   | D.InfoShutoff -> 'O'
   | D.InfoCrashed -> 'X'
 
-(* Sum Domain.block_stats structures together.  Missing fields
- * get forced to 0.  Empty list returns all 0.
- *)
-let zero_block_stats =
-  { D.rd_req = 0L; rd_bytes = 0L; wr_req = 0L; wr_bytes = 0L; errs = 0L }
-let add_block_stats bs1 bs2 =
-  let add f1 f2 = if f1 >= 0L && f2 >= 0L then f1 +^ f2 else 0L in
-  { D.rd_req = add bs1.D.rd_req   bs2.D.rd_req;
-    rd_bytes = add bs1.D.rd_bytes bs2.D.rd_bytes;
-    wr_req   = add bs1.D.wr_req   bs2.D.wr_req;
-    wr_bytes = add bs1.D.wr_bytes bs2.D.wr_bytes;
-    errs     = add bs1.D.errs     bs2.D.errs }
-let sum_block_stats =
-  List.fold_left add_block_stats zero_block_stats
-
-(* Get the difference between two block_stats structures.  Missing data
- * forces the difference to -1.
- *)
-let diff_block_stats curr prev =
-  let sub f1 f2 = if f1 >= 0L && f2 >= 0L then f1 -^ f2 else -1L in
-  { D.rd_req = sub curr.D.rd_req   prev.D.rd_req;
-    rd_bytes = sub curr.D.rd_bytes prev.D.rd_bytes;
-    wr_req   = sub curr.D.wr_req   prev.D.wr_req;
-    wr_bytes = sub curr.D.wr_bytes prev.D.wr_bytes;
-    errs     = sub curr.D.errs     prev.D.errs }
-
-(* Sum Domain.interface_stats structures together.  Missing fields
- * get forced to 0.  Empty list returns all 0.
- *)
-let zero_interface_stats =
-  { D.rx_bytes = 0L; rx_packets = 0L; rx_errs = 0L; rx_drop = 0L;
-    tx_bytes = 0L; tx_packets = 0L; tx_errs = 0L; tx_drop = 0L }
-let add_interface_stats is1 is2 =
-  let add f1 f2 = if f1 >= 0L && f2 >= 0L then f1 +^ f2 else 0L in
-  { D.rx_bytes = add is1.D.rx_bytes   is2.D.rx_bytes;
-    rx_packets = add is1.D.rx_packets is2.D.rx_packets;
-    rx_errs    = add is1.D.rx_errs    is2.D.rx_errs;
-    rx_drop    = add is1.D.rx_drop    is2.D.rx_drop;
-    tx_bytes   = add is1.D.tx_bytes   is2.D.tx_bytes;
-    tx_packets = add is1.D.tx_packets is2.D.tx_packets;
-    tx_errs    = add is1.D.tx_errs    is2.D.tx_errs;
-    tx_drop    = add is1.D.tx_drop    is2.D.tx_drop }
-let sum_interface_stats =
-  List.fold_left add_interface_stats zero_interface_stats
-
-(* Get the difference between two interface_stats structures.
- * Missing data forces the difference to -1.
- *)
-let diff_interface_stats curr prev =
-  let sub f1 f2 = if f1 >= 0L && f2 >= 0L then f1 -^ f2 else -1L in
-  { D.rx_bytes = sub curr.D.rx_bytes   prev.D.rx_bytes;
-    rx_packets = sub curr.D.rx_packets prev.D.rx_packets;
-    rx_errs    = sub curr.D.rx_errs    prev.D.rx_errs;
-    rx_drop    = sub curr.D.rx_drop    prev.D.rx_drop;
-    tx_bytes   = sub curr.D.tx_bytes   prev.D.tx_bytes;
-    tx_packets = sub curr.D.tx_packets prev.D.tx_packets;
-    tx_errs    = sub curr.D.tx_errs    prev.D.tx_errs;
-    tx_drop    = sub curr.D.tx_drop    prev.D.tx_drop }
-
-(* Update the display and sleep for given number of seconds. *)
-let sleep n = refresh (); Unix.sleep n
+(* Sleep in seconds. *)
+let sleep = Unix.sleep
+
+(* Sleep in milliseconds. *)
+let usleep n =
+  ignore (Unix.select [] [] [] (float n /. 1000.))
 
 (* The curses getstr/getnstr functions are just weird.
  * This helper function also enables echo temporarily.
  *)
 let get_string maxlen =
-  echo ();
+  ignore (echo ());
   let str = String.create maxlen in
   let ok = getstr str in (* Safe because binding calls getnstr. *)
-  noecho ();
+  ignore (noecho ());
   if not ok then ""
   else (
     (* Chop at first '\0'. *)
@@ -318,13 +356,6 @@ let get_string maxlen =
       Not_found -> str (* it is full maxlen bytes *)
   )
 
-(* Pad a string to the full width with spaces.  If too long, truncate. *)
-let pad width str =
-  let n = String.length str in
-  if n = width then str
-  else if n > width then String.sub str 0 width
-  else (* if n < width then *) str ^ String.make (width-n) ' '
-
 (* Line numbers. *)
 let top_lineno = 0
 let summary_lineno = 1 (* this takes 2 lines *)
@@ -332,20 +363,19 @@ let message_lineno = 3
 let header_lineno = 4
 let domains_lineno = 5
 
+(* Easier to use versions of curses functions addstr, mvaddstr, etc. *)
+let move y x = ignore (move y x)
+let refresh () = ignore (refresh ())
+let addch c = ignore (addch (int_of_char c))
+let addstr s = ignore (addstr s)
+let mvaddstr y x s = ignore (mvaddstr y x s)
+
 (* Print in the "message area". *)
 let clear_msg () = move message_lineno 0; clrtoeol ()
-let print_msg str = clear_msg (); mvaddstr message_lineno 0 str; ()
-
-(* Write CSV header row. *)
-let write_csv_header () =
-  (!csv_write) [ "Hostname"; "Time"; "Arch"; "Physical CPUs";
-                "Count"; "Running"; "Blocked"; "Paused"; "Shutdown";
-                "Shutoff"; "Crashed"; "Active"; "Inactive";
-                "%CPU"; "Total memory KB"; "Total guest memory KB";
-                "Total CPU time ns" ]
+let print_msg str = clear_msg (); mvaddstr message_lineno 0 str
 
 (* Intermediate "domain + stats" structure that we use to collect
- * everything we know about a domain within the redraw function.
+ * everything we know about a domain within the collect function.
  *)
 type rd_domain = Inactive | Active of rd_active
 and rd_active = {
@@ -371,8 +401,8 @@ and rd_active = {
   rd_net_tx_bytes : int64 option;      (* Number of bytes transmitted. *)
 }
 
-(* Redraw the display. *)
-let redraw, clear_pcpu_display_data =
+(* Collect stats. *)
+let collect, clear_pcpu_display_data =
   (* We cache the list of block devices and interfaces for each domain
    * here, so we don't need to reparse the XML each time.
    *)
@@ -399,16 +429,14 @@ let redraw, clear_pcpu_display_data =
   (* Save vcpuinfo structures across redraws too (only for pCPU display). *)
   let last_vcpu_info = Hashtbl.create 13 in
 
-  (* Keep a historical list of %CPU usages. *)
-  let historical_cpu = ref [] in
-  let historical_cpu_last_time = ref (Unix.gettimeofday ()) in
-
-  let redraw (conn, stdscr, node_info, hostname, _) =
-    clear ();
-
-    (* Get the screen/window size. *)
-    let lines, cols = get_size () in
+  let clear_pcpu_display_data () =
+    (* Clear out vcpu_info used by PCPUDisplay display_mode
+     * when we switch back to TaskDisplay mode.
+     *)
+    Hashtbl.clear last_vcpu_info
+  in
 
+  let collect (conn, _, _, _, node_info, _, _) =
     (* Number of physical CPUs (some may be disabled). *)
     let nr_pcpus = C.maxcpus_of_node_info node_info in
 
@@ -417,7 +445,6 @@ let redraw, clear_pcpu_display_data =
     let tm = Unix.localtime time in
     let printable_time =
       sprintf "%02d:%02d:%02d" tm.Unix.tm_hour tm.Unix.tm_min tm.Unix.tm_sec in
-    mvaddstr top_lineno 0 ("virt-top " ^ printable_time ^ " - ");
 
     (* What's the total CPU time elapsed since we were last called? (ns) *)
     let total_cpu_per_pcpu = 1_000_000_000. *. (time -. !last_time) in
@@ -426,15 +453,6 @@ let redraw, clear_pcpu_display_data =
       if total_cpu_per_pcpu <= 0. then 1. else total_cpu_per_pcpu in
     let total_cpu = float node_info.C.cpus *. total_cpu_per_pcpu in
 
-    (* Basic node_info. *)
-    addstr (sprintf "%s %d/%dCPU %dMHz %LdMB "
-             node_info.C.model node_info.C.cpus nr_pcpus node_info.C.mhz
-             (node_info.C.memory /^ 1024L));
-    (* Save the cursor position for when we come to draw the
-     * historical CPU times (down in this function).
-     *)
-    let historical_cursor = getyx stdscr in
-
     (* Get the domains.  Match up with their last_info (if any). *)
     let doms =
       (* Active domains. *)
@@ -455,12 +473,12 @@ let redraw, clear_pcpu_display_data =
              let block_stats =
                try List.map (fun dev -> dev, D.block_stats dom dev) blkdevs
                with
-               | Invalid_argument "virDomainBlockStats not supported"
+               | Libvirt.Not_supported "virDomainBlockStats"
                | Libvirt.Virterror _ -> [] in
              let interface_stats =
                try List.map (fun dev -> dev, D.interface_stats dom dev) netifs
                with
-               | Invalid_argument "virDomainInterfaceStats not supported"
+               | Libvirt.Not_supported "virDomainInterfaceStats"
                | Libvirt.Virterror _ -> [] in
 
              let prev_info, prev_block_stats, prev_interface_stats =
@@ -486,11 +504,18 @@ let redraw, clear_pcpu_display_data =
        ) ids in
 
       (* Inactive domains. *)
-      let n = C.num_of_defined_domains conn in
-      let names =
-       if n > 0 then Array.to_list (C.list_defined_domains conn n)
-       else [] in
-      let doms_inactive = List.map (fun name -> name, Inactive) names in
+      let doms_inactive =
+       try
+         let n = C.num_of_defined_domains conn in
+         let names =
+           if n > 0 then Array.to_list (C.list_defined_domains conn n)
+           else [] in
+         List.map (fun name -> name, Inactive) names
+       with
+         (* Ignore transient errors, in particular errors from
+          * num_of_defined_domains if it cannot contact xend.
+          *)
+       | Libvirt.Virterror _ -> [] in
 
       doms @ doms_inactive in
 
@@ -574,6 +599,178 @@ let redraw, clear_pcpu_display_data =
        | rd -> rd
       ) doms in
 
+    (* Collect some extra information in PCPUDisplay display_mode. *)
+    let pcpu_display =
+      if !display_mode = PCPUDisplay then (
+       (* Get the VCPU info and VCPU->PCPU mappings for active domains.
+        * Also cull some data we don't care about.
+        *)
+       let doms = List.filter_map (
+         function
+         | (name, Active rd) ->
+             (try
+                let domid = rd.rd_domid in
+                let maplen = C.cpumaplen nr_pcpus in
+                let maxinfo = rd.rd_info.D.nr_virt_cpu in
+                let nr_vcpus, vcpu_infos, cpumaps =
+                  D.get_vcpus rd.rd_dom maxinfo maplen in
+
+                (* Got previous vcpu_infos for this domain? *)
+                let prev_vcpu_infos =
+                  try Some (Hashtbl.find last_vcpu_info domid)
+                  with Not_found -> None in
+                (* Update last_vcpu_info. *)
+                Hashtbl.replace last_vcpu_info domid vcpu_infos;
+
+                (match prev_vcpu_infos with
+                 | Some prev_vcpu_infos
+                     when Array.length prev_vcpu_infos = Array.length vcpu_infos ->
+                     Some (domid, name, nr_vcpus, vcpu_infos, prev_vcpu_infos,
+                           cpumaps, maplen)
+                 | _ -> None (* ignore missing / unequal length prev_vcpu_infos *)
+                );
+              with
+                Libvirt.Virterror _ -> None(* ignore transient libvirt errs *)
+             )
+         | (_, Inactive) -> None (* ignore inactive doms *)
+       ) doms in
+       let nr_doms = List.length doms in
+
+       (* Rearrange the data into a matrix.  Major axis (down) is
+        * pCPUs.  Minor axis (right) is domains.  At each node we store:
+        *  cpu_time (on this pCPU only, nanosecs),
+        *  average? (if set, then cpu_time is an average because the
+        *     vCPU is pinned to more than one pCPU)
+        *  running? (if set, we were instantaneously running on this pCPU)
+        *)
+       let empty_node = (0L, false, false) in
+       let pcpus = Array.make_matrix nr_pcpus nr_doms empty_node in
+
+       List.iteri (
+         fun di (domid, name, nr_vcpus, vcpu_infos, prev_vcpu_infos,
+                 cpumaps, maplen) ->
+           (* Which pCPUs can this dom run on? *)
+           for v = 0 to nr_vcpus-1 do
+             let pcpu = vcpu_infos.(v).D.cpu in (* instantaneous pCPU *)
+             let nr_poss_pcpus = ref 0 in (* how many pcpus can it run on? *)
+             for p = 0 to nr_pcpus-1 do
+               (* vcpu v can reside on pcpu p *)
+               if C.cpu_usable cpumaps maplen v p then
+                 incr nr_poss_pcpus
+             done;
+             let nr_poss_pcpus = Int64.of_int !nr_poss_pcpus in
+             for p = 0 to nr_pcpus-1 do
+               (* vcpu v can reside on pcpu p *)
+               if C.cpu_usable cpumaps maplen v p then
+                 let vcpu_time_on_pcpu =
+                   vcpu_infos.(v).D.vcpu_time
+                   -^ prev_vcpu_infos.(v).D.vcpu_time in
+                 let vcpu_time_on_pcpu =
+                   vcpu_time_on_pcpu /^ nr_poss_pcpus in
+                 pcpus.(p).(di) <-
+                   (vcpu_time_on_pcpu, nr_poss_pcpus > 1L, p = pcpu)
+             done
+           done
+       ) doms;
+
+       (* Sum the CPU time used by each pCPU, for the %CPU column. *)
+       let pcpus_cpu_time = Array.map (
+         fun row ->
+           let cpu_time = ref 0L in
+           for di = 0 to Array.length row-1 do
+             let t, _, _ = row.(di) in
+             cpu_time := !cpu_time +^ t
+           done;
+           Int64.to_float !cpu_time
+       ) pcpus in
+
+       Some (doms, pcpus, pcpus_cpu_time)
+      ) else
+       None in
+
+    (* Calculate totals. *)
+    let totals = List.fold_left (
+      fun (count, running, blocked, paused, shutdown, shutoff,
+          crashed, active, inactive,
+          total_cpu_time, total_memory, total_domU_memory) ->
+       function
+       | (name, Active rd) ->
+           let test state orig =
+             if rd.rd_info.D.state = state then orig+1 else orig
+           in
+           let running = test D.InfoRunning running in
+           let blocked = test D.InfoBlocked blocked in
+           let paused = test D.InfoPaused paused in
+           let shutdown = test D.InfoShutdown shutdown in
+           let shutoff = test D.InfoShutoff shutoff in
+           let crashed = test D.InfoCrashed crashed in
+
+           let total_cpu_time = total_cpu_time +. rd.rd_cpu_time in
+           let total_memory = total_memory +^ rd.rd_info.D.memory in
+           let total_domU_memory = total_domU_memory +^
+             if rd.rd_domid > 0 then rd.rd_info.D.memory else 0L in
+
+           (count+1, running, blocked, paused, shutdown, shutoff,
+            crashed, active+1, inactive,
+            total_cpu_time, total_memory, total_domU_memory)
+
+       | (name, Inactive) -> (* inactive domain *)
+           (count+1, running, blocked, paused, shutdown, shutoff,
+            crashed, active, inactive+1,
+            total_cpu_time, total_memory, total_domU_memory)
+    ) (0,0,0,0,0,0,0,0,0, 0.,0L,0L) doms in
+
+    (* Update last_time, last_info. *)
+    last_time := time;
+    Hashtbl.clear last_info;
+    List.iter (
+      function
+      | (_, Active rd) ->
+         let info = rd.rd_info, rd.rd_block_stats, rd.rd_interface_stats in
+         Hashtbl.add last_info rd.rd_domid info
+      | _ -> ()
+    ) doms;
+
+    (doms,
+     time, printable_time,
+     nr_pcpus, total_cpu, total_cpu_per_pcpu,
+     totals,
+     pcpu_display)
+  in
+
+  collect, clear_pcpu_display_data
+
+(* Redraw the display. *)
+let redraw =
+  (* Keep a historical list of %CPU usages. *)
+  let historical_cpu = ref [] in
+  let historical_cpu_last_time = ref (Unix.gettimeofday ()) in
+  fun
+  (_, _, _, _, node_info, _, _) (* setup *)
+  (doms,
+   time, printable_time,
+   nr_pcpus, total_cpu, total_cpu_per_pcpu,
+   totals,
+   pcpu_display) (* state *) ->
+    clear ();
+
+    (* Get the screen/window size. *)
+    let lines, cols = get_size () in
+
+    (* Time. *)
+    mvaddstr top_lineno 0 (sprintf "virt-top %s - " printable_time);
+
+    (* Basic node_info. *)
+    addstr
+      (sprintf "%s %d/%dCPU %dMHz %LdMB "
+        node_info.C.model node_info.C.cpus nr_pcpus node_info.C.mhz
+        (node_info.C.memory /^ 1024L));
+    (* Save the cursor position for when we come to draw the
+     * historical CPU times (down in this function).
+     *)
+    let stdscr = stdscr () in
+    let historical_cursor = getyx stdscr in
+
     (match !display_mode with
      | TaskDisplay -> (*---------- Showing domains ----------*)
         (* Sort domains on current sort_order. *)
@@ -657,16 +854,16 @@ let redraw, clear_pcpu_display_data =
           | (name, Active rd) :: doms ->
               if lineno < lines then (
                 let state = show_state rd.rd_info.D.state in
-                let rd_req = show_int64_option rd.rd_block_rd_reqs in
-                let wr_req = show_int64_option rd.rd_block_wr_reqs in
-                let rx_bytes = show_int64_option rd.rd_net_rx_bytes in
-                let tx_bytes = show_int64_option rd.rd_net_tx_bytes in
-                let percent_cpu = show_percent rd.rd_percent_cpu in
+                let rd_req = Show.int64_option rd.rd_block_rd_reqs in
+                let wr_req = Show.int64_option rd.rd_block_wr_reqs in
+                let rx_bytes = Show.int64_option rd.rd_net_rx_bytes in
+                let tx_bytes = Show.int64_option rd.rd_net_tx_bytes in
+                let percent_cpu = Show.percent rd.rd_percent_cpu in
                 let percent_mem =
                   100L *^ rd.rd_info.D.memory /^ node_info.C.memory in
                 let percent_mem = Int64.to_float percent_mem in
-                let percent_mem = show_percent percent_mem in
-                let time = show_time rd.rd_info.D.cpu_time in
+                let percent_mem = Show.percent percent_mem in
+                let time = Show.time rd.rd_info.D.cpu_time in
 
                 let line = sprintf "%5d %c %s %s %s %s %s %s %s %s"
                   rd.rd_domid state rd_req wr_req rx_bytes tx_bytes
@@ -689,87 +886,10 @@ let redraw, clear_pcpu_display_data =
         loop domains_lineno doms
 
      | PCPUDisplay -> (*---------- Showing physical CPUs ----------*)
-        (* Get the VCPU info and VCPU->PCPU mappings for active domains.
-         * Also cull some data we don't care about.
-         *)
-        let doms = List.filter_map (
-          function
-          | (name, Active rd) ->
-              (try
-                 let domid = rd.rd_domid in
-                 let maplen = C.cpumaplen nr_pcpus in
-                 let maxinfo = rd.rd_info.D.nr_virt_cpu in
-                 let nr_vcpus, vcpu_infos, cpumaps =
-                   D.get_vcpus rd.rd_dom maxinfo maplen in
-
-                 (* Got previous vcpu_infos for this domain? *)
-                 let prev_vcpu_infos =
-                   try Some (Hashtbl.find last_vcpu_info domid)
-                   with Not_found -> None in
-                 (* Update last_vcpu_info. *)
-                 Hashtbl.replace last_vcpu_info domid vcpu_infos;
-
-                 (match prev_vcpu_infos with
-                  | Some prev_vcpu_infos
-                      when Array.length prev_vcpu_infos = Array.length vcpu_infos ->
-                      Some (domid, name, nr_vcpus, vcpu_infos, prev_vcpu_infos,
-                            cpumaps, maplen)
-                  | _ -> None (* ignore missing / unequal length prev_vcpu_infos *)
-                 );
-               with
-                 Libvirt.Virterror _ -> None(* ignore transient libvirt errs *)
-              )
-          | (_, Inactive) -> None (* ignore inactive doms *)
-        ) doms in
-        let nr_doms = List.length doms in
-
-        (* Rearrange the data into a matrix.  Major axis (down) is
-         * pCPUs.  Minor axis (right) is domains.  At each node we store:
-         *  cpu_time (on this pCPU only, nanosecs),
-         *  average? (if set, then cpu_time is an average because the
-         *     vCPU is pinned to more than one pCPU)
-         *  running? (if set, we were instantaneously running on this pCPU)
-         *)
-        let empty_node = (0L, false, false) in
-        let pcpus = Array.make_matrix nr_pcpus nr_doms empty_node in
-
-        List.iteri (
-          fun di (domid, name, nr_vcpus, vcpu_infos, prev_vcpu_infos,
-                  cpumaps, maplen) ->
-            (* Which pCPUs can this dom run on? *)
-            for v = 0 to nr_vcpus-1 do
-              let pcpu = vcpu_infos.(v).D.cpu in (* instantaneous pCPU *)
-              let nr_poss_pcpus = ref 0 in (* how many pcpus can it run on? *)
-              for p = 0 to nr_pcpus-1 do
-                (* vcpu v can reside on pcpu p *)
-                if C.cpu_usable cpumaps maplen v p then
-                  incr nr_poss_pcpus
-              done;
-              let nr_poss_pcpus = Int64.of_int !nr_poss_pcpus in
-              for p = 0 to nr_pcpus-1 do
-                (* vcpu v can reside on pcpu p *)
-                if C.cpu_usable cpumaps maplen v p then
-                  let vcpu_time_on_pcpu =
-                    vcpu_infos.(v).D.vcpu_time
-                    -^ prev_vcpu_infos.(v).D.vcpu_time in
-                  let vcpu_time_on_pcpu =
-                    vcpu_time_on_pcpu /^ nr_poss_pcpus in
-                  pcpus.(p).(di) <-
-                    (vcpu_time_on_pcpu, nr_poss_pcpus > 1L, p = pcpu)
-              done
-            done
-        ) doms;
-
-        (* Sum the CPU time used by each pCPU, for the %CPU column. *)
-        let pcpus_cpu_time = Array.map (
-          fun row ->
-            let cpu_time = ref 0L in
-            for di = 0 to Array.length row-1 do
-              let t, _, _ = row.(di) in
-              cpu_time := !cpu_time +^ t
-            done;
-            Int64.to_float !cpu_time
-        ) pcpus in
+        let doms, pcpus, pcpus_cpu_time =
+          match pcpu_display with
+          | Some p -> p
+          | None -> failwith "internal error: no pcpu_display data" in
 
         (* Display the pCPUs. *)
         let dom_names =
@@ -790,8 +910,8 @@ let redraw, clear_pcpu_display_data =
             mvaddstr (p+domains_lineno) 0 (sprintf "%4d   " p);
             let cpu_time = pcpus_cpu_time.(p) in (* ns used on this CPU *)
             let percent_cpu = 100. *. cpu_time /. total_cpu_per_pcpu in
-            addstr (show_percent percent_cpu);
-            addch 32;
+            addstr (Show.percent percent_cpu);
+            addch ' ';
 
             List.iteri (
               fun di (domid, name, _, _, _, _, _) ->
@@ -803,7 +923,7 @@ let redraw, clear_pcpu_display_data =
                   else (
                     let t = Int64.to_float t in
                     let percent = 100. *. t /. total_cpu_per_pcpu in
-                    sprintf "%s%c%c " (show_percent percent)
+                    sprintf "%s%c%c " (Show.percent percent)
                       (if is_average then '=' else ' ')
                       (if is_running then '#' else ' ')
                   ) in
@@ -864,7 +984,7 @@ let redraw, clear_pcpu_display_data =
                 (fun (_, { rd_domid = id1 }, _, { rd_domid = id2 }) ->
                    compare id1 id2)
             | Processor | Memory | Time | BlockRdRq | BlockWrRq
-                (* fallthrough to RXBY comparison. *)
+                  (* fallthrough to RXBY comparison. *)
             | NetRX ->
                 (fun ({ D.rx_bytes = b1 }, _, { D.rx_bytes = b2 }, _) ->
                    compare b2 b1)
@@ -885,7 +1005,7 @@ let redraw, clear_pcpu_display_data =
           (pad cols "   ID S RXBY TXBY RXPK TXPK DOMAIN       INTERFACE");
         attroff A.reverse;
 
-         (* Print domains and devices. *)
+        (* Print domains and devices. *)
         let rec loop lineno = function
           | [] -> ()
           | (dev, name, rd, stats) :: devs ->
@@ -893,19 +1013,19 @@ let redraw, clear_pcpu_display_data =
                 let state = show_state rd.rd_info.D.state in
                 let rx_bytes =
                   if stats.D.rx_bytes >= 0L
-                  then show_int64 stats.D.rx_bytes
+                  then Show.int64 stats.D.rx_bytes
                   else "    " in
                 let tx_bytes =
                   if stats.D.tx_bytes >= 0L
-                  then show_int64 stats.D.tx_bytes
+                  then Show.int64 stats.D.tx_bytes
                   else "    " in
                 let rx_packets =
                   if stats.D.rx_packets >= 0L
-                  then show_int64 stats.D.rx_packets
+                  then Show.int64 stats.D.rx_packets
                   else "    " in
                 let tx_packets =
                   if stats.D.tx_packets >= 0L
-                  then show_int64 stats.D.tx_packets
+                  then Show.int64 stats.D.tx_packets
                   else "    " in
 
                 let line = sprintf "%5d %c %s %s %s %s %-12s %s"
@@ -972,7 +1092,7 @@ let redraw, clear_pcpu_display_data =
                 (fun (_, { rd_domid = id1 }, _, { rd_domid = id2 }) ->
                    compare id1 id2)
             | Processor | Memory | Time | NetRX | NetTX
-                (* fallthrough to RDRQ comparison. *)
+                  (* fallthrough to RDRQ comparison. *)
             | BlockRdRq ->
                 (fun ({ D.rd_req = b1 }, _, { D.rd_req = b2 }, _) ->
                    compare b2 b1)
@@ -993,7 +1113,7 @@ let redraw, clear_pcpu_display_data =
           (pad cols "   ID S RDBY WRBY RDRQ WRRQ DOMAIN       DEVICE");
         attroff A.reverse;
 
-         (* Print domains and devices. *)
+        (* Print domains and devices. *)
         let rec loop lineno = function
           | [] -> ()
           | (dev, name, rd, stats) :: devs ->
@@ -1001,19 +1121,19 @@ let redraw, clear_pcpu_display_data =
                 let state = show_state rd.rd_info.D.state in
                 let rd_bytes =
                   if stats.D.rd_bytes >= 0L
-                  then show_int64 stats.D.rd_bytes
+                  then Show.int64 stats.D.rd_bytes
                   else "    " in
                 let wr_bytes =
                   if stats.D.wr_bytes >= 0L
-                  then show_int64 stats.D.wr_bytes
+                  then Show.int64 stats.D.wr_bytes
                   else "    " in
                 let rd_req =
                   if stats.D.rd_req >= 0L
-                  then show_int64 stats.D.rd_req
+                  then Show.int64 stats.D.rd_req
                   else "    " in
                 let wr_req =
                   if stats.D.wr_req >= 0L
-                  then show_int64 stats.D.wr_req
+                  then Show.int64 stats.D.wr_req
                   else "    " in
 
                 let line = sprintf "%5d %c %s %s %s %s %-12s %s"
@@ -1027,137 +1147,166 @@ let redraw, clear_pcpu_display_data =
               )
         in
         loop domains_lineno devs
+    ); (* end of display_mode conditional section *)
+
+    let (count, running, blocked, paused, shutdown, shutoff,
+        crashed, active, inactive,
+        total_cpu_time, total_memory, total_domU_memory) = totals in
+
+    mvaddstr summary_lineno 0
+      (sprintf
+        (f_"%d domains, %d active, %d running, %d sleeping, %d paused, %d inactive D:%d O:%d X:%d")
+        count active running blocked paused inactive shutdown shutoff crashed);
+
+    (* Total %CPU used, and memory summary. *)
+    let percent_cpu = 100. *. total_cpu_time /. total_cpu in
+    mvaddstr (summary_lineno+1) 0
+      (sprintf
+        (f_"CPU: %2.1f%%  Mem: %Ld MB (%Ld MB by guests)")
+        percent_cpu (total_memory /^ 1024L) (total_domU_memory /^ 1024L));
+
+    (* Time to grab another historical %CPU for the list? *)
+    if time >= !historical_cpu_last_time +. float !historical_cpu_delay
+    then (
+      historical_cpu := percent_cpu :: List.take 10 !historical_cpu;
+      historical_cpu_last_time := time
     );
 
-    (* Calculate and print totals. *)
+    (* Display historical CPU time. *)
     let () =
-      let totals = List.fold_left (
-       fun (count, running, blocked, paused, shutdown, shutoff,
-            crashed, active, inactive,
-            total_cpu_time, total_memory, total_domU_memory) ->
-         function
-         | (name, Active rd) ->
-             let test state orig =
-               if rd.rd_info.D.state = state then orig+1 else orig
-             in
-             let running = test D.InfoRunning running in
-             let blocked = test D.InfoBlocked blocked in
-             let paused = test D.InfoPaused paused in
-             let shutdown = test D.InfoShutdown shutdown in
-             let shutoff = test D.InfoShutoff shutoff in
-             let crashed = test D.InfoCrashed crashed in
-
-             let total_cpu_time = total_cpu_time +. rd.rd_cpu_time in
-             let total_memory = total_memory +^ rd.rd_info.D.memory in
-             let total_domU_memory = total_domU_memory +^
-               if rd.rd_domid > 0 then rd.rd_info.D.memory else 0L in
-
-             (count+1, running, blocked, paused, shutdown, shutoff,
-              crashed, active+1, inactive,
-              total_cpu_time, total_memory, total_domU_memory)
-
-         | (name, Inactive) -> (* inactive domain *)
-             (count+1, running, blocked, paused, shutdown, shutoff,
-              crashed, active, inactive+1,
-              total_cpu_time, total_memory, total_domU_memory)
-      ) (0,0,0,0,0,0,0,0,0, 0.,0L,0L) doms in
-
-      let (count, running, blocked, paused, shutdown, shutoff,
-          crashed, active, inactive,
-          total_cpu_time, total_memory, total_domU_memory) = totals in
-
-      mvaddstr summary_lineno 0
-       (sprintf "%d domains, %d active, %d running, %d sleeping, %d paused, %d inactive D:%d O:%d X:%d"
-          count active running blocked paused inactive shutdown shutoff
-          crashed);
-
-      (* Total %CPU used, and memory summary. *)
-      let percent_cpu = 100. *. total_cpu_time /. total_cpu in
-      mvaddstr (summary_lineno+1) 0
-       (sprintf "CPU: %2.1f%%  Mem: %Ld MB (%Ld MB by guests)"
-          percent_cpu (total_memory /^ 1024L) (total_domU_memory /^ 1024L));
-
-      (* Time to grab another historical %CPU for the list? *)
-      if time >= !historical_cpu_last_time +. float !historical_cpu_delay
-      then (
-       historical_cpu := percent_cpu :: List.take 10 !historical_cpu;
-       historical_cpu_last_time := time
-      );
-
-      (* Display historical CPU time. *)
-      let () =
-       let x, y = historical_cursor in (* Yes, it's a bug in ocaml-curses *)
-       let maxwidth = cols - x in
-       let line =
-         String.concat " "
-           (List.map (sprintf "%2.1f%%") !historical_cpu) in
-       let line = pad maxwidth line in
-       mvaddstr y x line;
-       () in
-
-      (* Write summary data to CSV file.  See also write_csv_header (). *)
-      if !csv_enabled then (
-       (!csv_write) [
-         hostname; printable_time; node_info.C.model; string_of_int nr_pcpus;
-         string_of_int count; string_of_int running; string_of_int blocked;
-         string_of_int paused; string_of_int shutdown; string_of_int shutoff;
-         string_of_int crashed; string_of_int active; string_of_int inactive;
-         sprintf "%2.1f" percent_cpu;
-         Int64.to_string total_memory; Int64.to_string total_domU_memory;
-         Int64.to_string (Int64.of_float total_cpu_time)
-       ]
-      );
-
-      ()
-    in
-
-    (* Update last_info, last_time. *)
-    last_time := time;
-    Hashtbl.clear last_info;
-    List.iter (
-      function
-      | (_, Active rd) ->
-         let info = rd.rd_info, rd.rd_block_stats, rd.rd_interface_stats in
-         Hashtbl.add last_info rd.rd_domid info
-      | _ -> ()
-    ) doms;
+      let y, x = historical_cursor in
+      let maxwidth = cols - x in
+      let line =
+       String.concat " "
+         (List.map (sprintf "%2.1f%%") !historical_cpu) in
+      let line = pad maxwidth line in
+      mvaddstr y x line;
+      () in
+
+    move message_lineno 0; (* Park cursor in message area, as with top. *)
+    refresh ()            (* Refresh the display. *)
 
-    move message_lineno 0 (* Park cursor in message area, as with top. *)
-  in
+(* Write CSV header row. *)
+let write_csv_header () =
+  (!csv_write) (
+    [ "Hostname"; "Time"; "Arch"; "Physical CPUs";
+      "Count"; "Running"; "Blocked"; "Paused"; "Shutdown";
+      "Shutoff"; "Crashed"; "Active"; "Inactive";
+      "%CPU";
+      "Total hardware memory (KB)";
+      "Total memory (KB)"; "Total guest memory (KB)";
+      "Total CPU time (ns)" ] @
+      (* These fields are repeated for each domain: *)
+    [ "Domain ID"; "Domain name"; ] @
+    (if !csv_cpu then [ "CPU (ns)"; "%CPU"; ] else []) @
+    (if !csv_block then [ "Block RDRQ"; "Block WRRQ"; ] else []) @
+    (if !csv_net then [ "Net RXBY"; "Net TXBY" ] else [])
+  )
 
-  let clear_pcpu_display_data () =
-    (* Clear out vcpu_info used by PCPUDisplay
-     * display_mode when we switch back to TaskDisplay mode.
-     *)
-    Hashtbl.clear last_vcpu_info
-  in
+(* Write summary data to CSV file. *)
+let append_csv
+    (_, _, _, _, node_info, hostname, _) (* setup *)
+    (doms,
+     _, printable_time,
+     nr_pcpus, total_cpu, _,
+     totals,
+     _) (* state *) =
+
+  (* The totals / summary fields. *)
+  let (count, running, blocked, paused, shutdown, shutoff,
+       crashed, active, inactive,
+       total_cpu_time, total_memory, total_domU_memory) = totals in
+
+  let percent_cpu = 100. *. total_cpu_time /. total_cpu in
+
+  let summary_fields = [
+    hostname; printable_time; node_info.C.model; string_of_int nr_pcpus;
+    string_of_int count; string_of_int running; string_of_int blocked;
+    string_of_int paused; string_of_int shutdown; string_of_int shutoff;
+    string_of_int crashed; string_of_int active; string_of_int inactive;
+    sprintf "%2.1f" percent_cpu;
+    Int64.to_string node_info.C.memory;
+    Int64.to_string total_memory; Int64.to_string total_domU_memory;
+    Int64.to_string (Int64.of_float total_cpu_time)
+  ] in
 
-  redraw, clear_pcpu_display_data
+  (* The domains.
+   *
+   * Sort them by ID so that the list of relatively stable.  Ignore
+   * inactive domains.
+   *)
+  let doms = List.filter_map (
+    function
+    | _, Inactive -> None              (* Ignore inactive domains. *)
+    | name, Active rd -> Some (name, rd)
+  ) doms in
+  let cmp (_, { rd_domid = rd_domid1 }) (_, { rd_domid = rd_domid2 }) =
+    compare rd_domid1 rd_domid2
+  in
+  let doms = List.sort ~cmp doms in
+
+  let string_of_int64_option = Option.map_default Int64.to_string "" in
+
+  let domain_fields = List.map (
+    fun (domname, rd) ->
+      [ string_of_int rd.rd_domid; domname ] @
+       (if !csv_cpu then [
+          string_of_float rd.rd_cpu_time; string_of_float rd.rd_percent_cpu
+        ] else []) @
+       (if !csv_block then [
+          string_of_int64_option rd.rd_block_rd_reqs;
+          string_of_int64_option rd.rd_block_wr_reqs;
+        ] else []) @
+       (if !csv_net then [
+          string_of_int64_option rd.rd_net_rx_bytes;
+          string_of_int64_option rd.rd_net_tx_bytes;
+        ] else [])
+  ) doms in
+  let domain_fields = List.flatten domain_fields in
+
+  (!csv_write) (summary_fields @ domain_fields)
 
 (* Main loop. *)
-let rec main_loop state =
-  if !csv_enabled then write_csv_header ();
+let rec main_loop ((_, batch_mode, script_mode, csv_enabled, _, _, _)
+                    as setup) =
+  if csv_enabled then write_csv_header ();
 
   while not !quit do
-    redraw state;
-    refresh ();
+    let state = collect setup in               (* Collect stats. *)
+    if not script_mode then redraw setup state; (* Redraw display. *)
+    if csv_enabled then append_csv setup state; (* Update CSV file. *)
 
     (* Clear up unused virDomainPtr objects. *)
     Gc.compact ();
 
-    if not !batch_mode then
-      get_key_press state
-    else (* Batch mode - just sleep, ignore keys. *)
-      Unix.sleep (!delay / 1000);
+    (* Get next key.  This does the sleep. *)
+    if not batch_mode && not script_mode then
+      get_key_press setup;
 
     (* Max iterations? *)
     if !iterations >= 0 then (
       decr iterations;
       if !iterations = 0 then quit := true
     );
+
+    (* End time? *)
+    (match !end_time with
+     | None -> ()
+     | Some end_time ->
+        let (_, time, _, _, _, _, _, _) = state in
+        let delay_secs = float !delay /. 1000. in
+        if end_time <= time +. delay_secs then quit := true
+    );
+
+    (* Batch mode or script mode.  We didn't call get_key_press above, so
+     * we didn't sleep.  Sleep now, unless we are about to quit.
+     *)
+    if batch_mode || script_mode then
+      if not !quit then
+       usleep !delay;
   done
 
-and get_key_press state =
+and get_key_press setup =
   (* Read the next key, waiting up to !delay milliseconds. *)
   timeout !delay;
   let k = getch () in
@@ -1166,7 +1315,7 @@ and get_key_press state =
   if k >= 0 && k <> 32 (* ' ' *) && k <> 12 (* ^L *) && k <> Key.resize
   then (
     if k = Char.code 'q' then quit := true
-    else if k = Char.code 'h' then show_help state
+    else if k = Char.code 'h' then show_help setup
     else if k = Char.code 's' || k = Char.code 'd' then change_delay ()
     else if k = Char.code 'M' then sort_order := Memory
     else if k = Char.code 'P' then sort_order := Processor
@@ -1177,32 +1326,35 @@ and get_key_press state =
     else if k = Char.code '1' then toggle_pcpu_display ()
     else if k = Char.code '2' then toggle_net_display ()
     else if k = Char.code '3' then toggle_block_display ()
+    else if k = Char.code 'W' then write_init_file ()
     else unknown_command k
   )
 
 and change_delay () =
-  print_msg (sprintf "Change delay from %.1f to: " (float !delay /. 1000.));
+  print_msg
+    (sprintf (f_"Change delay from %.1f to: ") (float !delay /. 1000.));
   let str = get_string 16 in
   (* Try to parse the number. *)
   let error =
     try
       let newdelay = float_of_string str in
       if newdelay <= 0. then (
-       print_msg "Delay must be > 0"; true
+       print_msg (s_"Delay must be > 0"); true
       ) else (
        delay := int_of_float (newdelay *. 1000.); false
       )
     with
       Failure "float_of_string" ->
-       print_msg "Not a valid number"; true in
+       print_msg (s_"Not a valid number"); true in
+  refresh ();
   sleep (if error then 2 else 1)
 
 and change_sort_order () =
   clear ();
   let lines, cols = get_size () in
 
-  mvaddstr top_lineno 0 "Set sort order for main display";
-  mvaddstr summary_lineno 0 "Type key or use up and down cursor keys.";
+  mvaddstr top_lineno 0 (s_"Set sort order for main display");
+  mvaddstr summary_lineno 0 (s_"Type key or use up and down cursor keys.");
 
   attron A.reverse;
   mvaddstr header_lineno 0 (pad cols "KEY   Sort field");
@@ -1284,7 +1436,10 @@ and change_sort_order () =
         sort_order := new_order;
         print_msg (sprintf "Sort order changed to: %s"
                      (printable_sort_order new_order));
-        if not loop then sleep 1
+        if not loop then (
+          refresh ();
+          sleep 1
+        )
     );
 
     if loop then change_sort_order ()
@@ -1317,7 +1472,84 @@ and toggle_block_display () =            (* key 3 *)
     | TaskDisplay | NetDisplay -> BlockDisplay
     | BlockDisplay -> TaskDisplay
 
-and show_help (_, _, _, hostname,
+(* Write an init file. *)
+and write_init_file () =
+  match !init_file with
+  | NoInitFile -> ()                   (* Do nothing if --no-init-file *)
+  | DefaultInitFile ->
+      let home = try Sys.getenv "HOME" with Not_found -> "/" in
+      let filename = home // rcfile in
+      _write_init_file filename
+  | InitFile filename ->
+      _write_init_file filename
+
+and _write_init_file filename =
+  try
+    (* Create the new file as filename.new. *)
+    let chan = open_out (filename ^ ".new") in
+
+    let time = Unix.gettimeofday () in
+    let tm = Unix.localtime time in
+    let printable_date_time =
+      sprintf "%04d-%02d-%02d %02d:%02d:%02d"
+       (tm.Unix.tm_year + 1900) (tm.Unix.tm_mon+1) tm.Unix.tm_mday
+       tm.Unix.tm_hour tm.Unix.tm_min tm.Unix.tm_sec in
+    let username =
+      try
+       let uid = Unix.geteuid () in
+       (Unix.getpwuid uid).Unix.pw_name
+      with
+       Not_found -> "unknown" in
+
+    let fp = fprintf in
+    let nl () = fp chan "\n" in
+    let () = fp chan (f_"# %s virt-top configuration file\n") rcfile in
+    let () = fp chan (f_"# generated on %s by %s\n") printable_date_time username in
+    nl ();
+    fp chan "display %s\n" (cli_of_display !display_mode);
+    fp chan "delay %g\n" (float !delay /. 1000.);
+    fp chan "hist-cpu %d\n" !historical_cpu_delay;
+    if !iterations <> -1 then fp chan "iterations %d\n" !iterations;
+    fp chan "sort %s\n" (cli_of_sort_order !sort_order);
+    (match !uri with
+     | None -> ()
+     | Some uri -> fp chan "connect %s\n" uri
+    );
+    if !batch_mode = true then fp chan "batch true\n";
+    if !secure_mode = true then fp chan "secure true\n";
+    nl ();
+    output_string chan (s_"# To send debug and error messages to a file, uncomment next line\n");
+    fp chan "#debug virt-top.out\n";
+    nl ();
+    output_string chan (s_"# Enable CSV output to the named file\n");
+    fp chan "#csv virt-top.csv\n";
+    nl ();
+    output_string chan (s_"# To protect this file from being overwritten, uncomment next line\n");
+    fp chan "#overwrite-init-file false\n";
+
+    close_out chan;
+
+    (* If the file exists, rename it as filename.old. *)
+    (try Unix.rename filename (filename ^ ".old")
+     with Unix.Unix_error _ -> ());
+
+    (* Rename filename.new to filename. *)
+    Unix.rename (filename ^ ".new") filename;
+
+    print_msg (sprintf (f_"Wrote settings to %s") filename);
+    refresh ();
+    sleep 2
+  with
+  | Sys_error err ->
+      print_msg (s_"Error" ^ ": " ^ err);
+      refresh (); sleep 2
+  | Unix.Unix_error (err, fn, str) ->
+      print_msg (s_"Error" ^ ": " ^
+                  (Unix.error_message err) ^ " " ^ fn ^ " " ^ str);
+      refresh ();
+      sleep 2
+
+and show_help (_, _, _, _, _, hostname,
               (libvirt_major, libvirt_minor, libvirt_release)) =
   clear ();
 
@@ -1326,8 +1558,10 @@ and show_help (_, _, _, hostname,
 
   (* Banner at the top of the screen. *)
   let banner =
-    sprintf "virt-top %s (libvirt %d.%d.%d) by Red Hat"
-      Libvirt_version.version libvirt_major libvirt_minor libvirt_release in
+    sprintf (f_"virt-top %s ocaml-libvirt %s libvirt %d.%d.%d by Red Hat")
+      Virt_top_version.version
+      Libvirt_version.version
+      libvirt_major libvirt_minor libvirt_release in
   let banner = pad cols banner in
   attron A.reverse;
   mvaddstr 0 0 banner;
@@ -1335,18 +1569,20 @@ and show_help (_, _, _, hostname,
 
   (* Status. *)
   mvaddstr 1 0
-    (sprintf "Delay: %.1f secs; Batch: %s; Secure: %s; Sort: %s"
+    (sprintf
+       (f_"Delay: %.1f secs; Batch: %s; Secure: %s; Sort: %s")
        (float !delay /. 1000.)
-       (if !batch_mode then "On" else "Off")
-       (if !secure_mode then "On" else "Off")
+       (if !batch_mode then s_"On" else s_"Off")
+       (if !secure_mode then s_"On" else s_"Off")
        (printable_sort_order !sort_order));
   mvaddstr 2 0
-    (sprintf "Connect: %s; Hostname: %s"
-       (match !uri with None -> "default" | Some s -> s)
+    (sprintf
+       (f_"Connect: %s; Hostname: %s")
+       (match !uri with None -> s_"default" | Some s -> s)
        hostname);
 
   (* Misc keys on left. *)
-  let banner = pad 38 "MAIN KEYS" in
+  let banner = pad 38 (s_"MAIN KEYS") in
   attron A.reverse;
   mvaddstr header_lineno 1 banner;
   attroff A.reverse;
@@ -1358,28 +1594,28 @@ and show_help (_, _, _, hostname,
   let key keys description =
     let lineno = get_lineno () in
     move lineno 1; attron A.bold; addstr keys; attroff A.bold;
-    move lineno 10; addstr description; ()
+    move lineno 10; addstr description
   in
-  key "space ^L" "Update display";
-  key "q"        "Quit";
-  key "d s"      "Set update interval";
-  key "h"        "Help";
+  key "space ^L" (s_"Update display");
+  key "q"        (s_"Quit");
+  key "d s"      (s_"Set update interval");
+  key "h"        (s_"Help");
 
   (* Sort order. *)
   ignore (get_lineno ());
-  let banner = pad 38 "SORTING" in
+  let banner = pad 38 (s_"SORTING") in
   attron A.reverse;
   mvaddstr (get_lineno ()) 1 banner;
   attroff A.reverse;
 
-  key "P" "Sort by %CPU";
-  key "M" "Sort by %MEM";
-  key "T" "Sort by TIME";
-  key "N" "Sort by ID";
-  key "F" "Select sort field";
+  key "P" (s_"Sort by %CPU");
+  key "M" (s_"Sort by %MEM");
+  key "T" (s_"Sort by TIME");
+  key "N" (s_"Sort by ID");
+  key "F" (s_"Select sort field");
 
   (* Display modes on right. *)
-  let banner = pad 39 "DISPLAY MODES" in
+  let banner = pad 39 (s_"DISPLAY MODES") in
   attron A.reverse;
   mvaddstr header_lineno 40 banner;
   attroff A.reverse;
@@ -1391,19 +1627,20 @@ and show_help (_, _, _, hostname,
   let key keys description =
     let lineno = get_lineno () in
     move lineno 40; attron A.bold; addstr keys; attroff A.bold;
-    move lineno 49; addstr description; ()
+    move lineno 49; addstr description
   in
-  key "0" "Domains display";
-  key "1" "Toggle physical CPUs";
-  key "2" "Toggle network interfaces";
-  key "3" "Toggle block devices";
+  key "0" (s_"Domains display");
+  key "1" (s_"Toggle physical CPUs");
+  key "2" (s_"Toggle network interfaces");
+  key "3" (s_"Toggle block devices");
 
   (* Update screen and wait for key press. *)
   mvaddstr (lines-1) 0
-    "More help in virt-top(1) man page. Press any key to return.";
+    (s_"More help in virt-top(1) man page. Press any key to return.");
   refresh ();
   ignore (getch ())
 
 and unknown_command k =
-  print_msg "Unknown command - try 'h' for help";
+  print_msg (s_"Unknown command - try 'h' for help");
+  refresh ();
   sleep 1