Add validation caching interface.

src/trusted/validator/validation_cache_test.cc

+/*
+ * 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.
+ */
+
+#include "gtest/gtest.h"
+
+#include "native_client/src/trusted/validator/ncvalidate.h"
+#include "native_client/src/trusted/validator/validation_cache.h"
+
+#if NACL_WINDOWS
+# define UNREFERENCED_PARAMETER(P) (P)
+#else
+# define UNREFERENCED_PARAMETER(P) do { (void) P; } while (0)
+#endif
+
+const char * nop =

[Mark Seaborn, 2012/02/29 21:33:34]

Spacing style.  Use an array: nops[32] = "..."

[Nick Bray (chromium), 2012/02/29 22:58:09]

Done.

+    "\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90"
+    "\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90";
+
+// ret
+const char * ret =
+    "\xc3\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90"
+    "\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90";
+
+// pblendw $0xc0,%xmm0,%xmm2
+const char * sse4 =

[Mark Seaborn, 2012/02/29 21:33:34]

Is this SSE4 or SSE4.1?

[Nick Bray (chromium), 2012/02/29 22:58:09]

Done.

+    "\x66\x0f\x3a\x0e\xd0\xc0\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90"
+    "\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90\x90";
+
+const int code_size = 32;
+
+struct MockContext {
+  /* Sanity check that we're getting the right object. */
+  int marker;
+  int query_result;
+  int set_validates_expected;
+  int query_destroyed;
+};
+
+struct MockQuery {
+  /* Sanity check that we're getting the right object. */
+  int marker;
+  int state;
+  int add_count;
+  MockContext *context;
+};
+
+void *MockCreateQuery(void *context) {
+  MockContext *mcontext = (MockContext *) context;
+  MockQuery *mquery = (MockQuery *) malloc(sizeof(MockQuery));
+  EXPECT_EQ(31, mcontext->marker);
+  mquery->marker = 37;
+  mquery->state = 0;
+  mquery->add_count = 0;
+  mquery->context = mcontext;
+  return mquery;
+}
+
+void MockAddData(void *query, const unsigned char *data, size_t length) {
+  MockQuery *mquery = (MockQuery *) query;
+  ASSERT_EQ(37, mquery->marker);
+  UNREFERENCED_PARAMETER(data);
+  EXPECT_EQ(0, mquery->state);
+  /* Small data is supicious. */
+  EXPECT_LE((size_t) 2, length);
+  mquery->add_count += 1;
+}
+
+int MockDoQuery(void *query) {
+  MockQuery *mquery = (MockQuery *) query;
+  EXPECT_EQ(37, mquery->marker);
+  EXPECT_EQ(0, mquery->state);
+  /* Less than two pieces of data is suspicious. */
+  EXPECT_LE(2, mquery->add_count);
+  mquery->state = 1;
+  return mquery->context->query_result;
+}
+
+void MockSetValidates(void *query) {
+  MockQuery *mquery = (MockQuery *) query;
+  ASSERT_EQ(37, mquery->marker);
+  EXPECT_EQ(1, mquery->state);
+  EXPECT_EQ(1, mquery->context->set_validates_expected);
+  mquery->state = 2;
+}
+
+void MockDestroyQuery(void *query) {
+  MockQuery *mquery = (MockQuery *) query;
+  ASSERT_EQ(37, mquery->marker);
+  EXPECT_EQ(mquery->context->set_validates_expected ? 2 : 1, mquery->state);
+  mquery->context->query_destroyed = 1;
+  free(mquery);
+}
+
+class ValidationCachingInterfaceTests : public ::testing::Test {
+ protected:
+  MockContext context;
+  NaClValidationCache cache;
+  NaClCPUFeatures cpu_features;
+  int bundle_size;
+
+  unsigned char code_buffer[32];
+
+  void SetUp() {
+    context.marker = 31;
+    context.query_result = 1;
+    context.set_validates_expected = 0;
+    context.query_destroyed = 0;
+
+    cache.create_query = MockCreateQuery;
+    cache.add_data = MockAddData;
+    cache.do_query = MockDoQuery;
+    cache.set_validates = MockSetValidates;
+    cache.destroy_query = MockDestroyQuery;
+
+    NaClSetAllCPUFeatures(&cpu_features);
+
+    bundle_size = 32;
+
+    memset(code_buffer, 0x90, 32);
+  }
+
+  NaClValidationStatus Validate() {
+    return NACL_SUBARCH_NAME(ApplyValidator,
+                             NACL_TARGET_ARCH,
+                             NACL_TARGET_SUBARCH)(
+                                 NACL_SB_DEFAULT,
+                                 NaClApplyCodeValidation,
+                                 0, code_buffer, 32,
+                                 bundle_size, &cpu_features,
+                                 &cache, &context);
+  }
+};
+
+TEST_F(ValidationCachingInterfaceTests, Sanity) {
+  void *query = cache.create_query(&context);
+  cache.add_data(query, 0, 6);
+  cache.add_data(query, 0, 128);
+  EXPECT_EQ(1, cache.do_query(query));
+  cache.destroy_query(query);
+  EXPECT_EQ(1, context.query_destroyed);
+}
+
+TEST_F(ValidationCachingInterfaceTests, NoCache) {
+  NaClValidationStatus status;
+  status = NACL_SUBARCH_NAME(ApplyValidator,
+                             NACL_TARGET_ARCH,
+                             NACL_TARGET_SUBARCH)(
+                                 NACL_SB_DEFAULT,
+                                 NaClApplyCodeValidation,
+                                 0, code_buffer, code_size,
+                                 bundle_size, &cpu_features,
+                                 NULL, NULL);
+  EXPECT_EQ(NaClValidationSucceeded, status);
+}
+
+TEST_F(ValidationCachingInterfaceTests, CacheHit) {
+  NaClValidationStatus status;
+  status = Validate();
+  EXPECT_EQ(NaClValidationSucceeded, status);
+  EXPECT_EQ(1, context.query_destroyed);
+}
+
+TEST_F(ValidationCachingInterfaceTests, CacheMiss) {
+  NaClValidationStatus status;
+  context.query_result = 0;
+  context.set_validates_expected = 1;
+  status = Validate();
+  EXPECT_EQ(NaClValidationSucceeded, status);
+  EXPECT_EQ(1, context.query_destroyed);
+}
+
+TEST_F(ValidationCachingInterfaceTests, SSE4Allowed) {
+  NaClValidationStatus status;
+  memcpy(code_buffer, sse4, 32);
+  context.query_result = 0;
+  context.set_validates_expected = 1;
+  status = Validate();
+  EXPECT_EQ(NaClValidationSucceeded, status);
+  EXPECT_EQ(1, context.query_destroyed);
+}
+
+TEST_F(ValidationCachingInterfaceTests, SSE4Stubout) {
+  NaClValidationStatus status;
+  memcpy(code_buffer, sse4, 32);
+  context.query_result = 0;
+  NaClSetCPUFeature(&cpu_features, NaClCPUFeature_SSE41, 0);
+  status = Validate();
+  EXPECT_EQ(NaClValidationSucceeded, status);
+  EXPECT_EQ(1, context.query_destroyed);
+}
+
+TEST_F(ValidationCachingInterfaceTests, IllegalInst) {
+  NaClValidationStatus status;
+  memcpy(code_buffer, ret, 32);
+  context.query_result = 0;
+  status = Validate();
+  EXPECT_EQ(NaClValidationFailed, status);
+  EXPECT_EQ(1, context.query_destroyed);
+}
+
+// Test driver function.
+int main(int argc, char *argv[]) {
+  testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}