tests: added unit test for gnutls_pkcs11_get_pin_function

Signed-off-by: Nikos Mavrogiannopoulos <nmav@gnutls.org>

2 files changed, 60 insertions, 1 deletions

diff --git a/tests/Makefile.am b/tests/Makefile.am
index 48d535d194..1edaff6840 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -218,7 +218,7 @@ ctests += pkcs11-cert-import-url-exts pkcs11-get-exts pkcs11-get-raw-issuer-exts
 	pkcs11/pkcs11-combo pkcs11/pkcs11-privkey pkcs11/pkcs11-pubkey-import-rsa pkcs11/pkcs11-pubkey-import-ecdsa \
 	pkcs11-import-url-privkey pkcs11-privkey-fork pkcs11/pkcs11-ec-privkey-test \
 	pkcs11-privkey-always-auth pkcs11-privkey-export pkcs11/pkcs11-import-with-pin \
-	pkcs11/pkcs11-privkey-pthread
+	pkcs11/pkcs11-privkey-pthread pkcs11/pkcs11-pin-func
 
 
 endif
diff --git a/tests/pkcs11/pkcs11-pin-func.c b/tests/pkcs11/pkcs11-pin-func.c
new file mode 100644
index 0000000000..b6d707ee72
--- /dev/null
+++ b/tests/pkcs11/pkcs11-pin-func.c
@@ -0,0 +1,59 @@
+/*
+ * Copyright (C) 2017 Nikos Mavrogiannopoulos
+ *
+ * Author: Nikos Mavrogiannopoulos
+ *
+ * This file is part of GnuTLS.
+ *
+ * GnuTLS 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * GnuTLS 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 GnuTLS; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <assert.h>
+#include <gnutls/gnutls.h>
+#include <gnutls/pkcs11.h>
+
+static
+int pin_func(void* userdata, int attempt, const char* url, const char *label,
+		unsigned flags, char *pin, size_t pin_max)
+{
+	if (attempt == 0) {
+		strcpy(pin, "xxx");
+		return 0;
+	}
+	return -1;
+}
+
+int main(int argc, char **argv)
+{
+	void *u;
+	gnutls_pin_callback_t cb;
+
+	gnutls_pkcs11_set_pin_function(pin_func, (void*)-1);
+
+	cb = gnutls_pkcs11_get_pin_function(&u);
+
+	assert(u==(void*)-1);
+	assert(cb == pin_func);
+
+	return 0;
+}