Function intercept_hook

plugins/slice/intercept.cc:26–82

#include "slice.h"

int
intercept_hook(TSCont contp, TSEvent event, void *edata)
{
  Data *const data = static_cast<Data *>(TSContDataGet(contp));

  if (nullptr == data) {
    ERROR_LOG("intercept_hook called without data");
    TSContDestroy(contp);
    return TS_EVENT_ERROR;
  }

  // After the initial TS_EVENT_NET_ACCEPT
  // any "events" will be handled by the vio read or write channel handler
  switch (event) {
  case TS_EVENT_NET_ACCEPT: {
    // set up reader from client
    TSVConn const downvc = static_cast<TSVConn>(edata);
    data->m_dnstream.setupConnection(downvc);
    data->m_dnstream.setupVioRead(contp, INT64_MAX);
  } break;

  case TS_EVENT_NET_ACCEPT_FAILED:
  case TS_EVENT_VCONN_INACTIVITY_TIMEOUT:
  case TS_EVENT_VCONN_ACTIVE_TIMEOUT:
  case TS_EVENT_ERROR: {
    abort(contp, data);
  } break;

  default: {
    // data from client -- only the initial header
    if (data->m_dnstream.m_read.isOpen() && edata == data->m_dnstream.m_read.m_vio) {
      if (handle_client_req(contp, event, data)) {
        // DEBUG_LOG("shutting down read from client pipe");
        TSVConnShutdown(data->m_dnstream.m_vc, 1, 0);
      }
    }
    // server wants more data from us, should never happen
    // every time TSHttpConnect is called this resets
    else if (data->m_upstream.m_write.isOpen() && edata == data->m_upstream.m_write.m_vio) {
      // DEBUG_LOG("shutting down send to server pipe");
      TSVConnShutdown(data->m_upstream.m_vc, 0, 1);
    }
    // server has data for us
    else if (data->m_upstream.m_read.isOpen() && edata == data->m_upstream.m_read.m_vio) {
      handle_server_resp(contp, event, data);
    }
    // client wants more data from us, only body content
    else if (data->m_dnstream.m_write.isOpen() && edata == data->m_dnstream.m_write.m_vio) {
      handle_client_resp(contp, event, data);
    } else {
      ERROR_LOG("Unhandled event: %d", event);
    }
  } break;
  }

  return TS_EVENT_CONTINUE;
}