bash: add set-quicly-ws function

[vpp.git] / extras / bash / functions.bash

diff --git a/extras/bash/functions.bash b/extras/bash/functions.bash
index 597b749..777dd94 100644 (file)
--- a/extras/bash/functions.bash
+++ b/extras/bash/functions.bash
@@ -1,4 +1,4 @@
-# Copyright (c) 2019 Cisco and/or its affiliates.
+# Copyright (c) 2021 Cisco and/or its affiliates.
 # 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:
@@ -15,7 +15,6 @@
 # This file is meant to be sourced in a .bashrc file to add useful
 # bash functions to an interactive shell
 
-
 # Bash function to run vpp 'make test' testcases
 # repeatedly, stopping on test failure or when
 # a test log contains the optionally specified text
@@ -30,19 +29,28 @@ vpp-make-test()
     local run_make_test
     local old_pwd
     local test_desc
+    local grep_results
+    local result
+    local fail
+    local i
+    local line
     local is_feature="false"
     local retry_count=100
     local tester=${GERRIT_USER:-$USER}
     local jobs="auto"
-    
+
     if [ -z "$WS_ROOT" ] ; then
-        echo "ERROR: WS_ROOT is not set!"
-        return
-    elif [ -z "$(find $WS_ROOT -type d -name vppinfra)" ] ; then
+        if [ -d "./extras/bash" ] ; then
+            export WS_ROOT="$(pwd)"
+        else
+            echo "ERROR: WS_ROOT is not set!"
+            return
+        fi
+    elif [ ! -d "$WS_ROOT/src/vppinfra" ] ; then
         echo "ERROR: WS_ROOT is not set to a VPP workspace!"
         return
     fi
-    
+
     options=$(getopt -o "adfg:j:r:" -- "$@")
     if [ $? -eq 1 ] ; then
         usage=true
@@ -89,7 +97,7 @@ vpp-make-test()
         esac
         shift
     done
-    
+
     if [ -n "$usage" ] || [ -z "$1" ] ; then
         if [ -z "$1" ] ; then
             echo "ERROR: no testcase specified!"
@@ -144,8 +152,79 @@ vpp-make-test()
             return
         fi
     done
-    
+
     echo -e "\n$line\nPASS [$((i-1))/$retry_count]: $test_desc\n$line\n"
     echo -e "Hey $tester, Life is good!!! :D\n"
     cd $old_pwd
 }
+
+# bash function to set up csit python virtual environment
+csit-env()
+{
+    if [ -f "$WS_ROOT/VPP_REPO_URL" ] && [ -f "$WS_ROOT/requirements.txt" ]; then
+        if [ -n "$(declare -f deactivate)" ]; then
+            echo "Deactivating Python Virtualenv!"
+            deactivate
+        fi
+        local PIP=pip
+        local setup_framework=$WS_ROOT/resources/libraries/python/SetupFramework.py
+        if [ -n "$(grep pip3 $setup_framework)" ]; then
+            PIP=pip3
+            local VENV_OPTS="-p python3"
+        fi
+        export CSIT_DIR=$WS_ROOT
+        export PYTHONPATH=$CSIT_DIR
+        rm -rf $PYTHONPATH/env && virtualenv $VENV_OPTS $PYTHONPATH/env \
+            && source $PYTHONPATH/env/bin/activate \
+            && $PIP install --upgrade -r $PYTHONPATH/requirements.txt \
+            && $PIP install --upgrade -r $PYTHONPATH/tox-requirements.txt
+    else
+        echo "ERROR: WS_ROOT not set to a CSIT workspace!"
+    fi
+}
+
+# bash function to set up VPP workspace with quicly source code
+set-quicly-ws ()
+{
+    local ext_quicly_version_file="/opt/vpp/external/x86_64/include/quicly/version.h"
+    if [ ! -f "$ext_quicly_version_file" ] ; then
+        echo -e "\nCannot find quicly version file: $ext_quicly_version_file"
+        echo -e "\nPlease run VPP 'make install-ext-deps' to install it."
+        return
+    fi
+
+    if [ -d "$1" ]; then
+        if [ -z "$WS_ROOT" ] ; then
+            if [ -d "./extras/bash" ] ; then
+                export WS_ROOT="$(pwd)"
+            else
+                echo "ERROR: WS_ROOT is not set!"
+                return
+            fi
+        elif [ ! -d "$WS_ROOT/extras/bash" ] ; then
+            echo "ERROR: WS_ROOT is not set to a VPP workspace!"
+            return
+        fi
+        export WS_QUICLY="$1"
+        export QUICLY_LIBRARY="$WS_QUICLY/libquicly.a"
+        export QUICLY_INCLUDE_DIR="$WS_QUICLY/include"
+        export WS_PICOTLS="$WS_QUICLY/deps/picotls"
+        export PICOTLS_INCLUDE_DIR="$WS_PICOTLS/include"
+        export PICOTLS_CORE_LIBRARY="$WS_PICOTLS/libpicotls-core.a"
+        export PICOTLS_OPENSSL_LIBRARY="$WS_PICOTLS/libpicotls-openssl.a"
+        export VPP_EXTRA_CMAKE_ARGS="-DQUICLY_LIBRARY=$QUICLY_LIBRARY -DQUICLY_INCLUDE_DIR=$QUICLY_INCLUDE_DIR -DPICOTLS_CORE_LIBRARY=$PICOTLS_CORE_LIBRARY -DPICOTLS_OPENSSL_LIBRARY=$PICOTLS_OPENSSL_LIBRARY -DPICOTLS_INCLUDE_DIR=$PICOTLS_INCLUDE_DIR"
+        local quicly_ws_version_file="$QUICLY_INCLUDE_DIR/quicly/version.h"
+        cp -f $ext_quicly_version_file $quicly_ws_version_file
+        local expected_quicly_version="$(grep 'set(EXPECTED_QUICLY_VERSION' $WS_ROOT/src/plugins/quic/CMakeLists.txt | cut -d'"' -f2)"
+        sed -ie "s/LIBQUICLY_VERSION \".*\"/LIBQUICLY_VERSION \"$expected_quicly_version\"/" $quicly_ws_version_file
+    else
+        echo -e "\nUsage: set-quicly-ws <path-to-quicly>"
+        echo -e "\nPrerequisites:"
+        echo -e "\n1. Clone quicly repo:\n   git clone https://github.com/h2o/quicly"
+        echo -e "\n2. Build quicly and picotls following instructions in:"
+        echo "   .../quicly/README.md"
+        echo "   .../quicly/deps/picotls/README.md"
+        echo -e "\n3. Run set-quicly-ws <path-to-quicly>"
+        echo -e "\n4. Build vpp as desired\n"
+    fi;
+}