-(* This would be so much simpler with ChriS's delimited
- * overloading macro XXX
- *)
-let i32_of_string_le v =
- let b0 = int_of_char (String.unsafe_get v 0) in
- let b1 = int_of_char (String.unsafe_get v 1) in
- let b2 = int_of_char (String.unsafe_get v 2) in
- let b3 = Int32.of_int (int_of_char (String.unsafe_get v 3)) in
- Int32.logor
- (Int32.of_int (b0 lor (b1 lsl 8) lor (b2 lsl 16)))
- (Int32.shift_left b3 24)
-
-let i32_of_string_be v =
- let b0 = Int32.of_int (int_of_char (String.unsafe_get v 0)) in
- let b1 = int_of_char (String.unsafe_get v 1) in
- let b2 = int_of_char (String.unsafe_get v 2) in
- let b3 = int_of_char (String.unsafe_get v 3) in
- Int32.logor
- (Int32.of_int (b3 lor (b2 lsl 8) lor (b1 lsl 16)))
- (Int32.shift_left b0 24)
-
-let i64_of_string_le v =
- let b0 = int_of_char (String.unsafe_get v 0) in
- let b1 = int_of_char (String.unsafe_get v 1) in
- let b2 = int_of_char (String.unsafe_get v 2) in
- let b3 = Int64.of_int (int_of_char (String.unsafe_get v 3)) in
- let b4 = Int64.of_int (int_of_char (String.unsafe_get v 4)) in
- let b5 = Int64.of_int (int_of_char (String.unsafe_get v 5)) in
- let b6 = Int64.of_int (int_of_char (String.unsafe_get v 6)) in
- let b7 = Int64.of_int (int_of_char (String.unsafe_get v 7)) in
- Int64.logor
- (Int64.logor
- (Int64.logor
- (Int64.logor
- (Int64.logor
- (Int64.of_int (b0 lor (b1 lsl 8) lor (b2 lsl 16)))
- (Int64.shift_left b3 24))
- (Int64.shift_left b4 32))
- (Int64.shift_left b5 40))
- (Int64.shift_left b6 48))
- (Int64.shift_left b7 56)
+module CE = CamomileLibraryDefault.Camomile.CharEncoding
+module UTF8 = CamomileLibraryDefault.Camomile.UTF8
+module UChar = CamomileLibraryDefault.Camomile.UChar
+let utf16le = CE.utf16le
+let utf8 = CE.utf8
+let recode = CE.recode_string ~in_enc:utf16le ~out_enc:utf8
+
+let windows_string_to_utf8 str =
+ let str = recode str in
+
+ (* Windows strings include the final \0 so remove this if present. *)
+ let len = UTF8.length str in
+ if len > 0 && UChar.code (UTF8.get str (len-1)) = 0 then
+ String.sub str 0 (UTF8.last str)
+ else
+ str
+
+(* Best effort convert hive value to printable string. *)
+let rec printable_hivex_value ?split_long_lines t v =
+ let hex = reg_hex_of_string ?split_long_lines in
+ match t with
+ | Hivex.REG_NONE -> if v = "" then "" else hex v
+ | Hivex.REG_SZ ->
+ (try windows_string_to_utf8 v with _ -> hex v)
+ | Hivex.REG_EXPAND_SZ ->
+ (try windows_string_to_utf8 v with _ -> hex v)
+ | Hivex.REG_BINARY -> hex v
+ | Hivex.REG_DWORD ->
+ (bitmatch Bitstring.bitstring_of_string v with
+ | { i : 32 : littleendian } -> sprintf "%08lx" i
+ | { _ } -> hex v)
+ | Hivex.REG_DWORD_BIG_ENDIAN ->
+ (bitmatch Bitstring.bitstring_of_string v with
+ | { i : 32 : bigendian } -> sprintf "%08lx" i
+ | { _ } -> hex v)
+ | Hivex.REG_LINK -> hex v
+ | Hivex.REG_MULTI_SZ -> (* XXX should be better for this one *)
+ hex v
+ | Hivex.REG_RESOURCE_LIST -> hex v
+ | Hivex.REG_FULL_RESOURCE_DESCRIPTOR -> hex v
+ | Hivex.REG_RESOURCE_REQUIREMENTS_LIST -> hex v
+ | Hivex.REG_QWORD ->
+ (bitmatch Bitstring.bitstring_of_string v with
+ | { i : 64 : littleendian } -> sprintf "%016Lx" i
+ | { _ } -> hex v)
+ | Hivex.REG_UNKNOWN i32 -> hex v
+
+(* Convert binary data to a hex string. This includes line breaks. *)
+and reg_hex_of_string ?(split_long_lines=false) v =
+ let vs = String.explode v in
+ let vs = List.mapi (
+ fun i c ->
+ sprintf "%s%02x"
+ (if split_long_lines && i mod 16 = 0 then "\n" else "")
+ (int_of_char c)
+ ) vs in
+ String.concat "," vs
+
+let local_file_exists filename =
+ try Unix.access filename [Unix.F_OK]; true
+ with Unix.Unix_error _ -> false
+
+let basename pathname =
+ let len = String.length pathname in
+ try
+ let i = String.rindex pathname '/' in
+ let r = String.sub pathname (i+1) (len-i-1) in
+ if r = "" then "root" else r
+ with
+ Not_found -> pathname
+
+let extension pathname =
+ let len = String.length pathname in
+ try
+ let i = String.rindex pathname '.' in
+ let r = String.sub pathname i (len-i) in
+ r
+ with
+ Not_found -> ""