WebSocketFrame.h

#pragma once
#include "WebSocketCommon.h"
#include "ByteBuffer.h"
#include <vector>
using namespace std;
namespace NPL
{
    namespace WebSocket
    {

        class WebSocketFrame
        {
        public:
            WebSocketFrame();
            ~WebSocketFrame();
            std::vector<byte>& getMask() {
                return mask;
            };
            byte getOpCode() {
                return (byte)(finRsvOp & 0x0F);
            };
            
            ByteBuffer getPayload() {
                return data;
            };
            int getPayloadLength() {
                return data.bytesRemaining();
            };
            bool isMasked() {
                return masked;
            }
            bool isFin() {
                return (byte)(finRsvOp & 0x80) != 0;
            };
            void setFin(bool fin) {
                // set bit 1
                finRsvOp = (byte)((finRsvOp & 0x7F) | (fin ? 0x80 : 0x00));
            };
            void setMask(const std::vector<byte>& maskingKey);
            void setMask(const byte* data, int nCount=4);
            void setMasked(bool mask) {
                masked = mask;
            };
            void setOpCode(byte op) {
                finRsvOp = (byte)((finRsvOp & 0xF0) | (op & 0x0F));
            };
            void setPayload(ByteBuffer& buffer) {
                data.clear();
                append(buffer,buffer.bytesRemaining());
            };
            void append(ByteBuffer& buffer,int needed_len);;
            void setRsv1(bool rsv1) {
                // set bit 2
                finRsvOp = (byte)((finRsvOp & 0xBF) | (rsv1 ? 0x40 : 0x00));
            };
            void setRsv2(bool rsv2) {
                // set bit 3
                finRsvOp = (byte)((finRsvOp & 0xDF) | (rsv2 ? 0x20 : 0x00));
            };
            void setRsv3(bool rsv3) {
                // set bit 4
                finRsvOp = (byte)((finRsvOp & 0xEF) | (rsv3 ? 0x10 : 0x00));
            };
            bool isRsv1() {
                return (byte)(finRsvOp & 0x40) != 0;
            };
            bool isRsv2() {
                return (byte)(finRsvOp & 0x20) != 0;
            };
            bool isRsv3() {
                return (byte)(finRsvOp & 0x10) != 0;
            };
            bool hasPayload() {
                return getPayloadLength() > 0;
            };
            void reset();;
            void assertValid();

            void loadData(std::vector<byte>& outData);
            int size() { return data.bytesRemaining(); };
        private:
            byte finRsvOp;
            bool masked;
            ByteBuffer data;
            std::vector<byte> mask;
        };
    }
}