convert.c 50.1 KB
Newer Older
1
/*-------
2
 * Module:               convert.c
3
 *
B
Bruce Momjian 已提交
4 5 6 7 8 9
 * Description:    This module contains routines related to
 *				   converting parameters and columns into requested data types.
 *				   Parameters are converted from their SQL_C data types into
 *				   the appropriate postgres type.  Columns are converted from
 *				   their postgres type (SQL type) into the appropriate SQL_C
 *				   data type.
10
 *
B
Bruce Momjian 已提交
11
 * Classes:		   n/a
12 13 14
 *
 * API functions:  none
 *
B
Bruce Momjian 已提交
15
 * Comments:	   See "notice.txt" for copyright and license information.
16
 *-------
17
 */
H
Hiroshi Inoue 已提交
18
/* Multibyte support  Eiji Tokuya	2001-03-15	*/
19

B
Byron Nikolaidis 已提交
20
#ifdef HAVE_CONFIG_H
21
#include "config.h"
B
Byron Nikolaidis 已提交
22 23
#endif

24 25
#include <stdio.h>
#include <string.h>
26
#include <ctype.h>
B
Byron Nikolaidis 已提交
27

B
 
Byron Nikolaidis 已提交
28 29
#include "psqlodbc.h"

H
Hiroshi Inoue 已提交
30 31 32 33
#ifdef MULTIBYTE
#include "multibyte.h"
#endif

34
#ifndef WIN32
B
Byron Nikolaidis 已提交
35 36 37 38
#include "iodbc.h"
#include "isql.h"
#include "isqlext.h"
#else
39
#include <windows.h>
40
#include <sql.h>
41
#include <sqlext.h>
B
Byron Nikolaidis 已提交
42 43
#endif

44
#include <time.h>
45
#include <math.h>
46
#include <stdlib.h>
47 48
#include "convert.h"
#include "statement.h"
B
Byron Nikolaidis 已提交
49
#include "qresult.h"
50 51
#include "bind.h"
#include "pgtypes.h"
52 53
#include "lobj.h"
#include "connection.h"
54
#include "pgapifunc.h"
55

56 57
#ifndef WIN32
#ifndef HAVE_STRICMP
B
Byron Nikolaidis 已提交
58 59 60 61 62
#define stricmp(s1,s2) strcasecmp(s1,s2)
#define strnicmp(s1,s2,n) strncasecmp(s1,s2,n)
#endif
#ifndef SCHAR
typedef signed char SCHAR;
B
Bruce Momjian 已提交
63

B
Byron Nikolaidis 已提交
64 65 66
#endif
#endif

67 68
extern GLOBAL_VALUES globals;

69 70 71 72
/*
 *	How to map ODBC scalar functions {fn func(args)} to Postgres.
 *	This is just a simple substitution.  List augmented from:
 *	http://www.merant.com/datadirect/download/docs/odbc16/Odbcref/rappc.htm
73
 *	- thomas 2000-04-03
74
 */
B
Bruce Momjian 已提交
75 76 77 78
char	   *mapFuncs[][2] = {
/*	{ "ASCII",		 "ascii"	  }, */
	{"CHAR", "chr"},
	{"CONCAT", "textcat"},
B
Bruce Momjian 已提交
79
/*	{ "DIFFERENCE",  "difference" }, */
B
Bruce Momjian 已提交
80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
/*	{ "INSERT",		 "insert"	  }, */
	{"LCASE", "lower"},
	{"LEFT", "ltrunc"},
	{"LOCATE", "strpos"},
	{"LENGTH", "char_length"},
/*	{ "LTRIM",		 "ltrim"	  }, */
	{"RIGHT", "rtrunc"},
/*	{ "REPEAT",		 "repeat"	  }, */
/*	{ "REPLACE",	 "replace"	  }, */
/*	{ "RTRIM",		 "rtrim"	  }, */
/*	{ "SOUNDEX",	 "soundex"	  }, */
	{"SUBSTRING", "substr"},
	{"UCASE", "upper"},

/*	{ "ABS",		 "abs"		  }, */
/*	{ "ACOS",		 "acos"		  }, */
/*	{ "ASIN",		 "asin"		  }, */
/*	{ "ATAN",		 "atan"		  }, */
/*	{ "ATAN2",		 "atan2"	  }, */
	{"CEILING", "ceil"},
/*	{ "COS",		 "cos"		  }, */
/*	{ "COT",		 "cot"		  }, */
/*	{ "DEGREES",	 "degrees"	  }, */
/*	{ "EXP",		 "exp"		  }, */
/*	{ "FLOOR",		 "floor"	  }, */
	{"LOG", "ln"},
	{"LOG10", "log"},
/*	{ "MOD",		 "mod"		  }, */
/*	{ "PI",			 "pi"		  }, */
	{"POWER", "pow"},
/*	{ "RADIANS",	 "radians"	  }, */
	{"RAND", "random"},
/*	{ "ROUND",		 "round"	  }, */
/*	{ "SIGN",		 "sign"		  }, */
/*	{ "SIN",		 "sin"		  }, */
/*	{ "SQRT",		 "sqrt"		  }, */
/*	{ "TAN",		 "tan"		  }, */
	{"TRUNCATE", "trunc"},

/*	{ "CURDATE",	 "curdate"	  }, */
/*	{ "CURTIME",	 "curtime"	  }, */
/*	{ "DAYNAME",	 "dayname"	  }, */
B
Bruce Momjian 已提交
122
/*	{ "DAYOFMONTH",  "dayofmonth" }, */
B
Bruce Momjian 已提交
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
/*	{ "DAYOFWEEK",	 "dayofweek"  }, */
/*	{ "DAYOFYEAR",	 "dayofyear"  }, */
/*	{ "HOUR",		 "hour"		  }, */
/*	{ "MINUTE",		 "minute"	  }, */
/*	{ "MONTH",		 "month"	  }, */
/*	{ "MONTHNAME",	 "monthname"  }, */
/*	{ "NOW",		 "now"		  }, */
/*	{ "QUARTER",	 "quarter"	  }, */
/*	{ "SECOND",		 "second"	  }, */
/*	{ "WEEK",		 "week"		  }, */
/*	{ "YEAR",		 "year"		  }, */

/*	{ "DATABASE",	 "database"   }, */
	{"IFNULL", "coalesce"},
	{"USER", "odbc_user"},
	{0, 0}
139 140
};

B
Bruce Momjian 已提交
141
char	   *mapFunction(char *func);
142 143
unsigned int conv_from_octal(unsigned char *s);
unsigned int conv_from_hex(unsigned char *s);
B
Bruce Momjian 已提交
144
char	   *conv_to_octal(unsigned char val);
145

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
/*---------
 *			A Guide for date/time/timestamp conversions
 *
 *			field_type		fCType				Output
 *			----------		------				----------
 *			PG_TYPE_DATE	SQL_C_DEFAULT		SQL_C_DATE
 *			PG_TYPE_DATE	SQL_C_DATE			SQL_C_DATE
 *			PG_TYPE_DATE	SQL_C_TIMESTAMP		SQL_C_TIMESTAMP		(time = 0 (midnight))
 *			PG_TYPE_TIME	SQL_C_DEFAULT		SQL_C_TIME
 *			PG_TYPE_TIME	SQL_C_TIME			SQL_C_TIME
 *			PG_TYPE_TIME	SQL_C_TIMESTAMP		SQL_C_TIMESTAMP		(date = current date)
 *			PG_TYPE_ABSTIME SQL_C_DEFAULT		SQL_C_TIMESTAMP
 *			PG_TYPE_ABSTIME SQL_C_DATE			SQL_C_DATE			(time is truncated)
 *			PG_TYPE_ABSTIME SQL_C_TIME			SQL_C_TIME			(date is truncated)
 *			PG_TYPE_ABSTIME SQL_C_TIMESTAMP		SQL_C_TIMESTAMP
 *---------
 */
163 164 165 166 167



/*	This is called by SQLFetch() */
int
B
Bruce Momjian 已提交
168
copy_and_convert_field_bindinfo(StatementClass *stmt, Int4 field_type, void *value, int col)
169
{
B
Bruce Momjian 已提交
170
	BindInfoClass *bic = &(stmt->bindings[col]);
171

B
Bruce Momjian 已提交
172 173
	return copy_and_convert_field(stmt, field_type, value, (Int2) bic->returntype, (PTR) bic->buffer,
							 (SDWORD) bic->buflen, (SDWORD *) bic->used);
174 175
}

176

177 178
/*	This is called by SQLGetData() */
int
B
Bruce Momjian 已提交
179
copy_and_convert_field(StatementClass *stmt, Int4 field_type, void *value, Int2 fCType,
B
Bruce Momjian 已提交
180
					   PTR rgbValue, SDWORD cbValueMax, SDWORD *pcbValue)
181
{
B
Bruce Momjian 已提交
182 183
	Int4		len = 0,
				copy_len = 0;
184
	SIMPLE_TIME st;
B
Bruce Momjian 已提交
185 186 187 188 189 190 191 192 193
	time_t		t = time(NULL);
	struct tm  *tim;
	int			pcbValueOffset,
				rgbValueOffset;
	char	   *rgbValueBindRow,
			   *ptr;
	int			bind_row = stmt->bind_row;
	int			bind_size = stmt->options.bind_size;
	int			result = COPY_OK;
194 195 196
	BOOL		changed;
	static		char *tempBuf= NULL;
	static		unsigned int tempBuflen = 0;
197

198 199
	if (!tempBuf)
		tempBuflen = 0;
200 201 202 203 204
	/*---------
	 *	rgbValueOffset is *ONLY* for character and binary data.
	 *	pcbValueOffset is for computing any pcbValue location
	 *---------
	 */
205

B
Bruce Momjian 已提交
206
	if (bind_size > 0)
207
		pcbValueOffset = rgbValueOffset = (bind_size * bind_row);
B
Bruce Momjian 已提交
208 209
	else
	{
210 211
		pcbValueOffset = bind_row * sizeof(SDWORD);
		rgbValueOffset = bind_row * cbValueMax;
212

213
	}
214 215 216

	memset(&st, 0, sizeof(SIMPLE_TIME));

B
Bruce Momjian 已提交
217
	/* Initialize current date */
218 219 220 221 222
	tim = localtime(&t);
	st.m = tim->tm_mon + 1;
	st.d = tim->tm_mday;
	st.y = tim->tm_year + 1900;

B
Bruce Momjian 已提交
223
	mylog("copy_and_convert: field_type = %d, fctype = %d, value = '%s', cbValueMax=%d\n", field_type, fCType, (value == NULL) ? "<NULL>" : value, cbValueMax);
224

B
Bruce Momjian 已提交
225 226
	if (!value)
	{
227

228
		/*
229 230
		 * handle a null just by returning SQL_NULL_DATA in pcbValue, and
		 * doing nothing to the buffer.
231
		 */
B
Bruce Momjian 已提交
232
		if (pcbValue)
233
			*(SDWORD *) ((char *) pcbValue + pcbValueOffset) = SQL_NULL_DATA;
B
Byron Nikolaidis 已提交
234 235 236
		return COPY_OK;
	}

B
Bruce Momjian 已提交
237 238 239 240 241 242 243 244 245
	if (stmt->hdbc->DataSourceToDriver != NULL)
	{
		int			length = strlen(value);

		stmt->hdbc->DataSourceToDriver(stmt->hdbc->translation_option,
									   SQL_CHAR,
									   value, length,
									   value, length, NULL,
									   NULL, 0, NULL);
B
Byron Nikolaidis 已提交
246 247
	}

248
	/*
249
	 * First convert any specific postgres types into more useable data.
250
	 *
251 252
	 * NOTE: Conversions from PG char/varchar of a date/time/timestamp value
	 * to SQL_C_DATE,SQL_C_TIME, SQL_C_TIMESTAMP not supported
253
	 */
B
Bruce Momjian 已提交
254 255
	switch (field_type)
	{
256 257 258 259 260

			/*
			 * $$$ need to add parsing for date/time/timestamp strings in
			 * PG_TYPE_CHAR,VARCHAR $$$
			 */
B
Bruce Momjian 已提交
261 262 263
		case PG_TYPE_DATE:
			sscanf(value, "%4d-%2d-%2d", &st.y, &st.m, &st.d);
			break;
264

B
Bruce Momjian 已提交
265 266 267
		case PG_TYPE_TIME:
			sscanf(value, "%2d:%2d:%2d", &st.hh, &st.mm, &st.ss);
			break;
268

B
Bruce Momjian 已提交
269 270 271 272 273 274
		case PG_TYPE_ABSTIME:
		case PG_TYPE_DATETIME:
		case PG_TYPE_TIMESTAMP:
			if (strnicmp(value, "invalid", 7) != 0)
				sscanf(value, "%4d-%2d-%2d %2d:%2d:%2d", &st.y, &st.m, &st.d, &st.hh, &st.mm, &st.ss);
			else
275
			{
276

277
				/*
278 279
				 * The timestamp is invalid so set something conspicuous,
				 * like the epoch
280
				 */
B
Bruce Momjian 已提交
281 282 283 284 285 286 287 288 289 290 291 292 293 294 295
				t = 0;
				tim = localtime(&t);
				st.m = tim->tm_mon + 1;
				st.d = tim->tm_mday;
				st.y = tim->tm_year + 1900;
				st.hh = tim->tm_hour;
				st.mm = tim->tm_min;
				st.ss = tim->tm_sec;
			}
			break;

		case PG_TYPE_BOOL:
			{					/* change T/F to 1/0 */
				char	   *s = (char *) value;

296 297 298 299 300 301 302 303 304 305 306 307 308
				/*  Aidan Mountford (aidan@oz.to) 1/08/2001:  
					
					>> if (s[0] == 'T' || s[0] == 't') <<< This wont work...

					When MoveFirst is called twice on one set of tuples, 
					this will have the effect of setting s[0] to 1 on the
					first pass, and s[0] on the second.

					This is bad ;)

				*/
					
				if (s[0] == 'T' || s[0] == 't' || s[0] == '1')
B
Bruce Momjian 已提交
309 310
					s[0] = '1';
				else
311 312
				    s[0] = '0';

B
Bruce Momjian 已提交
313 314 315
			}
			break;

316
			/* This is for internal use by SQLStatistics() */
B
Bruce Momjian 已提交
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350
		case PG_TYPE_INT2VECTOR:
			{
				int			nval,
							i;
				char	   *vp;

				/* this is an array of eight integers */
				short	   *short_array = (short *) ((char *) rgbValue + rgbValueOffset);

				len = 32;
				vp = value;
				nval = 0;
				mylog("index=(");
				for (i = 0; i < 16; i++)
				{
					if (sscanf(vp, "%hd", &short_array[i]) != 1)
						break;

					mylog(" %d", short_array[i]);
					nval++;

					/* skip the current token */
					while ((*vp != '\0') && (!isspace((unsigned char) *vp)))
						vp++;
					/* and skip the space to the next token */
					while ((*vp != '\0') && (isspace((unsigned char) *vp)))
						vp++;
					if (*vp == '\0')
						break;
				}
				mylog(") nval = %d\n", nval);

				for (i = nval; i < 16; i++)
					short_array[i] = 0;
B
Byron Nikolaidis 已提交
351

352
#if 0
B
Bruce Momjian 已提交
353 354 355 356 357 358 359 360 361
				sscanf(value, "%hd %hd %hd %hd %hd %hd %hd %hd",
					   &short_array[0],
					   &short_array[1],
					   &short_array[2],
					   &short_array[3],
					   &short_array[4],
					   &short_array[5],
					   &short_array[6],
					   &short_array[7]);
362
#endif
B
Byron Nikolaidis 已提交
363

B
Bruce Momjian 已提交
364 365 366
				/* There is no corresponding fCType for this. */
				if (pcbValue)
					*(SDWORD *) ((char *) pcbValue + pcbValueOffset) = len;
B
Byron Nikolaidis 已提交
367

B
Bruce Momjian 已提交
368 369 370
				return COPY_OK; /* dont go any further or the data will be
								 * trashed */
			}
B
Byron Nikolaidis 已提交
371

B
Bruce Momjian 已提交
372 373 374 375 376
			/*
			 * This is a large object OID, which is used to store
			 * LONGVARBINARY objects.
			 */
		case PG_TYPE_LO:
B
Byron Nikolaidis 已提交
377

B
Bruce Momjian 已提交
378
			return convert_lo(stmt, value, fCType, ((char *) rgbValue + rgbValueOffset), cbValueMax, (SDWORD *) ((char *) pcbValue + pcbValueOffset));
B
Byron Nikolaidis 已提交
379

B
Bruce Momjian 已提交
380
		default:
B
Byron Nikolaidis 已提交
381

B
Bruce Momjian 已提交
382 383 384
			if (field_type == stmt->hdbc->lobj_type)	/* hack until permanent
														 * type available */
				return convert_lo(stmt, value, fCType, ((char *) rgbValue + rgbValueOffset), cbValueMax, (SDWORD *) ((char *) pcbValue + pcbValueOffset));
B
Byron Nikolaidis 已提交
385 386
	}

B
Bruce Momjian 已提交
387 388 389
	/* Change default into something useable */
	if (fCType == SQL_C_DEFAULT)
	{
B
Byron Nikolaidis 已提交
390 391 392 393 394
		fCType = pgtype_to_ctype(stmt, field_type);

		mylog("copy_and_convert, SQL_C_DEFAULT: fCType = %d\n", fCType);
	}

395 396
	rgbValueBindRow = (char *) rgbValue + rgbValueOffset;

B
Bruce Momjian 已提交
397 398 399
	if (fCType == SQL_C_CHAR)
	{
		/* Special character formatting as required */
400

B
Bruce Momjian 已提交
401 402 403 404 405 406 407 408 409 410 411
		/*
		 * These really should return error if cbValueMax is not big
		 * enough.
		 */
		switch (field_type)
		{
			case PG_TYPE_DATE:
				len = 10;
				if (cbValueMax > len)
					sprintf(rgbValueBindRow, "%.4d-%.2d-%.2d", st.y, st.m, st.d);
				break;
412

B
Bruce Momjian 已提交
413 414 415 416 417
			case PG_TYPE_TIME:
				len = 8;
				if (cbValueMax > len)
					sprintf(rgbValueBindRow, "%.2d:%.2d:%.2d", st.hh, st.mm, st.ss);
				break;
418

B
Bruce Momjian 已提交
419 420 421 422 423 424 425 426
			case PG_TYPE_ABSTIME:
			case PG_TYPE_DATETIME:
			case PG_TYPE_TIMESTAMP:
				len = 19;
				if (cbValueMax > len)
					sprintf(rgbValueBindRow, "%.4d-%.2d-%.2d %.2d:%.2d:%.2d",
							st.y, st.m, st.d, st.hh, st.mm, st.ss);
				break;
427

B
Bruce Momjian 已提交
428 429 430 431 432 433 434 435
			case PG_TYPE_BOOL:
				len = 1;
				if (cbValueMax > len)
				{
					strcpy(rgbValueBindRow, value);
					mylog("PG_TYPE_BOOL: rgbValueBindRow = '%s'\n", rgbValueBindRow);
				}
				break;
436

B
Bruce Momjian 已提交
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455
				/*
				 * Currently, data is SILENTLY TRUNCATED for BYTEA and
				 * character data types if there is not enough room in
				 * cbValueMax because the driver can't handle multiple
				 * calls to SQLGetData for these, yet.	Most likely, the
				 * buffer passed in will be big enough to handle the
				 * maximum limit of postgres, anyway.
				 *
				 * LongVarBinary types are handled correctly above, observing
				 * truncation and all that stuff since there is
				 * essentially no limit on the large object used to store
				 * those.
				 */
			case PG_TYPE_BYTEA:/* convert binary data to hex strings
								 * (i.e, 255 = "FF") */
				len = convert_pgbinary_to_char(value, rgbValueBindRow, cbValueMax);

				/***** THIS IS NOT PROPERLY IMPLEMENTED *****/
				break;
456

B
Bruce Momjian 已提交
457
			default:
458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490
				if (stmt->current_col >= 0 && stmt->bindings[stmt->current_col].data_left == -2)
					stmt->bindings[stmt->current_col].data_left = (cbValueMax > 0) ? 0 : -1; /* This seems to be needed for ADO ? */
				if (stmt->current_col < 0 || stmt->bindings[stmt->current_col].data_left < 0)
				{
					/* convert linefeeds to carriage-return/linefeed */
					len = convert_linefeeds(value, NULL, 0, &changed);
					if (cbValueMax == 0) /* just returns length info */
					{
						result = COPY_RESULT_TRUNCATED;
						break;
					}
					if (changed || len >= cbValueMax)
					{
						if (len >= (int) tempBuflen)
						{
							tempBuf = realloc(tempBuf, len + 1);
							tempBuflen = len + 1;
						}
						convert_linefeeds(value, tempBuf, tempBuflen, &changed);
						ptr = tempBuf;
					}
					else
					{
						if (tempBuf)
						{
							free(tempBuf);
							tempBuf = NULL;
						}
						ptr = value;
					}
				}
				else
					ptr = tempBuf;
B
Bruce Momjian 已提交
491 492 493 494 495 496

				mylog("DEFAULT: len = %d, ptr = '%s'\n", len, ptr);

				if (stmt->current_col >= 0)
				{
					if (stmt->bindings[stmt->current_col].data_left == 0)
497 498 499 500 501 502 503 504
					{
						if (tempBuf)
						{
							free(tempBuf);
							tempBuf = NULL;
						}
						/* The following seems to be needed for ADO ? */
						stmt->bindings[stmt->current_col].data_left = -2;
B
Bruce Momjian 已提交
505
						return COPY_NO_DATA_FOUND;
506
					}
B
Bruce Momjian 已提交
507 508
					else if (stmt->bindings[stmt->current_col].data_left > 0)
					{
509
						ptr += strlen(ptr) - stmt->bindings[stmt->current_col].data_left;
B
Bruce Momjian 已提交
510 511 512
						len = stmt->bindings[stmt->current_col].data_left;
					}
					else
513
						stmt->bindings[stmt->current_col].data_left = len;
514 515
				}

B
Bruce Momjian 已提交
516 517 518
				if (cbValueMax > 0)
				{
					copy_len = (len >= cbValueMax) ? cbValueMax - 1 : len;
519

B
Bruce Momjian 已提交
520
					/* Copy the data */
521 522
					memcpy(rgbValueBindRow, ptr, copy_len);
					rgbValueBindRow[copy_len] = '\0';
523

B
Bruce Momjian 已提交
524 525 526
					/* Adjust data_left for next time */
					if (stmt->current_col >= 0)
						stmt->bindings[stmt->current_col].data_left -= copy_len;
527 528
				}

B
Bruce Momjian 已提交
529 530 531 532
				/*
				 * Finally, check for truncation so that proper status can
				 * be returned
				 */
533
				if (cbValueMax > 0 && len >= cbValueMax)
B
Bruce Momjian 已提交
534
					result = COPY_RESULT_TRUNCATED;
535 536 537 538 539 540 541 542
				else
				{
					if (tempBuf)
					{
						free(tempBuf);
						tempBuf = NULL;
					}
				}
543 544


B
Bruce Momjian 已提交
545 546
				mylog("    SQL_C_CHAR, default: len = %d, cbValueMax = %d, rgbValueBindRow = '%s'\n", len, cbValueMax, rgbValueBindRow);
				break;
B
Byron Nikolaidis 已提交
547
		}
548 549


B
Bruce Momjian 已提交
550 551 552
	}
	else
	{
553

B
Bruce Momjian 已提交
554 555 556 557 558
		/*
		 * for SQL_C_CHAR, it's probably ok to leave currency symbols in.
		 * But to convert to numeric types, it is necessary to get rid of
		 * those.
		 */
B
Byron Nikolaidis 已提交
559 560 561
		if (field_type == PG_TYPE_MONEY)
			convert_money(value);

B
Bruce Momjian 已提交
562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577
		switch (fCType)
		{
			case SQL_C_DATE:
				len = 6;
				{
					DATE_STRUCT *ds;

					if (bind_size > 0)
						ds = (DATE_STRUCT *) ((char *) rgbValue + (bind_row * bind_size));
					else
						ds = (DATE_STRUCT *) rgbValue + bind_row;
					ds->year = st.y;
					ds->month = st.m;
					ds->day = st.d;
				}
				break;
578

B
Bruce Momjian 已提交
579 580 581 582 583 584 585 586 587 588 589 590 591 592
			case SQL_C_TIME:
				len = 6;
				{
					TIME_STRUCT *ts;

					if (bind_size > 0)
						ts = (TIME_STRUCT *) ((char *) rgbValue + (bind_row * bind_size));
					else
						ts = (TIME_STRUCT *) rgbValue + bind_row;
					ts->hour = st.hh;
					ts->minute = st.mm;
					ts->second = st.ss;
				}
				break;
B
Byron Nikolaidis 已提交
593

B
Bruce Momjian 已提交
594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611
			case SQL_C_TIMESTAMP:
				len = 16;
				{
					TIMESTAMP_STRUCT *ts;

					if (bind_size > 0)
						ts = (TIMESTAMP_STRUCT *) ((char *) rgbValue + (bind_row * bind_size));
					else
						ts = (TIMESTAMP_STRUCT *) rgbValue + bind_row;
					ts->year = st.y;
					ts->month = st.m;
					ts->day = st.d;
					ts->hour = st.hh;
					ts->minute = st.mm;
					ts->second = st.ss;
					ts->fraction = 0;
				}
				break;
612

B
Bruce Momjian 已提交
613 614 615 616 617 618
			case SQL_C_BIT:
				len = 1;
				if (bind_size > 0)
					*(UCHAR *) ((char *) rgbValue + (bind_row * bind_size)) = atoi(value);
				else
					*((UCHAR *) rgbValue + bind_row) = atoi(value);
619

B
Bruce Momjian 已提交
620 621 622 623 624
				/*
				 * mylog("SQL_C_BIT: val = %d, cb = %d, rgb=%d\n",
				 * atoi(value), cbValueMax, *((UCHAR *)rgbValue));
				 */
				break;
625

B
Bruce Momjian 已提交
626 627 628 629 630 631 632 633
			case SQL_C_STINYINT:
			case SQL_C_TINYINT:
				len = 1;
				if (bind_size > 0)
					*(SCHAR *) ((char *) rgbValue + (bind_row * bind_size)) = atoi(value);
				else
					*((SCHAR *) rgbValue + bind_row) = atoi(value);
				break;
634

B
Bruce Momjian 已提交
635 636 637 638 639 640 641
			case SQL_C_UTINYINT:
				len = 1;
				if (bind_size > 0)
					*(UCHAR *) ((char *) rgbValue + (bind_row * bind_size)) = atoi(value);
				else
					*((UCHAR *) rgbValue + bind_row) = atoi(value);
				break;
642

B
Bruce Momjian 已提交
643 644 645 646 647 648 649
			case SQL_C_FLOAT:
				len = 4;
				if (bind_size > 0)
					*(SFLOAT *) ((char *) rgbValue + (bind_row * bind_size)) = (float) atof(value);
				else
					*((SFLOAT *) rgbValue + bind_row) = (float) atof(value);
				break;
650

B
Bruce Momjian 已提交
651 652 653 654 655 656 657
			case SQL_C_DOUBLE:
				len = 8;
				if (bind_size > 0)
					*(SDOUBLE *) ((char *) rgbValue + (bind_row * bind_size)) = atof(value);
				else
					*((SDOUBLE *) rgbValue + bind_row) = atof(value);
				break;
658

B
Bruce Momjian 已提交
659 660 661 662 663 664 665 666
			case SQL_C_SSHORT:
			case SQL_C_SHORT:
				len = 2;
				if (bind_size > 0)
					*(SWORD *) ((char *) rgbValue + (bind_row * bind_size)) = atoi(value);
				else
					*((SWORD *) rgbValue + bind_row) = atoi(value);
				break;
667

B
Bruce Momjian 已提交
668 669 670 671 672 673 674
			case SQL_C_USHORT:
				len = 2;
				if (bind_size > 0)
					*(UWORD *) ((char *) rgbValue + (bind_row * bind_size)) = atoi(value);
				else
					*((UWORD *) rgbValue + bind_row) = atoi(value);
				break;
675

B
Bruce Momjian 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691
			case SQL_C_SLONG:
			case SQL_C_LONG:
				len = 4;
				if (bind_size > 0)
					*(SDWORD *) ((char *) rgbValue + (bind_row * bind_size)) = atol(value);
				else
					*((SDWORD *) rgbValue + bind_row) = atol(value);
				break;

			case SQL_C_ULONG:
				len = 4;
				if (bind_size > 0)
					*(UDWORD *) ((char *) rgbValue + (bind_row * bind_size)) = atol(value);
				else
					*((UDWORD *) rgbValue + bind_row) = atol(value);
				break;
692

B
Bruce Momjian 已提交
693
			case SQL_C_BINARY:
694

B
Bruce Momjian 已提交
695 696
				/* truncate if necessary */
				/* convert octal escapes to bytes */
697

698 699 700 701 702 703
				if (len = strlen(value), len >= (int) tempBuflen)
				{
					tempBuf = realloc(tempBuf, len + 1);
					tempBuflen = len + 1;
				}
				len = convert_from_pgbinary(value, tempBuf, tempBuflen);
B
Bruce Momjian 已提交
704
				ptr = tempBuf;
705

B
Bruce Momjian 已提交
706 707 708 709
				if (stmt->current_col >= 0)
				{
					/* No more data left for this column */
					if (stmt->bindings[stmt->current_col].data_left == 0)
710 711 712
					{
						free(tempBuf);
						tempBuf = NULL;
B
Bruce Momjian 已提交
713
						return COPY_NO_DATA_FOUND;
714
					}
715

B
Bruce Momjian 已提交
716 717 718 719 720 721 722 723 724
					/*
					 * Second (or more) call to SQLGetData so move the
					 * pointer
					 */
					else if (stmt->bindings[stmt->current_col].data_left > 0)
					{
						ptr += len - stmt->bindings[stmt->current_col].data_left;
						len = stmt->bindings[stmt->current_col].data_left;
					}
725

B
Bruce Momjian 已提交
726 727 728
					/* First call to SQLGetData so initialize data_left */
					else
						stmt->bindings[stmt->current_col].data_left = len;
729

B
Bruce Momjian 已提交
730
				}
731

B
Bruce Momjian 已提交
732 733 734
				if (cbValueMax > 0)
				{
					copy_len = (len > cbValueMax) ? cbValueMax : len;
735

B
Bruce Momjian 已提交
736 737
					/* Copy the data */
					memcpy(rgbValueBindRow, ptr, copy_len);
738

B
Bruce Momjian 已提交
739 740 741
					/* Adjust data_left for next time */
					if (stmt->current_col >= 0)
						stmt->bindings[stmt->current_col].data_left -= copy_len;
742
				}
743

B
Bruce Momjian 已提交
744 745 746 747 748 749
				/*
				 * Finally, check for truncation so that proper status can
				 * be returned
				 */
				if (len > cbValueMax)
					result = COPY_RESULT_TRUNCATED;
750

751 752 753 754 755
				if (tempBuf)
				{
					free(tempBuf);
					tempBuf = NULL;
				}
B
Bruce Momjian 已提交
756 757 758 759 760
				mylog("SQL_C_BINARY: len = %d, copy_len = %d\n", len, copy_len);
				break;

			default:
				return COPY_UNSUPPORTED_TYPE;
B
Byron Nikolaidis 已提交
761 762
		}
	}
763

B
Bruce Momjian 已提交
764 765 766
	/* store the length of what was copied, if there's a place for it */
	if (pcbValue)
		*(SDWORD *) ((char *) pcbValue + pcbValueOffset) = len;
767

768
	return result;
769

770 771
}

772

773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943
/*--------------------------------------------------------------------
 *	Functions/Macros to get rid of query size limit.
 *
 *	I always used the follwoing macros to convert from
 *	old_statement to new_statement.	 Please improve it
 *	if you have a better way.	Hiroshi 2001/05/22
 *--------------------------------------------------------------------
 */
#define	INIT_MIN_ALLOC	4096
static int enlarge_statement(StatementClass *stmt, unsigned int newsize)
{
	unsigned int	newalsize = INIT_MIN_ALLOC;
	static char *func = "enlarge_statement";

	if (stmt->stmt_size_limit > 0 && stmt->stmt_size_limit < (int) newsize)
	{
		stmt->errormsg = "Query buffer overflow in copy_statement_with_parameters";
		stmt->errornumber = STMT_EXEC_ERROR;
		SC_log_error(func, "", stmt);
		return -1;
	}
	while (newalsize <= newsize)
		newalsize *= 2;
	if (!(stmt->stmt_with_params = realloc(stmt->stmt_with_params, newalsize)))
	{
		stmt->errormsg = "Query buffer allocate error in copy_statement_with_parameters";
		stmt->errornumber = STMT_EXEC_ERROR;
		SC_log_error(func, "", stmt);
		return 0;
	}
	return newalsize;
}

/*----------
 *	Enlarge stmt_with_params if necessary.
 *----------
 */
#define	ENLARGE_NEWSTATEMENT(newpos) \
	if (newpos >= new_stsize) \
	{ \
		if ((new_stsize = enlarge_statement(stmt, newpos)) <= 0) \
			return SQL_ERROR; \
		new_statement = stmt->stmt_with_params; \
	}
/*----------
 *	Initialize stmt_with_params, new_statement etc.
 *----------
 */
#define	CVT_INIT(size) \
{ \
	if (stmt->stmt_with_params) \
		free(stmt->stmt_with_params); \
	if (stmt->stmt_size_limit > 0) \
		new_stsize = stmt->stmt_size_limit; \
	else \
	{ \
		new_stsize = INIT_MIN_ALLOC; \
		while (new_stsize <= size) \
			new_stsize *= 2; \
	} \
	new_statement = malloc(new_stsize); \
	stmt->stmt_with_params = new_statement; \
	npos = 0; \
	new_statement[0] = '\0'; \
}
/*----------
 *	Terminate the stmt_with_params string with NULL.
 *----------
 */
#define	CVT_TERMINATE { new_statement[npos] = '\0'; }

/*----------
 *	Append a data.
 *----------
 */
#define	CVT_APPEND_DATA(s, len) \
{ \
	unsigned int	newpos = npos + len; \
	ENLARGE_NEWSTATEMENT(newpos) \
	memcpy(&new_statement[npos], s, len); \
	npos = newpos; \
	new_statement[npos] = '\0'; \
}
/*----------
 *	Append a string.
 *----------
 */
#define	CVT_APPEND_STR(s) \
{ \
	unsigned int len = strlen(s); \
	CVT_APPEND_DATA(s, len); \
}
/*----------
 *	Append a char.	
 *----------
 */
#define	CVT_APPEND_CHAR(c) \
{ \
	ENLARGE_NEWSTATEMENT(npos + 1); \
	new_statement[npos++] = c; \
}
/*----------
 *	Append a binary data.
 *	Newly reqeuired size may be overestimated currently. 
 *----------
 */
#define	CVT_APPEND_BINARY(buf, used) \
{ \
	unsigned int	newlimit = npos + 5 * used; \
	ENLARGE_NEWSTATEMENT(newlimit); \
	npos += convert_to_pgbinary(buf, &new_statement[npos], used); \
}
/*----------
 *
 *----------
 */
#define	CVT_SPECIAL_CHARS(buf, used) \
{ \
	int	cnvlen = convert_special_chars(buf, NULL, used); \
	unsigned int	newlimit = npos + cnvlen; \
\
	ENLARGE_NEWSTATEMENT(newlimit); \
	convert_special_chars(buf, &new_statement[npos], used); \
	npos += cnvlen; \
}

/*----------
 *	Check if the statement is	
 *	SELECT ... INTO table FROM .....
 *	This isn't really a strict check but ...
 *---------- 
 */
static BOOL
into_table_from(const char *stmt)
{
	if (strnicmp(stmt, "into", 4))
		return FALSE;
	stmt += 4;
	if (!isspace((unsigned char) *stmt))
		return FALSE;
	while (isspace((unsigned char) *(++stmt)));
	switch (*stmt)
	{
		case '\0':
		case ',':
		case '\'':
			return FALSE;
		case '\"': /* double quoted table name ? */
			do
			{
				do
				{
					 while (*(++stmt) != '\"' && *stmt);
				}
				while (*stmt && *(++stmt) == '\"');
				while (*stmt && !isspace((unsigned char) *stmt) && *stmt != '\"') stmt++;
			}
			while (*stmt == '\"');
			break;
		default:
			while (!isspace((unsigned char) *(++stmt)));
			break;
	}
	if (! *stmt)
		return FALSE;
	while (isspace((unsigned char) *(++stmt)));
	if (strnicmp(stmt, "from", 4))
		return FALSE;
	return isspace((unsigned char) stmt[4]);
}

944 945 946
/*
 *	This function inserts parameters into an SQL statements.
 *	It will also modify a SELECT statement for use with declare/fetch cursors.
947
 *	This function does a dynamic memory allocation to get rid of query siz elimit!
948
 */
949
int
B
Bruce Momjian 已提交
950
copy_statement_with_parameters(StatementClass *stmt)
951
{
B
Bruce Momjian 已提交
952 953 954 955 956 957
	static char *func = "copy_statement_with_parameters";
	unsigned int opos,
				npos,
				oldstmtlen;
	char		param_string[128],
				tmp[256],
958
				cbuf[PG_NUMERIC_MAX_PRECISION * 2]; /* seems big enough to handle the data in this function */
B
Bruce Momjian 已提交
959 960 961
	int			param_number;
	Int2		param_ctype,
				param_sqltype;
962
	char	   *old_statement = stmt->statement, oldchar;
B
Bruce Momjian 已提交
963
	char	   *new_statement = stmt->stmt_with_params;
964
	unsigned int	new_stsize = 0;
B
Bruce Momjian 已提交
965 966 967 968 969 970
	SIMPLE_TIME st;
	time_t		t = time(NULL);
	struct tm  *tim;
	SDWORD		used;
	char	   *buffer,
			   *buf;
971
	BOOL		in_quote = FALSE, in_dquote = FALSE, in_escape = FALSE;
B
Bruce Momjian 已提交
972 973 974
	Oid			lobj_oid;
	int			lobj_fd,
				retval;
975 976
	BOOL	check_select_into = FALSE; /* select into check */
	unsigned int	declare_pos;
977 978 979
#ifdef	DRIVER_CURSOR_IMPLEMENT
	BOOL ins_ctrl = FALSE;
#endif /* DRIVER_CURSOR_IMPLEMENT */
B
Bruce Momjian 已提交
980 981 982 983


	if (!old_statement)
	{
B
Byron Nikolaidis 已提交
984
		SC_log_error(func, "No statement string", stmt);
985
		return SQL_ERROR;
B
Byron Nikolaidis 已提交
986
	}
987

988 989
	memset(&st, 0, sizeof(SIMPLE_TIME));

B
Bruce Momjian 已提交
990
	/* Initialize current date */
991 992 993 994 995
	tim = localtime(&t);
	st.m = tim->tm_mon + 1;
	st.d = tim->tm_mday;
	st.y = tim->tm_year + 1900;

B
Bruce Momjian 已提交
996 997
	/* If the application hasn't set a cursor name, then generate one */
	if (stmt->cursor_name[0] == '\0')
998
		sprintf(stmt->cursor_name, "SQL_CUR%p", stmt);
999

1000 1001
	oldstmtlen = strlen(old_statement);
	CVT_INIT(oldstmtlen);
B
Bruce Momjian 已提交
1002 1003 1004
	/* For selects, prepend a declare cursor to the statement */
	if (stmt->statement_type == STMT_TYPE_SELECT && globals.use_declarefetch)
	{
1005
		sprintf(new_statement, "declare %s cursor for ", stmt->cursor_name);
1006
		npos = strlen(new_statement);
1007 1008
		check_select_into = TRUE;
		declare_pos = npos;
1009
	}
B
Bruce Momjian 已提交
1010
	param_number = -1;
1011

1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
#ifdef	DRIVER_CURSOR_IMPLEMENT
	if (stmt->statement_type != STMT_TYPE_SELECT)
	{
		stmt->options.cursor_type = SQL_CURSOR_FORWARD_ONLY;
		stmt->options.scroll_concurrency = SQL_CONCUR_READ_ONLY;
	}
	else if (stmt->options.cursor_type == SQL_CURSOR_FORWARD_ONLY)
    		stmt->options.scroll_concurrency = SQL_CONCUR_READ_ONLY;
	else if (stmt->options.scroll_concurrency != SQL_CONCUR_READ_ONLY)
	{
		if (stmt->parse_status == STMT_PARSE_NONE)
			parse_statement(stmt);
		if (stmt->parse_status != STMT_PARSE_COMPLETE)
			stmt->options.scroll_concurrency = SQL_CONCUR_READ_ONLY;
		else if (!stmt->ti || stmt->ntab != 1)
    			stmt->options.scroll_concurrency = SQL_CONCUR_READ_ONLY;
		else ins_ctrl = TRUE;
	}
#endif /* DRIVER_CURSOR_IMPLEMENT */
H
Hiroshi Inoue 已提交
1031
#ifdef MULTIBYTE
B
Bruce Momjian 已提交
1032
	multibyte_init();
H
Hiroshi Inoue 已提交
1033
#endif
1034

B
Bruce Momjian 已提交
1035 1036
	for (opos = 0; opos < oldstmtlen; opos++)
	{
1037
		oldchar = old_statement[opos];
1038
#ifdef MULTIBYTE
1039
		if (multibyte_char_check(oldchar) != 0)
1040
		{
1041
			CVT_APPEND_CHAR(oldchar);
1042 1043 1044 1045 1046 1047 1048
			continue;
		}
		/*
		 *	From here we are guaranteed to handle a
		 *	1-byte character.
		 */
#endif
1049

1050
		if (in_escape) /* escape check */
1051 1052
		{
			in_escape = FALSE;
1053
			CVT_APPEND_CHAR(oldchar);
1054 1055 1056 1057
			continue;
		}	
		else if (in_quote || in_dquote) /* quote/double quote check */
		{
1058 1059 1060
			if (oldchar == '\\')
				in_escape = TRUE;
			else if (oldchar == '\'' && in_quote)
1061
				in_quote = FALSE;
1062
			else if (oldchar == '\"' && in_dquote)
1063
				in_dquote = FALSE;
1064
			CVT_APPEND_CHAR(oldchar);
1065 1066 1067 1068
			continue;	
		}
		/*
		 *	From here we are guranteed to be in neither
1069
		 *	an escape, a quote nor a double quote.
1070
		 */
1071 1072 1073 1074
		/* Squeeze carriage-return/linefeed pairs to linefeed only */
		else if (oldchar == '\r' && opos + 1 < oldstmtlen &&
			old_statement[opos + 1] == '\n')
			continue;
B
Bruce Momjian 已提交
1075 1076 1077 1078
		/*
		 * Handle literals (date, time, timestamp) and ODBC scalar
		 * functions
		 */
1079
		else if (oldchar == '{')
B
Bruce Momjian 已提交
1080 1081 1082 1083
		{
			char	   *esc;
			char	   *begin = &old_statement[opos + 1];

H
Hiroshi Inoue 已提交
1084
#ifdef MULTIBYTE
B
Bruce Momjian 已提交
1085
			char	   *end = multibyte_strchr(begin, '}');
1086

H
Hiroshi Inoue 已提交
1087
#else
B
Bruce Momjian 已提交
1088
			char	   *end = strchr(begin, '}');
1089

H
Hiroshi Inoue 已提交
1090
#endif
1091

B
Bruce Momjian 已提交
1092
			if (!end)
1093
				continue;
H
Hiroshi Inoue 已提交
1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117
			/* procedure calls */
			if (stmt->statement_type == STMT_TYPE_PROCCALL)
			{
				while (isspace((unsigned char) old_statement[++opos]));
				if (old_statement[opos] == '?')
				{
					param_number++;
					while (isspace((unsigned char) old_statement[++opos]));
					if (old_statement[opos] != '=')
					{
						opos--;
						continue;
					}
					while (isspace((unsigned char) old_statement[++opos]));
				}
				if (strnicmp(&old_statement[opos], "call", 4))
				{
					opos--;
					continue;
				}
				opos += (4 - 1);
				CVT_APPEND_STR("SELECT");
				continue; 
			}
1118 1119 1120
			*end = '\0';

			esc = convert_escape(begin);
B
Bruce Momjian 已提交
1121 1122
			if (esc)
			{
1123
				CVT_APPEND_STR(esc);
1124
			}
B
Bruce Momjian 已提交
1125 1126 1127
			else
			{					/* it's not a valid literal so just copy */
				*end = '}';
1128
				CVT_APPEND_CHAR(oldchar);
1129 1130
				continue;
			}
1131

1132
			opos += end - begin + 1;
1133 1134 1135
			*end = '}';
			continue;
		}
H
Hiroshi Inoue 已提交
1136 1137 1138
		/* End of a procedure call */
		else if (oldchar == '}' && stmt->statement_type == STMT_TYPE_PROCCALL)
			continue;
1139

B
Bruce Momjian 已提交
1140 1141 1142 1143 1144
		/*
		 * Can you have parameter markers inside of quotes?  I dont think
		 * so. All the queries I've seen expect the driver to put quotes
		 * if needed.
		 */
1145
		else if (oldchar == '?')
B
Bruce Momjian 已提交
1146 1147 1148
			;					/* ok */
		else
		{
1149
			if (oldchar == '\'')
1150
				in_quote = TRUE;
1151
			else if (oldchar == '\\')
1152
				in_escape = TRUE;
1153
			else if (oldchar == '\"')
1154 1155 1156 1157 1158 1159 1160
				in_dquote = TRUE;
			else if (check_select_into && /* select into check */
    				 opos > 0 &&
    				 isspace((unsigned char) old_statement[opos - 1]) &&
    				 into_table_from(&old_statement[opos]))
			{
				stmt->statement_type = STMT_TYPE_CREATE;
1161
				stmt->options.scroll_concurrency = SQL_CONCUR_READ_ONLY;
1162 1163 1164
				memmove(new_statement, new_statement + declare_pos, npos - declare_pos);
				npos -= declare_pos;
			}
1165 1166 1167 1168 1169 1170 1171 1172 1173 1174
#ifdef	DRIVER_CURSOR_IMPLEMENT
			else if (ins_ctrl && /* select into check */
    				 opos > 0 &&
    				 isspace((unsigned char) old_statement[opos - 1]) &&
    				 strnicmp(&old_statement[opos], "from", 4) == 0)
			{
				ins_ctrl = FALSE;
				CVT_APPEND_STR(", CTID, OID ");
			}
#endif /* DRIVER_CURSOR_IMPLEMENT */
1175
			CVT_APPEND_CHAR(oldchar);
1176 1177 1178
			continue;
		}

1179 1180 1181
		/*
		 * Its a '?' parameter alright
		 */
1182 1183
		param_number++;

B
Bruce Momjian 已提交
1184
		if (param_number >= stmt->parameters_allocated)
1185 1186 1187
		{
			if (stmt->pre_executing)
			{
1188
				CVT_APPEND_STR("NULL");
1189 1190 1191 1192 1193
				stmt->inaccurate_result = TRUE;
				continue;
			}
			else
			{
1194
				CVT_APPEND_CHAR('?');
1195 1196 1197
				continue;
			}
		}
1198

B
Bruce Momjian 已提交
1199 1200 1201
		/* Assign correct buffers based on data at exec param or not */
		if (stmt->parameters[param_number].data_at_exec)
		{
1202 1203 1204
			used = stmt->parameters[param_number].EXEC_used ? *stmt->parameters[param_number].EXEC_used : SQL_NTS;
			buffer = stmt->parameters[param_number].EXEC_buffer;
		}
B
Bruce Momjian 已提交
1205 1206
		else
		{
1207 1208
			
			
1209
			used = stmt->parameters[param_number].used ? *stmt->parameters[param_number].used : SQL_NTS;
1210
			
1211 1212 1213
			buffer = stmt->parameters[param_number].buffer;
		}

B
Bruce Momjian 已提交
1214 1215 1216
		/* Handle NULL parameter data */
		if (used == SQL_NULL_DATA)
		{
1217
			CVT_APPEND_STR("NULL");
1218 1219 1220
			continue;
		}

B
Bruce Momjian 已提交
1221 1222 1223 1224 1225 1226
		/*
		 * If no buffer, and it's not null, then what the hell is it? Just
		 * leave it alone then.
		 */
		if (!buffer)
		{
1227 1228
			if (stmt->pre_executing)
			{
1229
				CVT_APPEND_STR("NULL");
1230 1231 1232 1233 1234
				stmt->inaccurate_result = TRUE;
				continue;
			}
			else
			{
1235
				CVT_APPEND_CHAR('?');
1236 1237
				continue;
			}
1238
		}
1239 1240 1241

		param_ctype = stmt->parameters[param_number].CType;
		param_sqltype = stmt->parameters[param_number].SQLType;
B
Bruce Momjian 已提交
1242

1243
		mylog("copy_statement_with_params: from(fcType)=%d, to(fSqlType)=%d\n", param_ctype, param_sqltype);
B
Bruce Momjian 已提交
1244

B
Bruce Momjian 已提交
1245
		/* replace DEFAULT with something we can use */
B
Bruce Momjian 已提交
1246
		if (param_ctype == SQL_C_DEFAULT)
1247 1248 1249 1250 1251 1252
			param_ctype = sqltype_to_default_ctype(param_sqltype);

		buf = NULL;
		param_string[0] = '\0';
		cbuf[0] = '\0';

B
Bruce Momjian 已提交
1253 1254 1255 1256 1257 1258 1259
		/* Convert input C type to a neutral format */
		switch (param_ctype)
		{
			case SQL_C_BINARY:
			case SQL_C_CHAR:
				buf = buffer;
				break;
1260

B
Bruce Momjian 已提交
1261
			case SQL_C_DOUBLE:
1262
				sprintf(param_string, "%.15g",
B
Bruce Momjian 已提交
1263 1264
						*((SDOUBLE *) buffer));
				break;
1265

B
Bruce Momjian 已提交
1266
			case SQL_C_FLOAT:
1267
				sprintf(param_string, "%.6g",
B
Bruce Momjian 已提交
1268 1269
						*((SFLOAT *) buffer));
				break;
1270

B
Bruce Momjian 已提交
1271 1272 1273 1274 1275
			case SQL_C_SLONG:
			case SQL_C_LONG:
				sprintf(param_string, "%ld",
						*((SDWORD *) buffer));
				break;
1276

B
Bruce Momjian 已提交
1277 1278 1279 1280 1281
			case SQL_C_SSHORT:
			case SQL_C_SHORT:
				sprintf(param_string, "%d",
						*((SWORD *) buffer));
				break;
1282

B
Bruce Momjian 已提交
1283 1284 1285 1286 1287
			case SQL_C_STINYINT:
			case SQL_C_TINYINT:
				sprintf(param_string, "%d",
						*((SCHAR *) buffer));
				break;
1288

B
Bruce Momjian 已提交
1289 1290 1291 1292
			case SQL_C_ULONG:
				sprintf(param_string, "%lu",
						*((UDWORD *) buffer));
				break;
1293

B
Bruce Momjian 已提交
1294 1295 1296 1297
			case SQL_C_USHORT:
				sprintf(param_string, "%u",
						*((UWORD *) buffer));
				break;
1298

B
Bruce Momjian 已提交
1299 1300 1301 1302
			case SQL_C_UTINYINT:
				sprintf(param_string, "%u",
						*((UCHAR *) buffer));
				break;
1303

B
Bruce Momjian 已提交
1304 1305 1306
			case SQL_C_BIT:
				{
					int			i = *((UCHAR *) buffer);
1307

B
Bruce Momjian 已提交
1308 1309 1310
					sprintf(param_string, "%d", i ? 1 : 0);
					break;
				}
1311

B
Bruce Momjian 已提交
1312 1313 1314
			case SQL_C_DATE:
				{
					DATE_STRUCT *ds = (DATE_STRUCT *) buffer;
1315

B
Bruce Momjian 已提交
1316 1317 1318
					st.m = ds->month;
					st.d = ds->day;
					st.y = ds->year;
1319

B
Bruce Momjian 已提交
1320 1321
					break;
				}
1322

B
Bruce Momjian 已提交
1323 1324 1325
			case SQL_C_TIME:
				{
					TIME_STRUCT *ts = (TIME_STRUCT *) buffer;
1326

B
Bruce Momjian 已提交
1327 1328 1329
					st.hh = ts->hour;
					st.mm = ts->minute;
					st.ss = ts->second;
1330

B
Bruce Momjian 已提交
1331 1332
					break;
				}
1333

B
Bruce Momjian 已提交
1334 1335 1336
			case SQL_C_TIMESTAMP:
				{
					TIMESTAMP_STRUCT *tss = (TIMESTAMP_STRUCT *) buffer;
1337

B
Bruce Momjian 已提交
1338 1339 1340 1341 1342 1343
					st.m = tss->month;
					st.d = tss->day;
					st.y = tss->year;
					st.hh = tss->hour;
					st.mm = tss->minute;
					st.ss = tss->second;
1344

B
Bruce Momjian 已提交
1345
					mylog("m=%d,d=%d,y=%d,hh=%d,mm=%d,ss=%d\n", st.m, st.d, st.y, st.hh, st.mm, st.ss);
1346

B
Bruce Momjian 已提交
1347
					break;
1348

B
Bruce Momjian 已提交
1349 1350 1351 1352 1353
				}
			default:
				/* error */
				stmt->errormsg = "Unrecognized C_parameter type in copy_statement_with_parameters";
				stmt->errornumber = STMT_NOT_IMPLEMENTED_ERROR;
1354
				CVT_TERMINATE	/* just in case */
B
Bruce Momjian 已提交
1355 1356 1357
				SC_log_error(func, "", stmt);
				return SQL_ERROR;
		}
1358

B
Bruce Momjian 已提交
1359 1360 1361 1362
		/*
		 * Now that the input data is in a neutral format, convert it to
		 * the desired output format (sqltype)
		 */
1363

B
Bruce Momjian 已提交
1364 1365 1366 1367 1368
		switch (param_sqltype)
		{
			case SQL_CHAR:
			case SQL_VARCHAR:
			case SQL_LONGVARCHAR:
1369

1370
				CVT_APPEND_CHAR('\'');	/* Open Quote */
1371

B
Bruce Momjian 已提交
1372 1373 1374
				/* it was a SQL_C_CHAR */
				if (buf)
				{
1375
					CVT_SPECIAL_CHARS(buf, used);
B
Bruce Momjian 已提交
1376
				}
1377

B
Bruce Momjian 已提交
1378 1379 1380
				/* it was a numeric type */
				else if (param_string[0] != '\0')
				{
1381
					CVT_APPEND_STR(param_string);
B
Bruce Momjian 已提交
1382
				}
1383

B
Bruce Momjian 已提交
1384 1385 1386 1387 1388
				/* it was date,time,timestamp -- use m,d,y,hh,mm,ss */
				else
				{
					sprintf(tmp, "%.4d-%.2d-%.2d %.2d:%.2d:%.2d",
							st.y, st.m, st.d, st.hh, st.mm, st.ss);
1389

1390
					CVT_APPEND_STR(tmp);
B
Bruce Momjian 已提交
1391
				}
1392

1393
				CVT_APPEND_CHAR('\'');	/* Close Quote */
1394

B
Bruce Momjian 已提交
1395
				break;
1396

B
Bruce Momjian 已提交
1397 1398 1399 1400 1401 1402
			case SQL_DATE:
				if (buf)
				{				/* copy char data to time */
					my_strcpy(cbuf, sizeof(cbuf), buf, used);
					parse_datetime(cbuf, &st);
				}
1403

B
Bruce Momjian 已提交
1404
				sprintf(tmp, "'%.4d-%.2d-%.2d'", st.y, st.m, st.d);
1405

1406
				CVT_APPEND_STR(tmp);
B
Bruce Momjian 已提交
1407
				break;
1408

B
Bruce Momjian 已提交
1409 1410 1411 1412 1413 1414
			case SQL_TIME:
				if (buf)
				{				/* copy char data to time */
					my_strcpy(cbuf, sizeof(cbuf), buf, used);
					parse_datetime(cbuf, &st);
				}
1415

B
Bruce Momjian 已提交
1416
				sprintf(tmp, "'%.2d:%.2d:%.2d'", st.hh, st.mm, st.ss);
1417

1418
				CVT_APPEND_STR(tmp);
B
Bruce Momjian 已提交
1419
				break;
1420

B
Bruce Momjian 已提交
1421
			case SQL_TIMESTAMP:
1422

B
Bruce Momjian 已提交
1423 1424 1425 1426 1427
				if (buf)
				{
					my_strcpy(cbuf, sizeof(cbuf), buf, used);
					parse_datetime(cbuf, &st);
				}
1428

B
Bruce Momjian 已提交
1429 1430
				sprintf(tmp, "'%.4d-%.2d-%.2d %.2d:%.2d:%.2d'",
						st.y, st.m, st.d, st.hh, st.mm, st.ss);
1431

1432
				CVT_APPEND_STR(tmp);
1433

B
Bruce Momjian 已提交
1434
				break;
1435

B
Bruce Momjian 已提交
1436 1437 1438
			case SQL_BINARY:
			case SQL_VARBINARY:/* non-ascii characters should be
								 * converted to octal */
1439
				CVT_APPEND_CHAR('\'');	/* Open Quote */
1440

B
Bruce Momjian 已提交
1441
				mylog("SQL_VARBINARY: about to call convert_to_pgbinary, used = %d\n", used);
1442

1443
				CVT_APPEND_BINARY(buf, used);
1444

1445
				CVT_APPEND_CHAR('\'');	/* Close Quote */
B
Byron Nikolaidis 已提交
1446

B
Bruce Momjian 已提交
1447
				break;
1448

B
Bruce Momjian 已提交
1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
			case SQL_LONGVARBINARY:

				if (stmt->parameters[param_number].data_at_exec)
					lobj_oid = stmt->parameters[param_number].lobj_oid;
				else
				{
					/* begin transaction if needed */
					if (!CC_is_in_trans(stmt->hdbc))
					{
						QResultClass *res;
						char		ok;

						res = CC_send_query(stmt->hdbc, "BEGIN", NULL);
						if (!res)
						{
							stmt->errormsg = "Could not begin (in-line) a transaction";
							stmt->errornumber = STMT_EXEC_ERROR;
							SC_log_error(func, "", stmt);
							return SQL_ERROR;
						}
						ok = QR_command_successful(res);
						QR_Destructor(res);
						if (!ok)
						{
							stmt->errormsg = "Could not begin (in-line) a transaction";
							stmt->errornumber = STMT_EXEC_ERROR;
							SC_log_error(func, "", stmt);
							return SQL_ERROR;
						}
1478

B
Bruce Momjian 已提交
1479 1480
						CC_set_in_trans(stmt->hdbc);
					}
B
Byron Nikolaidis 已提交
1481

B
Bruce Momjian 已提交
1482 1483 1484 1485
					/* store the oid */
					lobj_oid = lo_creat(stmt->hdbc, INV_READ | INV_WRITE);
					if (lobj_oid == 0)
					{
B
Byron Nikolaidis 已提交
1486
						stmt->errornumber = STMT_EXEC_ERROR;
B
Bruce Momjian 已提交
1487
						stmt->errormsg = "Couldnt create (in-line) large object.";
B
Byron Nikolaidis 已提交
1488 1489 1490
						SC_log_error(func, "", stmt);
						return SQL_ERROR;
					}
B
Bruce Momjian 已提交
1491 1492 1493 1494 1495

					/* store the fd */
					lobj_fd = lo_open(stmt->hdbc, lobj_oid, INV_WRITE);
					if (lobj_fd < 0)
					{
B
Byron Nikolaidis 已提交
1496
						stmt->errornumber = STMT_EXEC_ERROR;
B
Bruce Momjian 已提交
1497
						stmt->errormsg = "Couldnt open (in-line) large object for writing.";
B
Byron Nikolaidis 已提交
1498 1499 1500 1501
						SC_log_error(func, "", stmt);
						return SQL_ERROR;
					}

B
Bruce Momjian 已提交
1502
					retval = lo_write(stmt->hdbc, lobj_fd, buffer, used);
1503

B
Bruce Momjian 已提交
1504
					lo_close(stmt->hdbc, lobj_fd);
1505

B
Bruce Momjian 已提交
1506 1507 1508 1509 1510
					/* commit transaction if needed */
					if (!globals.use_declarefetch && CC_is_in_autocommit(stmt->hdbc))
					{
						QResultClass *res;
						char		ok;
1511

B
Bruce Momjian 已提交
1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
						res = CC_send_query(stmt->hdbc, "COMMIT", NULL);
						if (!res)
						{
							stmt->errormsg = "Could not commit (in-line) a transaction";
							stmt->errornumber = STMT_EXEC_ERROR;
							SC_log_error(func, "", stmt);
							return SQL_ERROR;
						}
						ok = QR_command_successful(res);
						QR_Destructor(res);
						if (!ok)
						{
							stmt->errormsg = "Could not commit (in-line) a transaction";
							stmt->errornumber = STMT_EXEC_ERROR;
							SC_log_error(func, "", stmt);
							return SQL_ERROR;
						}
1529

B
Bruce Momjian 已提交
1530 1531 1532
						CC_set_no_trans(stmt->hdbc);
					}
				}
1533

B
Bruce Momjian 已提交
1534 1535 1536 1537 1538 1539
				/*
				 * the oid of the large object -- just put that in for the
				 * parameter marker -- the data has already been sent to
				 * the large object
				 */
				sprintf(param_string, "'%d'", lobj_oid);
1540
				CVT_APPEND_STR(param_string);
1541

B
Bruce Momjian 已提交
1542 1543 1544 1545 1546 1547 1548
				break;

				/*
				 * because of no conversion operator for bool and int4,
				 * SQL_BIT
				 */
				/* must be quoted (0 or 1 is ok to use inside the quotes) */
1549

B
Bruce Momjian 已提交
1550 1551 1552 1553
			case SQL_REAL:
				if (buf)
					my_strcpy(param_string, sizeof(param_string), buf, used);
				sprintf(tmp, "'%s'::float4", param_string);
1554
				CVT_APPEND_STR(tmp);
B
Bruce Momjian 已提交
1555 1556 1557 1558 1559 1560
				break;
			case SQL_FLOAT:
			case SQL_DOUBLE:
				if (buf)
					my_strcpy(param_string, sizeof(param_string), buf, used);
				sprintf(tmp, "'%s'::float8", param_string);
1561
				CVT_APPEND_STR(tmp);
B
Bruce Momjian 已提交
1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573
				break;
			case SQL_NUMERIC:
				if (buf)
				{
					cbuf[0] = '\'';
					my_strcpy(cbuf + 1, sizeof(cbuf) - 12, buf, used);	/* 12 = 1('\'') +
																		 * strlen("'::numeric")
																		 * + 1('\0') */
					strcat(cbuf, "'::numeric");
				}
				else
					sprintf(cbuf, "'%s'::numeric", param_string);
1574
				CVT_APPEND_STR(cbuf);
B
Bruce Momjian 已提交
1575 1576 1577
				break;
			default:			/* a numeric type or SQL_BIT */
				if (param_sqltype == SQL_BIT)
1578
					CVT_APPEND_CHAR('\'');		/* Open Quote */
B
Bruce Momjian 已提交
1579 1580 1581

				if (buf)
				{
1582
					CVT_APPEND_DATA(buf, used);
B
Bruce Momjian 已提交
1583 1584 1585
				}
				else
				{
1586
					CVT_APPEND_STR(param_string);
B
Bruce Momjian 已提交
1587 1588 1589
				}

				if (param_sqltype == SQL_BIT)
1590
					CVT_APPEND_CHAR('\'');		/* Close Quote */
B
Bruce Momjian 已提交
1591 1592

				break;
1593
		}
B
Bruce Momjian 已提交
1594
	}							/* end, for */
1595

B
Bruce Momjian 已提交
1596
	/* make sure new_statement is always null-terminated */
1597
	CVT_TERMINATE
1598

B
Bruce Momjian 已提交
1599 1600 1601 1602 1603 1604 1605 1606 1607
	if (stmt->hdbc->DriverToDataSource != NULL)
	{
		int			length = strlen(new_statement);

		stmt->hdbc->DriverToDataSource(stmt->hdbc->translation_option,
									   SQL_CHAR,
									   new_statement, length,
									   new_statement, length, NULL,
									   NULL, 0, NULL);
B
Byron Nikolaidis 已提交
1608 1609
	}

1610 1611 1612 1613
#ifdef	DRIVER_CURSOR_IMPLEMENT
	if (ins_ctrl)
		stmt->options.scroll_concurrency = SQL_CONCUR_READ_ONLY;
#endif /* DRIVER_CURSOR_IMPLEMENT */
1614 1615 1616
	return SQL_SUCCESS;
}

1617

1618 1619 1620
char *
mapFunction(char *func)
{
B
Bruce Momjian 已提交
1621
	int			i;
1622 1623

	for (i = 0; mapFuncs[i][0]; i++)
B
Bruce Momjian 已提交
1624
		if (!stricmp(mapFuncs[i][0], func))
1625 1626 1627 1628
			return mapFuncs[i][1];

	return NULL;
}
1629

1630 1631 1632 1633

/*
 * convert_escape()
 *
1634 1635
 * This function returns a pointer to static memory!
 */
1636 1637 1638
char *
convert_escape(char *value)
{
B
Bruce Momjian 已提交
1639 1640
	static char escape[1024];
	char		key[33];
1641

1642
	/* Separate off the key, skipping leading and trailing whitespace */
B
Bruce Momjian 已提交
1643 1644
	while ((*value != '\0') && isspace((unsigned char) *value))
		value++;
1645
	sscanf(value, "%32s", key);
B
Bruce Momjian 已提交
1646 1647 1648 1649
	while ((*value != '\0') && (!isspace((unsigned char) *value)))
		value++;
	while ((*value != '\0') && isspace((unsigned char) *value))
		value++;
1650

1651
	mylog("convert_escape: key='%s', val='%s'\n", key, value);
1652

B
Bruce Momjian 已提交
1653 1654
	if ((strcmp(key, "d") == 0) ||
		(strcmp(key, "t") == 0) ||
1655 1656
		(strcmp(key, "oj") == 0) ||		/* {oj syntax support for 7.1
										 * servers */
B
Bruce Momjian 已提交
1657 1658
		(strcmp(key, "ts") == 0))
	{
1659
		/* Literal; return the escape part as-is */
B
Bruce Momjian 已提交
1660
		strncpy(escape, value, sizeof(escape) - 1);
1661
	}
B
Bruce Momjian 已提交
1662 1663
	else if (strcmp(key, "fn") == 0)
	{
1664

B
Bruce Momjian 已提交
1665 1666 1667
		/*
		 * Function invocation Separate off the func name, skipping
		 * trailing whitespace.
1668
		 */
B
Bruce Momjian 已提交
1669 1670 1671
		char	   *funcEnd = value;
		char		svchar;
		char	   *mapFunc;
1672 1673

		while ((*funcEnd != '\0') && (*funcEnd != '(') &&
B
Bruce Momjian 已提交
1674
			   (!isspace((unsigned char) *funcEnd)))
1675
			funcEnd++;
1676 1677 1678 1679
		svchar = *funcEnd;
		*funcEnd = '\0';
		sscanf(value, "%32s", key);
		*funcEnd = svchar;
1680 1681
		while ((*funcEnd != '\0') && isspace((unsigned char) *funcEnd))
			funcEnd++;
1682

B
Bruce Momjian 已提交
1683 1684 1685
		/*
		 * We expect left parenthesis here, else return fn body as-is
		 * since it is one of those "function constants".
1686
		 */
B
Bruce Momjian 已提交
1687 1688 1689
		if (*funcEnd != '(')
		{
			strncpy(escape, value, sizeof(escape) - 1);
1690
			return escape;
1691
		}
1692
		mapFunc = mapFunction(key);
B
Bruce Momjian 已提交
1693 1694 1695 1696

		/*
		 * We could have mapFunction() return key if not in table... -
		 * thomas 2000-04-03
1697
		 */
B
Bruce Momjian 已提交
1698 1699
		if (mapFunc == NULL)
		{
1700
			/* If unrecognized function name, return fn body as-is */
B
Bruce Momjian 已提交
1701
			strncpy(escape, value, sizeof(escape) - 1);
1702 1703 1704 1705
			return escape;
		}
		/* copy mapped name and remaining input string */
		strcpy(escape, mapFunc);
B
Bruce Momjian 已提交
1706
		strncat(escape, funcEnd, sizeof(escape) - 1 - strlen(mapFunc));
1707
	}
B
Bruce Momjian 已提交
1708 1709
	else
	{
1710
		/* Bogus key, leave untranslated */
1711 1712 1713 1714 1715 1716 1717 1718 1719 1720
		return NULL;
	}

	return escape;
}


char *
convert_money(char *s)
{
B
Bruce Momjian 已提交
1721 1722
	size_t		i = 0,
				out = 0;
1723

B
Bruce Momjian 已提交
1724 1725
	for (i = 0; i < strlen(s); i++)
	{
1726
		if (s[i] == '$' || s[i] == ',' || s[i] == ')')
B
Bruce Momjian 已提交
1727
			;					/* skip these characters */
1728 1729 1730 1731 1732 1733 1734 1735 1736 1737
		else if (s[i] == '(')
			s[out++] = '-';
		else
			s[out++] = s[i];
	}
	s[out] = '\0';
	return s;
}


1738 1739 1740 1741
/*
 *	This function parses a character string for date/time info and fills in SIMPLE_TIME
 *	It does not zero out SIMPLE_TIME in case it is desired to initialize it with a value
 */
1742
char
B
Bruce Momjian 已提交
1743
parse_datetime(char *buf, SIMPLE_TIME *st)
1744
{
B
Bruce Momjian 已提交
1745 1746 1747 1748 1749 1750 1751 1752
	int			y,
				m,
				d,
				hh,
				mm,
				ss;
	int			nf;

1753 1754
	y = m = d = hh = mm = ss = 0;

1755 1756 1757 1758 1759 1760 1761 1762
	/* escape sequence ? */
	if (buf[0] == '{')
	{
		while (*(++buf) && *buf != '\'');
		if (!(*buf))
			return FALSE;
		buf++;
	}
B
Bruce Momjian 已提交
1763 1764
	if (buf[4] == '-')			/* year first */
		nf = sscanf(buf, "%4d-%2d-%2d %2d:%2d:%2d", &y, &m, &d, &hh, &mm, &ss);
1765
	else
B
Bruce Momjian 已提交
1766
		nf = sscanf(buf, "%2d-%2d-%4d %2d:%2d:%2d", &m, &d, &y, &hh, &mm, &ss);
1767

B
Bruce Momjian 已提交
1768 1769
	if (nf == 5 || nf == 6)
	{
1770 1771 1772 1773 1774 1775 1776 1777 1778 1779
		st->y = y;
		st->m = m;
		st->d = d;
		st->hh = hh;
		st->mm = mm;
		st->ss = ss;

		return TRUE;
	}

B
Bruce Momjian 已提交
1780
	if (buf[4] == '-')			/* year first */
1781 1782 1783 1784
		nf = sscanf(buf, "%4d-%2d-%2d", &y, &m, &d);
	else
		nf = sscanf(buf, "%2d-%2d-%4d", &m, &d, &y);

B
Bruce Momjian 已提交
1785 1786
	if (nf == 3)
	{
1787 1788 1789 1790 1791 1792 1793 1794
		st->y = y;
		st->m = m;
		st->d = d;

		return TRUE;
	}

	nf = sscanf(buf, "%2d:%2d:%2d", &hh, &mm, &ss);
B
Bruce Momjian 已提交
1795 1796
	if (nf == 2 || nf == 3)
	{
1797 1798 1799 1800 1801 1802 1803 1804 1805
		st->hh = hh;
		st->mm = mm;
		st->ss = ss;

		return TRUE;
	}

	return FALSE;
}
1806

1807

1808
/*	Change linefeed to carriage-return/linefeed */
1809
int
1810
convert_linefeeds(const char *si, char *dst, size_t max, BOOL *changed)
1811
{
B
Bruce Momjian 已提交
1812 1813
	size_t		i = 0,
				out = 0;
1814

1815 1816 1817 1818
	if (max == 0)
		max = 0xffffffff;
	*changed = FALSE;
	for (i = 0; si[i] && out < max - 1; i++)
B
Bruce Momjian 已提交
1819 1820 1821 1822 1823 1824
	{
		if (si[i] == '\n')
		{
			/* Only add the carriage-return if needed */
			if (i > 0 && si[i - 1] == '\r')
			{
1825 1826 1827 1828
				if (dst)
					dst[out++] = si[i];
				else
					out++;
B
Byron Nikolaidis 已提交
1829 1830
				continue;
			}
1831
			*changed = TRUE;
B
Byron Nikolaidis 已提交
1832

1833 1834 1835 1836 1837 1838 1839
			if (dst)
			{
				dst[out++] = '\r';
				dst[out++] = '\n';
			}
			else
				out += 2;
1840 1841
		}
		else
1842 1843 1844 1845 1846 1847
		{
			if (dst)
				dst[out++] = si[i];
			else
				out++;
		}
1848
	}
1849 1850
	if (dst)
		dst[out] = '\0';
1851
	return out;
1852 1853
}

1854 1855 1856 1857 1858

/*
 *	Change carriage-return/linefeed to just linefeed
 *	Plus, escape any special characters.
 */
1859
int
1860 1861
convert_special_chars(char *si, char *dst, int used)
{
B
Bruce Momjian 已提交
1862 1863 1864
	size_t		i = 0,
				out = 0,
				max;
1865
	char	   *p = NULL;
1866 1867 1868 1869 1870 1871


	if (used == SQL_NTS)
		max = strlen(si);
	else
		max = used;
1872 1873 1874 1875 1876
	if (dst)
	{
		p = dst;
		p[0] = '\0';
	}
H
Hiroshi Inoue 已提交
1877 1878 1879
#ifdef MULTIBYTE
	multibyte_init();
#endif
1880

B
Bruce Momjian 已提交
1881 1882
	for (i = 0; i < max; i++)
	{
1883 1884 1885 1886 1887 1888 1889 1890 1891
#ifdef MULTIBYTE
		if (multibyte_char_check(si[i]) != 0)
		{
			if (p)
				p[out] = si[i];
			out++;
			continue;
		}
#endif
1892
		if (si[i] == '\r' && si[i + 1] == '\n')
1893
			continue;
B
Byron Nikolaidis 已提交
1894
		else if (si[i] == '\'' || si[i] == '\\')
1895 1896 1897 1898 1899 1900 1901 1902 1903 1904
		{
			if (p)
				p[out++] = '\\';
			else
				out++;
		}
		if (p)
			p[out++] = si[i];
		else
			out++;
1905
	}
1906 1907 1908
	if (p)
		p[out] = '\0';
	return out;
1909 1910
}

1911

1912 1913 1914 1915
/*	!!! Need to implement this function !!!  */
int
convert_pgbinary_to_char(char *value, char *rgbValue, int cbValueMax)
{
1916 1917 1918
	mylog("convert_pgbinary_to_char: value = '%s'\n", value);

	strncpy_null(rgbValue, value, cbValueMax);
1919 1920 1921
	return 0;
}

1922

1923 1924 1925
unsigned int
conv_from_octal(unsigned char *s)
{
B
Bruce Momjian 已提交
1926 1927
	int			i,
				y = 0;
1928

B
Bruce Momjian 已提交
1929 1930
	for (i = 1; i <= 3; i++)
		y += (s[i] - 48) * (int) pow(8, 3 - i);
1931 1932

	return y;
1933

1934 1935
}

1936

B
Byron Nikolaidis 已提交
1937 1938 1939
unsigned int
conv_from_hex(unsigned char *s)
{
B
Bruce Momjian 已提交
1940 1941 1942
	int			i,
				y = 0,
				val;
1943

B
Bruce Momjian 已提交
1944 1945 1946 1947 1948 1949 1950 1951
	for (i = 1; i <= 2; i++)
	{
		if (s[i] >= 'a' && s[i] <= 'f')
			val = s[i] - 'a' + 10;
		else if (s[i] >= 'A' && s[i] <= 'F')
			val = s[i] - 'A' + 10;
		else
			val = s[i] - '0';
B
Byron Nikolaidis 已提交
1952

B
Bruce Momjian 已提交
1953
		y += val * (int) pow(16, 2 - i);
B
Byron Nikolaidis 已提交
1954 1955 1956 1957 1958
	}

	return y;
}

1959

B
Bruce Momjian 已提交
1960
/*	convert octal escapes to bytes */
1961 1962 1963
int
convert_from_pgbinary(unsigned char *value, unsigned char *rgbValue, int cbValueMax)
{
1964
	size_t		i, ilen = strlen(value);
B
Bruce Momjian 已提交
1965
	int			o = 0;
1966

B
Bruce Momjian 已提交
1967

1968
	for (i = 0; i < ilen;)
B
Bruce Momjian 已提交
1969 1970 1971
	{
		if (value[i] == '\\')
		{
1972 1973 1974
			rgbValue[o] = conv_from_octal(&value[i]);
			i += 4;
		}
B
Bruce Momjian 已提交
1975
		else
1976 1977 1978 1979
			rgbValue[o] = value[i++];
		mylog("convert_from_pgbinary: i=%d, rgbValue[%d] = %d, %c\n", i, o, rgbValue[o], rgbValue[o]);
		o++;
	}
1980

B
Bruce Momjian 已提交
1981
	rgbValue[o] = '\0';			/* extra protection */
1982

1983 1984 1985 1986 1987 1988 1989
	return o;
}


char *
conv_to_octal(unsigned char val)
{
B
Bruce Momjian 已提交
1990 1991
	int			i;
	static char x[6];
1992 1993 1994 1995 1996

	x[0] = '\\';
	x[1] = '\\';
	x[5] = '\0';

B
Bruce Momjian 已提交
1997 1998
	for (i = 4; i > 1; i--)
	{
1999 2000 2001 2002 2003 2004 2005
		x[i] = (val & 7) + 48;
		val >>= 3;
	}

	return x;
}

2006

B
Bruce Momjian 已提交
2007
/*	convert non-ascii bytes to octal escape sequences */
2008 2009 2010
int
convert_to_pgbinary(unsigned char *in, char *out, int len)
{
B
Bruce Momjian 已提交
2011 2012
	int			i,
				o = 0;
2013

B
Bruce Momjian 已提交
2014 2015
	for (i = 0; i < len; i++)
	{
2016
		mylog("convert_to_pgbinary: in[%d] = %d, %c\n", i, in[i], in[i]);
B
Bruce Momjian 已提交
2017
		if (isalnum(in[i]) || in[i] == ' ')
2018
			out[o++] = in[i];
B
Bruce Momjian 已提交
2019 2020 2021
		else
		{
			strcpy(&out[o], conv_to_octal(in[i]));
2022 2023 2024 2025 2026 2027 2028 2029 2030 2031
			o += 5;
		}
	}

	mylog("convert_to_pgbinary: returning %d, out='%.*s'\n", o, o, out);

	return o;
}


B
Byron Nikolaidis 已提交
2032 2033 2034
void
encode(char *in, char *out)
{
2035
	unsigned int i, ilen = strlen(in),
B
Bruce Momjian 已提交
2036
				o = 0;
B
Byron Nikolaidis 已提交
2037

2038
	for (i = 0; i < ilen; i++)
B
Bruce Momjian 已提交
2039 2040 2041
	{
		if (in[i] == '+')
		{
B
Byron Nikolaidis 已提交
2042 2043 2044
			sprintf(&out[o], "%%2B");
			o += 3;
		}
B
Bruce Momjian 已提交
2045
		else if (isspace((unsigned char) in[i]))
B
Byron Nikolaidis 已提交
2046
			out[o++] = '+';
B
Bruce Momjian 已提交
2047 2048
		else if (!isalnum((unsigned char) in[i]))
		{
2049
			sprintf(&out[o], "%%%02x", (unsigned char) in[i]);
B
Byron Nikolaidis 已提交
2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061
			o += 3;
		}
		else
			out[o++] = in[i];
	}
	out[o++] = '\0';
}


void
decode(char *in, char *out)
{
2062
	unsigned int i, ilen = strlen(in),
B
Bruce Momjian 已提交
2063
				o = 0;
B
Byron Nikolaidis 已提交
2064

2065
	for (i = 0; i < ilen; i++)
B
Bruce Momjian 已提交
2066
	{
B
Byron Nikolaidis 已提交
2067 2068
		if (in[i] == '+')
			out[o++] = ' ';
B
Bruce Momjian 已提交
2069 2070
		else if (in[i] == '%')
		{
B
Byron Nikolaidis 已提交
2071
			sprintf(&out[o++], "%c", conv_from_hex(&in[i]));
B
Bruce Momjian 已提交
2072
			i += 2;
B
Byron Nikolaidis 已提交
2073 2074 2075 2076 2077 2078 2079 2080
		}
		else
			out[o++] = in[i];
	}
	out[o++] = '\0';
}


2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096
/*-------
 *	1. get oid (from 'value')
 *	2. open the large object
 *	3. read from the large object (handle multiple GetData)
 *	4. close when read less than requested?  -OR-
 *		lseek/read each time
 *		handle case where application receives truncated and
 *		decides not to continue reading.
 *
 *	CURRENTLY, ONLY LONGVARBINARY is handled, since that is the only
 *	data type currently mapped to a PG_TYPE_LO.  But, if any other types
 *	are desired to map to a large object (PG_TYPE_LO), then that would
 *	need to be handled here.  For example, LONGVARCHAR could possibly be
 *	mapped to PG_TYPE_LO someday, instead of PG_TYPE_TEXT as it is now.
 *-------
 */
2097
int
B
Bruce Momjian 已提交
2098
convert_lo(StatementClass *stmt, void *value, Int2 fCType, PTR rgbValue,
B
Bruce Momjian 已提交
2099
		   SDWORD cbValueMax, SDWORD *pcbValue)
2100
{
B
Bruce Momjian 已提交
2101 2102 2103 2104
	Oid			oid;
	int			retval,
				result,
				left = -1;
T
Tom Lane 已提交
2105
	BindInfoClass *bindInfo = NULL;
2106 2107


2108
	/* If using SQLGetData, then current_col will be set */
B
Bruce Momjian 已提交
2109 2110
	if (stmt->current_col >= 0)
	{
2111 2112 2113
		bindInfo = &stmt->bindings[stmt->current_col];
		left = bindInfo->data_left;
	}
2114

B
Bruce Momjian 已提交
2115 2116 2117 2118
	/*
	 * if this is the first call for this column, open the large object
	 * for reading
	 */
2119

B
Bruce Momjian 已提交
2120 2121
	if (!bindInfo || bindInfo->data_left == -1)
	{
B
Byron Nikolaidis 已提交
2122
		/* begin transaction if needed */
B
Bruce Momjian 已提交
2123 2124
		if (!CC_is_in_trans(stmt->hdbc))
		{
B
Byron Nikolaidis 已提交
2125
			QResultClass *res;
B
Bruce Momjian 已提交
2126
			char		ok;
B
Byron Nikolaidis 已提交
2127 2128

			res = CC_send_query(stmt->hdbc, "BEGIN", NULL);
B
Bruce Momjian 已提交
2129 2130
			if (!res)
			{
B
Byron Nikolaidis 已提交
2131 2132 2133 2134 2135 2136
				stmt->errormsg = "Could not begin (in-line) a transaction";
				stmt->errornumber = STMT_EXEC_ERROR;
				return COPY_GENERAL_ERROR;
			}
			ok = QR_command_successful(res);
			QR_Destructor(res);
B
Bruce Momjian 已提交
2137 2138
			if (!ok)
			{
B
Byron Nikolaidis 已提交
2139 2140 2141 2142 2143 2144 2145 2146
				stmt->errormsg = "Could not begin (in-line) a transaction";
				stmt->errornumber = STMT_EXEC_ERROR;
				return COPY_GENERAL_ERROR;
			}

			CC_set_in_trans(stmt->hdbc);
		}

2147 2148
		oid = atoi(value);
		stmt->lobj_fd = lo_open(stmt->hdbc, oid, INV_READ);
B
Bruce Momjian 已提交
2149 2150
		if (stmt->lobj_fd < 0)
		{
2151
			stmt->errornumber = STMT_EXEC_ERROR;
2152
			stmt->errormsg = "Couldnt open large object for reading.";
2153 2154
			return COPY_GENERAL_ERROR;
		}
2155

B
Bruce Momjian 已提交
2156
		/* Get the size */
2157
		retval = lo_lseek(stmt->hdbc, stmt->lobj_fd, 0L, SEEK_END);
B
Bruce Momjian 已提交
2158 2159
		if (retval >= 0)
		{
2160 2161 2162 2163
			left = lo_tell(stmt->hdbc, stmt->lobj_fd);
			if (bindInfo)
				bindInfo->data_left = left;

B
Bruce Momjian 已提交
2164
			/* return to beginning */
2165 2166 2167 2168
			lo_lseek(stmt->hdbc, stmt->lobj_fd, 0L, SEEK_SET);
		}
	}

B
Bruce Momjian 已提交
2169
	if (left == 0)
2170
		return COPY_NO_DATA_FOUND;
2171

B
Bruce Momjian 已提交
2172 2173
	if (stmt->lobj_fd < 0)
	{
2174 2175 2176 2177
		stmt->errornumber = STMT_EXEC_ERROR;
		stmt->errormsg = "Large object FD undefined for multiple read.";
		return COPY_GENERAL_ERROR;
	}
2178

2179
	retval = lo_read(stmt->hdbc, stmt->lobj_fd, (char *) rgbValue, cbValueMax);
B
Bruce Momjian 已提交
2180 2181
	if (retval < 0)
	{
2182
		lo_close(stmt->hdbc, stmt->lobj_fd);
B
Byron Nikolaidis 已提交
2183 2184

		/* commit transaction if needed */
B
Bruce Momjian 已提交
2185 2186
		if (!globals.use_declarefetch && CC_is_in_autocommit(stmt->hdbc))
		{
B
Byron Nikolaidis 已提交
2187
			QResultClass *res;
B
Bruce Momjian 已提交
2188
			char		ok;
B
Byron Nikolaidis 已提交
2189 2190

			res = CC_send_query(stmt->hdbc, "COMMIT", NULL);
B
Bruce Momjian 已提交
2191 2192
			if (!res)
			{
B
Byron Nikolaidis 已提交
2193 2194 2195 2196 2197 2198
				stmt->errormsg = "Could not commit (in-line) a transaction";
				stmt->errornumber = STMT_EXEC_ERROR;
				return COPY_GENERAL_ERROR;
			}
			ok = QR_command_successful(res);
			QR_Destructor(res);
B
Bruce Momjian 已提交
2199 2200
			if (!ok)
			{
B
Byron Nikolaidis 已提交
2201 2202 2203 2204 2205 2206 2207 2208
				stmt->errormsg = "Could not commit (in-line) a transaction";
				stmt->errornumber = STMT_EXEC_ERROR;
				return COPY_GENERAL_ERROR;
			}

			CC_set_no_trans(stmt->hdbc);
		}

2209 2210 2211 2212 2213 2214 2215
		stmt->lobj_fd = -1;

		stmt->errornumber = STMT_EXEC_ERROR;
		stmt->errormsg = "Error reading from large object.";
		return COPY_GENERAL_ERROR;
	}

2216 2217 2218 2219 2220 2221 2222
	if (retval < left)
		result = COPY_RESULT_TRUNCATED;
	else
		result = COPY_OK;

	if (pcbValue)
		*pcbValue = left < 0 ? SQL_NO_TOTAL : left;
2223

B
Bruce Momjian 已提交
2224
	if (bindInfo && bindInfo->data_left > 0)
2225
		bindInfo->data_left -= retval;
2226

B
Bruce Momjian 已提交
2227 2228
	if (!bindInfo || bindInfo->data_left == 0)
	{
2229
		lo_close(stmt->hdbc, stmt->lobj_fd);
B
Byron Nikolaidis 已提交
2230 2231

		/* commit transaction if needed */
B
Bruce Momjian 已提交
2232 2233
		if (!globals.use_declarefetch && CC_is_in_autocommit(stmt->hdbc))
		{
B
Byron Nikolaidis 已提交
2234
			QResultClass *res;
B
Bruce Momjian 已提交
2235
			char		ok;
B
Byron Nikolaidis 已提交
2236 2237

			res = CC_send_query(stmt->hdbc, "COMMIT", NULL);
B
Bruce Momjian 已提交
2238 2239
			if (!res)
			{
B
Byron Nikolaidis 已提交
2240 2241 2242 2243 2244 2245
				stmt->errormsg = "Could not commit (in-line) a transaction";
				stmt->errornumber = STMT_EXEC_ERROR;
				return COPY_GENERAL_ERROR;
			}
			ok = QR_command_successful(res);
			QR_Destructor(res);
B
Bruce Momjian 已提交
2246 2247
			if (!ok)
			{
B
Byron Nikolaidis 已提交
2248 2249 2250 2251 2252 2253 2254 2255
				stmt->errormsg = "Could not commit (in-line) a transaction";
				stmt->errornumber = STMT_EXEC_ERROR;
				return COPY_GENERAL_ERROR;
			}

			CC_set_no_trans(stmt->hdbc);
		}

B
Bruce Momjian 已提交
2256
		stmt->lobj_fd = -1;		/* prevent further reading */
2257
	}
2258 2259

	return result;
2260
}