Replace telnetlib with scrapli

.gitignore

@@ -2,8 +2,10 @@
 .idea
 venv
 *.qcow2
+*.qcow
 cisco*.bin
 *.gz
+*.tgz
 *.xz
 *.vmdk
 *.iso
@@ -24,4 +26,7 @@ cisco*.bin
 *.xml
 
 # ignore clab- dirs
-clab-*
+clab-*
+
+# Ignore container cidfile
+**cidfile

aoscx/docker/Dockerfile

@@ -1,25 +1,4 @@
-FROM debian:bullseye
-MAINTAINER Stefano Sasso <[email protected]>
-
-ENV DEBIAN_FRONTEND=noninteractive
-
-RUN apt-get update -qy \
- && apt-get upgrade -qy \
- && apt-get install -y \
-    bridge-utils \
-    iproute2 \
-    python3-ipy \
-    socat \
-    tcpdump \
-    ssh \
-    inetutils-ping \
-    dnsutils \
-    iptables \
-    telnet \
-    ftp \
-    qemu-system-x86=1:5.2+dfsg-11+deb11u2 \
-    qemu-utils=1:5.2+dfsg-11+deb11u2 \
- && rm -rf /var/lib/apt/lists/*
+FROM ghcr.io/srl-labs/vrnetlab-base:0.0.1
 
 ARG IMAGE
 COPY $IMAGE* /

aoscx/docker/launch.py

@@ -64,7 +64,7 @@ def bootstrap_spin(self):
             self.start()
             return
 
-        (ridx, match, res) = self.tn.expect([b"switch login:"], 1)
+        (ridx, match, res) = self.expect([b"switch login:"], 1)
         if match:  # got a match!
             if ridx == 0:  # login
                 self.logger.debug("trying to log in with 'admin'")
@@ -92,8 +92,8 @@ def bootstrap_spin(self):
 
         # no match, if we saw some output from the router it's probably
         # booting, so let's give it some more time
-        if res != b"":
-            self.logger.trace("OUTPUT: %s" % res.decode())
+        if res != "":
+            self.print(res)
             # reset spins if we saw some output
             self.spins = 0
 

build-base-image.sh

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+# this script builds the vrnetlab base container image
+# that is used in the dockerfiles of the NOS images
+
+set -e
+
+if [ -z "$1" ]; then
+    echo "Usage: $0 <version>"
+    exit 1
+fi
+
+sudo docker build -t ghcr.io/srl-labs/vrnetlab-base:$1 \
+    -f vrnetlab-base.dockerfile .

cisco/asav/Makefile

@@ -0,0 +1,21 @@
+VENDOR=cisco
+NAME=asav
+IMAGE_FORMAT=qcow2
+IMAGE_GLOB=*.qcow2
+VENDOR_SUBDIR=1
+
+# match versions like:
+# asav9-18-2.qcow2
+VERSION=$(shell echo $(IMAGE) | sed -e 's/.\+[^0-9]\([0-9]\+\-[0-9]\+\-[0-9]\+[a-z]\?\)\([^0-9].*\|$$\)/\1/')
+
+-include ../../makefile-sanity.include
+-include ../../makefile.include
+
+docker-pre-build:
+	-cat cidfile | xargs --no-run-if-empty docker rm -f
+	-rm cidfile
+
+docker-build: docker-build-common
+	docker run --cidfile cidfile --privileged $(REGISTRY)$(VENDOR)_$(NAME):$(VERSION) --trace --install
+	docker commit --change='ENTRYPOINT ["/launch.py"]' $$(cat cidfile) $(REGISTRY)$(VENDOR)_$(NAME):$(VERSION)
+	docker rm -f $$(cat cidfile)

cisco/asav/docker/Dockerfile

@@ -0,0 +1,11 @@
+FROM ghcr.io/srl-labs/vrnetlab-base:0.0.1
+
+ARG VERSION
+ENV VERSION=${VERSION}
+ARG IMAGE
+COPY $IMAGE* /
+COPY *.py /
+
+EXPOSE 22 161/udp 830 5000 10000-10099
+HEALTHCHECK CMD ["/healthcheck.py"]
+ENTRYPOINT ["/launch.py"]

asav/docker/launch.py → cisco/asav/docker/launch.py

@@ -11,6 +11,9 @@
 import vrnetlab
 
 
+DEFAULT_SMP = 4
+DEFAULT_RAM = 8192 # in MB
+
 def handle_SIGCHLD(signal, frame):
     os.waitpid(-1, os.WNOHANG)
 
@@ -37,16 +40,17 @@ def trace(self, message, *args, **kws):
 
 
 class ASAv_vm(vrnetlab.VM):
-    def __init__(self, username, password, install_mode=False):
+    def __init__(self, username, password, conn_mode, install_mode=False):
         for e in os.listdir("/"):
             if re.search(".qcow2$", e):
                 disk_image = "/" + e
 
         super(ASAv_vm, self).__init__(
-            username, password, disk_image=disk_image, ram=2048
+            username, password, disk_image=disk_image, ram=DEFAULT_RAM, smp=f"cores={DEFAULT_SMP},threads=1,sockets=1"
         )
         self.nic_type = "e1000"
         self.install_mode = install_mode
+        self.conn_mode = conn_mode
         self.num_nics = 8
 
     def bootstrap_spin(self):
@@ -58,18 +62,18 @@ def bootstrap_spin(self):
             self.start()
             return
 
-        (ridx, match, res) = self.tn.expect([b"ciscoasa>"], 1)
+        (ridx, match, res) = self.expect([b"ciscoasa>"], 1)
         if match:  # got a match!
             if ridx == 0:  # login
                 if self.install_mode:
-                    self.logger.debug("matched, ciscoasa>")
+                    self.logger.info("matched, ciscoasa>")
                     self.wait_write("", wait=None)
                     self.wait_write("", None)
                     self.wait_write("", wait="ciscoasa>")
                     self.running = True
                     return
 
-                self.logger.debug("matched, ciscoasa>")
+                self.logger.info("matched, ciscoasa>")
                 self.wait_write("", wait=None)
 
                 # run main config!
@@ -85,8 +89,8 @@ def bootstrap_spin(self):
 
         # no match, if we saw some output from the router it's probably
         # booting, so let's give it some more time
-        if res != b"":
-            self.logger.trace("OUTPUT: %s" % res.decode())
+        if res != "":
+            self.print(res)
             # reset spins if we saw some output
             self.spins = 0
 
@@ -103,8 +107,7 @@ def bootstrap_config(self):
         self.wait_write("VR-netlab9", wait="Repeat Password:")
         self.wait_write("", wait="ciscoasa#")
         self.wait_write("configure terminal", wait="#")
-        self.wait_write("N", wait="[Y]es, [N]o, [A]sk later:")
-        self.wait_write("", wait="(config)#")
+        self.wait_write("N", wait="[Y]es, [N]o, [A]sk later:", timeout=15)
         self.wait_write("aaa authentication ssh console LOCAL")
         self.wait_write("aaa authentication enable console LOCAL")
         self.wait_write(
@@ -114,6 +117,7 @@ def bootstrap_config(self):
         self.wait_write("nameif management")
         self.wait_write("ip address 10.0.0.15 255.255.255.0")
         self.wait_write("no shutdown")
+        self.wait_write("route management 0.0.0.0 0.0.0.0 10.0.0.2")
         self.wait_write("ssh 0.0.0.0 0.0.0.0 management")
         self.wait_write("ssh version 2")
         self.wait_write("ssh key-exchange group dh-group14-sha256")
@@ -124,17 +128,17 @@ def bootstrap_config(self):
 
 
 class ASAv(vrnetlab.VR):
-    def __init__(self, username, password):
+    def __init__(self, username, password, conn_mode):
         super(ASAv, self).__init__(username, password)
-        self.vms = [ASAv_vm(username, password)]
+        self.vms = [ASAv_vm(username, password, conn_mode)]
 
 
 class ASAv_installer(ASAv):
     """ASAv installer"""
 
-    def __init__(self, username, password):
+    def __init__(self, username, password, conn_mode):
         super(ASAv, self).__init__(username, password)
-        self.vms = [ASAv_vm(username, password, install_mode=True)]
+        self.vms = [ASAv_vm(username, password, conn_mode, install_mode=True)]
 
     def install(self):
         self.logger.info("Installing ASAv")
@@ -156,6 +160,11 @@ def install(self):
     parser.add_argument("--username", default="vrnetlab", help="Username")
     parser.add_argument("--password", default="VR-netlab9", help="Password")
     parser.add_argument("--install", action="store_true", help="Install ASAv")
+    parser.add_argument(
+        "--connection-mode",
+        default="vrxcon",
+        help="Connection mode to use in the datapath",
+    )
     args = parser.parse_args()
 
     LOG_FORMAT = "%(asctime)s: %(module)-10s %(levelname)-8s %(message)s"
@@ -167,8 +176,8 @@ def install(self):
         logger.setLevel(1)
 
     if args.install:
-        vr = ASAv_installer(args.username, args.password)
+        vr = ASAv_installer(args.username, args.password, args.connection_mode)
         vr.install()
     else:
-        vr = ASAv(args.username, args.password)
+        vr = ASAv(args.username, args.password, args.connection_mode)
         vr.start()

c8000v/Makefile → cisco/c8000v/Makefile

@@ -2,14 +2,18 @@ VENDOR=cisco
 NAME=c8000v
 IMAGE_FORMAT=qcow2
 IMAGE_GLOB=*.qcow2
+VENDOR_SUBDIR=1
 
 # match versions like:
 # c8000v-17.11.01a.qcow2
 VERSION=$(shell echo $(IMAGE) | sed -e 's/.\+[^0-9]\([0-9]\+\.[0-9]\+\.[0-9]\+[a-z]\?\)\([^0-9].*\|$$\)/\1/')
 
--include ../makefile-sanity.include
--include ../makefile.include
--include ../makefile-install.include
+-include ../../makefile-sanity.include
+-include ../../makefile.include
+
+docker-pre-build:
+	-cat cidfile | xargs --no-run-if-empty docker rm -f
+	-rm cidfile
 
 docker-build: docker-build-common
 	docker run --cidfile cidfile --privileged $(REGISTRY)$(VENDOR)_$(NAME):$(VERSION) --trace --install

cisco/c8000v/docker/Dockerfile

@@ -0,0 +1,11 @@
+FROM ghcr.io/srl-labs/vrnetlab-base:0.0.1
+
+ARG VERSION
+ENV VERSION=${VERSION}
+ARG IMAGE
+COPY $IMAGE* /
+COPY *.py /
+
+EXPOSE 22 161/udp 830 5000 10000-10099
+HEALTHCHECK CMD ["/healthcheck.py"]
+ENTRYPOINT ["/launch.py"]