/*
* =====================================================================================
*
* Filename: szbtool.c
*
* Description: 联想手机szb格式的制作工具，部分开源代码(仅提供程序思想)；
*
* Version: 1.0
* Created: 2013年03月25日 01时46分16秒
* Revision: none
* Compiler: gcc
*
* Author: linkscue (scue),
* Organization:
*
* =====================================================================================
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <time.h>
extern char *optarg;
extern int optind;
extern int opterr;
extern int optopt;
#define u8 unsigned char
#define u32 unsigned int
#define u16 unsigned short
#define INFOSIZE 256
#define BUFFER_SIZE 1024
typedef struct {
/*
* 保密信息：szb文件0~12字节
* */
}szb_header_t;
typedef struct {
/*
* 保密信息：szb文件12~256字节
*/
}szb_head_t;
typedef struct {
/*
* 保密信息：szb文件的结构体
*/
}szb_info;
typedef struct{
/*
*保密信息：存放image信息的结构体
* */
}szb_images_t;
/*获取文件长度*/
long getSize( FILE *fp )
{
long int save_pos; /* 保存当前文件指针 */
long size_of_file; /* 保存文件的大小 */
save_pos = ftell( fp ); /* 暂存当前文件指针位置 */
fseek( fp, 0L, SEEK_END ); /* 从文件开始位置移动至文件结尾 */
size_of_file = ftell( fp ); /* 获取文件指针位置 */
fseek( fp, save_pos, SEEK_SET ); /* 还原之前的文件指针位置 */
return( size_of_file ); /* 返回文件大小数值 */
}
unsigned int getSzbSum(char *file){
/*
*保密信息：获取szb文件校验码
* */
return sum;
}
/*计算一个文件的CRC校验*/
unsigned int getSum(FILE *fp)
{
/*
*保密信息：获取image文件校验码
* */
return sum; /* 这里的检验和的值是正确的 */
}
/* 分解szb函数 */
void splitFile(char *file){
FILE *fd = NULL;
FILE *ft = NULL;
int i;
get_szb_info(file); /* 显示szb文件信息 */
if ( (fd=fopen(file,"rb")) == NULL ) { /* 打开文件进行操作 */
printf ( "Extract szb file, open %s failure!\n", file );
exit(1);
}
fseek( fd, 0, SEEK_SET ); /* 重新定向文件指针位置 */
u32 imagecount = 0;
fseek( fd, 84, SEEK_SET ); /* 略过前边的84字节 */
fscanf( fd,"%4c", &imagecount ); /* 获取镜像文件个数 */
fseek( fd, 168, SEEK_CUR ); /* 总共略过前边的256字节的Header信息 */
szb_images_t images[10];
memset(images,0x00,sizeof(images));
for ( i=0; i < imagecount ; i++ ){
fscanf(fd,"%64c",&images[i].filename); /* 获取得文件名称 */
fscanf(fd,"%32c",&images[i].partname);
fscanf(fd,"%4c",&images[i].checksum);
fscanf(fd,"%4c",&images[i].timestamp);
fscanf(fd,"%4c",&images[i].imageoffset); /* 获取偏移位置 */
fscanf(fd,"%4c",&images[i].imagesize); /* 获取镜像文件的大小 */
fscanf(fd,"%4c",&images[i].eraseflag);
fscanf(fd,"%4c",&images[i].writeflag);
fscanf(fd,"%136c",&images[i].reserve);
}
//开始分解数据；
int size,n,offset,fp_local,end;
unsigned char imagename[32]="";
unsigned char buffer[BUFFER_SIZE]; /* 创建缓冲区 */
strncpy(buffer,"",sizeof(buffer)); /* 清空缓冲区内容 */
for( i=0; i < imagecount ; i++ ){
strncpy(imagename, images[i].filename, sizeof(imagename));
offset=images[i].imageoffset;
size=images[i].imagesize;
end=(offset+size);
if ( size != 0 ) {
if ( ( ft=fopen(imagename,"wb") ) == NULL ){
printf("Extract szb file, open %s failure!\n",imagename);
}
fseek( fd, offset, SEEK_SET); /* 跳转至数据段 */
printf("Extract %s..\n",imagename);
while ( fp_local != end ) {
n = fread(buffer,1, sizeof(buffer), fd);
fwrite(buffer, n, 1, ft);
fp_local=ftell(fd);
}
}
}
fclose(fd);
printf("Extract szb file done!\n");
}
/* 在尾部追加二进制文件(cat a >> b) */
void appendFile(char *fp, char *body) {
int n=0;
FILE *in,*out;
u8 buffer[BUFFER_SIZE];
if ( (in = fopen(fp, "rb")) == NULL){
printf ( "Open in file failure!\n" );
exit(1);
}
if ( (out=fopen( body , "ab")) == NULL ){
printf ( "Open out file failure!\n" );
exit(1);
}
while (!feof(in)) {
n = fread(buffer, 1, BUFFER_SIZE, in);
fwrite(buffer, 1, n, out);
}
fclose(in);
fclose(out);
}
/* 复制一个文件至另一个文件位置 */
//void copyFile(char *from, char *to){
// FILE * outfile, *infile;
// infile = fopen(from, "rb");
// outfile = fopen(to, "wb" );
// unsigned char buf[BUFFER_SIZE];
// if( outfile == NULL || infile == NULL )
// {
// printf("Copy file failure!");
// exit(1);
// }
// int rc;
// while( (rc = fread(buf,sizeof(unsigned char), BUFFER_SIZE ,infile)) != 0 )
// {
// fwrite( buf, sizeof( unsigned char ), rc, outfile );
// }
//// sleep(0.1);
// fclose(infile);
// fclose(outfile);
//}
/* 合并两个文件至一个文件 (cat a b > c) */
//void mergeFile(char *fp1,char *fp2,char *name){
// FILE *fd1,*fd2,*fp3;
// unsigned char buf[BUFFER_SIZE];
// char filename[100];
// strncpy(filename,name,sizeof(filename));
// int rc1,rc2;
// fd1 = fopen(fp1,"rb");
// fd2 = fopen(fp2,"rb");
// fp3 = fopen(filename, "wb" );
// while( (rc1 = fread(buf,sizeof(unsigned char), BUFFER_SIZE,fd1)) != 0 )
// {
// fwrite( buf, sizeof( unsigned char ), rc1, fp3 );
// }
// while( (rc2 = fread(buf,sizeof(unsigned char), BUFFER_SIZE,fd2)) != 0 )
// {
// fwrite( buf, sizeof( unsigned char ), rc2, fp3 );
// }
// sleep(0.1);
// fclose(fd1);
// fclose(fd2);
// fclose(fp3);
//}
/* appendImage(szb文件, image文件，文件名称，分区位置，偏移量，索引位置) */
u32 appendImage(char *szb, char *file, char *filename, char *partname, u32 offset, int index){
FILE *fp,*fb;
if((fp=fopen(file,"rb")) == NULL){
printf ( "Open %s failure!\n",filename );
exit(1);
}
szb_images_t image;
strncpy(image.filename, filename, sizeof(image.filename));
strncpy(image.partname, partname, sizeof(image.partname));
strncpy(image.reserve, "", sizeof(image.reserve));
time(&image.timestamp);
image.imageoffset=offset;
image.imagesize=getSize(fp);
image.checksum=getSum(fp);
image.eraseflag=1;
image.writeflag=1;
fclose(fp);
printf("Adding: %-12sOffset: 0x%08x Checksum: 0x%08x\n",image.partname, image.imageoffset, image.checksum);
if ( (fb=fopen(szb, "rb+")) == NULL) {
printf(" append %s to szb file, open szb file failure!\n", file);
exit(1);
}
fseek( fb, ( index * INFOSIZE ), SEEK_SET ); /* 移动文件指针的位置 */
fwrite(&image.filename, sizeof(image.filename), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.partname, sizeof(image.partname), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.checksum, sizeof(image.checksum), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.timestamp, sizeof(image.timestamp), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.imageoffset, sizeof(image.imageoffset), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.imagesize, sizeof(image.imagesize), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.eraseflag, sizeof(image.eraseflag), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.writeflag, sizeof(image.writeflag), 1, fb); /* 向szb文件写入Image的相关信息 */
fwrite(&image.reserve, sizeof(image.reserve), 1, fb); /* 向szb文件写入Image的相关信息 */
fclose(fb); /* 向szb文件写完Image信息后关闭 */
appendFile(file, szb); /* 向szb文件的尾部添加image数据 */
return (image.imagesize+image.imageoffset); /* 返回一个偏移数据，方便之后的判断 */
}
void get_szb_info(char *file){
FILE *fp;
int i,count;
szb_info info;
if ( (fp=fopen(file, "rb")) == NULL ) {
/* code */
}
fscanf(fp,"%8c",&info.magic);
/*
*保密信息: 获取szb头部信息
* */
fscanf(fp,"%136c",&info.reserve);
u8 magic[]="LmSzBfMt";
if ( strstr(info.magic,magic) == NULL ){
printf ( "This file not in szb format, abort!\n" );
exit(1);
}
struct tm *ptr; /* 格式化时间输出 */
time_t lt;
unsigned int times=info.timestamp;
char str[80];
lt=times;
ptr=localtime(<);
strftime(str,100,"%F%X",ptr);
printf ( "\n" );
printf ( "Header:\n" );
printf ( "Checksum:\t0x%08x\n",info.checksum );
printf ( "Size:\t\t0x%08x\n",info.filesize );
printf ( "Author:\t\t%s\n",info.author );
printf ( "Version:\t%s\n",info.version );
printf ( "Times:\t\t%s\n",str );
printf ( "Counts:\t\t0x%x\n",info.imagecount );
count=info.imagecount;
szb_images_t images[10];
printf ( "\n" );
printf ( "Images:\n" );
for ( i=0; i < info.imagecount ; i++ ){
fscanf(fp,"%64c",&images[i].filename);
fscanf(fp,"%32c",&images[i].partname);
fscanf(fp,"%4c",&images[i].checksum);
fscanf(fp,"%4c",&images[i].timestamp);
fscanf(fp,"%4c",&images[i].imageoffset);
fscanf(fp,"%4c",&images[i].imagesize);
fscanf(fp,"%4c",&images[i].eraseflag);
fscanf(fp,"%4c",&images[i].writeflag);
fscanf(fp,"%136c",&images[i].reserve);
if ( strlen(images[i].filename) != 0 ){
printf ( "Position: %-10sOffset: 0x%08x Checksum: 0x%08x\n",images[i].partname, images[i].imageoffset, images[i].checksum );}
else{
printf ( "Earse: %-10sCaution!\n",images[i].partname);
}
}
printf ( "\n" );
fclose(fp);
}
/* 显示szbtool作者信息 */
void author_info(){
printf ( "\n" );
printf ( "===============================================================\n" );
printf ( " Welcome to use Lenovo K860/K860i szbtool! \n" );
printf ( "\n" );
printf ( " -- version: 0.14 \n" );
printf ( " -- author: linkscue \n" );
printf ( "===============================================================\n" );
}
/* 显示使用帮助 */
void usage(){
printf ( "\n" );
printf ( "---------------------------------------------------------------\n" );
printf ( "usage:\n" );
printf ( "szbtool -b uboot # add uboot.bin\n" );
printf ( "szbtool -k boot # add boot.img\n" );
printf ( "szbtool -r recovery # add recovery.img\n" );
printf ( "szbtool -s system # add system.img\n" );
printf ( "szbtool -c cpimage # add cpimage.img\n" );
printf ( "szbtool -p preload # add preload.img\n" );
printf ( "szbtool -d userdata # add userdata.img\n" );
printf ( "szbtool -e erase # erase data & cache space\n" );
printf ( "szbtool -a author # specify author\n" );
printf ( "szbtool -v version # specify version\n" );
printf ( "szbtool -i szb # szb information\n" );
printf ( "szbtool -x szb # extract szb \n" );
printf ( "szbtool -h # help\n" );
printf ( "---------------------------------------------------------------\n" );
printf ( "e.g. szbtool -k boot.img -s system.img -a scue@Link -v Link.szb\n" );
printf ( "\n" );
}
/*主函数*/
int main ( int argc, char *argv[] )
{
FILE *fp,*fp1,*fp2,*fd1,*fd2,*sub_image;
char *file=NULL;
u8 magic[8];
u8 imagename[64];
u8 partname[32];
u8 version[32];
u8 author[32];
u32 filesize;
u32 offset=8192; /* 初始的偏移量 */
char opt; /* 获取输入的选项 */
int index=0; /* 判断szb中的分区索引数量 */
int DO_NOTHING=0; /* 依据此变量判断是否执行 */
int ERASE_DATA=0; /* 依据此变量判断是否擦除数据 */
//清空变量
memset(version,0x00,sizeof(version));
memset(imagename,0x00,sizeof(imagename));
memset(partname,0x00,sizeof(partname));
strncpy(author,"scue@Link",sizeof(author));
//显示作者信息
author_info();
//检测输入参数
if(argc == 1)
{
usage(); /* 显示使用手册 */
exit(0);
}
//判断选项中是否有版本
int i;
char tmp1[32]="";
char tmp2[32]="";
strncpy(tmp2,"-v",sizeof(tmp2));
for (i = 0; i < argc; i++) {
strncpy(tmp1,argv[i],sizeof(tmp1));
if ( strcmp(tmp1,tmp2) == 0 ) {
strncpy(version,argv[i+1],sizeof(version));
}
if ( strcmp(tmp1,"-x") == 0 || strcmp(tmp1,"-i")==0 ) {
DO_NOTHING = 1;
}
}
if ( strlen(version) == 0 && DO_NOTHING !=1 ){
printf ( "\n" );
printf ( "Please use -v to specify a version name!\n" );
usage();
exit(1);
}
if (DO_NOTHING != 1) { /* 当发现有事可做的时候... */
//创建文件
remove(version); /* delete the old szb file */
if ( (fp=fopen(version,"wb+")) == NULL ) {
printf ( "Create version file %s failure!\n",version);
exit(1);
}
u8 header_buffer[INFOSIZE*32];
strncpy(header_buffer, "LmSzBfMt", sizeof(header_buffer));
fwrite(header_buffer, sizeof(header_buffer) , 1, fp); /* 向其中写8192字节后退出 */
index++;
fclose(fp);
}
//开始获取选项信息
opterr=0; /* 不显示错误的选项信息 */
while ((opt = getopt(argc, argv, "b:k:r:s:c:p:d:x:i:v:a:e")) != -1)
switch (opt) {
case 'v':
strncpy(version,optarg,sizeof(version));
break;
case 'b':
file=optarg;
strncpy(partname,"bootloader",sizeof(partname));
strncpy(imagename,"uboot.bin",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 'k':
file=optarg;
strncpy(partname,"boot",sizeof(partname));
strncpy(imagename,"boot.img",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 'r':
file=optarg;
strncpy(partname,"recovery",sizeof(partname));
strncpy(imagename,"recovery.img",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 's':
file=optarg;
strncpy(partname,"system",sizeof(partname));
strncpy(imagename,"system.img",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 'c':
file=optarg;
strncpy(partname,"cpimage",sizeof(partname));
strncpy(imagename,"cpimage.img",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 'p':
file=optarg;
strncpy(partname,"preload",sizeof(partname));
strncpy(imagename,"preload.img",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 'd':
file=optarg;
strncpy(partname,"userdata",sizeof(partname));
strncpy(imagename,"userdata.img",sizeof(imagename));
offset=appendImage(version, file, imagename, partname, offset, index);
index++;
break;
case 'e':
ERASE_DATA=1;
break;
case 'a':
strncpy(author,optarg,sizeof(author));
break;
case 'x':
DO_NOTHING=1;
file=optarg;
splitFile(file);
break;
case 'i':
DO_NOTHING=1;
file=optarg;
get_szb_info(file);
break;
default:
DO_NOTHING=1;
usage();
}
argv += optind;
//开始执行制作szb文件
if ( DO_NOTHING != 1 ){
//检查是否输入格式名
if ( strlen(version) == 0 ){
DO_NOTHING = 1;
printf ( "\n" );
printf ( "Please use -v to specify a version name!\n" );
usage();
exit(1);
}
// 双清功能
if ( ERASE_DATA == 1) { /* 判断是否有“双清功能” */
u8 erase_data[32];
u8 erase_cache[32];
strncpy(erase_data, "userdata", sizeof(erase_data));
strncpy(erase_cache, "cache", sizeof(erase_data));
if ( (fp=fopen(version, "rb+")) == NULL ) {
printf("Add erase features, open file %s failure!\n", version);
exit(1);
}
fseek( fp, (INFOSIZE * index + 64 ), SEEK_SET ); /* erase userdata */
fwrite(erase_data, sizeof(erase_data), 1, fp);
index++;
fseek( fp, (INFOSIZE * index + 64 ), SEEK_SET ); /* erase cache */
fwrite(erase_cache, sizeof(erase_cache), 1, fp);
index++;
fclose(fp);
}
//制作SZB格式文件头部(256-12)部分
szb_head_t head;
if ( (fp=fopen(version,"rb+")) ==NULL ){
printf("Calculate file size, open file %s failure!\n", version);
exit(1);
}
head.filesize=getSize(fp); /* get szb file size */
/*
* 保密信息：向szb文件写入数据
* */
time(&head.timestamp); /* 写入时间信息 */
/*
* 保密信息：向szb文件写入数据
* */
fseek( fp, 12, SEEK_SET );
fwrite(&head,sizeof(szb_head_t),1,fp); /* 写入256-12字节部分 */
fclose(fp);
u32 szb_checksum;
szb_checksum=getSzbSum(version);
if ( (fp=fopen(version,"rb+")) ==NULL ){
printf("Calculate file size, open file %s failure!\n", version);
exit(1);
}
fseek(fp, 8, SEEK_SET);
fwrite(&szb_checksum,sizeof(szb_checksum),1,fp); /* 写入总的验证码字节部分 */
fclose(fp);
get_szb_info(version);
printf ( ">> Congratulations! The Official ROM %s done!\n",version);
printf ( "\n" );
}
return 0;
}
