#include <curl/curl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "memdebug.h"

#define MAIN_LOOP_HANG_TIMEOUT     90 * 1000
#define MULTI_PERFORM_HANG_TIMEOUT 60 * 1000

static int progress_callback(void *clientp, double dltotal, double dlnow, double ultotal, double ulnow)
{
  (void)clientp;
  fprintf(stderr, "Progress %f %f %f %f ...\n", dltotal, dlnow, ultotal, ulnow);
  return 0;
}

static int start_and_cancel(char *URL,int steps)
{
  int res = 0;
  int counter = 0;
  CURL *curl;
  int running;
  CURLM *m = NULL;

  fprintf(stderr, "Starting ...\n");
  if (curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
    fprintf(stderr, "curl_global_init() failed\n");
    return 2;
  }

  if ((curl = curl_easy_init()) == NULL) {
    fprintf(stderr, "curl_easy_init() failed\n");
    curl_global_cleanup();
    return 2;
  }

  curl_easy_setopt(curl, CURLOPT_URL, URL);
  curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
  curl_easy_setopt(curl, CURLOPT_FAILONERROR, 1);

  /* we want to use our own progress function */
  curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0L);
  curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);

  fprintf(stderr, "Initializing ...\n");
  if ((m = curl_multi_init()) == NULL) {
    fprintf(stderr, "curl_multi_init() failed\n");
    curl_easy_cleanup(curl);
    curl_global_cleanup();
    return 2;
  }

  if ((res = (int)curl_multi_add_handle(m, curl)) != CURLM_OK) {
    fprintf(stderr, "curl_multi_add_handle() failed, "
            "with code %d\n", res);
    curl_multi_cleanup(m);
    curl_easy_cleanup(curl);
    curl_global_cleanup();
    return 2;
  }

  fprintf(stderr, "Start at URL %s, steps = %d\n",URL,steps);

  for(counter=0; counter < steps; counter++) {
      usleep(1000);
      fprintf(stderr, "Step for URL %s:%d started ...\n",URL,counter);
      res = (int)curl_multi_perform(m, &running);
      fprintf(stderr, "Step for URL %s:%d finished, res = %d\n",URL,counter,res);
  }

  fprintf(stderr, "Canceling URL %s\n",URL);

  curl_easy_cleanup(curl);
  curl_multi_cleanup(m);
  curl_global_cleanup();

  sleep(2);

  return CURLE_OK;
}

int test(char *URL)
{
  int steps=0;
  for(steps=1; steps < 10; steps++) {
    start_and_cancel(URL,steps);
  }
}

int main(int ac, char **av)
{
    if( ac > 1 )
        test(av[1]);
    else {
        fprintf(stderr, "Usage: %s URL\n",av[0]);
        return 1;
    }
    return 0;
}