12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394 |
- /*
- * This file is part of UAS2.
- *
- * UAS2 is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * UAS2 is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
-
- * You should have received a copy of the GNU General Public License
- * along with UASv1; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- */
-
- /**
- * @file RecvPacket.h
- */
-
- #ifndef __RECVPACKET_H
- #define __RECVPACKET_H
-
- #include <cassert>
-
- #include "Shared.h"
-
- class cRecvPacket
- {
- public:
- cRecvPacket( )
- : m_wSize( 0 )
- {
- m_pcTH = reinterpret_cast< cTransportHeader * >( m_bData );
- }
-
- ~cRecvPacket( ) {}
-
- inline DWORD GetSequence ( ) { return m_pcTH->m_dwSequence; }
- inline DWORD GetFlags ( ) { return m_pcTH->m_dwFlags; }
- inline DWORD GetCRC ( ) { return m_pcTH->m_dwCRC; }
- inline WORD GetLogicalID( ) { return m_pcTH->m_wLogicalID; }
- inline WORD GetTime ( ) { return m_pcTH->m_wTime; }
- inline WORD GetTotalSize( ) { return m_pcTH->m_wTotalSize; }
- inline WORD GetTable ( ) { return m_pcTH->m_wTable; }
-
- inline BYTE *GetPayloadFront( ) { return m_bData + sizeof( cTransportHeader ); }
- inline BYTE *GetPayload ( WORD wOffset ) { return m_bData + sizeof( cTransportHeader ) + wOffset; }
- inline BYTE *GetPayloadBack ( ) { return m_bData + sizeof( cTransportHeader ) + m_pcTH->m_wTotalSize; }
-
- inline void CopyPayload( WORD wOffset, void *pDest, WORD wSize )
- {
- CopyMemory( pDest, m_bData + sizeof( cTransportHeader ) + wOffset, wSize );
- }
-
- inline BYTE& operator[]( int n )
- {
- #ifdef _DEBUG
- WORD wMaxSize = GetTotalSize( ) + sizeof( cTransportHeader );
- if ( wMaxSize > 0 )
- assert( n < wMaxSize );
- #endif
-
- return m_bData[n];
- }
-
- inline int GetPadding( BYTE *pbStart );
-
- BYTE m_bData[MAX_PACKET_SIZE];
- WORD m_wSize;
- SOCKADDR_IN m_saSockAddr;
-
- private:
- cTransportHeader *m_pcTH;
- };
-
- inline int cRecvPacket::GetPadding( BYTE *pbStart )
- {
- WORD wSize = *(WORD *)pbStart;
- pbStart += (sizeof( WORD ) + wSize);
-
- DWORD dwDiff = pbStart - m_bData;
-
- if( dwDiff < 4 )
- return 4 - dwDiff;
- else if ( (dwDiff % 4) != 0 )
- return 4 - (dwDiff % 4);
- else
- return 0;
- }
-
- #endif // #ifndef __RECVPACKET_H
|