X-Git-Url: http://git.annexia.org/?p=virt-top.git;a=blobdiff_plain;f=virt-top%2Fvirt_top.ml;h=631d04fd475de67455d05a02bd73b8dbbd75ef41;hp=a621a182d03ea97921db7e81ce9b1cf4c0951fc3;hb=6116cbd975924cc971e1eff565fee63ced296d5c;hpb=17777061b8970d1bd528e2b5ddf35da68808326b diff --git a/virt-top/virt_top.ml b/virt-top/virt_top.ml index a621a18..631d04f 100644 --- a/virt-top/virt_top.ml +++ b/virt-top/virt_top.ml @@ -6,6 +6,8 @@ open Printf open ExtList open Curses +open Virt_top_utils + module C = Libvirt.Connect module D = Libvirt.Domain module N = Libvirt.Network @@ -32,6 +34,7 @@ let (-^) = Int64.sub let ( *^ ) = Int64.mul let (/^) = Int64.div +(* Sort order. *) type sort_order = | DomainID | DomainName | Processor | Memory | Time | NetRX | NetTX | BlockRdRq | BlockWrRq @@ -49,10 +52,44 @@ let printable_sort_order = function | NetTX -> "Net TX bytes" | BlockRdRq -> "Block read reqs" | BlockWrRq -> "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 -> failwith (str ^ ": sort order should be: 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 -> failwith (str ^ ": display should be 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 *) @@ -65,6 +102,8 @@ let display_mode = ref TaskDisplay let uri = ref None let debug_file = ref "" let csv_enabled = ref false +let init_file = ref DefaultInitFile +let script_mode = ref false (* Function to read command line arguments and go into curses mode. *) let start_up () = @@ -74,22 +113,15 @@ let start_up () = failwith "-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 in let argspec = Arg.align [ "-1", Arg.Unit set_pcpu_mode, " Start by displaying pCPUs (default: tasks)"; @@ -102,9 +134,12 @@ let start_up () = "-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"; + "--init-file", Arg.String set_init_file, "file Set name of init file"; + "--no-init-file", Arg.Unit no_init_file, " Do not read init file"; "-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"; + "--script", Arg.Set script_mode, " Run from a script (no user interface)"; ] in let anon_fun str = raise (Arg.Bad (str ^ ": unknown parameter")) in let usage_msg = "virt-top : a 'top'-like utility for virtualization @@ -115,6 +150,38 @@ 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 + | _, "batch", b -> batch_mode := bool_of_string b + | _, "secure", b -> secure_mode := bool_of_string b + | _, "script", b -> script_mode := bool_of_string b + | _, "overwrite-init-file", "false" -> no_init_file () + | lineno, key, _ -> + eprintf "%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 // ".virt-toprc" 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. *) @@ -137,6 +204,9 @@ OPTIONS" in let hostname = try C.get_hostname conn with + (* qemu:/// and other URIs didn't support virConnectGetHostname until + * libvirt 0.3.3. Before that they'd throw a virterror. *) + | Libvirt.Virterror _ | Invalid_argument "virConnectGetHostname not supported" -> "unknown" in let libvirt_version = @@ -147,13 +217,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,7 +234,7 @@ OPTIONS" in ); (* Curses voodoo (see ncurses(3)). *) - let stdscr = + if not !script_mode then ( initscr (); cbreak (); noecho (); @@ -172,13 +242,16 @@ OPTIONS" in let stdscr = stdscr () in intrflush stdscr false; keypad stdscr true; - stdscr in + () + ); - (* 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 + (conn, + !batch_mode, !script_mode, !csv_enabled, (* immutable modes *) + node_info, hostname, libvirt_version) (* Show a percentage in 4 chars. *) let show_percent percent = @@ -345,7 +418,7 @@ let write_csv_header () = "Total CPU time ns" ] (* 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 +444,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 +472,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 @@ -426,15 +497,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. *) @@ -581,6 +643,174 @@ 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 + + (* 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. *) @@ -696,87 +926,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 = @@ -871,7 +1024,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) @@ -892,7 +1045,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 -> @@ -979,7 +1132,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) @@ -1000,7 +1153,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 -> @@ -1034,127 +1187,85 @@ 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 "%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 ); - (* 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; - - move message_lineno 0 (* Park cursor in message area, as with top. *) - 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 - - redraw, clear_pcpu_display_data + 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 + + move message_lineno 0; (* Park cursor in message area, as with top. *) + refresh (); (* Refresh the display. *) + () + +(* Write summary data to CSV file. See also write_csv_header (). *) +let append_csv + (_, _, _, _, node_info, hostname, _) (* setup *) + (_, + _, printable_time, + nr_pcpus, total_cpu, _, + totals, + _) (* state *) = + 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 + + (!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) + ] (* 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. *) + if not batch_mode && not script_mode then + get_key_press setup + else (* Batch mode or script mode - just sleep, ignore keys. *) Unix.sleep (!delay / 1000); (* Max iterations? *) @@ -1164,7 +1275,7 @@ let rec main_loop state = ); 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 @@ -1173,7 +1284,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 @@ -1184,6 +1295,7 @@ 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 ) @@ -1324,7 +1436,78 @@ 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 // ".virt-toprc" 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 + fp chan "# .virt-toprc virt-top configuration file\n"; + fp chan "# generated on %s by %s\n" printable_date_time username; + 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 (); + fp chan "# To send debug and error messages to a file, uncomment next line\n"; + fp chan "#debug virt-top.out\n"; + nl (); + fp chan "# Enable CSV output to the named file\n"; + fp chan "#csv virt-top.csv\n"; + nl (); + fp chan "# 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 "Wrote settings to %s" filename); sleep 2 + with + | Sys_error err -> print_msg "Error: %s"; sleep 2 + | Unix.Unix_error (err, fn, str) -> + print_msg (sprintf "Error: %s %s %s" (Unix.error_message err) fn str); + sleep 2 + +and show_help (_, _, _, _, _, hostname, (libvirt_major, libvirt_minor, libvirt_release)) = clear ();