/*
 * Copyright (c) 2015-2018 Nitrokey UG
 *
 * This file is part of libnitrokey.
 *
 * libnitrokey is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * any later version.
 *
 * libnitrokey is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with libnitrokey. If not, see <http://www.gnu.org/licenses/>.
 *
 * SPDX-License-Identifier: LGPL-3.0
 */

static const int TOO_LONG_STRING = 200;

#include "catch.hpp"

#include <iostream>
#include <string>
#include "log.h"
#include "../NK_C_API.h"

int login;

TEST_CASE("C API connect", "[BASIC]") {
  login = NK_login_auto();
  REQUIRE(login != 0);
  NK_logout();
  login = NK_login_auto();
  REQUIRE(login != 0);
  NK_logout();
  login = NK_login_auto();
  REQUIRE(login != 0);
}

TEST_CASE("Check retry count", "[BASIC]") {
  REQUIRE(login != 0);
  NK_set_debug_level(3);
  REQUIRE(NK_get_admin_retry_count() == 3);
  REQUIRE(NK_get_user_retry_count() == 3);
}

TEST_CASE("Check long strings", "[STANDARD]") {
  REQUIRE(login != 0);
  const char* longPin = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
  const char* pin = "123123123";
  auto result = NK_change_user_PIN(longPin, pin);
  REQUIRE(result == TOO_LONG_STRING);
  result = NK_change_user_PIN(pin, longPin);
  REQUIRE(result == TOO_LONG_STRING);
  CAPTURE(result);
}

#include <string.h>

TEST_CASE("multiple devices with ID", "[BASIC]") {
  NK_logout();
  NK_set_debug_level(3);
  auto s = NK_list_devices_by_cpuID();
  REQUIRE(s!=nullptr);
  REQUIRE(strnlen(s, 4096) < 4096);
  REQUIRE(strnlen(s, 4096) > 2*4);
  std::cout << s << std::endl;

  char *string, *token;
  int t;

  string = strndup(s, 4096);
  free (static_cast<void*>(const_cast<char*>(s)));

  while ((token = strsep(&string, ";")) != nullptr){
    if (strnlen(token, 4096) < 3) continue;
    std::cout << token << " connecting: ";
    std::cout << (t=NK_connect_with_ID(token)) << std::endl;
    REQUIRE(t == 1);
  }

  free (string);
}

TEST_CASE("Get device model", "[BASIC]") {
    NK_logout();
    NK_device_model model = NK_get_device_model();
    REQUIRE(model == NK_device_model::NK_DISCONNECTED);

    auto success = NK_login_auto() == 1;
    REQUIRE(success);
    model = NK_get_device_model();
    REQUIRE(model != NK_device_model::NK_DISCONNECTED);

    REQUIRE((model == NK_PRO || model == NK_STORAGE));
    NK_logout();
}