Socket開發之通訊協議及處理

所謂的協議就是通訊雙方協商並制定好要傳送的數據的結構與格式。並按制定好的格式去組合與分析數據。從而使數據得以被準確的理解和處理。
那麼我們如何去制定通訊協議呢?很簡單,就是指定數據中各個字節所代表的意義。比如說:第一位代表封包頭,第二位代表封類型,第三、四位代表封包的數據長度。然後後面是實際的數據內容。
如下面這個例子:
01 01 06 00 01 0f ef 87 56 34
協議類別 協議代碼 數據長度 實際數據

前面三部分稱之爲封包頭,它的長度是固定的,第四部分是封包數據,它的長度是不固定的,由第三部分標識其長度。因爲我們的協議將用在TCP中,所以我沒有加入校驗位。原因是TCP可以保證數據的完整性。校驗位是沒有必要存在的。
接下來我們要爲這個數據封包聲明一個類來封裝它:
    public class Message
    {
        private byte _class;
        private byte _flag;
        private int _size;
        private byte[] _content;

        public byte[] Content
        {
            get { return _content; }
            set {_content = value; }
        }

        public int Size
        {
            get { return _size; }
            set { _size = value; }
        }

        public byte Flag
        {
            get { return _flag; }
            set { _flag = value; }
        }

        public byte Class
        {
            get { return _class; }
            set { _class = value; }
        }

        public Message()
        {

        }

        public Message(byte @class, byte flag, byte[] content)
        {
            _class = @class;
            _flag = flag;
            _size = content.Length;
            _content = content;
        }

        public byte[] ToBytes()
        {
            byte[] _byte;
            using (MemoryStream mem = new MemoryStream())
            {
                BinaryWriter writer = new BinaryWriter(mem);
                writer.Write(_class);
                writer.Write(_flag);
                writer.Write(_size);
                if (_size > 0)
                {
                    writer.Write(_content);
                }
                _byte = mem.ToArray();
                writer.Close();
            }
            return _byte;
        }

        public static Message FromBytes(byte[] Buffer)
        {
            Message message = new Message();
            using (MemoryStream mem = new MemoryStream(Buffer))
            {
                BinaryReader reader = new BinaryReader(mem);
                message._class = reader.ReadByte();
                message._flag = reader.ReadByte();
                message._size = reader.ReadInt32();
                if (message._size > 0)
                {
                    message._content = reader.ReadBytes(message._size);
                }
                reader.Close();
            }
            return message;
        }

    } 
我們可以用Tobytes()和FromBytes()將封包轉換成二進制數組和從二進制數組轉換回來。
事情看起來已經解決了,但……真的是這樣子嗎?不然,我們知道,TCP數據是以流的形式被傳送的,我們並不知道一個數據包是否被傳送完畢,也不知道我們接收回來的數據包中是否有多個數據包,如果直接使用FromBytes()來轉換的話,很可能會因爲數據不完整而出現異常,也有可能會因爲數據中含有多個數據包而導致數據丟失(因爲你並不知道這些數據中含有多少個數據包)。那我們該怎麼辦?這也不難,我們先把接收回來的數據寫入一個流中。然後分析其中是否有完整的數據包,如果有,將其從流中取出,並將這部分數據從流中清除。直到流中沒有完整的數據爲止,以後接收回來的數據就將其寫入流的結尾處,並從頭繼續分析。直到結束。
讓我們來看看這部分的代碼:
    public class MessageStream
    {
        private byte[] _buffer;
        private int _position;
        private int _length;
        private int _capacity;

        public MessageStream()
        {
            _buffer = new byte[0];
            _position = 0;
            _length = 0;
            _capacity = 0;
        }

        private byte ReadByte()
        {
            if (this._position >= this._length)
            {
                return 0;
            }
            return this._buffer[this._position++];
        }

        private int ReadInt()
        {
            int num = this._position += 4;
            if (num > this._length)
            {
                this._position = this._length;
                return -1;
            }
            return (((this._buffer[num - 4] | (this._buffer[num - 3] << 8)) | (this._buffer[num - 2] << 0x10)) | (this._buffer[num - 1] << 0x18));
        }

        private byte[] ReadBytes(int count)
        {
            int num = this._length - this._position;
            if (num > count)
            {
                num = count;
            }
            if (num <= 0)
            {
                return null;
            }
            byte[] buffer = new byte[num];
            if (num <= 8)
            {
                int num2 = num;
                while (--num2 >= 0)
                {
                    buffer[num2] = this._buffer[this._position + num2];
                }
            }
            else
            {
                Buffer.BlockCopy(this._buffer, this._position, buffer, 0, num);
            }
            this._position += num;
            return buffer;
        }

        public bool Read(out Message message)
        {
            message = null;
            _position = 0;
            if (_length > 6)
            {
                message = new Message();
                message.Class = ReadByte();
                message.Flag = ReadByte();
                message.Size = ReadInt();
                if (message.Size <= 0 || message.Size <= _length - _position)
                {
                    if (message.Size > 0)
                    {
                        message.Content = ReadBytes(message.Size);
                    }
                    Remove(message.Size + 6);
                    return true;
                }
                else
                {
                    message = null;
                    return false;
                }
            }
            else
            {
                return false;
            }
        }

        private void EnsureCapacity(int value)
        {
            if (value <= this._capacity)
                return;
            int num1 = value;
            if (num1 < 0x100)
                num1 = 0x100;
            if (num1 < (this._capacity * 2))
                num1 = this._capacity * 2;
            byte[] buffer1 = new byte[num1];
            if (this._length > 0)
                Buffer.BlockCopy(this._buffer, 0, buffer1, 0, this._length);
            this._buffer = buffer1;
            this._capacity = num1;
        }

        public void Write(byte[] buffer, int offset, int count)
        {
            if (buffer.Length - offset < count)
            {
                count = buffer.Length - offset;
            }
            EnsureCapacity(buffer.Length + count);
            Array.Clear(_buffer, _length, _capacity - _length);
            Buffer.BlockCopy(buffer, offset, _buffer, _length, count);
            _length += count;
        }

        private void Remove(int count)
        {
            if (_length >= count)
            {
                Buffer.BlockCopy(_buffer, count, _buffer, 0, _length - count);
                _length -= count;
                Array.Clear(_buffer, _length, _capacity - _length);
            }
            else
            {
                _length = 0;
                Array.Clear(_buffer, 0, _capacity);
            }
        }
    }  
這個類的使用非常簡單,你只要用Write(byte[] buffer, int offset, int count)將接收到的數據寫入數據流中,並用bool Read(out Message message)將數據中的第一個數據包取出,如果函數返回True,就說明取回一個封包成功,如果返回False,則說明流中已經沒有完整的封包,你需要繼續接收後面的數據以組成一個完整的封包。
這們我們的數據分析就會變得非常簡單。我們可以在ReceiveCallBack回調函數中將接收到的數據寫入到流中並通知線程池中的工作者線程分析數據流並處理數據。我在前面的關於Socket異步操作的文章中的Analyzer函數就是用這兩個類來分析處理數據的。這樣的好處理就是,Socket工作線程只需要負責數據的接收,並將其寫入流,其它的事情由其它的線程這處理,就不會因爲處理的時間過長而導致接收操作被阻塞。從而影響Socket的性能。
本文所述方法只是協議處理的多種方法中的其中一種,而且可能並不是很優秀的方法,如果誰有更好的方法,還希望您能和我多多交流。好了,今天就到這裏了,關於Socket的文章到這裏可能就告一段落了,我現在在研究VS2008裏面的新東西,如果有什麼必得的話,我會繼續寫出來的。謝謝大家的支持。

發佈了221 篇原創文章 · 獲贊 3 · 訪問量 33萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章