MPackageEnt MPackageManagerNPK::findEntity(const char* name)
{
for(int i = M_MAX_PACKAGES-1; i >=0; --i)
{
if(m_packages[i])
{
MPackageNPK* pack = (MPackageNPK*)m_packages[i];
if(NPK_ENTITY ent = npk_package_get_entity(pack->package, name))
{
return ent;
}
}
}
return 0;
}
int libnpk_read_entity_partial( int argc, char * argv [] )
{
int teakey[4] = {98521,16322,7163,992};
NPK_PACKAGE pack = npk_package_open( "sample.npk", teakey );
CHECK( pack != NULL );
std::string entityNames[2] = { "sample.txt", "tea.txt" };
for( int i = 0; i < 2; ++i )
{
NPK_ENTITY entity = npk_package_get_entity( pack, entityNames[i].c_str() );
CHECK( entity != NULL );
NPK_SIZE size = npk_entity_get_size( entity );
void* buf = malloc( size );
NPK_FLAG flag;
npk_entity_get_current_flag( entity, &flag );
if( flag & ( NPK_ENTITY_ENCRYPT_TEA | NPK_ENTITY_ENCRYPT_XXTEA ) )
{
// offset must be aligned by 8 bytes
CHECK( (!npk_entity_read_partial( entity, buf, 9, 32 )) );
// Size can be not aligned by 8 bytes when read end of file
NPK_SIZE offset = size - (size % 8) - 32;
CHECK( npk_entity_read_partial( entity, buf, offset, size - offset ) );
CHECK_EQUAL_STR_WITH_FILE_PARTIAL( (const char*)buf, "sample.txt", offset, size - offset );
}
CHECK( npk_entity_read_partial( entity, buf, 8, 32 ) );
CHECK_EQUAL_STR_WITH_FILE_PARTIAL( (const char*)buf, "sample.txt", 8, 32 );
free( buf );
}
npk_package_close( pack );
return 0;
}
int libnpk_streamable( int argc, char * argv [] )
{
int teakey[4] = {1,2,3,4};
NPK_PACKAGE pack;
NPK_ENTITY entity;
// create a pack
CHECK( NPK_SUCCESS == npk_package_alloc( &pack, teakey ) );
CHECK( NPK_SUCCESS == npk_package_add_file( pack, "sample.txt", "sample.txt", &entity ) );
CHECK( NPK_SUCCESS == npk_package_add_file( pack, "sample.txt", "tea.txt", &entity ) );
CHECK( NPK_SUCCESS == npk_entity_set_flag( entity, NPK_ENTITY_ENCRYPT_TEA ) );
CHECK( NPK_SUCCESS == npk_package_add_file( pack, "sample.txt", "xxtea.txt", &entity ) );
CHECK( NPK_SUCCESS == npk_entity_set_flag( entity, NPK_ENTITY_ENCRYPT_XXTEA ) );
CHECK( NPK_SUCCESS == npk_package_add_file( pack, "sample.txt", "zip.txt", &entity ) );
CHECK( NPK_SUCCESS == npk_entity_set_flag( entity, NPK_ENTITY_COMPRESS_ZLIB ) );
CHECK( NPK_SUCCESS == npk_package_save( pack, "foo.npk", true ) );
npk_package_close( pack );
// simulate download
int rh = open( "foo.npk", O_RDONLY | O_BINARY );
size_t filesize = npk_seek( rh, 0, SEEK_END );
npk_seek( rh, 0, SEEK_SET );
int wh = open( "foo_2.npk", O_CREAT | O_RDWR | O_TRUNC | O_BINARY, S_IREAD | S_IWRITE );
// validation
std::string entityNames[4] = { "sample.txt", "tea.txt", "xxtea.txt", "zip.txt" };
pack = 0;
int i = 0;
size_t offset = 0;
char buf[255];
while( offset < filesize )
{
size_t r = rand()%16;
if( r + offset > filesize )
r = filesize - offset;
read( rh, &buf, sizeof(char)*r );
write( wh, buf, sizeof(char)*r );
printf( "offset %ld, reading %ld byte(s).\n", offset, r );
offset += r;
if( pack == 0 )
{
pack = npk_package_open( "foo_2.npk", teakey );
if( pack != 0 )
printf( " package loaded.\n" );
}
else
{
NPK_ENTITY entity = npk_package_get_entity( pack, entityNames[i].c_str() );
CHECK( entity != NULL );
NPK_SIZE size = npk_entity_get_size( entity );
if( npk_entity_is_ready( entity ) )
{
printf( " entity %s ready.\n", entityNames[i].c_str() );
void* buf = malloc( size );
CHECK( npk_entity_read( entity, buf ) );
CHECK_EQUAL_STR_WITH_FILE( (const char*)buf, "sample.txt" );
free( buf );
++i;
}
}
}
npk_package_close( pack );
close( wh );
close( rh );
return 0;
}