79 lines
2.4 KiB
C++
79 lines
2.4 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/*
|
|
* This file is part of the LibreOffice project.
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*
|
|
*/
|
|
|
|
#include "oox/crypto/DocumentEncryption.hxx"
|
|
|
|
#include <com/sun/star/io/XSeekable.hpp>
|
|
|
|
#include "oox/helper/binaryinputstream.hxx"
|
|
#include "oox/helper/binaryoutputstream.hxx"
|
|
|
|
namespace oox {
|
|
namespace core {
|
|
|
|
using namespace css::beans;
|
|
using namespace css::io;
|
|
using namespace css::lang;
|
|
using namespace css::uno;
|
|
|
|
using namespace std;
|
|
|
|
DocumentEncryption::DocumentEncryption(Reference< XStream > xDocumentStream, oox::ole::OleStorage& rOleStorage, const OUString& aPassword) :
|
|
mxDocumentStream(xDocumentStream),
|
|
mrOleStorage(rOleStorage),
|
|
maPassword(aPassword)
|
|
{}
|
|
|
|
bool DocumentEncryption::encrypt()
|
|
{
|
|
Reference< XInputStream > xInputStream ( mxDocumentStream->getInputStream(), UNO_SET_THROW );
|
|
Reference< XSeekable > xSeekable( xInputStream, UNO_QUERY );
|
|
|
|
if (!xSeekable.is())
|
|
return false;
|
|
|
|
sal_uInt32 aLength = xSeekable->getLength();
|
|
|
|
if (!mrOleStorage.isStorage())
|
|
return false;
|
|
|
|
Reference< XOutputStream > xEncryptionInfo( mrOleStorage.openOutputStream( "EncryptionInfo" ), UNO_SET_THROW );
|
|
BinaryXOutputStream aEncryptionInfoBinaryOutputStream( xEncryptionInfo, false );
|
|
|
|
mEngine.writeEncryptionInfo(maPassword, aEncryptionInfoBinaryOutputStream);
|
|
|
|
aEncryptionInfoBinaryOutputStream.close();
|
|
xEncryptionInfo->flush();
|
|
xEncryptionInfo->closeOutput();
|
|
|
|
Reference< XOutputStream > xEncryptedPackage( mrOleStorage.openOutputStream( "EncryptedPackage" ), UNO_SET_THROW );
|
|
BinaryXOutputStream aEncryptedPackageStream( xEncryptedPackage, false );
|
|
|
|
BinaryXInputStream aDocumentInputStream( xInputStream, false );
|
|
aDocumentInputStream.seekToStart();
|
|
|
|
aEncryptedPackageStream.writeValue<sal_uInt32>( aLength ); // size
|
|
aEncryptedPackageStream.writeValue<sal_uInt32>( 0 ); // reserved
|
|
|
|
mEngine.encrypt(aDocumentInputStream, aEncryptedPackageStream);
|
|
|
|
aEncryptedPackageStream.close();
|
|
aDocumentInputStream.close();
|
|
|
|
xEncryptedPackage->flush();
|
|
xEncryptedPackage->closeOutput();
|
|
|
|
return true;
|
|
}
|
|
|
|
} // namespace core
|
|
} // namespace oox
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|