Commit 03032941 authored by Xiao Liang's avatar Xiao Liang Committed by Jakub Kicinski
Browse files

selftests: net: Add python context manager for netns entering 



Change netns of current thread and switch back on context exit.
For example:

    with NetNSEnter("ns1"):
        ip("link add dummy0 type dummy")

The command be executed in netns "ns1".

Signed-off-by: default avatarXiao Liang <shaw.leon@gmail.com>
Link: https://patch.msgid.link/20250219125039.18024-13-shaw.leon@gmail.com


Signed-off-by: default avatarJakub Kicinski <kuba@kernel.org>
parent 7ca486d0

tools/testing/selftests/net/lib/py/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@ 


from .consts import KSRC

from .ksft import *

from .netns import NetNS

from .netns import NetNS, NetNSEnter

from .nsim import *

from .utils import *

from .ynl import NlError, YnlFamily, EthtoolFamily, NetdevFamily, RtnlFamily, RtnlAddrFamily

tools/testing/selftests/net/lib/py/netns.py

+18 −0
Original line number Diff line number Diff line 
# SPDX-License-Identifier: GPL-2.0



from .utils import ip

import ctypes

import random

import string



libc = ctypes.cdll.LoadLibrary('libc.so.6')





class NetNS:

    def __init__(self, name=None):
@@ -29,3 +32,18 @@ class NetNS: 


    def __repr__(self):

        return f"NetNS({self.name})"





class NetNSEnter:

    def __init__(self, ns_name):

        self.ns_path = f"/run/netns/{ns_name}"



    def __enter__(self):

        self.saved = open("/proc/thread-self/ns/net")

        with open(self.ns_path) as ns_file:

            libc.setns(ns_file.fileno(), 0)

        return self



    def __exit__(self, exc_type, exc_value, traceback):

        libc.setns(self.saved.fileno(), 0)

        self.saved.close()