NCBI C++ ToolKit
connection_string_parse.c
Go to the documentation of this file.

Go to the SVN repository for this file.

1 #include "common.h"
2 #include "freetds/odbc.h"
3 
4 #include <freetds/test_assert.h>
5 
6 
7 #ifdef _WIN32
8 HINSTANCE hinstFreeTDS;
9 #endif
10 
11 static void
12 assert_equal_dstr(DSTR a, const char *b)
13 {
14  assert(b && strcmp(tds_dstr_cstr(&a), b)==0);
15 }
16 
17 static void
18 assert_equal_str(TDS_PARSED_PARAM param, const char *b)
19 {
20  /* printf("param %.*s b %s\n", (int) param.len, param.p, b); */
21  assert(b && strlen(b) == param.len && strncmp(param.p, b, param.len)==0);
22 }
23 
24 static int
25 simple_string(void)
26 {
27  TDSLOGIN *login;
28  TDS_ERRS errs = {0};
29  TDSLOCALE *locale;
30  TDS_PARSED_PARAM parsed_params[ODBC_PARAM_SIZE];
31  const char *connect_string = "DRIVER=libtdsodbc.so;SERVER=127.0.0.1;PORT=1337;UID=test_username;PWD=test_password;DATABASE=test_db;ClientCharset=UTF-8;";
32 
33  const char *connect_string_end = connect_string + strlen(connect_string);
34  login = tds_alloc_login(0);
35  if (!tds_set_language(login, "us_english"))
36  return 1;
37  locale = tds_alloc_locale();
38  login = tds_init_login(login, locale);
39 
40  odbc_errs_reset(&errs);
41 
42  if (!odbc_parse_connect_string(&errs, connect_string, connect_string_end, login, parsed_params))
43  return 1;
44 
45  assert_equal_str(parsed_params[ODBC_PARAM_UID], "test_username");
46  assert_equal_dstr(login->server_name, "127.0.0.1");
47  assert_equal_dstr(login->password, "test_password");
48  assert_equal_str(parsed_params[ODBC_PARAM_PWD], "test_password");
49  assert(login->port == 1337);
50 
51  tds_free_login(login);
52  tds_free_locale(locale);
53 
54  return 0;
55 }
56 
57 static int
58 simple_escaped_string(void)
59 {
60  TDSLOGIN *login;
61  TDS_ERRS errs = {0};
62  TDSLOCALE *locale;
63  TDS_PARSED_PARAM parsed_params[ODBC_PARAM_SIZE];
64  const char *connect_string = "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={test_password};DATABASE={test_db};ClientCharset={UTF-8};";
65 
66  const char *connect_string_end = connect_string + strlen(connect_string);
67  login = tds_alloc_login(0);
68  if (!tds_set_language(login, "us_english"))
69  return 1;
70  locale = tds_alloc_locale();
71  login = tds_init_login(login, locale);
72 
73  odbc_errs_reset(&errs);
74 
75  if (!odbc_parse_connect_string(&errs, connect_string, connect_string_end, login, parsed_params))
76  return 1;
77 
78  assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
79  assert_equal_dstr(login->server_name, "127.0.0.1");
80  assert_equal_dstr(login->password, "test_password");
81  assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{test_password}");
82  assert(login->port == 1337);
83 
84  tds_free_login(login);
85  tds_free_locale(locale);
86 
87  return 0;
88 }
89 
90 static int
91 test_special_symbols(void)
92 {
93  TDSLOGIN *login;
94  TDS_ERRS errs = {0};
95  TDSLOCALE *locale;
96  TDS_PARSED_PARAM parsed_params[ODBC_PARAM_SIZE];
97  const char *connect_string = "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={[]{}}(),;?*=!@};DATABASE={test_db};ClientCharset={UTF-8};";
98 
99  const char *connect_string_end = connect_string + strlen(connect_string);
100  login = tds_alloc_login(0);
101  if (!tds_set_language(login, "us_english"))
102  return 1;
103  locale = tds_alloc_locale();
104  login = tds_init_login(login, locale);
105 
106  odbc_errs_reset(&errs);
107 
108  if (!odbc_parse_connect_string(&errs, connect_string, connect_string_end, login, parsed_params))
109  return 1;
110 
111  assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
112  assert_equal_dstr(login->server_name, "127.0.0.1");
113  assert_equal_dstr(login->password, "[]{}(),;?*=!@");
114  assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{[]{}}(),;?*=!@}");
115  assert(login->port == 1337);
116 
117  tds_free_login(login);
118  tds_free_locale(locale);
119 
120  return 0;
121 }
122 
123 static int
124 password_contains_curly_braces(void)
125 {
126  TDSLOGIN *login;
127  TDS_ERRS errs = {0};
128  TDSLOCALE *locale;
129  TDS_PARSED_PARAM parsed_params[ODBC_PARAM_SIZE];
130  const char *connect_string = "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={test{}}_password};DATABASE={test_db};ClientCharset={UTF-8};";
131 
132  const char *connect_string_end = connect_string + strlen(connect_string);
133  login = tds_alloc_login(0);
134  if (!tds_set_language(login, "us_english"))
135  return 1;
136  locale = tds_alloc_locale();
137  login = tds_init_login(login, locale);
138 
139  odbc_errs_reset(&errs);
140 
141  if (!odbc_parse_connect_string(&errs, connect_string, connect_string_end, login, parsed_params))
142  return 1;
143 
144  assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
145  assert_equal_dstr(login->server_name, "127.0.0.1");
146  assert_equal_dstr(login->password, "test{}_password");
147  assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{test{}}_password}");
148  assert(login->port == 1337);
149 
150  tds_free_login(login);
151  tds_free_locale(locale);
152 
153  return 0;
154 }
155 
156 static int
157 password_contains_curly_braces_and_separator(void)
158 {
159  TDSLOGIN *login;
160  TDS_ERRS errs = {0};
161  TDSLOCALE *locale;
162  TDS_PARSED_PARAM parsed_params[ODBC_PARAM_SIZE];
163  const char *connect_string = "DRIVER={libtdsodbc.so};SERVER={127.0.0.1};PORT={1337};UID={test_username};PWD={test{}};_password};DATABASE={test_db};ClientCharset={UTF-8};";
164 
165  const char *connect_string_end = connect_string + strlen(connect_string);
166  login = tds_alloc_login(0);
167  if (!tds_set_language(login, "us_english"))
168  return 1;
169  locale = tds_alloc_locale();
170  login = tds_init_login(login, locale);
171 
172  odbc_errs_reset(&errs);
173 
174  if (!odbc_parse_connect_string(&errs, connect_string, connect_string_end, login, parsed_params))
175  return 1;
176 
177  assert_equal_str(parsed_params[ODBC_PARAM_UID], "{test_username}");
178  assert_equal_dstr(login->server_name, "127.0.0.1");
179  assert_equal_dstr(login->password, "test{};_password");
180  assert_equal_str(parsed_params[ODBC_PARAM_PWD], "{test{}};_password}");
181  assert(login->port == 1337);
182 
183  tds_free_login(login);
184  tds_free_locale(locale);
185 
186  return 0;
187 }
188 
189 
190 int main(int argc, char *argv[])
191 {
192 #ifdef _WIN32
193  hinstFreeTDS = GetModuleHandle(NULL);
194 #endif
195 
196  if (simple_string() != 0)
197  return 1;
198 
199  if (simple_escaped_string() != 0)
200  return 1;
201 
202  if (password_contains_curly_braces() != 0)
203  return 1;
204 
205  if (test_special_symbols() != 0)
206  return 1;
207 
208  if (password_contains_curly_braces_and_separator() != 0)
209  return 1;
210 
211  return 0;
212 }
main
int main(int argc, char *argv[])
Definition: connection_string_parse.c:190
simple_escaped_string
static int simple_escaped_string(void)
Definition: connection_string_parse.c:58
assert_equal_str
static void assert_equal_str(TDS_PARSED_PARAM param, const char *b)
Definition: connection_string_parse.c:18
assert_equal_dstr
static void assert_equal_dstr(DSTR a, const char *b)
Definition: connection_string_parse.c:12
simple_string
static int simple_string(void)
Definition: connection_string_parse.c:25
test_special_symbols
static int test_special_symbols(void)
Definition: connection_string_parse.c:91
password_contains_curly_braces
static int password_contains_curly_braces(void)
Definition: connection_string_parse.c:124
password_contains_curly_braces_and_separator
static int password_contains_curly_braces_and_separator(void)
Definition: connection_string_parse.c:157
test_assert.h
odbc_errs_reset
void odbc_errs_reset(struct _sql_errors *errs)
reset errors
Definition: error.c:343
odbc_parse_connect_string
int odbc_parse_connect_string(TDS_ERRS *errs, const char *connect_string, const char *connect_string_end, TDSLOGIN *login, TDS_PARSED_PARAM *parsed_params)
Parses a connection string for SQLDriverConnect().
Definition: connectparams.c:275
ODBC_PARAM_PWD
@ ODBC_PARAM_PWD
Definition: odbc.h:528
ODBC_PARAM_SIZE
@ ODBC_PARAM_SIZE
Definition: odbc.h:529
ODBC_PARAM_UID
@ ODBC_PARAM_UID
Definition: odbc.h:528
hinstFreeTDS
HINSTANCE hinstFreeTDS
Definition: initnet.c:10
login
static TDSLOGIN * login
Definition: dataread.c:31
tds_free_login
#define tds_free_login
Definition: rename_ftds_tds.h:150
tds_init_login
#define tds_init_login
Definition: rename_ftds_tds.h:197
tds_alloc_login
#define tds_alloc_login
Definition: rename_ftds_tds.h:58
tds_set_language
#define tds_set_language
Definition: rename_ftds_tds.h:278
tds_free_locale
#define tds_free_locale
Definition: rename_ftds_tds.h:149
NULL
#define NULL
Definition: ncbistd.hpp:225
tds_dstr_cstr
static const char * tds_dstr_cstr(DSTR *s)
Returns a C version (NUL terminated string) of dstr.
Definition: string.h:66
tds_alloc_locale
#define tds_alloc_locale
Definition: rename_ftds_tds.h:57
util::strncmp
int strncmp(const char *str1, const char *str2, size_t count)
Definition: odbc_utils.hpp:133
util::strcmp
int strcmp(const char *str1, const char *str2)
Definition: odbc_utils.hpp:160
a
unsigned int a
Definition: ncbi_localip.c:102
locale
static char * locale
Definition: pcregrep.c:149
HINSTANCE
void * HINSTANCE
Definition: sqltypes.h:116
assert
#define assert(x)
Definition: srv_diag.hpp:58
TDS_PARSED_PARAM
Definition: odbc.h:538
TDS_PARSED_PARAM::len
size_t len
Definition: odbc.h:540
TDS_PARSED_PARAM::p
const char * p
Definition: odbc.h:539
_sql_errors
Definition: odbc.h:107
tds_dstr
Structure to hold a string.
Definition: tds.h:116
tds_locale
Definition: tds.h:647
tds_login
Definition: tds.h:584
tds_login::password
DSTR password
password of account login
Definition: tds.h:602
tds_login::server_name
DSTR server_name
server name (in freetds.conf)
Definition: tds.h:585
tds_login::port
int port
port of database service
Definition: tds.h:586
Generated by doxygen 1.9.0
Modified on Thu May 02 14:33:44 2024 by modify_doxy.py rev. 669887