Function handle_write_header

plugins/experimental/icap/icap_plugin.cc:530–574

}

static int
handle_write_header(TSCont contp, TransformData *data)
{
  data->state        = State::WRITE_HEADER;
  data->input_buf    = TSIOBufferCreate();
  data->input_reader = TSIOBufferReaderAlloc(data->input_buf);
  data->icap_vio     = TSVConnWrite(data->icap_vc, contp, data->input_reader, INT64_MAX);

  /* Acquire client request and server response header */
  TSMBuffer bufp_c, bufp_s;
  TSMLoc    req_loc, resp_loc;

  if (TSHttpTxnClientReqGet(data->txn, &bufp_c, &req_loc) != TS_SUCCESS) {
    TSError("[%s] Couldn't retrieve client request header", PLUGIN_NAME);
    return 0;
  }

  if (TSHttpTxnServerRespGet(data->txn, &bufp_s, &resp_loc) != TS_SUCCESS) {
    TSError("[%s] Couldn't retrieve server response header", PLUGIN_NAME);
    TSHandleMLocRelease(bufp_c, TS_NULL_MLOC, req_loc);
    return 0;
  }

  int64_t client_req_size  = TSHttpHdrLengthGet(bufp_c, req_loc);
  int64_t server_resp_size = TSHttpHdrLengthGet(bufp_s, resp_loc);
  /* formulate the ICAP request header */
  char res_buf[1000];
  memset(res_buf, 0, 1000);
  snprintf(res_buf, sizeof(res_buf),
           "RESPMOD %s ICAP/%s\r\n"
           "Host: %s\r\n"
           "Connection: close\r\n" // "Connection: close" is used since each scan creates a new connection
           "Encapsulated: req-hdr=0, res-hdr=%" PRIu64 ", res-body=%" PRIu64 "\r\n\r\n",
           ICAP_SERVICE_URL, ICAP_VERSION, server_ip.c_str(), client_req_size, server_resp_size + client_req_size);

  TSIOBufferWrite(data->input_buf, (const char *)res_buf, strlen(res_buf));
  TSHttpHdrPrint(bufp_c, req_loc, data->input_buf);
  TSHttpHdrPrint(bufp_s, resp_loc, data->input_buf);
  data->done_write += TSIOBufferReaderAvail(data->input_reader);

  TSHandleMLocRelease(bufp_c, TS_NULL_MLOC, req_loc);
  TSHandleMLocRelease(bufp_s, TS_NULL_MLOC, resp_loc);

  return transform_write_body(data);
}

static int
handle_write_body(TSCont contp, TransformData *data)