Line data Source code
1 : #include "common.h"
2 : #include <assert.h>
3 : #include "freetds/odbc.h"
4 :
5 :
6 : static void
7 120 : assert_equal_dstr(DSTR a, const char *b)
8 : {
9 240 : assert(b && strcmp(tds_dstr_cstr(&a), b)==0);
10 120 : }
11 :
12 : static void
13 120 : assert_equal_str(TDS_PARSED_PARAM param, const char *b)
14 : {
15 : /* printf("param %.*s b %s\n", (int) param.len, param.p, b); */
16 120 : assert(b && strlen(b) == param.len && strncmp(param.p, b, param.len)==0);
17 120 : }
18 :
19 : typedef void check_func_t(TDSLOGIN *login, TDS_PARSED_PARAM *parsed_params);
20 :
21 : static void
22 70 : test_common(const char *name, const char *connect_string, check_func_t *check_func)
23 : {
24 : TDSLOGIN *login;
25 70 : TDS_ERRS errs = {0};
26 : TDSLOCALE *locale;
27 : TDS_PARSED_PARAM parsed_params[ODBC_PARAM_SIZE];
28 :
29 70 : const char *connect_string_end = connect_string + strlen(connect_string);
30 70 : login = tds_alloc_login(false);
31 70 : if (!tds_set_language(login, "us_english")) {
32 0 : fprintf(stderr, "Error setting language in test '%s'\n", name);
33 0 : exit(1);
34 : }
35 70 : locale = tds_alloc_locale();
36 70 : login = tds_init_login(login, locale);
37 :
38 70 : odbc_errs_reset(&errs);
39 :
40 70 : if (!odbc_parse_connect_string(&errs, connect_string, connect_string_end, login, parsed_params)) {
41 10 : assert(errs.num_errors > 0);
42 10 : if (check_func) {
43 0 : fprintf(stderr, "Error parsing string in test '%s'\n", name);
44 0 : exit(1);
45 : }
46 : } else {
47 60 : assert(errs.num_errors == 0);
48 60 : assert(check_func != NULL);
49 60 : check_func(login, parsed_params);
50 : }
51 :
52 70 : odbc_errs_reset(&errs);
53 70 : tds_free_login(login);
54 70 : tds_free_locale(locale);
55 70 : }
56 :
57 : #define CHECK(name, s) \
58 : static const char *name ## _connect_string = s; \
59 : static void name ## _check(TDSLOGIN *login, TDS_PARSED_PARAM *parsed_params); \
60 : static void name(void) { \
61 : test_common(#name, name ## _connect_string, name ## _check); \
62 : } \
63 : static void name ## _check(TDSLOGIN *login, TDS_PARSED_PARAM *parsed_params)
64 :
65 : #define CHECK_ERROR(name, s) \
66 : static const char *name ## _connect_string = s; \
67 : static void name(void) { \
68 : test_common(#name, name ## _connect_string, NULL); \
69 : }
70 :
71 20 : CHECK(simple_string,
72 : "DRIVER=libtdsodbc.so;SERVER=127.0.0.1;PORT=1337;UID=test_username;PWD=test_password;DATABASE=test_db;"
73 : "ClientCharset=UTF-8;")
74 : {
75 10 : assert_equal_str(parsed_params[ODBC_PARAM_UID], "test_username");
76 10 : assert_equal_dstr(login->server_name, "127.0.0.1");
77 10 : assert_equal_dstr(login->password, "test_password");
78 10 : assert_equal_str(parsed_params[ODBC_PARAM_PWD], "test_password");
79 10 : assert(login->port == 1337);
80 10 : }
81 :
82 20 : CHECK(simple_escaped_string,
83 : "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={test_password};DATABASE={test_db};"
84 : "ClientCharset={UTF-8};")
85 : {
86 10 : assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
87 10 : assert_equal_dstr(login->server_name, "127.0.0.1");
88 10 : assert_equal_dstr(login->password, "test_password");
89 10 : assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{test_password}");
90 10 : assert(login->port == 1337);
91 10 : }
92 :
93 20 : CHECK(test_special_symbols,
94 : "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={[]{}}(),;?*=!@};DATABASE={test_db};"
95 : "ClientCharset={UTF-8};")
96 : {
97 10 : assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
98 10 : assert_equal_dstr(login->server_name, "127.0.0.1");
99 10 : assert_equal_dstr(login->password, "[]{}(),;?*=!@");
100 10 : assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{[]{}}(),;?*=!@}");
101 10 : assert(login->port == 1337);
102 10 : }
103 :
104 20 : CHECK(password_contains_curly_braces,
105 : "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={test{}}_password};DATABASE={test_db};"
106 : "ClientCharset={UTF-8};")
107 : {
108 10 : assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
109 10 : assert_equal_dstr(login->server_name, "127.0.0.1");
110 10 : assert_equal_dstr(login->password, "test{}_password");
111 10 : assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{test{}}_password}");
112 10 : assert(login->port == 1337);
113 10 : }
114 :
115 20 : CHECK(password_contains_curly_braces_and_separator,
116 : "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={test{}};_password};DATABASE={test_db};"
117 : "ClientCharset={UTF-8};")
118 : {
119 10 : assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
120 10 : assert_equal_dstr(login->server_name, "127.0.0.1");
121 10 : assert_equal_dstr(login->password, "test{};_password");
122 10 : assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{test{}};_password}");
123 10 : assert(login->port == 1337);
124 10 : }
125 :
126 20 : CHECK(password_bug_report,
127 : "Driver=FreeTDS;Server=1.2.3.4;Port=1433;Database=test;uid=test_user;pwd={p@ssw0rd}")
128 : {
129 10 : assert_equal_str(parsed_params[ODBC_PARAM_UID], "test_user");
130 10 : assert_equal_dstr(login->server_name, "1.2.3.4");
131 10 : assert_equal_dstr(login->password, "p@ssw0rd");
132 10 : assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{p@ssw0rd}");
133 10 : assert(login->port == 1433);
134 10 : }
135 :
136 : /* unfinished "pwd", the "Port" before "pwd" is to reveal a leak */
137 10 : CHECK_ERROR(unfinished,
138 : "Driver=FreeTDS;Server=1.2.3.4;Port=1433;pwd={p@ssw0rd");
139 :
140 : int
141 10 : main(void)
142 : {
143 : simple_string();
144 :
145 : simple_escaped_string();
146 :
147 : password_contains_curly_braces();
148 :
149 : test_special_symbols();
150 :
151 : password_contains_curly_braces_and_separator();
152 :
153 : password_bug_report();
154 :
155 : unfinished();
156 :
157 : return 0;
158 : }
|
