Code:
/ FX-1434 / FX-1434 / 1.0 / untmp / whidbey / REDBITS / ndp / fx / src / Data / System / Data / SqlClient / SqlCommandSet.cs / 1 / SqlCommandSet.cs
//------------------------------------------------------------------------------ //// Copyright (c) Microsoft Corporation. All rights reserved. // //[....] //[....] //----------------------------------------------------------------------------- namespace System.Data.SqlClient { using System; using System.Collections.Generic; using System.ComponentModel; using System.Data; using System.Data.Common; using System.Diagnostics; using System.Globalization; using System.Text; using System.Text.RegularExpressions; internal sealed class SqlCommandSet { private const string SqlIdentifierPattern = "^@[\\p{Lo}\\p{Lu}\\p{Ll}\\p{Lm}_@#][\\p{Lo}\\p{Lu}\\p{Ll}\\p{Lm}\\p{Nd}\uff3f_@#\\$]*$"; private static readonly Regex SqlIdentifierParser = new Regex(SqlIdentifierPattern, RegexOptions.ExplicitCapture|RegexOptions.Singleline); private List_commandList = new List (); private SqlCommand _batchCommand; private static int _objectTypeCount; // Bid counter internal readonly int _objectID = System.Threading.Interlocked.Increment(ref _objectTypeCount); private sealed class LocalCommand { internal readonly string CommandText; internal readonly SqlParameterCollection Parameters; internal readonly int ReturnParameterIndex; internal readonly CommandType CmdType; internal LocalCommand(string commandText, SqlParameterCollection parameters, int returnParameterIndex, CommandType cmdType) { Debug.Assert(0 <= commandText.Length, "no text"); this.CommandText = commandText; this.Parameters = parameters; this.ReturnParameterIndex = returnParameterIndex; this.CmdType = cmdType; } } internal SqlCommandSet() : base() { _batchCommand = new SqlCommand(); } private SqlCommand BatchCommand { get { SqlCommand command = _batchCommand; if (null == command) { throw ADP.ObjectDisposed(this); } return command; } } internal int CommandCount { get { return CommandList.Count; } } private List CommandList { get { List commandList = _commandList; if (null == commandList) { throw ADP.ObjectDisposed(this); } return commandList; } } internal int CommandTimeout { /*get { return BatchCommand.CommandTimeout; }*/ set { BatchCommand.CommandTimeout = value; } } internal SqlConnection Connection { get { return BatchCommand.Connection; } set { BatchCommand.Connection = value; } } internal SqlTransaction Transaction { /*get { return BatchCommand.Transaction; }*/ set { BatchCommand.Transaction = value; } } internal int ObjectID { get { return _objectID; } } internal void Append(SqlCommand command) { ADP.CheckArgumentNull(command, "command"); Bid.Trace(" %d#, command=%d, parameterCount=%d\n", ObjectID, command.ObjectID, command.Parameters.Count); string cmdText = command.CommandText; if (ADP.IsEmpty(cmdText)) { throw ADP.CommandTextRequired(ADP.Append); } CommandType commandType = command.CommandType; switch(commandType) { case CommandType.Text: case CommandType.StoredProcedure: break; case CommandType.TableDirect: Debug.Assert(false, "command.CommandType"); throw System.Data.SqlClient.SQL.NotSupportedCommandType(commandType); default: Debug.Assert(false, "command.CommandType"); throw ADP.InvalidCommandType(commandType); } SqlParameterCollection parameters = null; SqlParameterCollection collection = command.Parameters; if (0 < collection.Count) { parameters = new SqlParameterCollection(); // clone parameters so they aren't destroyed for(int i = 0; i < collection.Count; ++i) { SqlParameter p = new SqlParameter(); collection[i].CopyTo(p); parameters.Add(p); // SQL Injection awarene if (!SqlIdentifierParser.IsMatch(p.ParameterName)) { throw ADP.BadParameterName(p.ParameterName); } } foreach(SqlParameter p in parameters) { // deep clone the parameter value if byte[] or char[] object obj = p.Value; byte[] byteValues = (obj as byte[]); if (null != byteValues) { int offset = p.Offset; int size = p.Size; int countOfBytes = byteValues.Length - offset; if ((0 != size) && (size < countOfBytes)) { countOfBytes = size; } byte[] copy = new byte[Math.Max(countOfBytes, 0)]; Buffer.BlockCopy(byteValues, offset, copy, 0, copy.Length); p.Offset = 0; p.Value = copy; } else { char[] charValues = (obj as char[]); if (null != charValues) { int offset = p.Offset; int size = p.Size; int countOfChars = charValues.Length - offset; if ((0 != size) && (size < countOfChars)) { countOfChars = size; } char[] copy = new char[Math.Max(countOfChars, 0)]; Buffer.BlockCopy(charValues, offset, copy, 0, copy.Length*2); p.Offset = 0; p.Value = copy; } else { ICloneable cloneable = (obj as ICloneable); if (null != cloneable) { p.Value = cloneable.Clone(); } } } } } int returnParameterIndex = -1; if (null != parameters) { for(int i = 0; i < parameters.Count; ++i) { if (ParameterDirection.ReturnValue == parameters[i].Direction) { returnParameterIndex = i; break; } } } LocalCommand cmd = new LocalCommand(cmdText, parameters, returnParameterIndex, command.CommandType); CommandList.Add(cmd); } internal static void BuildStoredProcedureName(StringBuilder builder, string part) { if ((null != part) && (0 < part.Length)) { if ('[' == part[0]) { int count = 0; foreach(char c in part) { if (']' == c) { count++; } } if (1 == (count%2)) { builder.Append(part); return; } } builder.Append("["); builder.Append(part.Replace("]", "]]")); // SQL Injection awarene builder.Append("]"); } } internal void Clear() { Bid.Trace(" %d#", ObjectID); DbCommand batchCommand = BatchCommand; if (null != batchCommand) { batchCommand.Parameters.Clear(); batchCommand.CommandText = null; } List commandList = _commandList; if (null != commandList) { commandList.Clear(); } } internal void Dispose() { Bid.Trace(" %d#", ObjectID); SqlCommand command = _batchCommand; _commandList = null; _batchCommand = null; if (null != command) { command.Dispose(); } } internal int ExecuteNonQuery() { SqlConnection.ExecutePermission.Demand(); IntPtr hscp; Bid.ScopeEnter(out hscp, " %d#", ObjectID); try { if (Connection.IsContextConnection) { throw SQL.BatchedUpdatesNotAvailableOnContextConnection(); } ValidateCommandBehavior(ADP.ExecuteNonQuery, CommandBehavior.Default); BatchCommand.BatchRPCMode = true; BatchCommand.ClearBatchCommand(); BatchCommand.Parameters.Clear(); for (int ii = 0 ; ii < _commandList.Count; ii++) { LocalCommand cmd = _commandList[ii]; BatchCommand.AddBatchCommand(cmd.CommandText, cmd.Parameters, cmd.CmdType); } return BatchCommand.ExecuteBatchRPCCommand(); } finally { Bid.ScopeLeave(ref hscp); } } internal SqlParameter GetParameter(int commandIndex, int parameterIndex) { return CommandList[commandIndex].Parameters[parameterIndex]; } internal bool GetBatchedAffected(int commandIdentifier, out int recordsAffected, out Exception error) { error = BatchCommand.GetErrors(commandIdentifier); int? affected = BatchCommand.GetRecordsAffected(commandIdentifier); recordsAffected = affected.GetValueOrDefault(); return affected.HasValue; } internal int GetParameterCount(int commandIndex) { return CommandList[commandIndex].Parameters.Count; } private void ValidateCommandBehavior(string method, CommandBehavior behavior) { if (0 != (behavior & ~(CommandBehavior.SequentialAccess|CommandBehavior.CloseConnection))) { ADP.ValidateCommandBehavior(behavior); throw ADP.NotSupportedCommandBehavior(behavior & ~(CommandBehavior.SequentialAccess|CommandBehavior.CloseConnection), method); } } } } // File provided for Reference Use Only by Microsoft Corporation (c) 2007. // Copyright (c) Microsoft Corporation. All rights reserved.
Link Menu
This book is available now!
Buy at Amazon US or
Buy at Amazon UK
- GroupBox.cs
- ObjectStateManagerMetadata.cs
- RegexTree.cs
- BaseTemplateParser.cs
- XmlToDatasetMap.cs
- DesignerLoader.cs
- SqlProviderManifest.cs
- DataObject.cs
- InstanceDataCollection.cs
- NumericPagerField.cs
- ExceptionUtil.cs
- RegexBoyerMoore.cs
- ConfigsHelper.cs
- ColorAnimationBase.cs
- UnsafeNativeMethods.cs
- ToolStripItemCollection.cs
- XmlTypeAttribute.cs
- HtmlContainerControl.cs
- StringKeyFrameCollection.cs
- WSSecurityOneDotZeroReceiveSecurityHeader.cs
- RectKeyFrameCollection.cs
- Brush.cs
- AddInToken.cs
- QilExpression.cs
- TypeRestriction.cs
- _NetRes.cs
- RightsManagementLicense.cs
- OracleBoolean.cs
- RequestCacheEntry.cs
- LinearGradientBrush.cs
- XsdBuilder.cs
- KeySplineConverter.cs
- TdsValueSetter.cs
- HostingEnvironment.cs
- ReadOnlyNameValueCollection.cs
- ContainerParaClient.cs
- OneOfElement.cs
- EFAssociationProvider.cs
- Encoder.cs
- CodeSnippetExpression.cs
- RoleGroupCollection.cs
- AvtEvent.cs
- login.cs
- CqlBlock.cs
- ClientTargetSection.cs
- PrimitiveType.cs
- ObjectListCommand.cs
- BamlTreeNode.cs
- MobileTextWriter.cs
- VisualTarget.cs
- UshortList2.cs
- PinnedBufferMemoryStream.cs
- WinFormsComponentEditor.cs
- HttpModulesSection.cs
- DataGridViewColumn.cs
- TextRunCacheImp.cs
- ServiceModelStringsVersion1.cs
- BaseResourcesBuildProvider.cs
- InternalTypeHelper.cs
- PageThemeCodeDomTreeGenerator.cs
- ReadWriteSpinLock.cs
- TextElement.cs
- AsmxEndpointPickerExtension.cs
- EnvironmentPermission.cs
- BitVector32.cs
- TextLineResult.cs
- UnmanagedMemoryStreamWrapper.cs
- ContentWrapperAttribute.cs
- WorkflowTransactionService.cs
- BitmapEffect.cs
- CodeDesigner.cs
- StrongNameUtility.cs
- panel.cs
- ReadOnlyKeyedCollection.cs
- Utils.cs
- Selection.cs
- SafeSecurityHandles.cs
- DataGridRowHeader.cs
- TypedTableBaseExtensions.cs
- ClickablePoint.cs
- DataRecordInfo.cs
- SecurityElementBase.cs
- XmlCharCheckingWriter.cs
- ConnectionConsumerAttribute.cs
- DataGridRowHeaderAutomationPeer.cs
- LongValidatorAttribute.cs
- NameNode.cs
- XmlQualifiedName.cs
- FaultImportOptions.cs
- SynchronizedPool.cs
- ExpandSegmentCollection.cs
- Accessors.cs
- ComponentManagerBroker.cs
- DataSetMappper.cs
- XmlParser.cs
- CollectionsUtil.cs
- EditorZoneBase.cs
- TargetInvocationException.cs
- XmlDocumentFragment.cs
- ExpressionBindingsDialog.cs