//////////////////
//
// XPK fileformat:
// chunk-based, crunched data in chunks
//
// File header:
// starts as common IFF-style header:
// - 4-byte ID, 'XPKF'
// - 4-byte int for filesize minus header (8)
// - 4-byte sub-type (cruncher-ID) e.g. 'SQSH'
// - 4-byte int for total uncompressed length ?
// - first 16-bytes of original file
// - 1 byte for flags
// - 1 byte for header checksum ?
// - 1 byte for minor version of cruncher/library ?
// - actually version required of sub-library..
// - 1 byte for major version of cruncher/library ?
//
bool XpkTags::ReadStreamHeader(CIOBuffer &Buffer)
{
// should have enough data to actually parse file header
if (Buffer.GetSize() < sizeof(XpkStreamHeader))
{
return false;
}
if (isXpkFile(Buffer.GetBegin()) == false)
{
return false;
}
Buffer.SetCurrentPos(0); // start at beginning if not already..
// set default
m_formatType = XPKMODE_UPSTD;
// "XPKF", 0x58504b46 (XPK_COOKIE, magic number)
// note: support also XFD-packed files? ("XFDD")
m_streamHeader.xsh_PackID = GetULong(Buffer.GetNext(4));
if (m_streamHeader.xsh_PackID != MakeTag("XPKF"))
// this too ? && m_streamHeader.xsh_PackID != MakeTag("XFDD"))
{
return false;
}
// file length without IFF header (type+length = 8) ?
m_streamHeader.xsh_CompressedLen = GetULong(Buffer.GetNext(4));
// keep packer type as type name/ID,
// just access as-is
m_typeName.assign((char*)Buffer.GetAtCurrent(), 4);
// packer type, e.g. "SQSH", "NUKE", "RLEN"..
m_streamHeader.xsh_PackerType = GetULong(Buffer.GetNext(4));
// TODO: check supported types..? if != MakeTag()..
// -> caller/parent should do (knows libraries..)
// uncompressed length?
m_streamHeader.xsh_UnpackedLen = GetULong(Buffer.GetNext(4));
// first 16 bytes of original file
::memcpy(m_streamHeader.xsh_Initial, Buffer.GetNext(16), 16);
// flags
m_streamHeader.xsh_Flags = Buffer.GetNextByte();
/*
// also check "XFDD", 0x58464444 in file ID?
if (m_streamHeader.xsh_Flags & XMF_XFD && m_streamHeader.xsh_PackID == MakeTag("XFDD"))
{
m_formatType = XPKMODE_UPXFD;
}
if (m_streamHeader.xsh_Flags & XMF_PASSTHRU)
{
// unpacked?
m_formatType = XPKMODE_UPUP;
}
if (m_streamHeader.xsh_Flags & XPKSTREAMF_PASSWORD)
{
// password-protected file?
}
*/
// ..no idea.. header checksum?
m_streamHeader.xsh_HeaderChk = Buffer.GetNextByte();
// minor&major version of XPK master/cruncher?
m_streamHeader.xsh_SubVrs = Buffer.GetNextByte(); // sub-library version required?
m_streamHeader.xsh_MasVrs = Buffer.GetNextByte();
// TODO: remove later, debug-test..
if (Buffer.GetCurrentPos() != sizeof(XpkStreamHeader))
{
throw IOException("Read size does not match stream-header size");
}
// non-zero header checksum? (note where doing checksumming..)
if (hchecksum(Buffer.GetBegin(), sizeof(XpkStreamHeader)) != 0)
{
throw ArcException("Header checksum error", m_streamHeader.xsh_HeaderChk);
}
// has extended header?
if (m_streamHeader.xsh_Flags & XPKSTREAMF_EXTHEADER)
{
// size of extended header if present?
m_extHeaderLen = GetUWord(Buffer.GetNext(2));
/*
// this done above..
if(!hookread(xbuf, XIO_READ, &exthlen, sizeof(UWORD)))
goto Abort; -> exception
// no read, just skip it??
if(!hookread(xbuf, XIO_READ, NULL, exthlen))
goto Abort; -> exception
// later it is read anyway, why not just directly when detected?
*/
// increment by length-field size anyway..
m_extHeaderLen += sizeof(uint16_t); /* for unwinding while XpkExamine */
// note: increase buffer position by size of extended header?
}
// header parsed, should be valid file?
return true;
}
int CSocketServer::Run()
{
try
{
vector<WorkerThread *> workers;
workers.reserve( m_numThreads );
for ( size_t i = 0; i < m_numThreads; ++i )
{
/*
* Call to unqualified virtual function
*/
WorkerThread *pThread = CreateWorkerThread( m_iocp );
workers.push_back( pThread );
pThread->Start();
}
HANDLE handlesToWaitFor[2];
handlesToWaitFor[0] = m_shutdownEvent.GetEvent();
handlesToWaitFor[1] = m_acceptConnectionsEvent.GetEvent();
while ( !m_shutdownEvent.Wait( 0 ) )
{
DWORD waitResult = ::WaitForMultipleObjects( 2, handlesToWaitFor, false, INFINITE );
if ( waitResult == WAIT_OBJECT_0 )
{
/*
* Time to shutdown
*/
break;
}
else if ( waitResult == WAIT_OBJECT_0 + 1 )
{
/*
* accept connections
*/
while ( !m_shutdownEvent.Wait( 0 ) && m_acceptConnectionsEvent.Wait( 0 ) )
{
CIOBuffer *pAddress = Allocate();
int addressSize = (int)pAddress->GetSize();
SOCKET acceptedSocket = ::WSAAccept(
m_listeningSocket,
reinterpret_cast<sockaddr *>(const_cast<BYTE *>( pAddress->GetBuffer() ) ),
&addressSize,
0,
0);
pAddress->Use( addressSize );
if ( acceptedSocket != INVALID_SOCKET )
{
Socket *pSocket = AllocateSocket( acceptedSocket );
/*
* Call to unqualified virtual function
*/
OnConnectionEstablished( pSocket, pAddress );
}
else if ( m_acceptConnectionsEvent.Wait( 0 ) )
{
/*
* Call to unqualified virtual function
*/
OnError( _T("CSocketServer::Run() - WSAAccept:") + GetLastErrorMessage( ::WSAGetLastError() ) );
}
pAddress->Release();
}
}
else
{
/*
* Call to unqualified virtual function
*/
OnError( _T("CSocketServer::Run() - WaitForMultipleObjects: ") + GetLastErrorMessage( ::GetLastError() ) );
}
}
for ( i = 0; i < m_numThreads; ++i )
{
workers[i]->InitiateShutdown();
}
for ( i = 0; i < m_numThreads; ++i )
{
workers[i]->WaitForShutdownToComplete();
delete workers[i];
workers[i] = 0;
}
}
catch( const CException &e )
{
/*
* Call to unqualified virtual function
*/
OnError( _T("CSocketServer::Run() - Exception: ") + e.GetWhere() + _T(" - ") + e.GetMessage() );
}
catch(...)
{
/*
* Call to unqualified virtual function
*/
OnError( _T("CSocketServer::Run() - Unexpected exception") );
}
/*
* Call to unqualified virtual function
*/
OnShutdownComplete();
return 0;
}
