1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/os/ossrv/compressionlibs/ziplib/src/ezip/zipfilememberinputstream.cpp Fri Jun 15 03:10:57 2012 +0200
1.3 @@ -0,0 +1,294 @@
1.4 +// Copyright (c) 2003-2009 Nokia Corporation and/or its subsidiary(-ies).
1.5 +// All rights reserved.
1.6 +// This component and the accompanying materials are made available
1.7 +// under the terms of "Eclipse Public License v1.0"
1.8 +// which accompanies this distribution, and is available
1.9 +// at the URL "http://www.eclipse.org/legal/epl-v10.html".
1.10 +//
1.11 +// Initial Contributors:
1.12 +// Nokia Corporation - initial contribution.
1.13 +//
1.14 +// Contributors:
1.15 +//
1.16 +// Description:
1.17 +//
1.18 +
1.19 +
1.20 +#include <zipfile.h>
1.21 +#include <zipfilememberinputstream.h>
1.22 +#include "libzcore.h"
1.23 +
1.24 +RZipFileMemberReaderStream::RZipFileMemberReaderStream(
1.25 + CZipFile& aZipFile,
1.26 + TUint32 aDataOffset,
1.27 + TUint32 aCompressedSize,
1.28 + TUint32 aUncompressedSize,
1.29 + TUint32 aCompressionMethod):
1.30 + iZipFile(aZipFile),
1.31 + iCompressionMethod(aCompressionMethod),
1.32 + iCompressedSize(aCompressedSize),
1.33 + iUncompressedSize(aUncompressedSize),
1.34 + iFileOffset(aDataOffset)
1.35 + {
1.36 + }
1.37 +
1.38 +
1.39 +RZipFileMemberReaderStream* RZipFileMemberReaderStream::NewL(
1.40 + CZipFile& aZipFile,
1.41 + TUint32 aDataOffset,
1.42 + TUint32 aCompressedSize,
1.43 + TUint32 aUncompressedSize,
1.44 + TUint32 aCompressionMethod)
1.45 + {
1.46 + RZipFileMemberReaderStream* me = new(ELeave) RZipFileMemberReaderStream(aZipFile, aDataOffset, aCompressedSize,aUncompressedSize, aCompressionMethod);
1.47 + CleanupStack::PushL(me);
1.48 + me->ConstructL();
1.49 + CleanupStack::Pop(me);
1.50 + return me;
1.51 + }
1.52 +
1.53 +/**
1.54 +Creates input stream to be used for reading the contents of the compressed file.
1.55 +*/
1.56 +void RZipFileMemberReaderStream::ConstructL()
1.57 + {
1.58 + TInt err = inflateInit2_r(&iStream, -MAX_WBITS);
1.59 + if (err == Z_MEM_ERROR)
1.60 + {
1.61 + User::Leave(KErrNoMemory);
1.62 + }
1.63 + }
1.64 +
1.65 +/**
1.66 +Destructor. All dynamically allocated data structures for this stream are freed.
1.67 +*/
1.68 +EXPORT_C RZipFileMemberReaderStream::~RZipFileMemberReaderStream()
1.69 +{
1.70 + inflateEnd_r(&iStream);
1.71 +}
1.72 +
1.73 +/**
1.74 +Reads data from the stream buffer into the specified descriptor.
1.75 +On return, contains the data read from the stream buffer
1.76 +
1.77 +@param aDes The target descriptor for the data read from the stream buffer
1.78 +@param aLength The maximum number of bytes to be read
1.79 +@return KErrNone If all bytes read successfully.
1.80 +@return KErrCorrupt If reading fails.
1.81 +@return KErrEof If end of file is reached.
1.82 +@return ... Any one of the system-wide error codes for other errors.
1.83 +*/
1.84 +EXPORT_C TInt RZipFileMemberReaderStream::Read(TDes16& aDes, TInt aLength)
1.85 +{
1.86 + TUint32 numBytesRead = 0;
1.87 + TInt err = Read(CONST_CAST(TByte*,(const TByte*)aDes.Ptr()), 2*aLength, &numBytesRead);
1.88 + if (err != KErrNone)
1.89 + {
1.90 + aDes.SetLength( (err==KErrEof) ? numBytesRead>>2 : 0 );
1.91 + return err;
1.92 + }
1.93 +
1.94 + aDes.SetLength(numBytesRead>>2);
1.95 + return KErrNone;
1.96 +}
1.97 +
1.98 +TInt RZipFileMemberReaderStream::Read(void)
1.99 + {
1.100 + TByte b;
1.101 + TUint32 numRead = 0;
1.102 +
1.103 + if (Read(&b, 1, &numRead) == 1)
1.104 + {
1.105 + return b;
1.106 + }
1.107 + else
1.108 + {
1.109 + return -1;
1.110 + }
1.111 + }
1.112 +
1.113 +TInt RZipFileMemberReaderStream::Read(TDes8& aDes, TInt aLength)
1.114 + {
1.115 + TUint32 numBytesRead = 0;
1.116 + TInt err = Read(CONST_CAST(TByte*,aDes.Ptr()), aLength, &numBytesRead);
1.117 + if (err != KErrNone)
1.118 + {
1.119 + aDes.SetLength( (err==KErrEof) ? numBytesRead : 0 );
1.120 + return err;
1.121 + }
1.122 +
1.123 + aDes.SetLength(numBytesRead);
1.124 + return KErrNone;
1.125 + }
1.126 +
1.127 +void RZipFileMemberReaderStream::ReadL(TDes16& aDes, TInt aLength)
1.128 + {
1.129 + User::LeaveIfError(Read(aDes, aLength));
1.130 + }
1.131 +
1.132 +void RZipFileMemberReaderStream::Release()
1.133 +{}
1.134 +
1.135 +void RZipFileMemberReaderStream::Close()
1.136 +{}
1.137 +
1.138 +
1.139 +TInt RZipFileMemberReaderStream::Read(TByte* aBytes, TUint32 aLength, TUint32* aRetLength)
1.140 + {
1.141 + if (iCompressionMethod == CZipArchive::EDeflated)
1.142 + {
1.143 + return GetBytes(aBytes, aLength, aRetLength);
1.144 + }
1.145 + else
1.146 + {
1.147 + return GetStoredBytes(aBytes, aLength, aRetLength);
1.148 + }
1.149 + }
1.150 +
1.151 +TInt RZipFileMemberReaderStream::GetBytes(TByte* aBytes, TUint32 aLength, TUint32* aRetLength)
1.152 + {
1.153 + TUint32 bytesLeftToRead;
1.154 + TInt status;
1.155 +
1.156 + iBytesLength = 0;
1.157 +
1.158 + // Be careful not to confuse compressed bytes and uncompressed bytes.
1.159 + // The length passed in is in uncompressed bytes, compressed bytes
1.160 + // are mainly used in the GetCompressedBytes() function called.
1.161 + // iBytesLength refers to the number of bytes already read.
1.162 + // If the request is to read past the end of the file
1.163 + // we should return KErrNone on the first instance, i.e. return all bytes
1.164 + // read successfully. On the second instance return KErrEof, i.e. we have
1.165 + // already read all the bytes when another request comes in, so return KErrEof.
1.166 + // This follows the rules for reading an uncompressed file within this component
1.167 + // and this is also the way that RFile::ReadL() does it.
1.168 +
1.169 + if (aLength > iUncompressedSize)
1.170 + {
1.171 + aLength = iUncompressedSize; // no point trying to read more than we have
1.172 + }
1.173 +
1.174 + bytesLeftToRead = aLength;
1.175 +
1.176 + while (bytesLeftToRead > 0)
1.177 + {
1.178 + if (iStream.avail_in == 0)
1.179 + {
1.180 + if (GetCompressedBytes() != KErrNone)
1.181 + {
1.182 + return KErrCorrupt;
1.183 + }
1.184 + }
1.185 +
1.186 + // The decompression will be done in the user provided memory.
1.187 + iStream.next_out = &aBytes[iBytesLength];
1.188 + iStream.avail_out = aLength - iBytesLength;
1.189 + status = inflate_r(&iStream, Z_SYNC_FLUSH);
1.190 +
1.191 + switch (status)
1.192 + {
1.193 + case Z_OK:
1.194 + iBytesLength = aLength - iStream.avail_out;
1.195 + break;
1.196 +
1.197 + case Z_STREAM_END: //EOF
1.198 + if (iBytesLength == aLength - iStream.avail_out)
1.199 + {
1.200 + *aRetLength = iBytesLength;
1.201 + return KErrEof;
1.202 + }
1.203 + else
1.204 + {
1.205 + iBytesLength = aLength - iStream.avail_out;
1.206 + break;
1.207 + }
1.208 + case Z_MEM_ERROR:
1.209 + return KErrNoMemory;
1.210 +
1.211 + default:
1.212 + return KErrCorrupt;
1.213 +
1.214 + }
1.215 + bytesLeftToRead = aLength - iBytesLength;
1.216 + }
1.217 +
1.218 + *aRetLength = iBytesLength;
1.219 + return KErrNone;
1.220 + }
1.221 +
1.222 +TInt RZipFileMemberReaderStream::GetCompressedBytes(void)
1.223 + {
1.224 + if (iOffset < iCompressedSize)
1.225 + {
1.226 + TUint32 nBytesLeft;
1.227 + TUint32 nBytesToRead;
1.228 +
1.229 + nBytesLeft = iCompressedSize - iOffset;
1.230 + nBytesToRead = (nBytesLeft > sizeof(iCompressedBytes)) ? sizeof(iCompressedBytes) : nBytesLeft;
1.231 + if (iZipFile.Seek(iFileOffset) != KErrNone)
1.232 + {
1.233 + return KErrCorrupt;
1.234 + }
1.235 + else
1.236 + if (iZipFile.Read(iCompressedBytes, nBytesToRead) != KErrNone)
1.237 + {
1.238 + return KErrCorrupt;
1.239 + }
1.240 + iFileOffset += nBytesToRead;
1.241 + iOffset += nBytesToRead;
1.242 + iStream.next_in = iCompressedBytes;
1.243 + iStream.avail_in = nBytesToRead;
1.244 + return KErrNone;
1.245 + }
1.246 + else
1.247 + if (iDone == EFalse)
1.248 + {
1.249 + iCompressedBytes[0] = 0;
1.250 + iStream.avail_in = 1;
1.251 + iStream.next_in = iCompressedBytes;
1.252 + iDone = ETrue;
1.253 + return KErrNone;
1.254 + }
1.255 + else
1.256 + {
1.257 + return KErrCorrupt;
1.258 + }
1.259 + }
1.260 +
1.261 +TInt RZipFileMemberReaderStream::GetStoredBytes(TByte* aBytes, TUint32 aLength, TUint32* aRetLength)
1.262 + {
1.263 + TInt status;
1.264 +
1.265 + if (aLength > iUncompressedSize)
1.266 + {
1.267 + aLength = iUncompressedSize; // no point trying to read more than we have
1.268 + }
1.269 + if (aLength == 0) // empty file like a directory
1.270 + {
1.271 + return KErrNone;
1.272 + }
1.273 + if (iOffset == iCompressedSize) // end of zip item.
1.274 + {
1.275 + return KErrEof;
1.276 + }
1.277 + if ((iOffset + aLength) > iCompressedSize)
1.278 + {
1.279 + aLength = iCompressedSize - iOffset; // adjust read to what is left
1.280 + if ( aLength <= 0 )
1.281 + {
1.282 + return KErrCorrupt;
1.283 + }
1.284 + }
1.285 + if (iZipFile.Seek(iFileOffset) != KErrNone)
1.286 + {
1.287 + return KErrCorrupt;
1.288 + }
1.289 + status = iZipFile.Read(aBytes, aLength);
1.290 + if (status == KErrNone)
1.291 + {
1.292 + iFileOffset += aLength;
1.293 + iOffset += aLength;
1.294 + *aRetLength = aLength;
1.295 + }
1.296 + return status;
1.297 + }