[PATCH v2 3/3] Add test for C_GenerateRandom.

[Damien Goutte-Gattat]

* tests/Makefile.am (TESTS): Add t-generaterandom test.
* tests/t-generaterandom.c: New file.

Signed-off-by: Damien Goutte-Gattat <dgouttegattat at incenp.org>
---
 tests/Makefile.am        |   2 +-
 tests/t-generaterandom.c | 105 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 106 insertions(+), 1 deletion(-)
 create mode 100644 tests/t-generaterandom.c

diff --git a/tests/Makefile.am b/tests/Makefile.am
index 644eeb0..6c19071 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -33,7 +33,7 @@ noinst_HEADERS = t-support.h
 TESTS = t-link t-getfunctionlist t-initialize t-getinfo t-getslotlist \
 	t-getslotinfo t-gettokeninfo t-getmechanismlist t-getmechanisminfo \
 	t-opensession t-closeallsessions t-getsessioninfo \
-	t-findobjects t-getattribute t-auth
+	t-findobjects t-getattribute t-auth t-generaterandom
 
 noinst_PROGRAMS = $(TESTS)
 
diff --git a/tests/t-generaterandom.c b/tests/t-generaterandom.c
new file mode 100644
index 0000000..675138d
--- /dev/null
+++ b/tests/t-generaterandom.c
@@ -0,0 +1,105 @@
+/* t-generaterandom.c - Regression test.
+   Copyright (C) 2016 g10 Code GmbH
+
+   This file is part of Scute.
+
+   Scute 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.
+
+   Scute 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 Scute; if not, write to the Free Software Foundation,
+   Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+
+   In addition, as a special exception, g10 Code GmbH gives permission
+   to link this library: with the Mozilla Foundation's code for
+   Mozilla (or with modified versions of it that use the same license
+   as the "Mozilla" code), and distribute the linked executables.  You
+   must obey the GNU General Public License in all respects for all of
+   the code used other than "Mozilla".  If you modify this file, you
+   may extend this exception to your version of the file, but you are
+   not obligated to do so.  If you do not wish to do so, delete this
+   exception statement from your version.  */
+
+#include <stdio.h>
+#include <stdbool.h>
+
+#include "t-support.h"
+
+
+int
+main (int argc, char *argv[])
+{
+  CK_RV err;
+  CK_SLOT_ID_PTR slots;
+  CK_ULONG slots_count;
+  unsigned int i;
+
+  (void) argc;
+  (void) argv;
+
+  init_cryptoki ();
+
+  err = C_GetSlotList (true, NULL, &slots_count);
+  fail_if_err (err);
+
+  if (slots_count == 0)
+    {
+      printf ("Skipping test because no token is present.\n");
+      return 77;
+    }
+
+  printf ("Number of slots with tokens: %lu\n", slots_count);
+  slots = malloc (sizeof (CK_SLOT_ID) * slots_count);
+  if (!slots)
+    fail_if_err (CKR_HOST_MEMORY);
+
+  err = C_GetSlotList (true, slots, &slots_count);
+  fail_if_err (err);
+
+  for (i = 0; i < slots_count; i++)
+    {
+      CK_TOKEN_INFO info;
+
+      printf ("%2i. Slot ID %lu\n", i, slots[i]);
+
+      err = C_GetTokenInfo (slots[i], &info);
+      fail_if_err (err);
+
+      if ((info.flags & CKF_RNG) > 0)
+        {
+          CK_SESSION_HANDLE session;
+          unsigned char buffer[16];
+          unsigned int j;
+
+          printf("    RNG available\n");
+
+          err = C_OpenSession (slots[i], CKF_SERIAL_SESSION, NULL, NULL,
+                               &session);
+          fail_if_err (err);
+
+          printf ("    Session ID: %lu\n", session);
+
+          err = C_GenerateRandom (session, buffer, sizeof(buffer));
+          fail_if_err (err);
+
+          printf ("    Random bytes: 0x");
+          for (j = 0; j < sizeof(buffer); j++)
+            printf ("%02x", buffer[j]);
+          printf ("\n");
+
+          err = C_CloseSession (session);
+          fail_if_err (err);
+        }
+      else
+        printf ("    No RNG available on token\n");
+    }
+
+  return 0;
+}
-- 
2.7.4