Line data Source code
1 : /*
2 : * Test reading data with SQLBindCol
3 : */
4 : #include "common.h"
5 : #include <assert.h>
6 : #include <ctype.h>
7 : #include "parser.h"
8 :
9 : /*
10 : * This test is useful to test odbc_tds2sql function
11 : * odbc_tds2sql have some particular cases:
12 : * (1) numeric -> binary numeric is different in ODBC
13 : * (2) * -> binary dependent from libTDS representation and ODBC one
14 : * (3) binary -> char TODO
15 : * (4) date -> char different format
16 : * Also we have to check normal char and wide char
17 : */
18 :
19 : static int result = 0;
20 :
21 : static int ignore_select_error = 0;
22 : static int ignore_result = 0;
23 :
24 : static void
25 730 : Test(const char *type, const char *value_to_convert, SQLSMALLINT out_c_type, const char *expected)
26 : {
27 : char sbuf[1024];
28 : unsigned char out_buf[256];
29 730 : SQLLEN out_len = 0;
30 :
31 730 : SQLFreeStmt(odbc_stmt, SQL_UNBIND);
32 730 : SQLFreeStmt(odbc_stmt, SQL_RESET_PARAMS);
33 :
34 : /* execute a select to get data as wire */
35 730 : sprintf(sbuf, "SELECT CONVERT(%s, '%s') AS data", type, value_to_convert);
36 730 : if (strncmp(value_to_convert, "0x", 2) == 0) {
37 28 : if (odbc_db_is_microsoft())
38 24 : sprintf(sbuf, "SELECT CONVERT(%s, %s) COLLATE Latin1_General_CI_AS AS data", type, value_to_convert);
39 : else
40 4 : sprintf(sbuf, "SELECT CONVERT(%s, %s)", type, value_to_convert);
41 702 : } else if (strcmp(type, "SQL_VARIANT") == 0)
42 92 : sprintf(sbuf, "SELECT CONVERT(SQL_VARIANT, %s) AS data", value_to_convert);
43 610 : else if (strncmp(value_to_convert, "u&'", 3) == 0)
44 10 : sprintf(sbuf, "SELECT CONVERT(%s, %s) AS data", type, value_to_convert);
45 730 : if (ignore_select_error) {
46 730 : if (odbc_command2(sbuf, "SENo") == SQL_ERROR) {
47 20 : odbc_reset_statement();
48 20 : ignore_select_error = 0;
49 20 : ignore_result = 0;
50 20 : return;
51 : }
52 : } else {
53 0 : odbc_command(sbuf);
54 : }
55 710 : SQLBindCol(odbc_stmt, 1, out_c_type, out_buf, sizeof(out_buf), &out_len);
56 710 : CHKFetch("S");
57 710 : CHKFetch("No");
58 710 : CHKMoreResults("No");
59 :
60 : /* test results */
61 710 : odbc_c2string(sbuf, out_c_type, out_buf, out_len);
62 :
63 710 : if (!ignore_result && strcmp(sbuf, expected) != 0) {
64 0 : fprintf(stderr, "Wrong result\n Got: %s\n Expected: %s\n", sbuf, expected);
65 0 : result = 1;
66 : }
67 710 : ignore_select_error = 0;
68 710 : ignore_result = 0;
69 : }
70 :
71 : static int
72 0 : get_int(const char *s, odbc_parser *parser)
73 : {
74 : char *end;
75 : long l;
76 :
77 0 : if (!s)
78 0 : odbc_fatal(parser, ": NULL int\n");
79 0 : l = strtol(s, &end, 0);
80 0 : if (end[0])
81 0 : odbc_fatal(parser, ": Invalid int\n");
82 0 : return (int) l;
83 : }
84 :
85 : static int
86 1360 : lookup(const char *name, const struct odbc_lookup_int *table, odbc_parser *parser)
87 : {
88 : int res;
89 :
90 1360 : if (!table)
91 0 : return get_int(name, parser);
92 :
93 1360 : res = odbc_lookup(name, table, SQL_UNKNOWN_TYPE);
94 :
95 1360 : return res == SQL_UNKNOWN_TYPE ? get_int(name, parser) : res;
96 : }
97 :
98 : int
99 10 : main(void)
100 : {
101 10 : bool cond = true;
102 :
103 : #define TEST_FILE "data.in"
104 10 : const char *in_file = FREETDS_SRCDIR "/" TEST_FILE;
105 : FILE *f;
106 : odbc_parser *parser;
107 :
108 10 : odbc_connect();
109 :
110 10 : f = fopen(in_file, "r");
111 10 : if (!f)
112 0 : f = fopen(TEST_FILE, "r");
113 10 : if (!f) {
114 0 : fprintf(stderr, "error opening test file\n");
115 0 : exit(1);
116 : }
117 :
118 : /* cache version */
119 10 : odbc_tds_version();
120 :
121 10 : parser = odbc_init_parser(f);
122 1460 : for (;;) {
123 : char *p;
124 1470 : const char *cmd = odbc_get_cmd_line(parser, &p, &cond);
125 :
126 1470 : if (!cmd)
127 : break;
128 :
129 : /* select type */
130 1460 : if (!strcmp(cmd, "select")) {
131 1320 : const char *type = odbc_get_str(parser, &p);
132 1320 : const char *value = odbc_get_str(parser, &p);
133 1320 : int c_type = lookup(odbc_get_tok(&p), odbc_sql_c_types, parser);
134 1320 : const char *expected = odbc_get_str(parser, &p);
135 :
136 2780 : if (!cond) continue;
137 :
138 698 : ignore_select_error = 1;
139 698 : Test(type, value, c_type, expected);
140 698 : continue;
141 : }
142 : /* select type setting condition */
143 140 : if (!strcmp(cmd, "select_cond")) {
144 40 : const char *bool_name = odbc_get_tok(&p);
145 40 : const char *type = odbc_get_tok(&p);
146 40 : const char *value = odbc_get_str(parser, &p);
147 40 : int c_type = lookup(odbc_get_tok(&p), odbc_sql_c_types, parser);
148 40 : const char *expected = odbc_get_str(parser, &p);
149 40 : int save_result = result;
150 :
151 40 : if (!bool_name) odbc_fatal(parser, ": no condition name\n");
152 40 : if (!cond) continue;
153 :
154 32 : ignore_select_error = 1;
155 32 : ignore_result = 1;
156 32 : result = 0;
157 32 : Test(type, value, c_type, expected);
158 32 : odbc_set_bool(parser, bool_name, result == 0);
159 32 : result = save_result;
160 32 : continue;
161 : }
162 : /* execute a sql command */
163 100 : if (!strcmp(cmd, "sql")) {
164 30 : const char *sql = odbc_get_str(parser, &p);
165 :
166 30 : if (!cond) continue;
167 :
168 18 : odbc_command(sql);
169 18 : continue;
170 : }
171 70 : if (!strcmp(cmd, "sql_cond")) {
172 70 : const char *bool_name = odbc_get_tok(&p);
173 70 : const char *sql = odbc_get_str(parser, &p);
174 :
175 70 : if (!cond) continue;
176 :
177 54 : odbc_set_bool(parser, bool_name, odbc_command2(sql, "SENo") != SQL_ERROR);
178 54 : continue;
179 : }
180 0 : odbc_fatal(parser, ": unknown command\n");
181 : }
182 10 : odbc_free_parser(parser);
183 10 : fclose(f);
184 :
185 10 : printf("\n");
186 :
187 10 : odbc_disconnect();
188 :
189 10 : if (!result)
190 10 : printf("Done successfully!\n");
191 10 : return result;
192 : }
|
