void
tds_send_result(TDSSOCKET * tds, TDSRESULTINFO * resinfo)
{
TDSCOLUMN *curcol;
int i, totlen;
size_t len;
tds_put_byte(tds, TDS_RESULT_TOKEN);
totlen = 2;
for (i = 0; i < resinfo->num_cols; i++) {
curcol = resinfo->columns[i];
len = tds_dstr_len(&curcol->column_name);
totlen += 8;
totlen += len;
curcol = resinfo->columns[i];
if (!is_fixed_type(curcol->column_type)) {
totlen++;
}
}
tds_put_smallint(tds, totlen);
tds_put_smallint(tds, resinfo->num_cols);
for (i = 0; i < resinfo->num_cols; i++) {
curcol = resinfo->columns[i];
len = tds_dstr_len(&curcol->column_name);
tds_put_byte(tds, tds_dstr_len(&curcol->column_name));
tds_put_n(tds, tds_dstr_cstr(&curcol->column_name), len);
tds_put_byte(tds, '0');
tds_put_int(tds, curcol->column_usertype);
tds_put_byte(tds, curcol->column_type);
if (!is_fixed_type(curcol->column_type)) {
tds_put_byte(tds, curcol->column_size);
}
tds_put_byte(tds, 0);
}
}
/**
* Send a "done" token, marking the end of a table, stored procedure, or query.
* \param tds Where the token will be written to.
* \param token The appropriate type of "done" token for this context:
* TDS_DONE_TOKEN outside a stored procedure,
* TDS_DONEINPROC_TOKEN inside a stored procedure, or
* TDS_DONEPROC_TOKEN at the end of a stored procedure.
* \param flags Bitwise-OR of flags in the "enum tds_end" data type.
* TDS_DONE_FINAL for the last statement in a query,
* TDS_DONE_MORE_RESULTS if not the last statement,
* TDS_DONE_ERROR if the statement had an error,
* TDS_DONE_INXACT if a transaction is pending,
* TDS_DONE_PROC inside a stored procedure,
* TDS_DONE_COUNT if a table was sent (and rows counted)
* TDS_DONE_CANCELLED if the query was canceled, and
* TDS_DONE_EVENT if the token marks an event.
* \param numrows Number of rows, if flags has TDS_DONE_COUNT.
*/
void
tds_send_done(TDSSOCKET * tds, int token, TDS_SMALLINT flags, TDS_INT numrows)
{
tds_put_byte(tds, token);
tds_put_smallint(tds, flags);
tds_put_smallint(tds, 2); /* are these two bytes the transaction status? */
if (IS_TDS72_PLUS(tds->conn))
tds_put_int8(tds, numrows);
else
tds_put_int(tds, numrows);
}
void
tds7_send_result(TDSSOCKET * tds, TDSRESULTINFO * resinfo)
{
int i, j;
TDSCOLUMN *curcol;
/* TDS7+ uses TDS7_RESULT_TOKEN to send column names and info */
tds_put_byte(tds, TDS7_RESULT_TOKEN);
/* send the number of columns */
tds_put_smallint(tds, resinfo->num_cols);
/* send info about each column */
for (i = 0; i < resinfo->num_cols; i++) {
/* usertype, flags, and type */
curcol = resinfo->columns[i];
tds_put_smallint(tds, curcol->column_usertype);
tds_put_smallint(tds, curcol->column_flags);
tds_put_byte(tds, curcol->column_type); /* smallint? */
/* bytes in "size" field varies */
if (is_blob_type(curcol->column_type)) {
tds_put_int(tds, curcol->column_size);
} else if (curcol->column_type>=128) { /*is_large_type*/
tds_put_smallint(tds, curcol->column_size);
} else {
tds_put_tinyint(tds, curcol->column_size);
}
/* some types have extra info */
if (is_numeric_type(curcol->column_type)) {
tds_put_tinyint(tds, curcol->column_prec);
tds_put_tinyint(tds, curcol->column_scale);
} else if (is_blob_type(curcol->column_type)) {
size_t len = tds_dstr_len(&curcol->table_name);
const char *name = tds_dstr_cstr(&curcol->table_name);
tds_put_smallint(tds, 2 * len);
for (j = 0; name[j] != '\0'; j++){
tds_put_byte(tds, name[j]);
tds_put_byte(tds, 0);
}
}
/* finally the name, in UCS16 format */
assert(strlen(curcol->column_name) == curcol->column_namelen);
tds_put_byte(tds, curcol->column_namelen);
for (j = 0; j < curcol->column_namelen; j++) {
tds_put_byte(tds, curcol->column_name[j]);
tds_put_byte(tds, 0);
}
}
}
void
tds_send_msg(TDSSOCKET * tds, int msgno, int msgstate, int severity,
const char *msgtext, const char *srvname, const char *procname, int line)
{
int msgsz;
size_t len;
tds_put_byte(tds, TDS_INFO_TOKEN);
if (!procname)
procname = "";
len = strlen(procname);
msgsz = 4 /* msg no */
+ 1 /* msg state */
+ 1 /* severity */
/* FIXME ucs2 */
+ 4 + (IS_TDS7_PLUS(tds->conn) ? 2 : 1) * (strlen(msgtext) + strlen(srvname) + len)
+ (IS_TDS72_PLUS(tds->conn) ? 4 : 2); /* line number */
tds_put_smallint(tds, msgsz);
tds_put_int(tds, msgno);
tds_put_byte(tds, msgstate);
tds_put_byte(tds, severity);
tds_put_smallint(tds, strlen(msgtext));
/* FIXME ucs2 */
tds_put_string(tds, msgtext, strlen(msgtext));
tds_put_byte(tds, strlen(srvname));
/* FIXME ucs2 */
tds_put_string(tds, srvname, strlen(srvname));
if (len) {
tds_put_byte(tds, len);
/* FIXME ucs2 */
tds_put_string(tds, procname, len);
} else {
tds_put_byte(tds, 0);
}
if (IS_TDS72_PLUS(tds->conn))
tds_put_int(tds, line);
else
tds_put_smallint(tds, line);
}
static TDSRET
tds5_negotiate_handle_next(TDSSOCKET * tds, TDSAUTHENTICATION * tds_auth, size_t len)
{
TDS5NEGOTIATE *auth = (TDS5NEGOTIATE *) tds_auth;
TDSPARAMINFO *info;
void *rsa, *nonce = NULL;
size_t rsa_len, nonce_len = 0;
void *em;
size_t em_size;
TDSRET rc = TDS_FAIL;
/* send next data for authentication */
if (!tds->login)
goto error;
/* we only support RSA authentication, we should have send 2/3 parameters:
* 1- integer.. unknown actually 1 TODO
* 2- binary, rsa public key in PEM format
* 3- binary, nonce (optional)
*/
/* message not supported */
if (auth->msg_type != 0x1e)
goto error;
info = tds->param_info;
if (!info || info->num_cols < 2)
goto error;
if (info->columns[1]->column_type != SYBLONGBINARY)
goto error;
if (info->num_cols >= 3 && info->columns[2]->column_type != SYBLONGBINARY)
goto error;
rsa = ((TDSBLOB*) info->columns[1]->column_data)->textvalue;
rsa_len = info->columns[1]->column_size;
if (info->num_cols >= 3) {
nonce = ((TDSBLOB*) info->columns[2]->column_data)->textvalue;
nonce_len = info->columns[2]->column_size;
}
em = tds5_rsa_encrypt(rsa, rsa_len, nonce, nonce_len, tds_dstr_cstr(&tds->login->password), &em_size);
if (!em)
goto error;
tds->out_flag = TDS_NORMAL;
/* password */
tds5_send_msg(tds, 0x1f);
tds_put_n(tds, "\xec\x0e\x00\x01\x00\x00\x00\x00\x00\x00\x00\xe1\xff\xff\xff\x7f\x00", 0x11);
tds_put_byte(tds, TDS5_PARAMS_TOKEN);
tds_put_int(tds, em_size);
tds_put_n(tds, em, em_size);
/* remote password */
tds5_send_msg(tds, 0x20);
tds_put_n(tds, "\xec\x17\x00\x02\x00\x00\x00\x00\x00\x00\x00\x27\xff\x00\x00\x00\x00\x00\x00\x00\xe1\xff\xff\xff\x7f\x00", 0x1a);
tds_put_byte(tds, TDS5_PARAMS_TOKEN);
tds_put_byte(tds, 0);
tds_put_int(tds, em_size);
tds_put_n(tds, em, em_size);
free(em);
rc = tds_flush_packet(tds);
error:
tds5_negotiate_free(tds->conn, tds_auth);
tds->conn->authentication = NULL;
return rc;
}
/**
* Send one row of data to server
* \tds
* \param bcpinfo BCP information
* \param get_col_data function to call to retrieve data to be sent
* \param ignored function to call if we try to send NULL if not allowed (not used)
* \param offset passed to get_col_data and null_error to specify the row to get
* \return TDS_SUCCESS or TDS_FAIL.
*/
TDSRET
tds_bcp_send_record(TDSSOCKET *tds, TDSBCPINFO *bcpinfo, tds_bcp_get_col_data get_col_data, tds_bcp_null_error null_error, int offset)
{
TDSCOLUMN *bindcol;
int i, start_col = bcpinfo->next_col;
TDSRET rc;
tdsdump_log(TDS_DBG_FUNC, "tds_bcp_send_bcp_record(%p, %p, %p, %p, %d)\n", tds, bcpinfo, get_col_data, null_error, offset);
if (tds->out_flag != TDS_BULK || tds_set_state(tds, TDS_WRITING) != TDS_WRITING)
return TDS_FAIL;
if (start_col > 0) {
bindcol = bcpinfo->bindinfo->columns[start_col - 1];
*bindcol->column_lenbind
= MIN((TDS_INT) bindcol->column_bindlen - bcpinfo->text_sent,
*bindcol->column_lenbind);
tds_put_n(tds, bindcol->column_varaddr, *bindcol->column_lenbind);
bcpinfo->text_sent += *bindcol->column_lenbind;
if ((TDS_UINT) bcpinfo->text_sent < bindcol->column_bindlen) {
return TDS_SUCCESS; /* That's all for now. */
} else if (!IS_TDS7_PLUS(tds->conn)) {
bcpinfo->blob_cols++;
}
bcpinfo->next_col = 0;
bcpinfo->text_sent = 0;
}
if (IS_TDS7_PLUS(tds->conn)) {
if (start_col == 0) {
tds_put_byte(tds, TDS_ROW_TOKEN); /* 0xd1 */
}
for (i = start_col; i < bcpinfo->bindinfo->num_cols; i++) {
TDS_INT save_size;
unsigned char *save_data;
TDSBLOB blob;
int /* bool */ has_text = 0;
bindcol = bcpinfo->bindinfo->columns[i];
/*
* Don't send the (meta)data for timestamp columns or
* identity columns unless indentity_insert is enabled.
*/
if ((!bcpinfo->identity_insert_on && bindcol->column_identity) ||
bindcol->column_timestamp ||
!tds_bcp_is_bound(bcpinfo, bindcol)) {
continue;
}
rc = get_col_data(bcpinfo, bindcol, offset);
if (rc == TDS_FAIL) {
tdsdump_log(TDS_DBG_INFO1, "get_col_data (column %d) failed\n", i + 1);
goto cleanup;
} else if (rc != TDS_SUCCESS) { /* CS_BLK_HAS_TEXT? */
has_text = 1;
}
tdsdump_log(TDS_DBG_INFO1, "gotten column %d length %d null %d\n",
i + 1, bindcol->bcp_column_data->datalen, bindcol->bcp_column_data->is_null);
save_size = bindcol->column_cur_size;
save_data = bindcol->column_data;
assert(bindcol->column_data == NULL);
if (bindcol->bcp_column_data->is_null) {
if ( !bindcol->column_nullable
&& !is_nullable_type(bindcol->on_server.column_type) ) {
return TDS_FAIL;
}
bindcol->column_cur_size = -1;
} else if (has_text) {
bindcol->column_cur_size = bindcol->bcp_column_data->datalen;
} else if (is_blob_col(bindcol)) {
bindcol->column_cur_size = bindcol->bcp_column_data->datalen;
memset(&blob, 0, sizeof(blob));
blob.textvalue = (TDS_CHAR *) bindcol->bcp_column_data->data;
bindcol->column_data = (unsigned char *) &blob;
} else {
bindcol->column_cur_size = bindcol->bcp_column_data->datalen;
bindcol->column_data = bindcol->bcp_column_data->data;
}
rc = bindcol->funcs->put_data(tds, bindcol, 1);
bindcol->column_cur_size = save_size;
bindcol->column_data = save_data;
if (TDS_FAILED(rc))
goto cleanup;
else if (has_text) {
bcpinfo->next_col = i + 1;
/* bcpinfo->text_sent = 0; */
break;
}
}
} /* IS_TDS7_PLUS */
else {
if (start_col == 0) {
int row_pos;
int row_sz_pos;
int var_cols_written = 0;
TDS_INT old_record_size = bcpinfo->bindinfo->row_size;
unsigned char *record = bcpinfo->bindinfo->current_row;
memset(record, '\0', old_record_size); /* zero the rowbuffer */
/*
* offset 0 = number of var columns
* offset 1 = row number. zeroed (datasever assigns)
*/
row_pos = 2;
rc = TDS_FAIL;
if ((row_pos = tds_bcp_add_fixed_columns(bcpinfo, get_col_data, null_error, offset, record, row_pos)) < 0)
goto cleanup;
row_sz_pos = row_pos;
/* potential variable columns to write */
if ((row_pos = tds_bcp_add_variable_columns(bcpinfo, get_col_data, null_error, offset, record, row_pos, &var_cols_written)) < 0)
goto cleanup;
if (var_cols_written) {
TDS_PUT_UA2(&record[row_sz_pos], row_pos);
record[0] = var_cols_written;
}
tdsdump_log(TDS_DBG_INFO1, "old_record_size = %d new size = %d \n", old_record_size, row_pos);
tds_put_smallint(tds, row_pos);
tds_put_n(tds, record, row_pos);
/* row is done, now handle any text/image data */
bcpinfo->blob_cols = 0;
}
for (i = start_col; i < bcpinfo->bindinfo->num_cols; i++) {
bindcol = bcpinfo->bindinfo->columns[i];
if (is_blob_type(bindcol->column_type)) {
/* Elide trailing NULLs */
if (bindcol->bcp_column_data->is_null) {
int j;
for (j = i + 1; j < bcpinfo->bindinfo->num_cols; ++j) {
TDSCOLUMN *bindcol2 = bcpinfo->bindinfo->columns[j];
if (is_blob_type(bindcol2->column_type)
&& !bindcol2->bcp_column_data->is_null) {
break;
}
}
if (j == bcpinfo->bindinfo->num_cols) {
i = j;
break;
}
}
rc = get_col_data(bcpinfo, bindcol, offset);
if (rc == TDS_FAIL)
goto cleanup;
/* unknown but zero */
tds_put_smallint(tds, 0);
tds_put_byte(tds, (unsigned char) bindcol->column_type);
tds_put_byte(tds, 0xff - bcpinfo->blob_cols);
/*
* offset of txptr we stashed during variable
* column processing
*/
tds_put_smallint(tds, bindcol->column_textpos);
tds_put_int(tds, bindcol->bcp_column_data->datalen);
if (rc != TDS_SUCCESS) { /* CS_BLK_HAS_TEXT? */
bcpinfo->next_col = i + 1;
/* bcpinfo->text_sent = 0; */
break;
}
tds_put_n(tds, bindcol->bcp_column_data->data, bindcol->bcp_column_data->datalen);
bcpinfo->blob_cols++;
}
}
}
if (i == bcpinfo->bindinfo->num_cols) {
tds_set_state(tds, TDS_SENDING);
bcpinfo->next_col = 0;
}
return TDS_SUCCESS;
cleanup:
tds_set_state(tds, TDS_SENDING);
return rc;
}
