mirror of
https://github.com/Ascend/ascend-docker-runtime.git
synced 2025-10-18 20:04:52 +08:00
Match-id-333af697bff0299c9522aeb1f41de79301684546
This commit is contained in:
BianTanggui
223
build/build.sh
223
build/build.sh
@@ -20,15 +20,6 @@ HOOKSRCNAME="main.go"
|
|||||||
RUNTIMEDIR=${ROOT}/runtime
|
RUNTIMEDIR=${ROOT}/runtime
|
||||||
RUNTIMESRCNAME="main.go"
|
RUNTIMESRCNAME="main.go"
|
||||||
|
|
||||||
DEBPACK=${ROOT}/debpack
|
|
||||||
BINDIR=${DEBPACK}/usr/bin
|
|
||||||
DEBDIR=${DEBPACK}/DEBIAN
|
|
||||||
|
|
||||||
RPMPACK=${ROOT}/rpmpack
|
|
||||||
RPMSOURCESDIR=${RPMPACK}/SOURCES
|
|
||||||
RPMSPECDIR=${RPMPACK}/SPECS
|
|
||||||
|
|
||||||
|
|
||||||
CLISRCPATH=`find ${CLIDIR} -name "${CLISRCNAME}"`
|
CLISRCPATH=`find ${CLIDIR} -name "${CLISRCNAME}"`
|
||||||
CLISRCDIR=${CLISRCPATH%/${CLISRCNAME}}
|
CLISRCDIR=${CLISRCPATH%/${CLISRCNAME}}
|
||||||
INSTALLHELPERSRCPATH=`find ${INSTALLHELPERDIR} -name "${INSTALLHELPERSRCNAME}"`
|
INSTALLHELPERSRCPATH=`find ${INSTALLHELPERDIR} -name "${INSTALLHELPERSRCNAME}"`
|
||||||
@@ -38,144 +29,114 @@ HOOKSRCDIR=${HOOKSRCPATH%/${HOOKSRCNAME}}
|
|||||||
RUNTIMESRCPATH=`find ${RUNTIMEDIR} -name "${RUNTIMESRCNAME}"`
|
RUNTIMESRCPATH=`find ${RUNTIMEDIR} -name "${RUNTIMESRCNAME}"`
|
||||||
RUNTIMESRCDIR=${RUNTIMESRCPATH%/${RUNTIMESRCNAME}}
|
RUNTIMESRCDIR=${RUNTIMESRCPATH%/${RUNTIMESRCNAME}}
|
||||||
|
|
||||||
SOFT_VERSION=`cat $TOP_DIR/CI/config/version.ini | grep "PackageName" | cut -d "=" -f 2`
|
VERSION=`cat $TOP_DIR/CI/config/version.ini | grep "PackageName" | cut -d "=" -f 2`
|
||||||
VERSION=$SOFT_VERSION
|
PACKAGENAME="Ascend-docker-runtime"
|
||||||
RELEASE="1"
|
|
||||||
PACKAGENAEM="ascend-docker-runtime"
|
|
||||||
CPUARCH=`uname -m`
|
CPUARCH=`uname -m`
|
||||||
|
|
||||||
funcbuild(){
|
function build_bin()
|
||||||
echo "make cli"
|
{
|
||||||
[ -d "${CLISRCDIR}/build" ]&&rm -rf ${CLISRCDIR}/build
|
echo "make cli"
|
||||||
mkdir ${CLISRCDIR}/build&&cd ${CLISRCDIR}/build
|
[ -d "${CLISRCDIR}/build" ] && rm -rf ${CLISRCDIR}/build
|
||||||
cmake ../
|
mkdir ${CLISRCDIR}/build && cd ${CLISRCDIR}/build
|
||||||
make clean
|
cmake ../
|
||||||
make
|
make clean
|
||||||
|
make
|
||||||
|
|
||||||
echo "make installhelper"
|
echo "make installhelper"
|
||||||
[ -d "${INSTALLHELPERSRCDIR}/build" ]&&rm -rf ${INSTALLHELPERSRCDIR}/build
|
[ -d "${INSTALLHELPERSRCDIR}/build" ] && rm -rf ${INSTALLHELPERSRCDIR}/build
|
||||||
mkdir ${INSTALLHELPERSRCDIR}/build&&cd ${INSTALLHELPERSRCDIR}/build
|
mkdir ${INSTALLHELPERSRCDIR}/build && cd ${INSTALLHELPERSRCDIR}/build
|
||||||
cmake ../
|
cmake ../
|
||||||
make clean
|
make clean
|
||||||
make
|
make
|
||||||
|
|
||||||
[ -d "${ROOT}/opensource/src" ]&&rm -rf ${ROOT}/opensource/src
|
[ -d "${ROOT}/opensource/src" ] && rm -rf ${ROOT}/opensource/src
|
||||||
mkdir ${ROOT}/opensource/src
|
mkdir ${ROOT}/opensource/src
|
||||||
/bin/cp -rf ${HOOKSRCDIR}/vendor/* ${ROOT}/opensource/src
|
/bin/cp -rf ${HOOKSRCDIR}/vendor/* ${ROOT}/opensource/src
|
||||||
export GOPATH="${GOPATH}:${ROOT}/opensource"
|
export GOPATH="${GOPATH}:${ROOT}/opensource"
|
||||||
export GO111MODULE=off
|
export GO111MODULE=off
|
||||||
|
|
||||||
echo "make hook"
|
echo "make hook"
|
||||||
[ -d "${HOOKSRCDIR}/build" ]&&rm -rf ${HOOKSRCDIR}/build
|
[ -d "${HOOKSRCDIR}/build" ] && rm -rf ${HOOKSRCDIR}/build
|
||||||
mkdir ${HOOKSRCDIR}/build&&cd ${HOOKSRCDIR}/build
|
mkdir ${HOOKSRCDIR}/build && cd ${HOOKSRCDIR}/build
|
||||||
go build -ldflags "-buildid=IdNetCheck" -trimpath ../${HOOKSRCNAME}
|
go build -ldflags "-buildid=IdNetCheck" -trimpath ../${HOOKSRCNAME}
|
||||||
mv main ascend-docker-hook
|
mv main ascend-docker-hook
|
||||||
|
|
||||||
echo "make runtime"
|
echo "make runtime"
|
||||||
[ -d "${RUNTIMESRCDIR}/build" ]&&rm -rf ${RUNTIMESRCDIR}/build
|
[ -d "${RUNTIMESRCDIR}/build" ] && rm -rf ${RUNTIMESRCDIR}/build
|
||||||
mkdir ${RUNTIMESRCDIR}/build&&cd ${RUNTIMESRCDIR}/build
|
mkdir ${RUNTIMESRCDIR}/build&&cd ${RUNTIMESRCDIR}/build
|
||||||
go build -ldflags "-buildid=IdNetCheck" -trimpath ../${RUNTIMESRCNAME}
|
go build -ldflags "-buildid=IdNetCheck" -trimpath ../${RUNTIMESRCNAME}
|
||||||
mv main ascend-docker-runtime
|
mv main ascend-docker-runtime
|
||||||
}
|
}
|
||||||
|
|
||||||
fillcontrol(){
|
function build_run_package()
|
||||||
sed -i "1i\Package: ${PACKAGENAEM}" ${DEBDIR}/control
|
{
|
||||||
sed -i "2a\Architecture: all" ${DEBDIR}/control
|
cd ${BUILD}
|
||||||
sed -i "3a\Version: ${VERSION}" ${DEBDIR}/control
|
mkdir run_pkg
|
||||||
|
|
||||||
|
/bin/cp -f {${RUNTIMESRCDIR},${HOOKSRCDIR},${INSTALLHELPERSRCDIR},${CLISRCDIR}}/build/ascend-docker* run_pkg
|
||||||
|
FILECNT=`ls -l run_pkg |grep "^-"|wc -l`
|
||||||
|
echo "prepare package $FILECNT bins"
|
||||||
|
if [ $FILECNT -ne 4 ]; then
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
/bin/cp -f scripts/run_main.sh run_pkg
|
||||||
|
chmod 550 run_pkg/run_main.sh
|
||||||
|
|
||||||
|
RUN_PKG_NAME="${PACKAGENAME}-${CPUARCH}.run"
|
||||||
|
DATE=`date -u "+%Y-%m-%d"`
|
||||||
|
bash makeself.sh --nomd5 --nocrc --help-header scripts/help.info --packaging-date ${DATE} \
|
||||||
|
--tar-extra "--mtime=${DATE}" run_pkg "${RUN_PKG_NAME}" ascend-docker-runtime ./run_main.sh
|
||||||
|
mv ${RUN_PKG_NAME} ${OUTPUT}
|
||||||
}
|
}
|
||||||
|
|
||||||
funcmakedeb(){
|
function make_clean()
|
||||||
cd ${BUILD}
|
{
|
||||||
mkdir -pv {${DEBDIR},${BINDIR}}
|
[ -d "${OUTPUT}" ] && cd ${OUTPUT}&&rm -rf *
|
||||||
/bin/cp -f {${RUNTIMESRCDIR},${HOOKSRCDIR},${INSTALLHELPERSRCDIR},${CLISRCDIR}}/build/ascend-docker* ${BINDIR}
|
|
||||||
FILECNT=`ls -l ${BINDIR} |grep "^-"|wc -l`
|
|
||||||
echo "prepare package $FILECNT bins"
|
|
||||||
if [ $FILECNT -ne 4 ]; then
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
CONPATH=`find ${INSTALLHELPERDIR} -name "control"`
|
|
||||||
INSTPATH=`find ${INSTALLHELPERDIR} -name "postinst"`
|
|
||||||
RMPATH=`find ${INSTALLHELPERDIR} -name "prerm"`
|
|
||||||
/bin/cp -f ${CONPATH} ${INSTPATH} ${RMPATH} ${DEBDIR}
|
|
||||||
echo ${INSTPATH}
|
|
||||||
fillcontrol
|
|
||||||
chmod 555 ${DEBDIR}/postinst
|
|
||||||
chmod 555 ${DEBDIR}/prerm
|
|
||||||
dpkg-deb -b ${DEBPACK} ${PACKAGENAEM}_${VERSION}-${RELEASE}_${CPUARCH}.deb
|
|
||||||
DEBS=`find ${BUILD} -name "*.deb"`
|
|
||||||
/bin/cp ${DEBS} ${OUTPUT}
|
|
||||||
}
|
|
||||||
|
|
||||||
funcfillspec(){
|
|
||||||
sed -i "4a\BuildArch: $CPUARCH" ${RPMSPECDIR}/*.spec
|
|
||||||
}
|
}
|
||||||
|
|
||||||
funcmakerpm(){
|
function make_pull()
|
||||||
mkdir -pv ${RPMPACK}/{BUILD,BUILDROOT,RPMS,SOURCES,SPECS,SRPMS}
|
{
|
||||||
/bin/cp -f {${RUNTIMESRCDIR},${HOOKSRCDIR},${INSTALLHELPERSRCDIR},${CLISRCDIR}}/build/ascend-docker* ${RPMSOURCESDIR}
|
cd ${OPENSRC}
|
||||||
FILECNT=`ls -l ${RPMSOURCESDIR} |grep "^-"|wc -l`
|
wget -O cJSON.tar.gz https://github.com/DaveGamble/cJSON/archive/v1.7.13.tar.gz --no-check-certificate
|
||||||
echo "prepare package $FILECNT bins"
|
|
||||||
if [ $FILECNT -ne 4 ]; then
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
SPECPATH=`find ${INSTALLHELPERDIR} -name "*.spec"`
|
|
||||||
sed -i "2c Version: ${SOFT_VERSION}" ${SPECPATH}
|
|
||||||
dos2unix ${SPECPATH}
|
|
||||||
/bin/cp -f ${SPECPATH} ${RPMSPECDIR}
|
|
||||||
funcfillspec
|
|
||||||
rpmbuild --define "_topdir ${RPMPACK}"
|
|
||||||
rpmbuild --showrc | grep topdir
|
|
||||||
echo ${RPMPACK}
|
|
||||||
echo "%_topdir ${RPMPACK}" > ~/.rpmmacros
|
|
||||||
rpmbuild -bb ${RPMPACK}/SPECS/ascend-docker-plgugin.spec
|
|
||||||
RPMS=`find ${RPMPACK} -name "*.rpm"`
|
|
||||||
/bin/cp ${RPMS} ${OUTPUT}
|
|
||||||
}
|
|
||||||
|
|
||||||
funcmakeclean(){
|
|
||||||
[ -d "${RPMPACK}" ]&&rm -rf ${RPMPACK}
|
|
||||||
[ -d "${DEBPACK}" ]&&rm -rf ${DEBPACK}
|
|
||||||
[ -d "${OUTPUT}" ]&&cd ${OUTPUT}&&rm -rf *
|
|
||||||
}
|
}
|
||||||
|
|
||||||
funcmakepull(){
|
function make_unzip()
|
||||||
cd ${OPENSRC}
|
{
|
||||||
wget -O cJSON.tar.gz https://github.com/DaveGamble/cJSON/archive/v1.7.13.tar.gz --no-check-certificate
|
cd ${OPENSRC}
|
||||||
|
tar -xzvf cJSON*.tar.gz
|
||||||
|
CJSONS=`find . -name "cJSON.*"`
|
||||||
|
CJSONSLIB=${INSTALLHELPERDIR}/deb/src/cjson
|
||||||
|
/bin/cp -f ${CJSONS} ${CJSONSLIB}
|
||||||
|
|
||||||
|
unzip makeself-release-*.zip
|
||||||
|
rm -f makeself-release-*.zip
|
||||||
|
MAKESELF_DIR=`find . -name "makeself-release-*"`
|
||||||
|
/bin/cp -f ${MAKESELF_DIR}/makeself.sh ${BUILD}
|
||||||
|
/bin/cp -f ${MAKESELF_DIR}/makeself-header.sh ${BUILD}
|
||||||
|
cd ${BUILD} && /bin/cp -f scripts/mkselfmodify.patch ./
|
||||||
|
patch -p0 < mkselfmodify.patch
|
||||||
|
rm -f mkselfmodify.patch
|
||||||
|
|
||||||
|
cd ${PLATFORM}
|
||||||
|
tar -xzvf HuaweiSecureC.tar.gz
|
||||||
|
SECURECSRC=`find . -name "src"`
|
||||||
|
SECURECINC=`find . -name "include"`
|
||||||
|
|
||||||
|
SECURECLIB=${INSTALLHELPERDIR}/deb/src/HuaweiSecureC
|
||||||
|
/bin/cp -f ${SECURECSRC}/* ${SECURECLIB}
|
||||||
|
/bin/cp -f ${SECURECINC}/* ${SECURECLIB}
|
||||||
|
|
||||||
|
SECURECLIB=${CLIDIR}/src/HuaweiSecureC
|
||||||
|
/bin/cp -f ${SECURECSRC}/* ${SECURECLIB}
|
||||||
|
/bin/cp -f ${SECURECINC}/* ${SECURECLIB}
|
||||||
}
|
}
|
||||||
|
|
||||||
funcmakeunzip(){
|
make_clean
|
||||||
cd ${OPENSRC}
|
|
||||||
tar -xzvf cJSON*.tar.gz
|
|
||||||
CJSONS=`find . -name "cJSON.*"`
|
|
||||||
CJSONSLIB=${INSTALLHELPERDIR}/deb/src/cjson
|
|
||||||
/bin/cp -f ${CJSONS} ${CJSONSLIB}
|
|
||||||
|
|
||||||
cd ${PLATFORM}
|
|
||||||
tar -xzvf HuaweiSecureC.tar.gz
|
|
||||||
SECURECSRC=`find . -name "src"`
|
|
||||||
SECURECINC=`find . -name "include"`
|
|
||||||
|
|
||||||
SECURECLIB=${INSTALLHELPERDIR}/deb/src/HuaweiSecureC
|
|
||||||
/bin/cp -f ${SECURECSRC}/* ${SECURECLIB}
|
|
||||||
/bin/cp -f ${SECURECINC}/* ${SECURECLIB}
|
|
||||||
|
|
||||||
SECURECLIB=${CLIDIR}/src/HuaweiSecureC
|
|
||||||
/bin/cp -f ${SECURECSRC}/* ${SECURECLIB}
|
|
||||||
/bin/cp -f ${SECURECINC}/* ${SECURECLIB}
|
|
||||||
}
|
|
||||||
|
|
||||||
funcmakeclean
|
|
||||||
if [ $1 == "pull" ]; then
|
if [ $1 == "pull" ]; then
|
||||||
funcmakepull
|
make_pull
|
||||||
fi
|
fi
|
||||||
|
|
||||||
funcmakeunzip
|
make_unzip
|
||||||
funcbuild
|
build_bin
|
||||||
|
build_run_package
|
||||||
if [ -f /etc/centos-release ]; then
|
|
||||||
funcmakerpm
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -f /etc/debian_version ]; then
|
|
||||||
funcmakedeb
|
|
||||||
fi
|
|
||||||
|
|||||||
5
build/scripts/help.info
Normal file
5
build/scripts/help.info
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
--install Install into this system
|
||||||
|
--install-path Specify the installation path (default: /usr/local/Ascend/Ascend-Docker-Runtime)
|
||||||
|
--uninstall Uninstall the installed ascend-docker-runtime tool
|
||||||
|
--upgrade Upgrade the installed ascend-docker-runtime tool
|
||||||
|
--devel Install as devel mode
|
||||||
228
build/scripts/mkselfmodify.patch
Normal file
228
build/scripts/mkselfmodify.patch
Normal file
@@ -0,0 +1,228 @@
|
|||||||
|
--- makeself-header.sh 2020-04-27 18:59:03.000000000 +0800
|
||||||
|
+++ makeself-header_new.sh 2020-07-09 18:04:43.044000000 +0800
|
||||||
|
@@ -1,12 +1,9 @@
|
||||||
|
cat << EOF > "$archname"
|
||||||
|
-#!/bin/sh
|
||||||
|
+#!/bin/bash
|
||||||
|
# This script was generated using Makeself $MS_VERSION
|
||||||
|
# The license covering this archive and its contents, if any, is wholly independent of the Makeself license (GPL)
|
||||||
|
|
||||||
|
ORIG_UMASK=\`umask\`
|
||||||
|
-if test "$KEEP_UMASK" = n; then
|
||||||
|
- umask 077
|
||||||
|
-fi
|
||||||
|
|
||||||
|
CRCsum="$CRCsum"
|
||||||
|
MD5="$MD5sum"
|
||||||
|
@@ -16,6 +13,9 @@
|
||||||
|
export USER_PWD
|
||||||
|
ARCHIVE_DIR=`dirname \$0`
|
||||||
|
export ARCHIVE_DIR
|
||||||
|
+name_of_file="\$0 "
|
||||||
|
+pwd_of_file="\$PWD "
|
||||||
|
+
|
||||||
|
|
||||||
|
label="$LABEL"
|
||||||
|
script="$SCRIPT"
|
||||||
|
@@ -148,6 +148,24 @@
|
||||||
|
MS_Help()
|
||||||
|
{
|
||||||
|
cat << EOH >&2
|
||||||
|
+Usage: \$0 [options]
|
||||||
|
+Options:
|
||||||
|
+ --help | -h Print this message
|
||||||
|
+ --info Print embedded info : title, default target directory, embedded script ...
|
||||||
|
+ --list Print the list of files in the archive
|
||||||
|
+ --check Checks integrity of the archive
|
||||||
|
+ --quiet Quiet install mode, skip human-computer interactions
|
||||||
|
+ --noexec Do not run embedded script
|
||||||
|
+ --extract=<path> Extract directly to a target directory (absolute or relative)
|
||||||
|
+ Usually used with --noexec to just extract files without running
|
||||||
|
+ --tar arg1 [arg2 ...] Access the contents of the archive through the tar command
|
||||||
|
+\${helpheader}
|
||||||
|
+EOH
|
||||||
|
+}
|
||||||
|
+
|
||||||
|
+MS_Help1()
|
||||||
|
+{
|
||||||
|
+ cat << EOH >&2
|
||||||
|
\${helpheader}Makeself version $MS_VERSION
|
||||||
|
1) Getting help or info about \$0 :
|
||||||
|
\$0 --help Print this message
|
||||||
|
@@ -212,7 +230,7 @@
|
||||||
|
fi
|
||||||
|
sha=\`echo \$SHA | cut -d" " -f\$i\`
|
||||||
|
if test x"\$sha" = x0000000000000000000000000000000000000000000000000000000000000000; then
|
||||||
|
- test x"\$verb" = xy && echo " \$1 does not contain an embedded SHA256 checksum." >&2
|
||||||
|
+ test x"\$verb" = xy
|
||||||
|
else
|
||||||
|
shasum=\`MS_dd_Progress "\$1" \$offset \$s | eval "\$SHA_PATH \$SHA_ARG" | cut -b-64\`;
|
||||||
|
if test x"\$shasum" != x"\$sha"; then
|
||||||
|
@@ -230,7 +248,7 @@
|
||||||
|
fi
|
||||||
|
md5=\`echo \$MD5 | cut -d" " -f\$i\`
|
||||||
|
if test x"\$md5" = x00000000000000000000000000000000; then
|
||||||
|
- test x"\$verb" = xy && echo " \$1 does not contain an embedded MD5 checksum." >&2
|
||||||
|
+ test x"\$verb" = xy
|
||||||
|
else
|
||||||
|
md5sum=\`MS_dd_Progress "\$1" \$offset \$s | eval "\$MD5_PATH \$MD5_ARG" | cut -b-32\`;
|
||||||
|
if test x"\$md5sum" != x"\$md5"; then
|
||||||
|
@@ -243,7 +261,7 @@
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
if test x"\$crc" = x0000000000; then
|
||||||
|
- test x"\$verb" = xy && echo " \$1 does not contain a CRC checksum." >&2
|
||||||
|
+ test x"\$verb" = xy
|
||||||
|
else
|
||||||
|
sum1=\`MS_dd_Progress "\$1" \$offset \$s | CMD_ENV=xpg4 cksum | awk '{print \$1}'\`
|
||||||
|
if test x"\$sum1" != x"\$crc"; then
|
||||||
|
@@ -300,6 +318,22 @@
|
||||||
|
eval \$finish; exit 15
|
||||||
|
}
|
||||||
|
|
||||||
|
+Script_Args_Check()
|
||||||
|
+{
|
||||||
|
+ script_supported_args=\$(echo \${helpheader} | grep -o -P "\-\-[^ ]+" | awk -F"=" {'print \$1'})
|
||||||
|
+ arg_to_test=\$(echo \$1|awk -F"=" {'print \$1'})
|
||||||
|
+
|
||||||
|
+ for arg in \${script_supported_args[@]};
|
||||||
|
+ do
|
||||||
|
+ if test x"\$arg_to_test" = x"\$arg" ;then
|
||||||
|
+ return
|
||||||
|
+ fi
|
||||||
|
+ done
|
||||||
|
+
|
||||||
|
+ MS_Help
|
||||||
|
+ exit 1
|
||||||
|
+}
|
||||||
|
+
|
||||||
|
finish=true
|
||||||
|
xterm_loop=
|
||||||
|
noprogress=$NOPROGRESS
|
||||||
|
@@ -312,7 +346,7 @@
|
||||||
|
|
||||||
|
initargs="\$@"
|
||||||
|
|
||||||
|
-while true
|
||||||
|
+while [ -n "\$*" ]
|
||||||
|
do
|
||||||
|
case "\$1" in
|
||||||
|
-h | --help)
|
||||||
|
@@ -324,10 +358,6 @@
|
||||||
|
noprogress=y
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
- --accept)
|
||||||
|
- accept=y
|
||||||
|
- shift
|
||||||
|
- ;;
|
||||||
|
--info)
|
||||||
|
echo Identification: "\$label"
|
||||||
|
echo Target directory: "\$targetdir"
|
||||||
|
@@ -372,14 +402,6 @@
|
||||||
|
echo SKIP=\"\$skip\"
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
- --lsm)
|
||||||
|
-cat << EOLSM
|
||||||
|
-EOF
|
||||||
|
-eval "$LSM_CMD"
|
||||||
|
-cat << EOF >> "$archname"
|
||||||
|
-EOLSM
|
||||||
|
- exit 0
|
||||||
|
- ;;
|
||||||
|
--list)
|
||||||
|
echo Target directory: \$targetdir
|
||||||
|
offset=\`head -n "\$skip" "\$0" | wc -c | tr -d " "\`
|
||||||
|
@@ -422,10 +444,10 @@
|
||||||
|
keep=y
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
- --target)
|
||||||
|
+ --extract=*)
|
||||||
|
keep=y
|
||||||
|
- targetdir="\${2:-.}"
|
||||||
|
- if ! shift 2; then MS_Help; exit 1; fi
|
||||||
|
+ targetdir=\`echo \$1 | cut -d"=" -f2 \`
|
||||||
|
+ if ! shift; then MS_Help; exit 1; fi
|
||||||
|
;;
|
||||||
|
--noprogress)
|
||||||
|
noprogress=y
|
||||||
|
@@ -470,19 +492,34 @@
|
||||||
|
cleanupargs="\$2"
|
||||||
|
if ! shift 2; then MS_help; exit 1; fi
|
||||||
|
;;
|
||||||
|
- --)
|
||||||
|
- shift
|
||||||
|
- break ;;
|
||||||
|
- -*)
|
||||||
|
- echo Unrecognized flag : "\$1" >&2
|
||||||
|
- MS_Help
|
||||||
|
- exit 1
|
||||||
|
- ;;
|
||||||
|
*)
|
||||||
|
- break ;;
|
||||||
|
+ Script_Args_Check \$1
|
||||||
|
+ scriptargs="\$scriptargs \$1"
|
||||||
|
+ shift
|
||||||
|
+ ;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
+quiet_para=""
|
||||||
|
+
|
||||||
|
+if test x"\$quiet" = xy; then
|
||||||
|
+ quiet_para="--quiet "
|
||||||
|
+fi
|
||||||
|
+
|
||||||
|
+keep_para=""
|
||||||
|
+
|
||||||
|
+if test x"\$keep" = xy; then
|
||||||
|
+ keep_para="--keep "
|
||||||
|
+fi
|
||||||
|
+
|
||||||
|
+confirm_para=""
|
||||||
|
+
|
||||||
|
+if test x"\$verbose" = xy; then
|
||||||
|
+ confirm_para="--confirm "
|
||||||
|
+fi
|
||||||
|
+
|
||||||
|
+
|
||||||
|
+scriptargs="--\$name_of_file""--\$pwd_of_file""\$quiet_para""\$keep_para""\$confirm_para""\$scriptargs"
|
||||||
|
if test x"\$quiet" = xy -a x"\$verbose" = xy; then
|
||||||
|
echo Cannot be verbose and quiet at the same time. >&2
|
||||||
|
exit 1
|
||||||
|
@@ -559,7 +596,7 @@
|
||||||
|
fi
|
||||||
|
mkdir \$dashp "\$tmpdir" || {
|
||||||
|
echo 'Cannot create target directory' \$tmpdir >&2
|
||||||
|
- echo 'You should try option --target dir' >&2
|
||||||
|
+ echo 'You should try option --extract=<path>' >&2
|
||||||
|
eval \$finish
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
|
@@ -643,11 +680,19 @@
|
||||||
|
fi
|
||||||
|
|
||||||
|
if test x"\$verbose" = x"y"; then
|
||||||
|
- MS_Printf "OK to execute: \$script \$scriptargs \$* ? [Y/n] "
|
||||||
|
- read yn
|
||||||
|
- if test x"\$yn" = x -o x"\$yn" = xy -o x"\$yn" = xY; then
|
||||||
|
- eval "\"\$script\" \$scriptargs \"\\\$@\""; res=\$?;
|
||||||
|
- fi
|
||||||
|
+ yn="x"
|
||||||
|
+ while test x"\$yn" != x -a x"\$yn" != xy -a x"\$yn" != xY -a x"\$yn" != xn -a x"\$yn" != xN
|
||||||
|
+ do
|
||||||
|
+ MS_Printf "OK to execute: \$script \$scriptargs \$* ? [Y/n] "
|
||||||
|
+ read yn
|
||||||
|
+ if test x"\$yn" = x -o x"\$yn" = xy -o x"\$yn" = xY; then
|
||||||
|
+ eval "\"\$script\" \$scriptargs \"\\\$@\""; res=\$?;
|
||||||
|
+ elif test x"\$yn" = xn -o x"\$yn" = xN; then
|
||||||
|
+ echo "Unable to decompress \$script ,because of aborting! ";res=\$?
|
||||||
|
+ else
|
||||||
|
+ echo "Input value is unacceptable,please try again."
|
||||||
|
+ fi
|
||||||
|
+ done
|
||||||
|
else
|
||||||
|
eval "\"\$script\" \$scriptargs \"\\\$@\""; res=\$?
|
||||||
|
fi
|
||||||
158
build/scripts/run_main.sh
Normal file
158
build/scripts/run_main.sh
Normal file
@@ -0,0 +1,158 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
DOCKER_CONFIG_DIR=/etc/docker
|
||||||
|
INSTALL_PATH=/usr/local/Ascend/Ascend-Docker-Runtime
|
||||||
|
|
||||||
|
function install()
|
||||||
|
{
|
||||||
|
echo 'installing ascend docker runtime'
|
||||||
|
|
||||||
|
if [ ! -d "${INSTALL_PATH}" ]; then
|
||||||
|
mkdir -p ${INSTALL_PATH}
|
||||||
|
fi
|
||||||
|
|
||||||
|
cp -f ./ascend-docker-runtime ${INSTALL_PATH}/ascend-docker-runtime
|
||||||
|
cp -f ./ascend-docker-hook ${INSTALL_PATH}/ascend-docker-hook
|
||||||
|
cp -f ./ascend-docker-cli ${INSTALL_PATH}/ascend-docker-cli
|
||||||
|
chmod 550 ${INSTALL_PATH}/ascend-docker-runtime
|
||||||
|
chmod 550 ${INSTALL_PATH}/ascend-docker-hook
|
||||||
|
chmod 550 ${INSTALL_PATH}/ascend-docker-cli
|
||||||
|
echo 'install executable files success'
|
||||||
|
|
||||||
|
if [ ! -d "${DOCKER_CONFIG_DIR}" ]; then
|
||||||
|
mkdir -p ${DOCKER_CONFIG_DIR}
|
||||||
|
fi
|
||||||
|
|
||||||
|
SRC="${DOCKER_CONFIG_DIR}/daemon.json.${PPID}"
|
||||||
|
DST="${DOCKER_CONFIG_DIR}/daemon.json"
|
||||||
|
./ascend-docker-plugin-install-helper add ${DST} ${SRC} ${INSTALL_PATH}/ascend-docker-runtime
|
||||||
|
if [ "$?" != "0" ]; then
|
||||||
|
echo 'create damon.json failed'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
mv ${SRC} ${DST}
|
||||||
|
echo 'create damom.json success'
|
||||||
|
echo 'please reboot docker daemon to take effect'
|
||||||
|
}
|
||||||
|
|
||||||
|
function uninstall()
|
||||||
|
{
|
||||||
|
echo 'uninstalling ascend docker runtime'
|
||||||
|
|
||||||
|
if [ ! -d "${INSTALL_PATH}" ]; then
|
||||||
|
echo 'WARNING: the specified install path does not exist, skipping'
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
rm -rf ${INSTALL_PATH}
|
||||||
|
echo 'remove executable files success'
|
||||||
|
|
||||||
|
SRC="${DOCKER_CONFIG_DIR}/daemon.json.${PPID}"
|
||||||
|
DST="${DOCKER_CONFIG_DIR}/daemon.json"
|
||||||
|
if [ ! -f "${DST}" ]; then
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
./ascend-docker-plugin-install-helper rm ${DST} ${SRC}
|
||||||
|
if [ "$?" != "0" ]; then
|
||||||
|
echo 'ERROR: del damon.json failed'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
mv ${SRC} ${DST}
|
||||||
|
echo 'del damom.json success'
|
||||||
|
}
|
||||||
|
|
||||||
|
function upgrade()
|
||||||
|
{
|
||||||
|
echo 'upgrading ascend docker runtime'
|
||||||
|
|
||||||
|
if [ ! -d "${INSTALL_PATH}" ]; then
|
||||||
|
echo 'ERROR: the specified install path does not exist, stopping upgrading'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
cp -f ./ascend-docker-runtime ${INSTALL_PATH}/ascend-docker-runtime
|
||||||
|
cp -f ./ascend-docker-hook ${INSTALL_PATH}/ascend-docker-hook
|
||||||
|
cp -f ./ascend-docker-cli ${INSTALL_PATH}/ascend-docker-cli
|
||||||
|
chmod 550 ${INSTALL_PATH}/ascend-docker-runtime
|
||||||
|
chmod 550 ${INSTALL_PATH}/ascend-docker-hook
|
||||||
|
chmod 550 ${INSTALL_PATH}/ascend-docker-cli
|
||||||
|
|
||||||
|
echo 'upgrade ascend docker runtime success'
|
||||||
|
}
|
||||||
|
|
||||||
|
INSTALL_FLAG=n
|
||||||
|
INSTALL_PATH_FLAG=n
|
||||||
|
UNINSTALL_FLAG=n
|
||||||
|
UPGRADE_FLAG=n
|
||||||
|
DEVEL_FLAG=n
|
||||||
|
|
||||||
|
while true
|
||||||
|
do
|
||||||
|
case "$3" in
|
||||||
|
--install)
|
||||||
|
INSTALL_FLAG=y
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
--uninstall)
|
||||||
|
UNINSTALL_FLAG=y
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
--install-path=*)
|
||||||
|
INSTALL_PATH_FLAG=y
|
||||||
|
INSTALL_PATH=`echo $3 | cut -d"=" -f2 `
|
||||||
|
INSTALL_PATH=`echo ${INSTALL_PATH} | sed "s/\/*$//g"`
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
--upgrade)
|
||||||
|
UPGRADE_FLAG=y
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
--devel)
|
||||||
|
DEVEL_FLAG=y
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
break
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
# 安装为相对路径时报错
|
||||||
|
if [ "${INSTALL_PATH}" == ".." ] || [ "${INSTALL_PATH}" == "." ]; then
|
||||||
|
echo "error :Please follow the installation address after the --install-path=<Absolute path>"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 单纯只有--install-path的判定处理
|
||||||
|
if [ "${INSTALL_PATH_FLAG}" == "y" ] && \
|
||||||
|
[ "${INSTALL_FLAG}" == "n" ] && \
|
||||||
|
[ "${UNINSTALL_FLAG}" == "n" ] && \
|
||||||
|
[ "${UPGRADE_FLAG}" == "n" ] && \
|
||||||
|
[ "${DEVEL_FLAG}" == "n" ]; then
|
||||||
|
echo "Error:only input <install_path> command. When use --install-path you also need intput --install or --uninstall or --upgrade or --devel"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 需root用户权限
|
||||||
|
if [ "${USER}" != "root" ]; then
|
||||||
|
echo 'please run with root permission'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "${INSTALL_FLAG}" == "y" ] || [ "${DEVEL_FLAG}" == "y" ]; then
|
||||||
|
install
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "${UNINSTALL_FLAG}" == "y" ]; then
|
||||||
|
uninstall
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "${UPGRADE_FLAG}" == "y" ]; then
|
||||||
|
upgrade
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
@@ -32,6 +32,20 @@
|
|||||||
</copy>
|
</copy>
|
||||||
</copies>
|
</copies>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<versionType>3rd-party</versionType>
|
||||||
|
<repoType>Generic</repoType>
|
||||||
|
<id>
|
||||||
|
<artifactName>makeself-release-2.4.2.zip</artifactName>
|
||||||
|
<pdmCode>05836ECT</pdmCode>
|
||||||
|
</id>
|
||||||
|
<copies>
|
||||||
|
<copy>
|
||||||
|
<source>/*</source>
|
||||||
|
<dest>opensource</dest>
|
||||||
|
</copy>
|
||||||
|
</copies>
|
||||||
|
</dependency>
|
||||||
<!-- Platform -->
|
<!-- Platform -->
|
||||||
<dependency>
|
<dependency>
|
||||||
<versionType>BVersion</versionType>
|
<versionType>BVersion</versionType>
|
||||||
|
|||||||
15
hook/main.go
15
hook/main.go
@@ -10,7 +10,6 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
|
||||||
"path"
|
"path"
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
"strconv"
|
||||||
@@ -25,7 +24,7 @@ const (
|
|||||||
ascendVisibleDevices = "ASCEND_VISIBLE_DEVICES"
|
ascendVisibleDevices = "ASCEND_VISIBLE_DEVICES"
|
||||||
ascendRuntimeOptions = "ASCEND_RUNTIME_OPTIONS"
|
ascendRuntimeOptions = "ASCEND_RUNTIME_OPTIONS"
|
||||||
ascendDockerCli = "ascend-docker-cli"
|
ascendDockerCli = "ascend-docker-cli"
|
||||||
defaultAscendDockerCli = "/usr/bin/ascend-docker-cli"
|
defaultAscendDockerCli = "/usr/local/bin/ascend-docker-cli"
|
||||||
|
|
||||||
borderNum = 2
|
borderNum = 2
|
||||||
kvPairSize = 2
|
kvPairSize = 2
|
||||||
@@ -221,14 +220,14 @@ func doPrestartHook() error {
|
|||||||
return fmt.Errorf("failed to parse runtime options: %w", err)
|
return fmt.Errorf("failed to parse runtime options: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cliPath, err := exec.LookPath(ascendDockerCliName)
|
currentExecPath, err := os.Executable()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_, err = os.Stat(defaultAscendDockerCliName)
|
return fmt.Errorf("cannot get the path of ascend-docker-hook: %w", err)
|
||||||
if err != nil {
|
}
|
||||||
return fmt.Errorf("could not found ascend docker cli")
|
|
||||||
}
|
|
||||||
|
|
||||||
cliPath = defaultAscendDockerCliName
|
cliPath := path.Join(path.Dir(currentExecPath), ascendDockerCliName)
|
||||||
|
if _, err = os.Stat(cliPath); err != nil {
|
||||||
|
return fmt.Errorf("cannot find ascend-docker-cli executable file at %s: %w", cliPath, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
args := append([]string{cliPath},
|
args := append([]string{cliPath},
|
||||||
|
|||||||
@@ -9,13 +9,14 @@
|
|||||||
#include "cJSON.h"
|
#include "cJSON.h"
|
||||||
|
|
||||||
#define MAX_JSON_FILE_SIZE 65535
|
#define MAX_JSON_FILE_SIZE 65535
|
||||||
#define NUM_ARGS 4
|
#define MIN_ARGS_NUM 4
|
||||||
|
#define ADD_CMD_ARGS_NUM 5
|
||||||
#define ADD_CMD "add"
|
#define ADD_CMD "add"
|
||||||
#define RM_CMD "rm"
|
#define RM_CMD "rm"
|
||||||
#define CMD_INDEX 1
|
#define CMD_INDEX 1
|
||||||
#define FINAL_FILE_INDEX 2
|
#define FINAL_FILE_INDEX 2
|
||||||
#define TEMP_FILE_INDEX 3
|
#define TEMP_FILE_INDEX 3
|
||||||
#define ASCEND_RUNTIME_PATH_VALUE "/usr/bin/ascend-docker-runtime"
|
#define RUNTIME_PATH_INDEX 4
|
||||||
#define ASCEND_RUNTIME_PATH_KEY "path"
|
#define ASCEND_RUNTIME_PATH_KEY "path"
|
||||||
#define ASCEND_RUNTIME_ARGS_KEY "runtimeArgs"
|
#define ASCEND_RUNTIME_ARGS_KEY "runtimeArgs"
|
||||||
#define RUNTIME_KEY "runtimes"
|
#define RUNTIME_KEY "runtimes"
|
||||||
@@ -38,7 +39,7 @@ void ReadJsonFile(FILE *pf, char *text, int maxBufferSize)
|
|||||||
text[size] = '\0';
|
text[size] = '\0';
|
||||||
}
|
}
|
||||||
|
|
||||||
cJSON *CreateAscendRuntimeInfo()
|
cJSON *CreateAscendRuntimeInfo(const char *runtimePath)
|
||||||
{
|
{
|
||||||
cJSON *root = NULL;
|
cJSON *root = NULL;
|
||||||
root = cJSON_CreateObject();
|
root = cJSON_CreateObject();
|
||||||
@@ -46,15 +47,15 @@ cJSON *CreateAscendRuntimeInfo()
|
|||||||
fprintf(stderr, "create ascend runtime info root err\n");
|
fprintf(stderr, "create ascend runtime info root err\n");
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
cJSON *newString = NULL;
|
cJSON *newString = NULL;
|
||||||
newString = cJSON_CreateString(ASCEND_RUNTIME_PATH_VALUE);
|
newString = cJSON_CreateString(runtimePath);
|
||||||
if (newString == NULL) {
|
if (newString == NULL) {
|
||||||
fprintf(stderr, "create ascend runtime info path value err\n");
|
fprintf(stderr, "create ascend runtime info path value err\n");
|
||||||
cJSON_Delete(root);
|
cJSON_Delete(root);
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
cJSON *paraArray = NULL;
|
cJSON *paraArray = NULL;
|
||||||
paraArray = cJSON_CreateArray();
|
paraArray = cJSON_CreateArray();
|
||||||
if (paraArray == NULL) {
|
if (paraArray == NULL) {
|
||||||
@@ -70,10 +71,10 @@ cJSON *CreateAscendRuntimeInfo()
|
|||||||
return root;
|
return root;
|
||||||
}
|
}
|
||||||
|
|
||||||
cJSON *CreateRuntimes()
|
cJSON *CreateRuntimes(const char *runtimePath)
|
||||||
{
|
{
|
||||||
cJSON *ascendRuntime = NULL;
|
cJSON *ascendRuntime = NULL;
|
||||||
ascendRuntime = CreateAscendRuntimeInfo();
|
ascendRuntime = CreateAscendRuntimeInfo(runtimePath);
|
||||||
if (ascendRuntime == NULL) {
|
if (ascendRuntime == NULL) {
|
||||||
fprintf(stderr, "create ascendruntime err\n");
|
fprintf(stderr, "create ascendruntime err\n");
|
||||||
return NULL;
|
return NULL;
|
||||||
@@ -111,11 +112,11 @@ int DelJsonContent(cJSON *root, const char *key)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
cJSON *CreateContent()
|
cJSON *CreateContent(const char *runtimePath)
|
||||||
{
|
{
|
||||||
/* 插入ascend runtime */
|
/* 插入ascend runtime */
|
||||||
cJSON *runtimes = NULL;
|
cJSON *runtimes = NULL;
|
||||||
runtimes = CreateRuntimes();
|
runtimes = CreateRuntimes(runtimePath);
|
||||||
if (runtimes == NULL) {
|
if (runtimes == NULL) {
|
||||||
fprintf(stderr, "create runtimes err\n");
|
fprintf(stderr, "create runtimes err\n");
|
||||||
return NULL;
|
return NULL;
|
||||||
@@ -145,7 +146,7 @@ cJSON *CreateContent()
|
|||||||
return root;
|
return root;
|
||||||
}
|
}
|
||||||
|
|
||||||
cJSON *ModifyContent(FILE *pf)
|
cJSON *ModifyContent(FILE *pf, const char *runtimePath)
|
||||||
{
|
{
|
||||||
char jsonStr[MAX_JSON_FILE_SIZE] = {0x0};
|
char jsonStr[MAX_JSON_FILE_SIZE] = {0x0};
|
||||||
ReadJsonFile(pf, &jsonStr[0], MAX_JSON_FILE_SIZE);
|
ReadJsonFile(pf, &jsonStr[0], MAX_JSON_FILE_SIZE);
|
||||||
@@ -161,7 +162,7 @@ cJSON *ModifyContent(FILE *pf)
|
|||||||
cJSON *runtimes = NULL;
|
cJSON *runtimes = NULL;
|
||||||
runtimes = cJSON_GetObjectItem(root, "runtimes");
|
runtimes = cJSON_GetObjectItem(root, "runtimes");
|
||||||
if (runtimes == NULL) {
|
if (runtimes == NULL) {
|
||||||
runtimes = CreateRuntimes();
|
runtimes = CreateRuntimes(runtimePath);
|
||||||
if (runtimes == NULL) {
|
if (runtimes == NULL) {
|
||||||
cJSON_Delete(root);
|
cJSON_Delete(root);
|
||||||
return NULL;
|
return NULL;
|
||||||
@@ -174,7 +175,7 @@ cJSON *ModifyContent(FILE *pf)
|
|||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
cJSON *ascendRuntime = NULL;
|
cJSON *ascendRuntime = NULL;
|
||||||
ascendRuntime = CreateAscendRuntimeInfo();
|
ascendRuntime = CreateAscendRuntimeInfo(runtimePath);
|
||||||
if (ascendRuntime == NULL) {
|
if (ascendRuntime == NULL) {
|
||||||
cJSON_Delete(root);
|
cJSON_Delete(root);
|
||||||
return NULL;
|
return NULL;
|
||||||
@@ -236,15 +237,15 @@ cJSON *RemoveContent(FILE *pf)
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
int DetectAndCreateJsonFile(const char *filePath, const char *tempPath)
|
int DetectAndCreateJsonFile(const char *filePath, const char *tempPath, const char *runtimePath)
|
||||||
{
|
{
|
||||||
cJSON *root = NULL;
|
cJSON *root = NULL;
|
||||||
FILE *pf = NULL;
|
FILE *pf = NULL;
|
||||||
pf = fopen(filePath, "r+");
|
pf = fopen(filePath, "r+");
|
||||||
if (pf == NULL) {
|
if (pf == NULL) {
|
||||||
root = CreateContent();
|
root = CreateContent(runtimePath);
|
||||||
} else {
|
} else {
|
||||||
root = ModifyContent(pf);
|
root = ModifyContent(pf, runtimePath);
|
||||||
fclose(pf);
|
fclose(pf);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -302,14 +303,21 @@ int CreateRevisedJsonFile(const char *filePath, const char *tempPath)
|
|||||||
/* 该函数只负责生成json.bak文件，由调用者进行覆盖操作 */
|
/* 该函数只负责生成json.bak文件，由调用者进行覆盖操作 */
|
||||||
int main(int argc, char *argv[])
|
int main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
if (argc != NUM_ARGS) {
|
if (argc < MIN_ARGS_NUM) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
printf("%s\n", argv[FINAL_FILE_INDEX]);
|
printf("%s\n", argv[FINAL_FILE_INDEX]);
|
||||||
printf("%s\n", argv[TEMP_FILE_INDEX]);
|
printf("%s\n", argv[TEMP_FILE_INDEX]);
|
||||||
printf("%s\n", argv[CMD_INDEX]);
|
printf("%s\n", argv[CMD_INDEX]);
|
||||||
|
|
||||||
if (strcmp(argv[CMD_INDEX], ADD_CMD) == 0) {
|
if (strcmp(argv[CMD_INDEX], ADD_CMD) == 0) {
|
||||||
return DetectAndCreateJsonFile(argv[FINAL_FILE_INDEX], argv[TEMP_FILE_INDEX]);
|
if (argc != ADD_CMD_ARGS_NUM) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return DetectAndCreateJsonFile(argv[FINAL_FILE_INDEX], argv[TEMP_FILE_INDEX], argv[RUNTIME_PATH_INDEX]);
|
||||||
}
|
}
|
||||||
|
|
||||||
return CreateRevisedJsonFile(argv[FINAL_FILE_INDEX], argv[TEMP_FILE_INDEX]);
|
return CreateRevisedJsonFile(argv[FINAL_FILE_INDEX], argv[TEMP_FILE_INDEX]);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ import (
|
|||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
|
|
||||||
@@ -20,7 +21,7 @@ import (
|
|||||||
const (
|
const (
|
||||||
loggingPrefix = "ascend-docker-runtime"
|
loggingPrefix = "ascend-docker-runtime"
|
||||||
hookCli = "ascend-docker-hook"
|
hookCli = "ascend-docker-hook"
|
||||||
hookDefaultFilePath = "/usr/bin/ascend-docker-hook"
|
hookDefaultFilePath = "/usr/local/bin/ascend-docker-hook"
|
||||||
dockerRuncFile = "docker-runc"
|
dockerRuncFile = "docker-runc"
|
||||||
runcFile = "runc"
|
runcFile = "runc"
|
||||||
)
|
)
|
||||||
@@ -71,12 +72,14 @@ var execRunc = func() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func addHook(spec *specs.Spec) error {
|
func addHook(spec *specs.Spec) error {
|
||||||
path, err := exec.LookPath(hookCliPath)
|
currentExecPath, err := os.Executable()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
path = hookDefaultFile
|
return fmt.Errorf("cannot get the path of ascend-docker-runtime: %w", err)
|
||||||
if _, err = os.Stat(path); err != nil {
|
}
|
||||||
return fmt.Errorf("cannot find the hook")
|
|
||||||
}
|
hookCliPath = path.Join(path.Dir(currentExecPath), hookCli)
|
||||||
|
if _, err = os.Stat(hookCliPath); err != nil {
|
||||||
|
return fmt.Errorf("cannot find ascend-docker-hook executable file at %s: %w", hookCliPath, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if spec.Hooks == nil {
|
if spec.Hooks == nil {
|
||||||
@@ -91,8 +94,8 @@ func addHook(spec *specs.Spec) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
spec.Hooks.Prestart = append(spec.Hooks.Prestart, specs.Hook{
|
spec.Hooks.Prestart = append(spec.Hooks.Prestart, specs.Hook{
|
||||||
Path: path,
|
Path: hookCliPath,
|
||||||
Args: []string{path},
|
Args: []string{hookCliPath},
|
||||||
})
|
})
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
Reference in New Issue
Block a user