public StringRowReader(NpgsqlRowDescription rowDesc, Stream inputStream, byte[] buffer, ByteBuffer bytes)
: base(rowDesc, inputStream, buffer, bytes)
{
_messageSize = PGUtil.ReadInt32(inputStream, buffer);
if (PGUtil.ReadInt16(inputStream, buffer) != rowDesc.NumFields)
{
throw new DataException();
}
}
public FieldData(Stream stream, NpgsqlBackendTypeMapping typeMapping, byte[] buffer, ByteBuffer queue)
{
Name = PGUtil.ReadString(stream, queue);
TableOID = PGUtil.ReadInt32(stream, buffer);
ColumnAttributeNumber = PGUtil.ReadInt16(stream, buffer);
TypeOID = PGUtil.ReadInt32(stream, buffer);
TypeInfo = typeMapping[TypeOID];
TypeSize = PGUtil.ReadInt16(stream, buffer);
TypeModifier = PGUtil.ReadInt32(stream, buffer);
FormatCode = (FormatCode)PGUtil.ReadInt16(stream, buffer);
}
private static NpgsqlCopyFormat ReadCopyHeader(Stream stream, byte[] buffer)
{
byte copyFormat = (byte)stream.ReadByte();
short numCopyFields = PGUtil.ReadInt16(stream, buffer);
short[] copyFieldFormats = new short[numCopyFields];
for (Int16 i = 0; i < numCopyFields; i++)
{
copyFieldFormats[i] = PGUtil.ReadInt16(stream, buffer);
}
return(new NpgsqlCopyFormat(copyFormat, copyFieldFormats));
}
public int NextMessage(Stream inputStream)
{
disposed = false;
_stream = inputStream;
_messageSize = PGUtil.ReadInt32(_stream, buffer);
if (PGUtil.ReadInt16(_stream, buffer) != _rowDesc.NumFields)
{
throw new DataException();
}
_nextFieldSize = null;
return(_messageSize);
}
public NpgsqlRowDescription(Stream stream, NpgsqlBackendTypeMapping type_mapping, Version compatVersion, byte[] buffer, ByteBuffer queue)
{
_compatVersion = compatVersion;
PGUtil.EatStreamBytes(stream, 4);
var num = PGUtil.ReadInt16(stream, buffer);
fields_data = new FieldData[num];
for (int i = 0; i < fields_data.Length; i++)
{
fields_data[i] = new FieldData(stream, type_mapping, buffer, queue);
}
}
public NpgsqlRowDescription Process(Stream stream, NpgsqlBackendTypeMapping type_mapping, byte[] buffer, ByteBuffer queue)
{
PGUtil.EatStreamBytes(stream, 4);
var num = PGUtil.ReadInt16(stream, buffer);
fields_data_len = num;
if (num > fields_data.Length)
{
fields_data = new FieldData[num];
}
for (int i = 0; i < fields_data_len; i++)
{
fields_data[i] = new FieldData(stream, type_mapping, buffer, queue);
}
_field_name_index_table = null;
_caseInsensitiveNameIndexTable = null;
return(this);
}
protected IEnumerable <IServerResponseObject> ProcessBackendResponses_Ver_3(NpgsqlConnector context)
{
try
{
Stream stream = context.Stream;
NpgsqlMediator mediator = context.Mediator;
var buffer = context.TmpBuffer;
var queue = context.ArrayBuffer;
List <NpgsqlError> errors = null;
for (; ;)
{
// Check the first Byte of response.
BackEndMessageCode message = (BackEndMessageCode)stream.ReadByte();
switch (message)
{
case BackEndMessageCode.ErrorResponse:
NpgsqlError error = new NpgsqlError(stream, buffer, queue);
error.ErrorSql = mediator.SqlSent;
if (errors == null)
{
errors = new List <NpgsqlError>();
}
errors.Add(error);
// Return imediately if it is in the startup state or connected state as
// there is no more messages to consume.
// Possible error in the NpgsqlStartupState:
// Invalid password.
// Possible error in the NpgsqlConnectedState:
// No pg_hba.conf configured.
if (!context.RequireReadyForQuery)
{
throw new NpgsqlException(errors);
}
break;
case BackEndMessageCode.AuthenticationRequest:
// Get the length in case we're getting AuthenticationGSSContinue
int authDataLength = PGUtil.ReadInt32(stream, buffer) - 8;
AuthenticationRequestType authType = (AuthenticationRequestType)PGUtil.ReadInt32(stream, buffer);
switch (authType)
{
case AuthenticationRequestType.AuthenticationOk:
break;
case AuthenticationRequestType.AuthenticationClearTextPassword:
// Send the PasswordPacket.
ChangeState(context, NpgsqlStartupState.Instance);
context.Authenticate(context.Password);
break;
case AuthenticationRequestType.AuthenticationMD5Password:
// Now do the "MD5-Thing"
// for this the Password has to be:
// 1. md5-hashed with the username as salt
// 2. md5-hashed again with the salt we get from the backend
MD5 md5 = MD5.Create();
// 1.
byte[] passwd = context.Password;
byte[] saltUserName = ENCODING_UTF8.GetBytes(context.UserName);
byte[] crypt_buf = new byte[passwd.Length + saltUserName.Length];
passwd.CopyTo(crypt_buf, 0);
saltUserName.CopyTo(crypt_buf, passwd.Length);
StringBuilder sb = new StringBuilder();
byte[] hashResult = md5.ComputeHash(crypt_buf);
foreach (byte b in hashResult)
{
sb.Append(b.ToString("x2"));
}
String prehash = sb.ToString();
byte[] prehashbytes = ENCODING_UTF8.GetBytes(prehash);
crypt_buf = new byte[prehashbytes.Length + 4];
stream.Read(crypt_buf, prehashbytes.Length, 4);
// Send the PasswordPacket.
ChangeState(context, NpgsqlStartupState.Instance);
// 2.
prehashbytes.CopyTo(crypt_buf, 0);
sb = new StringBuilder("md5"); // This is needed as the backend expects md5 result starts with "md5"
hashResult = md5.ComputeHash(crypt_buf);
foreach (byte b in hashResult)
{
sb.Append(b.ToString("x2"));
}
context.Authenticate(ENCODING_UTF8.GetBytes(sb.ToString()));
break;
#if WINDOWS && UNMANAGED
case AuthenticationRequestType.AuthenticationSSPI:
{
if (context.IntegratedSecurity)
{
// For SSPI we have to get the IP-Address (hostname doesn't work)
string ipAddressString = ((IPEndPoint)context.Socket.RemoteEndPoint).Address.ToString();
context.SSPI = new SSPIHandler(ipAddressString, "POSTGRES");
ChangeState(context, NpgsqlStartupState.Instance);
context.Authenticate(context.SSPI.Continue(null));
break;
}
else
{
// TODO: correct exception
throw new Exception();
}
}
case AuthenticationRequestType.AuthenticationGSSContinue:
{
byte[] authData = new byte[authDataLength];
PGUtil.CheckedStreamRead(stream, authData, 0, authDataLength);
byte[] passwd_read = context.SSPI.Continue(authData);
if (passwd_read.Length != 0)
{
context.Authenticate(passwd_read);
}
break;
}
#endif
default:
// Only AuthenticationClearTextPassword and AuthenticationMD5Password supported for now.
if (errors == null)
{
errors = new List <NpgsqlError>();
}
errors.Add(
new NpgsqlError(String.Format(resman.GetString("Exception_AuthenticationMethodNotSupported"), authType)));
throw new NpgsqlException(errors);
}
break;
case BackEndMessageCode.RowDescription:
yield return(context.RowDescription());
break;
case BackEndMessageCode.ParameterDescription:
// Do nothing,for instance, just read...
int length = PGUtil.ReadInt32(stream, buffer);
int nb_param = PGUtil.ReadInt16(stream, buffer);
//WTF
for (int i = 0; i < nb_param; i++)
{
int typeoid = PGUtil.ReadInt32(stream, buffer);
}
break;
case BackEndMessageCode.DataRow:
yield return(context.NextRow());
break;
case BackEndMessageCode.ReadyForQuery:
// Possible status bytes returned:
// I = Idle (no transaction active).
// T = In transaction, ready for more.
// E = Error in transaction, queries will fail until transaction aborted.
// Just eat the status byte, we have no use for it at this time.
PGUtil.ReadInt32(stream, buffer);
stream.ReadByte();
ChangeState(context, NpgsqlReadyState.Instance);
if (errors != null)
{
throw new NpgsqlException(errors);
}
yield break;
case BackEndMessageCode.BackendKeyData:
// BackendKeyData message.
NpgsqlBackEndKeyData backend_keydata = new NpgsqlBackEndKeyData(stream, buffer);
context.BackEndKeyData = backend_keydata;
// Wait for ReadForQuery message
break;
case BackEndMessageCode.NoticeResponse:
// Notices and errors are identical except that we
// just throw notices away completely ignored.
context.FireNotice(new NpgsqlError(stream, buffer, queue));
break;
case BackEndMessageCode.CompletedResponse:
PGUtil.ReadInt32(stream, buffer);
yield return(new CompletedResponse(stream, queue));
break;
case BackEndMessageCode.ParseComplete:
// Just read up the message length.
PGUtil.ReadInt32(stream, buffer);
yield break;
case BackEndMessageCode.BindComplete:
// Just read up the message length.
PGUtil.ReadInt32(stream, buffer);
yield break;
case BackEndMessageCode.EmptyQueryResponse:
PGUtil.ReadInt32(stream, buffer);
break;
case BackEndMessageCode.NotificationResponse:
// Eat the length
PGUtil.ReadInt32(stream, buffer);
context.FireNotification(new NpgsqlNotificationEventArgs(stream, true, buffer, queue));
if (context.IsNotificationThreadRunning)
{
yield break;
}
break;
case BackEndMessageCode.ParameterStatus:
NpgsqlParameterStatus parameterStatus = new NpgsqlParameterStatus(stream, queue);
context.AddParameterStatus(parameterStatus);
if (parameterStatus.Parameter == "server_version")
{
// Deal with this here so that if there are
// changes in a future backend version, we can handle it here in the
// protocol handler and leave everybody else put of it.
string versionString = parameterStatus.ParameterValue.Trim();
for (int idx = 0; idx != versionString.Length; ++idx)
{
char c = parameterStatus.ParameterValue[idx];
if (!char.IsDigit(c) && c != '.')
{
versionString = versionString.Substring(0, idx);
break;
}
}
context.ServerVersion = new Version(versionString);
}
break;
case BackEndMessageCode.NoData:
// This nodata message may be generated by prepare commands issued with queries which doesn't return rows
// for example insert, update or delete.
// Just eat the message.
PGUtil.ReadInt32(stream, buffer);
break;
case BackEndMessageCode.CopyInResponse:
// Enter COPY sub protocol and start pushing data to server
ChangeState(context, NpgsqlCopyInState.Instance);
PGUtil.ReadInt32(stream, buffer); // length redundant
context.CurrentState.StartCopy(context, ReadCopyHeader(stream, buffer));
yield break;
// Either StartCopy called us again to finish the operation or control should be passed for user to feed copy data
case BackEndMessageCode.CopyOutResponse:
// Enter COPY sub protocol and start pulling data from server
ChangeState(context, NpgsqlCopyOutState.Instance);
PGUtil.ReadInt32(stream, buffer); // length redundant
context.CurrentState.StartCopy(context, ReadCopyHeader(stream, buffer));
yield break;
// Either StartCopy called us again to finish the operation or control should be passed for user to feed copy data
case BackEndMessageCode.CopyData:
Int32 len = PGUtil.ReadInt32(stream, buffer) - 4;
byte[] buf = new byte[len];
PGUtil.ReadBytes(stream, buf, 0, len);
context.Mediator.ReceivedCopyData = buf;
yield break; // read data from server one chunk at a time while staying in copy operation mode
case BackEndMessageCode.CopyDone:
PGUtil.ReadInt32(stream, buffer); // CopyDone can not have content so this is always 4
// This will be followed by normal CommandComplete + ReadyForQuery so no op needed
break;
case BackEndMessageCode.IO_ERROR:
// Connection broken. Mono returns -1 instead of throwing an exception as ms.net does.
throw new IOException();
default:
// This could mean a number of things
// We've gotten out of sync with the backend?
// We need to implement this type?
// Backend has gone insane?
// FIXME
// what exception should we really throw here?
throw new NotSupportedException(String.Format("Backend sent unrecognized response type: {0}", (Char)message));
}
}
}
finally
{
context.RequireReadyForQuery = true;
}
}