* SqlCommand.cs: Moved connection reset before the throw.
[mono.git] / mcs / class / System.Data / System.Data.SqlClient / SqlCommand.cs
1 //
2 // System.Data.SqlClient.SqlCommand.cs
3 //
4 // Author:
5 //   Rodrigo Moya (rodrigo@ximian.com)
6 //   Daniel Morgan (danmorg@sc.rr.com)
7 //   Tim Coleman (tim@timcoleman.com)
8 //   Diego Caravana (diego@toth.it)
9 //
10 // (C) Ximian, Inc 2002 http://www.ximian.com/
11 // (C) Daniel Morgan, 2002
12 // Copyright (C) Tim Coleman, 2002
13 //
14
15 //
16 // Copyright (C) 2004 Novell, Inc (http://www.novell.com)
17 //
18 // Permission is hereby granted, free of charge, to any person obtaining
19 // a copy of this software and associated documentation files (the
20 // "Software"), to deal in the Software without restriction, including
21 // without limitation the rights to use, copy, modify, merge, publish,
22 // distribute, sublicense, and/or sell copies of the Software, and to
23 // permit persons to whom the Software is furnished to do so, subject to
24 // the following conditions:
25 // 
26 // The above copyright notice and this permission notice shall be
27 // included in all copies or substantial portions of the Software.
28 // 
29 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
30 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
31 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
32 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
33 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
34 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
35 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
36 //
37
38 using Mono.Data.Tds;
39 using Mono.Data.Tds.Protocol;
40 using System;
41 using System.IO;
42 using System.Collections;
43 using System.Collections.Specialized;
44 using System.ComponentModel;
45 using System.Data;
46 using System.Data.Common;
47 #if NET_2_0
48 using System.Data.Sql;
49 #endif
50 using System.Runtime.InteropServices;
51 using System.Text;
52 using System.Xml;
53
54 namespace System.Data.SqlClient {
55         [DesignerAttribute ("Microsoft.VSDesigner.Data.VS.SqlCommandDesigner, "+ Consts.AssemblyMicrosoft_VSDesigner, "System.ComponentModel.Design.IDesigner")]
56         [ToolboxItemAttribute ("System.Drawing.Design.ToolboxItem, "+ Consts.AssemblySystem_Drawing)]
57 #if NET_2_0
58         [DefaultEventAttribute ("RecordsAffected")]
59         public sealed class SqlCommand : DbCommand, IDbCommand, ICloneable
60 #else
61         public sealed class SqlCommand : Component, IDbCommand, ICloneable
62 #endif // NET_2_0
63         {
64                 #region Fields
65
66                 const int DEFAULT_COMMAND_TIMEOUT = 30;
67
68                 int commandTimeout;
69                 bool designTimeVisible;
70                 string commandText;
71                 CommandType commandType;
72                 SqlConnection connection;
73                 SqlTransaction transaction;
74                 UpdateRowSource updatedRowSource;
75                 CommandBehavior behavior = CommandBehavior.Default;
76                 SqlParameterCollection parameters;
77                 string preparedStatement;
78 #if NET_2_0
79                 bool disposed;
80                 SqlNotificationRequest notification;
81                 bool notificationAutoEnlist;
82 #endif
83
84                 #endregion // Fields
85
86                 #region Constructors
87
88                 public SqlCommand() 
89                         : this (String.Empty, null, null)
90                 {
91                 }
92
93                 public SqlCommand (string cmdText)
94                         : this (cmdText, null, null)
95                 {
96                 }
97
98                 public SqlCommand (string cmdText, SqlConnection connection)
99                         : this (cmdText, connection, null)
100                 {
101                 }
102
103                 public SqlCommand (string cmdText, SqlConnection connection, SqlTransaction transaction) 
104                 {
105                         this.commandText = cmdText;
106                         this.connection = connection;
107                         this.transaction = transaction;
108                         this.commandType = CommandType.Text;
109                         this.updatedRowSource = UpdateRowSource.Both;
110
111                         this.commandTimeout = DEFAULT_COMMAND_TIMEOUT;
112 #if NET_2_0
113                         notificationAutoEnlist = true;
114 #endif
115                         designTimeVisible = true;
116                         parameters = new SqlParameterCollection (this);
117                 }
118
119                 private SqlCommand(string commandText, SqlConnection connection, SqlTransaction transaction, CommandType commandType, UpdateRowSource updatedRowSource, bool designTimeVisible, int commandTimeout, SqlParameterCollection parameters)
120                 {
121                         this.commandText = commandText;
122                         this.connection = connection;
123                         this.transaction = transaction;
124                         this.commandType = commandType;
125                         this.updatedRowSource = updatedRowSource;
126                         this.designTimeVisible = designTimeVisible;
127                         this.commandTimeout = commandTimeout;
128                         this.parameters = new SqlParameterCollection(this);
129                         for (int i = 0;i < parameters.Count;i++)
130                                 this.parameters.Add(((ICloneable)parameters[i]).Clone());
131                 }
132
133                 #endregion // Constructors
134
135                 #region Properties
136
137                 internal CommandBehavior CommandBehavior {
138                         get { return behavior; }
139                 }
140
141 #if !NET_2_0
142                 [DataSysDescription ("Command text to execute.")]
143 #endif
144                 [DefaultValue ("")]
145                 [EditorAttribute ("Microsoft.VSDesigner.Data.SQL.Design.SqlCommandTextEditor, "+ Consts.AssemblyMicrosoft_VSDesigner, "System.Drawing.Design.UITypeEditor, "+ Consts.AssemblySystem_Drawing )]
146                 [RefreshProperties (RefreshProperties.All)]
147                 public
148 #if NET_2_0
149                 override
150 #endif //NET_2_0
151                 string CommandText {
152                         get {
153                                 if (commandText == null)
154                                         return string.Empty;
155                                 return commandText;
156                         }
157                         set {
158                                 if (value != commandText && preparedStatement != null)
159                                         Unprepare ();
160                                 commandText = value;
161                         }
162                 }
163
164 #if !NET_2_0
165                 [DataSysDescription ("Time to wait for command to execute.")]
166                 [DefaultValue (DEFAULT_COMMAND_TIMEOUT)]
167 #endif
168                 public
169 #if NET_2_0
170                 override
171 #endif //NET_2_0
172                 int CommandTimeout {
173                         get { return commandTimeout; }
174                         set { 
175                                 if (value < 0)
176 #if NET_2_0
177                                         throw new ArgumentException ("The property value assigned is less than 0.",
178                                                 "CommandTimeout");
179 #else
180                                         throw new ArgumentException ("The property value assigned is less than 0.");
181 #endif
182                                 commandTimeout = value; 
183                         }
184                 }
185
186 #if !NET_2_0
187                 [DataSysDescription ("How to interpret the CommandText.")]
188 #endif
189                 [DefaultValue (CommandType.Text)]
190                 [RefreshProperties (RefreshProperties.All)]
191                 public
192 #if NET_2_0
193                 override
194 #endif //NET_2_0
195                 CommandType CommandType {
196                         get { return commandType; }
197                         set { 
198                                 if (value == CommandType.TableDirect)
199 #if NET_2_0
200                                         throw new ArgumentOutOfRangeException ("CommandType.TableDirect is not supported " +
201                                                 "by the Mono SqlClient Data Provider.");
202 #else
203                                         throw new ArgumentException ("CommandType.TableDirect is not supported by the Mono SqlClient Data Provider.");
204 #endif
205
206                                 ExceptionHelper.CheckEnumValue (typeof (CommandType), value);
207                                 commandType = value; 
208                         }
209                 }
210
211                 [DefaultValue (null)]
212 #if !NET_2_0
213                 [DataSysDescription ("Connection used by the command.")]
214 #endif
215                 [EditorAttribute ("Microsoft.VSDesigner.Data.Design.DbConnectionEditor, "+ Consts.AssemblyMicrosoft_VSDesigner, "System.Drawing.Design.UITypeEditor, "+ Consts.AssemblySystem_Drawing )]                
216                 public
217 #if NET_2_0
218                 new
219 #endif //NET_2_0
220                 SqlConnection Connection {
221                         get { return connection; }
222                         set
223                         {
224 #if ONLY_1_1
225                                 if (connection != null && connection.DataReader != null)
226                                         throw new InvalidOperationException ("The connection is busy fetching data.");
227 #endif
228                                 connection = value;
229                         }
230                 }
231
232                 [Browsable (false)]
233                 [DefaultValue (true)]
234                 [DesignOnly (true)]
235 #if NET_2_0
236                 [EditorBrowsable (EditorBrowsableState.Never)]
237 #endif
238                 public
239 #if NET_2_0
240                 override
241 #endif //NET_2_0
242                 bool DesignTimeVisible {
243                         get { return designTimeVisible; } 
244                         set { designTimeVisible = value; }
245                 }
246
247 #if !NET_2_0
248                 [DataSysDescription ("The parameters collection.")]
249 #endif
250                 [DesignerSerializationVisibility (DesignerSerializationVisibility.Content)]
251                 public
252 #if NET_2_0
253                 new
254 #endif //NET_2_0
255                 SqlParameterCollection Parameters {
256                         get { return parameters; }
257                 }
258
259                 internal Tds Tds {
260                         get { return Connection.Tds; }
261                 }
262
263 #if !NET_2_0
264                 IDbConnection IDbCommand.Connection {
265                         get { return Connection; }
266                         set { 
267                                 if (!(value == null || value is SqlConnection))
268                                         throw new InvalidCastException ("The value was not a valid SqlConnection.");
269                                 Connection = (SqlConnection) value;
270                         }
271                 }
272
273                 IDataParameterCollection IDbCommand.Parameters {
274                         get { return Parameters; }
275                 }
276
277                 IDbTransaction IDbCommand.Transaction {
278                         get { return Transaction; }
279                         set {
280                                 if (!(value == null || value is SqlTransaction))
281                                         throw new ArgumentException ();
282                                 Transaction = (SqlTransaction) value; 
283                         }
284                 }
285 #endif
286
287                 [Browsable (false)]
288 #if !NET_2_0
289                 [DataSysDescription ("The transaction used by the command.")]
290 #endif
291                 [DesignerSerializationVisibility (DesignerSerializationVisibility.Hidden)]
292                 public new SqlTransaction Transaction {
293                         get {
294                                 if (transaction != null && !transaction.IsOpen)
295                                         transaction = null;
296                                 return transaction;
297                         }
298                         set
299                         {
300 #if ONLY_1_1
301                                 if (connection != null && connection.DataReader != null)
302                                         throw new InvalidOperationException ("The connection is busy fetching data.");
303 #endif
304                                 transaction = value;
305                         }
306                 }
307
308 #if !NET_2_0
309                 [DataSysDescription ("When used by a DataAdapter.Update, how command results are applied to the current DataRow.")]
310 #endif
311                 [DefaultValue (UpdateRowSource.Both)]
312                 public
313 #if NET_2_0
314                 override
315 #endif // NET_2_0
316                 UpdateRowSource UpdatedRowSource {
317                         get { return updatedRowSource; }
318                         set {
319                                 ExceptionHelper.CheckEnumValue (typeof (UpdateRowSource), value);
320                                 updatedRowSource = value;
321                         }
322                 }
323
324 #if NET_2_0
325                 [Browsable (false)]
326                 [DesignerSerializationVisibility (DesignerSerializationVisibility.Hidden)]
327                 public SqlNotificationRequest Notification {
328                         get { return notification; }
329                         set { notification = value; }
330                 }
331
332                 [DefaultValue (true)]
333                 public bool NotificationAutoEnlist {
334                         get { return notificationAutoEnlist; }
335                         set { notificationAutoEnlist = value; }
336                 }
337 #endif
338                 #endregion // Fields
339
340                 #region Methods
341
342                 public
343 #if NET_2_0
344                 override
345 #endif // NET_2_0
346                 void Cancel ()
347                 {
348                         if (Connection == null || Connection.Tds == null)
349                                 return;
350                         Connection.Tds.Cancel ();
351                 }
352
353 #if NET_2_0
354                 public SqlCommand Clone ()
355                 {
356                         return new SqlCommand (commandText, connection, transaction, commandType, updatedRowSource, designTimeVisible, commandTimeout, parameters);
357                 }
358 #endif // NET_2_0
359
360                 internal void CloseDataReader ()
361                 {
362                         if (Connection != null) {
363                                 Connection.DataReader = null;
364
365                                 if ((behavior & CommandBehavior.CloseConnection) != 0)
366                                         Connection.Close ();
367
368                                 if (Tds != null)
369                                         Tds.SequentialAccess = false;
370                         }
371
372                         // Reset the behavior
373                         behavior = CommandBehavior.Default;
374                 }
375
376                 public new SqlParameter CreateParameter ()
377                 {
378                         return new SqlParameter ();
379                 }
380
381                 private string EscapeProcName (string name, bool schema)
382                 {
383                         string procName;
384                         string tmpProcName = name.Trim ();
385                         int procNameLen = tmpProcName.Length;
386                         char[] brkts = new char [] {'[', ']'};
387                         bool foundMatching = false;
388                         int start = 0, count = procNameLen;
389                         int sindex = -1, eindex = -1;
390                         
391                         // We try to match most of the "brackets" combination here, however
392                         // there could be other combinations that may generate a different type
393                         // of exception in MS.NET
394                         
395                         if (procNameLen > 1) {
396                                 if ((sindex = tmpProcName.IndexOf ('[')) <= 0)
397                                         foundMatching = true;
398                                 else
399                                         foundMatching = false;
400                         
401                                 if (foundMatching == true && sindex > -1) {
402                                         eindex = tmpProcName.IndexOf (']');
403                                         if (sindex > eindex && eindex != -1) {
404                                                 foundMatching = false;
405                                         } else if (eindex == procNameLen-1) {
406                                                 if (tmpProcName.IndexOfAny (brkts, 1, procNameLen-2) != -1) {
407                                                         foundMatching = false;
408                                                 } else {
409                                                         start = 1;
410                                                         count = procNameLen - 2;
411                                                 }
412                                         } else if (eindex == -1 && schema) {
413                                                 foundMatching = true;
414                                         } else {
415                                                 foundMatching = false;
416                                         }
417                                 }
418                         
419                                 if (foundMatching)
420                                         procName = tmpProcName.Substring (start, count);
421                                 else
422                                         throw new ArgumentException (String.Format ("SqlCommand.CommandText property value is an invalid multipart name {0}, incorrect usage of quotes", CommandText));
423                         } else {
424                                 procName = tmpProcName;
425                         }
426                         
427                         return procName;
428                 }
429                 internal void DeriveParameters ()
430                 {
431                         if (commandType != CommandType.StoredProcedure)
432                                 throw new InvalidOperationException (String.Format ("SqlCommand DeriveParameters only supports CommandType.StoredProcedure, not CommandType.{0}", commandType));
433                         ValidateCommand ("DeriveParameters", false);
434
435                         string procName = CommandText;
436                         string schemaName = String.Empty;
437                         int dotPosition = procName.IndexOf ('.');
438                         if (dotPosition >= 0) {
439                                 schemaName = procName.Substring (0, dotPosition);
440                                 procName = procName.Substring (dotPosition + 1);
441                         }
442                         
443                         procName = EscapeProcName (procName, false);
444                         schemaName = EscapeProcName (schemaName, true);
445                         
446                         SqlParameterCollection localParameters = new SqlParameterCollection (this);
447                         localParameters.Add ("@procedure_name", SqlDbType.NVarChar, procName.Length).Value = procName;
448                         if (schemaName.Length > 0)
449                                 localParameters.Add ("@procedure_schema", SqlDbType.NVarChar, schemaName.Length).Value = schemaName;
450                         
451                         string sql = "sp_procedure_params_rowset";
452
453                         try {
454                                 Connection.Tds.ExecProc (sql, localParameters.MetaParameters, 0, true);
455                         } catch (TdsTimeoutException ex) {
456                                 Connection.Tds.Reset ();
457                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
458                         } catch (TdsInternalException ex) {
459                                 Connection.Close ();
460                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
461                         }
462
463                         SqlDataReader reader = new SqlDataReader (this);
464                         parameters.Clear ();
465                         object[] dbValues = new object[reader.FieldCount];
466
467                         while (reader.Read ()) {
468                                 reader.GetValues (dbValues);
469                                 parameters.Add (new SqlParameter (dbValues));
470                         }
471                         reader.Close ();
472
473                         if (parameters.Count == 0)
474                                 throw new InvalidOperationException ("Stored procedure '" + procName + "' does not exist.");
475                 }
476
477                 private void Execute (bool wantResults)
478                 {
479                         int index = 0;
480                         Connection.Tds.RecordsAffected = -1;
481                         TdsMetaParameterCollection parms = Parameters.MetaParameters;
482                         foreach (TdsMetaParameter param in parms) {
483                                 param.Validate (index++);
484                         }
485
486                         if (preparedStatement == null) {
487                                 bool schemaOnly = ((behavior & CommandBehavior.SchemaOnly) > 0);
488                                 bool keyInfo = ((behavior & CommandBehavior.KeyInfo) > 0);
489
490                                 StringBuilder sql1 = new StringBuilder ();
491                                 StringBuilder sql2 = new StringBuilder ();
492
493                                 if (schemaOnly || keyInfo)
494                                         sql1.Append ("SET FMTONLY OFF;");
495                                 if (keyInfo) {
496                                         sql1.Append ("SET NO_BROWSETABLE ON;");
497                                         sql2.Append ("SET NO_BROWSETABLE OFF;");
498                                 }
499                                 if (schemaOnly) {
500                                         sql1.Append ("SET FMTONLY ON;");
501                                         sql2.Append ("SET FMTONLY OFF;");
502                                 }
503                                 
504                                 switch (CommandType) {
505                                 case CommandType.StoredProcedure:
506                                         try {
507                                                 if (keyInfo || schemaOnly)
508                                                         Connection.Tds.Execute (sql1.ToString ());
509                                                 Connection.Tds.ExecProc (CommandText, parms, CommandTimeout, wantResults);
510                                                 if (keyInfo || schemaOnly)
511                                                         Connection.Tds.Execute (sql2.ToString ());
512                                         } catch (TdsTimeoutException ex) {
513                                                 // If it is a timeout exception there can be many reasons:
514                                                 // 1) Network is down/server is down/not reachable
515                                                 // 2) Somebody has an exclusive lock on Table/DB
516                                                 // In any of these cases, don't close the connection. Let the user do it
517                                                 Connection.Tds.Reset ();
518                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
519                                         } catch (TdsInternalException ex) {
520                                                 Connection.Close ();
521                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
522                                         }
523                                         break;
524                                 case CommandType.Text:
525                                         string sql;
526                                         if (sql2.Length > 0) {
527                                                 sql = String.Format ("{0}{1};{2}", sql1.ToString (), CommandText, sql2.ToString ());
528                                         } else {
529                                                 sql = String.Format ("{0}{1}", sql1.ToString (), CommandText);
530                                         }
531                                         try {
532                                                 Connection.Tds.Execute (sql, parms, CommandTimeout, wantResults);
533                                         } catch (TdsTimeoutException ex) {
534                                                 Connection.Tds.Reset ();
535                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
536                                         } catch (TdsInternalException ex) {
537                                                 Connection.Close ();
538                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
539                                         }
540                                         break;
541                                 }
542                         }
543                         else {
544                                 try {
545                                         Connection.Tds.ExecPrepared (preparedStatement, parms, CommandTimeout, wantResults);
546                                 } catch (TdsTimeoutException ex) {
547                                         Connection.Tds.Reset ();
548                                         throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
549                                 } catch (TdsInternalException ex) {
550                                         Connection.Close ();
551                                         throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
552                                 }
553                         }
554                 }
555
556                 public
557 #if NET_2_0
558                 override
559 #endif // NET_2_0
560                 int ExecuteNonQuery ()
561                 {
562                         ValidateCommand ("ExecuteNonQuery", false);
563                         int result = 0;
564                         behavior = CommandBehavior.Default;
565
566                         try {
567                                 Execute (false);
568                                 result = Connection.Tds.RecordsAffected;
569                         } catch (TdsTimeoutException e) {
570                                 Connection.Tds.Reset ();
571                                 throw SqlException.FromTdsInternalException ((TdsInternalException) e);
572                         }
573
574                         GetOutputParameters ();
575                         return result;
576                 }
577
578                 public new SqlDataReader ExecuteReader ()
579                 {
580                         return ExecuteReader (CommandBehavior.Default);
581                 }
582
583                 public new SqlDataReader ExecuteReader (CommandBehavior behavior)
584                 {
585                         ValidateCommand ("ExecuteReader", false);
586                         if ((behavior & CommandBehavior.SingleRow) != 0)
587                                 behavior |= CommandBehavior.SingleResult;
588                         this.behavior = behavior;
589                         if ((behavior & CommandBehavior.SequentialAccess) != 0)
590                                 Tds.SequentialAccess = true;
591                         Execute (true);
592                         Connection.DataReader = new SqlDataReader (this);
593                         return Connection.DataReader;
594                 }
595
596                 public
597 #if NET_2_0
598                 override
599 #endif // NET_2_0
600                 object ExecuteScalar ()
601                 {
602                         try {
603                                 object result = null;
604 #if NET_2_0
605                                 ValidateCommand ("ExecuteScalar", false);
606 #else
607                                 ValidateCommand ("ExecuteReader", false);
608 #endif
609                                 behavior = CommandBehavior.Default;
610                                 Execute (true);
611
612                                 try {
613                                         if (Connection.Tds.NextResult () && Connection.Tds.NextRow ())
614                                                 result = Connection.Tds.ColumnValues[0];
615
616                                         if (commandType == CommandType.StoredProcedure) {
617                                                 Connection.Tds.SkipToEnd ();
618                                                 GetOutputParameters ();
619                                         }
620                                 } catch (TdsTimeoutException ex) {
621                                         Connection.Tds.Reset ();
622                                         throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
623                                 } catch (TdsInternalException ex) {
624                                         Connection.Close ();
625                                         throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
626                                 }
627
628                                 return result;
629                         } finally {
630                                 CloseDataReader ();
631                         }
632                 }
633
634                 public XmlReader ExecuteXmlReader ()
635                 {
636                         ValidateCommand ("ExecuteXmlReader", false);
637                         behavior = CommandBehavior.Default;
638                         try {
639                                 Execute (true);
640                         } catch (TdsTimeoutException e) {
641                                 Connection.Tds.Reset ();
642                                 throw SqlException.FromTdsInternalException ((TdsInternalException) e);
643                         }
644
645                         SqlDataReader dataReader = new SqlDataReader (this);
646                         SqlXmlTextReader textReader = new SqlXmlTextReader (dataReader);
647                         XmlReader xmlReader = new XmlTextReader (textReader);
648                         return xmlReader;
649                 }
650
651                 internal void GetOutputParameters ()
652                 {
653                         IList list = Connection.Tds.OutputParameters;
654
655                         if (list != null && list.Count > 0) {
656
657                                 int index = 0;
658                                 foreach (SqlParameter parameter in parameters) {
659                                         if (parameter.Direction != ParameterDirection.Input &&
660                                             parameter.Direction != ParameterDirection.ReturnValue) {
661                                                 parameter.Value = list [index];
662                                                 index += 1;
663                                         }
664                                         if (index >= list.Count)
665                                                 break;
666                                 }
667                         }
668                 }
669
670                 object ICloneable.Clone ()
671                 {
672                         return new SqlCommand (commandText, connection, transaction, commandType, updatedRowSource, designTimeVisible, commandTimeout, parameters);
673
674                 }
675
676 #if !NET_2_0
677                 IDbDataParameter IDbCommand.CreateParameter ()
678                 {
679                         return CreateParameter ();
680                 }
681
682                 IDataReader IDbCommand.ExecuteReader ()
683                 {
684                         return ExecuteReader ();
685                 }
686
687                 IDataReader IDbCommand.ExecuteReader (CommandBehavior behavior)
688                 {
689                         return ExecuteReader (behavior);
690                 }
691 #endif
692
693 #if NET_2_0
694                 protected override void Dispose (bool disposing)
695                 {
696                         if (disposed) return;
697                         if (disposing) {
698                                 parameters.Clear();
699                         }
700                         base.Dispose (disposing);
701                         disposed = true;
702                 }
703 #endif
704
705                 public
706 #if NET_2_0
707                 override
708 #endif // NET_2_0
709                 void Prepare ()
710                 {
711 #if NET_2_0
712                         if (Connection == null)
713                                 throw new NullReferenceException ();
714 #endif
715
716                         if (CommandType == CommandType.StoredProcedure || CommandType == CommandType.Text && Parameters.Count == 0)
717                                 return;
718
719                         ValidateCommand ("Prepare", false);
720
721                         try {
722                                 foreach (SqlParameter param in Parameters)
723                                         param.CheckIfInitialized ();
724                         } catch (Exception e) {
725                                 throw new InvalidOperationException ("SqlCommand.Prepare requires " + e.Message);
726                         }
727
728                         preparedStatement = Connection.Tds.Prepare (CommandText, Parameters.MetaParameters);
729                 }
730
731                 public void ResetCommandTimeout ()
732                 {
733                         commandTimeout = DEFAULT_COMMAND_TIMEOUT;
734                 }
735
736                 private void Unprepare ()
737                 {
738                         Connection.Tds.Unprepare (preparedStatement);
739                         preparedStatement = null;
740                 }
741
742                 private void ValidateCommand (string method, bool async)
743                 {
744                         if (Connection == null)
745                                 throw new InvalidOperationException (String.Format ("{0}: A Connection object is required to continue.", method));
746                         if (Transaction == null && Connection.Transaction != null)
747                                 throw new InvalidOperationException (String.Format (
748                                         "{0} requires a transaction if the command's connection is in a pending transaction.",
749 #if NET_2_0
750                                         method));
751 #else
752                                         "Execute"));
753 #endif
754                         if (Transaction != null && Transaction.Connection != Connection)
755                                 throw new InvalidOperationException ("The connection does not have the same transaction as the command.");
756                         if (Connection.State != ConnectionState.Open)
757                                 throw new InvalidOperationException (String.Format ("{0} requires an open connection to continue. This connection is closed.", method));
758                         if (CommandText.Length == 0)
759                                 throw new InvalidOperationException (String.Format ("{0}: CommandText has not been set for this Command.", method));
760                         if (Connection.DataReader != null)
761                                 throw new InvalidOperationException ("There is already an open DataReader associated with this Connection which must be closed first.");
762                         if (Connection.XmlReader != null)
763                                 throw new InvalidOperationException ("There is already an open XmlReader associated with this Connection which must be closed first.");
764 #if NET_2_0
765                         if (async && !Connection.AsyncProcessing)
766                                 throw new InvalidOperationException ("This Connection object is not " + 
767                                         "in Asynchronous mode. Use 'Asynchronous" +
768                                         " Processing = true' to set it.");
769 #endif // NET_2_0
770                 }
771
772 #if NET_2_0
773                 protected override DbParameter CreateDbParameter ()
774                 {
775                         return CreateParameter ();
776                 }
777
778                 protected override DbDataReader ExecuteDbDataReader (CommandBehavior behavior)
779                 {
780                         return ExecuteReader (behavior);
781                 }
782
783                 protected override DbConnection DbConnection {
784                         get { return Connection; }
785                         set { Connection = (SqlConnection) value; }
786                 }
787
788                 protected override DbParameterCollection DbParameterCollection {
789                         get { return Parameters; }
790                 }
791
792                 protected override DbTransaction DbTransaction {
793                         get { return Transaction; }
794                         set { Transaction = (SqlTransaction) value; }
795                 }
796 #endif // NET_2_0
797
798                 #endregion // Methods
799
800 #if NET_2_0
801                 #region Asynchronous Methods
802
803                 internal IAsyncResult BeginExecuteInternal (CommandBehavior behavior, 
804                                                                                                         bool wantResults,
805                                                                                                         AsyncCallback callback, 
806                                                                                                         object state)
807                 {
808                         IAsyncResult ar = null;
809                         Connection.Tds.RecordsAffected = -1;
810                         TdsMetaParameterCollection parms = Parameters.MetaParameters;
811                         if (preparedStatement == null) {
812                                 bool schemaOnly = ((behavior & CommandBehavior.SchemaOnly) > 0);
813                                 bool keyInfo = ((behavior & CommandBehavior.KeyInfo) > 0);
814
815                                 StringBuilder sql1 = new StringBuilder ();
816                                 StringBuilder sql2 = new StringBuilder ();
817
818                                 if (schemaOnly || keyInfo)
819                                         sql1.Append ("SET FMTONLY OFF;");
820                                 if (keyInfo) {
821                                         sql1.Append ("SET NO_BROWSETABLE ON;");
822                                         sql2.Append ("SET NO_BROWSETABLE OFF;");
823                                 }
824                                 if (schemaOnly) {
825                                         sql1.Append ("SET FMTONLY ON;");
826                                         sql2.Append ("SET FMTONLY OFF;");
827                                 }
828                                 switch (CommandType) {
829                                 case CommandType.StoredProcedure:
830                                         string prolog = "";
831                                         string epilog = "";
832                                         if (keyInfo || schemaOnly)
833                                                 prolog = sql1.ToString ();
834                                         if (keyInfo || schemaOnly)
835                                                 epilog = sql2.ToString ();
836                                         try {
837                                                 Connection.Tds.BeginExecuteProcedure (prolog,
838                                                                                       epilog,
839                                                                                       CommandText,
840                                                                                       !wantResults,
841                                                                                       parms,
842                                                                                       callback,
843                                                                                       state);
844                                         } catch (TdsTimeoutException ex) {
845                                                 Connection.Tds.Reset ();
846                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
847                                         } catch (TdsInternalException ex) {
848                                                 Connection.Close ();
849                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
850                                         }
851                                         break;
852                                 case CommandType.Text:
853                                         string sql = String.Format ("{0}{1};{2}", sql1.ToString (), CommandText, sql2.ToString ());
854                                         try {
855                                                 if (wantResults)
856                                                         ar = Connection.Tds.BeginExecuteQuery (sql, parms, callback, state);
857                                                 else
858                                                         ar = Connection.Tds.BeginExecuteNonQuery (sql, parms, callback, state);
859                                         } catch (TdsTimeoutException ex) {
860                                                 Connection.Tds.Reset ();
861                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
862                                         } catch (TdsInternalException ex) {
863                                                 Connection.Close ();
864                                                 throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
865                                         }
866                                         break;
867                                 }
868                         }
869                         else {
870                                 try {
871                                         Connection.Tds.ExecPrepared (preparedStatement, parms, CommandTimeout, wantResults);
872                                 } catch (TdsTimeoutException ex) {
873                                         Connection.Tds.Reset ();
874                                         throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
875                                 } catch (TdsInternalException ex) {
876                                         Connection.Close ();
877                                         throw SqlException.FromTdsInternalException ((TdsInternalException) ex);
878                                 }
879                         }
880                         return ar;
881                 }
882
883                 internal void EndExecuteInternal (IAsyncResult ar)
884                 {
885                         SqlAsyncResult sqlResult = ( (SqlAsyncResult) ar);
886                         Connection.Tds.WaitFor (sqlResult.InternalResult);
887                         Connection.Tds.CheckAndThrowException (sqlResult.InternalResult);
888                 }
889
890                 public IAsyncResult BeginExecuteNonQuery ()
891                 {
892                         return BeginExecuteNonQuery (null, null);
893                 }
894
895                 public IAsyncResult BeginExecuteNonQuery (AsyncCallback callback, object stateObject)
896                 {
897                         ValidateCommand ("BeginExecuteNonQuery", true);
898                         SqlAsyncResult ar = new SqlAsyncResult (callback, stateObject);
899                         ar.EndMethod = "EndExecuteNonQuery";
900                         ar.InternalResult = BeginExecuteInternal (CommandBehavior.Default, false, ar.BubbleCallback, ar);
901                         return ar;
902                 }
903
904                 public int EndExecuteNonQuery (IAsyncResult asyncResult)
905                 {
906                         ValidateAsyncResult (asyncResult, "EndExecuteNonQuery");
907                         EndExecuteInternal (asyncResult);
908
909                         int ret = Connection.Tds.RecordsAffected;
910
911                         GetOutputParameters ();
912                         ((SqlAsyncResult) asyncResult).Ended = true;
913                         return ret;
914                 }
915
916                 public IAsyncResult BeginExecuteReader ()
917                 {
918                         return BeginExecuteReader (null, null, CommandBehavior.Default);
919                 }
920
921                 public IAsyncResult BeginExecuteReader (CommandBehavior behavior)
922                 {
923                         return BeginExecuteReader (null, null, behavior);
924                 }
925
926                 public IAsyncResult BeginExecuteReader (AsyncCallback callback, object stateObject)
927                 {
928                         return BeginExecuteReader (callback, stateObject, CommandBehavior.Default);
929                 }
930
931                 public IAsyncResult BeginExecuteReader (AsyncCallback callback, object stateObject, CommandBehavior behavior)
932                 {
933                         ValidateCommand ("BeginExecuteReader", true);
934                         this.behavior = behavior;
935                         SqlAsyncResult ar = new SqlAsyncResult (callback, stateObject);
936                         ar.EndMethod = "EndExecuteReader";
937                         IAsyncResult tdsResult = BeginExecuteInternal (behavior, true, 
938                                 ar.BubbleCallback, stateObject);
939                         ar.InternalResult = tdsResult;
940                         return ar;
941                 }
942
943                 public SqlDataReader EndExecuteReader (IAsyncResult asyncResult)
944                 {
945                         ValidateAsyncResult (asyncResult, "EndExecuteReader");
946                         EndExecuteInternal (asyncResult);
947                         SqlDataReader reader = null;
948                         try {
949                                 reader = new SqlDataReader (this);
950                         } catch (TdsTimeoutException e) {
951                                 throw SqlException.FromTdsInternalException ((TdsInternalException) e);
952                         } catch (TdsInternalException e) {
953                                 // if behavior is closeconnection, even if it throws exception
954                                 // the connection has to be closed.
955                                 if ((behavior & CommandBehavior.CloseConnection) != 0)
956                                         Connection.Close ();
957                                 throw SqlException.FromTdsInternalException ((TdsInternalException) e);
958                         }
959
960                         ((SqlAsyncResult) asyncResult).Ended = true;
961                         return reader;
962                 }
963
964                 public IAsyncResult BeginExecuteXmlReader (AsyncCallback callback, object stateObject)
965                 {
966                         ValidateCommand ("BeginExecuteXmlReader", true);
967                         SqlAsyncResult ar = new SqlAsyncResult (callback, stateObject);
968                         ar.EndMethod = "EndExecuteXmlReader";
969                         ar.InternalResult = BeginExecuteInternal (behavior, true, 
970                                 ar.BubbleCallback, stateObject);
971                         return ar;
972                 }
973
974                 public IAsyncResult BeginExecuteXmlReader ()
975                 {
976                         return BeginExecuteXmlReader (null, null);
977                 }
978                 
979
980                 public XmlReader EndExecuteXmlReader (IAsyncResult asyncResult)
981                 {
982                         ValidateAsyncResult (asyncResult, "EndExecuteXmlReader");
983                         EndExecuteInternal (asyncResult);
984                         SqlDataReader reader = new SqlDataReader (this);
985                         SqlXmlTextReader textReader = new SqlXmlTextReader (reader);
986                         XmlReader xmlReader = new XmlTextReader (textReader);
987                         ((SqlAsyncResult) asyncResult).Ended = true;
988                         return xmlReader;
989                 }
990
991                 internal void ValidateAsyncResult (IAsyncResult ar, string endMethod)
992                 {
993                         if (ar == null)
994                                 throw new ArgumentException ("result passed is null!");
995                         if (! (ar is SqlAsyncResult))
996                                 throw new ArgumentException (String.Format ("cannot test validity of types {0}",
997                                         ar.GetType ()));
998                         SqlAsyncResult result = (SqlAsyncResult) ar;
999                         if (result.EndMethod != endMethod)
1000                                 throw new InvalidOperationException (String.Format ("Mismatched {0} called for AsyncResult. " + 
1001                                         "Expected call to {1} but {0} is called instead.",
1002                                         endMethod, result.EndMethod));
1003                         if (result.Ended)
1004                                 throw new InvalidOperationException (String.Format ("The method {0} cannot be called " + 
1005                                         "more than once for the same AsyncResult.", endMethod));
1006                 }
1007
1008                 #endregion // Asynchronous Methods
1009
1010                 public event StatementCompletedEventHandler StatementCompleted;
1011 #endif // NET_2_0
1012         }
1013 }