/***************************************************************************
 *                                  _   _ ____  _
 *  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"

static void t1940_showem(CURL *curl, int header_request, unsigned int type)
{
  static const char * const testdata[] = {
    "daTE",
    "Server",
    "content-type",
    "content-length",
    "location",
    "set-cookie",
    "silly-thing",
    "fold",
    "blank",
    "Blank2",
    "Test",
    NULL
  };

  int i;
  struct curl_header *header;
  for(i = 0; testdata[i]; i++) {
    if(CURLHE_OK == curl_easy_header(curl, testdata[i], 0,
                                     type, header_request, &header)) {
      if(header->amount > 1) {
        /* more than one, iterate over them */
        size_t index = 0;
        size_t amount = header->amount;
        do {
          curl_mprintf("- %s == %s (%zu/%zu)\n", header->name, header->value,
                       index, amount);

          if(++index == amount)
            break;
          if(CURLHE_OK != curl_easy_header(curl, testdata[i], index,
                                           type, header_request, &header))
            break;
        } while(1);
      }
      else {
        /* only one of this */
        curl_mprintf(" %s == %s\n", header->name, header->value);
      }
    }
  }
}

static CURLHcode t1940_negative(CURL *curl, int req_index)
{
  struct curl_header *hd1, *hd2;
  CURLHcode result;

  result = curl_easy_header(NULL, "x", 0, CURLH_HEADER, -1, &hd1);
  if(result != CURLHE_BAD_ARGUMENT)
    return result;

  result = curl_easy_header(curl, NULL, 0, CURLH_HEADER, -1, &hd1);
  if(result != CURLHE_BAD_ARGUMENT)
    return result;

  result = curl_easy_header(curl, "x", 0, CURLH_HEADER, -1, NULL);
  if(result != CURLHE_BAD_ARGUMENT)
    return result;

  result = curl_easy_header(curl, "x", 0, 0, -1, &hd1);
  if(result != CURLHE_BAD_ARGUMENT)
    return result;

  result = curl_easy_header(curl, "date", 0, CURLH_HEADER, req_index, &hd1);
  if(result != CURLHE_OK)
    return result;
  if(!hd1)
    return CURLHE_NOHEADERS;

  /* Should give the first header */
  hd2 = curl_easy_nextheader(curl, CURLH_HEADER, -1, NULL);
  if(!hd2)
    return CURLHE_BADINDEX;

  hd2 = curl_easy_nextheader(NULL, CURLH_HEADER, -1, hd1);
  if(hd2) /* should not work */
    return CURLHE_BADINDEX;

  return CURLHE_OK;
}

static CURLcode test_lib1940(const char *URL)
{
  CURL *curl = NULL;
  CURLcode result = CURLE_OK;

  int header_request;
  if(testnum == 1946) {
    header_request = 0;
  }
  else {
    header_request = -1;
  }

  global_init(CURL_GLOBAL_DEFAULT);
  easy_init(curl);
  easy_setopt(curl, CURLOPT_URL, URL);
  easy_setopt(curl, CURLOPT_VERBOSE, 1L);
  easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
  /* ignores any content */
  easy_setopt(curl, CURLOPT_WRITEFUNCTION, tutil_throwaway_cb);

  /* if there is a proxy set, use it */
  if(libtest_arg2 && *libtest_arg2) {
    easy_setopt(curl, CURLOPT_PROXY, libtest_arg2);
    easy_setopt(curl, CURLOPT_HTTPPROXYTUNNEL, 1L);
  }
  result = curl_easy_perform(curl);
  if(result)
    goto test_cleanup;

  t1940_showem(curl, header_request, CURLH_HEADER);
  if(libtest_arg2 && *libtest_arg2) {
    /* now show connect headers only */
    t1940_showem(curl, header_request, CURLH_CONNECT);
  }
  t1940_showem(curl, header_request, CURLH_1XX);
  t1940_showem(curl, header_request, CURLH_TRAILER);

  result = (CURLcode)t1940_negative(curl, header_request);

test_cleanup:
  curl_easy_cleanup(curl);
  curl_global_cleanup();
  return result;
}
