From fd215a8adbf0021758c963487b2d908c07418224 Mon Sep 17 00:00:00 2001
From: dece <shgck@pistache.land>
Date: Sun, 19 Apr 2020 15:26:21 +0200
Subject: [PATCH] dcx: WIP parsing

---
 src/parsers/bhd.rs | 14 +++--------
 src/parsers/dcx.rs | 60 +++++++++++++++++++++++++++++++++++-----------
 2 files changed, 49 insertions(+), 25 deletions(-)

diff --git a/src/parsers/bhd.rs b/src/parsers/bhd.rs
index 4c8bbc4..ce53fc8 100644
--- a/src/parsers/bhd.rs
+++ b/src/parsers/bhd.rs
@@ -17,17 +17,9 @@ pub struct BhdHeader {
     pub ofs_buckets: u32,
 }
 
-const MAGIC: u32 = 0x35444842;
-
 fn parse_header(i: &[u8]) -> IResult<&[u8], BhdHeader> {
-    let (i, (magic, flags, unk08, file_len, num_buckets, ofs_buckets)) = tuple((
-        tag(b"BHD5"),
-        count(le_i8, 4),
-        le_u32,
-        le_u32,
-        le_u32,
-        le_u32,
-    ))(i)?;
+    let (i, (magic, flags, unk08, file_len, num_buckets, ofs_buckets)) =
+        tuple((tag(b"BHD5"), count(le_i8, 4), le_u32, le_u32, le_u32, le_u32))(i)?;
     Ok((
         i,
         BhdHeader {
@@ -40,7 +32,7 @@ fn parse_header(i: &[u8]) -> IResult<&[u8], BhdHeader> {
             file_len,
             num_buckets,
             ofs_buckets,
-        },
+        }
     ))
 }
 
diff --git a/src/parsers/dcx.rs b/src/parsers/dcx.rs
index 5c0581b..a05c6b9 100644
--- a/src/parsers/dcx.rs
+++ b/src/parsers/dcx.rs
@@ -45,6 +45,7 @@
 
 use nom::IResult;
 use nom::bytes::complete::tag;
+use nom::choice::alt;
 use nom::multi::count;
 use nom::number::complete::*;
 use nom::sequence::tuple;
@@ -59,40 +60,71 @@ pub struct DcxHeader {
     pub unk14: u32,
 }
 
-const HEADER_MAGIC: u32 = 0x00504344;
-
 fn parse_header(i: &[u8]) -> IResult<&[u8], DcxHeader> {
-    let (i, (magic, unk04, ofs_dcs, ofs_dcp, unk10, unk14)) = tuple((
-        tag(b"DCX\0"),
-        be_u32,
-        be_u32,
-        be_u32,
-        be_u32,
-        be_u32,
-    ))(i)?;
+    let (i, (magic, unk04, ofs_dcs, ofs_dcp, unk10, unk14)) =
+        tuple((tag(b"DCX\0"), be_u32, be_u32, be_u32, be_u32, be_u32))(i)?;
     Ok((i, DcxHeader { magic: magic.to_vec(), unk04, ofs_dcs, ofs_dcp, unk10, unk14 }))
 }
 
 #[derive(Debug)]
 pub struct DcxSizes {
-    pub magic: u32,
+    pub magic: Vec<u8>,
     pub uncompressed_size: u32,
     pub compressed_size: u32,
 }
 
+fn parse_sizes(i: &[u8]) -> IResult<&[u8], DcxSizes> {
+    let (i, (magic, uncompressed_size, compressed_size)) =
+        tuple((tag(b"DCS\0"), be_u32, be_u32))(i)?;
+    Ok((i, DcxSizes { magic: magic.to_vec(), uncompressed_size, compressed_size }))
+}
+
 #[allow(non_snake_case)]
 #[derive(Debug)]
 pub struct DcxParams {
-    pub magic: u32,
-    pub method: [u8; 4],
+    pub magic: Vec<u8>,
+    pub method: Vec<u8>,
     pub ofs_dca: u32,
-    pub unk0C: u32,
+    pub unk0C: u8,
+    pub unk0D: u8,
+    pub unk0E: u8,
+    pub unk0F: u8,
     pub unk10: u32,
     pub unk14: u32,
     pub unk18: u32,
     pub unk1C: u32,
 }
 
+fn parse_params(i: &[u8]) {
+    let (i, (magic, method, ofs_dca, flags, unk10, unk14, unk18, unk1C)) =
+        tuple((
+            tag(b"DCP\0"),
+            alt((tag(b"DFLT"), tag(b"EDGE"), tag(b"KRAK"))),
+            be_u32,
+            count(be_u8, 4),
+            be_u32,
+            be_u32,
+            be_u32,
+            be_u32,
+        ))(i)?;
+    Ok((
+        i,
+        DcxParams {
+            magic: magic.to_vec(),
+            method,
+            ofs_dca,
+            unk0C: flags[0],
+            unk0D: flags[1],
+            unk0E: flags[2],
+            unk0F: flags[3],
+            unk10,
+            unk14,
+            unk18,
+            unk1C
+        }
+    ))
+}
+
 #[derive(Debug)]
 pub struct Dcx {
     pub header: DcxHeader,