cpuid.c 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. /* ===================================================================
  2. *
  3. * Copyright (c) 2014, Legrandin <helderijs@gmail.com>
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions
  8. * are met:
  9. *
  10. * 1. Redistributions of source code must retain the above copyright
  11. * notice, this list of conditions and the following disclaimer.
  12. * 2. Redistributions in binary form must reproduce the above copyright
  13. * notice, this list of conditions and the following disclaimer in
  14. * the documentation and/or other materials provided with the
  15. * distribution.
  16. *
  17. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  20. * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  21. * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  22. * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  23. * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  25. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  27. * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28. * POSSIBILITY OF SUCH DAMAGE.
  29. * ===================================================================
  30. */
  31. #include "pycrypto_common.h"
  32. FAKE_INIT(cpuid)
  33. #if defined HAVE_CPUID_H
  34. #include <cpuid.h>
  35. #elif defined HAVE_INTRIN_H
  36. #include <intrin.h>
  37. #endif
  38. /** Return 1 if the CPU supports the AESNI extension **/
  39. EXPORT_SYM int have_aes_ni(void)
  40. {
  41. uint32_t info[4];
  42. memset(info, 0, sizeof info);
  43. /* Call cpuid to retrieve x86 Processor Info and Feature bits.
  44. * info[2] is ecx. If bit 25 is set, the CPU supports the
  45. * AES-NI extension. */
  46. #if defined HAVE_CPUID_H
  47. __get_cpuid(1, info, info+1, info+2, info+3);
  48. #elif defined HAVE_INTRIN_H
  49. __cpuidex(info, 1, 0);
  50. #endif
  51. return (info[2] & ((int)1<<25)) ? 1 : 0;
  52. }