Fixes from Kazuho's review.
diff --git a/lib/fusion.c b/lib/fusion.c
index 3dc03cd..41e7bd3 100644
--- a/lib/fusion.c
+++ b/lib/fusion.c
@@ -38,6 +38,7 @@
* IN THE SOFTWARE.
*/
#include <stdint.h>
+
#include <stdlib.h>
#include <string.h>
#include <immintrin.h>
@@ -293,11 +294,7 @@
__m128i ek0, bits0, bits1, bits2, bits3, bits4, bits5 = _mm_setzero_si128();
const __m128i *bits4keys = ctx->ecb.keys; /* is changed to supp->ctx.keys when calcurating suppout */
-#ifdef _WINDOWS
- struct ptls_fusion_gfmul_state gstate = {0};
-#else
- struct ptls_fusion_gfmul_state gstate = {};
-#endif
+ struct ptls_fusion_gfmul_state gstate = { 0 };
__m128i gdatabuf[6];
__m128i ac = _mm_shuffle_epi8(_mm_set_epi32(0, (int)aadlen * 8, 0, (int)inlen * 8), bswap8);
@@ -495,11 +492,7 @@
{
__m128i ek0 = _mm_setzero_si128(), bits0, bits1 = _mm_setzero_si128(), bits2 = _mm_setzero_si128(), bits3 = _mm_setzero_si128(),
bits4 = _mm_setzero_si128(), bits5 = _mm_setzero_si128();
-#ifdef _WINDOWS
struct ptls_fusion_gfmul_state gstate = { 0 };
-#else
- struct ptls_fusion_gfmul_state gstate = {};
-#endif
__m128i gdatabuf[6];
__m128i ac = _mm_shuffle_epi8(_mm_set_epi32(0, (int)aadlen * 8, 0, (int)inlen * 8), bswap8);
struct ptls_fusion_aesgcm_ghash_precompute *ghash_precompute = ctx->ghash + (aadlen + 15) / 16 + (inlen + 15) / 16 + 1;
@@ -991,10 +984,39 @@
sizeof(struct aesgcm_context),
aes256gcm_setup};
-#ifdef _WINDOWS
+/* #ifdef _WINDOWS */
+#if 1
+/**
+ * ptls_fusion_is_supported_by_cpu:
+ * Check that the CPU has extended instructions for PCMUL, AES and AVX2.
+ * This test assumes that the CPU is following the x86/x64 architecture.
+ * A slightly more refined test could check that the cpu_info spells out
+ * "genuineIntel" or "authenticAMD", but would fail in presence of
+ * little known CPU brands or some VM */
int ptls_fusion_is_supported_by_cpu(void)
{
- return 1;
+ uint32_t cpu_info[4];
+ uint32_t nb_ids;
+ int is_supported = 0;
+
+ __cpuid(cpu_info, 0);
+ nb_ids = cpu_info[0];
+
+ if (nb_ids >= 7) {
+ uint32_t leaf1_ecx;
+ __cpuid(cpu_info, 1);
+ leaf1_ecx = cpu_info[2];
+
+ if (/* PCLMUL */ (leaf1_ecx & (1 << 5)) != 0 && /* AES */ (leaf1_ecx & (1 << 25)) != 0) {
+ uint32_t leaf7_ebx;
+ __cpuid(cpu_info, 7);
+ leaf7_ebx = cpu_info[1];
+
+ is_supported = /* AVX2 */ (leaf7_ebx & (1 << 5)) != 0;
+ }
+ }
+
+ return is_supported;
}
#else
int ptls_fusion_is_supported_by_cpu(void)
diff --git a/lib/picotls.c b/lib/picotls.c
index 09846b9..9f84120 100644
--- a/lib/picotls.c
+++ b/lib/picotls.c
@@ -5140,11 +5140,7 @@
ptls_iovec_t hash_value, const char *label_prefix)
{
ptls_aead_context_t *ctx = NULL;
-#ifdef _WINDOWS
- uint8_t key_iv[PTLS_MAX_DIGEST_SIZE + PTLS_MAX_IV_SIZE];
-#else
- uint8_t key_iv[aead->key_size + aead->iv_size];
-#endif
+ uint8_t key_iv[PTLS_MAX_SECRET_SIZE + PTLS_MAX_IV_SIZE];
int ret;
if ((ret = get_traffic_key(hash, key_iv, aead->key_size, 0, secret, hash_value, label_prefix)) != 0)
diff --git a/t/fusion.c b/t/fusion.c
index c72836a..a7a9685 100644
--- a/t/fusion.c
+++ b/t/fusion.c
@@ -24,13 +24,8 @@
#include <string.h>
#include "picotls/fusion.h"
#include "picotls/minicrypto.h"
-#ifdef _WINDOWS
-#include "deps/picotest/picotest.h"
-#include "lib/fusion.c"
-#else
#include "../deps/picotest/picotest.h"
#include "../lib/fusion.c"
-#endif
static const char *tostr(const void *_p, size_t len)
{
@@ -55,11 +50,7 @@
static void test_loadn(void)
{
-#ifdef _WINDOWS
uint8_t buf[8192] = { 0 };
-#else
- uint8_t buf[8192] = {};
-#endif
for (size_t off = 0; off < 8192 - 15; ++off) {
uint8_t *src = buf + off;
@@ -73,11 +64,9 @@
}
ok(!!"success");
}
-#ifdef _WINDOWS
+
static const uint8_t zero[16384] = { 0 };
-#else
-static const uint8_t zero[16384] = {};
-#endif
+
static void test_ecb(void)
{
ptls_fusion_aesecb_context_t ecb;
@@ -219,19 +208,14 @@
ptls_cipher_encrypt(rand, &aadlen, zero, sizeof(aadlen));
ptls_cipher_encrypt(rand, &textlen, zero, sizeof(textlen));
ptls_cipher_encrypt(rand, &seq, zero, sizeof(seq));
-#ifdef _WINDOWS
+
uint8_t aad[256], text[256];
-#else
- uint8_t aad[aadlen], text[textlen];
-#endif
+
ptls_cipher_encrypt(rand, aad, zero, sizeof(aad));
ptls_cipher_encrypt(rand, text, zero, sizeof(text));
-#ifdef _WINDOWS
uint8_t encrypted[272], decrypted[256];
-#else
- uint8_t encrypted[textlen + 16], decrypted[textlen];
-#endif
+
memset(encrypted, 0x55, sizeof(encrypted));
memset(decrypted, 0xcc, sizeof(decrypted));
