/*
 * s3fs - FUSE-based file system backed by Amazon S3
 *
 * Copyright(C) 2014 Andrew Gaul <andrew@gaul.org>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#include <cstdlib>
#include <limits>
#include <stdint.h>
#include <string>

#include "s3fs_logger.h"
#include "string_util.h"
#include "test_util.h"

//-------------------------------------------------------------------
// Global variables for test_string_util
//-------------------------------------------------------------------
bool foreground                   = false;
std::string instance_name;

void test_trim()
{
    ASSERT_EQUALS(std::string("1234"), trim("  1234  "));
    ASSERT_EQUALS(std::string("1234"), trim("1234  "));
    ASSERT_EQUALS(std::string("1234"), trim("  1234"));
    ASSERT_EQUALS(std::string("1234"), trim("1234"));

    ASSERT_EQUALS(std::string("1234  "), trim_left("  1234  "));
    ASSERT_EQUALS(std::string("1234  "), trim_left("1234  "));
    ASSERT_EQUALS(std::string("1234"), trim_left("  1234"));
    ASSERT_EQUALS(std::string("1234"), trim_left("1234"));

    ASSERT_EQUALS(std::string("  1234"), trim_right("  1234  "));
    ASSERT_EQUALS(std::string("1234"), trim_right("1234  "));
    ASSERT_EQUALS(std::string("  1234"), trim_right("  1234"));
    ASSERT_EQUALS(std::string("1234"), trim_right("1234"));
}

void test_base64()
{
    unsigned char *buf;
    size_t len;

    ASSERT_STREQUALS(s3fs_base64(nullptr, 0).c_str(), nullptr);
    buf = s3fs_decode64(nullptr, 0, &len);
    ASSERT_BUFEQUALS(reinterpret_cast<const char *>(buf), len, nullptr, 0);

    ASSERT_STREQUALS(s3fs_base64(reinterpret_cast<const unsigned char *>(""), 0).c_str(), nullptr);
    buf = s3fs_decode64("", 0, &len);
    ASSERT_BUFEQUALS(reinterpret_cast<const char *>(buf), len, nullptr, 0);

    ASSERT_EQUALS(s3fs_base64(reinterpret_cast<const unsigned char *>("1"), 1), std::string("MQ=="));
    buf = s3fs_decode64("MQ==", 4, &len);
    ASSERT_BUFEQUALS(reinterpret_cast<const char *>(buf), len, "1", 1);
    ASSERT_EQUALS(len, static_cast<size_t>(1));

    ASSERT_EQUALS(s3fs_base64(reinterpret_cast<const unsigned char *>("12"), 2), std::string("MTI="));
    buf = s3fs_decode64("MTI=", 4, &len);
    ASSERT_BUFEQUALS(reinterpret_cast<const char *>(buf), len, "12", 2);
    ASSERT_EQUALS(len, static_cast<size_t>(2));

    ASSERT_EQUALS(s3fs_base64(reinterpret_cast<const unsigned char *>("123"), 3), std::string("MTIz"));
    buf = s3fs_decode64("MTIz", 4, &len);
    ASSERT_BUFEQUALS(reinterpret_cast<const char *>(buf), len, "123", 3);
    ASSERT_EQUALS(len, static_cast<size_t>(3));

    ASSERT_EQUALS(s3fs_base64(reinterpret_cast<const unsigned char *>("1234"), 4), std::string("MTIzNA=="));
    buf = s3fs_decode64("MTIzNA==", 8, &len);
    ASSERT_BUFEQUALS(reinterpret_cast<const char *>(buf), len, "1234", 4);
    ASSERT_EQUALS(len, static_cast<size_t>(4));

    // TODO: invalid input
}

void test_strtoofft()
{
    off_t value;

    ASSERT_TRUE(s3fs_strtoofft(&value, "0"));
    ASSERT_EQUALS(value, static_cast<off_t>(0L));

    ASSERT_TRUE(s3fs_strtoofft(&value, "9"));
    ASSERT_EQUALS(value, static_cast<off_t>(9L));

    ASSERT_FALSE(s3fs_strtoofft(&value, "A"));

    ASSERT_TRUE(s3fs_strtoofft(&value, "A", /*base=*/ 16));
    ASSERT_EQUALS(value, static_cast<off_t>(10L));

    ASSERT_TRUE(s3fs_strtoofft(&value, "F", /*base=*/ 16));
    ASSERT_EQUALS(value, static_cast<off_t>(15L));

    ASSERT_TRUE(s3fs_strtoofft(&value, "a", /*base=*/ 16));
    ASSERT_EQUALS(value, static_cast<off_t>(10L));

    ASSERT_TRUE(s3fs_strtoofft(&value, "f", /*base=*/ 16));
    ASSERT_EQUALS(value, static_cast<off_t>(15L));

    ASSERT_TRUE(s3fs_strtoofft(&value, "deadbeef", /*base=*/ 16));
    ASSERT_EQUALS(value, static_cast<off_t>(3735928559L));
}

void test_wtf8_encoding()
{
    std::string ascii("normal std::string");
    std::string utf8("Hyld\xc3\xbdpi \xc3\xbej\xc3\xb3\xc3\xb0""f\xc3\xa9lagsins vex \xc3\xbar k\xc3\xa6rkomnu b\xc3\xb6li \xc3\xad \xc3\xa1st");
    std::string cp1252("Hyld\xfdpi \xfej\xf3\xf0""f\xe9lagsins vex \xfar k\xe6rkomnu b\xf6li \xed \xe1st");
    std::string broken = utf8;
    broken[14] = 0x97;
    std::string mixed = ascii + utf8 + cp1252;

    ASSERT_EQUALS(s3fs_wtf8_encode(ascii), ascii);
    ASSERT_EQUALS(s3fs_wtf8_decode(ascii), ascii);
    ASSERT_EQUALS(s3fs_wtf8_encode(utf8), utf8);
    ASSERT_EQUALS(s3fs_wtf8_decode(utf8), utf8);

    ASSERT_NEQUALS(s3fs_wtf8_encode(cp1252), cp1252);
    ASSERT_EQUALS(s3fs_wtf8_decode(s3fs_wtf8_encode(cp1252)), cp1252);

    ASSERT_NEQUALS(s3fs_wtf8_encode(broken), broken);
    ASSERT_EQUALS(s3fs_wtf8_decode(s3fs_wtf8_encode(broken)), broken);

    ASSERT_NEQUALS(s3fs_wtf8_encode(mixed), mixed);
    ASSERT_EQUALS(s3fs_wtf8_decode(s3fs_wtf8_encode(mixed)), mixed);
}

void test_cr_encoding()
{
    // bse strings
    std::string base_no("STR");

    std::string base_end_cr1("STR\r");
    std::string base_mid_cr1("STR\rSTR");
    std::string base_end_cr2("STR\r\r");
    std::string base_mid_cr2("STR\r\rSTR");

    std::string base_end_per1("STR%");
    std::string base_mid_per1("STR%STR");
    std::string base_end_per2("STR%%");
    std::string base_mid_per2("STR%%STR");

    std::string base_end_crlf1("STR\r\n");
    std::string base_mid_crlf1("STR\r\nSTR");
    std::string base_end_crlf2("STR\r\n\r\n");
    std::string base_mid_crlf2("STR\r\n\r\nSTR");

    std::string base_end_crper1("STR%\r");
    std::string base_mid_crper1("STR%\rSTR");
    std::string base_end_crper2("STR%\r%\r");
    std::string base_mid_crper2("STR%\r%\rSTR");

    // encode->decode->compare
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_no.c_str()).c_str()),         base_no);

    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_cr1.c_str()).c_str()),    base_end_cr1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_cr1.c_str()).c_str()),    base_mid_cr1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_cr2.c_str()).c_str()),    base_end_cr2);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_cr2.c_str()).c_str()),    base_mid_cr2);

    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_per1.c_str()).c_str()),   base_end_per1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_per1.c_str()).c_str()),   base_mid_per1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_per2.c_str()).c_str()),   base_end_per2);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_per2.c_str()).c_str()),   base_mid_per2);

    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_crlf1.c_str()).c_str()),  base_end_crlf1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_crlf1.c_str()).c_str()),  base_mid_crlf1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_crlf2.c_str()).c_str()),  base_end_crlf2);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_crlf2.c_str()).c_str()),  base_mid_crlf2);

    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_crper1.c_str()).c_str()), base_end_crper1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_crper1.c_str()).c_str()), base_mid_crper1);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_end_crper2.c_str()).c_str()), base_end_crper2);
    ASSERT_EQUALS(get_decoded_cr_code(get_encoded_cr_code(base_mid_crper2.c_str()).c_str()), base_mid_crper2);
}

int main(int argc, char *argv[])
{
    S3fsLog singletonLog;

    test_trim();
    test_base64();
    test_strtoofft();
    test_wtf8_encoding();
    test_cr_encoding();

    return 0;
}

/*
* Local variables:
* tab-width: 4
* c-basic-offset: 4
* End:
* vim600: expandtab sw=4 ts=4 fdm=marker
* vim<600: expandtab sw=4 ts=4
*/