ps/virt_ps.ml

   1 (* Memory info for virtual domains.
   2    (C) Copyright 2008 Richard W.M. Jones, Red Hat Inc.
   3    http://libvirt.org/
   4
   5    This program is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 2 of the License, or
   8    (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *)
  19
  20 open Printf
  21
  22 open Virt_mem_gettext.Gettext
  23 open Virt_mem_utils
  24 open Virt_mem_types
  25
  26 open Kernel_task_struct
  27
  28 let run debug { domname = domname; mem = mem }
  29     { utsname = utsname; tasks = tasks } =
  30   let utsname = Option.get utsname in
  31   let kernel_version = utsname.kernel_release in
  32   let init_task_addr = Option.get tasks in
  33
  34   (* Starting at init_task, navigate through the linked list of
  35    * tasks (through tasks.next).  The main program has already made
  36    * sure these are mapped into memory.
  37    *)
  38   let tasks =
  39     let rec loop acc task =
  40       let next = task.task_struct_tasks'next in
  41       if next <> init_task_addr then (
  42         let task = get_task_struct kernel_version mem next in
  43         let task = {
  44           task with
  45             task_struct_comm = truncate_c_string task.task_struct_comm
  46         } in
  47         let acc = task :: acc in
  48         loop acc task
  49       ) else
  50         acc
  51     in
  52     loop [] (get_task_struct kernel_version mem init_task_addr) in
  53
  54   (* Sort tasks by PID. *)
  55   let cmp { task_struct_pid = p1 } { task_struct_pid = p2 } = compare p1 p2 in
  56   let tasks = List.sort cmp tasks in
  57
  58   printf "  PID STAT COMMAND\n";
  59
  60   List.iter (
  61     fun task ->
  62       printf "%5Ld      %s\n" task.task_struct_pid task.task_struct_comm
  63   ) tasks
  64
  65 let summary = s_"list processes in virtual machine"
  66 let description = s_"\
  67 virt-ps prints a process listing for virtual machines running under
  68 libvirt."
  69
  70 let () =
  71   Virt_mem.register "ps" summary description
  72     ~needs_utsname:true ~needs_tasks:true ~run