Remove bogus '() with' (thanks Bluestorm).
[virt-mem.git] / extract / codegen / kerneldb_to_parser.ml
index e7b4142..f94de2f 100644 (file)
    and fields we try to parse.
 *)
 
-let what = [
-  "task_struct", (
-    "struct task_struct {", "};", true,
-    [ "state"; "prio"; "normal_prio"; "static_prio";
-      "tasks'prev"; "tasks'next"; "comm"]
-  );
+type struct_t = {
+  opener : string;     (* String in pa_hole file which starts this struct. *)
+  closer : string;     (* String in pa_hole file which ends this struct. *)
+  mandatory_struct : bool; (* Is this struct mandatory? *)
+  fields : (string * field_t) list;   (* List of interesting fields. *)
+}
+and field_t = {
+  mandatory_field : bool;  (* Is this field mandatory? *)
+}
+
+let structs = [
+  "task_struct", {
+    opener = "struct task_struct {"; closer = "};"; mandatory_struct = true;
+    fields = [
+      "state",       { mandatory_field = true };
+      "prio",        { mandatory_field = true };
+      "normal_prio", { mandatory_field = true };
+      "static_prio", { mandatory_field = true };
+      "tasks'prev",  { mandatory_field = true };
+      "tasks'next",  { mandatory_field = true };
+      "mm",          { mandatory_field = true };
+      "active_mm",   { mandatory_field = true };
+      "comm",        { mandatory_field = true };
+      "pid",         { mandatory_field = true };
+    ]
+  };
 (*
   "mm_struct", (
     "struct mm_struct {", "};", true,
     [ ]
   );
 *)
-  "net_device", (
-    "struct net_device {", "};", true,
-    [ "name"; "dev_addr" ]
-  );
+  "net_device", {
+    opener = "struct net_device {"; closer = "};"; mandatory_struct = true;
+    fields = [
+      "dev_list'prev", { mandatory_field = false };
+      "dev_list'next", { mandatory_field = false };
+      "next",          { mandatory_field = false };
+      "name",          { mandatory_field = true };
+      "dev_addr",      { mandatory_field = true };
+    ]
+  };
+  "net", {
+    opener = "struct net {"; closer = "};"; mandatory_struct = false;
+    fields = [
+      "dev_base_head'next", { mandatory_field = true };
+    ]
+  };
 ]
 
-let debug = true
+let debug = false
 
 open Camlp4.PreCast
 open Syntax
@@ -58,6 +90,56 @@ open Printf
 
 let (//) = Filename.concat
 
+(* Couple of handy camlp4 construction functions which do some
+ * things that ought to be easy/obvious but aren't.
+ *
+ * 'concat_str_items' concatenates a list of str_item together into
+ * one big str_item.
+ *
+ * 'concat_record_fields' concatenates a list of records fields into
+ * a record.  The list must have at least one element.
+ *
+ * 'build_record' builds a record out of record fields.
+ * 
+ * 'build_tuple_from_exprs' builds an arbitrary length tuple from
+ * a list of expressions of length >= 2.
+ *
+ * Thanks to bluestorm on #ocaml for getting these working.
+ *)
+let concat_str_items _loc items =
+  match items with
+  | [] -> <:str_item< >>
+  | x :: xs ->
+      List.fold_left (fun xs x -> <:str_item< $xs$ $x$ >>) x xs
+
+let concat_sig_items _loc items =
+  match items with
+  | [] -> <:sig_item< >>
+  | x :: xs ->
+      List.fold_left (fun xs x -> <:sig_item< $xs$ $x$ >>) x xs
+
+let concat_record_fields _loc fields =
+  match fields with
+    | [] -> assert false
+    | f :: fs ->
+       List.fold_left (fun fs f -> <:ctyp< $fs$ ; $f$ >>) f fs
+
+let concat_record_bindings _loc rbs =
+  match rbs with
+    | [] -> assert false
+    | rb :: rbs ->
+       List.fold_left (fun rbs rb -> <:rec_binding< $rbs$ ; $rb$ >>) rb rbs
+
+let build_record _loc rbs =
+  Ast.ExRec (_loc, rbs, Ast.ExNil _loc)
+
+let build_tuple_from_exprs _loc exprs =
+  match exprs with
+  | [] | [_] -> assert false
+  | x :: xs ->
+      Ast.ExTup (_loc,
+                List.fold_left (fun xs x -> Ast.ExCom (_loc, x, xs)) x xs)
+
 let () =
   let args = Array.to_list Sys.argv in
 
@@ -80,7 +162,7 @@ Example (from toplevel of virt-mem source tree):
   let infos = Sys.readdir kernelsdir in
   let infos = Array.to_list infos in
   let infos = List.filter (fun name -> String.ends_with name ".info") infos in
-  let infos = List.map ((//) kernelsdir) infos in
+  let infos = List.map ( (//) kernelsdir) infos in
 
   (* Regular expressions.  We really really should use ocaml-mikmatch ... *)
   let re_oldformat = Pcre.regexp "^RPM: \\d+: \\(build \\d+\\) ([-\\w]+) ([\\w.]+) ([\\w.]+) \\(.*?\\) (\\w+)" in
@@ -146,16 +228,20 @@ Example (from toplevel of virt-mem source tree):
       (basename, version, arch)
   ) infos in
 
+  let nr_kernels = List.length infos in
+
   (* For quick access to the opener strings, build a hash. *)
   let openers = Hashtbl.create 13 in
   List.iter (
-    fun (name, (opener, closer, _, _)) ->
+    fun (name, { opener = opener; closer = closer }) ->
       Hashtbl.add openers opener (closer, name)
-  ) what;
+  ) structs;
 
   (* Now read the data files and parse out the structures of interest. *)
-  let datas = List.map (
-    fun (basename, version, arch) ->
+  let kernels = List.mapi (
+    fun i (basename, version, arch) ->
+      printf "Loading kernel data file %d/%d\r%!" (i+1) nr_kernels;
+
       let file_exists name =
        try Unix.access name [Unix.F_OK]; true
        with Unix.Unix_error _ -> false
@@ -222,10 +308,10 @@ Example (from toplevel of virt-mem source tree):
 
       (* Make sure we got all the mandatory structures. *)
       List.iter (
-        fun (name, (_, _, mandatory, _)) ->
+        fun (name, { mandatory_struct = mandatory }) ->
           if mandatory && not (Hashtbl.mem bodies name) then
             failwith (sprintf "%s: structure %s not found in this kernel" basename name)
-      ) what;
+      ) structs;
 
       (basename, version, arch, bodies)
   ) infos in
@@ -286,9 +372,9 @@ Example (from toplevel of virt-mem source tree):
        match struct_name with
        | None -> nested_fields
        | Some prefix ->
-           List.map (
-             fun (name, details) -> (prefix ^ "'" ^ name, details)
-           ) nested_fields in
+            List.map (
+              fun (name, details) -> (prefix ^ "'" ^ name, details)
+            ) nested_fields in
 
       (* Parse the rest. *)
       nested_fields @ parse basename rest
@@ -315,7 +401,8 @@ Example (from toplevel of virt-mem source tree):
         let subs = Pcre.exec ~rex:re_offsetsize line in
         let offset = int_of_string (Pcre.get_substring subs 1) in
         let size = int_of_string (Pcre.get_substring subs 2) in
-        (name, (`Ptr struct_name, offset, size)) :: parse basename lines
+        (name, (`Ptr struct_name, offset, size))
+          :: parse basename lines
        with
         Not_found -> parse basename lines
       );
@@ -329,7 +416,8 @@ Example (from toplevel of virt-mem source tree):
         let subs = Pcre.exec ~rex:re_offsetsize line in
         let offset = int_of_string (Pcre.get_substring subs 1) in
         let size = int_of_string (Pcre.get_substring subs 2) in
-        (name, (`Str width, offset, size)) :: parse basename lines
+        (name, (`Str width, offset, size))
+          :: parse basename lines
        with
         Not_found -> parse basename lines
       );
@@ -340,12 +428,13 @@ Example (from toplevel of virt-mem source tree):
 
   in
 
-  let datas = List.map (
+  let kernels = List.map (
     fun (basename, version, arch, bodies) ->
       let structures = List.filter_map (
-       fun (name, (_, _, _, wanted_fields)) ->
+       fun (struct_name, { fields = wanted_fields }) ->
          let body =
-           try Some (Hashtbl.find bodies name) with Not_found -> None in
+           try Some (Hashtbl.find bodies struct_name)
+           with Not_found -> None in
          match body with
          | None -> None
          | Some body ->
@@ -363,21 +452,26 @@ Example (from toplevel of virt-mem source tree):
               * the wanted_fields.
               *)
              let fields = List.filter (
-               fun (name, _) -> List.mem name wanted_fields
+               fun (name, _) -> List.mem_assoc name wanted_fields
              ) fields in
 
-             (* Also check we have all the wanted fields. *)
+             (* Also check we have all the mandatory fields. *)
              List.iter (
-               fun wanted_field ->
-                 if not (List.mem_assoc wanted_field fields) then
-                   failwith (sprintf "%s: structure %s is missing required field %s" basename name wanted_field)
+               fun (wanted_field, { mandatory_field = mandatory }) ->
+                 if mandatory && not (List.mem_assoc wanted_field fields) then
+                   failwith (sprintf "%s: structure %s is missing required field %s" basename struct_name wanted_field)
              ) wanted_fields;
 
-             Some (name, (fields, total_size))
-      ) what in
+             (* Prefix all the field names with the structure name. *)
+             let fields =
+               List.map (fun (name, details) ->
+                           struct_name ^ "_" ^ name, details) fields in
+
+             Some (struct_name, (fields, total_size))
+      ) structs in
 
       (basename, version, arch, structures)
-  ) datas in
+  ) kernels in
 
   if debug then
     List.iter (
@@ -400,50 +494,109 @@ Example (from toplevel of virt-mem source tree):
            ) fields;
            printf "  } /* %d bytes */\n\n" total_size;
        ) structures;
-    ) datas;
+    ) kernels;
+
+  (* First output file is a simple list of kernels, to support the
+   * 'virt-mem --list-kernels' option.
+   *)
+  let () =
+    let _loc = Loc.ghost in
+
+    let versions = List.map (fun (_, version, _, _) -> version) kernels in
+
+    (* Sort them in reverse because we are going to generate the
+     * final list in reverse.
+     *)
+    let cmp a b = compare b a in
+    let versions = List.sort ~cmp versions in
+
+    let xs =
+      List.fold_left (fun xs version -> <:expr< $str:version$ :: $xs$ >>)
+      <:expr< [] >> versions in
+
+    let code = <:str_item<
+      let kernels = $xs$
+    >> in
+
+    let output_file = outputdir // "virt_mem_kernels.ml" in
+    printf "Writing list of kernels to %s ...\n%!" output_file;
+    Printers.OCaml.print_implem ~output_file code in
 
   (* We'll generate a code file for each structure type (eg. task_struct
-   * across all kernel versions), so rearrange 'datas' for that purpose.
+   * across all kernel versions), so rearrange 'kernels' for that purpose.
    *
    * XXX This loop is O(n^3), luckily n is small!
    *)
   let files =
     List.map (
       fun (name, _) ->
-       name,
-       List.filter_map (
-         fun (basename, version, arch, structures) ->
-           try Some (basename, version, arch, List.assoc name structures)
-           with Not_found -> None
-       ) datas
-    ) what in
+       let kernels =
+         List.filter_map (
+           fun (basename, version, arch, structures) ->
+             try Some (basename, version, arch, List.assoc name structures)
+             with Not_found -> None
+         ) kernels in
 
-  let datas = () in ignore datas; (* garbage collect *)
+       (* Sort the kernels, which makes the generated output more stable
+        * and makes patches more useful.
+        *)
+       let kernels = List.sort kernels in
 
-  (* Get just the field types.  It's plausible that a field with the
-   * same name has a different type between kernel versions, so we must
-   * check that didn't happen.
+       name, kernels
+    ) structs in
+
+  let kernels = () in ignore kernels; (* garbage collect *)
+
+  (* Get just the field types.
+   *
+   * It's plausible that a field with the same name has a different
+   * type between kernel versions, so we must check that didn't
+   * happen.
+   *
+   * This is complicated because of non-mandatory fields, which don't
+   * appear in every kernel version.
    *)
   let files = List.map (
     fun (struct_name, kernels) ->
       let field_types =
-       match kernels with
-       | [] -> []
-       | (_, _, _, (fields, _)) :: kernels ->
-           let field_types_of_fields fields =
-             List.sort (
-               List.map (
-                 fun (field_name, (typ, _, _)) -> field_name, typ
-               ) fields
-             )
-           in
-           let field_types = field_types_of_fields fields in
-           List.iter (
-             fun (_, _, _, (fields, _)) ->
-               if field_types <> field_types_of_fields fields then
-                 failwith (sprintf "%s: one of the structure fields changed type between kernel versions" struct_name)
-           ) kernels;
-           field_types in
+       (* Get the list of fields expected in this structure. *)
+       let { fields = struct_fields } = List.assoc struct_name structs in
+
+       (* Get the list of fields that we found in each kernel version. *)
+       let found_fields =
+         List.flatten
+           (List.map (fun (_, _, _, (fields, _)) -> fields) kernels) in
+
+       (* Determine a hash from each field name to the type.  As we add
+        * fields, we might get a conflicting type (meaning the type
+        * changed between kernel versions).
+        *)
+       let hash = Hashtbl.create 13 in
+
+       List.iter (
+         fun (field_name, (typ, _, _)) ->
+           try
+             let field_type = Hashtbl.find hash field_name in
+             if typ <> field_type then
+               failwith (sprintf "%s.%s: structure field changed type between kernel versions" struct_name field_name);
+           with Not_found ->
+             Hashtbl.add hash field_name typ
+       ) found_fields;
+
+       (* Now get a type for each structure field. *)
+       List.filter_map (
+         fun (field_name, { mandatory_field = mandatory }) ->
+           try
+             let field_name = struct_name ^ "_" ^ field_name in
+             let typ = Hashtbl.find hash field_name in
+             Some (field_name, (typ, mandatory))
+           with Not_found ->
+             let msg =
+               sprintf "%s.%s: this field was not found in any kernel version"
+                 struct_name field_name in
+             if mandatory then failwith msg else prerr_endline msg;
+             None
+       ) struct_fields in
       (struct_name, kernels, field_types)
   ) files in
 
@@ -488,7 +641,8 @@ Example (from toplevel of virt-mem source tree):
                  !i in
              (basename, version, arch, total_size, j)
          ) kernels in
-       struct_name, kernels, field_types, List.rev !xs
+       let parsers = List.rev !xs in
+       struct_name, kernels, field_types, parsers
     ) files in
 
   (* How much did we save by sharing? *)
@@ -501,6 +655,25 @@ Example (from toplevel of virt-mem source tree):
          (List.length parsers)
     ) files;
 
+  (* Extend the parsers fields by adding on any optional fields which
+   * are not actually present in the specific kernel.
+   *)
+  let files =
+    List.map (
+      fun (struct_name, kernels, field_types, parsers) ->
+       let parsers = List.map (
+         fun (i, (endian, fields)) ->
+           let fields_not_present =
+             List.filter_map (
+               fun (field_name, _) ->
+                 if List.mem_assoc field_name fields then None
+                 else Some field_name
+             ) field_types in
+           (i, (endian, fields, fields_not_present))
+       ) parsers in
+       (struct_name, kernels, field_types, parsers)
+    ) files in
+
   (* Let's generate some code! *)
   let files =
     List.map (
@@ -514,30 +687,79 @@ Example (from toplevel of virt-mem source tree):
        let struct_type, struct_sig =
          let fields = List.map (
            function
-           | (name, `Int) ->
-               <:ctyp< $lid:name$ : int >>
-           | (name, `Ptr struct_name) ->
-               <:ctyp< $lid:name$ : [`$lid:struct_name$] int64 >>
-           | (name, `Str _) ->
+           | (name, (`Int, true)) ->
+               <:ctyp< $lid:name$ : int64 >>
+           | (name, (`Int, false)) ->
+               <:ctyp< $lid:name$ : int64 option >>
+           | (name, (`Ptr _, true)) ->
+               <:ctyp< $lid:name$ : Virt_mem_mmap.addr >>
+           | (name, (`Ptr _, false)) ->
+               <:ctyp< $lid:name$ : Virt_mem_mmap.addr option >>
+           | (name, (`Str _, true)) ->
                <:ctyp< $lid:name$ : string >>
+           | (name, (`Str _, false)) ->
+               <:ctyp< $lid:name$ : string option >>
          ) field_types in
-         let f, fs = match fields with
-           | [] -> failwith (sprintf "%s: structure has no fields" struct_name)
-           | f :: fs -> f, fs in
-         let fields = List.fold_left (
-           fun fs f -> <:ctyp< $fs$ ; $f$ >>
-         ) f fs in
-
+         let fields = concat_record_fields _loc fields in
          let struct_type = <:str_item< type t = { $fields$ } >> in
          let struct_sig = <:sig_item< type t = { $fields$ } >> in
          struct_type, struct_sig in
 
+       (* Create a "field signature" which describes certain aspects
+        * of the fields which vary between kernel versions.
+        *)
+       let fieldsig_type, fieldsigs =
+         let fieldsig_type =
+           let fields = List.map (
+             fun (name, _) ->
+               let fsname = "__fs_" ^ name in
+               <:ctyp< $lid:fsname$ : Virt_mem_types.fieldsig >>
+           ) field_types in
+           let fields = concat_record_fields _loc fields in
+           <:str_item< type fs_t = { $fields$ } >> in
+
+         let fieldsigs = List.map (
+           fun (i, (_, fields, fields_not_present)) ->
+             let make_fieldsig field_name available offset =
+               let available =
+                 if available then <:expr< true >> else <:expr< false >> in
+               let fsname = "__fs_" ^ field_name in
+               <:rec_binding<
+                 $lid:fsname$ =
+                     { Virt_mem_types.field_available = $available$;
+                       field_offset = $`int:offset$ }
+               >>
+             in
+              let fields = List.map (
+               fun (field_name, (_, offset, _)) ->
+                 make_fieldsig field_name true offset
+             ) fields in
+              let fields_not_present = List.map (
+               fun field_name ->
+                 make_fieldsig field_name false (-1)
+             ) fields_not_present in
+
+             let fieldsigs = fields @ fields_not_present in
+             let fsname = sprintf "fieldsig_%d" i in
+             let fieldsigs = concat_record_bindings _loc fieldsigs in
+             let fieldsigs = build_record _loc fieldsigs in
+             <:str_item<
+               let $lid:fsname$ = $fieldsigs$
+             >>
+         ) parsers in
+
+         let fieldsigs = concat_str_items _loc fieldsigs in
+
+         fieldsig_type, fieldsigs in
+
        (* The shared parser functions.
+        * 
         * We could include bitmatch statements directly in here, but
         * what happens is that the macros get expanded here, resulting
-        * in unreadable generated code.  So instead just do a textual
-        * substitution later by post-processing the generated files.
-        * Not type-safe, but we can't have everything.
+        * in (even more) unreadable generated code.  So instead just
+        * do a textual substitution later by post-processing the
+        * generated files.  Not type-safe, but we can't have
+        * everything.
         *)
        let parser_stmts, parser_subs =
          let parser_stmts = List.map (
@@ -548,22 +770,74 @@ Example (from toplevel of virt-mem source tree):
              >>
          ) parsers in
 
-         let parser_stmts =
-           match parser_stmts with
-           | [] -> <:str_item< >>
-           | p :: ps ->
-               List.fold_left (fun ps p -> <:str_item< $ps$ $p$ >>) p ps in
+         let parser_stmts = concat_str_items _loc parser_stmts in
 
          (* What gets substituted for "parser_NN" ... *)
          let parser_subs = List.map (
-           fun (i, (endian, fields)) ->
+           fun (i, (endian, fields, fields_not_present)) ->
              let fnname = sprintf "parser_%d" i in
-             let patterns = "" and assignments = "" in (* XXX *)
+             let endian =
+               match endian with
+               | Bitstring.LittleEndian -> "littleendian"
+               | Bitstring.BigEndian -> "bigendian"
+               | _ -> assert false in
+             let patterns =
+               (* Fields must be sorted by offset, otherwise bitmatch
+                * will complain.
+                *)
+               let cmp (_, (_, o1, _)) (_, (_, o2, _)) = compare o1 o2 in
+               let fields = List.sort ~cmp fields in
+               String.concat ";\n      " (
+                 List.map (
+                   function
+                   | (field_name, (`Int, offset, size))
+                   | (field_name, (`Ptr _, offset, size)) ->
+                       (* 'zero+' is a hack to force the type to int64. *)
+                       sprintf "%s : zero+%d : offset(%d), %s"
+                         field_name (size*8) (offset*8) endian
+                   | (field_name, (`Str width, offset, size)) ->
+                       sprintf "%s : %d : offset(%d), string"
+                         field_name (width*8) (offset*8)
+                 ) fields
+               ) in
+             let assignments =
+               List.map (
+                 fun (field_name, typ) ->
+                   let (_, mandatory) =
+                     try List.assoc field_name field_types
+                     with Not_found ->
+                       failwith (sprintf "%s: not found in field_types"
+                                   field_name) in
+                   match typ, mandatory with
+                   | (`Ptr "list_head", offset, size), true ->
+                       sprintf "%s = Int64.sub %s %dL"
+                         field_name field_name offset
+                   | (`Ptr "list_head", offset, size), false ->
+                       sprintf "%s = Some (Int64.sub %s %dL)"
+                         field_name field_name offset
+                   | _, true ->
+                       sprintf "%s = %s" field_name field_name
+                   | _, false ->
+                       sprintf "%s = Some %s" field_name field_name
+               ) fields in
+             let assignments_not_present =
+               List.map (
+                 fun field_name -> sprintf "%s = None" field_name
+               ) fields_not_present in
+
+             let assignments =
+               String.concat ";\n        "
+                 (assignments @ assignments_not_present) in
+
              let sub =
-               sprintf "bitmatch bits with
-                         | { %s } -> { %s }
-                         | { _ } -> raise (ParseError (%S, %S, \"failed to match kernel structure\"))"
-                 patterns assignments struct_name fnname in
+               sprintf "
+  bitmatch bits with
+  | { %s } ->
+      { %s }
+  | { _ } ->
+      raise (Virt_mem_types.ParseError (struct_name, %S, match_err))"
+                 patterns assignments fnname in
+
              fnname, sub
          ) parsers in
 
@@ -574,58 +848,140 @@ Example (from toplevel of virt-mem source tree):
          let stmts = List.fold_left (
            fun stmts (_, version, arch, total_size, i) ->
              let parserfn = sprintf "parser_%d" i in
+             let fsname = sprintf "fieldsig_%d" i in
              <:str_item<
                $stmts$
-               let v = ($lid:parserfn$, $`int:total_size$)
+               let v = ($lid:parserfn$, $`int:total_size$, $lid:fsname$)
                let map = StringMap.add $str:version$ v map
              >>
          ) <:str_item< let map = StringMap.empty >> kernels in
 
          <:str_item<
-            module StringMap = Map.Make (String)
+            module StringMap = Map.Make (String) ;;
            $stmts$
          >> in
 
+       (* Accessors for the field signatures. *)
+       let fsaccess, fsaccess_sig =
+         let fields = List.map (
+           fun (field_name, _) ->
+             let fsname = "__fs_" ^ field_name in
+             <:str_item<
+               let $lid:"field_signature_of_"^field_name$ version =
+                 let _, _, fs = StringMap.find version map in
+                 fs.$lid:fsname$
+             >>
+         ) field_types in
+
+         let fsaccess = concat_str_items _loc fields in
+
+         let fields = List.map (
+           fun (field_name, _) ->
+             <:sig_item<
+               val $lid:"field_signature_of_"^field_name$ : kernel_version ->
+                 Virt_mem_types.fieldsig
+             >>
+         ) field_types in
+
+         let fsaccess_sig = concat_sig_items _loc fields in
+
+         fsaccess, fsaccess_sig in
+
        (* Code (.ml file). *)
        let code = <:str_item<
-         let warning = "This code is automatically generated from the kernel database by kerneldb-to-parser program.  Any edits you make will be lost."
-          exception ParseError of string ;;
+          let zero = 0
+         let struct_name = $str:struct_name$
+         let match_err = "failed to match kernel structure" ;;
          $struct_type$
+         $fieldsig_type$
+         $fieldsigs$
          $parser_stmts$
          $version_map$
 
          type kernel_version = string
-         let known version = StringMap.mem version map
-         let size version =
-           let _, size = StringMap.find version map in
+         let $lid:struct_name^"_known"$ version = StringMap.mem version map
+         let $lid:struct_name^"_size"$ version =
+           let _, size, _ = StringMap.find version map in
            size
-         let get version bits =
-           let parsefn, _ = StringMap.find version map in
+         let $lid:struct_name^"_of_bits"$ version bits =
+           let parsefn, _, _ = StringMap.find version map in
            parsefn bits
+         let $lid:"get_"^struct_name$ version mem addr =
+           let parsefn, size, _ = StringMap.find version map in
+           let bytes = Virt_mem_mmap.get_bytes mem addr size in
+           let bits = Bitstring.bitstring_of_string bytes in
+           parsefn bits ;;
+         $fsaccess$
        >> in
 
        (* Interface (.mli file). *)
        let interface = <:sig_item<
-          exception ParseError of string ;;
          $struct_sig$
 
+          val struct_name : string
          type kernel_version = string
-         val known : kernel_version -> bool
-         val size : kernel_version -> int
-         val get : kernel_version -> Bitstring.bitstring -> t
+         val $lid:struct_name^"_known"$ : kernel_version -> bool
+         val $lid:struct_name^"_size"$ : kernel_version -> int
+         val $lid:struct_name^"_of_bits"$ :
+           kernel_version -> Bitstring.bitstring -> t
+         val $lid:"get_"^struct_name$ : kernel_version ->
+           ('a, 'b, [`HasMapping]) Virt_mem_mmap.t -> Virt_mem_mmap.addr -> t;;
+         $fsaccess_sig$
        >> in
 
-       (struct_name, code, interface)
+       (struct_name, code, interface, parser_subs)
     ) files in
 
   (* Finally generate the output files. *)
+  let re_subst = Pcre.regexp "^(.*)\"(parser_\\d+)\"(.*)$" in
+
   List.iter (
-    fun (struct_name, code, interface) ->
+    fun (struct_name, code, interface, parser_subs) ->
+      (* Interface (.mli file). *)
+      let output_file = outputdir // "kernel_" ^ struct_name ^ ".mli" in
+      printf "Writing %s interface to %s ...\n%!" struct_name output_file;
+      Printers.OCaml.print_interf ~output_file interface;
+
+      (* Implementation (.ml file). *)
       let output_file = outputdir // "kernel_" ^ struct_name ^ ".ml" in
       printf "Writing %s implementation to %s ...\n%!" struct_name output_file;
-      Printers.OCaml.print_implem ~output_file code;
 
-      let output_file = outputdir // "kernel_" ^ struct_name ^ ".mli" in
-      printf "Writing %s interface to %s ...\n%!" struct_name output_file;
-      Printers.OCaml.print_interf ~output_file interface
+      let new_output_file = output_file ^ ".new" in
+      Printers.OCaml.print_implem ~output_file:new_output_file code;
+
+      (* Substitute the parser bodies in the output file. *)
+      let ichan = open_in new_output_file in
+      let ochan = open_out output_file in
+
+      output_string ochan "\
+(* WARNING: This file and the corresponding mli (interface) are
+ * automatically generated by the extract/codegen/kerneldb_to_parser.ml
+ * program.
+ *
+ * Any edits you make to this file will be lost.
+ *
+ * To update this file from the latest kernel database, it is recommended
+ * that you do 'make update-kernel-structs'.
+ *)\n\n";
+
+      let rec loop () =
+       let line = input_line ichan in
+       let line =
+         if Pcre.pmatch ~rex:re_subst line then (
+           let subs = Pcre.exec ~rex:re_subst line in
+           let start = Pcre.get_substring subs 1 in
+           let template = Pcre.get_substring subs 2 in
+           let rest = Pcre.get_substring subs 3 in
+           let sub = List.assoc template parser_subs in
+           start ^ sub ^ rest
+         ) else line in
+       output_string ochan line; output_char ochan '\n';
+       loop ()
+      in
+      (try loop () with End_of_file -> ());
+
+      close_out ochan;
+      close_in ichan;
+
+      Unix.unlink new_output_file
   ) files