int npk_open(lua_State* L)
{
npkg* pkg = new npkg;
snprintf(pkg->path, 255, lua_tostring(L, -1));
int key[4] = {0, 0, 0, 0};
FILE* fp = fopen(pkg->path, "r");
if(fp == 0)
npk_package_alloc(&pkg->pkg, key);
else
pkg->pkg = npk_package_open(pkg->path, key);
lua_pushnumber(L, pkg->id);
return 1;
}
MPackage MPackageManagerNPK::loadPackage(const char* packageName)
{
MPackageNPK* pack = new MPackageNPK;
if(!(pack->package = npk_package_open(packageName, teakey)))
{
delete pack;
return 0;
}
pack->filename = packageName;
if(!mountPackage(pack))
{
npk_package_close(pack->package);
delete pack;
pack = 0;
}
return pack;
}
MPackage MPackageManagerNPK::openPackage(const char* packageName)
{
#ifdef M_PACKAGE_WRITABLE
MPackageNPK* package = new MPackageNPK;
if(isFileExist(packageName))
{
package->package = npk_package_open(packageName, teakey);
}
else
{
npk_package_alloc(&package->package, teakey);
}
if(package->package)
{
package->filename = packageName;
return package;
}
delete package;
#endif
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;
}
