254 lines
6.6 KiB
C++
Executable File
254 lines
6.6 KiB
C++
Executable File
/*===========================================================================
|
|
FILE:
|
|
DB2TextFile.h
|
|
|
|
DESCRIPTION:
|
|
Implementation of cDB2TextFile class
|
|
|
|
PUBLIC CLASSES AND METHODS:
|
|
cDB2TextFile
|
|
The cDB2TextFile class provides the simple ability to read in an
|
|
ANSI/UNICODE file and copy it to a dynamically allocated buffer
|
|
|
|
The sole difference between this and CStdioFile is that the issues
|
|
stemming from supporting both ANSI and UNICODE files are handled
|
|
in a simpler fashion at the expense of a bit of performance
|
|
|
|
Copyright (c) 2011, Code Aurora Forum. All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are met:
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
* Neither the name of Code Aurora Forum nor
|
|
the names of its contributors may be used to endorse or promote
|
|
products derived from this software without specific prior written
|
|
permission.
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
POSSIBILITY OF SUCH DAMAGE.
|
|
===========================================================================*/
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Include Files
|
|
//-----------------------------------------------------------------------------
|
|
#include "StdAfx.h"
|
|
#include "DB2TextFile.h"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Definitions
|
|
//-----------------------------------------------------------------------------
|
|
|
|
// Maximum size of a file this interface will try to open (8 MB)
|
|
const DWORD MAX_FILE_SZ = 1024 * 1024 * 8;
|
|
|
|
// Maximum number of characters to run UNICODE check over
|
|
const INT MAX_UNICODE_CHECK_LEN = 128;
|
|
|
|
/*=========================================================================*/
|
|
// cDB2TextFile Methods
|
|
/*=========================================================================*/
|
|
|
|
/*===========================================================================
|
|
METHOD:
|
|
cDB2TextFile (Public Method)
|
|
|
|
DESCRIPTION:
|
|
Construct object/load file into memory
|
|
|
|
PARAMETERS
|
|
pFileName [ I ] - File name
|
|
|
|
RETURN VALUE:
|
|
None
|
|
===========================================================================*/
|
|
cDB2TextFile::cDB2TextFile( LPCSTR pFileName )
|
|
: mText( "" ),
|
|
mStatus( ERROR_FILE_NOT_FOUND ),
|
|
mCurrentPos( 0 )
|
|
{
|
|
if (pFileName == 0 || pFileName[0] == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Open the file
|
|
std::ifstream tempFStream;
|
|
tempFStream.open( pFileName, std::ios::in | std::ios::binary );
|
|
if (tempFStream.fail() == true)
|
|
{
|
|
mStatus = ERROR_FILE_NOT_FOUND;
|
|
return;
|
|
}
|
|
|
|
// Get size
|
|
LONG nFileSize = tempFStream.rdbuf()->in_avail();
|
|
if (nFileSize == -1 || nFileSize > MAX_FILE_SZ)
|
|
{
|
|
tempFStream.close();
|
|
mStatus = ERROR_GEN_FAILURE;
|
|
return;
|
|
}
|
|
|
|
// Read file into pTempBuffer
|
|
CHAR * pTempBuffer = new char[ nFileSize ];
|
|
if (pTempBuffer == NULL)
|
|
{
|
|
tempFStream.close();
|
|
mStatus = ERROR_GEN_FAILURE;
|
|
return;
|
|
}
|
|
|
|
tempFStream.read( pTempBuffer, nFileSize );
|
|
if (tempFStream.fail() == true)
|
|
{
|
|
tempFStream.close();
|
|
delete [] pTempBuffer;
|
|
mStatus = ERROR_GEN_FAILURE;
|
|
return;
|
|
}
|
|
|
|
tempFStream.close();
|
|
|
|
// Convert to string
|
|
mText = std::string( pTempBuffer, nFileSize );
|
|
|
|
delete [] pTempBuffer;
|
|
|
|
// Final check
|
|
if (mText.size() != nFileSize)
|
|
{
|
|
mStatus = ERROR_GEN_FAILURE;
|
|
return;
|
|
}
|
|
|
|
// Success!
|
|
mStatus = NO_ERROR;
|
|
}
|
|
|
|
/*===========================================================================
|
|
METHOD:
|
|
cDB2TextFile (Public Method)
|
|
|
|
DESCRIPTION:
|
|
Construct object/copy from buffer into memory
|
|
|
|
PARAMETERS
|
|
pBuffer [ I ] - Buffer to copy from
|
|
bufferLen [ I ] - Size of above buffer
|
|
|
|
RETURN VALUE:
|
|
None
|
|
===========================================================================*/
|
|
cDB2TextFile::cDB2TextFile(
|
|
LPCSTR pBuffer,
|
|
ULONG bufferLen )
|
|
: mText( "" ),
|
|
mStatus( ERROR_FILE_NOT_FOUND ),
|
|
mCurrentPos( 0 )
|
|
{
|
|
// Convert to string
|
|
mText = std::string( pBuffer, bufferLen );
|
|
|
|
// Final check
|
|
if (mText.size() != bufferLen)
|
|
{
|
|
mStatus = ERROR_GEN_FAILURE;
|
|
return;
|
|
}
|
|
|
|
// Success!
|
|
mStatus = NO_ERROR;
|
|
}
|
|
|
|
/*===========================================================================
|
|
METHOD:
|
|
~cDB2TextFile (Public Method)
|
|
|
|
DESCRIPTION:
|
|
Destructor
|
|
|
|
RETURN VALUE:
|
|
None
|
|
===========================================================================*/
|
|
cDB2TextFile::~cDB2TextFile()
|
|
{
|
|
// Nothing to do
|
|
}
|
|
|
|
/*===========================================================================
|
|
METHOD:
|
|
ReadLine (Public Method)
|
|
|
|
DESCRIPTION:
|
|
Read the next available line
|
|
|
|
PARAMETERS
|
|
line [ O ] - Line (minus CR/LF)
|
|
|
|
RETURN VALUE:
|
|
None
|
|
===========================================================================*/
|
|
bool cDB2TextFile::ReadLine( std::string & line )
|
|
{
|
|
if (IsValid() == false)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int len = mText.size();
|
|
if (mCurrentPos >= len)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int newIdx = mText.find( '\n', mCurrentPos );
|
|
if (newIdx == -1)
|
|
{
|
|
// Possibly the end of the file
|
|
newIdx = len;
|
|
}
|
|
|
|
if (newIdx < mCurrentPos)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (newIdx == mCurrentPos)
|
|
{
|
|
// Return an empty line
|
|
mCurrentPos++;
|
|
line = "";
|
|
}
|
|
else
|
|
{
|
|
// Grab line
|
|
line = mText.substr( mCurrentPos, (newIdx - mCurrentPos) );
|
|
|
|
// Check for CR
|
|
int nCR = line.find( '\r' );
|
|
if (nCR != -1)
|
|
{
|
|
line.erase( nCR, 1 );
|
|
}
|
|
|
|
mCurrentPos = newIdx + 1;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|