Ejemplo n.º 1
0
        private void ReceiveEvent()
        {
            //Log.Info("Receive Event: " + eventArgs.BytesTransferred + "," + dataOffset + "," + dataLength + "," + inHeader);
            if (usingSocketTimeout)
            {
                eventReceived = true;
            }

            if (eventArgs.BytesTransferred <= 0)
            {
                ConnectionFailed(new AerospikeException.Connection("Connection closed"));
                return;
            }

            dataOffset += eventArgs.BytesTransferred;

            if (dataOffset < dataLength)
            {
                eventArgs.SetBuffer(dataOffset, dataLength - dataOffset);
                Receive();
                return;
            }
            dataOffset = segment.offset;

            if (inHeader)
            {
                int length = (int)(ByteUtil.BytesToLong(dataBuffer, dataOffset) & 0xFFFFFFFFFFFFL);

                if (length <= 0)
                {
                    ReceiveBegin();
                    return;
                }

                inHeader = false;

                if (length > segment.size)
                {
                    ResizeBuffer(length);
                    dataBuffer = segment.buffer;
                    dataOffset = segment.offset;
                }
                eventArgs.SetBuffer(dataBuffer, dataOffset, length);
                dataLength = dataOffset + length;
                Receive();
            }
            else
            {
                if (inAuthenticate)
                {
                    inAuthenticate = false;
                    inHeader       = true;

                    int resultCode = dataBuffer[dataOffset + 1];

                    if (resultCode != 0)
                    {
                        // Authentication failed. Session token probably expired.
                        // Signal tend thread to perform node login, so future
                        // transactions do not fail.
                        node.SignalLogin();

                        // This is a rare event because the client tracks session
                        // expiration and will relogin before session expiration.
                        // Do not try to login on same socket because login can take
                        // a long time and thousands of simultaneous logins could
                        // overwhelm server.
                        throw new AerospikeException(resultCode);
                    }
                    ConnectionReady();
                    return;
                }
                ParseCommand();
            }
        }
        internal void ReceiveEvent()
        {
            if (eventArgs.BytesTransferred <= 0)
            {
                Fail("Connection closed");
                return;
            }

            dataOffset += eventArgs.BytesTransferred;

            if (dataOffset < dataLength)
            {
                eventArgs.SetBuffer(dataOffset, dataLength - dataOffset);
                Receive();
                return;
            }
            dataOffset = 0;

            if (inHeader)
            {
                long proto  = ByteUtil.BytesToLong(dataBuffer, dataOffset);
                int  length = (int)(proto & 0xFFFFFFFFFFFFL);

                if (length <= 0)
                {
                    ReceiveBegin();
                    return;
                }

                inHeader = false;

                if (length > dataBuffer.Length)
                {
                    Fail("Invalid auth response");
                    return;
                }
                eventArgs.SetBuffer(dataOffset, length);
                dataLength = dataOffset + length;
                Receive();
            }
            else
            {
                int resultCode = dataBuffer[dataOffset + 1];

                if (resultCode != 0 && resultCode != ResultCode.SECURITY_NOT_ENABLED)
                {
                    // Authentication failed. Session token probably expired.
                    // Signal tend thread to perform node login, so future
                    // transactions do not fail.
                    node.SignalLogin();

                    // This is a rare event because the client tracks session
                    // expiration and will relogin before session expiration.
                    // Do not try to login on same socket because login can take
                    // a long time and thousands of simultaneous logins could
                    // overwhelm server.
                    Fail("Failed to authenticate: " + resultCode);
                    return;
                }
                ConnectionReady();
            }
        }
Ejemplo n.º 3
0
        private void ReceiveEvent()
        {
            //Log.Info("Receive Event: " + eventArgs.BytesTransferred + "," + dataOffset + "," + dataLength + "," + inHeader);
            if (socketWatch != null)
            {
                eventReceived = true;
            }

            if (eventArgs.BytesTransferred <= 0)
            {
                ConnectionFailed(new AerospikeException.Connection("Connection closed"));
                return;
            }

            dataOffset += eventArgs.BytesTransferred;

            if (dataOffset < dataLength)
            {
                eventArgs.SetBuffer(dataOffset, dataLength - dataOffset);
                Receive();
                return;
            }
            dataOffset = segment.offset;

            if (inHeader)
            {
                long proto  = ByteUtil.BytesToLong(dataBuffer, dataOffset);
                int  length = (int)(proto & 0xFFFFFFFFFFFFL);

                if (length <= 0)
                {
                    ReceiveBegin();
                    return;
                }

                compressed = ((proto >> 48) & 0xFF) == (long)Command.MSG_TYPE_COMPRESSED;
                inHeader   = false;

                if (length > segment.size)
                {
                    ResizeBuffer(length);
                    dataBuffer = segment.buffer;
                    dataOffset = segment.offset;
                }
                eventArgs.SetBuffer(dataBuffer, dataOffset, length);
                dataLength = dataOffset + length;
                Receive();
            }
            else
            {
                if (inAuthenticate)
                {
                    inAuthenticate = false;
                    inHeader       = true;

                    int resultCode = dataBuffer[dataOffset + 1];

                    if (resultCode != 0 && resultCode != ResultCode.SECURITY_NOT_ENABLED)
                    {
                        // Authentication failed. Session token probably expired.
                        // Signal tend thread to perform node login, so future
                        // transactions do not fail.
                        node.SignalLogin();

                        // This is a rare event because the client tracks session
                        // expiration and will relogin before session expiration.
                        // Do not try to login on same socket because login can take
                        // a long time and thousands of simultaneous logins could
                        // overwhelm server.
                        throw new AerospikeException(resultCode);
                    }
                    ConnectionReady();
                    return;
                }

                conn.UpdateLastUsed();

                if (compressed)
                {
                    int usize = (int)ByteUtil.BytesToLong(dataBuffer, dataOffset);
                    dataOffset += 8;
                    byte[] ubuf = new byte[usize];

                    ByteUtil.Decompress(dataBuffer, dataOffset, dataLength, ubuf, usize);
                    dataBuffer = ubuf;
                    dataOffset = 8;
                    dataLength = usize;
                }

                ParseCommand();
            }
        }