tkt-b72787b1.test 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. # 2011 February 21
  2. #
  3. # The author disclaims copyright to this source code. In place of
  4. # a legal notice, here is a blessing:
  5. #
  6. # May you do good and not evil.
  7. # May you find forgiveness for yourself and forgive others.
  8. # May you share freely, never taking more than you give.
  9. #
  10. #***********************************************************************
  11. # This file implements regression tests for SQLite library.
  12. #
  13. # This file implements tests to verify that ticket [b72787b1a7] has been
  14. # fixed. From the ticket:
  15. #
  16. # The sqlite3ExpirePreparedStatements routine marks all statements
  17. # as expired. This includes statements that are not expired.
  18. #
  19. # Steps to reproduce:
  20. #
  21. # * Prepare a statement (A)
  22. # * Alter the schema to invalidate cookie in A
  23. # * Prepare a statement (B)
  24. # * Run B and have A run as part of B
  25. # * A will find a bad cookie and cause *all* statements
  26. # to be expired including the currently running B by calling
  27. # sqlite3ExpirePreparedStatements
  28. # * When control returns to B it will then abort
  29. #
  30. # The bug is that sqlite3ExpirePreparedStatements expires all statements.
  31. # Note that B was prepared after the schema change and hence is perfectly
  32. # valid and then is marked as expired while running.
  33. #
  34. set testdir [file dirname $argv0]
  35. source $testdir/tester.tcl
  36. ifcapable !compound {
  37. finish_test
  38. return
  39. }
  40. unset -nocomplain ::STMT
  41. proc runsql {} {
  42. db eval {CREATE TABLE IF NOT EXISTS t4(q)}
  43. sqlite3_step $::STMT
  44. set rc [sqlite3_column_int $::STMT 0]
  45. sqlite3_reset $::STMT
  46. return $rc
  47. }
  48. do_test tkt-b72787b1.1 {
  49. db eval {
  50. CREATE TABLE t1(x);
  51. INSERT INTO t1 VALUES(1);
  52. INSERT INTO t1 VALUES(2);
  53. CREATE TABLE t2(y);
  54. INSERT INTO t2 SELECT x+2 FROM t1;
  55. INSERT INTO t2 SELECT x+4 FROM t1;
  56. }
  57. db func runsql ::runsql
  58. set DB [sqlite3_connection_pointer db]
  59. set sql {SELECT max(x) FROM t1}
  60. set ::STMT [sqlite3_prepare_v2 $DB $sql -1 TAIL]
  61. # The runsql() call on the second row of the first query will
  62. # cause all $::STMT to hit an expired cookie. Prior to the fix
  63. # for [b72787b1a7, the bad cookie would expire all statements, including
  64. # the following compound SELECT, which would cause a fault when the
  65. # second SELECT was reached. After the fix, the current statement
  66. # continues to completion.
  67. db eval {
  68. SELECT CASE WHEN y=3 THEN y+100 WHEN y==4 THEN runsql()+200
  69. ELSE 300+y END FROM t2
  70. UNION ALL
  71. SELECT * FROM t1;
  72. }
  73. } {103 202 305 306 1 2}
  74. sqlite3_finalize $::STMT
  75. finish_test