static void
unfinished_query_test(TDSSOCKET *tds)
{
int char_len;
char *buf, *p;
int i, len;
union {
TDS_USMALLINT si;
TDS_UINT i;
TDS_INT8 i8;
char buf[8];
} conv;
if (IS_TDS72_PLUS(tds->conn))
return;
tds_init_write_buf(tds);
/* try to build an invalid (unfinished) query split in two packets */
char_len = IS_TDS7_PLUS(tds->conn) ? 2 : 1;
buf = calloc(1, tds->out_buf_max + 200);
memset(buf, '-', tds->out_buf_max + 200);
strcpy(buf + (tds->out_buf_max - 8) / char_len - strlen(select_query) + 1, select_query);
memset(strchr(buf, 0), 0, 16);
/* convert if needed */
len = strlen(buf);
for (i = len; --i >= 0; ) {
char c = buf[i];
buf[i * char_len + 0] = c;
if (IS_TDS7_PLUS(tds->conn))
buf[i * char_len + 1] = 0;
}
len *= char_len;
/* send the query using tds_put_int8, non allineati */
tds->out_flag = TDS_QUERY;
if (tds_set_state(tds, TDS_WRITING) != TDS_WRITING)
exit(1);
p = buf;
memcpy(conv.buf, p, 2);
tds_put_smallint(tds, conv.si);
p += 2;
for (; p < buf + len; p += 8) {
CHECK_TDS_EXTRA(tds);
memcpy(conv.buf, p, 8);
tds_put_int8(tds, conv.i8);
}
tds_flush_packet(tds);
tds_set_state(tds, TDS_PENDING);
/* check result was fine */
if (TDS_FAILED(tds_process_simple_query(tds))) {
fprintf(stderr, "Error in prepared query\n");
exit(1);
}
free(buf);
}
/*
* if a dead connection on the client side left this member in a questionable
* state, let's bring in a correct one
* We are not sure what the client did so we must try to clean as much as
* possible.
* Use pool_free_member if the state is really broken.
*/
void
pool_reset_member(TDS_POOL_MEMBER * pmbr)
{
// FIXME not wait for server !!! asyncronous
TDSSOCKET *tds = pmbr->tds;
if (pmbr->current_user) {
pmbr->current_user->assigned_member = NULL;
pool_free_user(pmbr->current_user);
pmbr->current_user = NULL;
}
/* cancel whatever pending */
tds->state = TDS_IDLE;
tds_init_write_buf(tds);
tds->out_flag = TDS_CANCEL;
tds_flush_packet(tds);
tds->state = TDS_PENDING;
if (tds_read_packet(tds) < 0) {
pool_free_member(pmbr);
return;
}
if (IS_TDS71_PLUS(tds->conn)) {
/* this 0x9 final reset the state from mssql 2000 */
tds_init_write_buf(tds);
tds->out_flag = TDS_QUERY;
tds_write_packet(tds, 0x9);
tds->state = TDS_PENDING;
if (tds_read_packet(tds) < 0) {
pool_free_member(pmbr);
return;
}
}
pmbr->state = TDS_IDLE;
}