Add CPUID::print

2 files changed, 29 insertions, 0 deletions

diff --git a/src/utils/cpuid.cpp b/src/utils/cpuid.cpp
index c727471bc..75a88c6b8 100644
--- a/src/utils/cpuid.cpp
+++ b/src/utils/cpuid.cpp
@@ -9,6 +9,7 @@
 #include <botan/types.h>
 #include <botan/get_byte.h>
 #include <botan/mem_ops.h>
+#include <ostream>
 
 #if defined(BOTAN_TARGET_CPU_IS_PPC_FAMILY)
 
@@ -153,6 +154,31 @@ bool altivec_check_pvr_emul()
 
 }
 
+void CPUID::print(std::ostream& o)
+   {
+   o << "CPUID flags: ";
+
+#define CPUID_PRINT(flag) do { if(has_##flag()) o << #flag << " "; } while(0)
+   CPUID_PRINT(sse2);
+   CPUID_PRINT(ssse3);
+   CPUID_PRINT(sse41);
+   CPUID_PRINT(sse42);
+   CPUID_PRINT(avx2);
+   CPUID_PRINT(avx512f);
+   CPUID_PRINT(altivec);
+
+   CPUID_PRINT(rdtsc);
+   CPUID_PRINT(bmi2);
+   CPUID_PRINT(clmul);
+   CPUID_PRINT(aes_ni);
+   CPUID_PRINT(rdrand);
+   CPUID_PRINT(rdseed);
+   CPUID_PRINT(intel_sha);
+   CPUID_PRINT(adx);
+#undef CPUID_PRINT
+   o << "\n";
+   }
+
 void CPUID::initialize()
    {
 #if defined(BOTAN_TARGET_CPU_IS_PPC_FAMILY)
diff --git a/src/utils/cpuid.h b/src/utils/cpuid.h
index 67657d2ee..1c4c1df0b 100644
--- a/src/utils/cpuid.h
+++ b/src/utils/cpuid.h
@@ -9,6 +9,7 @@
 #define BOTAN_CPUID_H__
 
 #include <botan/types.h>
+#include <iosfwd>
 
 namespace Botan {
 
@@ -116,6 +117,8 @@ class BOTAN_DLL CPUID
       * Check if the processor supports AltiVec/VMX
       */
       static bool has_altivec() { return m_altivec_capable; }
+
+      static void print(std::ostream& o);
    private:
       enum CPUID_bits {
          CPUID_RDTSC_BIT = 4,