simg2img.c revision 28fa5bc347390480fe190294c6c385b6a9f0d68b
128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross/* 228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * Copyright (C) 2010 The Android Open Source Project 328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * 428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * Licensed under the Apache License, Version 2.0 (the "License"); 528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * you may not use this file except in compliance with the License. 628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * You may obtain a copy of the License at 728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * 828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * http://www.apache.org/licenses/LICENSE-2.0 928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * 1028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * Unless required by applicable law or agreed to in writing, software 1128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * distributed under the License is distributed on an "AS IS" BASIS, 1228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * See the License for the specific language governing permissions and 1428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * limitations under the License. 1528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross */ 1628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 1728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#define _FILE_OFFSET_BITS 64 1828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#define _LARGEFILE64_SOURCE 1 1928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <sys/types.h> 2028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <unistd.h> 2128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 2228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include "sparse_defs.h" 2328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include "sparse_format.h" 2428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include "sparse_crc32.h" 2528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 2628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <fcntl.h> 2728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <stdio.h> 2828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <stdlib.h> 2928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <string.h> 3028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <sys/types.h> 3128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <sys/stat.h> 3228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <sys/types.h> 3328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <sys/mman.h> 3428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#include <unistd.h> 3528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 3628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#define COPY_BUF_SIZE (1024*1024) 3728fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossu8 *copybuf; 3828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 3928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross/* This will be malloc'ed with the size of blk_sz from the sparse file header */ 4028fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossu8* zerobuf; 4128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 4228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#define SPARSE_HEADER_MAJOR_VER 1 4328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#define SPARSE_HEADER_LEN (sizeof(sparse_header_t)) 4428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross#define CHUNK_HEADER_LEN (sizeof(chunk_header_t)) 4528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 4628fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossvoid usage() 4728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 4828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Usage: simg2img <sparse_image_file> <raw_image_file>\n"); 4928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 5028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 5128fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossstatic int read_all(int fd, void *buf, size_t len) 5228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 5328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross size_t total = 0; 5428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int ret; 5528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross char *ptr = buf; 5628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 5728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross while (total < len) { 5828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = read(fd, ptr, len - total); 5928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 6028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret < 0) 6128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return ret; 6228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 6328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret == 0) 6428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return total; 6528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 6628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ptr += ret; 6728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross total += ret; 6828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 6928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 7028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return total; 7128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 7228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 7328fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossstatic int write_all(int fd, void *buf, size_t len) 7428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 7528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross size_t total = 0; 7628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int ret; 7728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross char *ptr = buf; 7828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 7928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross while (total < len) { 8028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = write(fd, ptr, len - total); 8128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 8228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret < 0) 8328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return ret; 8428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 8528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret == 0) 8628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return total; 8728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 8828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ptr += ret; 8928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross total += ret; 9028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 9128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 9228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return total; 9328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 9428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 9528fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossint process_raw_chunk(int in, int out, u32 blocks, u32 blk_sz, u32 *crc32) 9628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 9728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u64 len = (u64)blocks * blk_sz; 9828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int ret; 9928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int chunk; 10028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 10128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross while (len) { 10228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross chunk = (len > COPY_BUF_SIZE) ? COPY_BUF_SIZE : len; 10328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = read_all(in, copybuf, chunk); 10428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret != chunk) { 10528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "read returned an error copying a raw chunk: %d %d\n", 10628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret, chunk); 10728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 10828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 10928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross *crc32 = sparse_crc32(*crc32, copybuf, chunk); 11028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = write_all(out, copybuf, chunk); 11128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret != chunk) { 11228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "write returned an error copying a raw chunk\n"); 11328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 11428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 11528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross len -= chunk; 11628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 11728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 11828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return blocks; 11928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 12028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 12128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 12228fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossint process_fill_chunk(int in, int out, u32 blocks, u32 blk_sz, u32 *crc32) 12328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 12428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u64 len = (u64)blocks * blk_sz; 12528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int ret; 12628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int chunk; 12728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u32 fill_val; 12828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u32 *fillbuf; 12928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross unsigned int i; 13028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 13128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross /* Fill copy_buf with the fill value */ 13228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = read_all(in, &fill_val, sizeof(fill_val)); 13328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fillbuf = (u32 *)copybuf; 13428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross for (i = 0; i < (COPY_BUF_SIZE / sizeof(fill_val)); i++) { 13528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fillbuf[i] = fill_val; 13628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 13728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 13828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross while (len) { 13928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross chunk = (len > COPY_BUF_SIZE) ? COPY_BUF_SIZE : len; 14028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross *crc32 = sparse_crc32(*crc32, copybuf, chunk); 14128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = write_all(out, copybuf, chunk); 14228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret != chunk) { 14328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "write returned an error copying a raw chunk\n"); 14428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 14528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 14628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross len -= chunk; 14728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 14828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 14928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return blocks; 15028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 15128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 15228fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossint process_skip_chunk(int out, u32 blocks, u32 blk_sz, u32 *crc32) 15328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 15428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross /* len needs to be 64 bits, as the sparse file specifies the skip amount 15528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * as a 32 bit value of blocks. 15628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross */ 15728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u64 len = (u64)blocks * blk_sz; 15828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 15928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross lseek64(out, len, SEEK_CUR); 16028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 16128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return blocks; 16228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 16328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 16428fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossint process_crc32_chunk(int in, u32 crc32) 16528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 16628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u32 file_crc32; 16728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int ret; 16828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 16928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = read_all(in, &file_crc32, 4); 17028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret != 4) { 17128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "read returned an error copying a crc32 chunk\n"); 17228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 17328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 17428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 17528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (file_crc32 != crc32) { 17628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "computed crc32 of 0x%8.8x, expected 0x%8.8x\n", 17728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross crc32, file_crc32); 17828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 17928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 18028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 18128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross return 0; 18228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 18328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 18428fa5bc347390480fe190294c6c385b6a9f0d68bColin Crossint main(int argc, char *argv[]) 18528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross{ 18628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int in; 18728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int out; 18828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross unsigned int i; 18928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross sparse_header_t sparse_header; 19028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross chunk_header_t chunk_header; 19128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u32 crc32 = 0; 19228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross u32 total_blocks = 0; 19328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross int ret; 19428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 19528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (argc != 3) { 19628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross usage(); 19728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 19828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 19928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 20028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if ( (copybuf = malloc(COPY_BUF_SIZE)) == 0) { 20128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Cannot malloc copy buf\n"); 20228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 20328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 20428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 20528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (strcmp(argv[1], "-") == 0) { 20628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross in = STDIN_FILENO; 20728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } else { 20828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if ((in = open(argv[1], O_RDONLY)) == 0) { 20928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Cannot open input file %s\n", argv[1]); 21028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 21128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 21228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 21328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 21428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (strcmp(argv[2], "-") == 0) { 21528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross out = STDOUT_FILENO; 21628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } else { 21728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if ((out = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0666)) == 0) { 21828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Cannot open output file %s\n", argv[2]); 21928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 22028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 22128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 22228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 22328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = read_all(in, &sparse_header, sizeof(sparse_header)); 22428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret != sizeof(sparse_header)) { 22528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Error reading sparse file header\n"); 22628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 22728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 22828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 22928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (sparse_header.magic != SPARSE_HEADER_MAGIC) { 23028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Bad magic\n"); 23128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 23228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 23328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 23428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (sparse_header.major_version != SPARSE_HEADER_MAJOR_VER) { 23528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Unknown major version number\n"); 23628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 23728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 23828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 23928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (sparse_header.file_hdr_sz > SPARSE_HEADER_LEN) { 24028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross /* Skip the remaining bytes in a header that is longer than 24128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * we expected. 24228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross */ 24328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross lseek64(in, sparse_header.file_hdr_sz - SPARSE_HEADER_LEN, SEEK_CUR); 24428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 24528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 24628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if ( (zerobuf = malloc(sparse_header.blk_sz)) == 0) { 24728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Cannot malloc zero buf\n"); 24828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 24928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 25028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 25128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross for (i=0; i<sparse_header.total_chunks; i++) { 25228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross ret = read_all(in, &chunk_header, sizeof(chunk_header)); 25328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ret != sizeof(chunk_header)) { 25428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Error reading chunk header\n"); 25528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 25628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 25728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 25828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (sparse_header.chunk_hdr_sz > CHUNK_HEADER_LEN) { 25928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross /* Skip the remaining bytes in a header that is longer than 26028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * we expected. 26128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross */ 26228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross lseek64(in, sparse_header.chunk_hdr_sz - CHUNK_HEADER_LEN, SEEK_CUR); 26328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 26428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 26528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross switch (chunk_header.chunk_type) { 26628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross case CHUNK_TYPE_RAW: 26728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (chunk_header.total_sz != (sparse_header.chunk_hdr_sz + 26828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross (chunk_header.chunk_sz * sparse_header.blk_sz)) ) { 26928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Bogus chunk size for chunk %d, type Raw\n", i); 27028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 27128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 27228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross total_blocks += process_raw_chunk(in, out, 27328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross chunk_header.chunk_sz, sparse_header.blk_sz, &crc32); 27428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross break; 27528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross case CHUNK_TYPE_FILL: 27628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (chunk_header.total_sz != (sparse_header.chunk_hdr_sz + sizeof(u32)) ) { 27728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Bogus chunk size for chunk %d, type Fill\n", i); 27828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 27928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 28028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross total_blocks += process_fill_chunk(in, out, 28128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross chunk_header.chunk_sz, sparse_header.blk_sz, &crc32); 28228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross break; 28328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross case CHUNK_TYPE_DONT_CARE: 28428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (chunk_header.total_sz != sparse_header.chunk_hdr_sz) { 28528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Bogus chunk size for chunk %d, type Dont Care\n", i); 28628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 28728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 28828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross total_blocks += process_skip_chunk(out, 28928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross chunk_header.chunk_sz, sparse_header.blk_sz, &crc32); 29028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross break; 29128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross case CHUNK_TYPE_CRC32: 29228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross process_crc32_chunk(in, crc32); 29328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross break; 29428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross default: 29528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Unknown chunk type 0x%4.4x\n", chunk_header.chunk_type); 29628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 29728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 29828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 29928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 30028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross /* If the last chunk was a skip, then the code just did a seek, but 30128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * no write, and the file won't actually be the correct size. This 30228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * will make the file the correct size. Make sure the offset is 30328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross * computed in 64 bits, and the function called can handle 64 bits. 30428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross */ 30528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (ftruncate64(out, (u64)total_blocks * sparse_header.blk_sz)) { 30628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Error calling ftruncate() to set the image size\n"); 30728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 30828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 30928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 31028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross close(in); 31128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross close(out); 31228fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 31328fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross if (sparse_header.total_blks != total_blocks) { 31428fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross fprintf(stderr, "Wrote %d blocks, expected to write %d blocks\n", 31528fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross total_blocks, sparse_header.total_blks); 31628fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(-1); 31728fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross } 31828fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 31928fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross exit(0); 32028fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross} 32128fa5bc347390480fe190294c6c385b6a9f0d68bColin Cross 322