Get rid of lookup_ksym function, replace with a map.

[virt-mem.git] / lib / virt_mem_capture.ml

diff --git a/lib/virt_mem_capture.ml b/lib/virt_mem_capture.ml
index 154f76d..0c1ff57 100644 (file)
--- a/lib/virt_mem_capture.ml
+++ b/lib/virt_mem_capture.ml
@@ -22,6 +22,9 @@
 open Printf
 open ExtString
 
+module D = Libvirt.Domain
+
+open Virt_mem_types
 open Virt_mem_gettext.Gettext
 
 (* This will contain what is passed by the user as '-o' option. *)
@@ -51,11 +54,11 @@ let rec beforeksyms debug = function
        * is the domain ID (if known) or a mangled domain name.
        *)
       List.iter (
-       fun ((domid, domname, _, _) as image) ->
+       fun ({ dom = dom; domname = domname } as image) ->
          let filename =
            !output_filename ^ "." ^
-           match domid with
-           | Some id -> string_of_int id
+           match dom with
+           | Some dom -> string_of_int (D.get_id dom)
            | None ->
                let f = function
                  | ('a'..'z'|'A'..'Z'|'0'..'9'|'_' as c) -> String.make 1 c
@@ -65,7 +68,7 @@ let rec beforeksyms debug = function
          save_image image filename
       ) images
 
-and save_image (_, domname, arch, mem) filename =
+and save_image { domname = domname } filename =
   assert false;
 
   let chan = open_out filename in