Exemplo n.º 1
0
        public AwaitableWrapper <Msg> RecvMsgR(BytesView buf)
        {
            if (_RecvMsgR_ra == null)
            {
                _RecvMsgR_ra   = new ReusableAwaiter <Msg>();
                _RecvMsgR_cont = () => {
                    if (!_RecvMsgR_awaiter.TryGetResult(out var r, out var ex))
                    {
                        _RecvMsgR_ra.SetException(ex);
                    }
                    else
                    {
                        _RecvMsgR_ra.SetResult(new Msg(r.bv));
                    }
                };
            }
            _RecvMsgR_ra.Reset();
            var awaiter = _RecvMsgR_awaiter = _readAsyncR(new BytesSegment(buf?.bytes, buf?.offset ?? 0, buf?.len ?? 0));

            if (awaiter.IsCompleted)
            {
                _RecvMsgR_cont();
            }
            else
            {
                awaiter.OnCompleted(_RecvMsgR_cont);
            }
            return(new AwaitableWrapper <Msg>(_RecvMsgR_ra));
        }
Exemplo n.º 2
0
 public void Reset()
 {
     this.bytes    = null;
     this.offset   = 0;
     this.len      = 0;
     this.nextNode = null;
 }
Exemplo n.º 3
0
 public static AwaitableWrapper <Msg> RecvMsgR(this IMsgStream ms, BytesView bv)
 {
     if (ms is IMsgStreamReadR r)
     {
         return(r.RecvMsgR(bv));
     }
     return(new AwaitableWrapper <Msg>(ms.RecvMsg(bv)));
 }
Exemplo n.º 4
0
 public void OnWrite(BytesView bv)
 {
     if (WriteFilters?.Count > 0)
     {
         lock (WriteFilters)
             foreach (var item in WriteFilters)
             {
                 item(bv);
             }
     }
 }
Exemplo n.º 5
0
        public static Action <BytesView> GetAesFilter(bool isEncrypt, byte[] iv, byte[] key)
        {
            int keySize = key.Length * 8, blockSize = iv.Length * 8;
            int blockBytesSize = blockSize / 8;

            byte[]    buf    = new byte[blockSize / 8];
            BytesView bvBuf  = new BytesView(buf);
            var       aesalg = Aes.Create();

            aesalg.Padding   = PaddingMode.PKCS7;
            aesalg.KeySize   = keySize;
            aesalg.BlockSize = blockSize;
            aesalg.IV        = iv;
            aesalg.Key       = key;
            return((bv) => {
                if (bv.len == 0)
                {
                    return;
                }
                var pos = 0;
                if (isEncrypt)
                {
                    using (var ms = new MemoryStream(bv.bytes, bv.offset, bv.len))
                        using (var cryStream = new CryptoStream(ms, aesalg.CreateEncryptor(), CryptoStreamMode.Read)) {
                            int read;
                            int oldbytesSize = bv.len - (bv.len % blockBytesSize);
                            bv.len = oldbytesSize;
                            bv.nextNode = bvBuf;
                            while ((read = cryStream.Read(bv.bytes, bv.offset + pos, oldbytesSize - pos)) > 0)
                            {
                                pos += read;
                            }
                            while ((read = cryStream.Read(buf, pos - oldbytesSize, buf.Length - (pos - oldbytesSize))) > 0)
                            {
                                pos += read;
                            }
                        }
                }
                else
                {
                    using (var cryStream = new CryptoStream(new MemoryStream(bv.bytes, bv.offset, bv.len), aesalg.CreateDecryptor(), CryptoStreamMode.Read)) {
                        int read;
                        while ((read = cryStream.Read(buf, 0, blockBytesSize)) > 0)
                        {
                            for (int i = 0; i < read; i++)
                            {
                                bv.bytes[bv.offset + pos++] = buf[i];
                            }
                        }
                        bv.len = pos;
                    }
                }
            });
        }
Exemplo n.º 6
0
 public void OnRead(BytesView bv)
 {
     if (ReadFilters?.Count > 0)
     {
         lock (ReadFilters)
             for (int i = ReadFilters.Count - 1; i >= 0; i--)
             {
                 Action <BytesView> item = ReadFilters[i];
                 item(bv);
             }
     }
 }
Exemplo n.º 7
0
 public void TryRecycle()
 {
     if (Data == null)
     {
         return;
     }
     if (Data.nextNode != null)
     {
         Logging.logWithStackTrace("lastMsg.nextNode != null", Logging.Level.Warning);
     }
     BufferPool.GlobalPut(Data.bytes);
     Data = null;
 }
Exemplo n.º 8
0
 public void Set(BytesView bv)
 {
     this.bytes  = bv.bytes;
     this.offset = bv.offset;
     this.len    = bv.len;
 }
Exemplo n.º 9
0
        public static Action <BytesView> GetAesStreamFilter(bool isEncrypt, byte[] key) // is actually AES OFB
        {
            uint keySize = (uint)key.Length * 8, blockSize = 128;
            uint blockBytesSize     = blockSize / 8;
            uint encryptedZerosSize = blockBytesSize;

            byte[]    buf    = new byte[blockSize / 8];
            BytesView bvBuf  = new BytesView(buf);
            var       aesalg = new AesCryptoServiceProvider();

            aesalg.Mode      = CipherMode.CBC; // to generate OFB keystream, use CBC with all zeros byte array as input.
            aesalg.KeySize   = (int)keySize;
            aesalg.BlockSize = (int)blockSize;
            aesalg.Key       = key;
            ICryptoTransform enc = null;

            byte[] keystreamBuffer    = null;
            uint   keystreamBufferPos = 0;

            return(bv => {
                if (enc == null)
                {
                    if (isEncrypt)
                    {
                        bv.nextNode = bv.Clone();
                        bv.Set(aesalg.IV);
                        bv = bv.nextNode;
                    }
                    else
                    {
                        aesalg.IV = bv.GetBytes(0, (int)blockBytesSize);
                        for (uint i = 0; i < bv.len - blockBytesSize; i++)
                        {
                            bv[i] = bv[i + blockBytesSize];
                        }
                        bv.len -= (int)blockBytesSize;
                    }
                    enc = aesalg.CreateEncryptor();
                    if (enc.CanTransformMultipleBlocks)
                    {
                        encryptedZerosSize *= blocksPerPass;
                    }
                    keystreamBuffer = new byte[encryptedZerosSize];
                    keystreamBufferPos = encryptedZerosSize;
                }
                unsafe
                {
                    fixed(byte *ksBuf = keystreamBuffer)
                    do
                    {
                        var pos = (uint)bv.offset;
                        var end = pos + (uint)bv.len;
                        if (bv.bytes == null)
                        {
                            throw new ArgumentNullException("bv.bytes");
                        }
                        if (bv.bytes.Length < end)
                            throw new ArgumentException("bv.bytes.Length < offset + len");
                        fixed(byte *bytes = bv.bytes)
                        while (pos < end)
                        {
                            var remainningTmp = (uint)encryptedZerosSize - keystreamBufferPos;
                            if (remainningTmp == 0)
                            {
                                remainningTmp = encryptedZerosSize;
                                keystreamBufferPos = 0;
                                enc.TransformBlock(zeroesBytes, 0, (int)encryptedZerosSize, keystreamBuffer, 0);
                            }
                            var tmpEnd = pos + remainningTmp;
                            var thisEnd = end < tmpEnd ? end : tmpEnd;
                            var thisCount = thisEnd - pos;
                            NaiveUtils.XorBytesUnsafe(ksBuf + keystreamBufferPos, bytes + pos, (uint)thisCount);
                            keystreamBufferPos += thisCount;
                            pos += thisCount;
                        }

                        bv = bv.nextNode;
                    }while (bv != null);
                }
            });
        }
Exemplo n.º 10
0
        public static Action <BytesView> GetAesFilter2(bool isEncrypting, byte[] key)
        {
            int keySize = key.Length * 8, blockSize = 128;
            int blockBytesSize = blockSize / 8;

            byte[]    buf    = new byte[blockSize / 8];
            BytesView bvBuf  = new BytesView(buf);
            var       aesalg = new AesCryptoServiceProvider();

            aesalg.Padding = PaddingMode.PKCS7;
            //aesalg.Mode = CipherMode.CBC;
            aesalg.KeySize   = keySize;
            aesalg.BlockSize = blockSize;
            aesalg.Key       = key;
            bool firstPacket = true;

            if (isEncrypting)
            {
                return((bv) => {
                    if (firstPacket)
                    {
                        firstPacket = false;
                        aesalg.GenerateIV();
                        var tmp = bv.Clone();
                        bv.Set(aesalg.IV);
                        bv.nextNode = tmp;
                        bv = tmp;
                    }
                    if (bv.len == 0)
                    {
                        return;
                    }
                    var pos = 0;
                    using (var ms = new MemoryStream(bv.bytes, bv.offset, bv.len))
                        using (var cryStream = new CryptoStream(ms, aesalg.CreateEncryptor(), CryptoStreamMode.Read)) {
                            int read;
                            int oldbytesSize = bv.len - (bv.len % blockBytesSize);
                            bv.len = oldbytesSize;
                            bv.nextNode = bvBuf;
                            while ((read = cryStream.Read(bv.bytes, bv.offset + pos, oldbytesSize - pos)) > 0)
                            {
                                pos += read;
                            }
                            while ((read = cryStream.Read(buf, pos - oldbytesSize, buf.Length - (pos - oldbytesSize))) >
                                   0)
                            {
                                pos += read;
                            }
                            aesalg.IV = buf;
                        }
                });
            }
            else
            {
                return((bv) => {
                    if (firstPacket)
                    {
                        firstPacket = false;
                        aesalg.IV = bv.GetBytes(0, blockBytesSize);
                        //bv.offset += blockBytesSize;
                        for (int i = 0; i < bv.len - blockBytesSize; i++)
                        {
                            bv[i] = bv[i + blockBytesSize];
                        }
                        bv.len -= blockBytesSize;
                    }
                    if (bv.len == 0)
                    {
                        return;
                    }
                    var pos = 0;
                    using (var cryStream = new CryptoStream(new MemoryStream(bv.bytes, bv.offset, bv.len),
                                                            aesalg.CreateDecryptor(), CryptoStreamMode.Read)) {
                        int read;
                        int lastBlockPos = bv.len - blockBytesSize;
                        for (int i = 0; i < blockBytesSize; i++)
                        {
                            buf[i] = bv[lastBlockPos + i];
                        }
                        aesalg.IV = buf;
                        while ((read = cryStream.Read(bv.bytes, bv.offset + pos, bv.len - pos)) > 0)
                        {
                            pos += read;
                        }
                        bv.len = pos;
                    }
                });
            }
        }
Exemplo n.º 11
0
        public async Task <Msg> RecvMsg(BytesView buf)
        {
            var frame = await _readAsyncR(new BytesSegment(buf?.bytes, buf?.offset ?? 0, buf?.len ?? 0));

            return(new Msg(frame.bv));
        }
Exemplo n.º 12
0
 public Msg(BytesView data)
 {
     this.Data = data;
 }
Exemplo n.º 13
0
 public BytesSegment(BytesView bv)
 {
     this.Bytes  = bv.bytes;
     this.Offset = bv.offset;
     this.Len    = bv.len;
 }