| /* |
| ******************************************************************************* |
| * |
| * Copyright (C) 1999, International Business Machines |
| * Corporation and others. All Rights Reserved. |
| * |
| ******************************************************************************* |
| * file name: gencmn.c |
| * encoding: US-ASCII |
| * tab size: 8 (not used) |
| * indentation:4 |
| * |
| * created on: 1999nov01 |
| * created by: Markus W. Scherer |
| * |
| * This program reads a list of data files and combines them |
| * into one common, memory-mappable file. |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include "unicode/utypes.h" |
| #include "cmemory.h" |
| #include "cstring.h" |
| #include "filestrm.h" |
| #include "toolutil.h" |
| #include "unewdata.h" |
| |
| #define STRING_STORE_SIZE 100000 |
| #define MAX_FILE_COUNT 2000 |
| |
| #define COMMON_DATA_NAME "icudata" |
| #define DATA_TYPE "dat" |
| |
| /* UDataInfo cf. udata.h */ |
| static const UDataInfo dataInfo={ |
| sizeof(UDataInfo), |
| 0, |
| |
| U_IS_BIG_ENDIAN, |
| U_CHARSET_FAMILY, |
| sizeof(UChar), |
| 0, |
| |
| 0x43, 0x6d, 0x6e, 0x44, /* dataFormat="CmnD" */ |
| 1, 0, 0, 0, /* formatVersion */ |
| 3, 0, 0, 0 /* dataVersion */ |
| }; |
| |
| static uint32_t maxSize; |
| |
| static char stringStore[STRING_STORE_SIZE]; |
| static uint32_t stringTop=0, basenameTotal=0; |
| |
| typedef struct { |
| char *pathname, *basename; |
| uint32_t basenameLength, basenameOffset, fileSize, fileOffset; |
| } File; |
| |
| static File files[MAX_FILE_COUNT]; |
| static uint32_t fileCount=0; |
| |
| /* prototypes --------------------------------------------------------------- */ |
| |
| static void |
| addFile(const char *filename); |
| |
| static char * |
| allocString(uint32_t length); |
| |
| static int |
| compareFiles(const void *file1, const void *file2); |
| |
| /* -------------------------------------------------------------------------- */ |
| |
| extern int |
| main(int argc, char *argv[]) { |
| static uint8_t buffer[4096]; |
| char line[512]; |
| FileStream *in, *file; |
| UNewDataMemory *out; |
| char *s; |
| UErrorCode errorCode=U_ZERO_ERROR; |
| uint32_t i, fileOffset, basenameOffset, length; |
| |
| if(argc<=1) { |
| fprintf(stderr, |
| "usage: %s maxsize [list-filename]\n" |
| "\tread the list file (default: stdin) and \n" |
| "\tcreate " COMMON_DATA_NAME "." DATA_TYPE " from all the files listed but each not larger than maxsize\n", |
| argv[0]); |
| } |
| |
| if(argc<2) { |
| return U_ILLEGAL_ARGUMENT_ERROR; |
| } |
| |
| maxSize=uprv_strtoul(argv[1], NULL, 0); |
| if(maxSize==0) { |
| fprintf(stderr, "gencmn: maxSize %s not valid\n", argv[1]); |
| exit(U_ILLEGAL_ARGUMENT_ERROR); |
| } |
| |
| if(argc==2) { |
| in=T_FileStream_stdin(); |
| } else { |
| in=T_FileStream_open(argv[2], "r"); |
| if(in==NULL) { |
| fprintf(stderr, "gencmn: unable to open input file %s\n", argv[2]); |
| exit(U_FILE_ACCESS_ERROR); |
| } |
| } |
| |
| /* read the list of files and get their lengths */ |
| while(T_FileStream_readLine(in, line, sizeof(line))!=NULL) { |
| /* remove trailing newline characters */ |
| s=line; |
| while(*s!=0) { |
| if(*s=='\r' || *s=='\n') { |
| *s=0; |
| break; |
| } |
| ++s; |
| } |
| |
| addFile(getLongPathname(line)); |
| } |
| |
| if(in!=T_FileStream_stdin()) { |
| T_FileStream_close(in); |
| } |
| |
| /* sort the files by basename */ |
| qsort(files, fileCount, sizeof(File), compareFiles); |
| |
| /* determine the offsets of all basenames and files in this common one */ |
| basenameOffset=4+8*fileCount; |
| fileOffset=basenameOffset+(basenameTotal+15)&~0xf; |
| for(i=0; i<fileCount; ++i) { |
| files[i].fileOffset=fileOffset; |
| fileOffset+=(files[i].fileSize+15)&~0xf; |
| files[i].basenameOffset=basenameOffset; |
| basenameOffset+=files[i].basenameLength; |
| } |
| |
| /* create the output file */ |
| out=udata_create(DATA_TYPE, COMMON_DATA_NAME, &dataInfo, U_COPYRIGHT_STRING, &errorCode); |
| if(U_FAILURE(errorCode)) { |
| fprintf(stderr, "gencmn: unable to open output file - error %s\n", u_errorName(errorCode)); |
| exit(errorCode); |
| } |
| |
| /* write the table of contents */ |
| udata_write32(out, fileCount); |
| for(i=0; i<fileCount; ++i) { |
| udata_write32(out, files[i].basenameOffset); |
| udata_write32(out, files[i].fileOffset); |
| } |
| |
| /* write the basenames */ |
| for(i=0; i<fileCount; ++i) { |
| udata_writeString(out, files[i].basename, files[i].basenameLength); |
| } |
| length=4+8*fileCount+basenameTotal; |
| |
| /* copy the files */ |
| for(i=0; i<fileCount; ++i) { |
| /* pad to 16-align the next file */ |
| length&=0xf; |
| if(length!=0) { |
| udata_writePadding(out, 16-length); |
| } |
| |
| /* copy the next file */ |
| file=T_FileStream_open(files[i].pathname, "rb"); |
| if(file==NULL) { |
| fprintf(stderr, "gencmn: unable to open listed file %s\n", files[i].pathname); |
| exit(U_FILE_ACCESS_ERROR); |
| } |
| for(;;) { |
| length=T_FileStream_read(file, buffer, sizeof(buffer)); |
| if(length==0) { |
| break; |
| } |
| udata_writeBlock(out, buffer, length); |
| } |
| T_FileStream_close(file); |
| length=files[i].fileSize; |
| } |
| |
| /* finish */ |
| udata_finish(out, &errorCode); |
| if(U_FAILURE(errorCode)) { |
| fprintf(stderr, "gencmn: error finishing output file - %s\n", u_errorName(errorCode)); |
| exit(errorCode); |
| } |
| |
| return 0; |
| } |
| |
| static void |
| addFile(const char *filename) { |
| FileStream *file; |
| char *s; |
| uint32_t length; |
| |
| if(fileCount==MAX_FILE_COUNT) { |
| fprintf(stderr, "gencmn: too many files\n"); |
| exit(U_BUFFER_OVERFLOW_ERROR); |
| } |
| |
| /* try to open the file */ |
| file=T_FileStream_open(filename, "rb"); |
| if(file==NULL) { |
| fprintf(stderr, "gencmn: unable to open listed file %s\n", filename); |
| exit(U_FILE_ACCESS_ERROR); |
| } |
| |
| /* get the file length */ |
| length=T_FileStream_size(file); |
| if(T_FileStream_error(file) || length<=20) { |
| fprintf(stderr, "gencmn: unable to get length of listed file %s\n", filename); |
| exit(U_FILE_ACCESS_ERROR); |
| } |
| T_FileStream_close(file); |
| |
| /* do not add files that are longer than maxSize */ |
| if(length>maxSize) { |
| return; |
| } |
| files[fileCount].fileSize=length; |
| |
| /* store the pathname */ |
| length=uprv_strlen(filename)+1; |
| s=allocString(length); |
| uprv_memcpy(s, filename, length); |
| files[fileCount].pathname=s; |
| |
| /* get the basename */ |
| s=(char *)findBasename(s); |
| files[fileCount].basename=s; |
| length=uprv_strlen(s)+1; |
| files[fileCount].basenameLength=length; |
| basenameTotal+=length; |
| |
| ++fileCount; |
| } |
| |
| static char * |
| allocString(uint32_t length) { |
| uint32_t top=stringTop+length; |
| char *p; |
| |
| if(top>STRING_STORE_SIZE) { |
| fprintf(stderr, "gencmn: out of memory\n"); |
| exit(U_MEMORY_ALLOCATION_ERROR); |
| } |
| p=stringStore+stringTop; |
| stringTop=top; |
| return p; |
| } |
| |
| static int |
| compareFiles(const void *file1, const void *file2) { |
| /* sort by basename */ |
| return uprv_strcmp(((File *)file1)->basename, ((File *)file2)->basename); |
| } |