curl/tests/libtest/lib670.c

/***************************************************************************
 *                                  _   _ ____  _
 *  Project                     ___| | | |  _ \| |
 *                             / __| | | | |_) | |
 *                            | (__| |_| |  _ <| |___
 *                             \___|\___/|_| \_\_____|
 *
 * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
 *
 * This software is licensed as described in the file COPYING, which
 * you should have received as part of this distribution. The terms
 * are also available at https://curl.se/docs/copyright.html.
 *
 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
 * copies of the Software, and permit persons to whom the Software is
 * furnished to do so, under the terms of the COPYING file.
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 * KIND, either express or implied.
 *
 * SPDX-License-Identifier: curl
 *
 ***************************************************************************/
#include "first.h"

#define PAUSE_TIME      5

struct t670_ReadThis {
  CURL *curl;
  time_t origin;
  int count;
};

static size_t t670_read_cb(char *ptr, size_t size, size_t nmemb, void *userp)
{
  struct t670_ReadThis *pooh = (struct t670_ReadThis *) userp;
  time_t delta;

  if(size * nmemb < 1)
    return 0;

  switch(pooh->count++) {
  case 0:
    *ptr = '\x41'; /* ASCII A. */
    return 1;
  case 1:
    pooh->origin = time(NULL);
    return CURL_READFUNC_PAUSE;
  case 2:
    delta = time(NULL) - pooh->origin;
    *ptr = delta >= PAUSE_TIME ? '\x42' : '\x41'; /* ASCII A or B. */
    return 1;
  case 3:
    return 0;
  }
  curl_mfprintf(stderr, "Read callback called after EOF\n");
  exit(1);
}

static int t670_xferinfo(void *clientp,
                         curl_off_t dltotal, curl_off_t dlnow,
                         curl_off_t ultotal, curl_off_t ulnow)
{
  struct t670_ReadThis *pooh = (struct t670_ReadThis *) clientp;

  (void)dltotal;
  (void)dlnow;
  (void)ultotal;
  (void)ulnow;

  if(pooh->origin) {
    time_t delta = time(NULL) - pooh->origin;

    if(delta >= 4 * PAUSE_TIME) {
      curl_mfprintf(stderr, "unpausing failed: drain problem?\n");
      return CURLE_ABORTED_BY_CALLBACK;
    }

    if(delta >= PAUSE_TIME)
      curl_easy_pause(pooh->curl, CURLPAUSE_CONT);
  }

  return 0;
}

static CURLcode test_lib670(const char *URL)
{
  static const char testname[] = "field";

  curl_mime *mime = NULL;
  struct curl_httppost *formpost = NULL;

  struct t670_ReadThis pooh;
  CURLcode res = TEST_ERR_FAILURE;

  /*
   * Check proper pausing/unpausing from a mime or form read callback.
   */

  if(curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
    curl_mfprintf(stderr, "curl_global_init() failed\n");
    return TEST_ERR_MAJOR_BAD;
  }

  pooh.origin = (time_t) 0;
  pooh.count = 0;
  pooh.curl = curl_easy_init();

  /* First set the URL that is about to receive our POST. */
  test_setopt(pooh.curl, CURLOPT_URL, URL);

  /* get verbose debug output please */
  test_setopt(pooh.curl, CURLOPT_VERBOSE, 1L);

  /* include headers in the output */
  test_setopt(pooh.curl, CURLOPT_HEADER, 1L);

  if(testnum == 670 || testnum == 671) {
    curl_mimepart *part;
    /* Build the mime tree. */
    mime = curl_mime_init(pooh.curl);
    part = curl_mime_addpart(mime);
    res = curl_mime_name(part, testname);
    if(res != CURLE_OK) {
      curl_mfprintf(stderr,
                 "Something went wrong when building the mime structure: %d\n",
                 res);
      goto test_cleanup;
    }

    res = curl_mime_data_cb(part, (curl_off_t) 2, t670_read_cb,
                            NULL, NULL, &pooh);

    /* Bind mime data to its easy handle. */
    if(res == CURLE_OK)
      test_setopt(pooh.curl, CURLOPT_MIMEPOST, mime);
  }
  else {
    struct curl_httppost *lastptr = NULL;
    CURLFORMcode formrc;
    /* Build the form. */
    formrc = curl_formadd(&formpost, &lastptr,
                          CURLFORM_COPYNAME, testname,
                          CURLFORM_STREAM, &pooh,
                          CURLFORM_CONTENTLEN, (curl_off_t) 2,
                          CURLFORM_END);
    if(formrc) {
      curl_mfprintf(stderr, "curl_formadd() = %d\n", formrc);
      goto test_cleanup;
    }

    /* We want to use our own read function. */
    test_setopt(pooh.curl, CURLOPT_READFUNCTION, t670_read_cb);

    /* Send a multi-part formpost. */
    test_setopt(pooh.curl, CURLOPT_HTTPPOST, formpost);
  }

  if(testnum == 670 || testnum == 672) {
    CURLMcode mres;
    CURLM *multi;
    /* Use the multi interface. */
    multi = curl_multi_init();
    mres = curl_multi_add_handle(multi, pooh.curl);
    while(!mres) {
      struct timeval timeout;
      int rc = 0;
      fd_set fdread;
      fd_set fdwrite;
      fd_set fdexcept;
      int maxfd = -1;
      int still_running = 0;

      mres = curl_multi_perform(multi, &still_running);
      if(!still_running || mres != CURLM_OK)
        break;

      if(pooh.origin) {
        time_t delta = time(NULL) - pooh.origin;

        if(delta >= 4 * PAUSE_TIME) {
          curl_mfprintf(stderr, "unpausing failed: drain problem?\n");
          res = CURLE_OPERATION_TIMEDOUT;
          break;
        }

        if(delta >= PAUSE_TIME)
          curl_easy_pause(pooh.curl, CURLPAUSE_CONT);
      }

      FD_ZERO(&fdread);
      FD_ZERO(&fdwrite);
      FD_ZERO(&fdexcept);
      timeout.tv_sec = 0;
      timeout.tv_usec = 1000000 * PAUSE_TIME / 10;
      mres = curl_multi_fdset(multi, &fdread, &fdwrite, &fdexcept, &maxfd);
      if(mres)
        break;
#ifdef _WIN32
      if(maxfd == -1)
        curlx_wait_ms(100);
      else
#endif
      rc = select(maxfd + 1, &fdread, &fdwrite, &fdexcept, &timeout);
      if(rc == -1) {
        curl_mfprintf(stderr, "Select error\n");
        break;
      }
    }

    if(mres != CURLM_OK)
      for(;;) {
        int msgs_left;
        CURLMsg *msg;
        msg = curl_multi_info_read(multi, &msgs_left);
        if(!msg)
          break;
        if(msg->msg == CURLMSG_DONE) {
          res = msg->data.result;
        }
      }

    curl_multi_remove_handle(multi, pooh.curl);
    curl_multi_cleanup(multi);
  }
  else {
    /* Use the easy interface. */
    test_setopt(pooh.curl, CURLOPT_XFERINFODATA, &pooh);
    test_setopt(pooh.curl, CURLOPT_XFERINFOFUNCTION, t670_xferinfo);
    test_setopt(pooh.curl, CURLOPT_NOPROGRESS, 0L);
    res = curl_easy_perform(pooh.curl);
  }

test_cleanup:
  curl_easy_cleanup(pooh.curl);

  if(testnum == 670 || testnum == 671) {
    curl_mime_free(mime);
  }
  else {
    curl_formfree(formpost);
  }

  curl_global_cleanup();
  return res;
}