iridium-c/test/test-crypto.c

#include <glib.h>
#include <gio/gio.h>
#include <string.h>
#include <json-glib/json-glib.h>
#include "iridium-crypto.h"
#include "iridium-standard-file.h"

typedef struct {
  JsonParser *parser;
} Dump;

static void
dump_003_set_up (Dump *dump, gconstpointer user_data)
{
  GBytes *data;
  GError *error = NULL;

  data = g_resources_lookup_data ("/net/bloerg/Test/dumps/003.encrypted.json",
                                  G_RESOURCE_LOOKUP_FLAGS_NONE, NULL);
  g_assert_nonnull (data);

  dump->parser = json_parser_new_immutable ();

  g_assert (json_parser_load_from_data (dump->parser, g_bytes_get_data (data, NULL),
                                        g_bytes_get_size(data), &error));

  g_assert_null (error);
  g_bytes_unref (data);
}

static void
dump_003_tear_down (Dump *dump, gconstpointer user_data)
{
  g_object_unref (dump->parser);
}

static void
test_hexlify (void)
{
  guint8 data[] = { 'A', '1', 'x', ';' };
  gchar *hexlified = iridium_crypto_hexlify (data, 4);
  g_assert_cmpstr (hexlified, ==, "4131783b");
  g_free (hexlified);
}

static void
test_unhexlify (void)
{
  gchar *hexlified = "466f6f62617221";
  guint8 *data = iridium_crypto_unhexlify (hexlified, strlen (hexlified));
  g_assert_cmpstr ((gchar *) data, ==, "Foobar!");
  g_free (data);
}

static void
test_decrypt_003_wrong_password (Dump *dump, gconstpointer user_data)
{
  IridiumAuthParams *params;
  GList *items;
  GError *error = NULL;

  params = iridium_crypto_auth_params_new ();
  iridium_crypto_derive_keys (params, "foo");
  items = iridium_standard_file_get_items (dump->parser, params, &error);
  g_assert_null (items);
  g_assert_nonnull (error);
  g_assert (error->code == IRIDIUM_CRYPTO_ERROR_DECRYPT_ENCRYPTION_KEY);
  iridium_crypto_auth_params_free (params);
}

static void
test_decrypt_003_correct_password (Dump *dump, gconstpointer user_data)
{
  IridiumAuthParams *params;
  JsonNode *root;
  JsonNode *nonce_node;
  GList *items;
  GError *error = NULL;

  params = iridium_crypto_auth_params_new ();
  root = json_parser_get_root (dump->parser);
  g_assert_nonnull (root);

  nonce_node = json_path_query ("$.auth_params.pw_nonce", root, &error);
  g_assert_nonnull (nonce_node);
  g_assert_null (error);
  g_free (params->salt);
  params->salt = g_strdup(json_node_get_string(
      json_array_get_element(json_node_get_array(nonce_node), 0)));
  json_node_unref (nonce_node);

  iridium_crypto_derive_keys (params, "test123");
  items = iridium_standard_file_get_items (dump->parser, params, &error);
  g_assert_nonnull (items);
  g_assert_null (error);
  iridium_crypto_auth_params_free (params);
}

int
main (int argc, char * argv[])
{
  g_test_init (&argc, &argv, NULL);
  g_test_add_func ("/crypto/hexlify", test_hexlify);
  g_test_add_func ("/crypto/unhexlify", test_unhexlify);

  g_test_add ("/crypto/decrypt/003/wrong-password", Dump, NULL,
              dump_003_set_up, test_decrypt_003_wrong_password, dump_003_tear_down);

  g_test_add ("/crypto/decrypt/003/correct-password", Dump, NULL,
              dump_003_set_up, test_decrypt_003_correct_password, dump_003_tear_down);

  return g_test_run ();
}