Line data Source code
1 : #include "common.h"
2 :
3 : /*
4 : * This test try do discovery how dblib process token looking for state
5 : * at every iteration. It issue a query to server and check
6 : * - row count (valid number means DONE processed)
7 : * - if possible to send another query (means state IDLE)
8 : * - if error readed (means ERROR token readed)
9 : * - if status present (PARAMRESULT token readed)
10 : * - if parameter present (PARAM token readed)
11 : * It try these query types:
12 : * - normal row
13 : * - normal row with no count
14 : * - normal row without rows
15 : * - error query
16 : * - store procedure call with output parameters
17 : */
18 :
19 : /* Forward declarations of the error handler and message handler. */
20 : static int err_handler(DBPROCESS * dbproc, int severity, int dberr, int oserr, char *dberrstr, char *oserrstr);
21 : static int msg_handler(DBPROCESS * dbproc, DBINT msgno, int msgstate, int severity, char *msgtext, char *srvname, char *procname,
22 : int line);
23 :
24 : static DBPROCESS *dbproc;
25 : static int silent = 0;
26 : static int check_idle = 0;
27 :
28 : /* print functions adapted from src/dblib/dblib.c */
29 : static const char *
30 254 : prdbretcode(int retcode)
31 : {
32 : static char unknown[24];
33 254 : switch(retcode) {
34 : case REG_ROW: return "REG_ROW/MORE_ROWS";
35 214 : case NO_MORE_ROWS: return "NO_MORE_ROWS";
36 0 : case BUF_FULL: return "BUF_FULL";
37 0 : case NO_MORE_RESULTS: return "NO_MORE_RESULTS";
38 0 : case SUCCEED: return "SUCCEED";
39 0 : case FAIL: return "FAIL";
40 0 : default:
41 0 : sprintf(unknown, "oops: %u ??", retcode);
42 : }
43 0 : return unknown;
44 : }
45 :
46 : static const char *
47 252 : prretcode(int retcode)
48 : {
49 : static char unknown[24];
50 252 : switch(retcode) {
51 : case SUCCEED: return "SUCCEED";
52 10 : case FAIL: return "FAIL";
53 138 : case NO_MORE_RESULTS: return "NO_MORE_RESULTS";
54 0 : default:
55 0 : sprintf(unknown, "oops: %u ??", retcode);
56 : }
57 0 : return unknown;
58 : }
59 :
60 : static void
61 100 : query(const char comment[])
62 : {
63 100 : if (comment)
64 40 : printf("%s\n", comment);
65 100 : sql_cmd(dbproc);
66 100 : dbsqlexec(dbproc);
67 100 : while (dbresults(dbproc) == SUCCEED) {
68 : /* nop */
69 : }
70 100 : }
71 :
72 : typedef const char* (*prfunc)(int);
73 :
74 : static void
75 506 : check_state(const char name[], prfunc print, int erc)
76 : {
77 506 : printf("State %-15s %-20s ", name, print(erc));
78 506 : if (dbnumcols(dbproc) > 0)
79 376 : printf("COLS(%d) ", dbnumcols(dbproc));
80 : /* row count */
81 506 : if (dbcount(dbproc) >= 0)
82 88 : printf("ROWS(%d) ", (int) dbcount(dbproc));
83 506 : silent = 1;
84 506 : if (dbdata(dbproc, 1))
85 130 : printf("DATA ");
86 506 : silent = 0;
87 : /* if status present */
88 506 : if (dbretstatus(dbproc) == TRUE)
89 0 : printf("STATUS %d ", (int) dbretstatus(dbproc));
90 : /* if parameter present */
91 506 : if (dbnumrets(dbproc) > 0)
92 8 : printf("PARAMS ");
93 : /*
94 : * if possible to send another query
95 : * NOTE this must be the last
96 : */
97 506 : if (check_idle) {
98 60 : silent = 1;
99 60 : dbcmd(dbproc, "declare @i int ");
100 60 : if (FAIL != dbsqlexec(dbproc))
101 20 : printf("IDLE ");
102 60 : silent = 0;
103 : }
104 506 : printf("\n");
105 506 : }
106 :
107 : static void
108 60 : do_test(const char comment[])
109 : {
110 : int ret;
111 60 : prfunc print_with = NULL;
112 :
113 60 : if (comment)
114 60 : printf("%s\n", comment);
115 60 : sql_cmd(dbproc);
116 :
117 60 : check_state("sqlexec ", prretcode, dbsqlexec(dbproc));
118 :
119 60 : check_state("nextrow ", prdbretcode, dbnextrow(dbproc));
120 60 : check_state("nextrow ", prdbretcode, dbnextrow(dbproc));
121 60 : check_state("results ", prretcode, dbresults(dbproc));
122 60 : check_state("nextrow ", prdbretcode, dbnextrow(dbproc));
123 60 : check_state("nextrow ", prdbretcode, dbnextrow(dbproc));
124 :
125 60 : check_idle = 0;
126 : for (;;) {
127 72 : ret = dbresults(dbproc);
128 72 : check_state("results ", prretcode, ret);
129 72 : if (ret != SUCCEED) {
130 60 : print_with = prretcode;
131 : break;
132 : }
133 :
134 : do {
135 14 : ret = dbnextrow(dbproc);
136 14 : check_state("nextrow ", prdbretcode, ret);
137 14 : } while (ret == REG_ROW);
138 : print_with = prdbretcode;
139 : }
140 60 : check_state("more results?", print_with, ret);
141 60 : }
142 :
143 : int
144 10 : main(int argc, char *argv[])
145 : {
146 : static const int invalid_column_name = 207;
147 : LOGINREC *login; /* Our login information. */
148 : int i;
149 :
150 10 : setbuf(stdout, NULL);
151 10 : read_login_info(argc, argv);
152 :
153 10 : if (dbinit() == FAIL)
154 0 : exit(1);
155 :
156 10 : dberrhandle(err_handler);
157 10 : dbmsghandle(msg_handler);
158 :
159 : #if 0
160 : /*
161 : * FIXME: Should be able to use the common err/msg handlers, but something about
162 : * the IDLE checking causes them to fail. Not sure about purpose of IDLE checking.
163 : * -- jkl January 2009
164 : */
165 : dberrhandle(syb_err_handler);
166 : dbmsghandle(syb_msg_handler);
167 : #endif
168 :
169 10 : login = dblogin();
170 10 : DBSETLUSER(login, USER);
171 10 : DBSETLPWD(login, PASSWORD);
172 10 : DBSETLAPP(login, "done_handling");
173 :
174 10 : dbproc = dbopen(login, SERVER);
175 10 : dbloginfree(login);
176 10 : if (!dbproc)
177 0 : exit(1);
178 10 : if (strlen(DATABASE))
179 10 : dbuse(dbproc, DATABASE);
180 :
181 60 : for (i=0; i < 6; i++)
182 60 : query(NULL);
183 : #if 0
184 : check_state("setup done ", prretcode, erc);
185 :
186 : printf("wasting results\n");
187 : while ((erc = dbresults(dbproc)) == SUCCEED) {
188 : while (dbnextrow(dbproc) == REG_ROW); /* no-op */
189 : }
190 : #endif
191 10 : check_idle = 1;
192 :
193 10 : do_test("normal row with rowcount on");
194 10 : query("turn rowcount off");
195 10 : do_test("normal row with rowcount off");
196 10 : query("turn rowcount back on");
197 10 : do_test("normal row without rows");
198 10 : dbsetuserdata(dbproc, (BYTE*) &invalid_column_name);
199 10 : do_test("error query");
200 10 : do_test("stored procedure call with output parameters");
201 :
202 10 : do_test("execute done2");
203 :
204 10 : query("drop done_test");
205 :
206 10 : query("drop done_test2");
207 :
208 10 : dbexit();
209 : return 0;
210 : }
211 :
212 : static int
213 50 : err_handler(DBPROCESS * dbproc TDS_UNUSED, int severity, int dberr TDS_UNUSED, int oserr, char *dberrstr, char *oserrstr)
214 : {
215 50 : if (silent)
216 : return INT_CANCEL;
217 :
218 10 : fflush(stdout);
219 10 : fprintf(stderr, "DB-Library error (severity %d):\n\t%s\n", severity, dberrstr);
220 :
221 10 : if (oserr != DBNOERR)
222 0 : fprintf(stderr, "Operating-system error:\n\t%s\n", oserrstr ? oserrstr : "(null)");
223 10 : fflush(stderr);
224 :
225 10 : return INT_CANCEL;
226 : }
227 :
228 : static int
229 38 : msg_handler(DBPROCESS * dbproc TDS_UNUSED, DBINT msgno, int msgstate, int severity,
230 : char *msgtext, char *srvname, char *procname, int line)
231 : {
232 38 : if (silent)
233 : return 0;
234 :
235 38 : fflush(stdout);
236 38 : fprintf(stderr, "Msg %d, Level %d, State %d\n", (int) msgno, severity, msgstate);
237 :
238 38 : if (strlen(srvname) > 0)
239 38 : fprintf(stderr, "Server '%s', ", srvname);
240 38 : if (procname && strlen(procname) > 0) {
241 0 : fprintf(stderr, "Procedure '%s', ", procname);
242 0 : if (line > 0)
243 0 : fprintf(stderr, "Line %d", line);
244 : }
245 :
246 38 : fprintf(stderr, "\n\t%s\n", msgtext);
247 38 : fflush(stderr);
248 :
249 38 : return 0;
250 : }
|
