Line data    Source code

       1             : /*
       2             :  * Purpose: Test we can cancel a query inside message handler
       3             :  * Some server some additional tokens which can prevent second query to work
       4             :  * correctly.
       5             :  */
       6             : 
       7             : #include "common.h"
       8             : 
       9             : static int
      10           8 : cancel_msg_handler(DBPROCESS * dbproc, DBINT msgno, int msgstate, int severity, char *msgtext, char *srvname, char *procname, int line)
      11             : {
      12           8 :         dbcancel(dbproc);
      13           8 :         return 0;
      14             : }
      15             : 
      16             : int
      17           8 : main(int argc, char **argv)
      18             : {
      19             :         LOGINREC *login;
      20             :         DBPROCESS *dbproc;
      21             :         int i;
      22           8 :         DBINT testint = -1;
      23             : 
      24           8 :         read_login_info(argc, argv);
      25             : 
      26           8 :         printf("Starting %s\n", argv[0]);
      27             : 
      28           8 :         dbinit();
      29             : 
      30           8 :         dberrhandle(syb_err_handler);
      31           8 :         dbmsghandle(syb_msg_handler);
      32             : 
      33           8 :         printf("About to logon\n");
      34             : 
      35           8 :         login = dblogin();
      36           8 :         DBSETLPWD(login, PASSWORD);
      37           8 :         DBSETLUSER(login, USER);
      38           8 :         DBSETLAPP(login, "cancel");
      39             : 
      40           8 :         printf("About to open\n");
      41             : 
      42           8 :         dbproc = dbopen(login, SERVER);
      43           8 :         if (strlen(DATABASE))
      44           8 :                 dbuse(dbproc, DATABASE);
      45           8 :         dbloginfree(login);
      46             : 
      47             :         /* first query, select with error  */
      48           8 :         dbmsghandle(cancel_msg_handler);
      49           8 :         sql_cmd(dbproc);
      50           8 :         dbsqlexec(dbproc);
      51             : 
      52           8 :         dbmsghandle(syb_msg_handler);
      53             : 
      54             :         /* second query, select */
      55           8 :         printf("second select\n");
      56             : 
      57           8 :         if (SUCCEED != sql_cmd(dbproc) || SUCCEED != dbsqlexec(dbproc)) {
      58           0 :                 fprintf(stderr, "%s:%d: dbcmd failed\n", __FILE__, __LINE__);
      59           0 :                 exit(1);
      60             :         }
      61             : 
      62           8 :         if (dbresults(dbproc) != SUCCEED) {
      63           0 :                 fprintf(stderr, "Was expecting a result set.");
      64           0 :                 exit(1);
      65             :         }
      66             : 
      67           8 :         for (i = 1; i <= dbnumcols(dbproc); i++)
      68           8 :                 printf("col %d is %s\n", i, dbcolname(dbproc, i));
      69             : 
      70           8 :         dbbind(dbproc, 1, INTBIND, 0, (BYTE *) & testint);
      71             : 
      72           8 :         if (REG_ROW != dbnextrow(dbproc)) {
      73           0 :                 fprintf(stderr, "Failed.  Expected a row\n");
      74           0 :                 exit(1);
      75             :         }
      76           8 :         if (testint != 1) {
      77           0 :                 fprintf(stderr, "Failed.  Expected i to be %d, was %d\n", i, (int) testint);
      78           0 :                 exit(1);
      79             :         }
      80           8 :         if (dbnextrow(dbproc) != NO_MORE_ROWS) {
      81           0 :                 fprintf(stderr, "No other rows expected\n");
      82             :         }
      83             : 
      84           8 :         dbexit();
      85             : 
      86           8 :         printf("ok\n");
      87             :         return 0;
      88             : }