/* GIO TLS tests
*
* Copyright (C) 2011 Collabora, Ltd.
*
* This library 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 2 of the License, or (at your option) any later version.
*
* This library 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 library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place, Suite 330,
* Boston, MA 02111-1307, USA.
*
* In addition, when the library is used with OpenSSL, a special
* exception applies. Refer to the LICENSE_EXCEPTION file for details.
*
* Author: Stef Walter <stefw@collabora.co.uk>
*/
#include <gio/gio.h>
#include <sys/types.h>
#include <string.h>
#include "pkcs11/gpkcs11pin.h"
typedef struct {
GTlsPassword *pin;
} TestPin;
static void
setup_pin (TestPin *test,
gconstpointer unused)
{
test->pin = g_pkcs11_pin_new (G_TLS_PASSWORD_RETRY, "Test description");
g_assert (G_IS_PKCS11_PIN (test->pin));
g_assert (G_IS_TLS_PASSWORD (test->pin));
}
static void
teardown_pin (TestPin *test,
gconstpointer unused)
{
g_assert_cmpint (G_OBJECT (test->pin)->ref_count, ==, 1);
g_object_unref (test->pin);
}
static void
test_attributes (TestPin *test,
gconstpointer data)
{
GTlsPasswordFlags flags;
const gchar *description;
flags = g_tls_password_get_flags (test->pin);
g_assert_cmpuint (flags, ==, G_TLS_PASSWORD_RETRY);
description = g_tls_password_get_description (test->pin);
g_assert_cmpstr (description, ==, "Test description");
}
static void
test_warnings (TestPin *test,
gconstpointer data)
{
const gchar *warning;
g_tls_password_set_flags (test->pin, G_TLS_PASSWORD_RETRY);
warning = g_tls_password_get_warning (test->pin);
g_assert (warning != NULL);
g_tls_password_set_flags (test->pin, G_TLS_PASSWORD_FINAL_TRY);
warning = g_tls_password_get_warning (test->pin);
g_assert (warning != NULL);
g_tls_password_set_flags (test->pin, G_TLS_PASSWORD_MANY_TRIES);
warning = g_tls_password_get_warning (test->pin);
g_assert (warning != NULL);
g_tls_password_set_flags (test->pin, (GTlsPasswordFlags)0x10000000);
warning = g_tls_password_get_warning (test->pin);
g_assert (warning == NULL);
}
static void
test_set_get_value (TestPin *test,
gconstpointer data)
{
const guchar *value;
gsize n_value = G_MAXSIZE;
value = g_tls_password_get_value (test->pin, &n_value);
g_assert_cmpuint (n_value, ==, 0);
g_assert (value == NULL);
g_tls_password_set_value (test->pin, (const guchar *)"secret", -1);
value = g_tls_password_get_value (test->pin, &n_value);
g_assert_cmpuint (n_value, ==, 6);
g_assert (!strncmp ((const gchar *)value, "secret", n_value));
g_tls_password_set_value (test->pin, (const guchar *)"other", 5);
value = g_tls_password_get_value (test->pin, &n_value);
g_assert_cmpuint (n_value, ==, 5);
g_assert (!strncmp ((const gchar *)value, "other", n_value));
}
static void
test_internal_pin (TestPin *test,
gconstpointer data)
{
P11KitPin *pin;
const unsigned char *value;
size_t n_value;
g_tls_password_set_value (test->pin, (const guchar *)"secret", -1);
pin = g_pkcs11_pin_steal_internal (G_PKCS11_PIN (test->pin));
value = p11_kit_pin_get_value (pin, &n_value);
g_assert_cmpuint (n_value, ==, 6);
g_assert (!strncmp ((const gchar *)value, "secret", n_value));
p11_kit_pin_unref (pin);
}
int
main (int argc,
char *argv[])
{
g_test_init (&argc, &argv, NULL);
g_test_add ("/pkcs11/pin/attributes", TestPin, NULL,
setup_pin, test_attributes, teardown_pin);
g_test_add ("/pkcs11/pin/warnings", TestPin, NULL,
setup_pin, test_warnings, teardown_pin);
g_test_add ("/pkcs11/pin/set-get-value", TestPin, NULL,
setup_pin, test_set_get_value, teardown_pin);
g_test_add ("/pkcs11/pin/internal-pin", TestPin, NULL,
setup_pin, test_internal_pin, teardown_pin);
return g_test_run();
}