From c562db9be1f91fb8f560e246aca7b4a5c0e57365 Mon Sep 17 00:00:00 2001
From: rajendra-dendukuri
 <47423477+rajendra-dendukuri@users.noreply.github.com>
Date: Wed, 4 Dec 2019 07:50:56 -0500
Subject: [PATCH] [sonic-ztp]: Build sonic-ztp package (#3299)

* Build sonic-ztp package

- Add changes in make rules to conditionally include sonic-ztp package

Signed-off-by: Rajendra Dendukuri <rajendra.dendukuri@broadcom.com>
---
 Makefile.work                                   |  2 ++
 files/build_templates/sonic_debian_extension.j2 |  6 ++++++
 rules/config                                    |  3 +++
 rules/sonic-ztp.mk                              | 17 +++++++++++++++++
 slave.mk                                        |  1 +
 5 files changed, 29 insertions(+)
 create mode 100644 rules/sonic-ztp.mk

diff --git a/Makefile.work b/Makefile.work
index c81586ecf577..a1ccf8715567 100644
--- a/Makefile.work
+++ b/Makefile.work
@@ -7,6 +7,7 @@
 #  * BUILD_NUMBER: Desired version-number to pass to the building-system.
 #  * ENABLE_DHCP_GRAPH_SERVICE: Enables get-graph service to fetch minigraph files
 #    through http.
+#  * ENABLE_ZTP: Enables zero touch provisioning.
 #  * SHUTDOWN_BGP_ON_START: Sets admin-down state for all bgp peerings after restart.
 #  * ENABLE_PFCWD_ON_START: Enable PFC Watchdog (PFCWD) on server-facing ports
 #  * by default for TOR switch.
@@ -162,6 +163,7 @@ SONIC_BUILD_INSTRUCTION :=  make \
                            BUILD_NUMBER=$(BUILD_NUMBER) \
                            BUILD_TIMESTAMP=$(BUILD_TIMESTAMP) \
                            ENABLE_DHCP_GRAPH_SERVICE=$(ENABLE_DHCP_GRAPH_SERVICE) \
+                           ENABLE_ZTP=$(ENABLE_ZTP) \
                            SHUTDOWN_BGP_ON_START=$(SHUTDOWN_BGP_ON_START) \
                            SONIC_ENABLE_PFCWD_ON_START=$(ENABLE_PFCWD_ON_START) \
                            SONIC_ENABLE_SYNCD_RPC=$(ENABLE_SYNCD_RPC) \
diff --git a/files/build_templates/sonic_debian_extension.j2 b/files/build_templates/sonic_debian_extension.j2
index aa44ca703507..5ba38a38502a 100644
--- a/files/build_templates/sonic_debian_extension.j2
+++ b/files/build_templates/sonic_debian_extension.j2
@@ -147,6 +147,12 @@ sudo https_proxy=$https_proxy LANG=C chroot $FILESYSTEM_ROOT pip install tabulat
 sudo dpkg --root=$FILESYSTEM_ROOT -i $python_debs_path/python-sonic-utilities_*.deb || \
     sudo LANG=C DEBIAN_FRONTEND=noninteractive chroot $FILESYSTEM_ROOT apt-get -y install -f
 
+{% if enable_ztp == "y" %}
+# Install ZTP (and its dependencies via 'apt-get -y install -f')
+sudo dpkg --root=$FILESYSTEM_ROOT -i $debs_path/sonic-ztp_*.deb || \
+    sudo LANG=C DEBIAN_FRONTEND=noninteractive chroot $FILESYSTEM_ROOT apt-get -y install -f
+{% endif %}
+
 # SONiC utilities installs bash-completion as a dependency. However, it is disabled by default
 # in bash.bashrc, so we copy a version of the file with it enabled here.
 sudo cp -f $IMAGE_CONFIGS/bash/bash.bashrc $FILESYSTEM_ROOT/etc/
diff --git a/rules/config b/rules/config
index 5d4f01e05746..c168fa5a964e 100644
--- a/rules/config
+++ b/rules/config
@@ -44,6 +44,9 @@ DEFAULT_PASSWORD = YourPaSsWoRd
 # If not set (default behavior) the default minigraph built into the image will be used.
 # ENABLE_DHCP_GRAPH_SERVICE = y
 
+# ENABLE_ZTP - installs Zero Touch Provisioning support.
+# ENABLE_ZTP = y
+
 # SHUTDOWN_BGP_ON_START - if set to y all bgp sessions will be in admin down state when
 # bgp service starts.
 # SHUTDOWN_BGP_ON_START = y
diff --git a/rules/sonic-ztp.mk b/rules/sonic-ztp.mk
new file mode 100644
index 000000000000..43615b7dcc8a
--- /dev/null
+++ b/rules/sonic-ztp.mk
@@ -0,0 +1,17 @@
+# SONiC ztp package
+#
+
+ifeq ($(ENABLE_ZTP), y)
+
+SONIC_ZTP_VERSION = 1.0.0
+
+SONIC_ZTP = sonic-ztp_$(SONIC_ZTP_VERSION)_all.deb
+$(SONIC_ZTP)_SRC_PATH = $(SRC_PATH)/sonic-ztp
+SONIC_DPKG_DEBS += $(SONIC_ZTP)
+SONIC_STRETCH_DEBS += $(SONIC_ZTP)
+
+export SONIC_ZTP_VERSION
+export SONIC_ZTP
+
+endif
+
diff --git a/slave.mk b/slave.mk
index f630aaf47f4e..16c4d4ef6a8b 100644
--- a/slave.mk
+++ b/slave.mk
@@ -632,6 +632,7 @@ $(addprefix $(TARGET_PATH)/, $(SONIC_INSTALLERS)) : $(TARGET_PATH)/% : \
                 $(MONIT)) \
         $$(addprefix $(TARGET_PATH)/,$$($$*_DOCKERS)) \
         $$(addprefix $(FILES_PATH)/,$$($$*_FILES)) \
+	$(if $(findstring y,$(ENABLE_ZTP)),$(addprefix $(DEBS_PATH)/,$(SONIC_ZTP))) \
 	$(addprefix $(STRETCH_FILES_PATH)/, $(if $(filter $(CONFIGURED_ARCH),amd64), $(IXGBE_DRIVER))) \
         $(addprefix $(PYTHON_DEBS_PATH)/,$(SONIC_UTILS)) \
         $(addprefix $(PYTHON_WHEELS_PATH)/,$(SONIC_CONFIG_ENGINE)) \