Function http_event_handler

plugins/experimental/wasm/wasm_main.cc:374–531

handler for transaction event

// handler for transaction event
static int
http_event_handler(TSCont contp, TSEvent event, void *data)
{
  int   result   = -1;
  auto *context  = static_cast<ats_wasm::Context *>(TSContDataGet(contp));
  auto *old_wasm = static_cast<ats_wasm::Wasm *>(context->wasm());

  context->resetTxnReenable();

  TSMutexLock(old_wasm->mutex());
  std::shared_ptr<ats_wasm::Wasm> temp = nullptr;
  auto                           *txnp = static_cast<TSHttpTxn>(data);

  TSMBuffer buf     = nullptr;
  TSMLoc    hdr_loc = nullptr;
  int       count   = 0;

  switch (event) {
  case TS_EVENT_HTTP_TXN_START:
    break;

  case TS_EVENT_HTTP_READ_REQUEST_HDR:
    if (TSHttpTxnClientReqGet(txnp, &buf, &hdr_loc) != TS_SUCCESS) {
      TSError("[wasm][%s] cannot retrieve client request", __FUNCTION__);
      TSMutexUnlock(old_wasm->mutex());
      TSHttpTxnReenable(txnp, TS_EVENT_HTTP_CONTINUE);
      context->setTxnReenable();
      return 0;
    }
    count = TSMimeHdrFieldsCount(buf, hdr_loc);
    TSHandleMLocRelease(buf, TS_NULL_MLOC, hdr_loc);

    result = context->onRequestHeaders(count, false) == proxy_wasm::FilterHeadersStatus::Continue ? 0 : 1;
    break;

  case TS_EVENT_HTTP_POST_REMAP:
    break;

  case TS_EVENT_HTTP_CACHE_LOOKUP_COMPLETE:
    break;

  case TS_EVENT_HTTP_SEND_REQUEST_HDR:
    break;

  case TS_EVENT_HTTP_READ_RESPONSE_HDR:
    if (TSHttpTxnServerRespGet(txnp, &buf, &hdr_loc) != TS_SUCCESS) {
      TSError("[wasm][%s] cannot retrieve server response", __FUNCTION__);
      TSMutexUnlock(old_wasm->mutex());
      TSHttpTxnReenable(txnp, TS_EVENT_HTTP_CONTINUE);
      context->setTxnReenable();
      return 0;
    }
    count = TSMimeHdrFieldsCount(buf, hdr_loc);
    TSHandleMLocRelease(buf, TS_NULL_MLOC, hdr_loc);

    result = context->onResponseHeaders(count, false) == proxy_wasm::FilterHeadersStatus::Continue ? 0 : 1;
    break;