/*****************************************************************************
 * Author:   Valient Gough <vgough@pobox.com>
 *
 *****************************************************************************
 * Copyright (c) 2013 Valient Gough
 *
 * This program 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 (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 Lesser General Public License for more
 * details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


#include <string.h>

#include <gtest/gtest.h>

#include "base/shared_ptr.h"
#include "cipher/PBKDF.h"
#include "cipher/testing.h"

using namespace encfs;

namespace {

TEST(PKCS5_PBKDF2_HMAC_SHA1, PBKDF) {
  Registry<PBKDF> registry = PBKDF::GetRegistry();
  shared_ptr<PBKDF> impl( registry.CreateForMatch(NAME_PKCS5_PBKDF2_HMAC_SHA1));
  ASSERT_FALSE(!impl);

  // Test cases from rfc6070
  // Test case 1
  {
    CipherKey key(20);
    bool ok = impl->makeKey("password", 8, 
                            (byte*)"salt", 4,
                            1, &key);
    ASSERT_TRUE(ok);
    ASSERT_EQ("0c60c80f961f0e71f3a9b524af6012062fe037a6", stringToHex(key));
  }

  {
    CipherKey key(25);
    bool ok = impl->makeKey("passwordPASSWORDpassword", 24, 
                            (byte*)"saltSALTsaltSALTsaltSALTsaltSALTsalt", 36,
                            4096, &key);
    ASSERT_TRUE(ok);
    ASSERT_EQ("3d2eec4fe41c849b80c8d83662c0e44a8b291a964cf2f07038",
              stringToHex(key));
  }

  {
    CipherKey key(16);
    bool ok = impl->makeKey("pass\0word", 9, 
                            (byte*)"sa\0lt", 5,
                            4096, &key);
    ASSERT_TRUE(ok);
    ASSERT_EQ("56fa6aa75548099dcc37d7f03425e0c3", stringToHex(key));
  }
}


}  // namespace