1 2 // 3 // This file is part of the aMule Project. 4 // 5 // Copyright (c) 2003-2011 aMule Team ( admin@amule.org / http://www.amule.org ) 6 // Copyright (c) 2002-2011 Merkur ( devs@emule-project.net / http://www.emule-project.net ) 7 // 8 // Any parts of this program derived from the xMule, lMule or eMule project, 9 // or contributed by third-party developers are copyrighted by their 10 // respective authors. 11 // 12 // This program is free software; you can redistribute it and/or modify 13 // it under the terms of the GNU General Public License as published by 14 // the Free Software Foundation; either version 2 of the License, or 15 // (at your option) any later version. 16 // 17 // This program is distributed in the hope that it will be useful, 18 // but WITHOUT ANY WARRANTY; without even the implied warranty of 19 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 20 // GNU General Public License for more details. 21 // 22 // You should have received a copy of the GNU General Public License 23 // along with this program; if not, write to the Free Software 24 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA 25 // 26 27 #ifndef PACKET_H 28 #define PACKET_H 29 30 #include "Types.h" // Needed for int8, int32, uint8 and uint32 31 32 class CMemFile; 33 class wxString; 34 35 // CLIENT TO SERVER 36 37 // PACKET CLASS 38 // TODO some parts could need some work to make it more efficient 39 40 class CPacket { 41 public: 42 CPacket(CPacket &p); 43 CPacket(uint8 protocol); 44 CPacket(byte* header, byte *buf); // only used for receiving packets 45 CPacket(const CMemFile& datafile, uint8 protocol, uint8 ucOpcode); 46 CPacket(int8 in_opcode, uint32 in_size, uint8 protocol, bool bFromPF = true); 47 CPacket(byte* pPacketPart, uint32 nSize, bool bLast, bool bFromPF = true); // only used for splitted packets! 48 49 ~CPacket(); 50 51 byte* GetHeader(); 52 byte* GetUDPHeader(); 53 byte* GetPacket(); 54 byte* DetachPacket(); GetRealPacketSize()55 uint32 GetRealPacketSize() const { return size + 6; } 56 static uint32 GetPacketSizeFromHeader(const byte* rawHeader); IsSplitted()57 bool IsSplitted() { return m_bSplitted; } IsLastSplitted()58 bool IsLastSplitted() { return m_bLastSplitted; } 59 void PackPacket(); 60 bool UnPackPacket(uint32 uMaxDecompressedSize = 50000); 61 // -khaos--+++> Returns either -1, 0 or 1. -1 is unset, 0 is from complete file, 1 is from part file IsFromPF()62 bool IsFromPF() { return m_bFromPF; } 63 GetOpCode()64 uint8 GetOpCode() const { return opcode; } SetOpCode(uint8 oc)65 void SetOpCode(uint8 oc) { opcode = oc; } GetPacketSize()66 uint32 GetPacketSize() const { return size; } GetProtocol()67 uint8 GetProtocol() const { return prot; } SetProtocol(uint8 p)68 void SetProtocol(uint8 p) { prot = p; } GetDataBuffer(void)69 const byte* GetDataBuffer(void) const { return pBuffer; } 70 void Copy16ToDataBuffer(const void* data); 71 void CopyToDataBuffer(unsigned int offset, const byte* data, unsigned int n); 72 void CopyUInt32ToDataBuffer(uint32 data, unsigned int offset = 0); 73 74 private: 75 //! CPacket is not assignable. 76 CPacket& operator=(const CPacket&); 77 78 uint32 size; 79 uint8 opcode; 80 uint8 prot; 81 bool m_bSplitted; 82 bool m_bLastSplitted; 83 bool m_bPacked; 84 bool m_bFromPF; 85 byte head[6]; 86 byte* tempbuffer; 87 byte* completebuffer; 88 byte* pBuffer; 89 }; 90 91 #endif // PACKET_H 92 // File_checked_for_headers 93