ctlib: Add support for setting hints for bulk copy

This is an extension as Sybase does not support hints.
Microsoft supports hints using bcp_control.
Hints are passed as a string using blk_props.

Signed-off-by: Frediano Ziglio <freddy77@gmail.com>

Author: freddy77

include/bkpublic.h

@@ -35,6 +35,9 @@ extern "C"
 /* bulk properties start with 1 */
 #define BLK_IDENTITY 1
 
+/* FreeTDS extention, set MS bulk hints */
+#define BLK_HINTS 1001
+
 CS_RETCODE blk_alloc(CS_CONNECTION * connection, CS_INT version, CS_BLKDESC ** blk_pointer);
 CS_RETCODE blk_bind(CS_BLKDESC * blkdesc, CS_INT colnum, CS_DATAFMT * datafmt, CS_VOID * buffer, CS_INT * datalen,
 		    CS_SMALLINT * indicator);

include/ctlib.h

@@ -402,6 +402,8 @@ const CS_DATAFMT_LARGE *_ct_datafmt_conv_in(CS_CONTEXT * ctx, const CS_DATAFMT *
 CS_DATAFMT_LARGE *_ct_datafmt_conv_prepare(CS_CONTEXT * ctx, CS_DATAFMT * datafmt, CS_DATAFMT_LARGE * fmtbuf);
 void _ct_datafmt_conv_back(CS_DATAFMT * datafmt, CS_DATAFMT_LARGE * fmtbuf);
 
+CS_RETCODE _ct_props_dstr(CS_CONNECTION * con, DSTR *s, CS_INT action, CS_VOID * buffer, CS_INT buflen, CS_INT * outlen);
+
 #ifdef __cplusplus
 #if 0
 {

src/ctlib/blk.c

@@ -405,6 +405,9 @@ blk_props(CS_BLKDESC * blkdesc, CS_INT action, CS_INT property, CS_VOID * buffer
 		}
 		break;
 
+	case BLK_HINTS:
+		return _ct_props_dstr(CONN(blkdesc), &blkdesc->bcpinfo.hint, action, buffer, buflen, outlen);
+
 	default:
 		_ctclient_msg(NULL, CONN(blkdesc), "blk_props", 2, 5, 1, 141, "%s, %d", "property", property);
 		break;

src/ctlib/ctutil.c

@@ -334,3 +334,27 @@ _ct_get_string_length(const char *buf, CS_INT buflen)
 
 	return buflen;
 }
+
+CS_RETCODE
+_ct_props_dstr(CS_CONNECTION * con, DSTR *s, CS_INT action, CS_VOID * buffer, CS_INT buflen, CS_INT * out_len)
+{
+	if (action == CS_SET) {
+		buflen = _ct_get_string_length(buffer, buflen);
+		if (buflen < 0) {
+			/* TODO what error ?? */
+			_ctclient_msg(NULL, con, "ct_con_props(SET,APPNAME)", 1, 1, 1, 5, "%d, %s", buflen, "buflen");
+			return CS_FAIL;
+		}
+		if (tds_dstr_copyn(s, buffer, buflen) != NULL)
+			return CS_SUCCEED;
+	} else if (action == CS_GET) {
+		if (out_len)
+			*out_len = tds_dstr_len(s);
+		strlcpy((char *) buffer, tds_dstr_cstr(s), buflen);
+		return CS_SUCCEED;
+	} else if (action == CS_CLEAR) {
+		tds_dstr_empty(s);
+		return CS_SUCCEED;
+	}
+	return CS_FAIL;
+}

src/ctlib/unittests/blk_in2.c

@@ -24,6 +24,8 @@ hoge_blkin(CS_CONNECTION * con, CS_BLKDESC * blk, char *table, char *data)
 	check_call(ct_cancel, (con, NULL, CS_CANCEL_ALL));
 	check_call(blk_init, (blk, CS_BLK_IN, table, CS_NULLTERM));
 
+	check_call(blk_props, (blk, CS_SET, BLK_HINTS, "TABLOCK", CS_NULLTERM, NULL));
+
 	meta.count = 1;
 	meta.datatype = CS_CHAR_TYPE;
 	meta.format = CS_FMT_PADBLANK;