Merge pull request #3591 from directhex/mono_libdir_fallback
[mono.git] / mcs / class / System.Data / System.Data.SqlClient / SqlCommand.cs
index 7d931bdf9a8e55df8eeeb8e8b305508b21ebc909..1fd5443df85192ea7e3fcffd5461c19356952b59 100644 (file)
@@ -44,28 +44,23 @@ using System.Collections.Specialized;
 using System.ComponentModel;
 using System.Data;
 using System.Data.Common;
-#if NET_2_0
 using System.Data.Sql;
-#endif
 using System.Runtime.InteropServices;
 using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
 using System.Xml;
 
 namespace System.Data.SqlClient {
        [DesignerAttribute ("Microsoft.VSDesigner.Data.VS.SqlCommandDesigner, "+ Consts.AssemblyMicrosoft_VSDesigner, "System.ComponentModel.Design.IDesigner")]
        [ToolboxItemAttribute ("System.Drawing.Design.ToolboxItem, "+ Consts.AssemblySystem_Drawing)]
-#if NET_2_0
        [DefaultEventAttribute ("RecordsAffected")]
        public sealed class SqlCommand : DbCommand, IDbCommand, ICloneable
-#else
-       public sealed class SqlCommand : Component, IDbCommand, ICloneable
-#endif // NET_2_0
        {
                #region Fields
 
                const int DEFAULT_COMMAND_TIMEOUT = 30;
 
-               bool disposed;
                int commandTimeout;
                bool designTimeVisible;
                string commandText;
@@ -76,10 +71,9 @@ namespace System.Data.SqlClient {
                CommandBehavior behavior = CommandBehavior.Default;
                SqlParameterCollection parameters;
                string preparedStatement;
-#if NET_2_0
+               bool disposed;
                SqlNotificationRequest notification;
                bool notificationAutoEnlist;
-#endif
 
                #endregion // Fields
 
@@ -109,9 +103,7 @@ namespace System.Data.SqlClient {
                        this.updatedRowSource = UpdateRowSource.Both;
 
                        this.commandTimeout = DEFAULT_COMMAND_TIMEOUT;
-#if NET_2_0
                        notificationAutoEnlist = true;
-#endif
                        designTimeVisible = true;
                        parameters = new SqlParameterCollection (this);
                }
@@ -138,16 +130,11 @@ namespace System.Data.SqlClient {
                        get { return behavior; }
                }
 
-#if !NET_2_0
-               [DataSysDescription ("Command text to execute.")]
-#endif
                [DefaultValue ("")]
                [EditorAttribute ("Microsoft.VSDesigner.Data.SQL.Design.SqlCommandTextEditor, "+ Consts.AssemblyMicrosoft_VSDesigner, "System.Drawing.Design.UITypeEditor, "+ Consts.AssemblySystem_Drawing )]
                [RefreshProperties (RefreshProperties.All)]
                public
-#if NET_2_0
                override
-#endif //NET_2_0
                string CommandText {
                        get {
                                if (commandText == null)
@@ -157,46 +144,32 @@ namespace System.Data.SqlClient {
                        set {
                                if (value != commandText && preparedStatement != null)
                                        Unprepare ();
-                               commandText = value; 
+                               commandText = value;
                        }
                }
 
-#if !NET_2_0
-               [DataSysDescription ("Time to wait for command to execute.")]
-               [DefaultValue (DEFAULT_COMMAND_TIMEOUT)]
-#endif
                public
-#if NET_2_0
                override
-#endif //NET_2_0
                int CommandTimeout {
                        get { return commandTimeout; }
                        set { 
                                if (value < 0)
-                                       throw new ArgumentException ("The property value assigned is less than 0.");
+                                       throw new ArgumentException ("The property value assigned is less than 0.",
+                                               "CommandTimeout");
                                commandTimeout = value; 
                        }
                }
 
-#if !NET_2_0
-               [DataSysDescription ("How to interpret the CommandText.")]
-#endif
                [DefaultValue (CommandType.Text)]
                [RefreshProperties (RefreshProperties.All)]
                public
-#if NET_2_0
                override
-#endif //NET_2_0
                CommandType CommandType {
                        get { return commandType; }
                        set { 
                                if (value == CommandType.TableDirect)
-#if NET_2_0
                                        throw new ArgumentOutOfRangeException ("CommandType.TableDirect is not supported " +
                                                "by the Mono SqlClient Data Provider.");
-#else
-                                       throw new ArgumentException ("CommandType.TableDirect is not supported by the Mono SqlClient Data Provider.");
-#endif
 
                                ExceptionHelper.CheckEnumValue (typeof (CommandType), value);
                                commandType = value; 
@@ -204,22 +177,13 @@ namespace System.Data.SqlClient {
                }
 
                [DefaultValue (null)]
-#if !NET_2_0
-               [DataSysDescription ("Connection used by the command.")]
-#endif
                [EditorAttribute ("Microsoft.VSDesigner.Data.Design.DbConnectionEditor, "+ Consts.AssemblyMicrosoft_VSDesigner, "System.Drawing.Design.UITypeEditor, "+ Consts.AssemblySystem_Drawing )]                
                public
-#if NET_2_0
                new
-#endif //NET_2_0
                SqlConnection Connection {
                        get { return connection; }
                        set
                        {
-#if ONLY_1_1
-                               if (connection != null && connection.DataReader != null)
-                                       throw new InvalidOperationException ("The connection is busy fetching data.");
-#endif
                                connection = value;
                        }
                }
@@ -227,26 +191,17 @@ namespace System.Data.SqlClient {
                [Browsable (false)]
                [DefaultValue (true)]
                [DesignOnly (true)]
-#if NET_2_0
                [EditorBrowsable (EditorBrowsableState.Never)]
-#endif
                public
-#if NET_2_0
                override
-#endif //NET_2_0
                bool DesignTimeVisible {
                        get { return designTimeVisible; } 
                        set { designTimeVisible = value; }
                }
 
-#if !NET_2_0
-               [DataSysDescription ("The parameters collection.")]
-#endif
                [DesignerSerializationVisibility (DesignerSerializationVisibility.Content)]
                public
-#if NET_2_0
                new
-#endif //NET_2_0
                SqlParameterCollection Parameters {
                        get { return parameters; }
                }
@@ -255,34 +210,8 @@ namespace System.Data.SqlClient {
                        get { return Connection.Tds; }
                }
 
-#if !NET_2_0
-               IDbConnection IDbCommand.Connection {
-                       get { return Connection; }
-                       set { 
-                               if (!(value == null || value is SqlConnection))
-                                       throw new InvalidCastException ("The value was not a valid SqlConnection.");
-                               Connection = (SqlConnection) value;
-                       }
-               }
-
-               IDataParameterCollection IDbCommand.Parameters {
-                       get { return Parameters; }
-               }
-
-               IDbTransaction IDbCommand.Transaction {
-                       get { return Transaction; }
-                       set {
-                               if (!(value == null || value is SqlTransaction))
-                                       throw new ArgumentException ();
-                               Transaction = (SqlTransaction) value; 
-                       }
-               }
-#endif
 
                [Browsable (false)]
-#if !NET_2_0
-               [DataSysDescription ("The transaction used by the command.")]
-#endif
                [DesignerSerializationVisibility (DesignerSerializationVisibility.Hidden)]
                public new SqlTransaction Transaction {
                        get {
@@ -292,22 +221,13 @@ namespace System.Data.SqlClient {
                        }
                        set
                        {
-#if ONLY_1_1
-                               if (connection != null && connection.DataReader != null)
-                                       throw new InvalidOperationException ("The connection is busy fetching data.");
-#endif
                                transaction = value;
                        }
                }
 
-#if !NET_2_0
-               [DataSysDescription ("When used by a DataAdapter.Update, how command results are applied to the current DataRow.")]
-#endif
                [DefaultValue (UpdateRowSource.Both)]
                public
-#if NET_2_0
                override
-#endif // NET_2_0
                UpdateRowSource UpdatedRowSource {
                        get { return updatedRowSource; }
                        set {
@@ -316,7 +236,6 @@ namespace System.Data.SqlClient {
                        }
                }
 
-#if NET_2_0
                [Browsable (false)]
                [DesignerSerializationVisibility (DesignerSerializationVisibility.Hidden)]
                public SqlNotificationRequest Notification {
@@ -329,15 +248,12 @@ namespace System.Data.SqlClient {
                        get { return notificationAutoEnlist; }
                        set { notificationAutoEnlist = value; }
                }
-#endif
                #endregion // Fields
 
                #region Methods
 
                public
-#if NET_2_0
                override
-#endif // NET_2_0
                void Cancel ()
                {
                        if (Connection == null || Connection.Tds == null)
@@ -345,12 +261,10 @@ namespace System.Data.SqlClient {
                        Connection.Tds.Cancel ();
                }
 
-#if NET_2_0
                public SqlCommand Clone ()
                {
                        return new SqlCommand (commandText, connection, transaction, commandType, updatedRowSource, designTimeVisible, commandTimeout, parameters);
                }
-#endif // NET_2_0
 
                internal void CloseDataReader ()
                {
@@ -373,6 +287,54 @@ namespace System.Data.SqlClient {
                        return new SqlParameter ();
                }
 
+               private string EscapeProcName (string name, bool schema)
+               {
+                       string procName;
+                       string tmpProcName = name.Trim ();
+                       int procNameLen = tmpProcName.Length;
+                       char[] brkts = new char [] {'[', ']'};
+                       bool foundMatching = false;
+                       int start = 0, count = procNameLen;
+                       int sindex = -1, eindex = -1;
+                       
+                       // We try to match most of the "brackets" combination here, however
+                       // there could be other combinations that may generate a different type
+                       // of exception in MS.NET
+                       
+                       if (procNameLen > 1) {
+                               if ((sindex = tmpProcName.IndexOf ('[')) <= 0)
+                                       foundMatching = true;
+                               else
+                                       foundMatching = false;
+                       
+                               if (foundMatching == true && sindex > -1) {
+                                       eindex = tmpProcName.IndexOf (']');
+                                       if (sindex > eindex && eindex != -1) {
+                                               foundMatching = false;
+                                       } else if (eindex == procNameLen-1) {
+                                               if (tmpProcName.IndexOfAny (brkts, 1, procNameLen-2) != -1) {
+                                                       foundMatching = false;
+                                               } else {
+                                                       start = 1;
+                                                       count = procNameLen - 2;
+                                               }
+                                       } else if (eindex == -1 && schema) {
+                                               foundMatching = true;
+                                       } else {
+                                               foundMatching = false;
+                                       }
+                               }
+                       
+                               if (foundMatching)
+                                       procName = tmpProcName.Substring (start, count);
+                               else
+                                       throw new ArgumentException (String.Format ("SqlCommand.CommandText property value is an invalid multipart name {0}, incorrect usage of quotes", CommandText));
+                       } else {
+                               procName = tmpProcName;
+                       }
+                       
+                       return procName;
+               }
                internal void DeriveParameters ()
                {
                        if (commandType != CommandType.StoredProcedure)
@@ -381,12 +343,19 @@ namespace System.Data.SqlClient {
 
                        string procName = CommandText;
                        string schemaName = String.Empty;
-                       int dotPosition = procName.IndexOf ('.');
+                       int dotPosition = procName.LastIndexOf ('.');
+
+                       // Procedure name can be: [database].[user].[procname]
                        if (dotPosition >= 0) {
                                schemaName = procName.Substring (0, dotPosition);
                                procName = procName.Substring (dotPosition + 1);
+                               if ((dotPosition = schemaName.LastIndexOf ('.')) >= 0)
+                                       schemaName = schemaName.Substring (dotPosition + 1);
                        }
                        
+                       procName = EscapeProcName (procName, false);
+                       schemaName = EscapeProcName (schemaName, true);
+                       
                        SqlParameterCollection localParameters = new SqlParameterCollection (this);
                        localParameters.Add ("@procedure_name", SqlDbType.NVarChar, procName.Length).Value = procName;
                        if (schemaName.Length > 0)
@@ -397,6 +366,7 @@ namespace System.Data.SqlClient {
                        try {
                                Connection.Tds.ExecProc (sql, localParameters.MetaParameters, 0, true);
                        } catch (TdsTimeoutException ex) {
+                               Connection.Tds.Reset ();
                                throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                        } catch (TdsInternalException ex) {
                                Connection.Close ();
@@ -457,6 +427,7 @@ namespace System.Data.SqlClient {
                                                // 1) Network is down/server is down/not reachable
                                                // 2) Somebody has an exclusive lock on Table/DB
                                                // In any of these cases, don't close the connection. Let the user do it
+                                               Connection.Tds.Reset ();
                                                throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                        } catch (TdsInternalException ex) {
                                                Connection.Close ();
@@ -473,6 +444,7 @@ namespace System.Data.SqlClient {
                                        try {
                                                Connection.Tds.Execute (sql, parms, CommandTimeout, wantResults);
                                        } catch (TdsTimeoutException ex) {
+                                               Connection.Tds.Reset ();
                                                throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                        } catch (TdsInternalException ex) {
                                                Connection.Close ();
@@ -485,7 +457,8 @@ namespace System.Data.SqlClient {
                                try {
                                        Connection.Tds.ExecPrepared (preparedStatement, parms, CommandTimeout, wantResults);
                                } catch (TdsTimeoutException ex) {
-                                               throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
+                                       Connection.Tds.Reset ();
+                                       throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                } catch (TdsInternalException ex) {
                                        Connection.Close ();
                                        throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
@@ -494,9 +467,7 @@ namespace System.Data.SqlClient {
                }
 
                public
-#if NET_2_0
                override
-#endif // NET_2_0
                int ExecuteNonQuery ()
                {
                        ValidateCommand ("ExecuteNonQuery", false);
@@ -507,6 +478,7 @@ namespace System.Data.SqlClient {
                                Execute (false);
                                result = Connection.Tds.RecordsAffected;
                        } catch (TdsTimeoutException e) {
+                               Connection.Tds.Reset ();
                                throw SqlException.FromTdsInternalException ((TdsInternalException) e);
                        }
 
@@ -522,28 +494,129 @@ namespace System.Data.SqlClient {
                public new SqlDataReader ExecuteReader (CommandBehavior behavior)
                {
                        ValidateCommand ("ExecuteReader", false);
+                       if ((behavior & CommandBehavior.SingleRow) != 0)
+                               behavior |= CommandBehavior.SingleResult;
                        this.behavior = behavior;
                        if ((behavior & CommandBehavior.SequentialAccess) != 0)
                                Tds.SequentialAccess = true;
-                       Execute (true);
-                       Connection.DataReader = new SqlDataReader (this);
-                       
-                       return Connection.DataReader;
+                       try {
+                               Execute (true);
+                               Connection.DataReader = new SqlDataReader (this);
+                               return Connection.DataReader;
+                       } catch {
+                               if ((behavior & CommandBehavior.CloseConnection) != 0)
+                                       Connection.Close ();
+                               throw;
+                       }
+               }
+
+               public new Task<SqlDataReader> ExecuteReaderAsync ()
+               {
+                       return ExecuteReaderAsync (CommandBehavior.Default, CancellationToken.None);
+               }
+
+               public new Task<SqlDataReader> ExecuteReaderAsync (CancellationToken cancellationToken)
+               {
+                       return ExecuteReaderAsync (behavior, CancellationToken.None);
+               }
+
+               public new Task<SqlDataReader> ExecuteReaderAsync (CommandBehavior behavior)
+               {
+                       return ExecuteReaderAsync (CommandBehavior.Default, CancellationToken.None);
+               }
+
+               public new Task<SqlDataReader> ExecuteReaderAsync (CommandBehavior behavior, CancellationToken cancellationToken)
+               {
+                       TaskCompletionSource<SqlDataReader> source = new TaskCompletionSource<SqlDataReader>();
+
+                       CancellationTokenRegistration registration = new CancellationTokenRegistration();
+                       if (cancellationToken.CanBeCanceled) {
+                               if (cancellationToken.IsCancellationRequested) {
+                                       source.SetCanceled();
+                                       return source.Task;
+                               }
+                               registration = cancellationToken.Register(CancelIgnoreFailure);
+                       }
+
+                       Task<SqlDataReader> returnedTask = source.Task;
+                       try {
+                               // TODO: RegisterForConnectionCloseNotification(ref returnedTask);
+
+                               Task<SqlDataReader>.Factory.FromAsync(BeginExecuteReaderAsync, EndExecuteReader, behavior, null).ContinueWith((t) => {
+                                       registration.Dispose();
+                                       if (t.IsFaulted) {
+                                               Exception e = t.Exception.InnerException;
+                                               source.SetException(e);
+                                       }
+                                       else {
+                                               if (t.IsCanceled) {
+                                                       source.SetCanceled();
+                                               }
+                                               else {
+                                                       source.SetResult(t.Result);
+                                               }
+                                       }
+                               }, TaskScheduler.Default);
+                       }
+                       catch (Exception e) {
+                               source.SetException(e);
+                       }
+
+                       return returnedTask;
+               }
+
+               public Task<XmlReader> ExecuteXmlReaderAsync ()
+               {
+                       return ExecuteXmlReaderAsync (CancellationToken.None);
+               }
+               public Task<XmlReader> ExecuteXmlReaderAsync (CancellationToken cancellationToken)
+               {
+                       TaskCompletionSource<XmlReader> source = new TaskCompletionSource<XmlReader>();
+
+                       CancellationTokenRegistration registration = new CancellationTokenRegistration();
+                       if (cancellationToken.CanBeCanceled) {
+                               if (cancellationToken.IsCancellationRequested) {
+                                       source.SetCanceled();
+                                       return source.Task;
+                               }
+                               registration = cancellationToken.Register(CancelIgnoreFailure);
+                       }
+
+                       Task<XmlReader> returnedTask = source.Task;
+                       try {
+                               // TODO: RegisterForConnectionCloseNotification(ref returnedTask);
+
+                               Task<XmlReader>.Factory.FromAsync(BeginExecuteXmlReader, EndExecuteXmlReader, null).ContinueWith((t) => {
+                                       registration.Dispose();
+                                       if (t.IsFaulted) {
+                                               Exception e = t.Exception.InnerException;
+                                               source.SetException(e);
+                                       }
+                                       else {
+                                               if (t.IsCanceled) {
+                                                       source.SetCanceled();
+                                               }
+                                               else {
+                                                       source.SetResult(t.Result);
+                                               }
+                                       }
+                               }, TaskScheduler.Default);
+                       }
+                       catch (Exception e) {
+                               source.SetException(e);
+                       }
+
+                       return returnedTask;
                }
 
                public
-#if NET_2_0
                override
-#endif // NET_2_0
                object ExecuteScalar ()
                {
                        try {
                                object result = null;
-#if NET_2_0
                                ValidateCommand ("ExecuteScalar", false);
-#else
-                               ValidateCommand ("ExecuteReader", false);
-#endif
                                behavior = CommandBehavior.Default;
                                Execute (true);
 
@@ -556,6 +629,7 @@ namespace System.Data.SqlClient {
                                                GetOutputParameters ();
                                        }
                                } catch (TdsTimeoutException ex) {
+                                       Connection.Tds.Reset ();
                                        throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                } catch (TdsInternalException ex) {
                                        Connection.Close ();
@@ -575,6 +649,7 @@ namespace System.Data.SqlClient {
                        try {
                                Execute (true);
                        } catch (TdsTimeoutException e) {
+                               Connection.Tds.Reset ();
                                throw SqlException.FromTdsInternalException ((TdsInternalException) e);
                        }
 
@@ -609,46 +684,31 @@ namespace System.Data.SqlClient {
 
                }
 
-#if !NET_2_0
-               IDbDataParameter IDbCommand.CreateParameter ()
-               {
-                       return CreateParameter ();
-               }
-
-               IDataReader IDbCommand.ExecuteReader ()
-               {
-                       return ExecuteReader ();
-               }
-
-               IDataReader IDbCommand.ExecuteReader (CommandBehavior behavior)
-               {
-                       return ExecuteReader (behavior);
-               }
-#endif
 
-#if NET_2_0
                protected override void Dispose (bool disposing)
                {
                        if (disposed) return;
                        if (disposing) {
                                parameters.Clear();
+                               if (Connection != null)
+                                       Connection.DataReader = null;
                        }
                        base.Dispose (disposing);
                        disposed = true;
                }
-#endif
 
                public
-#if NET_2_0
                override
-#endif // NET_2_0
                void Prepare ()
                {
-                       ValidateCommand ("Prepare", false);
+                       if (Connection == null)
+                               throw new NullReferenceException ();
 
-                       if (CommandType == CommandType.StoredProcedure)
+                       if (CommandType == CommandType.StoredProcedure || CommandType == CommandType.Text && Parameters.Count == 0)
                                return;
 
+                       ValidateCommand ("Prepare", false);
+
                        try {
                                foreach (SqlParameter param in Parameters)
                                        param.CheckIfInitialized ();
@@ -674,25 +734,26 @@ namespace System.Data.SqlClient {
                {
                        if (Connection == null)
                                throw new InvalidOperationException (String.Format ("{0}: A Connection object is required to continue.", method));
-                       if (Connection.Transaction != null && transaction != Connection.Transaction)
-                               throw new InvalidOperationException ("The Connection object does not have the same transaction as the command object.");
+                       if (Transaction == null && Connection.Transaction != null)
+                               throw new InvalidOperationException (String.Format (
+                                       "{0} requires a transaction if the command's connection is in a pending transaction.",
+                                       method));
+                       if (Transaction != null && Transaction.Connection != Connection)
+                               throw new InvalidOperationException ("The connection does not have the same transaction as the command.");
                        if (Connection.State != ConnectionState.Open)
-                               throw new InvalidOperationException (String.Format ("{0} requires an open Connection object to continue. This connection is closed.", method));
+                               throw new InvalidOperationException (String.Format ("{0} requires an open connection to continue. This connection is closed.", method));
                        if (CommandText.Length == 0)
-                               throw new InvalidOperationException ("The command text for this Command has not been set.");
+                               throw new InvalidOperationException (String.Format ("{0}: CommandText has not been set for this Command.", method));
                        if (Connection.DataReader != null)
                                throw new InvalidOperationException ("There is already an open DataReader associated with this Connection which must be closed first.");
                        if (Connection.XmlReader != null)
                                throw new InvalidOperationException ("There is already an open XmlReader associated with this Connection which must be closed first.");
-#if NET_2_0
                        if (async && !Connection.AsyncProcessing)
                                throw new InvalidOperationException ("This Connection object is not " + 
                                        "in Asynchronous mode. Use 'Asynchronous" +
                                        " Processing = true' to set it.");
-#endif // NET_2_0
                }
 
-#if NET_2_0
                protected override DbParameter CreateDbParameter ()
                {
                        return CreateParameter ();
@@ -716,11 +777,9 @@ namespace System.Data.SqlClient {
                        get { return Transaction; }
                        set { Transaction = (SqlTransaction) value; }
                }
-#endif // NET_2_0
 
                #endregion // Methods
 
-#if NET_2_0
                #region Asynchronous Methods
 
                internal IAsyncResult BeginExecuteInternal (CommandBehavior behavior, 
@@ -757,7 +816,7 @@ namespace System.Data.SqlClient {
                                        if (keyInfo || schemaOnly)
                                                epilog = sql2.ToString ();
                                        try {
-                                               Connection.Tds.BeginExecuteProcedure (prolog,
+                                               ar = Connection.Tds.BeginExecuteProcedure (prolog,
                                                                                      epilog,
                                                                                      CommandText,
                                                                                      !wantResults,
@@ -765,6 +824,7 @@ namespace System.Data.SqlClient {
                                                                                      callback,
                                                                                      state);
                                        } catch (TdsTimeoutException ex) {
+                                               Connection.Tds.Reset ();
                                                throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                        } catch (TdsInternalException ex) {
                                                Connection.Close ();
@@ -779,6 +839,7 @@ namespace System.Data.SqlClient {
                                                else
                                                        ar = Connection.Tds.BeginExecuteNonQuery (sql, parms, callback, state);
                                        } catch (TdsTimeoutException ex) {
+                                               Connection.Tds.Reset ();
                                                throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                        } catch (TdsInternalException ex) {
                                                Connection.Close ();
@@ -791,6 +852,7 @@ namespace System.Data.SqlClient {
                                try {
                                        Connection.Tds.ExecPrepared (preparedStatement, parms, CommandTimeout, wantResults);
                                } catch (TdsTimeoutException ex) {
+                                       Connection.Tds.Reset ();
                                        throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
                                } catch (TdsInternalException ex) {
                                        Connection.Close ();
@@ -848,6 +910,11 @@ namespace System.Data.SqlClient {
                        return BeginExecuteReader (callback, stateObject, CommandBehavior.Default);
                }
 
+               IAsyncResult BeginExecuteReaderAsync(CommandBehavior behavior, AsyncCallback callback, object stateObject)
+               {
+                       return BeginExecuteReader (callback, stateObject, behavior);
+               }
+
                public IAsyncResult BeginExecuteReader (AsyncCallback callback, object stateObject, CommandBehavior behavior)
                {
                        ValidateCommand ("BeginExecuteReader", true);
@@ -927,7 +994,9 @@ namespace System.Data.SqlClient {
 
                #endregion // Asynchronous Methods
 
+#pragma warning disable 0067
+               // TODO: Not implemented
                public event StatementCompletedEventHandler StatementCompleted;
-#endif // NET_2_0
+#pragma warning restore
        }
 }