From 4d8f87b1cada5041fa60012dcb502cd452a2bc42 Mon Sep 17 00:00:00 2001
From: Daniel King <damaki.gh@gmail.com>
Date: Wed, 18 May 2016 10:09:28 -0300
Subject: [PATCH] Add ChaCha20/Poly1305 ciphers to the selftest program

---
 programs/test/selftest.c | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/programs/test/selftest.c b/programs/test/selftest.c
index 72a37342f..57f9924ce 100644
--- a/programs/test/selftest.c
+++ b/programs/test/selftest.c
@@ -44,6 +44,9 @@
 #include "mbedtls/des.h"
 #include "mbedtls/aes.h"
 #include "mbedtls/camellia.h"
+#include "mbedtls/chacha20.h"
+#include "mbedtls/poly1305.h"
+#include "mbedtls/aead_chacha20_poly1305.h"
 #include "mbedtls/base64.h"
 #include "mbedtls/bignum.h"
 #include "mbedtls/rsa.h"
@@ -207,6 +210,15 @@ const selftest_t selftests[] =
 #if defined(MBEDTLS_CMAC_C)
     {"cmac", mbedtls_cmac_self_test},
 #endif
+#if defined(MBEDTLS_CHACHA20_C)
+    {"chacha20", mbedtls_chacha20_self_test},
+#endif
+#if defined(MBEDTLS_POLY1305_C)
+    {"poly1305", mbedtls_poly1305_self_test},
+#endif
+#if defined(MBEDTLS_AEAD_CHACHA20_POLY1305_C)
+    {"chacha20-poly1305", mbedtls_aead_chacha20_poly1305_self_test},
+#endif
 #if defined(MBEDTLS_BASE64_C)
     {"base64", mbedtls_base64_self_test},
 #endif