/*
 * Copyright 2016 Chris Johns <chrisj@rtems.org>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>

#include <assert.h>
#include <ctype.h>
#include <errno.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sysexits.h>
#include <unistd.h>

#include <machine/rtems-bsd-rc-conf.h>
#include <machine/rtems-bsd-rc-conf-services.h>

#define TEST_NAME "LIBBSD RC.CONF 1"

static const char* rc_conf_regex = \
  "\n"                              /* empty line */ \
  "#\n"                             /* comment then nothing */  \
  "# comment\n"                     /* comment */
  "   #   \n"                       /* whitespace (ws), commend, ws */ \
  "test_regex_1=\n"                 /* name, empty value */ \
  "test_regex_2=\"\"\n"             /* name, quoted empty value */ \
  "test_regex_3=\"1 2 3\"\n"        /* name, 3 args */ \
  "  test_regex_4=\"04\"\n"         /* ws, name, 1 arg */ \
  "\ttest_regex_5=\"05\" \n"        /* ws, name, 1 arg, ws */ \
  "test_regex_6=\" 1  2\t 3 \"\n";  /* name then ws and 3 args */

#define NUM_OF_TEST_REGEX_ 6
static bool test_regex_results[NUM_OF_TEST_REGEX_];
static int  test_regex_last_num;
static int  test_service_last_num;

static const char* rc_conf_not_found = \
  "# invalid directive.\n" \
  "abc_def_0=\"not found\"\n";

static int
test_service(rtems_bsd_rc_conf* rc_conf)
{
  rtems_bsd_rc_conf_argc_argv* aa;
  int                          r;

  test_service_last_num = 1;

  /*
   * Fill match of anything.
   */
  puts("test_service regex: 'test_regex_.*'");
  test_regex_last_num = 0;

  assert((aa = rtems_bsd_rc_conf_argc_argv_create()) != NULL);
  r = rtems_bsd_rc_conf_find(rc_conf, "test_regex_.*", aa);
  assert(r == 0 || (r < 0 && errno == ENOENT));
  if (r < 0 && errno == ENOENT)
    return -1;

  while (r == 0) {
    int num;
    int arg;
    rtems_bsd_rc_conf_print_cmd(rc_conf, "test_service[1]", aa->argc, aa->argv);
    assert(strncasecmp(aa->argv[0], "test_regex_", strlen("test_regex_")) == 0);
    num = atoi(aa->argv[0] + strlen("test_regex_"));
    assert(num == (test_regex_last_num + 1));
    assert((num - 1) < NUM_OF_TEST_REGEX_);
    for (arg = 0; arg < aa->argc; ++arg) {
      const char* a = aa->argv[arg];
      size_t      l = strlen(a);
      if (l > 0) {
        assert(!isspace(a[0]));
        assert(!isspace(a[l - 1]));
        assert(a[0] != '"');
        assert(a[l - 1] != '"');
      }
    }
    test_regex_results[num - 1] = true;
    ++test_regex_last_num;
    r = rtems_bsd_rc_conf_find_next(rc_conf, aa);
    assert(r == 0 || (r < 0 && errno == ENOENT));
  }

  /*
   * Specific match of only numbers.
   */
  puts("test_service regex: 'test_regex_[0-9]+'");
  test_regex_last_num = 0;

  assert((aa = rtems_bsd_rc_conf_argc_argv_create()) != NULL);
  r = rtems_bsd_rc_conf_find(rc_conf, "test_regex_[0-9]+", aa);
  assert(r == 0 || (r < 0 && errno == ENOENT));
  if (r < 0 && errno == ENOENT)
    return -1;

  while (r == 0) {
    int num;
    int arg;
    rtems_bsd_rc_conf_print_cmd(rc_conf, "test_service[2]", aa->argc, aa->argv);
    assert(strncasecmp(aa->argv[0], "test_regex_", strlen("test_regex_")) == 0);
    num = atoi(aa->argv[0] + strlen("test_regex_"));
    assert(num == (test_regex_last_num + 1));
    assert((num - 1) < NUM_OF_TEST_REGEX_);
    for (arg = 0; arg < aa->argc; ++arg) {
      const char* a = aa->argv[arg];
      size_t      l = strlen(a);
      if (l > 0) {
        assert(!isspace(a[0]));
        assert(!isspace(a[l - 1]));
        assert(a[0] != '"');
        assert(a[l - 1] != '"');
      }
    }
    test_regex_results[num - 1] = true;
    ++test_regex_last_num;
    r = rtems_bsd_rc_conf_find_next(rc_conf, aa);
    assert(r == 0 || (r < 0 && errno == ENOENT));
  }

  rtems_bsd_rc_conf_argc_argv_destroy(aa);
  puts("test_service done");
  return 0;
}

static int
test_service_2(rtems_bsd_rc_conf* rc_conf)
{
  puts("test_service_2");
  assert(test_service_last_num == 1);
  test_service_last_num = 2;
  return 0;
}

static int
test_service_3(rtems_bsd_rc_conf* rc_conf)
{
  puts("test_service_3");
  assert(test_service_last_num == 2);
  test_service_last_num = 3;
  return 0;
}

static int
test_service_4(rtems_bsd_rc_conf* rc_conf)
{
  puts("test_service_4");
  assert(test_service_last_num == 3);
  test_service_last_num = 4;
  return 0;
}

static int
test_service_5(rtems_bsd_rc_conf* rc_conf)
{
  puts("test_service_5");
  assert(test_service_last_num == 4);
  test_service_last_num = 5;
  return 0;
}

static int
test_service_bad(rtems_bsd_rc_conf* rc_conf)
{
  puts("test_service_bad");
  return -1;
}

static void
make_rc_conf(const char* rc_conf)
{
  FILE* f;
  unlink(rc_conf); /* Ignore any errors */
  assert((f = fopen(rc_conf, "w")) != NULL);
  assert(fwrite(rc_conf_regex, strlen(rc_conf_regex), 1, f) == 1);
  assert(fclose(f) == 0);
}

static void
test_regex_check(void)
{
  int i;
  assert(test_regex_last_num == NUM_OF_TEST_REGEX_); /* all items found? */
  for (i = 0; i < NUM_OF_TEST_REGEX_; ++i)
    assert(test_regex_results[i]);
}

static void
test_regex_reset(void)
{
  memset(&test_regex_results[0], 0, sizeof(test_regex_results));
  test_regex_last_num = 0;
  test_service_last_num = 0;
 }

static void
test_etc_rc_conf(void)
{
  puts("test_etc_rc_conf");
  make_rc_conf("/etc/rc.conf");
  test_regex_reset();
  assert(rtems_bsd_run_etc_rc_conf(0, true) == 0);
  test_regex_check();
}

static void
test_rc_conf(void)
{
  puts("test_rc_conf");
  make_rc_conf("/my_rc.conf");
  test_regex_reset();
  assert(rtems_bsd_run_rc_conf("/my_rc.conf", 0, true) == 0);
  test_regex_check();
}

static void
test_rc_conf_script(void)
{
  puts("test_rc_conf_conf");
  test_regex_reset();
  assert(rtems_bsd_run_rc_conf_script("internal", rc_conf_regex, 0, true) == 0);
  test_regex_check();
}

static void
test_rc_conf_script_not_found(void)
{
  puts("test_rc_conf_conf_not_found");
  test_regex_reset();
  assert(rtems_bsd_run_rc_conf_script("internal", rc_conf_not_found, 0, true) < 0);
  assert(test_regex_last_num == 0);
}

static void
setup(void)
{
  struct stat sb;
  mkdir("/etc", S_IRWXU | S_IRWXG | S_IRWXO); /* ignore errors, check the dir after. */
  assert(stat("/etc", &sb) == 0);
  assert(S_ISDIR(sb.st_mode));
  assert(rtems_bsd_rc_conf_service_add("test_service_2",
                                       "before:last;",
                                       test_service_2) == 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_5",
                                       "after:test_service_2;",
                                       test_service_5) == 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_4",
                                       "before:test_service_5;",
                                       test_service_4) == 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_3",
                                       "before:test_service_4;after:test_service_2;",
                                       test_service_3) == 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_bad",
                                       "before:first;",
                                       test_service_bad) < 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_bad",
                                       "after:last;",
                                       test_service_bad) < 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_bad",
                                       "after:xxxx,xxxx",
                                       test_service_bad) < 0);
  assert(rtems_bsd_rc_conf_service_add("test_service_bad",
                                       "yyyy:xxxx;",
                                       test_service_bad) < 0);
  assert(rtems_bsd_rc_conf_service_add("test_service",
                                       "after:first;",
                                       test_service) == 0);
}

static void
test_main(void)
{
  setup();

  test_etc_rc_conf();
  test_rc_conf();
  test_rc_conf_script();
  test_rc_conf_script_not_found();

  exit(0);
}

#include <rtems/bsd/test/default-init.h>