NPK_PACKAGE npk_package_open( NPK_CSTR filename, NPK_TEAKEY teakey[4] )
{
NPK_PACKAGEBODY* pb = NULL;
NPK_RESULT res;
res = npk_package_alloc( (NPK_PACKAGE*)&pb, teakey );
if( res != NPK_SUCCESS )
return NULL;
if( npk_package_init( pb ) != NPK_SUCCESS )
goto npk_package_open_return_null_with_free;
if( npk_open( &pb->handle_, filename, false, false ) != NPK_SUCCESS )
goto npk_package_open_return_null_with_free;
res = __npk_package_open( pb, filename, 0, teakey );
if( res != NPK_SUCCESS )
goto npk_package_open_return_null_with_free;
return (NPK_PACKAGE*)pb;
npk_package_open_return_null_with_free:
if( pb )
npk_package_close( pb );
return NULL;
}
NPK_PACKAGE npk_package_open_with_fd( NPK_CSTR name, int fd, long offset, long size, NPK_TEAKEY teakey[4] )
{
NPK_PACKAGEBODY* pb = NULL;
NPK_RESULT res;
res = npk_package_alloc( (NPK_PACKAGE*)&pb, teakey );
if( res != NPK_SUCCESS )
return NULL;
if( npk_package_init( pb ) != NPK_SUCCESS )
goto npk_package_open_return_null_with_free;
pb->handle_ = fd;
pb->usingFdopen_ = true;
pb->offsetJump_ = offset;
npk_seek( fd, offset, SEEK_CUR );
res = __npk_package_open( pb, name, size, teakey );
if( res != NPK_SUCCESS )
goto npk_package_open_return_null_with_free;
return (NPK_PACKAGE*)pb;
npk_package_open_return_null_with_free:
if( pb )
npk_package_close( pb );
return NULL;
}
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::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_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;
}
