TdsValueSetter.cs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620
  1. //------------------------------------------------------------------------------
  2. // <copyright file="TdsValueSetter.cs" company="Microsoft">
  3. // Copyright (c) Microsoft Corporation. All rights reserved.
  4. // </copyright>
  5. // <owner current="true" primary="true">[....]</owner>
  6. // <owner current="true" primary="false">[....]</owner>
  7. //------------------------------------------------------------------------------
  8. namespace System.Data.SqlClient {
  9. using Microsoft.SqlServer.Server;
  10. using System;
  11. using System.Data;
  12. using System.Data.Common;
  13. using System.Data.SqlClient;
  14. using System.Data.SqlTypes;
  15. using System.Diagnostics;
  16. using System.Text;
  17. using MSS = Microsoft.SqlServer.Server;
  18. // TdsValueSetter handles writing a single value out to a TDS stream
  19. // This class can easily be extended to handle multiple versions of TDS by sub-classing and virtualizing
  20. // methods that have different formats or are not supported in one or the other version.
  21. internal class TdsValueSetter {
  22. #region Private fields
  23. private TdsParserStateObject _stateObj; // target to write to
  24. private SmiMetaData _metaData; // metadata describing value
  25. private bool _isPlp; // should this column be sent in PLP format?
  26. private bool _plpUnknownSent;// did we send initial UNKNOWN_LENGTH marker?
  27. private Encoder _encoder; // required for chunking character type data
  28. private SmiMetaData _variantType; // required for sql_variant
  29. #if DEBUG
  30. private int _currentOffset; // for chunking, verify that caller is using correct offsets
  31. #endif
  32. #endregion
  33. #region Exposed Construct/factory methods
  34. internal TdsValueSetter(TdsParserStateObject stateObj, SmiMetaData md) {
  35. _stateObj = stateObj;
  36. _metaData = md;
  37. _isPlp = MetaDataUtilsSmi.IsPlpFormat(md);
  38. _plpUnknownSent = false;
  39. _encoder = null;
  40. #if DEBUG
  41. _currentOffset = 0;
  42. #endif
  43. }
  44. #endregion
  45. #region Setters
  46. // Set value to null
  47. // valid for all types
  48. internal void SetDBNull() {
  49. Debug.Assert(!_plpUnknownSent, "Setting a column to null that we already stated sending!");
  50. if (_isPlp) {
  51. _stateObj.Parser.WriteUnsignedLong(TdsEnums.SQL_PLP_NULL, _stateObj);
  52. }
  53. else {
  54. switch(_metaData.SqlDbType) {
  55. case SqlDbType.BigInt:
  56. case SqlDbType.Bit:
  57. case SqlDbType.DateTime:
  58. case SqlDbType.Decimal:
  59. case SqlDbType.Float:
  60. case SqlDbType.Int:
  61. case SqlDbType.Money:
  62. case SqlDbType.Real:
  63. case SqlDbType.UniqueIdentifier:
  64. case SqlDbType.SmallDateTime:
  65. case SqlDbType.SmallInt:
  66. case SqlDbType.SmallMoney:
  67. case SqlDbType.TinyInt:
  68. case SqlDbType.Date:
  69. case SqlDbType.Time:
  70. case SqlDbType.DateTime2:
  71. case SqlDbType.DateTimeOffset:
  72. _stateObj.WriteByte(TdsEnums.FIXEDNULL);
  73. break;
  74. case SqlDbType.Binary:
  75. case SqlDbType.Char:
  76. case SqlDbType.Image:
  77. case SqlDbType.NChar:
  78. case SqlDbType.NText:
  79. case SqlDbType.NVarChar:
  80. case SqlDbType.Text:
  81. case SqlDbType.Timestamp:
  82. case SqlDbType.VarBinary:
  83. case SqlDbType.VarChar:
  84. _stateObj.Parser.WriteShort(TdsEnums.VARNULL, _stateObj);
  85. break;
  86. case SqlDbType.Udt:
  87. case SqlDbType.Xml:
  88. Debug.Assert(false, "PLP-only types shouldn't get to this point. Type: " + _metaData.SqlDbType);
  89. break;
  90. case SqlDbType.Variant:
  91. _stateObj.Parser.WriteInt(TdsEnums.FIXEDNULL, _stateObj);
  92. break;
  93. case SqlDbType.Structured:
  94. Debug.Assert(false, "Not yet implemented. Not needed until Structured UDTs");
  95. break;
  96. default:
  97. Debug.Assert(false, "Unexpected SqlDbType: " + _metaData.SqlDbType);
  98. break;
  99. }
  100. }
  101. }
  102. // valid for SqlDbType.Bit
  103. internal void SetBoolean(Boolean value) {
  104. Debug.Assert(
  105. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetBoolean));
  106. if (SqlDbType.Variant == _metaData.SqlDbType) {
  107. _stateObj.Parser.WriteSqlVariantHeader(3, TdsEnums.SQLBIT, 0, _stateObj);
  108. }
  109. else {
  110. _stateObj.WriteByte((byte)_metaData.MaxLength);
  111. }
  112. if (value) {
  113. _stateObj.WriteByte(1);
  114. }
  115. else {
  116. _stateObj.WriteByte(0);
  117. }
  118. }
  119. // valid for SqlDbType.TinyInt
  120. internal void SetByte(Byte value) {
  121. Debug.Assert(
  122. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetByte));
  123. if (SqlDbType.Variant == _metaData.SqlDbType) {
  124. _stateObj.Parser.WriteSqlVariantHeader(3, TdsEnums.SQLINT1, 0, _stateObj);
  125. }
  126. else {
  127. _stateObj.WriteByte((byte)_metaData.MaxLength);
  128. }
  129. _stateObj.WriteByte(value);
  130. }
  131. // Semantics for SetBytes are to modify existing value, not overwrite
  132. // Use in combination with SetLength to ensure overwriting when necessary
  133. // valid for SqlDbTypes: Binary, VarBinary, Image, Udt, Xml
  134. // (VarBinary assumed for variants)
  135. internal int SetBytes(long fieldOffset, byte[] buffer, int bufferOffset, int length) {
  136. Debug.Assert(
  137. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetBytes));
  138. CheckSettingOffset(fieldOffset);
  139. SetBytesNoOffsetHandling(fieldOffset, buffer, bufferOffset, length);
  140. #if DEBUG
  141. _currentOffset += length;
  142. #endif
  143. return length;
  144. }
  145. private void SetBytesNoOffsetHandling(long fieldOffset, byte[] buffer, int bufferOffset, int length) {
  146. if (_isPlp) {
  147. if (!_plpUnknownSent) {
  148. _stateObj.Parser.WriteUnsignedLong(TdsEnums.SQL_PLP_UNKNOWNLEN, _stateObj);
  149. _plpUnknownSent = true;
  150. }
  151. // Write chunk length & chunk
  152. _stateObj.Parser.WriteInt(length, _stateObj);
  153. _stateObj.WriteByteArray(buffer, length, bufferOffset);
  154. }
  155. else {
  156. // Non-plp data must be sent in one chunk for now.
  157. #if DEBUG
  158. Debug.Assert(0 == _currentOffset, "SetBytes doesn't yet support chunking for non-plp data: " + _currentOffset);
  159. #endif
  160. Debug.Assert(!MetaType.GetMetaTypeFromSqlDbType(_metaData.SqlDbType, _metaData.IsMultiValued).IsLong,
  161. "We're assuming long length types are sent as PLP. SqlDbType = " + _metaData.SqlDbType);
  162. if (SqlDbType.Variant == _metaData.SqlDbType) {
  163. _stateObj.Parser.WriteSqlVariantHeader(4 + length, TdsEnums.SQLBIGVARBINARY, 2, _stateObj);
  164. }
  165. _stateObj.Parser.WriteShort(length, _stateObj);
  166. _stateObj.WriteByteArray(buffer, length, bufferOffset);
  167. }
  168. }
  169. internal void SetBytesLength(long length) {
  170. Debug.Assert(
  171. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetBytes));
  172. CheckSettingOffset(length);
  173. if (0 == length) {
  174. if (_isPlp) {
  175. Debug.Assert(!_plpUnknownSent, "A plpUnknown has already been sent before setting length to zero.");
  176. _stateObj.Parser.WriteLong(0, _stateObj);
  177. _plpUnknownSent = true;
  178. }
  179. else {
  180. Debug.Assert(!MetaType.GetMetaTypeFromSqlDbType(_metaData.SqlDbType, _metaData.IsMultiValued).IsLong,
  181. "We're assuming long length types are sent as PLP. SqlDbType = " + _metaData.SqlDbType);
  182. if (SqlDbType.Variant == _metaData.SqlDbType) {
  183. _stateObj.Parser.WriteSqlVariantHeader(4, TdsEnums.SQLBIGVARBINARY, 2, _stateObj);
  184. }
  185. _stateObj.Parser.WriteShort(0, _stateObj);
  186. }
  187. }
  188. if (_plpUnknownSent) {
  189. _stateObj.Parser.WriteInt(TdsEnums.SQL_PLP_CHUNK_TERMINATOR, _stateObj);
  190. _plpUnknownSent = false;
  191. }
  192. #if DEBUG
  193. //
  194. _currentOffset = 0;
  195. #endif
  196. }
  197. // Semantics for SetChars are to modify existing value, not overwrite
  198. // Use in combination with SetLength to ensure overwriting when necessary
  199. // valid for character types: Char, VarChar, Text, NChar, NVarChar, NText
  200. // (NVarChar and global clr collation assumed for variants)
  201. internal int SetChars(long fieldOffset, char[] buffer, int bufferOffset, int length) {
  202. Debug.Assert(
  203. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetChars));
  204. // ANSI types must convert to byte[] because that's the tool we have.
  205. if (MetaDataUtilsSmi.IsAnsiType(_metaData.SqlDbType)) {
  206. if (null == _encoder) {
  207. _encoder = _stateObj.Parser._defaultEncoding.GetEncoder();
  208. }
  209. byte[] bytes = new byte[_encoder.GetByteCount(buffer, bufferOffset, length, false)];
  210. _encoder.GetBytes(buffer, bufferOffset, length, bytes, 0, false);
  211. SetBytesNoOffsetHandling(fieldOffset, bytes, 0, bytes.Length);
  212. }
  213. else {
  214. CheckSettingOffset(fieldOffset);
  215. // Send via PLP format if we can.
  216. if (_isPlp) {
  217. // Handle initial PLP markers
  218. if (!_plpUnknownSent) {
  219. _stateObj.Parser.WriteUnsignedLong(TdsEnums.SQL_PLP_UNKNOWNLEN, _stateObj);
  220. _plpUnknownSent = true;
  221. }
  222. // Write chunk length
  223. _stateObj.Parser.WriteInt(length*ADP.CharSize, _stateObj);
  224. _stateObj.Parser.WriteCharArray(buffer, length, bufferOffset, _stateObj);
  225. }
  226. else {
  227. // Non-plp data must be sent in one chunk for now.
  228. #if DEBUG
  229. Debug.Assert(0 == _currentOffset, "SetChars doesn't yet support chunking for non-plp data: " + _currentOffset);
  230. #endif
  231. if (SqlDbType.Variant == _metaData.SqlDbType) {
  232. _stateObj.Parser.WriteSqlVariantValue(new String(buffer, bufferOffset, length), length, 0, _stateObj);
  233. }
  234. else {
  235. Debug.Assert(!MetaType.GetMetaTypeFromSqlDbType(_metaData.SqlDbType, _metaData.IsMultiValued).IsLong,
  236. "We're assuming long length types are sent as PLP. SqlDbType = " + _metaData.SqlDbType);
  237. _stateObj.Parser.WriteShort(length*ADP.CharSize, _stateObj);
  238. _stateObj.Parser.WriteCharArray(buffer, length, bufferOffset, _stateObj);
  239. }
  240. }
  241. }
  242. #if DEBUG
  243. _currentOffset += length;
  244. #endif
  245. return length;
  246. }
  247. internal void SetCharsLength(long length) {
  248. Debug.Assert(
  249. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetChars));
  250. CheckSettingOffset(length);
  251. if (0 == length) {
  252. if (_isPlp) {
  253. Debug.Assert(!_plpUnknownSent, "A plpUnknown has already been sent before setting length to zero.");
  254. _stateObj.Parser.WriteLong(0, _stateObj);
  255. _plpUnknownSent = true;
  256. }
  257. else {
  258. Debug.Assert(!MetaType.GetMetaTypeFromSqlDbType(_metaData.SqlDbType, _metaData.IsMultiValued).IsLong,
  259. "We're assuming long length types are sent as PLP. SqlDbType = " + _metaData.SqlDbType);
  260. _stateObj.Parser.WriteShort(0, _stateObj);
  261. }
  262. }
  263. if (_plpUnknownSent) {
  264. _stateObj.Parser.WriteInt(TdsEnums.SQL_PLP_CHUNK_TERMINATOR, _stateObj);
  265. _plpUnknownSent = false;
  266. }
  267. _encoder = null;
  268. #if DEBUG
  269. //
  270. _currentOffset = 0;
  271. #endif
  272. }
  273. // valid for character types: Char, VarChar, Text, NChar, NVarChar, NText
  274. internal void SetString(string value, int offset, int length) {
  275. Debug.Assert(
  276. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetString));
  277. // ANSI types must convert to byte[] because that's the tool we have.
  278. if (MetaDataUtilsSmi.IsAnsiType(_metaData.SqlDbType)) {
  279. byte[] bytes;
  280. // Optimize for common case of writing entire string
  281. if (offset == 0 && value.Length <= length) {
  282. bytes = _stateObj.Parser._defaultEncoding.GetBytes(value);
  283. }
  284. else {
  285. char[] chars = value.ToCharArray(offset, length);
  286. bytes = _stateObj.Parser._defaultEncoding.GetBytes(chars);
  287. }
  288. SetBytes(0, bytes, 0, bytes.Length);
  289. SetBytesLength(bytes.Length);
  290. }
  291. else if (SqlDbType.Variant == _metaData.SqlDbType) {
  292. Debug.Assert(null != _variantType && SqlDbType.NVarChar == _variantType.SqlDbType, "Invalid variant type");
  293. SqlCollation collation = new SqlCollation();
  294. collation.LCID = checked((int)_variantType.LocaleId);
  295. collation.SqlCompareOptions = _variantType.CompareOptions;
  296. if (length * ADP.CharSize > TdsEnums.TYPE_SIZE_LIMIT) { // send as varchar for length greater than 4000
  297. byte[] bytes;
  298. // Optimize for common case of writing entire string
  299. if (offset == 0 && value.Length <= length) {
  300. bytes = _stateObj.Parser._defaultEncoding.GetBytes(value);
  301. }
  302. else {
  303. bytes = _stateObj.Parser._defaultEncoding.GetBytes(value.ToCharArray(offset, length));
  304. }
  305. _stateObj.Parser.WriteSqlVariantHeader(9 + bytes.Length, TdsEnums.SQLBIGVARCHAR, 7, _stateObj);
  306. _stateObj.Parser.WriteUnsignedInt(collation.info, _stateObj); // propbytes: collation.Info
  307. _stateObj.WriteByte(collation.sortId); // propbytes: collation.SortId
  308. _stateObj.Parser.WriteShort(bytes.Length, _stateObj); // propbyte: varlen
  309. _stateObj.WriteByteArray(bytes, bytes.Length, 0);
  310. }
  311. else {
  312. _stateObj.Parser.WriteSqlVariantHeader(9 + length * ADP.CharSize, TdsEnums.SQLNVARCHAR, 7, _stateObj);
  313. _stateObj.Parser.WriteUnsignedInt(collation.info, _stateObj); // propbytes: collation.Info
  314. _stateObj.WriteByte(collation.sortId); // propbytes: collation.SortId
  315. _stateObj.Parser.WriteShort(length * ADP.CharSize, _stateObj); // propbyte: varlen
  316. _stateObj.Parser.WriteString(value, length, offset, _stateObj);
  317. }
  318. _variantType = null;
  319. }
  320. else if (_isPlp) {
  321. // Send the string as a complete PLP chunk.
  322. _stateObj.Parser.WriteLong(length*ADP.CharSize, _stateObj); // PLP total length
  323. _stateObj.Parser.WriteInt(length*ADP.CharSize, _stateObj); // Chunk length
  324. _stateObj.Parser.WriteString(value, length, offset, _stateObj); // Data
  325. if (length != 0) {
  326. _stateObj.Parser.WriteInt(TdsEnums.SQL_PLP_CHUNK_TERMINATOR, _stateObj); // Terminator
  327. }
  328. }
  329. else {
  330. _stateObj.Parser.WriteShort(length*ADP.CharSize, _stateObj);
  331. _stateObj.Parser.WriteString(value, length, offset, _stateObj);
  332. }
  333. }
  334. // valid for SqlDbType.SmallInt
  335. internal void SetInt16(Int16 value) {
  336. Debug.Assert(
  337. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetInt16));
  338. if (SqlDbType.Variant == _metaData.SqlDbType) {
  339. _stateObj.Parser.WriteSqlVariantHeader(4, TdsEnums.SQLINT2, 0, _stateObj);
  340. }
  341. else {
  342. _stateObj.WriteByte((byte)_metaData.MaxLength);
  343. }
  344. _stateObj.Parser.WriteShort(value, _stateObj);
  345. }
  346. // valid for SqlDbType.Int
  347. internal void SetInt32(Int32 value) {
  348. Debug.Assert(
  349. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetInt32));
  350. if (SqlDbType.Variant == _metaData.SqlDbType) {
  351. _stateObj.Parser.WriteSqlVariantHeader(6, TdsEnums.SQLINT4, 0, _stateObj);
  352. }
  353. else {
  354. _stateObj.WriteByte((byte)_metaData.MaxLength);
  355. }
  356. _stateObj.Parser.WriteInt(value, _stateObj);
  357. }
  358. // valid for SqlDbType.BigInt, SqlDbType.Money, SqlDbType.SmallMoney
  359. internal void SetInt64(Int64 value) {
  360. Debug.Assert(
  361. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetInt64));
  362. if (SqlDbType.Variant == _metaData.SqlDbType) {
  363. if (null == _variantType) {
  364. _stateObj.Parser.WriteSqlVariantHeader(10, TdsEnums.SQLINT8, 0, _stateObj);
  365. _stateObj.Parser.WriteLong(value, _stateObj);
  366. }
  367. else {
  368. Debug.Assert(SqlDbType.Money == _variantType.SqlDbType, "Invalid variant type");
  369. _stateObj.Parser.WriteSqlVariantHeader(10, TdsEnums.SQLMONEY, 0, _stateObj);
  370. _stateObj.Parser.WriteInt((int)(value >> 0x20), _stateObj);
  371. _stateObj.Parser.WriteInt((int)value, _stateObj);
  372. _variantType = null;
  373. }
  374. }
  375. else {
  376. _stateObj.WriteByte((byte)_metaData.MaxLength);
  377. if (SqlDbType.SmallMoney == _metaData.SqlDbType) {
  378. _stateObj.Parser.WriteInt((int)value, _stateObj);
  379. }
  380. else if (SqlDbType.Money == _metaData.SqlDbType) {
  381. _stateObj.Parser.WriteInt((int)(value >> 0x20), _stateObj);
  382. _stateObj.Parser.WriteInt((int)value, _stateObj);
  383. }
  384. else {
  385. _stateObj.Parser.WriteLong(value, _stateObj);
  386. }
  387. }
  388. }
  389. // valid for SqlDbType.Real
  390. internal void SetSingle(Single value) {
  391. Debug.Assert(
  392. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetSingle));
  393. if (SqlDbType.Variant == _metaData.SqlDbType) {
  394. _stateObj.Parser.WriteSqlVariantHeader(6, TdsEnums.SQLFLT4, 0, _stateObj);
  395. }
  396. else {
  397. _stateObj.WriteByte((byte)_metaData.MaxLength);
  398. }
  399. _stateObj.Parser.WriteFloat(value, _stateObj);
  400. }
  401. // valid for SqlDbType.Float
  402. internal void SetDouble(Double value) {
  403. Debug.Assert(
  404. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetDouble));
  405. if (SqlDbType.Variant == _metaData.SqlDbType) {
  406. _stateObj.Parser.WriteSqlVariantHeader(10, TdsEnums.SQLFLT8, 0, _stateObj);
  407. }
  408. else {
  409. _stateObj.WriteByte((byte)_metaData.MaxLength);
  410. }
  411. _stateObj.Parser.WriteDouble(value, _stateObj);
  412. }
  413. // valid for SqlDbType.Numeric (uses SqlDecimal since Decimal cannot hold full range)
  414. internal void SetSqlDecimal(SqlDecimal value) {
  415. Debug.Assert(
  416. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetSqlDecimal));
  417. if (SqlDbType.Variant == _metaData.SqlDbType) {
  418. _stateObj.Parser.WriteSqlVariantHeader(21, TdsEnums.SQLNUMERICN, 2, _stateObj);
  419. _stateObj.WriteByte(value.Precision); // propbytes: precision
  420. _stateObj.WriteByte(value.Scale); // propbytes: scale
  421. _stateObj.Parser.WriteSqlDecimal(value, _stateObj);
  422. }
  423. else {
  424. _stateObj.WriteByte(checked((byte)MetaType.MetaDecimal.FixedLength)); // SmiMetaData's length and actual wire format's length are different
  425. _stateObj.Parser.WriteSqlDecimal(SqlDecimal.ConvertToPrecScale(value, _metaData.Precision, _metaData.Scale), _stateObj);
  426. }
  427. }
  428. // valid for DateTime, SmallDateTime, Date, DateTime2
  429. internal void SetDateTime(DateTime value) {
  430. Debug.Assert(
  431. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetDateTime));
  432. if (SqlDbType.Variant == _metaData.SqlDbType) {
  433. if ((_variantType != null) && (_variantType.SqlDbType == SqlDbType.DateTime2))
  434. {
  435. _stateObj.Parser.WriteSqlVariantDateTime2(value, _stateObj);
  436. }
  437. else if ((_variantType != null) && (_variantType.SqlDbType == SqlDbType.Date))
  438. {
  439. _stateObj.Parser.WriteSqlVariantDate(value, _stateObj);
  440. }
  441. else
  442. {
  443. TdsDateTime dt = MetaType.FromDateTime(value, 8);
  444. _stateObj.Parser.WriteSqlVariantHeader(10, TdsEnums.SQLDATETIME, 0, _stateObj);
  445. _stateObj.Parser.WriteInt(dt.days, _stateObj);
  446. _stateObj.Parser.WriteInt(dt.time, _stateObj);
  447. }
  448. // Clean the variant metadata to prevent sharing it with next row.
  449. // As a reminder, SetVariantType raises an assert if _variantType is not clean
  450. _variantType = null;
  451. }
  452. else {
  453. _stateObj.WriteByte((byte)_metaData.MaxLength);
  454. if (SqlDbType.SmallDateTime == _metaData.SqlDbType) {
  455. TdsDateTime dt = MetaType.FromDateTime(value, (byte)_metaData.MaxLength);
  456. Debug.Assert (0 <= dt.days && dt.days <= UInt16.MaxValue, "Invalid DateTime '" + value + "' for SmallDateTime");
  457. _stateObj.Parser.WriteShort(dt.days, _stateObj);
  458. _stateObj.Parser.WriteShort(dt.time, _stateObj);
  459. } else if (SqlDbType.DateTime == _metaData.SqlDbType) {
  460. TdsDateTime dt = MetaType.FromDateTime(value, (byte)_metaData.MaxLength);
  461. _stateObj.Parser.WriteInt(dt.days, _stateObj);
  462. _stateObj.Parser.WriteInt(dt.time, _stateObj);
  463. } else { // date and datetime2
  464. int days = value.Subtract(DateTime.MinValue).Days;
  465. if (SqlDbType.DateTime2 == _metaData.SqlDbType) {
  466. Int64 time = value.TimeOfDay.Ticks / TdsEnums.TICKS_FROM_SCALE[_metaData.Scale];
  467. _stateObj.WriteByteArray(BitConverter.GetBytes(time), (int)_metaData.MaxLength - 3, 0);
  468. }
  469. _stateObj.WriteByteArray(BitConverter.GetBytes(days), 3, 0);
  470. }
  471. }
  472. }
  473. // valid for UniqueIdentifier
  474. internal void SetGuid(Guid value) {
  475. Debug.Assert(
  476. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetGuid));
  477. byte[] bytes = value.ToByteArray();
  478. Debug.Assert(SmiMetaData.DefaultUniqueIdentifier.MaxLength == bytes.Length, "Invalid length for guid bytes: " + bytes.Length);
  479. if (SqlDbType.Variant == _metaData.SqlDbType) {
  480. _stateObj.Parser.WriteSqlVariantHeader(18, TdsEnums.SQLUNIQUEID, 0, _stateObj);
  481. }
  482. else {
  483. Debug.Assert(_metaData.MaxLength == bytes.Length, "Unexpected uniqueid metadata length: " + _metaData.MaxLength);
  484. _stateObj.WriteByte((byte)_metaData.MaxLength);
  485. }
  486. _stateObj.WriteByteArray(bytes, bytes.Length, 0);
  487. }
  488. // valid for SqlDbType.Time
  489. internal void SetTimeSpan(TimeSpan value) {
  490. Debug.Assert(
  491. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetTime));
  492. byte scale;
  493. byte length;
  494. if (SqlDbType.Variant == _metaData.SqlDbType) {
  495. scale = SmiMetaData.DefaultTime.Scale;
  496. length = (byte)SmiMetaData.DefaultTime.MaxLength;
  497. _stateObj.Parser.WriteSqlVariantHeader(8, TdsEnums.SQLTIME, 1, _stateObj);
  498. _stateObj.WriteByte(scale); //propbytes: scale
  499. } else {
  500. scale = _metaData.Scale;
  501. length = (byte)_metaData.MaxLength;
  502. _stateObj.WriteByte(length);
  503. }
  504. Int64 time = value.Ticks / TdsEnums.TICKS_FROM_SCALE[scale];
  505. _stateObj.WriteByteArray(BitConverter.GetBytes(time), length, 0);
  506. }
  507. // valid for DateTimeOffset
  508. internal void SetDateTimeOffset(DateTimeOffset value) {
  509. Debug.Assert(
  510. SmiXetterAccessMap.IsSetterAccessValid(_metaData, SmiXetterTypeCode.XetDateTimeOffset));
  511. byte scale;
  512. byte length;
  513. if (SqlDbType.Variant == _metaData.SqlDbType) {
  514. // VSTFDevDiv #885208 - DateTimeOffset throws ArgumentException for when passing DateTimeOffset value to a sql_variant TVP
  515. // using a SqlDataRecord or SqlDataReader
  516. MSS.SmiMetaData dateTimeOffsetMetaData = MSS.SmiMetaData.DefaultDateTimeOffset;
  517. scale = MetaType.MetaDateTimeOffset.Scale;
  518. length = (byte)dateTimeOffsetMetaData.MaxLength;
  519. _stateObj.Parser.WriteSqlVariantHeader(13, TdsEnums.SQLDATETIMEOFFSET, 1, _stateObj);
  520. _stateObj.WriteByte(scale); //propbytes: scale
  521. } else {
  522. scale = _metaData.Scale;
  523. length = (byte)_metaData.MaxLength;
  524. _stateObj.WriteByte(length);
  525. }
  526. DateTime utcDateTime = value.UtcDateTime;
  527. Int64 time = utcDateTime.TimeOfDay.Ticks / TdsEnums.TICKS_FROM_SCALE[scale];
  528. int days = utcDateTime.Subtract(DateTime.MinValue).Days;
  529. Int16 offset = (Int16)value.Offset.TotalMinutes;
  530. _stateObj.WriteByteArray(BitConverter.GetBytes(time), length - 5, 0); // time
  531. _stateObj.WriteByteArray(BitConverter.GetBytes(days), 3, 0); // date
  532. _stateObj.WriteByte((byte)(offset & 0xff)); // offset byte 1
  533. _stateObj.WriteByte((byte)((offset >> 8) & 0xff)); // offset byte 2
  534. }
  535. internal void SetVariantType(SmiMetaData value) {
  536. Debug.Assert(null == _variantType, "Variant type can only be set once");
  537. Debug.Assert(value != null &&
  538. (value.SqlDbType == SqlDbType.Money ||
  539. value.SqlDbType == SqlDbType.NVarChar ||
  540. value.SqlDbType == SqlDbType.Date ||
  541. value.SqlDbType == SqlDbType.DateTime ||
  542. value.SqlDbType == SqlDbType.DateTime2 ||
  543. value.SqlDbType == SqlDbType.DateTimeOffset ||
  544. value.SqlDbType == SqlDbType.SmallDateTime
  545. ), "Invalid variant type");
  546. _variantType = value;
  547. }
  548. #endregion
  549. #region private methods
  550. [Conditional("DEBUG")]
  551. private void CheckSettingOffset(long offset) {
  552. #if DEBUG
  553. Debug.Assert(offset == _currentOffset, "Invalid offset passed. Should be: " + _currentOffset + ", but was: " + offset);
  554. #endif
  555. }
  556. #endregion
  557. }
  558. }