static char *
request_body_get(TSHttpTxn txnp, int *len)
{
char *ret = NULL;
TSIOBufferReader post_buffer_reader = TSHttpTxnPostBufferReaderGet(txnp);
int64_t read_avail = TSIOBufferReaderAvail(post_buffer_reader);
if (read_avail == 0) {
TSIOBufferReaderFree(post_buffer_reader);
return NULL;
}
ret = (char *)TSmalloc(sizeof(char) * read_avail);
int64_t consumed = 0;
int64_t data_len = 0;
const char *char_data = NULL;
TSIOBufferBlock block = TSIOBufferReaderStart(post_buffer_reader);
while (block != NULL) {
char_data = TSIOBufferBlockReadStart(block, post_buffer_reader, &data_len);
memcpy(ret + consumed, char_data, data_len);
consumed += data_len;
block = TSIOBufferBlockNext(block);
}
TSIOBufferReaderFree(post_buffer_reader);
*len = (int)consumed;
return ret;
}
/*-------------------------------------------------------------------------
cont_data_destroy
Deallocate ContData structure associated to a transaction
Input:
data structure to deallocate
Output:
Return Value:
none
-------------------------------------------------------------------------*/
static void
cont_data_destroy(ContData * data)
{
TSDebug(DBG_TAG, "Destroying continuation data");
if (data) {
TSAssert(data->magic == MAGIC_ALIVE);
if (data->output_reader) {
TSIOBufferReaderFree(data->output_reader);
data->output_reader = NULL;
}
if (data->output_buffer) {
TSIOBufferDestroy(data->output_buffer);
data->output_buffer = NULL;
}
if (data->psi_reader) {
TSIOBufferReaderFree(data->psi_reader);
data->psi_reader = NULL;
}
if (data->psi_buffer) {
TSIOBufferDestroy(data->psi_buffer);
data->psi_buffer = NULL;
}
data->magic = MAGIC_DEAD;
TSfree(data);
}
}
void
ts_http_fetcher_release(http_fetcher *fch)
{
if (fch->http_vc) {
TSVConnClose(fch->http_vc);
fch->http_vc = NULL;
}
if (fch->http_parser)
TSHttpParserDestroy(fch->http_parser);
if (fch->hdr_loc) {
TSMimeHdrDestroy(fch->hdr_bufp, fch->hdr_loc);
TSHandleMLocRelease(fch->hdr_bufp, TS_NULL_MLOC, fch->hdr_loc);
}
if (fch->hdr_bufp)
TSMBufferDestroy(fch->hdr_bufp);
if (fch->hdr_reader)
TSIOBufferReaderFree(fch->hdr_reader);
if (fch->hdr_buffer)
TSIOBufferDestroy(fch->hdr_buffer);
if (fch->resp_reader)
TSIOBufferReaderFree(fch->resp_reader);
if (fch->resp_buffer)
TSIOBufferDestroy(fch->resp_buffer);
if (fch->body_reader)
TSIOBufferReaderFree(fch->body_reader);
if (fch->body_buffer)
TSIOBufferDestroy(fch->body_buffer);
if (fch->flow_reader)
TSIOBufferReaderFree(fch->flow_reader);
if (fch->flow_buffer)
TSIOBufferDestroy(fch->flow_buffer);
if (fch->req_reader)
TSIOBufferReaderFree(fch->req_reader);
if (fch->req_buffer)
TSIOBufferDestroy(fch->req_buffer);
TSContDestroy(fch->fetch_contp);
TSfree(fch);
fch = NULL;
}
static int
transform_connect(TSCont contp, TransformData *data)
{
TSAction action;
int content_length;
struct sockaddr_in ip_addr;
data->state = STATE_CONNECT;
content_length = TSIOBufferReaderAvail(data->input_reader);
if (content_length >= 0) {
data->content_length = content_length;
data->content_length = htonl(data->content_length);
/* Prepend the content length to the buffer.
* If we decide to not send the content to the transforming
* server then we need to make sure and skip input_reader
* over the content length.
*/
{
TSIOBuffer temp;
TSIOBufferReader tempReader;
temp = TSIOBufferCreate();
tempReader = TSIOBufferReaderAlloc(temp);
TSIOBufferWrite(temp, (const char *)&content_length, sizeof(int));
TSIOBufferCopy(temp, data->input_reader, content_length, 0);
TSIOBufferReaderFree(data->input_reader);
TSIOBufferDestroy(data->input_buf);
data->input_buf = temp;
data->input_reader = tempReader;
}
} else {
TSError("[%s] TSIOBufferReaderAvail returns TS_ERROR", PLUGIN_NAME);
return 0;
}
/* TODO: This only supports IPv4, probably should be changed at some point, but
it's an example ... */
memset(&ip_addr, 0, sizeof(ip_addr));
ip_addr.sin_family = AF_INET;
ip_addr.sin_addr.s_addr = server_ip; /* Should be in network byte order */
ip_addr.sin_port = server_port;
TSDebug(PLUGIN_NAME, "net connect.");
action = TSNetConnect(contp, (struct sockaddr const *)&ip_addr);
if (!TSActionDone(action)) {
data->pending_action = action;
}
return 0;
}
static void
ts_chunked_destroy_transform_ctx(ts_chunked_transform_ctx *transform_ctx)
{
if (!transform_ctx)
return;
if (transform_ctx->output_reader)
TSIOBufferReaderFree(transform_ctx->output_reader);
if (transform_ctx->output_buffer)
TSIOBufferDestroy(transform_ctx->output_buffer);
TSfree(transform_ctx);
}
static void
handle_dns(TSHttpTxn txnp, TSCont contp ATS_UNUSED)
{
TSMBuffer bufp;
TSMLoc hdr_loc;
TSIOBuffer output_buffer;
TSIOBufferReader reader;
int total_avail;
TSIOBufferBlock block;
const char *block_start;
int64_t block_avail;
char *output_string;
int64_t output_len;
if (TSHttpTxnClientReqGet(txnp, &bufp, &hdr_loc) != TS_SUCCESS) {
TSDebug(PLUGIN_NAME, "couldn't retrieve client request header");
TSError("[%s] Couldn't retrieve client request header", PLUGIN_NAME);
goto done;
}
output_buffer = TSIOBufferCreate();
reader = TSIOBufferReaderAlloc(output_buffer);
/* This will print just MIMEFields and not
the http request line */
TSDebug(PLUGIN_NAME, "Printing the hdrs ... ");
TSMimeHdrPrint(bufp, hdr_loc, output_buffer);
if (TSHandleMLocRelease(bufp, TS_NULL_MLOC, hdr_loc) == TS_ERROR) {
TSDebug(PLUGIN_NAME, "non-fatal: error releasing MLoc");
TSError("[%s] non-fatal: Couldn't release MLoc", PLUGIN_NAME);
}
/* Find out how the big the complete header is by
seeing the total bytes in the buffer. We need to
look at the buffer rather than the first block to
see the size of the entire header */
total_avail = TSIOBufferReaderAvail(reader);
/* Allocate the string with an extra byte for the string
terminator */
output_string = (char *)TSmalloc(total_avail + 1);
output_len = 0;
/* We need to loop over all the buffer blocks to make
sure we get the complete header since the header can
be in multiple blocks */
block = TSIOBufferReaderStart(reader);
while (block) {
block_start = TSIOBufferBlockReadStart(block, reader, &block_avail);
/* We'll get a block pointer back even if there is no data
left to read so check for this condition and break out of
the loop. A block with no data to read means we've exhausted
buffer of data since if there was more data on a later
block in the chain, this block would have been skipped over */
if (block_avail == 0) {
break;
}
memcpy(output_string + output_len, block_start, block_avail);
output_len += block_avail;
/* Consume the data so that we get to the next block */
TSIOBufferReaderConsume(reader, block_avail);
/* Get the next block now that we've consumed the
data off the last block */
block = TSIOBufferReaderStart(reader);
}
/* Terminate the string */
output_string[output_len] = '\0';
output_len++;
/* Free up the TSIOBuffer that we used to print out the header */
TSIOBufferReaderFree(reader);
TSIOBufferDestroy(output_buffer);
/* Although I'd never do this a production plugin, printf
the header so that we can see it's all there */
TSDebug(PLUGIN_NAME, "%s", output_string);
TSfree(output_string);
done:
TSHttpTxnReenable(txnp, TS_EVENT_HTTP_CONTINUE);
}
