1 # Copyright (c) 2019 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
6 # http://www.apache.org/licenses/LICENSE-2.0
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
14 """Linux limit library."""
16 from resources.libraries.python.ssh import exec_cmd_no_error
18 __all__ = [u"LimitUtil"]
22 """Class contains methods for getting or setting process resource limits."""
25 def get_pid_limit(node, pid):
26 """Get process resource limits.
28 :param node: Node in the topology.
29 :param pid: Process ID.
33 command = f"prlimit --noheadings --pid={pid}"
34 message = f"Node {node[u'host']} failed to run: {command}"
36 exec_cmd_no_error(node, command, sudo=True, message=message)
39 def set_pid_limit(node, pid, resource, limit):
40 """Set process resource limits.
42 :param node: Node in the topology.
43 :param pid: Process ID.
44 :param resource: Resource to set limits.
45 :param limit: Limit value.
51 command = f"prlimit --{resource}={limit} --pid={pid}"
52 message = f"Node {node[u'host']} failed to run: {command}"
54 exec_cmd_no_error(node, command, sudo=True, message=message)