commit 9028d91fd86a2d517c7ce163c1d88b41de961ba8
author Lorenzo Colitti <lorenzo@google.com> Mon Mar 28 02:34:54 2016 +0900
committer Lorenzo Colitti <lorenzo@google.com> Wed Mar 30 16:43:53 2016 +0900
tree ff68d271eeb55bf1a88c058964f37bb032b06ced
parent 0f15055ba127da5c8bff3617f04ac6df0b4ee2f4

Add a test for StrictController.

Bug: 21725996
Bug: 25691379
Change-Id: I24b838161eaf98dede2ae897157ba42414fc926f

server/StrictController.h

diff --git a/server/StrictController.h b/server/StrictController.h
index 52a6779..6f5fed2 100644
--- a/server/StrictController.h
+++ b/server/StrictController.h
@@ -19,6 +19,8 @@
 
 #include <string>
 
+#include "NetdConstants.h"
+
 enum StrictPenalty { INVALID, ACCEPT, LOG, REJECT };
 
 /*
@@ -39,6 +41,11 @@
     static const char* LOCAL_CLEAR_CAUGHT;
     static const char* LOCAL_PENALTY_LOG;
     static const char* LOCAL_PENALTY_REJECT;
+
+protected:
+    // For testing.
+    friend class StrictControllerTest;
+    static int (*execIptables)(IptablesTarget target, ...);
 };
 
 #endif