#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "pid.h"
#define LOADER_SIZE (64<<10)

static sercomm_pid_t image_pid;

int search_pid(char *start,char *end,char *string)
{
	int len=strlen("sErCoMm");
	char *pt=start;
	
	while(memcmp(pt,"sErCoMm",len)!=0 && (pt+len) < end) 
		pt++;		
	
	/* Found */
	if(pt+len < end){
		memcpy(&image_pid,pt,sizeof(image_pid));
		return strcmp(image_pid.hw_id,string);
	}
	return -1;
}

int my_read_file(char *filename,char *buf,int size)
{
	FILE *fp;
	int len=0;
	if((fp=fopen(filename,"rb"))==NULL){
		fprintf(stderr,"Error opening %s for output\n",filename);
		exit(-1);
	}
	fseek(fp,0,SEEK_END);
	len=ftell(fp);
	
	if(len > size){
		fclose(fp);
		fprintf(stderr,"Sorry, %s must less than %d\n", filename,size);
		exit(-1);
	}
	fseek(fp, 0L, SEEK_SET);
	len=fread(buf,1,size,fp);
	fclose(fp);
	return len;
}

int main(int argc,char **argv)
{
	FILE	*fp;
	char    buf[4096<<10];
	int	len=0;
#if 1	
	if (argc != 4){
		printf("Usage: %s outImage inImage fs\n", argv[0]);
		return -1;
	}
#endif
	memset(buf,0xFF,sizeof(buf));
	/* adam2 */
	len=my_read_file(argv[2],buf,sizeof(buf));
	
	if(search_pid(buf+(len-(9<<10)),buf+len,"DG834GT")!=0){
		printf("Image Error!\n");
		return -1;
	}

	memset(buf+LOADER_SIZE,0xFF,sizeof(buf)-LOADER_SIZE);
	len=my_read_file(argv[3],buf+LOADER_SIZE,sizeof(buf)-LOADER_SIZE);
	
	memcpy(buf+LOADER_SIZE+len,&image_pid,sizeof(image_pid));
		
	/* write image */
	fp=fopen(argv[1],"wb");	
	if(fp==NULL){
		fprintf(stderr,"Error opening %s for output\n",argv[0]);
		return -1;
		
	}
	len=(1+(LOADER_SIZE+len+sizeof(image_pid))/8192)*8192;
	fwrite(&buf,len,1,fp);
	fclose(fp);
	return 0;
}