Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add python3 virtual environment for docker-ptf #10599

Merged
merged 6 commits into from
Apr 26, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitmodules
Original file line number Diff line number Diff line change
Expand Up @@ -103,3 +103,6 @@
[submodule "src/sonic-p4rt/sonic-pins"]
path = src/sonic-p4rt/sonic-pins
url = https://github.com/Azure/sonic-pins.git
[submodule "src/ptf-py3"]
path = src/ptf-py3
url = https://github.com/p4lang/ptf.git
67 changes: 66 additions & 1 deletion dockers/docker-ptf/Dockerfile.j2
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,11 @@ FROM {{ prefix }}multiarch/debian-debootstrap:arm64-stretch
FROM {{ prefix }}debian:buster
{% endif %}

{% from "dockers/dockerfile-macros.j2" import install_python_wheels, copy_files %}

USER root
WORKDIR /root

MAINTAINER Pavel Shirshov

RUN echo "deb [arch=amd64] http://debian-archive.trafficmanager.net/debian buster-backports main" >> /etc/apt/sources.list
Expand Down Expand Up @@ -51,6 +56,13 @@ RUN sed --in-place 's/httpredir.debian.org/debian-archive.trafficmanager.net/' /
python-libpcap \
python-scapy \
python-six \
python3 \
python3-venv \
python3-pip \
python3-dev \
python3-scapy \
python3-six \
libpcap-dev \
tacacs+ \
rsyslog \
ntp \
Expand All @@ -59,7 +71,9 @@ RUN sed --in-place 's/httpredir.debian.org/debian-archive.trafficmanager.net/' /
arping \
bridge-utils \
libteam-utils \
gdb
gdb \
automake \
iproute2

# Install all python modules from pypi. python-scapy is exception, ptf debian package requires python-scapy
# TODO: Clean up this step
Expand Down Expand Up @@ -117,10 +131,61 @@ RUN rm -rf /debs \
&& pip install pybrctl pyro4 rpyc yabgp \
&& pip install unittest-xml-reporting \
&& pip install pyrasite \
&& pip install retrying \
&& mkdir -p /opt \
&& cd /opt \
&& wget https://raw.githubusercontent.com/p4lang/ptf/master/ptf_nn/ptf_nn_agent.py

RUN python3 -m venv env-python3

# Activating a virtualenv. The virtualenv automatically works for RUN, ENV and CMD.
ENV VIRTUAL_ENV=/root/env-python3
ARG BACKUP_OF_PATH="$PATH"
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 PYTHONIOENCODING=UTF-8

RUN python3 -m pip install --upgrade --ignore-installed pip

# Install all python modules from pypi. python3-scapy is exception, ptf debian package requires python3-scapy
RUN python3 -m pip install setuptools \
&& pip3 install supervisor \
&& pip3 install ipython==5.4.1 \
&& pip3 install Cython \
&& pip3 install cffi \
&& pip3 install nnpy \
&& pip3 install dpkt \
&& pip3 install ipaddress \
&& pip3 install pysubnettree \
&& pip3 install paramiko \
&& pip3 install Flask \
&& pip3 install exabgp \
&& pip3 install pyaml \
&& pip3 install pybrctl pyro4 rpyc yabgp \
&& pip3 install unittest-xml-reporting \
&& pip3 install pyrasite \
&& pip3 install python-libpcap \
&& pip3 install enum34 \
&& pip3 install grpcio \
&& pip3 install grpcio-tools \
&& pip3 install protobuf \
&& pip3 install six==1.16.0 \
&& pip3 install itsdangerous \
&& pip3 install retrying \
&& pip3 install jinja2 \
&& pip3 install scapy==2.4.5

{% if docker_ptf_whls.strip() -%}
# Copy locally-built Python wheel dependencies
{{ copy_files("python-wheels/", docker_ptf_whls.split(' '), "/python-wheels/") }}

# Install locally-built Python wheel dependencies
{{ install_python_wheels(docker_ptf_whls.split(' ')) }}
{% endif %}

# Deactivating a virtualenv.
ENV PATH="$BACKUP_OF_PATH"

## Adjust sshd settings
RUN mkdir /var/run/sshd \
&& echo 'root:root' | chpasswd \
Expand Down
1 change: 1 addition & 0 deletions platform/vs/docker-ptf.mk
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# docker image for docker-ptf

DOCKER_PTF = docker-ptf.gz
$(DOCKER_PTF)_PYTHON_WHEELS += $(PTF_PY3)
$(DOCKER_PTF)_PATH = $(DOCKERS_PATH)/docker-ptf
$(DOCKER_PTF)_DEPENDS += $(LIBTHRIFT) $(PYTHON_THRIFT) $(PTF) $(PYTHON_SAITHRIFT)
SONIC_DOCKER_IMAGES += $(DOCKER_PTF)
Expand Down
11 changes: 11 additions & 0 deletions rules/ptf-py3.dep
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@

SPATH := $($(PTF_PY3)_SRC_PATH)
DEP_FILES := $(SONIC_COMMON_FILES_LIST) rules/ptf-py3.mk rules/ptf-py3.dep
DEP_FILES += $(SONIC_COMMON_BASE_FILES_LIST)
SMDEP_FILES := $(addprefix $(SPATH)/,$(shell cd $(SPATH) && git ls-files))

$(PTF_PY3)_CACHE_MODE := GIT_CONTENT_SHA
$(PTF_PY3)_DEP_FLAGS := $(SONIC_COMMON_FLAGS_LIST)
$(PTF_PY3)_DEP_FILES := $(DEP_FILES)
$(PTF_PY3)_SMDEP_FILES := $(SMDEP_FILES)
$(PTF_PY3)_SMDEP_PATHS := $(SPATH)
7 changes: 7 additions & 0 deletions rules/ptf-py3.mk
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
# ptf package

PTF_PY3 = ptf-0.9.3-py3-none-any.whl
$(PTF_PY3)_SRC_PATH = $(SRC_PATH)/ptf-py3
$(PTF_PY3)_PYTHON_VERSION = 3
$(PTF_PY3)_TEST = n
SONIC_PYTHON_WHEELS += $(PTF_PY3)
1 change: 1 addition & 0 deletions src/ptf-py3
Submodule ptf-py3 added at 405513
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
From a8b13b9fbaa16ddd305ba2df2238ef606ef222a7 Mon Sep 17 00:00:00 2001
From: Zhaohui Sun <zhaohuisun@microsoft.com>
Date: Wed, 13 Apr 2022 09:24:46 +0000
Subject: [PATCH 1/2] Remove ord in get_mac() to avoid TypeError

Signed-off-by: Zhaohui Sun <zhaohuisun@microsoft.com>
---
src/ptf/netutils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ptf/netutils.py b/src/ptf/netutils.py
index 752e76c..6aabe79 100644
--- a/src/ptf/netutils.py
+++ b/src/ptf/netutils.py
@@ -54,7 +54,7 @@ def get_if_index(iff):


def get_mac(iff):
- return ":".join(["%02x" % ord(char) for char in get_if(iff, SIOCGIFHWADDR)[18:24]])
+ return ":".join(["%02x" % char for char in get_if(iff, SIOCGIFHWADDR)[18:24]])


def set_promisc(s, iff, val=1):
--
2.25.1

Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
From 6e570e00ea05882d2db1e480ed041ea631bf37da Mon Sep 17 00:00:00 2001
From: Zhaohui Sun <zhaohuisun@microsoft.com>
Date: Wed, 13 Apr 2022 09:25:28 +0000
Subject: [PATCH 2/2] Fill byte formatted client mac address in DHCP Discover
packet

Signed-off-by: Zhaohui Sun <zhaohuisun@microsoft.com>
---
src/ptf/testutils.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/ptf/testutils.py b/src/ptf/testutils.py
index ab67cea..83a9075 100755
--- a/src/ptf/testutils.py
+++ b/src/ptf/testutils.py
@@ -2712,12 +2712,13 @@ def __dhcp_mac_to_chaddr(mac_addr="00:01:02:03:04:05"):
"""
Private helper function to convert a 6-byte MAC address of form:
'00:01:02:03:04:05'
- into a 16-byte chaddr byte string of form:
- '\x00\x01\x02\x03\x04\x05\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
+ into a 16-byte chaddr byte of form:
+ b'\x00\x01\x02\x03\x04\x05\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'

"""
- chaddr = "".join([chr(int(octet, 16)) for octet in mac_addr.split(":")])
- chaddr += "\x00" * 10
+ import binascii
+ chaddr = binascii.unhexlify(mac_addr.replace(':', ''))
+ chaddr += b'\x00\x00\x00\x00\x00\x00'
return chaddr


--
2.25.1

2 changes: 2 additions & 0 deletions src/ptf-py3.patch/series
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
0001-Remove-ord-in-get_mac-to-avoid-TypeError.patch
0002-Fill-byte-formatted-client-mac-address-in-DHCP-Disco.patch