commit 0a1a9a11d96d937e7563659f92edf04fafa766c9
author Chenbo Feng <fengc@google.com> Tue Apr 09 12:05:04 2019 -0700
committer Maciej Żenczykowski <maze@google.com> Wed Apr 10 05:54:00 2019 +0000
tree 3eec639b277d14a24fc410787cb5842dba254fe4
parent ee121646cf60988e69eb2c8b1e87e11f5e4932cc

Add helper function to set rlimit for tests

Add a helper function to set sufficient MEMLOCK rlimit
for bpf related unit tests.

Bug: 119279144
Bug: 129246448
Test: libbpf_android_test
Change-Id: I5390f4d3b21436abff69a661d1c6e6a6749542ed

libbpf_android/BpfMapTest.cpp

diff --git a/libbpf_android/BpfMapTest.cpp b/libbpf_android/BpfMapTest.cpp
index cfa9d88..895dda6 100644
--- a/libbpf_android/BpfMapTest.cpp
+++ b/libbpf_android/BpfMapTest.cpp
@@ -57,11 +57,13 @@
     void SetUp() {
         SKIP_IF_BPF_NOT_SUPPORTED;
 
+        EXPECT_EQ(0, setrlimitForTest());
         if (!access(PINNED_MAP_PATH, R_OK)) {
             EXPECT_EQ(0, remove(PINNED_MAP_PATH));
         }
         mMapFd = createMap(BPF_MAP_TYPE_HASH, sizeof(uint32_t), sizeof(uint32_t), TEST_MAP_SIZE,
                            BPF_F_NO_PREALLOC);
+        EXPECT_LE(0, mMapFd);
     }
 
     void TearDown() {