virt-top/virt_top_main.ml

   1 (* 'top'-like tool for libvirt domains.
   2    (C) Copyright 2007-2009 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    Just contains the main function.
  20 *)
  21
  22 open Curses
  23
  24 open Virt_top_gettext.Gettext
  25 open Virt_top
  26
  27 (* Note: make sure we catch any exceptions and clean up the display.
  28  *
  29  * Note (2): make sure all exit paths call the GC so that we can check
  30  * that all allocated resources are being counted properly (by running
  31  * the program under --debug ...).
  32  *)
  33
  34 let () = Printexc.record_backtrace true
  35
  36 let ((_, _, script_mode, _, stream_mode, _, _, _) as setup) =
  37   try start_up ()
  38   with
  39   | Failure msg ->
  40       prerr_endline msg;
  41       Printexc.print_backtrace stderr;
  42       exit 1
  43   | exn ->
  44       prerr_endline (s_ "Error" ^ ": " ^ Printexc.to_string exn);
  45       Printexc.print_backtrace stderr;
  46       exit 1
  47
  48 let () =
  49   try
  50     main_loop setup;
  51     if not script_mode && not stream_mode then endwin ();
  52     Gc.compact ();
  53     exit 0
  54   with
  55   | Libvirt.Virterror err ->
  56       if not script_mode && not stream_mode then endwin ();
  57       prerr_endline (Libvirt.Virterror.to_string err);
  58       Printexc.print_backtrace stderr;
  59       Gc.compact ();
  60       exit 1
  61   | Failure msg ->
  62       if not script_mode && not stream_mode then endwin ();
  63       prerr_endline msg;
  64       Printexc.print_backtrace stderr;
  65       Gc.compact ();
  66       exit 1
  67   | exn ->
  68       if not script_mode && not stream_mode then endwin ();
  69       prerr_endline (s_ "Error" ^ ": " ^ Printexc.to_string exn);
  70       Printexc.print_backtrace stderr;
  71       Gc.compact ();
  72       exit 1