WebSVN – QNX 8.QNX8 IFS tool – Diff – /dump.c

    size_t imageheader_offset = 0;
    size_t imagetrailer_offset = 0;
    fsentry_t **fsentries = NULL; // mallocated
    size_t fsentry_count = 0;
    fsentry_t *current_fsentry = NULL;
+   buffer_t decompression_dst;
    size_t startupfile_blobsize = 0;
+   size_t compressed_blocksize;
    struct utimbuf file_times = { 0, 0 };
    void *reallocated_ptr;
+   uint8_t *decompressor_out;
+   uint8_t *decompressor_in;
+   size_t decompressor_outlen;
    size_t bootfile_blobsize = 0;
    size_t current_offset;
    size_t fsentry_index;
    size_t nearest_distance;
    size_t nearest_index;
    size_t byte_index;
    buffer_t file;
    FILE *fp;
+   int cf;
    // open and read IFS file
    if (!Buffer_ReadFromFile (&file, ifs_pathname))
       DIE_WITH_EXITCODE (1, "can't open \"%s\" for reading: %s\n", ifs_pathname, strerror (errno));
    // parse file from start to end
    current_offset = 0;
    for (;;)
+   {
       // does a startup header start here ?
+      if ((current_offset + sizeof (startup_header_t) < file.size)
+          && (startup_header == NULL)
-      if ((current_offset + sizeof (startup_header_t) < file.size) && (memcmp (&file.bytes[current_offset], "\xeb\x7e\xff\x00", 4) == 0))
+          && (memcmp (&file.bytes[current_offset], "\xeb\x7e\xff\x00", 4) == 0))
+      {
          startupheader_offset = current_offset;
          startup_header = (startup_header_t *) &file.bytes[startupheader_offset];
          // layout:
          if (current_offset + startup_header->startup_size > file.size)
+         {
             LOG_WARNING ("this IFS file is corrupted (startup trailer extends past end of file)");
             goto endofdata;
+         }
+         // take note of the image compression flags
+         cf = startup_header->flags1 & STARTUP_HDR_FLAGS1_COMPRESS_MASK;
          // locate the right startup trailer at the right offset
          if (startup_header->flags1 & STARTUP_HDR_FLAGS1_TRAILER_V2)
             startupfile_blobsize = startup_header->startup_size - sizeof (startup_header_t) - sizeof (startup_trailer_v2_t);
          else // old V1 trailer
             startupfile_blobsize = startup_header->startup_size - sizeof (startup_header_t) - sizeof (startup_trailer_v1_t);
          current_offset += sizeof (startup_header_t); // jump over the startup header and reach the startup blob
          // write startup blob
          sprintf_s (outfile_pathname, sizeof (outfile_pathname), "%s/startup.bin", outdir);
          fopen_s (&fp, outfile_pathname, "wb");
          ASSERT (fp, "failed to open '%s': %s", outfile_pathname, strerror (errno));
          fwrite (&file.bytes[current_offset], 1, startupfile_blobsize, fp);
          fclose (fp);
          current_offset += startupfile_blobsize; // jump over the startup blob and reach the startup trailer
          current_offset += (startup_header->flags1 & STARTUP_HDR_FLAGS1_TRAILER_V2 ? sizeof (startup_trailer_v2_t) : sizeof (startup_trailer_v1_t)); // jump over the startup trailer and reach the next segment
+      }
       // else does an image header start here ?
+      else if ((current_offset + sizeof (image_header_t) < file.size)
+               && (image_header == NULL)
-      else if ((current_offset + sizeof (image_header_t) < file.size) && (memcmp (&file.bytes[current_offset], "imagefs", 7) == 0))
+               && (((cf == STARTUP_HDR_FLAGS1_COMPRESS_NONE) && (memcmp (&file.bytes[current_offset], "imagefs", 7) == 0))
+               || (cf != STARTUP_HDR_FLAGS1_COMPRESS_NONE) && (startup_header->imagefs_size > 0)))
+      {
          imageheader_offset = current_offset;
          image_header = (image_header_t *) &file.bytes[imageheader_offset];
+         // should we decompress it ?
+         if (cf != STARTUP_HDR_FLAGS1_COMPRESS_NONE)
+         {
+            // it appears mkifs compresses data in blocks, prefixed by 2-byte block size in BIG ENDIAN
+            Buffer_InitWithSize (&decompression_dst, startup_header->imagefs_size * 11 / 10); // mallocate and add 10% for safety
+            decompression_dst.size = 0;
+            if (cf == STARTUP_HDR_FLAGS1_COMPRESS_UCL)
+               ASSERT (ucl_init () == UCL_E_OK, "UCL library initialization failed -- please recompile this tool with less aggressive optimizations");
+            else if (cf == STARTUP_HDR_FLAGS1_COMPRESS_LZO)
+               ASSERT (lzo_init () == LZO_E_OK, "LZO library initialization failed -- please recompile this tool with less aggressive optimizations");
+            else if (cf == STARTUP_HDR_FLAGS1_COMPRESS_ZLIB)
+            {
+               LOG_WARNING ("unimplemented compression scheme: zlib (FIXME)");
+               goto endofdata;
+            }
+            else
+            {
+               LOG_WARNING ("unsupported compression flags: 0x%2x", cf);
+               goto endofdata;
+            }
+            // run the compressed payload (the imagefs) through the right decompression algorithm
+            for (;;)
+            {
+               compressed_blocksize = (file.bytes[current_offset + 0] << 8) | (file.bytes[current_offset + 1] << 0); // read block size word (in big engian)
+               current_offset += 2; // skip it
+               if (compressed_blocksize == 0)
+                  break; // a nil block size means end of stream is reached
+               LOG_DEBUG ("about to decompress block of %zd bytes", compressed_blocksize);
+               decompressor_in = &file.bytes[current_offset];
+               decompressor_out = &decompression_dst.bytes[decompression_dst.size];
+               decompressor_outlen = 0;
+               if (cf == STARTUP_HDR_FLAGS1_COMPRESS_UCL)
+               {
+                  // UCL compression. NOTE: the decompressor function used in startup-x86 is "ucl_nrv2b_decompress_8 / ucl_nrv2b_decompress_le16 / ucl_nrv2b_decompress_le32"
+                  static ucl_uint ucl_outlen; // have a different variable because of pointer size mismatch
+                  if (ucl_nrv2b_decompress_8 (decompressor_in, (ucl_uint) compressed_blocksize, decompressor_out, &ucl_outlen, NULL) != UCL_E_OK)
+                  {
+                     LOG_WARNING ("this IFS file is corrupted (UCL decompression failed)");
+                     goto endofdata;
+                  }
+                  decompressor_outlen = ucl_outlen;
+               }
+               else if (cf == STARTUP_HDR_FLAGS1_COMPRESS_LZO)
+               {
+                  // LZO decompression. NOTE: mkifs uses the full LZO package, whereas I use minilzo.
+                  static lzo_uint lzo_outlen; // have a different variable because of pointer size mismatch
+                  if (lzo1x_decompress (decompressor_in, (lzo_uint) compressed_blocksize, decompressor_out, &lzo_outlen, NULL) != LZO_E_OK)
+                  {
+                     LOG_WARNING ("this IFS file is corrupted (UCL decompression failed)");
+                     goto endofdata;
+                  }
+                  decompressor_outlen = lzo_outlen;
+               }
+               else if (cf == STARTUP_HDR_FLAGS1_COMPRESS_ZLIB)
+                  ; // TODO
+               current_offset += compressed_blocksize;
+               decompression_dst.size += decompressor_outlen;
+            }
+            LOG_INFO ("decompressed %zd bytes into %zd bytes\n", file.size - imageheader_offset, decompression_dst.size);
+            // now place the decompressed buffer in the payload at the imagefs offset
+            ASSERT_WITH_ERRNO (Buffer_WriteBufferAt (&file, imageheader_offset, &decompression_dst));
+            current_offset = imageheader_offset; // jump back to where we put the uncompressed data
+            file.size = imageheader_offset + decompression_dst.size; // update IFS data size
+            startup_header = (startup_header_t *) &file.bytes[startupheader_offset]; // fix the pointers that might have changed
+            image_header = (image_header_t *) &file.bytes[imageheader_offset]; // fix the pointers that might have changed
+         }
          // layout:
          // [IMAGE HEADER]
          // [image directory entries]
          // [smallest file blobs up to KERNEL]

Subversion Repositories QNX 8.QNX8 IFS tool

QNX 8.QNX8 IFS tool/dump.c – Rev 26 → 27