public static int GetTokenLength(this TdsPackageReader reader, byte token)
{
switch (token & TdsEnums.SQLLenMask)
{
case TdsEnums.SQLZeroLen:
return(0);
case TdsEnums.SQLFixedLen:
return(0x01 << ((token & 0x0c) >> 2));
default:
switch (token)
{
case TdsEnums.SQLFEATUREEXTACK:
return(-1);
case TdsEnums.SQLSESSIONSTATE:
return(reader.ReadInt32());
;
case TdsEnums.SQLRETURNVALUE:
return(-1); // In Yukon, the RETURNVALUE token stream no longer has length
default:
return(reader.ReadUInt16());
}
}
}
public static int ReadTdsTypeLen(this TdsPackageReader reader, int len) =>
len == 0
? 0
: len == 1
? reader.ReadByte()
: len == 2
? reader.ReadUInt16()
: reader.ReadInt32();
public static void SqlErrorAndInfo(this TdsPackageReader reader, byte token, int tokenLength)
{
var start = reader.GetReadPos();
var error = new SqlInfoAndError
{
Number = reader.ReadInt32(),
State = reader.ReadByte(),
Class = reader.ReadByte(),
Message = reader.ReadString(reader.ReadUInt16()),
Server = reader.ReadString(reader.ReadByte()),
Procedure = reader.ReadString(reader.ReadByte())
};
var current = reader.GetReadPos();
error.LineNumber = tokenLength - (current - start) > 2 ? reader.ReadInt32() : reader.ReadInt16();
if (error.Class >= TdsEnums.MIN_ERROR_CLASS)
{
throw new Exception(error.Message);
}
reader.CurrentSession.Errors.Add(error);
}
public static void ParseSessionState(this TdsPackageReader reader, int length)
{
var sdata = reader.CurrentSession.CurrentSessionData;
if (length < 5)
{
throw SQL.ParsingError();
}
var seqNum = reader.ReadUInt32();
if (seqNum == uint.MaxValue)
{
reader.CurrentSession.DoNotPoolThisConnection = true;
}
var status = reader.ReadByte();
if (status > 1)
{
throw SQL.ParsingError();
}
var recoverable = status != 0;
length -= 5;
while (length > 0)
{
var stateId = reader.ReadByte();
var stateLenByte = reader.ReadByte();
var stateLen = stateLenByte < 0xFF
? stateLenByte
: reader.ReadInt32();
byte[] buffer = null;
lock (sdata.Delta)
{
if (sdata.Delta[stateId] == null)
{
buffer = new byte[stateLen];
sdata.Delta[stateId] = new SessionStateRecord {
Version = seqNum, DataLength = stateLen, Data = buffer, Recoverable = recoverable
};
sdata.DeltaDirty = true;
if (!recoverable)
{
checked
{
sdata.UnrecoverableStatesCount++;
}
}
}
else
{
if (sdata.Delta[stateId].Version <= seqNum)
{
var sv = sdata.Delta[stateId];
sv.Version = seqNum;
sv.DataLength = stateLen;
if (sv.Recoverable != recoverable)
{
if (recoverable)
{
Debug.Assert(sdata.UnrecoverableStatesCount > 0, "Unrecoverable states count >0");
sdata.UnrecoverableStatesCount--;
}
else
{
checked
{
sdata.UnrecoverableStatesCount++;
}
}
sv.Recoverable = recoverable;
}
buffer = sv.Data;
if (buffer.Length < stateLen)
{
buffer = new byte[stateLen];
sv.Data = buffer;
}
}
}
}
if (buffer != null)
{
reader.ReadByteArray(buffer, 0, stateLen);
}
else
{
reader.GetBytes(stateLen);
}
length -= stateLenByte < 0xFF
? 2 + stateLen
: 6 + stateLen;
}
}
