t0021.c

来自「unix上的TDS协议实现」· C语言 代码 · 共 79 行

/* 
** test for proper return code from dbsqlexec()
*/
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <stdio.h>
#ifdef _WIN32
#define DBNTWIN32
#include <windows.h>
#endif
#include <sqlfront.h>
#include <sqldb.h>

static char  software_version[]   = "$Id: t0021.c,v 1.1 2002/07/12 03:09:06 brianb Exp $";
static void *no_unused_var_warn[] = {software_version,
                                     no_unused_var_warn};



int failed = 0;
/* unsafestr must contain one quote of each type */
char *unsafestr = "This is a string with ' and \" in it.";
/* safestr must be at least strlen(unsafestr) + 3 */
char *safestr[100];

int main()
{
LOGINREC   *login;
DBPROCESS   *dbproc;
int         i, len;
RETCODE ret;

#ifdef __FreeBSD__
   /*
    * Options for malloc   A- all warnings are fatal, J- init memory to 0xD0,
    * R- always move memory block on a realloc.
    */
   extern char *malloc_options;
   malloc_options = "AJR";
#endif

   fprintf(stdout, "Start\n");

   /* Fortify_EnterScope(); */
   dbinit();


   len = strlen(unsafestr);
   ret = dbsafestr(NULL, unsafestr, -1, safestr, len, DBSINGLE);
   if (ret!=FAIL) failed++;
   fprintf(stdout, "short buffer, single\n%s\n", safestr);
   /* plus one for termination and one for the quote */
   ret = dbsafestr(NULL, unsafestr, -1, safestr, len+2, DBSINGLE);
   if (ret!=SUCCEED) failed++;
   if (strlen(safestr) != len+1) failed++;
   fprintf(stdout, "single quote\n%s\n", safestr);
   ret = dbsafestr(NULL, unsafestr, -1, safestr, len+2, DBDOUBLE);
   if (ret!=SUCCEED) failed++;
   if (strlen(safestr) != len+1) failed++;
   fprintf(stdout, "double quote\n%s\n", safestr);
   ret = dbsafestr(NULL, unsafestr, -1, safestr, len+3, DBBOTH);
   if (ret!=SUCCEED) failed++;
   if (strlen(safestr) != len+2) failed++;
   fprintf(stdout, "both quotes\n%s\n", safestr);

   dbexit();

   fprintf(stdout, "dblib %s on %s\n", 
           (failed?"failed!":"okay"),
           __FILE__);
   return failed ? 1 : 0; 
}