size_t CDBL_SendDataCmd::SendChunk(const void* pChunk, size_t nof_bytes)
{
CHECK_DRIVER_ERROR(
!pChunk || !nof_bytes,
"Wrong (zero) arguments." + GetDbgInfo(),
290000 );
if (!GetBytes2Go())
return 0;
if (nof_bytes > GetBytes2Go())
nof_bytes = GetBytes2Go();
if (Check(dbmoretext(GetCmd(), (DBINT) nof_bytes, (BYTE*) pChunk)) != SUCCEED) {
Check(dbcancel(GetCmd()));
DATABASE_DRIVER_ERROR( "dbmoretext failed." + GetDbgInfo(), 290001 );
}
SetBytes2Go(GetBytes2Go() - nof_bytes);
if (GetBytes2Go() <= 0) {
// if (dbsqlok(m_Cmd) != SUCCEED || dbresults(m_Cmd) == FAIL) {
if (Check(dbsqlok(GetCmd())) != SUCCEED || GetConnection().x_Results(GetCmd()) == FAIL) {
DATABASE_DRIVER_ERROR( "dbsqlok/results failed." + GetDbgInfo(), 290002 );
}
}
return nof_bytes;
}
VALUE rb_tinytds_raise_error(DBPROCESS *dbproc, int is_message, int cancel, const char *error, const char *source, int severity, int dberr, int oserr) {
VALUE e;
GET_CLIENT_USERDATA(dbproc);
if (cancel && !dbdead(dbproc) && userdata && !userdata->closed) {
userdata->dbsqlok_sent = 1;
dbsqlok(dbproc);
userdata->dbcancel_sent = 1;
dbcancel(dbproc);
}
e = rb_exc_new2(cTinyTdsError, error);
rb_funcall(e, intern_source_eql, 1, rb_str_new2(source));
if (severity)
rb_funcall(e, intern_severity_eql, 1, INT2FIX(severity));
if (dberr)
rb_funcall(e, intern_db_error_number_eql, 1, INT2FIX(dberr));
if (oserr)
rb_funcall(e, intern_os_error_number_eql, 1, INT2FIX(oserr));
if (severity <= 10 && is_message) {
VALUE message_handler = userdata && userdata->message_handler ? userdata->message_handler : Qnil;
if (message_handler && message_handler != Qnil && rb_respond_to(message_handler, intern_call) != 0) {
rb_funcall(message_handler, intern_call, 1, e);
}
return Qnil;
}
rb_exc_raise(e);
return Qnil;
}
CDB_SendDataCmd* CDBL_Connection::SendDataCmd(I_BlobDescriptor& descr_in,
size_t data_size,
bool log_it,
bool /*dump_results*/)
{
x_SetExtraMsg(descr_in, data_size);
CHECK_DRIVER_ERROR( data_size < 1, "Wrong (zero) data size." + GetDbgInfo(), 210092 );
I_BlobDescriptor* p_desc= 0;
// check what type of descriptor we've got
if(descr_in.DescriptorType() != CDBL_BLOB_DESCRIPTOR_TYPE_MAGNUM) {
// this is not a native descriptor
p_desc = x_GetNativeBlobDescriptor
(dynamic_cast<CDB_BlobDescriptor&>(descr_in));
if (p_desc == NULL) {
return NULL;
} else {
x_SetExtraMsg(*p_desc, data_size);
}
}
unique_ptr<I_BlobDescriptor> d_guard(p_desc);
CDBL_BlobDescriptor& desc
= p_desc ? dynamic_cast<CDBL_BlobDescriptor&>(*p_desc) :
dynamic_cast<CDBL_BlobDescriptor&>(descr_in);
if (Check(dbwritetext(GetDBLibConnection(),
(char*) desc.m_ObjName.c_str(),
desc.m_TxtPtr_is_NULL ? 0 : desc.m_TxtPtr,
DBTXPLEN,
desc.m_TimeStamp_is_NULL ? 0 : desc.m_TimeStamp,
log_it ? TRUE : FALSE,
(DBINT) data_size, 0)) != SUCCEED ||
Check(dbsqlok(GetDBLibConnection())) != SUCCEED ||
// dbresults(GetDBLibConnection()) == FAIL) {
x_Results(GetDBLibConnection()) == FAIL) {
DATABASE_DRIVER_ERROR( "dbwritetext/dbsqlok/dbresults failed." + GetDbgInfo(), 210093 );
}
CDBL_SendDataCmd* sd_cmd = new CDBL_SendDataCmd(*this, GetDBLibConnection(), data_size);
return Create_SendDataCmd(*sd_cmd);
}
static VALUE rb_tinytds_raise_error(DBPROCESS *dbproc, int cancel, char *error, char *source, int severity, int dberr, int oserr) {
GET_CLIENT_USERDATA(dbproc);
if (cancel && !dbdead(dbproc) && userdata && !userdata->closed) {
userdata->dbsqlok_sent = 1;
dbsqlok(dbproc);
userdata->dbcancel_sent = 1;
dbcancel(dbproc);
}
VALUE e = rb_exc_new2(cTinyTdsError, error);
rb_funcall(e, intern_source_eql, 1, rb_str_new2(source));
if (severity)
rb_funcall(e, intern_severity_eql, 1, INT2FIX(severity));
if (dberr)
rb_funcall(e, intern_db_error_number_eql, 1, INT2FIX(dberr));
if (oserr)
rb_funcall(e, intern_os_error_number_eql, 1, INT2FIX(oserr));
rb_exc_raise(e);
return Qnil;
}
int
main(int argc, char **argv)
{
LOGINREC *login;
DBPROCESS *dbproc;
RETPARAM save_param, save_varchar_tds7_param, save_nvarchar_tds7_param;
char teststr[8000+1], abbrev_data[10+3+1], *output;
char *retname = NULL;
int i, failed = 0;
int rettype = 0, retlen = 0, return_status = 0;
char proc[] = "#t0022";
char *proc_name = proc;
int num_resultset = 0, num_empty_resultset = 0;
int num_params = 6;
static const char dashes30[] = "------------------------------";
static const char *dashes5 = dashes30 + (sizeof(dashes30) - 5),
*dashes20 = dashes30 + (sizeof(dashes30) - 20);
RETCODE erc, row_code;
set_malloc_options();
memset(&save_param, 0, sizeof(save_param));
memset(&save_varchar_tds7_param, 0, sizeof(save_varchar_tds7_param));
memset(&save_nvarchar_tds7_param, 0, sizeof(save_nvarchar_tds7_param));
read_login_info(argc, argv);
printf("Starting %s\n", argv[0]);
dbinit();
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
printf("About to logon\n");
login = dblogin();
DBSETLPWD(login, PASSWORD);
DBSETLUSER(login, USER);
DBSETLAPP(login, "rpc");
dberrhandle(ignore_err_handler);
DBSETLPACKET(login, -1);
dberrhandle(syb_err_handler);
printf("About to open %s.%s\n", SERVER, DATABASE);
dbproc = dbopen(login, SERVER);
if (strlen(DATABASE))
dbuse(dbproc, DATABASE);
dbloginfree(login);
printf("Check if server support long identifiers\n");
sql_cmd(dbproc);
i = 103;
dbsetuserdata(dbproc, (BYTE*) &i);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS)
while (dbnextrow(dbproc) != NO_MORE_ROWS)
continue;
dbsetuserdata(dbproc, NULL);
if (i == 0) {
fprintf(stderr, "This server does not support long identifiers\n");
dbexit();
return 0;
}
dberrhandle(ignore_err_handler);
dbmsghandle(ignore_msg_handler);
printf("trying to create a temporary stored procedure\n");
if (FAIL == init_proc(dbproc, proc_name)) {
num_params = 4;
printf("trying to create a permanent stored procedure\n");
if (FAIL == init_proc(dbproc, ++proc_name))
exit(EXIT_FAILURE);
}
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
printf("Created procedure %s\n", proc_name);
/* set up and send the rpc */
printf("executing dbrpcinit\n");
erc = dbrpcinit(dbproc, proc_name, 0); /* no options */
if (erc == FAIL) {
fprintf(stderr, "Failed line %d: dbrpcinit\n", __LINE__);
failed = 1;
}
for (pb = bindings, i = 0; pb < bindings + sizeof(bindings)/sizeof(bindings[0]); pb++, i++) {
printf("executing dbrpcparam for %s\n", pb->name);
if (num_params == 4 && (i == 3 || i == 4))
continue;
if ((erc = dbrpcparam(dbproc, pb->name, pb->status, pb->type, pb->maxlen, pb->datalen, pb->value)) == FAIL) {
fprintf(stderr, "Failed line %d: dbrpcparam\n", __LINE__);
failed++;
}
}
printf("executing dbrpcsend\n");
param_data5 = 0x11223344;
erc = dbrpcsend(dbproc);
if (erc == FAIL) {
fprintf(stderr, "Failed line %d: dbrpcsend\n", __LINE__);
exit(1);
}
/* wait for it to execute */
printf("executing dbsqlok\n");
erc = dbsqlok(dbproc);
if (erc == FAIL) {
fprintf(stderr, "Failed line %d: dbsqlok\n", __LINE__);
exit(1);
}
/* retrieve outputs per usual */
printf("fetching results\n");
while ((erc = dbresults(dbproc)) != NO_MORE_RESULTS) {
printf("fetched resultset %d %s:\n", 1+num_resultset, erc==SUCCEED? "successfully":"unsuccessfully");
if (erc == SUCCEED) {
const int ncol = dbnumcols(dbproc);
int empty_resultset = 1, c;
enum {buflen=1024, nbuf=5};
char bound_buffers[nbuf][buflen] = { "one", "two", "three", "four", "five" };
++num_resultset;
for( c=0; c < ncol && c < nbuf; c++ ) {
printf("column %d (%s) is %d wide, ", c+1, dbcolname(dbproc, c+1), colwidth(dbproc, c+1));
printf("buffer initialized to '%s'\n", bound_buffers[c]);
}
for( c=0; c < ncol && c < nbuf; c++ ) {
erc = dbbind(dbproc, c+1, STRINGBIND, 0, (BYTE *) bound_buffers[c]);
if (erc == FAIL) {
fprintf(stderr, "Failed line %d: dbbind\n", __LINE__);
exit(1);
}
printf("%-*s ", colwidth(dbproc, c+1), dbcolname(dbproc, c+1));
}
printf("\n");
while ((row_code = dbnextrow(dbproc)) != NO_MORE_ROWS) {
empty_resultset = 0;
if (row_code == REG_ROW) {
int c;
for( c=0; c < ncol && c < nbuf; c++ ) {
printf("%-*s ", colwidth(dbproc, c+1), bound_buffers[c]);
}
printf("\n");
} else {
/* not supporting computed rows in this unit test */
failed = 1;
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
}
printf("row count %d\n", (int) dbcount(dbproc));
printf("hasretstatus %d\n", dbhasretstat(dbproc));
if (num_resultset == 4 && !dbhasretstat(dbproc)) {
fprintf(stderr, "dbnextrow should have set hasretstatus after last recordset\n");
exit(1);
}
if (empty_resultset)
++num_empty_resultset;
} else {
fprintf(stderr, "Expected a result set.\n");
exit(1);
}
} /* while dbresults */
/* check return status */
printf("retrieving return status...\n");
if (dbhasretstat(dbproc) == TRUE) {
printf("%d\n", return_status = dbretstatus(dbproc));
} else {
printf("none\n");
}
/*
* Check output parameter values
*/
if (dbnumrets(dbproc) != num_params) { /* dbnumrets missed something */
fprintf(stderr, "Expected %d output parameters.\n", num_params);
exit(1);
}
printf("retrieving output parameters...\n");
printf("%-5s %-20s %5s %6s %-30s\n", "param", "name", "type", "length", "data");
printf("%-5s %-20s %5s %5s- %-30s\n", dashes5, dashes20, dashes5, dashes5, dashes30);
for (i = 1; i <= dbnumrets(dbproc); i++) {
retname = dbretname(dbproc, i);
rettype = dbrettype(dbproc, i);
retlen = dbretlen(dbproc, i);
dbconvert(dbproc, rettype, dbretdata(dbproc, i), retlen, SYBVARCHAR, (BYTE*) teststr, -1);
if(retlen <= 10) {
output = teststr;
} else {
memcpy(abbrev_data, teststr, 10);
sprintf(&abbrev_data[10], "...");
output = abbrev_data;
}
printf("%-5d %-20s %5d %6d %-30s\n", i, retname, rettype, retlen, output);
save_retparam(&save_param, retname, teststr, rettype, retlen);
if (i == 4) {
save_retparam(&save_varchar_tds7_param, retname, teststr, rettype, retlen);
}
if (i == 5) {
save_retparam(&save_nvarchar_tds7_param, retname, teststr, rettype, retlen);
}
}
/*
* Test the last parameter for expected outcome
*/
if ((save_param.name == NULL) || strcmp(save_param.name, bindings[5].name)) {
fprintf(stderr, "Expected retname to be '%s', got ", bindings[5].name);
if (save_param.name == NULL)
fprintf(stderr, "<NULL> instead.\n");
else
fprintf(stderr, "'%s' instead.\n", save_param.name);
exit(1);
}
if (strcmp(save_param.value, "3")) {
fprintf(stderr, "Expected retdata to be 3.\n");
exit(1);
}
if (save_param.type != SYBINT4) {
fprintf(stderr, "Expected rettype to be SYBINT4 was %d.\n", save_param.type);
exit(1);
}
if (save_param.len != 4) {
fprintf(stderr, "Expected retlen to be 4.\n");
exit(1);
}
if (num_params == 6) {
/*
* Test name, size, contents of the VARCHAR(8000) output parameter
*/
if ((save_varchar_tds7_param.name == NULL) || strcmp(save_varchar_tds7_param.name, bindings[3].name)) {
fprintf(stderr, "Expected retname to be '%s', got ", bindings[3].name);
if (save_varchar_tds7_param.name == NULL)
fprintf(stderr, "<NULL> instead.\n");
else
fprintf(stderr, "'%s' instead.\n", save_varchar_tds7_param.name);
exit(1);
}
if (save_varchar_tds7_param.type != SYBVARCHAR) {
fprintf(stderr, "Expected rettype to be SYBVARCHAR was %d.\n", save_varchar_tds7_param.type);
exit(1);
}
if (save_varchar_tds7_param.len != 8000) {
fprintf(stderr, "Expected retlen to be 8000 was %d.\n", save_varchar_tds7_param.len);
exit(1);
}
/*
* Test name, size, contents of the NVARCHAR(4000) output parameter
*/
if ((save_nvarchar_tds7_param.name == NULL) || strcmp(save_nvarchar_tds7_param.name, bindings[4].name)) {
fprintf(stderr, "Expected retname to be '%s', got ", bindings[4].name);
if (save_varchar_tds7_param.name == NULL)
fprintf(stderr, "<NULL> instead.\n");
else
fprintf(stderr, "'%s' instead.\n", save_nvarchar_tds7_param.name);
exit(1);
}
if (save_nvarchar_tds7_param.len != 4000) {
fprintf(stderr, "Expected retlen to be 4000 was %d.\n", save_nvarchar_tds7_param.len);
exit(1);
}
}
if(42 != return_status) {
fprintf(stderr, "Expected status to be 42.\n");
exit(1);
}
printf("Good: Got 6 output parameters and 1 return status of %d.\n", return_status);
/* Test number of result sets */
if (num_resultset != 4) {
fprintf(stderr, "Expected 4 resultset got %d.\n", num_resultset);
exit(1);
}
if (num_empty_resultset != 1) {
fprintf(stderr, "Expected an empty resultset got %d.\n", num_empty_resultset);
exit(1);
}
printf("Good: Got %d resultsets and %d empty resultset.\n", num_resultset, num_empty_resultset);
printf("Dropping procedure\n");
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
dbexit();
printf("%s %s\n", __FILE__, (failed ? "failed!" : "OK"));
free_retparam(&save_param);
free_retparam(&save_varchar_tds7_param);
free_retparam(&save_nvarchar_tds7_param);
return failed ? 1 : 0;
}
static void
test(int bind_type)
{
LOGINREC *login;
DBPROCESS *dbproc;
DBNUMERIC *num = NULL, *num2 = NULL;
RETCODE ret;
int i;
sql_rewind();
login = dblogin();
DBSETLUSER(login, USER);
DBSETLPWD(login, PASSWORD);
DBSETLAPP(login, "numeric");
dbsetmaxprocs(25);
DBSETLHOST(login, SERVER);
dbproc = dbopen(login, SERVER);
dbloginfree(login);
login = NULL;
if (strlen(DATABASE))
dbuse(dbproc, DATABASE);
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
if (DBTDS_5_0 < DBTDS(dbproc)) {
ret = dbcmd(dbproc,
"SET ARITHABORT ON;"
"SET CONCAT_NULL_YIELDS_NULL ON;"
"SET ANSI_NULLS ON;"
"SET ANSI_NULL_DFLT_ON ON;"
"SET ANSI_PADDING ON;"
"SET ANSI_WARNINGS ON;"
"SET ANSI_NULL_DFLT_ON ON;"
"SET CURSOR_CLOSE_ON_COMMIT ON;"
"SET QUOTED_IDENTIFIER ON");
chk(ret, "dbcmd");
ret = dbsqlexec(dbproc);
chk(ret, "dbsqlexec");
ret = dbcancel(dbproc);
chk(ret, "dbcancel");
}
ret = dbrpcinit(dbproc, "testDecimal", 0);
chk(ret, "dbrpcinit");
num = (DBDECIMAL *) calloc(1, sizeof(DBDECIMAL));
num->scale = 5;
num->precision = 16;
dbconvert(dbproc, SYBVARCHAR, (const BYTE *) "123.45", -1, SYBDECIMAL, (BYTE *) num, sizeof(*num));
ret = dbrpcparam(dbproc, "@idecimal", 0, SYBDECIMAL, -1, sizeof(DBDECIMAL), (BYTE *) num);
chk(ret, "dbrpcparam");
ret = dbrpcsend(dbproc);
chk(ret, "dbrpcsend");
ret = dbsqlok(dbproc);
chk(ret, "dbsqlok");
/* TODO check MS/Sybase format */
num2 = (DBDECIMAL *) calloc(1, sizeof(DBDECIMAL));
num2->precision = 20;
num2->scale = 10;
dbconvert(dbproc, SYBVARCHAR, (const BYTE *) "246.9", -1, SYBDECIMAL, (BYTE *) num2, sizeof(*num2));
for (i=0; (ret = dbresults(dbproc)) != NO_MORE_RESULTS; ++i) {
RETCODE row_code;
switch (ret) {
case SUCCEED:
if (DBROWS(dbproc) == FAIL)
continue;
assert(DBROWS(dbproc) == SUCCEED);
printf("dbrows() returned SUCCEED, processing rows\n");
memset(num, 0, sizeof(*num));
num->precision = num2->precision;
num->scale = num2->scale;
dbbind(dbproc, 1, bind_type, 0, (BYTE *) num);
while ((row_code = dbnextrow(dbproc)) != NO_MORE_ROWS) {
if (row_code == REG_ROW) {
if (memcmp(num, num2, sizeof(*num)) != 0) {
fprintf(stderr, "Failed. Output results does not match\n");
dump_addr(stderr, "numeric: ", num, sizeof(*num));
dump_addr(stderr, "numeric2:", num2, sizeof(*num2));
exit(1);
}
} else {
/* not supporting computed rows in this unit test */
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
}
break;
case FAIL:
fprintf(stderr, "dbresults returned FAIL\n");
exit(1);
default:
fprintf(stderr, "unexpected return code %d from dbresults\n", ret);
exit(1);
}
} /* while dbresults */
sql_cmd(dbproc);
free(num2);
free(num);
dbclose(dbproc);
}
static int
test(int argc, char **argv, int over4k)
{
const int rows_to_add = 3;
LOGINREC *login;
DBPROCESS *dbproc;
DBPROCESS *blobproc;
int i;
DBINT testint;
FILE *fp;
long result, isiz;
char *blob, *rblob;
unsigned char *textPtr, *timeStamp;
char objname[256];
char sqlCmd[256];
char rbuf[BLOB_BLOCK_SIZE];
long numread;
int numtowrite, numwritten;
set_malloc_options();
read_login_info(argc, argv);
fprintf(stdout, "Starting %s\n", argv[0]);
dbinit();
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
fprintf(stdout, "About to logon\n");
login = dblogin();
DBSETLPWD(login, PASSWORD);
DBSETLUSER(login, USER);
DBSETLAPP(login, "t0014");
fprintf(stdout, "About to open %s..%s for user '%s'\n", SERVER, DATABASE, USER);
dbproc = dbopen(login, SERVER);
blobproc = dbopen(login, SERVER);
if (strlen(DATABASE)) {
dbuse(dbproc, DATABASE);
dbuse(blobproc, DATABASE);
}
dbloginfree(login);
fprintf(stdout, "After logon\n");
fprintf(stdout, "About to read binary input file\n");
if (argc == 1) {
argv = testargs;
argc = 3;
}
if (argc < 3) {
fprintf(stderr, "Usage: %s infile outfile\n", argv[0]);
return 1;
}
if ((fp = fopen(argv[1], "rb")) == NULL) {
fprintf(stderr, "Cannot open input file: %s\n", argv[1]);
return 2;
}
result = fseek(fp, 0, SEEK_END);
isiz = ftell(fp);
result = fseek(fp, 0, SEEK_SET);
blob = (char *) malloc(isiz);
result = fread((void *) blob, isiz, 1, fp);
assert(result == 1);
fclose(fp);
/* FIXME this test seem to not work using temporary tables (sybase?)... */
fprintf(stdout, "Dropping table\n");
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
fprintf(stdout, "creating table\n");
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
fprintf(stdout, "insert\n");
for (i = 0; i < rows_to_add; i++) {
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
}
for (i = 0; i < rows_to_add; i++) {
sql_cmd(dbproc);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
fprintf(stderr, "Error inserting blob\n");
return 4;
}
while ((result = dbnextrow(dbproc)) != NO_MORE_ROWS) {
result = REG_ROW;
result = DBTXPLEN;
strcpy(objname, "dblib0014.PigTure");
textPtr = dbtxptr(dbproc, 1);
timeStamp = dbtxtimestamp(dbproc, 1);
if (!textPtr && !timeStamp && dbtds(dbproc) >= DBTDS_7_2) {
printf("Protocol 7.2+ detected, test not supported\n");
free(blob);
dbexit();
exit(0);
}
/*
* Use #ifdef if you want to test dbmoretext mode (needed for 16-bit apps)
* Use #ifndef for big buffer version (32-bit)
*/
if (over4k) {
if (dbwritetext(blobproc, objname, textPtr, DBTXPLEN, timeStamp, TRUE, isiz, (BYTE*) blob) != SUCCEED)
return 5;
} else {
if (dbwritetext(blobproc, objname, textPtr, DBTXPLEN, timeStamp, TRUE, isiz, NULL) != SUCCEED)
return 15;
dbsqlok(blobproc);
dbresults(blobproc);
numtowrite = 0;
/* Send the update value in chunks. */
for (numwritten = 0; numwritten < isiz; numwritten += numtowrite) {
numtowrite = (isiz - numwritten);
if (numtowrite > BLOB_BLOCK_SIZE)
numtowrite = BLOB_BLOCK_SIZE;
dbmoretext(blobproc, (DBINT) numtowrite, (BYTE *) (blob + numwritten));
}
dbsqlok(blobproc);
while (dbresults(blobproc) != NO_MORE_RESULTS);
}
}
}
fprintf(stdout, "select\n");
sql_cmd(dbproc);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
fprintf(stdout, "Was expecting a result set.");
exit(1);
}
for (i = 1; i <= dbnumcols(dbproc); i++) {
printf("col %d is %s\n", i, dbcolname(dbproc, i));
}
if (SUCCEED != dbbind(dbproc, 1, INTBIND, -1, (BYTE *) & testint)) {
fprintf(stderr, "Had problem with bind\n");
abort();
}
for (i = 0; i < rows_to_add; i++) {
char expected[1024];
sprintf(expected, "row %03d", i);
if (REG_ROW != dbnextrow(dbproc)) {
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
if (testint != i) {
fprintf(stderr, "Failed. Expected i to be %d, was %d\n", i, (int) testint);
abort();
}
/* get the image */
strcpy(sqlCmd, "SET TEXTSIZE 2147483647");
dbcmd(blobproc, sqlCmd);
dbsqlexec(blobproc);
if (dbresults(blobproc) != SUCCEED) {
dbcancel(blobproc);
return 16;
}
sprintf(sqlCmd, "SELECT PigTure FROM dblib0014 WHERE i = %d", i);
dbcmd(blobproc, sqlCmd);
dbsqlexec(blobproc);
if (dbresults(blobproc) != SUCCEED) {
fprintf(stderr, "Error extracting blob\n");
return 6;
}
numread = 0;
rblob = NULL;
while ((result = dbreadtext(blobproc, rbuf, BLOB_BLOCK_SIZE)) != NO_MORE_ROWS) {
if (result != 0) { /* this indicates not end of row */
rblob = (char*) realloc(rblob, result + numread);
memcpy((void *) (rblob + numread), (void *) rbuf, result);
numread += result;
}
}
if (i == 0) {
printf("Saving first blob data row to file: %s\n", argv[2]);
if ((fp = fopen(argv[2], "wb")) == NULL) {
fprintf(stderr, "Unable to open output file: %s\n", argv[2]);
return 3;
}
result = fwrite((void *) rblob, numread, 1, fp);
fclose(fp);
}
printf("Read blob data row %d --> %s %ld byte comparison\n",
(int) testint, (memcmp(blob, rblob, numread)) ? "failed" : "PASSED", numread);
free(rblob);
}
if (dbnextrow(dbproc) != NO_MORE_ROWS) {
fprintf(stderr, "Was expecting no more rows\n");
exit(1);
}
free(blob);
fprintf(stdout, "Dropping table\n");
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
dbexit();
return 0;
}
static void waitForDataResponse(eio_req *req) {
data_callback_t *callbackData = (data_callback_t *) req->data;
//dbsqlok
req->result = dbsqlok(callbackData->dbconn);
}
/**
* The purpose of this program is threefold:
*
* 1. To provide a generalized SQL processor suitable for testing db-lib.
* 2. To offer a robust batch-oriented SQL processor suitable for use in a production environment.
* 3. To serve as a model example of how to use db-lib functions.
*
* These purposes may be somewhat at odds with one another. For instance, the tutorial aspect calls for
* explanatory comments that wouldn't appear in production code. Nevertheless, I hope the experienced
* reader will forgive the verbosity and still find the program useful.
*
* \todo The error/message handlers are not robust enough. They should anticipate certain conditions
* and cause the application to retry the operation.
*/
int
main(int argc, char *argv[])
{
LOGINREC *login;
DBPROCESS *dbproc;
RETCODE erc;
setlocale(LC_ALL, "");
/* Initialize db-lib */
erc = dbinit();
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbinit() failed\n", options.appname, __LINE__);
exit(1);
}
/* Install our error and message handlers */
dberrhandle(err_handler);
dbmsghandle(msg_handler);
memset(&options, 0, sizeof(options));
options.headers = stderr;
login = get_login(argc, argv, &options); /* get command-line parameters and call dblogin() */
assert(login != NULL);
/*
* Override stdin, stdout, and stderr, as required
*/
if (options.input_filename) {
if (freopen(options.input_filename, "r", stdin) == NULL) {
fprintf(stderr, "%s: unable to open %s: %s\n", options.appname, options.input_filename, strerror(errno));
exit(1);
}
}
if (options.output_filename) {
if (freopen(options.output_filename, "w", stdout) == NULL) {
fprintf(stderr, "%s: unable to open %s: %s\n", options.appname, options.output_filename, strerror(errno));
exit(1);
}
}
if (options.error_filename) {
if (freopen(options.error_filename, "w", stderr) == NULL) {
fprintf(stderr, "%s: unable to open %s: %s\n", options.appname, options.error_filename, strerror(errno));
exit(1);
}
}
if (options.fverbose) {
options.verbose = stderr;
} else {
static const char null_device[] = NULL_DEVICE;
options.verbose = fopen(null_device, "w");
if (options.verbose == NULL) {
fprintf(stderr, "%s:%d unable to open %s for verbose operation: %s\n",
options.appname, __LINE__, null_device, strerror(errno));
exit(1);
}
}
fprintf(options.verbose, "%s:%d: Verbose operation enabled\n", options.appname, __LINE__);
/*
* Connect to the server
*/
if ((dbproc = dbopen(login, options.servername)) == NULL)
return 1;
/* Switch to the specified database, if any */
if (options.database)
dbuse(dbproc, options.database);
/*
* Read the queries and write the results
*/
while (next_query(dbproc) != -1 ) {
/* Send the query to the server (we could use dbsqlexec(), instead) */
erc = dbsqlsend(dbproc);
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbsqlsend() failed\n", options.appname, __LINE__);
exit(1);
}
fprintf(options.verbose, "%s:%d: dbsqlsend(): OK\n", options.appname, __LINE__);
/* Wait for it to execute */
erc = dbsqlok(dbproc);
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbsqlok() failed\n", options.appname, __LINE__);
exit(1);
}
fprintf(options.verbose, "%s:%d: dbsqlok(): OK\n", options.appname, __LINE__);
/* Write the output */
print_results(dbproc);
}
return 0;
}
static ngx_int_t
ngx_dbd_freetds_query(ngx_dbd_t *dbd)
{
ngx_dbd_freetds_ctx_t *ctx;
ngx_log_debug0(NGX_LOG_DEBUG_MYSQL, dbd->log, 0, "dbd freetds query");
ctx = dbd->ctx;
#if 0
if (dbrpcinit(ctx->db, ctx->sql, 0) != SUCCEED) {
return NGX_ERROR;
}
/* TODO: dbreginit */
/* dbrpcparam */
if (dbrpcexec(ctx->db) != SUCCEED) {
return NGX_ERROR;
}
if (dbrpcsend(ctx->db) != SUCCEED) {
return NGX_ERROR;
}
/* dbregparam */
/* dbregexec */
ctx->err = dbfcmd(ctx->db, ctx->sql);
if (ctx->err != SUCCEED) {
return NGX_ERROR;
}
/* dbbind */
/* dbhasretstat */
/* dbretstatus */
/* dbnumrets */
/* dbretdata */
/* dbrettype */
/* dbretlen */
/* dbretname */
if (dbsqlok(ctx->db) != SUCCEED) {
return NGX_ERROR;
}
#endif
ctx->err = dbcmd(ctx->db, (LPCSTR) ctx->sql);
if (ctx->err != SUCCEED) {
ngx_log_error(NGX_LOG_ALERT, dbd->log, 0, "dbcmd() failed");
return NGX_ERROR;
}
ctx->err = dbsqlexec(ctx->db);
if (ctx->err != SUCCEED) {
ngx_log_error(NGX_LOG_ALERT, dbd->log, 0, "dbsqlexec() failed");
return NGX_ERROR;
}
/* TODO: dbsetopt(DBBUFFER); */
/* dbclropt(ctx->db, DBBUFFER, "100"); */
/* dbisopt; */
#if 0
ctx->err = dbsetopt(ctx->db, DBBUFFER, "100");
#endif
ctx->err = dbresults(ctx->db);
if (ctx->err != SUCCEED) {
ngx_log_error(NGX_LOG_ALERT, dbd->log, 0, "dbresults() failed");
return NGX_ERROR;
}
ctx->cur_col = 0;
ctx->num_fields = (uint64_t) dbnumcols(ctx->db);
/* dbnumcompute(ctx->db); */
/* dbnumalts */
return NGX_OK;
}
/**
* The purpose of this program is to load or extract the text of a stored procedure.
* This first cut does extract only.
* TODO: support loading procedures onto the server.
*/
int
main(int argc, char *argv[])
{
LOGINREC *login;
DBPROCESS *dbproc;
PROCEDURE procedure;
RETCODE erc;
int i, nrows;
/* Initialize db-lib */
erc = dbinit();
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbinit() failed\n", options.appname, __LINE__);
exit(1);
}
memset(&options, 0, sizeof(options));
login = get_login(argc, argv, &options); /* get command-line parameters and call dblogin() */
assert(login != NULL);
/* Install our error and message handlers */
dberrhandle(err_handler);
dbmsghandle(msg_handler);
/*
* Override stdin, stdout, and stderr, as required
*/
if (options.input_filename) {
if (freopen(options.input_filename, "r", stdin) == NULL) {
fprintf(stderr, "%s: unable to open %s: %s\n", options.appname, options.input_filename, strerror(errno));
exit(1);
}
}
if (options.output_filename) {
if (freopen(options.output_filename, "w", stdout) == NULL) {
fprintf(stderr, "%s: unable to open %s: %s\n", options.appname, options.output_filename, strerror(errno));
exit(1);
}
}
/*
* Connect to the server
*/
dbproc = dbopen(login, options.servername);
assert(dbproc != NULL);
/* Switch to the specified database, if any */
if (options.database)
dbuse(dbproc, options.database);
/*
* Read the procedure names and move their texts.
*/
for (i=options.optind; i < argc; i++) {
static const char query[] = " select c.text"
" from syscomments as c"
" join sysobjects as o"
" on o.id = c.id"
" where o.name = '%s'"
" and o.uid = user_id('%s')"
" and o.type not in ('U', 'S')" /* no user or system tables */
" order by c.colid"
;
static const char query_table[] = " execute sp_help '%s.%s' ";
parse_argument(argv[i], &procedure);
erc = dbfcmd(dbproc, query, procedure.name, procedure.owner);
/* Send the query to the server (we could use dbsqlexec(), instead) */
erc = dbsqlsend(dbproc);
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbsqlsend() failed\n", options.appname, __LINE__);
exit(1);
}
/* Wait for it to execute */
erc = dbsqlok(dbproc);
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbsqlok() failed\n", options.appname, __LINE__);
exit(1);
}
/* Write the output */
nrows = print_results(dbproc);
if (0 == nrows) {
erc = dbfcmd(dbproc, query_table, procedure.owner, procedure.name);
erc = dbsqlexec(dbproc);
if (erc == FAIL) {
fprintf(stderr, "%s:%d: dbsqlexec() failed\n", options.appname, __LINE__);
exit(1);
}
nrows = print_ddl(dbproc, &procedure);
}
switch (nrows) {
case -1:
return 1;
case 0:
fprintf(stderr, "%s: error: %s.%s.%s.%s not found\n", options.appname,
options.servername, options.database, procedure.owner, procedure.name);
return 2;
default:
break;
}
}
return 0;
}
int
main(int argc, char **argv)
{
LOGINREC *login;
DBPROCESS *dbproc;
RETPARAM save_param;
int i, r;
char teststr[1024];
int failed = 0;
char *retname = NULL;
int rettype = 0, retlen = 0;
int return_status = 0;
char proc[] = "#t0022",
param0[] = "@null_input",
param1[] = "@first_type",
param2[] = "@nullout",
param3[] = "@nrows",
param4[] = "@c";
char *proc_name = proc;
char param_data1[64];
int param_data2;
int param_data3;
RETCODE erc, row_code;
int num_resultset = 0;
int num_empty_resultset = 0;
static const char dashes5[] = "-----",
dashes15[] = "---------------",
dashes30[] = "------------------------------";
set_malloc_options();
memset(&save_param, 0, sizeof(save_param));
read_login_info(argc, argv);
fprintf(stdout, "Start\n");
add_bread_crumb();
dbinit();
add_bread_crumb();
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
fprintf(stdout, "About to logon\n");
add_bread_crumb();
login = dblogin();
DBSETLPWD(login, PASSWORD);
DBSETLUSER(login, USER);
DBSETLAPP(login, "#t0022");
dberrhandle(ignore_err_handler);
DBSETLPACKET(login, -1);
dberrhandle(syb_err_handler);
fprintf(stdout, "About to open %s.%s\n", SERVER, DATABASE);
add_bread_crumb();
dbproc = dbopen(login, SERVER);
if (strlen(DATABASE))
dbuse(dbproc, DATABASE);
add_bread_crumb();
dbloginfree(login);
add_bread_crumb();
add_bread_crumb();
dberrhandle(ignore_err_handler);
dbmsghandle(ignore_msg_handler);
printf("trying to create a temporary stored procedure\n");
if (FAIL == init_proc(dbproc, proc_name)) {
printf("trying to create a permanent stored procedure\n");
if (FAIL == init_proc(dbproc, ++proc_name))
exit(EXIT_FAILURE);
}
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
fprintf(stdout, "Created procedure %s\n", proc_name);
/* set up and send the rpc */
printf("executing dbrpcinit\n");
erc = dbrpcinit(dbproc, proc_name, 0); /* no options */
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcinit\n");
failed = 1;
}
printf("executing dbrpcparam\n");
erc = dbrpcparam(dbproc, param0, DBRPCRETURN, SYBCHAR, /*maxlen= */ -1, /* datlen= */ 0, NULL);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcparam\n");
failed = 1;
}
printf("executing dbrpcparam\n");
erc = dbrpcparam(dbproc, param1, DBRPCRETURN, SYBCHAR, /*maxlen= */ sizeof(param_data1), /* datlen= */ 0, (BYTE *) & param_data1);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcparam\n");
failed = 1;
}
printf("executing dbrpcparam\n");
erc = dbrpcparam(dbproc, param2, DBRPCRETURN, SYBINT4, /*maxlen= */ -1, /* datalen= */ 0, (BYTE *) & param_data2);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcparam\n");
failed = 1;
}
printf("executing dbrpcparam\n");
erc = dbrpcparam(dbproc, param3, DBRPCRETURN, SYBINT4, /*maxlen= */ -1, /* datalen= */ -1, (BYTE *) & param_data3);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcparam\n");
failed = 1;
}
/* test for strange parameters using input */
printf("executing dbrpcparam\n");
erc = dbrpcparam(dbproc, param4, 0, SYBVARCHAR, /*maxlen= */ 0, /* datalen= */ 0, NULL);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcparam\n");
failed = 1;
}
printf("executing dbrpcsend\n");
param_data3 = 0x11223344;
erc = dbrpcsend(dbproc);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbrpcsend\n");
exit(1);
}
/* wait for it to execute */
printf("executing dbsqlok\n");
erc = dbsqlok(dbproc);
if (erc == FAIL) {
fprintf(stderr, "Failed: dbsqlok\n");
exit(1);
}
add_bread_crumb();
/* retrieve outputs per usual */
r = 0;
while ((erc = dbresults(dbproc)) != NO_MORE_RESULTS) {
if (erc == SUCCEED) {
const int ncols = dbnumcols(dbproc);
int empty_resultset = 1;
++num_resultset;
printf("bound 1 of %d columns ('%s') in result %d.\n", ncols, dbcolname(dbproc, 1), ++r);
dbbind(dbproc, 1, STRINGBIND, 0, (BYTE *) teststr);
printf("\t%s\n\t-----------\n", dbcolname(dbproc, 1));
while ((row_code = dbnextrow(dbproc)) != NO_MORE_ROWS) {
empty_resultset = 0;
if (row_code == REG_ROW) {
printf("\t%s\n", teststr);
} else {
/* not supporting computed rows in this unit test */
failed = 1;
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
}
printf("row count %d\n", (int) dbcount(dbproc));
if (empty_resultset)
++num_empty_resultset;
} else {
add_bread_crumb();
fprintf(stderr, "Expected a result set.\n");
exit(1);
}
} /* while dbresults */
/* check return status */
printf("retrieving return status...\n");
if (dbhasretstat(dbproc) == TRUE) {
printf("%d\n", return_status = dbretstatus(dbproc));
} else {
printf("none\n");
}
/*
* Check output parameter values
*/
if (dbnumrets(dbproc) < 4) { /* dbnumrets missed something */
fprintf(stderr, "Expected 4 output parameters.\n");
exit(1);
}
printf("retrieving output parameters...\n");
printf("%-5s %-15s %5s %6s %-30s\n", "param", "name", "type", "length", "data");
printf("%-5s %-15s %5s %5s- %-30s\n", dashes5, dashes15, dashes5, dashes5, dashes30);
for (i = 1; i <= dbnumrets(dbproc); i++) {
add_bread_crumb();
retname = dbretname(dbproc, i);
rettype = dbrettype(dbproc, i);
retlen = dbretlen(dbproc, i);
dbconvert(dbproc, rettype, dbretdata(dbproc, i), retlen, SYBVARCHAR, (BYTE*) teststr, -1);
printf("%-5d %-15s %5d %6d %-30s\n", i, retname, rettype, retlen, teststr);
add_bread_crumb();
save_retparam(&save_param, retname, teststr, rettype, retlen);
}
/*
* Test the last parameter for expected outcome
*/
if ((save_param.name == NULL) || strcmp(save_param.name, param3)) {
fprintf(stderr, "Expected retname to be '%s', got ", param3);
if (save_param.name == NULL)
fprintf(stderr, "<NULL> instead.\n");
else
fprintf(stderr, "'%s' instead.\n", save_param.name);
exit(1);
}
if (strcmp(save_param.value, "3")) {
fprintf(stderr, "Expected retdata to be 3.\n");
exit(1);
}
if (save_param.type != SYBINT4) {
fprintf(stderr, "Expected rettype to be SYBINT4 was %d.\n", save_param.type);
exit(1);
}
if (save_param.len != 4) {
fprintf(stderr, "Expected retlen to be 4.\n");
exit(1);
}
if(42 != return_status) {
fprintf(stderr, "Expected status to be 42.\n");
exit(1);
}
printf("Good: Got 4 output parameters and 1 return status of %d.\n", return_status);
/* Test number of result sets */
if (num_resultset != 3) {
fprintf(stderr, "Expected 3 resultset got %d.\n", num_resultset);
exit(1);
}
if (num_empty_resultset != 1) {
fprintf(stderr, "Expected an empty resultset got %d.\n", num_empty_resultset);
exit(1);
}
printf("Good: Got %d resultsets and %d empty resultset.\n", num_resultset, num_empty_resultset);
add_bread_crumb();
fprintf(stdout, "Dropping procedure\n");
add_bread_crumb();
sprintf(cmd, "DROP PROCEDURE %s", proc_name);
dbcmd(dbproc, cmd);
add_bread_crumb();
dbsqlexec(dbproc);
add_bread_crumb();
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
add_bread_crumb();
dbexit();
add_bread_crumb();
fprintf(stdout, "dblib %s on %s\n", (failed ? "failed!" : "okay"), __FILE__);
free_bread_crumb();
free(save_param.name);
free(save_param.value);
return failed ? 1 : 0;
}
int
main(int argc, const char *argv[])
{
LOGINREC *login;
DBPROCESS *dbproc;
int i;
DBINT testint;
FILE *fp;
long result, isiz;
char *blob, *rblob;
unsigned char *textPtr = NULL, *timeStamp = NULL;
char objname[256];
char sqlCmd[256];
char rbuf[BLOB_BLOCK_SIZE];
long numread;
BOOL readFirstImage;
char cmd[1024];
set_malloc_options();
read_login_info();
fprintf(stdout, "Start\n");
dbinit();
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
fprintf(stdout, "About to logon\n");
login = dblogin();
DBSETLPWD(login, PASSWORD);
DBSETLUSER(login, USER);
DBSETLAPP(login, "t0013");
fprintf(stdout, "About to open, PASSWORD: %s, USER: %s, SERVER: %s\n", "", "", ""); /* PASSWORD, USER, SERVER); */
dbproc = dbopen(login, SERVER);
if (strlen(DATABASE)) {
dbuse(dbproc, DATABASE);
}
fprintf(stdout, "After logon\n");
fprintf(stdout, "About to read binary input file\n");
if (argc == 1) {
argv = testargs;
argc = 3;
}
if (argc < 3) {
fprintf(stderr, "Usage: %s infile outfile\n", argv[0]);
return 1;
}
if ((fp = fopen(argv[1], "rb")) == NULL) {
fprintf(stderr, "Cannot open input file: %s\n", argv[1]);
return 2;
}
result = fseek(fp, 0, SEEK_END);
isiz = ftell(fp);
result = fseek(fp, 0, SEEK_SET);
blob = (char *) malloc(isiz);
fread((void *) blob, isiz, 1, fp);
fclose(fp);
fprintf(stdout, "Dropping table\n");
dbcmd(dbproc, "drop table #dblib0013");
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
fprintf(stdout, "creating table\n");
dbcmd(dbproc, "create table #dblib0013 (i int not null, PigTure image not null)");
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
fprintf(stdout, "insert\n");
sprintf(cmd, "insert into #dblib0013 values (1, '')");
fprintf(stdout, "%s\n", cmd);
dbcmd(dbproc, cmd);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
sprintf(sqlCmd, "SELECT PigTure FROM #dblib0013 WHERE i = 1");
dbcmd(dbproc, sqlCmd);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
fprintf(stderr, "Error inserting blob\n");
return 4;
}
while ((result = dbnextrow(dbproc)) != NO_MORE_ROWS) {
result = REG_ROW;
result = DBTXPLEN;
strcpy(objname, "#dblib0013.PigTure");
textPtr = dbtxptr(dbproc, 1);
timeStamp = dbtxtimestamp(dbproc, 1);
}
/* Use #ifdef if you want to test dbmoretext mode (needed for 16-bit apps)
* Use #ifndef for big buffer version (32-bit) */
#ifndef DBWRITE_OK_FOR_OVER_4K
if (dbwritetext(dbproc, objname, textPtr, DBTXPLEN, timeStamp, FALSE, isiz, (BYTE*) blob) != SUCCEED)
return 5;
#else
if (dbwritetext(dbproc, objname, textPtr, DBTXPLEN, timeStamp, FALSE, isiz, NULL) != SUCCEED)
return 15;
dbsqlok(dbproc);
dbresults(dbproc);
numtowrite = 0;
/* Send the update value in chunks. */
for (numwritten = 0; numwritten < isiz; numwritten += numtowrite) {
numtowrite = (isiz - numwritten);
if (numtowrite > BLOB_BLOCK_SIZE)
numtowrite = BLOB_BLOCK_SIZE;
dbmoretext(dbproc, (DBINT) numtowrite, blob + numwritten);
}
dbsqlok(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS);
#endif
fprintf(stdout, "select\n");
dbcmd(dbproc, "select * from #dblib0013 order by i");
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
failed = 1;
fprintf(stdout, "Was expecting a result set.");
exit(1);
}
for (i = 1; i <= dbnumcols(dbproc); i++) {
printf("col %d is %s\n", i, dbcolname(dbproc, i));
}
if (SUCCEED != dbbind(dbproc, 1, INTBIND, -1, (BYTE *) & testint)) {
failed = 1;
fprintf(stderr, "Had problem with bind\n");
abort();
}
if (REG_ROW != dbnextrow(dbproc)) {
failed = 1;
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
if (testint != 1) {
failed = 1;
fprintf(stderr, "Failed. Expected i to be %d, was %d\n", i, (int) testint);
abort();
}
dbnextrow(dbproc);
/* get the image */
strcpy(sqlCmd, "SET TEXTSIZE 2147483647");
dbcmd(dbproc, sqlCmd);
dbsqlexec(dbproc);
dbresults(dbproc);
fprintf(stdout, "select 2\n");
sprintf(sqlCmd, "SELECT PigTure FROM #dblib0013 WHERE i = 1");
dbcmd(dbproc, sqlCmd);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
fprintf(stderr, "Error extracting blob\n");
return 6;
}
numread = 0;
rblob = NULL;
readFirstImage = FALSE;
while ((result = dbreadtext(dbproc, rbuf, BLOB_BLOCK_SIZE)) != NO_MORE_ROWS) {
if (result == 0) { /* this indicates end of row */
readFirstImage = TRUE;
} else {
rblob = (char*) realloc(rblob, result + numread);
memcpy((void *) (rblob + numread), (void *) rbuf, result);
numread += result;
}
}
printf("Saving first blob data row to file: %s\n", argv[2]);
if ((fp = fopen(argv[2], "wb")) == NULL) {
fprintf(stderr, "Unable to open output file: %s\n", argv[2]);
return 3;
}
result = fwrite((void *) rblob, numread, 1, fp);
fclose(fp);
printf("Read blob data row %d --> %s %ld byte comparison\n",
(int) testint, (memcmp(blob, rblob, numread)) ? "failed" : "PASSED", numread);
free(rblob);
if (dbnextrow(dbproc) != NO_MORE_ROWS) {
failed = 1;
fprintf(stderr, "Was expecting no more rows\n");
exit(1);
}
dbexit();
fprintf(stdout, "dblib %s on %s\n", (failed ? "failed!" : "okay"), __FILE__);
return failed ? 1 : 0;
}
static int
test(int argc, char **argv, int over4k)
{
LOGINREC *login;
int i;
DBINT testint;
FILE *fp;
long result, isiz;
char *blob, *rblob;
DBBINARY *textPtr = NULL, *timeStamp = NULL;
char objname[256];
char rbuf[BLOB_BLOCK_SIZE];
long numread;
int data_ok;
int numtowrite, numwritten;
set_malloc_options();
read_login_info(argc, argv);
fprintf(stdout, "Starting %s\n", argv[0]);
dbinit();
dberrhandle(syb_err_handler);
dbmsghandle(syb_msg_handler);
fprintf(stdout, "About to logon\n");
login = dblogin();
DBSETLPWD(login, PASSWORD);
DBSETLUSER(login, USER);
DBSETLAPP(login, "t0013");
fprintf(stdout, "About to open, PASSWORD: %s, USER: %s, SERVER: %s\n", "", "", ""); /* PASSWORD, USER, SERVER); */
dbproc = dbopen(login, SERVER);
dbprocw = dbopen(login, SERVER);
if (strlen(DATABASE)) {
dbuse(dbproc, DATABASE);
dbuse(dbprocw, DATABASE);
}
dbloginfree(login);
fprintf(stdout, "logged in\n");
if (argc == 1) {
argv = testargs;
argc = 3;
}
if (argc < 3) {
fprintf(stderr, "Usage: %s infile outfile\n", argv[0]);
return 1;
}
if ((fp = fopen(argv[1], "rb")) == NULL) {
fprintf(stderr, "Cannot open input file: %s\n", argv[1]);
return 2;
}
fprintf(stdout, "Reading binary input file\n");
fseek(fp, 0, SEEK_END);
isiz = ftell(fp);
fseek(fp, 0, SEEK_SET);
blob = malloc(isiz);
assert(blob);
result = fread((void *) blob, isiz, 1, fp);
assert(result == 1);
fclose(fp);
drop_table();
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
sql_cmd(dbproc);
dbsqlexec(dbproc);
while (dbresults(dbproc) != NO_MORE_RESULTS) {
/* nop */
}
sql_cmd(dbproc);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
fprintf(stderr, "Error inserting blob\n");
return 4;
}
for (i=0; (result = dbnextrow(dbproc)) != NO_MORE_ROWS; i++) {
assert(REG_ROW == result);
printf("fetching row %d\n", i+1);
strcpy(objname, TABLE_NAME ".PigTure");
textPtr = dbtxptr(dbproc, 1);
timeStamp = dbtxtimestamp(dbproc, 1);
break; /* can't proceed until no more rows */
}
assert(REG_ROW == result || 0 < i);
if (!textPtr && !timeStamp && dbtds(dbproc) >= DBTDS_7_2) {
printf("Protocol 7.2+ detected, test not supported\n");
free(blob);
dbclose(dbproc);
dbproc = NULL;
dbexit();
exit(0);
}
if (!textPtr) {
fprintf(stderr, "Error getting textPtr\n");
exit(1);
}
/*
* Use #ifdef if you want to test dbmoretext mode (needed for 16-bit apps)
* Use #ifndef for big buffer version (32-bit)
*/
fprintf(stdout, "writing text ... ");
if (over4k) {
if (dbwritetext(dbprocw, objname, textPtr, DBTXPLEN, timeStamp, FALSE, isiz, (BYTE*) blob) != SUCCEED)
return 5;
fprintf(stdout, "done (in one shot)\n");
for (; (result = dbnextrow(dbproc)) != NO_MORE_ROWS; i++) {
assert(REG_ROW == result);
printf("fetching row %d?\n", i+1);
}
} else {
if (dbwritetext(dbprocw, objname, textPtr, DBTXPLEN, timeStamp, FALSE, isiz, NULL) != SUCCEED)
return 15;
fprintf(stdout, "done\n");
fprintf(stdout, "dbsqlok\n");
dbsqlok(dbprocw);
fprintf(stdout, "dbresults\n");
dbresults(dbprocw);
numtowrite = 0;
/* Send the update value in chunks. */
for (numwritten = 0; numwritten < isiz; numwritten += numtowrite) {
fprintf(stdout, "dbmoretext %d\n", 1 + numwritten);
numtowrite = (isiz - numwritten);
if (numtowrite > BLOB_BLOCK_SIZE)
numtowrite = BLOB_BLOCK_SIZE;
dbmoretext(dbprocw, (DBINT) numtowrite, (BYTE *) (blob + numwritten));
}
dbsqlok(dbprocw);
while (dbresults(dbprocw) != NO_MORE_RESULTS){
printf("suprise!\n");
}
}
#if 0
if (SUCCEED != dbclose(dbproc)){
fprintf(stdout, "dbclose failed");
exit(1);
}
dbproc = dbopen(login, SERVER);
assert(dbproc);
if (strlen(DATABASE)) {
dbuse(dbproc, DATABASE);
}
#endif
sql_cmd(dbproc);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
failed = 1;
fprintf(stdout, "Was expecting a result set.");
exit(1);
}
for (i = 1; i <= dbnumcols(dbproc); i++) {
printf("col %d, \"%s\", is type %s\n",
i, dbcolname(dbproc, i), dbprtype(dbcoltype(dbproc, i)));
}
if (2 != dbnumcols(dbproc)) {
fprintf(stderr, "Failed. Expected 2 columns\n");
exit(1);
}
if (SUCCEED != dbbind(dbproc, 1, INTBIND, -1, (BYTE *) & testint)) {
fprintf(stderr, "Had problem with bind\n");
exit(1);
}
if (REG_ROW != dbnextrow(dbproc)) {
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
if (testint != 1) {
failed = 1;
fprintf(stderr, "Failed. Expected i to be %d, was %d\n", 1, (int) testint);
exit(1);
}
dbnextrow(dbproc);
/* get the image */
sql_cmd(dbproc);
dbsqlexec(dbproc);
dbresults(dbproc);
fprintf(stdout, "select 2\n");
sql_cmd(dbproc);
dbsqlexec(dbproc);
if (dbresults(dbproc) != SUCCEED) {
fprintf(stderr, "Error extracting blob\n");
return 6;
}
numread = 0;
rblob = NULL;
while ((result = dbreadtext(dbproc, rbuf, BLOB_BLOCK_SIZE)) != NO_MORE_ROWS) {
if (result != 0) { /* this indicates not end of row */
rblob = (char*) realloc(rblob, result + numread);
memcpy((void *) (rblob + numread), (void *) rbuf, result);
numread += result;
}
}
data_ok = 1;
if (memcmp(blob, rblob, numread) != 0) {
printf("Saving first blob data row to file: %s\n", argv[2]);
if ((fp = fopen(argv[2], "wb")) == NULL) {
fprintf(stderr, "Unable to open output file: %s\n", argv[2]);
return 3;
}
fwrite((void *) rblob, numread, 1, fp);
fclose(fp);
failed = 1;
data_ok = 0;
}
printf("Read blob data row %d --> %s %ld byte comparison\n",
(int) testint, data_ok ? "PASSED" : "failed", numread);
free(rblob);
if (dbnextrow(dbproc) != NO_MORE_ROWS) {
failed = 1;
fprintf(stderr, "Was expecting no more rows\n");
exit(1);
}
free(blob);
drop_table();
dbclose(dbproc);
dbproc = NULL;
dbexit();
return 0;
}
int
main(int argc, char **argv)
{
LOGINREC *login;
DBPROCESS *dbproc;
int i,r, failed = 0;
RETCODE erc, row_code;
int num_resultset = 0;
char teststr[1024];
/*
* Connect to server
*/
set_malloc_options();
read_login_info(argc, argv);
fprintf(stdout, "Starting %s\n", argv[0]);
dbinit();
dberrhandle(timeout_err_handler);
dbmsghandle(syb_msg_handler);
fprintf(stdout, "About to logon\n");
login = dblogin();
DBSETLPWD(login, PASSWORD);
DBSETLUSER(login, USER);
DBSETLAPP(login, "#t0022");
fprintf(stdout, "About to open %s.%s\n", SERVER, DATABASE);
/*
* One way to test the login timeout is to connect to a discard server (grep discard /etc/services).
* It's normally supported by inetd.
*/
printf ("using %d 1-second login timeouts\n", max_timeouts);
dbsetlogintime(1);
start_time = time(NULL); /* keep track of when we started for reporting purposes */
if (NULL == (dbproc = dbopen(login, SERVER))){
fprintf(stderr, "Failed: dbopen\n");
exit(1);
}
printf ("connected.\n");
if (strlen(DATABASE))
dbuse(dbproc, DATABASE);
dbloginfree(login);
/* send something that will take awhile to execute */
printf ("using %d %d-second query timeouts\n", max_timeouts, timeout_seconds);
if (FAIL == dbsettime(timeout_seconds)) {
fprintf(stderr, "Failed: dbsettime\n");
exit(1);
}
printf ("issuing a query that will take 30 seconds\n");
if (FAIL == sql_cmd(dbproc)) {
fprintf(stderr, "Failed: dbcmd\n");
exit(1);
}
start_time = time(NULL);
ntimeouts = 0;
dbsetinterrupt(dbproc, (void*)chkintr, (void*)hndlintr);
if (FAIL == dbsqlsend(dbproc)) {
fprintf(stderr, "Failed: dbsend\n");
exit(1);
}
/* wait for it to execute */
printf("executing dbsqlok\n");
erc = dbsqlok(dbproc);
if (erc != FAIL) {
fprintf(stderr, "dbsqlok should fail for timeout\n");
exit(1);
}
/* retrieve outputs per usual */
r = 0;
for (i=0; (erc = dbresults(dbproc)) != NO_MORE_RESULTS; i++) {
int nrows, ncols;
switch (erc) {
case SUCCEED:
if (DBROWS(dbproc) == FAIL){
r++;
continue;
}
assert(DBROWS(dbproc) == SUCCEED);
printf("dbrows() returned SUCCEED, processing rows\n");
ncols = dbnumcols(dbproc);
++num_resultset;
printf("bound 1 of %d columns ('%s') in result %d.\n", ncols, dbcolname(dbproc, 1), ++r);
dbbind(dbproc, 1, STRINGBIND, 0, (BYTE *) teststr);
printf("\t%s\n\t-----------\n", dbcolname(dbproc, 1));
while ((row_code = dbnextrow(dbproc)) != NO_MORE_ROWS) {
if (row_code == REG_ROW) {
printf("\t%s\n", teststr);
} else {
/* not supporting computed rows in this unit test */
failed = 1;
fprintf(stderr, "Failed. Expected a row\n");
exit(1);
}
}
nrows = (int) dbcount(dbproc);
printf("row count %d\n", nrows);
if (nrows < 0){
failed++;
printf("error: dbrows() returned SUCCEED, but dbcount() returned -1\n");
}
break;
case FAIL:
printf("dbresults returned FAIL\n");
exit(1);
default:
printf("unexpected return code %d from dbresults\n", erc);
exit(1);
}
if ( i > 1) {
failed++;
break;
}
} /* while dbresults */
dbexit();
fprintf(stdout, "%s %s\n", __FILE__, (failed ? "failed!" : "OK"));
return failed ? 1 : 0;
}
bool CDBL_Connection::x_SendData(I_BlobDescriptor& descr_in,
CDB_Stream& stream, bool log_it)
{
size_t size = stream.Size();
if (size < 1)
return false;
x_SetExtraMsg(descr_in, size);
I_BlobDescriptor* p_desc= 0;
// check what type of descriptor we've got
if(descr_in.DescriptorType() != CDBL_BLOB_DESCRIPTOR_TYPE_MAGNUM){
// this is not a native descriptor
p_desc = x_GetNativeBlobDescriptor
(dynamic_cast<CDB_BlobDescriptor&>(descr_in));
if(p_desc == 0) return false;
x_SetExtraMsg(*p_desc, size);
}
unique_ptr<I_BlobDescriptor> d_guard(p_desc);
CDBL_BlobDescriptor& desc
= p_desc ? dynamic_cast<CDBL_BlobDescriptor&>(*p_desc) :
dynamic_cast<CDBL_BlobDescriptor&>(descr_in);
char buff[1800]; // maximal page size
if (size <= sizeof(buff)) { // we could write a blob in one chunk
size_t s = stream.Read(buff, sizeof(buff));
if (Check(dbwritetext(GetDBLibConnection(), (char*) desc.m_ObjName.c_str(),
desc.m_TxtPtr_is_NULL ? 0 : desc.m_TxtPtr,
DBTXPLEN,
desc.m_TimeStamp_is_NULL ? 0 : desc.m_TimeStamp,
log_it ? TRUE : FALSE, (DBINT) s, (BYTE*) buff))
!= SUCCEED) {
DATABASE_DRIVER_ERROR( "dbwritetext failed." + GetDbgInfo(), 210030 );
}
return true;
}
// write it in chunks
if (Check(dbwritetext(GetDBLibConnection(), (char*) desc.m_ObjName.c_str(),
desc.m_TxtPtr_is_NULL ? 0 : desc.m_TxtPtr,
DBTXPLEN,
desc.m_TimeStamp_is_NULL ? 0 : desc.m_TimeStamp,
log_it ? TRUE : FALSE, (DBINT) size, 0)) != SUCCEED ||
Check(dbsqlok(GetDBLibConnection())) != SUCCEED ||
// dbresults(GetDBLibConnection()) == FAIL) {
x_Results(GetDBLibConnection()) == FAIL) {
DATABASE_DRIVER_ERROR( "dbwritetext/dbsqlok/dbresults failed." + GetDbgInfo(), 210031 );
}
while (size > 0) {
size_t s = stream.Read(buff, sizeof(buff));
if (s < 1) {
Check(dbcancel(GetDBLibConnection()));
DATABASE_DRIVER_ERROR("BLOB data corrupted." + GetDbgInfo(),
210032);
}
if (Check(dbmoretext(GetDBLibConnection(), (DBINT) s, (BYTE*) buff)) != SUCCEED) {
Check(dbcancel(GetDBLibConnection()));
DATABASE_DRIVER_ERROR( "dbmoretext failed." + GetDbgInfo(), 210033 );
}
size -= s;
}
// if (dbsqlok(GetDBLibConnection()) != SUCCEED || dbresults(GetDBLibConnection()) == FAIL) {
if (Check(dbsqlok(GetDBLibConnection())) != SUCCEED || x_Results(GetDBLibConnection()) == FAIL) {
DATABASE_DRIVER_ERROR( "dbsqlok/dbresults failed." + GetDbgInfo(), 210034 );
}
return true;
}
