/* Copyright Fedor Indutny. All rights reserved.

 *

 * Permission is hereby granted, free of charge, to any person obtaining a copy

 * of this software and associated documentation files (the "Software"), to

 * deal in the Software without restriction, including without limitation the

 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or

 * sell copies of the Software, and to permit persons to whom the Software is

 * furnished to do so, subject to the following conditions:

 *

 * The above copyright notice and this permission notice shall be included in

 * all copies or substantial portions of the Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS

 * IN THE SOFTWARE.

 */

#include "http_parser.h"

#include <assert.h>

#include <stdio.h>

#include <string.h>

#include <sys/time.h>

static const char data[] =

    "POST /joyent/http-parser HTTP/1.1\r\n"

    "Host: github.com\r\n"

    "DNT: 1\r\n"

    "Accept-Encoding: gzip, deflate, sdch\r\n"

    "Accept-Language: ru-RU,ru;q=0.8,en-US;q=0.6,en;q=0.4\r\n"

    "User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) "

        "AppleWebKit/537.36 (KHTML, like Gecko) "

        "Chrome/39.0.2171.65 Safari/537.36\r\n"

    "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,"

        "image/webp,*/*;q=0.8\r\n"

    "Referer: https://github.com/joyent/http-parser\r\n"

    "Connection: keep-alive\r\n"

    "Transfer-Encoding: chunked\r\n"

    "Cache-Control: max-age=0\r\n\r\nb\r\nhello world\r\n0\r\n\r\n";

static const size_t data_len = sizeof(data) - 1;

static int on_info(http_parser* p) {

  return 0;

}

static int on_data(http_parser* p, const char *at, size_t length) {

  return 0;

}

static http_parser_settings settings = {

  .on_message_begin = on_info,

  .on_headers_complete = on_info,

  .on_message_complete = on_info,

  .on_header_field = on_data,

  .on_header_value = on_data,

  .on_url = on_data,

  .on_status = on_data,

  .on_body = on_data

};

int bench(int iter_count, int silent) {

  struct http_parser parser;

  int i;

  int err;

  struct timeval start;

  struct timeval end;

  float rps;

  if (!silent) {

    err = gettimeofday(&start, NULL);

    assert(err == 0);

  }

  for (i = 0; i < iter_count; i++) {

    size_t parsed;

    http_parser_init(&parser, HTTP_REQUEST);

    parsed = http_parser_execute(&parser, &settings, data, data_len);

    assert(parsed == data_len);

  }

  if (!silent) {

    err = gettimeofday(&end, NULL);

    assert(err == 0);

    fprintf(stdout, "Benchmark result:\n");

    rps = (float) (end.tv_sec - start.tv_sec) +

          (end.tv_usec - start.tv_usec) * 1e-6f;

    fprintf(stdout, "Took %f seconds to run\n", rps);

    rps = (float) iter_count / rps;

    fprintf(stdout, "%f req/sec\n", rps);

    fflush(stdout);

  }

  return 0;

}

int main(int argc, char** argv) {

  if (argc == 2 && strcmp(argv[1], "infinite") == 0) {

    for (;;)

      bench(5000000, 1);

    return 0;

  } else {

    return bench(5000000, 0);

  }

}