Function psi_include

example/plugins/c-api/thread_pool/psi.cc:457–530

------------------------------------------------------------------------- psi_include Read file to include. Copy its content into an iobuffer. This is the function doing blocking calls and called by the plugin's threads Input: data continuation for the current transaction Output : data->psi_buffer contains the file content data->psi_success 0 if include failed, 1 if

    1  if success
  -------------------------------------------------------------------------*/
static int
psi_include(TSCont contp, void *edata ATS_UNUSED)
{
#define BUFFER_SIZE 1024
  ContData *data;
  TSFile    filep;
  char      inc_file[PSI_PATH_MAX_SIZE + PSI_FILENAME_MAX_SIZE];

  /* We manipulate plugin continuation data from a separate thread.
     Grab mutex to avoid concurrent access */
  TSMutexLock(TSContMutexGet(contp));
  data = contDataGet(contp);
  TSAssert(data->magic == MAGIC_ALIVE);

  if (!data->psi_buffer) {
    data->psi_buffer = TSIOBufferCreate();
    data->psi_reader = TSIOBufferReaderAlloc(data->psi_buffer);
  }

  /* For security reason, we do not allow to include files that are
     not in the directory <plugin_path>/include.
     Also include file cannot contain any path. */
  snprintf(inc_file, sizeof(inc_file), "%s/%s", psi_directory, _basename(data->psi_filename));

  /* Read the include file and copy content into iobuffer */
  if ((filep = TSfopen(inc_file, "r")) != nullptr) {
    Dbg(dbg_ctl, "Reading include file %s", inc_file);

    char buf[BUFFER_SIZE];
    while (TSfgets(filep, buf, BUFFER_SIZE) != nullptr) {
      int64_t len, ndone, ntodo;

      len   = strlen(buf);
      ndone = 0;
      ntodo = len;
      while (ntodo > 0) {
        /* TSIOBufferStart allocates more blocks if required */
        TSIOBufferBlock block = TSIOBufferStart(data->psi_buffer);
        int64_t         avail;
        char           *ptr_block = TSIOBufferBlockWriteStart(block, &avail);
        int64_t         towrite   = MIN(ntodo, avail);

        memcpy(ptr_block, buf + ndone, towrite);
        TSIOBufferProduce(data->psi_buffer, towrite);
        ntodo -= towrite;
        ndone += towrite;
      }
    }
    TSfclose(filep);
    data->psi_success = 1;
    if (ts_log) {
      TSTextLogObjectWrite(ts_log, "Successfully included file: %s", inc_file);
    }
  } else {
    data->psi_success = 0;
    if (ts_log) {
      TSTextLogObjectWrite(ts_log, "Failed to include file: %s", inc_file);
    }