void extract( POLE::Storage* storage, char* stream_name, char* outfile ) { POLE::Stream* stream = new POLE::Stream( storage, stream_name ); if( !stream ) return; if( stream->fail() ) return; std::ofstream file; file.open( outfile, std::ios::binary|std::ios::out ); unsigned char buffer[16]; for( ;; ) { unsigned read = stream->read( buffer, sizeof( buffer ) ); file.write( (const char*)buffer, read ); if( read < sizeof( buffer ) ) break; } file.close(); delete stream; }
void CFStream::copy( std::wstring streamNameCreate, POLE::Storage * storageOut) { stream_->seek(0); int size_stream = stream_->size(); POLE::Stream *streamNew = new POLE::Stream(storageOut, streamNameCreate, true, size_stream); if (!streamNew) return; unsigned char buffer[4096]; int bytesRead = 0; while(true) { int bytesToRead = size_stream - bytesRead; if (bytesToRead <= 0) break; if (bytesToRead > 4096) bytesToRead = 4096; stream_->read(buffer, bytesToRead); streamNew->write(buffer, bytesToRead); bytesRead += bytesToRead; } //unsigned char* data_stream = new unsigned char[size_stream + 64]; //memset(data_stream, 0, size_stream + 64); //if (data_stream) //{ // stream->read(data_stream, size_stream); // streamNew->write(data_stream, size_stream); // delete []data_stream; // data_stream = NULL; //} streamNew->flush(); delete streamNew; }
void dump( POLE::Storage* storage, char* stream_name ) { POLE::Stream* stream = new POLE::Stream( storage, stream_name ); if( !stream ) return; if( stream->fail() ) return; // std::cout << "Size: " << stream->size() << " bytes" << std::endl; unsigned char buffer[16]; for( ;; ) { unsigned read = stream->read( buffer, sizeof( buffer ) ); for( unsigned i = 0; i < read; i++ ) printf( "%02x ", buffer[i] ); std::cout << " "; for( unsigned i = 0; i < read; i++ ) printf( "%c", ((buffer[i]>=32)&&(buffer[i]<128)) ? buffer[i] : '.' ); std::cout << std::endl; if( read < sizeof( buffer ) ) break; } delete stream; }
void visit( int indent, POLE::Storage* storage, std::string path ) { std::list<std::string> entries; entries = storage->entries( path ); std::list<std::string>::iterator it; for( it = entries.begin(); it != entries.end(); ++it ) { std::string name = *it; std::string fullname = path + name; for( int j = 0; j < indent; j++ ) std::cout << " "; POLE::Stream* ss = new POLE::Stream( storage, fullname ); std::cout << name; if( ss ) if( !ss->fail() )std::cout << " (" << ss->size() << ")"; std::cout << std::endl; delete ss; if( storage->isDirectory( fullname ) ) visit( indent+1, storage, fullname + "/" ); } }
// Extract encryption data from Microsoft Word file // Place 48 bytes at record_out: // 16 byte Salt, followed by // 16 byte EncryptedVerifier, followed by // 16 byte EncryptedVerifierHash extern "C" void extract_doc(const char *file_name, unsigned char *record_out) { int n; // Used for number of bytes read POLE::Storage* storage = new POLE::Storage(file_name ); storage->open(); if( storage->result() != POLE::Storage::Ok ) { std::cout << "Error on file " << std::endl; exit(1); return; } // Check File Information Block to see if document is encrypted POLE::Stream* stream = new POLE::Stream( storage, "WordDocument" ); if (!stream || stream->fail() ) { std::cerr << "Could not open WordDocument stream\n"; exit(1); } // FibBase should be at the beginning of the WordDocument stream unsigned char wIdent[2]; n = stream->read(wIdent, 2); if (n != 2) return; if (! (wIdent[0] == 0xEC && wIdent[1] == 0xA5 )) { std::cerr << "FibBase not found\n"; exit(1); return; } // fEncrypted is in 12th byte of FibBase. 2 bytes have been read // already and 12 = 2 + 10. unsigned char byte; for (int i = 1; i <= 10; i++) { n = stream->read(&byte, 1); if (n != 1) { std::cerr << "Error reading FibBase\n"; exit(1); } } if ((byte & 0x01) == 0) { std::cerr << "File is not encrypted\n"; exit(1); } // Look for encryption header in 1Table or 0Table stream // See http://msdn.microsoft.com/en-us/library/dd923367(v=office.12).aspx // and http://msdn.microsoft.com/en-us/library/dd908560(v=office.12).aspx delete stream; stream = new POLE::Stream( storage, "1Table" ); if (!stream || stream->fail() ) { stream = new POLE::Stream( storage, "0Table" ); } if (!stream || stream->fail() ) { std::cerr << "Couldn't open 1Table or 0Table stream\n"; exit(1); } unsigned char EncryptionHeader[52]; n = stream->read(EncryptionHeader, 52); memcpy(record_out, EncryptionHeader + 4, 48); }
KoFilter::ConversionStatus HancomWordImport::convert( const QByteArray& from, const QByteArray& to ) { if ( from != "application/x-hwp" ) return KoFilter::NotImplemented; if ( to != "application/vnd.oasis.opendocument.text" ) return KoFilter::NotImplemented; d->inputFile = m_chain->inputFile(); d->outputFile = m_chain->outputFile(); d->paragraphs.clear(); POLE::Storage storage( QFile::encodeName(d->inputFile) ); if( !storage.open() ) return KoFilter::WrongFormat; POLE::Stream* stream; stream = new POLE::Stream( &storage, "/PrvText" ); if( stream->fail() || (stream->size() == 0) ) { delete stream; return KoFilter::WrongFormat; } int len = stream->size() / 2; QString plaindoc; plaindoc.reserve( len ); unsigned char* buf = new unsigned char [stream->size()]; stream->read( buf, stream->size()); for(int i = 0; i < len; i++ ) plaindoc.append( QChar((int)readU16(buf + i*2) ) ); delete[] buf; delete stream; // split into paragraphs d->paragraphs = QStringList::split( "\n", plaindoc, true ); // create output store KoStore* storeout; storeout = KoStore::createStore( d->outputFile, KoStore::Write, "application/vnd.oasis.opendocument.text", KoStore::Zip ); if ( !storeout ) { kWarning() << "Couldn't open the requested file."; return KoFilter::FileNotFound; } if ( !storeout->open( "styles.xml" ) ) { kWarning() << "Couldn't open the file 'styles.xml'."; return KoFilter::CreationError; } storeout->write( d->createStyles() ); storeout->close(); if ( !storeout->open( "content.xml" ) ) { kWarning() << "Couldn't open the file 'content.xml'."; return KoFilter::CreationError; } storeout->write( d->createContent() ); storeout->close(); // store document manifest storeout->enterDirectory( "META-INF" ); if ( !storeout->open( "manifest.xml" ) ) { kWarning() << "Couldn't open the file 'META-INF/manifest.xml'."; return KoFilter::CreationError; } storeout->write( d->createManifest() ); storeout->close(); // we are done! d->inputFile.clear(); d->outputFile.clear(); delete storeout; return KoFilter::OK; }