MPackageFile(MPackageEnt ent) { m_size = npk_entity_get_size(ent); m_buffer = new char[m_size]; npk_entity_read(ent, m_buffer); m_pos = m_buffer; }
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; }