Форум программистов [Powered by Invision Power Board]

Код

#pragma pack(push,2)
        struct packetbuf
        {
                unsigned short length;
                char data[USHRT_MAX+1-2];
        };
#pragma pack(pop)

class basic_packetstream 
{
public:
        void skip(int n)        {pos+=n;}
        void reset()            {pos=&buf.data[0];}
        char* back() const      {return pos;}
protected:
        basic_packetstream() {}
        packetbuf buf;
        char* pos;
};

class ipacketstream: public basic_packetstream 
{
public:
        //простые типы
#define IMPL(T) {data=*(T*)pos;pos+=sizeof(T);return *this;}
        ipacketstream& operator>>(char& data) IMPL(char)
        ipacketstream& operator>>(unsigned char& data) IMPL(unsigned char)
        ipacketstream& operator>>(short& data) IMPL(short)
        ipacketstream& operator>>(unsigned short& data) IMPL(unsigned short)
        ipacketstream& operator>>(int& data) IMPL(int)
        ipacketstream& operator>>(unsigned int& data) IMPL(unsigned int)
        ipacketstream& operator>>(__int64& data) IMPL(__int64)
        ipacketstream& operator>>(unsigned __int64& data) IMPL(unsigned __int64)
        ipacketstream& operator>>(float& data) IMPL(float)
        ipacketstream& operator>>(double& data) IMPL(double)
#undef IMPL
        //составные типы
        ipacketstream& operator>>(std::string&);
        ipacketstream& operator>>(std::wstring&);
        //манипуляторы
        ipacketstream& operator>> (ipacketstream&(*manipulator)(ipacketstream&))
        {       return manipulator(*this);      }
        template<typename Func>
        ipacketstream& operator>> (Func& manipulator)
        {       return manipulator(*this);      }       
public:
        size_t size() {return buf.length;}
        packetbuf& buffer() { return buf; }     
};

class opacketstream: public basic_packetstream
{
public:
        opacketstream() {reset();};
public:
        //простые типы
#define IMPL(T) {*(T*)pos=data;pos+=sizeof(T);return *this;}
        opacketstream& operator<<(char data) IMPL(char)
        opacketstream& operator<<(unsigned char data) IMPL(unsigned char)
        opacketstream& operator<<(short data) IMPL(short)
        opacketstream& operator<<(unsigned short data) IMPL(unsigned short)
        opacketstream& operator<<(int data) IMPL(int)
        opacketstream& operator<<(unsigned int data) IMPL(unsigned int)
        opacketstream& operator<<(__int64 data) IMPL(__int64)
        opacketstream& operator<<(unsigned __int64 data) IMPL(unsigned __int64)
        opacketstream& operator<<(float data) IMPL(float)
        opacketstream& operator<<(double data) IMPL(double)
#undef IMPL
        //составные типы
        opacketstream& operator<<(std::string&);
        opacketstream& operator<<(std::wstring&);
        //манипуляторы
        opacketstream& operator<< (opacketstream&(*manipulator)(opacketstream&))
        {       return manipulator(*this);      }
        template<typename Func>
        opacketstream& operator<< (Func& manipulator)
        {       return manipulator(*this);      }
public:
        size_t size() { return pos-&buf.data[0];        }
        packetbuf& buffer() 
        { 
                buf.length=size();
                return buf;
        }
};