Rewrite non-temporal instructions

src/trusted/validator_ragel/dfa_validate_common.c

 /*
  * Copyright (c) 2012 The Native Client Authors. All rights reserved.
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 /* Implement the functions common for ia32 and x86-64 architectures.  */
 #include "native_client/src/trusted/validator_ragel/dfa_validate_common.h"
 
 #include <string.h>
@@ skipping 15 matching lines @@
 Bool NaClDfaProcessValidationError(const uint8_t *begin, const uint8_t *end,
                                    uint32_t info, void *callback_data) {
   UNREFERENCED_PARAMETER(begin);
   UNREFERENCED_PARAMETER(end);
   UNREFERENCED_PARAMETER(info);
   UNREFERENCED_PARAMETER(callback_data);
 
   return FALSE;
 }
 
+static Bool IsREX(uint8_t byte) {
+  return byte >= 0x40 && byte <= 0x4f;
+}
+
+static Bool RewriteUnsupportedInstruction(const uint8_t *begin,
+                                          const uint8_t *end) {
+  uint8_t *ptr = (uint8_t *) begin;
+  /* We usually only check and rewrite the first few bytes without examining
+   * further because this function is only called when the validator tells us
+   * that it is an 'unsupported instruction' and there are no other validation
+   * failures.
+   */
+  if (memcmp(begin, "\x0f\xe7", 2) == 0) {
+    /* ia32: movntq => movq */
+    ptr[1] = 0x7f;
+    return TRUE;
+  } else if (memcmp(begin, "\x66\x0f\xe7", 3) == 0) {
+    /* ia32: movntdq => movdqa */
+    ptr[2] = 0x7f;
+    return TRUE;
+  } else if (IsREX(begin[0])) {

[bradn, 2015/08/04 22:03:30]

Make this arch conditional (#ifdef), so you don't waste time checking for things
from the other arch?

[ruiq, 2015/08/05 20:11:03]

Done.

+    if (memcmp(begin + 1, "\x0f\x2b", 2) == 0) {
+      /* amd64: movntps => movaps */
+      ptr[2] = 0x29;
+      return TRUE;
+    } else if (memcmp(begin + 1, "\x0f\xc3", 2) == 0) {
+      /* amd64: movnti => mov */
+      ptr[1] = 0x89;
+      memmove(ptr + 2, ptr + 3, end - begin - 3);
+      ptr[end - begin - 1] = 0x90;
+      return TRUE;
+    } else if (memcmp(begin + 1, "\x0f\x18", 2) == 0) {
+      /* amd64: prefetchnta => nop */
+      memset(ptr, 0x90, end - begin);
+      return TRUE;
+    }
+  } else if (begin[0] == 0x66 && IsREX(begin[1]) &&
+      memcmp(begin + 2, "\x0f\xe7", 2) == 0) {
+    /* amd64: movntdq => movdqa */
+    ptr[3] = 0x7f;
+    return TRUE;
+  }
+  return FALSE;
+}
+
 Bool NaClDfaStubOutUnsupportedInstruction(const uint8_t *begin,
                                           const uint8_t *end,
                                           uint32_t info,
                                           void *callback_data) {
   struct StubOutCallbackData *data = callback_data;
   /* Stub-out instructions unsupported on this CPU, but valid on other CPUs.  */
   if ((info & VALIDATION_ERRORS_MASK) == CPUID_UNSUPPORTED_INSTRUCTION) {
     data->did_rewrite = 1;
     memset((uint8_t *)begin, NACL_HALT_OPCODE, end - begin);
     return TRUE;
   } else if ((info & VALIDATION_ERRORS_MASK) == UNSUPPORTED_INSTRUCTION) {
     if (data->flags & NACL_DISABLE_NONTEMPORALS_X86) {
       return FALSE;
     } else {
-      /* TODO(ruiq): rewrite instruction. For now, we keep the original
-       * instruction and indicate validation success, which is consistent
-       * with current validation results. */
-      data->did_rewrite = 0;
-      return TRUE;
+      if (RewriteUnsupportedInstruction(begin, end)) {
+        data->did_rewrite = 1;
+        return TRUE;
+      } else {
+        data->did_rewrite = 0;
+        return FALSE;
+      }
     }
   } else {
     return FALSE;
   }
 }
 
 Bool NaClDfaProcessCodeCopyInstruction(const uint8_t *begin_new,
                                        const uint8_t *end_new,
                                        uint32_t info_new,
                                        void *callback_data) {
@@ skipping 13 matching lines @@
 
 Bool NaClDfaCodeReplacementIsStubouted(const uint8_t *begin_existing,
                                        size_t instruction_length) {
 
   /* Unsupported instruction must have been replaced with HLTs.  */
   if (memcmp(kStubOutMem, begin_existing, instruction_length) == 0)
     return TRUE;
   else
     return FALSE;
 }