From b66aaf2d1b90ff51f54bcd2a344a6ab50ac6fe86 Mon Sep 17 00:00:00 2001
From: xleroy <xleroy@fca1b0fc-160b-0410-b1d3-a4f43f01ea2e>
Date: Wed, 7 Apr 2010 08:31:55 +0000
Subject: Static initialization of structs with bitfields

git-svn-id: https://yquem.inria.fr/compcert/svn/compcert/trunk@1311 fca1b0fc-160b-0410-b1d3-a4f43f01ea2e
---
 cparser/Bitfields.ml               | 75 +++++++++++++++++++++++++++++++++++++-
 test/regression/Results/bitfields2 |  3 ++
 test/regression/bitfields2.c       | 32 ++++++++++++++++
 3 files changed, 109 insertions(+), 1 deletion(-)
 create mode 100644 test/regression/Results/bitfields2
 create mode 100644 test/regression/bitfields2.c

diff --git a/cparser/Bitfields.ml b/cparser/Bitfields.ml
index 4f7bcf16..27d58957 100644
--- a/cparser/Bitfields.ml
+++ b/cparser/Bitfields.ml
@@ -317,7 +317,80 @@ let rec transf_stmt env s =
 let transf_fundef env f =
   { f with fd_body = transf_stmt env f.fd_body }
 
+(* Initializers *)
+
+let bitfield_initializer bf i =
+  match i with
+  | Init_single e ->
+      let m = Int64.pred (Int64.shift_left 1L bf.bf_size) in
+      let e_mask =
+        {edesc = EConst(CInt(m, IUInt, sprintf "0x%LXU" m));
+         etyp = TInt(IUInt, [])} in
+      let e_and =
+        {edesc = EBinop(Oand, e, e_mask, TInt(IUInt,[]));
+         etyp = TInt(IUInt,[])} in
+      {edesc = EBinop(Oshl, e_and, intconst (Int64.of_int bf.bf_pos) IInt,
+                      TInt(IUInt, []));
+       etyp = TInt(IUInt, [])}
+  | _ -> assert false
+
+let rec pack_bitfield_init id carrier fld_init_list =
+  match fld_init_list with
+  | [] -> ([], [])
+  | (fld, i) :: rem ->
+      try
+        let bf = Hashtbl.find bitfield_table (id, fld.fld_name) in
+        if bf.bf_carrier <> carrier then
+          ([], fld_init_list)
+        else begin
+          let (el, rem') = pack_bitfield_init id carrier rem in
+          (bitfield_initializer bf i :: el, rem')
+        end
+      with Not_found ->
+        ([], fld_init_list)
+
+let rec or_expr_list = function
+  | [] -> assert false
+  | [e] -> e
+  | e1 :: el ->
+      {edesc = EBinop(Oor, e1, or_expr_list el, TInt(IUInt,[]));
+       etyp = TInt(IUInt,[])}
+
+let rec transf_struct_init id fld_init_list =
+  match fld_init_list with
+  | [] -> []
+  | (fld, i) :: rem ->
+      try
+        let bf = Hashtbl.find bitfield_table (id, fld.fld_name) in
+        let (el, rem') =
+          pack_bitfield_init id bf.bf_carrier fld_init_list in
+        ({fld_name = bf.bf_carrier; fld_typ = bf.bf_carrier_typ;
+          fld_bitfield = None},
+         Init_single {edesc = ECast(bf.bf_carrier_typ, or_expr_list el);
+                      etyp = bf.bf_carrier_typ})
+        :: transf_struct_init id rem'
+      with Not_found ->
+        (fld, i) :: transf_struct_init id rem
+
+let rec transf_init env i =
+  match i with
+  | Init_single e -> Init_single (transf_expr env e)
+  | Init_array il -> Init_array (List.map (transf_init env) il)
+  | Init_struct(id, fld_init_list) ->
+      let fld_init_list' =
+        List.map (fun (f, i) -> (f, transf_init env i)) fld_init_list in
+        Init_struct(id, transf_struct_init id fld_init_list')
+  | Init_union(id, fld, i) -> Init_union(id, fld, transf_init env i)
+
+let transf_decl env (sto, id, ty, init_opt) =
+  (sto, id, ty,
+   match init_opt with None -> None | Some i -> Some(transf_init env i))
+
 (* Programs *)
 
 let program p =
-  Transform.program ~composite:transf_composite ~fundef:transf_fundef p
+  Transform.program
+    ~composite:transf_composite
+    ~decl: transf_decl
+    ~fundef:transf_fundef 
+    p
diff --git a/test/regression/Results/bitfields2 b/test/regression/Results/bitfields2
new file mode 100644
index 00000000..0ab7773f
--- /dev/null
+++ b/test/regression/Results/bitfields2
@@ -0,0 +1,3 @@
+x = { a = 1, b = 2 }
+y = { c = 3, d = 0; e = 5 }
+z = { f = 6, g = 7; h = 8 }
diff --git a/test/regression/bitfields2.c b/test/regression/bitfields2.c
new file mode 100644
index 00000000..b555f1eb
--- /dev/null
+++ b/test/regression/bitfields2.c
@@ -0,0 +1,32 @@
+#include <stdio.h>
+
+struct s {
+  signed char a: 6;
+  unsigned int b: 2;
+};
+
+struct t {
+  unsigned int c: 16;
+  unsigned int d: 1;
+  short e: 8;
+};
+
+struct u {
+  unsigned int f: 18;
+  unsigned int g: 24;
+  char h;
+};
+
+struct s x = { 1, 2 };
+
+struct t y = { 3, 4, 5 };
+
+struct u z = { 6, 7, 8 };
+
+int main()
+{
+  printf("x = { a = %d, b = %d }\n", x.a, x.b);
+  printf("y = { c = %d, d = %d; e = %d }\n", y.c, y.d, y.e);
+  printf("z = { f = %d, g = %d; h = %d }\n", z.f, z.g, z.h);
+}
+
-- 
cgit