8f453a2dda
The CNI daemon should always be run in its own cgroup. That typically can take two forms: - Running inside a container - Running as a systemd service This patch changes the way the memory usage is tracked so that both of the cgroup memberships listed above are supported. Thanks to using cgroups for tracking the memory usage, we will finally take into account the CNI daemon children memory usage. Change-Id: I0ef48742653d5c17ea0cc787ae3a997d5d315c5a Closes-Bug: 1752939 Signed-off-by: Antoni Segura Puimedon <antonisp@celebdor.com>
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
# Copyright (c) 2017 NEC Technologies India Pvt Ltd.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
PROC_ONE_CGROUP_PATH = '/proc/1/cgroup'
|
|
CONTAINER_RUNTIME_CGROUP_IDS = (
|
|
'docker', # This is set by docker/moby
|
|
'libpod', # This is set by podman
|
|
)
|
|
|
|
|
|
def running_under_container_runtime(proc_one_cg_path=PROC_ONE_CGROUP_PATH):
|
|
"""Returns True iff the CNI process is under a known container runtime."""
|
|
with open(proc_one_cg_path, 'r') as cgroup_info:
|
|
proc_one_cg_info = cgroup_info.read()
|
|
return any(runtime in proc_one_cg_info for runtime in
|
|
CONTAINER_RUNTIME_CGROUP_IDS)
|
|
|
|
|
|
class CNIConfig(dict):
|
|
def __init__(self, cfg):
|
|
super(CNIConfig, self).__init__(cfg)
|
|
|
|
for k, v in self.items():
|
|
if not k.startswith('_'):
|
|
setattr(self, k, v)
|
|
|
|
|
|
class CNIArgs(object):
|
|
def __init__(self, value):
|
|
for item in value.split(';'):
|
|
k, v = item.split('=', 1)
|
|
if not k.startswith('_'):
|
|
setattr(self, k, v)
|
|
|
|
|
|
class CNIParameters(object):
|
|
def __init__(self, env, cfg=None):
|
|
for k, v in env.items():
|
|
if k.startswith('CNI_'):
|
|
setattr(self, k, v)
|
|
if cfg is None:
|
|
self.config = CNIConfig(env['config_kuryr'])
|
|
else:
|
|
self.config = cfg
|
|
self.args = CNIArgs(self.CNI_ARGS)
|
|
|
|
def __repr__(self):
|
|
return repr({key: value for key, value in self.__dict__.items() if
|
|
key.startswith('CNI_')})
|