1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
#!/usr/bin/env python3
#
# USAGE: test_usdt.py
#
# Copyright 2018 Facebook, Inc
# Licensed under the Apache License, Version 2.0 (the "License")
from __future__ import print_function
from bcc import BPF
from unittest import main, skipUnless, TestCase
from utils import kernel_version_ge
import os, resource
@skipUnless(not kernel_version_ge(5, 11), "Since d5299b67dd59 \"bpf: Memcg-based memory accounting for bpf maps\""\
",map mem has been counted against memcg, not rlimit")
class TestRlimitMemlock(TestCase):
def testRlimitMemlock(self):
text = b"""
BPF_HASH(unused, u64, u64, 65536);
int test() { return 0; }
"""
# save the original memlock limits
memlock_limit = resource.getrlimit(resource.RLIMIT_MEMLOCK)
# set a small RLIMIT_MEMLOCK limit
resource.setrlimit(resource.RLIMIT_MEMLOCK, (4096, 4096))
# below will fail
failed = 0
try:
b = BPF(text=text, allow_rlimit=False)
except:
failed = 1
self.assertEqual(failed, 1)
# below should succeed
b = BPF(text=text, allow_rlimit=True)
# reset to the original memlock limits
resource.setrlimit(resource.RLIMIT_MEMLOCK, memlock_limit)
if __name__ == "__main__":
main()
|
