Support some last missing features from Kaitai: bit fields, instance search order and stream EOF.

1 files changed, 33 insertions, 12 deletions

diff --git a/plugins/kaitai/parsers/struct.c b/plugins/kaitai/parsers/struct.c
index f23a264..62dc25a 100644
--- a/plugins/kaitai/parsers/struct.c
+++ b/plugins/kaitai/parsers/struct.c
@@ -64,7 +64,7 @@ static bool g_kaitai_structure_load_imports(GKaitaiStruct *);
 
 
 /* Parcourt un contenu binaire selon des spécifications Kaitai. */
-static bool g_kaitai_structure_parse_content(GKaitaiStruct *, kaitai_scope_t *, GBinContent *, vmpa2t *, GMatchRecord **);
+static bool g_kaitai_structure_parse_content(GKaitaiStruct *, kaitai_scope_t *, GBinContent *, ext_vmpa_t *, GMatchRecord **);
 
 
 
@@ -712,13 +712,17 @@ GMatchRecord *g_kaitai_structure_parse(GKaitaiStruct *kstruct, GBinContent *cont
     GMatchRecord *result;                   /* Arborescence à retourner    */ 
     vmpa2t pos;                             /* Tête de lecture             */
     kaitai_scope_t locals;                  /* Variables locales           */
-    bool status;                            /* Bilan de l'analyse          */
+    ext_vmpa_t epos;                        /* Tête de lecture complète    */
 
     g_binary_content_compute_start_pos(content, &pos);
 
     init_record_scope(&locals, kstruct->meta);
 
-    status = g_kaitai_parser_parse_content(G_KAITAI_PARSER(kstruct), &locals, content, &pos, &result);
+    init_evmpa_from_vmpa(&epos, &pos);
+
+    g_kaitai_parser_parse_content(G_KAITAI_PARSER(kstruct), &locals, content, &epos, &result);
+
+    printf("FINAL STATUS :: %d\n", result != NULL);
 
     return result;
 
@@ -736,7 +740,7 @@ GMatchRecord *g_kaitai_structure_parse(GKaitaiStruct *kstruct, GBinContent *cont
 *  Paramètres  : kstruct = structure Kaitai en cours de parcours.             *
 *                locals  = variables locales pour les résolutions de types.   *
 *                content = données binaires à analyser et traduire.           *
-*                pos     = tête de lecture courante. [OUT]                    *
+*                epos    = tête de lecture courante. [OUT]                    *
 *                record  = noeud d'arborescence d'éléments rencontrés. [OUT]  *
 *                                                                             *
 *  Description : Parcourt un contenu binaire selon des spécifications Kaitai. *
@@ -747,7 +751,7 @@ GMatchRecord *g_kaitai_structure_parse(GKaitaiStruct *kstruct, GBinContent *cont
 *                                                                             *
 ******************************************************************************/
 
-static bool g_kaitai_structure_parse_content(GKaitaiStruct *kstruct, kaitai_scope_t *locals, GBinContent *content, vmpa2t *pos, GMatchRecord **record)
+static bool g_kaitai_structure_parse_content(GKaitaiStruct *kstruct, kaitai_scope_t *locals, GBinContent *content, ext_vmpa_t *epos, GMatchRecord **record)
 {
     bool result;                            /* Bilan à retourner           */
     GRecordGroup *group;                    /* Ensemble à constituer       */
@@ -760,7 +764,7 @@ static bool g_kaitai_structure_parse_content(GKaitaiStruct *kstruct, kaitai_scop
     /* Si le groupe est vide */
     if ((kstruct->seq_items_count + kstruct->instances_count) == 0)
     {
-        *record = G_MATCH_RECORD(g_record_empty_new(G_KAITAI_PARSER(kstruct), content, pos));
+        *record = G_MATCH_RECORD(g_record_empty_new(G_KAITAI_PARSER(kstruct), content, &epos->base));
 
         if (locals->root == NULL)
             locals->root = *record;
@@ -779,10 +783,18 @@ static bool g_kaitai_structure_parse_content(GKaitaiStruct *kstruct, kaitai_scop
         old = locals->parent;
         locals->parent = *record;
 
-        for (i = 0; i < kstruct->seq_items_count; i++)
+        /**
+         * Les instances sont à charger avant les éléments fixes car
+         * des références au premières peuvent être attendues dans ces derniers.
+         *
+         * Les évolutions de la tête de lecture n'ont en revanche normalement
+         * pas d'incidence sur le chargement des éléments fixes.
+         */
+
+        for (i = 0; i < kstruct->instances_count; i++)
         {
-            result = g_kaitai_parser_parse_content(G_KAITAI_PARSER(kstruct->seq_items[i]),
-                                                   locals, content, pos, &child);
+            result = g_kaitai_parser_parse_content(G_KAITAI_PARSER(kstruct->instances[i]),
+                                                   locals, content, epos, &child);
             if (!result) goto exit;
 
             if (child != NULL)
@@ -793,10 +805,16 @@ static bool g_kaitai_structure_parse_content(GKaitaiStruct *kstruct, kaitai_scop
 
         }
 
-        for (i = 0; i < kstruct->instances_count; i++)
+        /**
+         * Seconde phase.
+         */
+
+        locals->parent = *record;
+
+        for (i = 0; i < kstruct->seq_items_count; i++)
         {
-            result = g_kaitai_parser_parse_content(G_KAITAI_PARSER(kstruct->instances[i]),
-                                                   locals, content, pos, &child);
+            result = g_kaitai_parser_parse_content(G_KAITAI_PARSER(kstruct->seq_items[i]),
+                                                   locals, content, epos, &child);
             if (!result) goto exit;
 
             if (child != NULL)
@@ -811,6 +829,9 @@ static bool g_kaitai_structure_parse_content(GKaitaiStruct *kstruct, kaitai_scop
 
         locals->parent = old;
 
+        if (!result)
+            g_clear_object(record);
+
     }
 
     return result;