/* * Copyright (c) 2021 The Linux Foundation. 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 The Linux Foundation 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 "AS IS" AND ANY EXPRESS OR IMPLIED * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT * 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. */ #ifndef NETWORK_TRAFFIC_TCPHEADER_H #define NETWORK_TRAFFIC_TCPHEADER_H #include "TransportHeader.h" class TcpHeader: public TransportHeader { public: const static unsigned int mSizeBits {160}; const static unsigned int mSize {20}; DECLARE_BITSET(SourcePort, 16, 56789); DECLARE_BITSET(DestPort, 16, 64886); DECLARE_BITSET(SequenceNumber, 32, 1077280711); DECLARE_BITSET(AckNumber, 32, 3090542765); DECLARE_BITSET(DataOffset, 4, 5); DECLARE_BITSET(Reserved, 3, 0); DECLARE_BITSET(NS, 1, 0); DECLARE_BITSET(CWR, 1, 0); DECLARE_BITSET(ECE, 1, 0); DECLARE_BITSET(URG, 1, 0); DECLARE_BITSET(ACK, 1, 1); DECLARE_BITSET(PSH, 1, 1); DECLARE_BITSET(RST, 1, 0); DECLARE_BITSET(SYN, 1, 0); DECLARE_BITSET(FIN, 1, 0); DECLARE_BITSET(WindowSize, 16, 473); DECLARE_BITSET(Checksum, 16, 0); DECLARE_BITSET(UrgentPtr, 16, 0); TcpHeader() = default; explicit TcpHeader(const uint8_t *start) { unsigned int bufIndex = 0; setBitsFromArray(mSourcePort, start, bufIndex); setBitsFromArray(mDestPort, start, bufIndex); setBitsFromArray(mSequenceNumber, start, bufIndex); setBitsFromArray(mAckNumber, start, bufIndex); setBitsFromArray(mDataOffset, start, bufIndex); setBitsFromArray(mReserved, start, bufIndex); setBitsFromArray(mNS, start, bufIndex); setBitsFromArray(mCWR, start, bufIndex); setBitsFromArray(mECE, start, bufIndex); setBitsFromArray(mURG, start, bufIndex); setBitsFromArray(mACK, start, bufIndex); setBitsFromArray(mPSH, start, bufIndex); setBitsFromArray(mRST, start, bufIndex); setBitsFromArray(mSYN, start, bufIndex); setBitsFromArray(mFIN, start, bufIndex); setBitsFromArray(mWindowSize, start, bufIndex); setBitsFromArray(mChecksum, start, bufIndex); setBitsFromArray(mUrgentPtr, start, bufIndex); } vector asVector() const override { vector outVec; auto inserter = [](vector& vec, auto val){ vector valAsVector = bitsetToVector(val); vec.insert(vec.end(), valAsVector.begin(), valAsVector.end());}; auto inserter2 = [](vector& vec, auto val){ vector valAsVector = bitsetToVector(val); toggleLsbMsb(valAsVector, CHAR_BIT); vec.insert(vec.end(), valAsVector.begin(), valAsVector.end());}; inserter2(outVec, mSourcePort); inserter2(outVec, mDestPort); inserter2(outVec, mSequenceNumber); inserter2(outVec, mAckNumber); vector v = bitsetToVector<4>(mDataOffset); vector v2 = bitsetToVector<3>(mReserved); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mNS); v.insert(v.end(), v2.begin(), v2.end()); outVec.insert(outVec.end(), v.rbegin(), v.rend()); v = bitsetToVector<1>(mCWR); v2 = bitsetToVector<1>(mECE); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mURG); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mACK); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mPSH); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mRST); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mSYN); v.insert(v.end(), v2.begin(), v2.end()); v2 = bitsetToVector<1>(mFIN); v.insert(v.end(), v2.begin(), v2.end()); outVec.insert(outVec.end(), v.rbegin(), v.rend()); inserter2(outVec, mWindowSize); inserter2(outVec, mChecksum); inserter(outVec, mUrgentPtr); return outVec; } uint32_t getSeqNum() const { return static_cast(mSequenceNumber.to_ulong()); } size_t size() const override { return TcpHeader::mSize; } static uint8_t protocolNum(){ return 6; } void adjust(uint8_t *checksumBuf, size_t count){ fixChecksum(checksumBuf, count); } void zeroChecksum(){ mChecksum = 0; } string name() const override { return string("TCP"); } void streamFields(std::ostream &out) const override { out << "Source Port: " << mSourcePort.to_ulong() << ", " << "Dest Port: " << mDestPort.to_ulong() << ", " << "Sequence Number: " << mSequenceNumber.to_ulong() << ", " << "Ack Number: " << mAckNumber.to_ulong() << ", " << "Data Offset: " << mDataOffset.to_ulong() << ", " << "Reserved: " << mReserved.to_ulong() << ", " << "NS: " << mNS.to_ulong() << ", " << "CWR: " << mCWR.to_ulong() << ", " << "URG: " << mURG.to_ulong() << ", " << "ACK: " << mACK.to_ulong() << ", " << "PSH: " << mPSH.to_ulong() << ", " << "RST: " << mRST.to_ulong() << ", " << "SYN: " << mSYN.to_ulong() << ", " << "FIN: " << mFIN.to_ulong() << ", " << "Window Size: " << mWindowSize.to_ulong() << ", " << "Checksum: " << mChecksum.to_ulong() << ", " << "Urgent Ptr: " << mUrgentPtr << std::endl; } struct flags { bool ns; bool cwr; bool ece; bool urg; bool ack; bool psh; bool rst; bool syn; bool fin; }; flags getFlags() const { return flags { .ns = mNS.test(0), .cwr = mCWR.test(0), .ece = mECE.test(0), .urg = mURG.test(0), .ack = mACK.test(0), .psh = mPSH.test(0), .rst = mRST.test(0), .syn = mSYN.test(0), .fin = mFIN.test(0) }; } private: void fixChecksum(uint8_t *checksumBuf, size_t count){ mChecksum = 0; mChecksum = computeChecksum(reinterpret_cast(checksumBuf), count); } }; #endif //NETWORK_TRAFFIC_TCPHEADER_H