Implement off-the-wire validation scheme for enum types.

chrome/tools/ipclist/ipcfuzz.cc

 // Copyright (c) 2012 The Chromium 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 <algorithm>
 #include <ostream>
 #include <set>
 #include <vector>
 
 #include "base/command_line.h"
@@ skipping 492 matching lines @@
   static void CopyMessageID(PickleCracker *dst, PickleCracker *src) {
     memcpy(dst->mutable_payload(), src->payload(), sizeof(int));
   }
 };
 
 // Redefine macros to generate fuzzing from traits declarations.
 // Null out all the macros that need nulling.
 #include "ipc/ipc_message_null_macros.h"
 
 // STRUCT declarations cause corresponding STRUCT_TRAITS declarations to occur.
-#undef IPC_STRUCT_BEGIN
 #undef IPC_STRUCT_BEGIN_WITH_PARENT
 #undef IPC_STRUCT_MEMBER
 #undef IPC_STRUCT_END
 #define IPC_STRUCT_BEGIN_WITH_PARENT(struct_name, parent)\
-  IPC_STRUCT_BEGIN(struct_name)
-#define IPC_STRUCT_BEGIN(struct_name) IPC_STRUCT_TRAITS_BEGIN(struct_name)
+  IPC_STRUCT_TRAITS_BEGIN(struct_name)
 #define IPC_STRUCT_MEMBER(type, name, ...) IPC_STRUCT_TRAITS_MEMBER(name)
 #define IPC_STRUCT_END() IPC_STRUCT_TRAITS_END()
 
 // Set up so next include will generate fuzz trait classes.
 #undef IPC_STRUCT_TRAITS_BEGIN
 #undef IPC_STRUCT_TRAITS_MEMBER
 #undef IPC_STRUCT_TRAITS_PARENT
 #undef IPC_STRUCT_TRAITS_END
 #define IPC_STRUCT_TRAITS_BEGIN(struct_name) \
   template <> \
   struct FuzzTraits<struct_name> { \
     static void Fuzz(struct_name *p, IPC::Fuzzer* fuzzer) { \
 
 #define IPC_STRUCT_TRAITS_MEMBER(name) \
       FuzzParam(&p->name, fuzzer);
 
 #define IPC_STRUCT_TRAITS_PARENT(type) \
       FuzzParam(static_cast<type*>(p), fuzzer);
 
 #define IPC_STRUCT_TRAITS_END() \
     } \
   };
 
-#undef IPC_ENUM_TRAITS
-#define IPC_ENUM_TRAITS(enum_name) \
+#undef IPC_ENUM_TRAITS_VALIDATE
+#define IPC_ENUM_TRAITS_VALIDATE(enum_name, validation_expression) \
   template <> \
   struct FuzzTraits<enum_name> { \
     static void Fuzz(enum_name* p, IPC::Fuzzer* fuzzer) { \
       FuzzParam(reinterpret_cast<int*>(p), fuzzer); \
     } \
   };
 
 // Bring them into existence.
 #include "chrome/common/all_messages.h"
 #include "content/common/all_messages.h"
@@ skipping 129 matching lines @@
 
 // Entry point avoiding mangled names.
 extern "C" {
   __attribute__((visibility("default")))
   IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void);
 }
 
 IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void) {
   return &g_ipcfuzz;
 }