don't assume that some bytes are the filetype

[rsbs2] / firmware.c

diff --git a/firmware.c b/firmware.c
index 6ef5971461c7773eb7655e7b5cf9353f180a2f7f..cd7927b8f9854d86c20696adcfaabd8285f18dbf 100644 (file)
--- a/firmware.c
+++ b/firmware.c
@@ -8,7 +8,7 @@
 #include <string.h>
 #include <strings.h>
 #include "rsb-crc.h"
-#include "rsb-lz.h"
+#include "extract.h"
 
 #define FINDSTR(addr, str) (!strncmp((char*)addr, str, strlen(str)))
 
@@ -462,7 +462,7 @@ int main(int argc, char **argv)
        }
 
        if (extract) {
-               search_lz_sections(fw, statbuf.st_size - 4);
+               extract_files(fw, statbuf.st_size - 4);
        }
 
        if (update_crc || patch_fw || patch_bd) {