X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;f=virt-top%2Fvirt_top_main.ml;h=e8c4425c3754a6665eef67e0ea1412daa6cba668;hb=4a2d0ccd91f07d0a2009e8553a29fcf4cf752ba3;hp=58413629bbcb35e76bda36b80ff9cff76453b828;hpb=a8b837d5018c488a130fcbea425904817a862210;p=virt-top.git

diff --git a/virt-top/virt_top_main.ml b/virt-top/virt_top_main.ml
index 5841362..e8c4425 100644
--- a/virt-top/virt_top_main.ml
+++ b/virt-top/virt_top_main.ml
@@ -1,11 +1,27 @@
 (* 'top'-like tool for libvirt domains.
- * $Id: virt_top_main.ml,v 1.2 2007/08/30 13:16:57 rjones Exp $
- *
- * Just contains the main function.
- *)
+   (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.
+
+   Just contains the main function.
+*)
 
 open Curses
 
+open Virt_top_gettext.Gettext
 open Virt_top
 
 (* Note: make sure we catch any exceptions and clean up the display.
@@ -14,24 +30,43 @@ open Virt_top
  * that all allocated resources are being counted properly (by running
  * the program under --debug ...).
  *)
-let error =
-  let state = start_up () in
 
+let () = Printexc.record_backtrace true
+
+let ((_, _, script_mode, _, stream_mode, _, _, _) as setup) =
+  try start_up ()
+  with
+  | Failure msg ->
+      prerr_endline msg;
+      Printexc.print_backtrace stderr;
+      exit 1
+  | exn ->
+      prerr_endline (s_ "Error" ^ ": " ^ Printexc.to_string exn);
+      Printexc.print_backtrace stderr;
+      exit 1
+
+let () =
   try
-    main_loop state;
-    endwin ();
-    false
+    main_loop setup;
+    if not script_mode && not stream_mode then endwin ();
+    Gc.compact ();
+    exit 0
   with
   | Libvirt.Virterror err ->
-      endwin ();
+      if not script_mode && not stream_mode then endwin ();
       prerr_endline (Libvirt.Virterror.to_string err);
-      true
+      Printexc.print_backtrace stderr;
+      Gc.compact ();
+      exit 1
+  | Failure msg ->
+      if not script_mode && not stream_mode then endwin ();
+      prerr_endline msg;
+      Printexc.print_backtrace stderr;
+      Gc.compact ();
+      exit 1
   | exn ->
-      endwin ();
-      prerr_endline ("Error: " ^ Printexc.to_string exn);
-      true
-
-let () =
-  Gc.compact (); (* See note above. *)
-
-  exit (if error then 1 else 0)
+      if not script_mode && not stream_mode then endwin ();
+      prerr_endline (s_ "Error" ^ ": " ^ Printexc.to_string exn);
+      Printexc.print_backtrace stderr;
+      Gc.compact ();
+      exit 1