source test-header.sh

# ======================================================================

#

# Initial setup.

#

# ======================================================================

PYARMOR="${PYTHON} pyarmor.py"

csih_inform "Python is $PYTHON"

csih_inform "Tested Package: $pkgfile"

csih_inform "PyArmor is $PYARMOR"

csih_inform "Make workpath ${workpath}"

rm -rf ${workpath}

mkdir -p ${workpath} || csih_error "Make workpath FAILED"

csih_inform "Clean pyarmor data"

rm -rf ~/.pyarmor ~/.pyarmor_capsule.*

[[ -n "$USERPROFILE" ]] && rm -rf "$USERPROFILE\\.pyarmor*"

datafile=$(pwd)/data/pyarmor-data.tar.gz

capsulefile=$(pwd)/data/pyarmor-test-0001.zip

# Form v5.9.0, support PYARMOR_HOME

workhome=pyarmor_home

export PYARMOR_HOME="$workhome"

cd ${workpath}

[[ ${pkgfile} == *.zip ]] && unzip ${pkgfile} > /dev/null 2>&1

[[ ${pkgfile} == *.tar.bz2 ]] && tar xjf ${pkgfile}

cd pyarmor-$version || csih_error "Invalid pyarmor package file"

# From pyarmor 3.5.1, main scripts are moved to src

[[ -d src ]] && mv src/* ./

# Fix issue: assert_builtin(open) fails in python 3.0

patch_cross_protection_code_for_python3.0

csih_inform "Extract test data"

mkdir data && (cd data; tar xzf $datafile)

cp $capsulefile data/

# From pyarmor 4.5.4, platform name is renamed

# From pyarmor 5.7.5, platform name is changed

csih_inform "Add execute permission to dynamic library"

find ./platforms -name _pytransform.dll -exec chmod +x {} \;

csih_inform "Prepare for function testing"

echo ""

# ======================================================================

#

# Start test with trial version.

#

# ======================================================================

echo ""

echo "-------------------- Test Trial Version ------------------------"

echo ""

csih_inform "1. Show version information"

$PYARMOR --version >result.log 2>&1 || csih_bug "show version FAILED"

check_file_exists $workhome/license.lic

csih_inform "2. Obfuscate foo.py"

$PYARMOR obfuscate examples/helloworld/foo.py >result.log 2>&1

check_return_value

check_file_exists $workhome/.pyarmor_capsule.zip

csih_inform "3. Run obfuscated foo.py"

(cd dist; $PYTHON foo.py >result.log 2>&1)

check_return_value

check_file_content dist/result.log "is never expired"

check_file_content dist/result.log "Hello world"

check_file_content dist/result.log "1 + 1 = 2"

csih_inform "4. Generate expired license"

$PYARMOR licenses -e $(next_month) Joker >result.log 2>&1

check_return_value

check_file_exists licenses/Joker/license.lic

csih_inform "5. Run obfuscated foo.py with expired license"

$PYARMOR obfuscate --with-license licenses/Joker/license.lic \

examples/helloworld/foo.py >result.log 2>&1

check_return_value

(cd dist; $PYTHON foo.py >result.log 2>&1)

check_return_value

check_file_content dist/result.log "This license for Joker will be expired in"

check_file_content dist/result.log "Hello world"

check_file_content dist/result.log "1 + 1 = 2"

csih_inform "6. Import obfuscated package"

$PYARMOR obfuscate -O dist/mypkg examples/testpkg/mypkg/__init__.py >result.log 2>&1

(cd dist; $PYTHON -c "from mypkg import foo

foo.hello('pyarmor')" >result.log 2>&1)

check_return_value

check_file_content dist/result.log "Hello!"

csih_inform "7. Import many obfuscated packages"

PROPATH=test-many-packages

mkdir -p $PROPATH/pkg1

mkdir -p $PROPATH/pkg2

echo "name = 'This is first package'" > $PROPATH/pkg1/__init__.py

echo "name = 'This is second package'" > $PROPATH/pkg2/__init__.py

$PYARMOR init --src=$PROPATH/pkg1 --entry=__init__.py $PROPATH/pkg1 >result.log 2>&1

$PYARMOR init --src=$PROPATH/pkg2 --entry=__init__.py $PROPATH/pkg2 >result.log 2>&1

$PYARMOR build --output $PROPATH/obf --no-runtime $PROPATH/pkg1 >result.log 2>&1

$PYARMOR build --output $PROPATH/obf --no-runtime $PROPATH/pkg2 >result.log 2>&1

$PYARMOR build --output $PROPATH/obf --only-runtime $PROPATH/pkg1 >result.log 2>&1

cat <<EOF > $PROPATH/obf/main.py

from pkg1 import name as pkg_name1

from pkg2 import name as pkg_name2

print(pkg_name1)

print(pkg_name2)

EOF

(cd $PROPATH/obf; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/obf/result.log 'This is first package'

check_file_content $PROPATH/obf/result.log 'This is second package'

csih_inform "8. Obfuscate scripts with advanced mode"

$PYARMOR obfuscate --advanced 1 --output dist-trial-advanced \

examples/simple/queens.py >result.log 2>&1

check_return_value

check_file_exists dist-trial-advanced/queens.py

(cd dist-trial-advanced; $PYTHON queens.py >result.log 2>&1)

check_return_value

check_file_content dist-trial-advanced/result.log 'Found 92 solutions'

csih_inform "9. Obfuscate scripts with advanced mode but more than 30 functions"

let -i n=0

while (( n < 36 )) ; do

(( n++ ))

echo "def foo$n(i):

return i + 1" >> t32.py

done

echo "print('Hello world')" >> t32.py

$PYARMOR obfuscate --advanced 1 --output dist-trial-advanced-2 --exact t32.py >result.log 2>&1

check_file_content result.log 'In trial version the limitation is about'

csih_inform "10. Run big array scripts"

ascript="big_array.py"

$PYTHON -c"

with open('$ascript', 'w') as f:

for i in range(100):

f.write('a{0} = {1}\n'.format(i, [1] * 1000))"

$PYARMOR obfuscate --exact -O dist-big-array $ascript >result.log 2>&1

check_file_content result.log 'Too big code object, the limitation is'

csih_inform "11. Obfuscate big code object without wrap mode"

PROPATH=projects/test_big_code_object

$PYTHON -c"

with open('big_array.py', 'w') as f:

for i in range(100):

f.write('a{0} = {1}\n'.format(i, [1] * 1000))

f.write('print(\"a99 = %s\" % a99)')"

$PYARMOR init --src=. --entry=big_array.py -t app $PROPATH >result.log 2>&1

$PYARMOR config --wrap-mode=0 --manifest="include big_array.py" $PROPATH >result.log 2>&1

(cd $PROPATH; $ARMOR build >result.log 2>&1)

check_file_content $PROPATH/result.log 'Too big code object, the limitation is'

csih_inform "12. Obfuscate the scripts with option --enable-suffix"

$PYARMOR obfuscate -O dist-suffix --enable-suffix \

examples/simple/queens.py >result.log 2>&1

check_return_value

check_file_exists dist-suffix/pytransform/__init__.py

(cd dist-suffix; $PYTHON queens.py >result.log 2>&1)

check_return_value

check_file_content dist-suffix/result.log 'Found 92 solutions'

csih_inform "13. Generate runtime package with option --enable-suffix"

$PYARMOR runtime -O test-runtime-suffix --enable-suffix >result.log 2>&1

check_return_value

check_file_exists test-runtime-suffix/pytransform/__init__.py

check_file_content test-runtime-suffix/pytransform_bootstrap.py \

'from pytransform import pyarmor_runtime'

$PYARMOR runtime -O test-runtime-suffix-2 --no-package --enable-suffix >result.log 2>&1

check_return_value

check_file_exists test-runtime-suffix-2/pytransform.py

check_file_content test-runtime-suffix-2/pytransform_bootstrap.py \

'from pytransform import pyarmor_runtime'

csih_inform "14. Build project with option --enable-suffix"

PROPATH=projects/test_enable_suffix

mkdir $PROPATH

echo "print('Hello this is a project with suffix')" > $PROPATH/main.py

$PYARMOR init --src=$PROPATH --entry=main.py $PROPATH >result.log 2>&1

$PYARMOR config --enable-suffix 1 $PROPATH >result.log 2>&1

check_return_value

(cd $PROPATH; $ARMOR build >result.log 2>&1)

check_return_value

check_file_exists $PROPATH/dist/pytransform/__init__.py

(cd $PROPATH/dist; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/dist/result.log 'Hello this is a project with suffix'

csih_inform "15. Import the package obfuscated by others"

PROPATH=test-import-other-package

mkdir -p $PROPATH/pkg1 $PROPATH/pkg2

echo "print('This is package 1')" > $PROPATH/pkg1/__init__.py

echo "print('This is package 2')" > $PROPATH/pkg2/__init__.py

cat <<EOF > $PROPATH/main.py

from time import sleep

sleep(1.0)

import pkg1, pkg2

print('This is main script')

EOF

$PYARMOR obfuscate -O $PROPATH/dist/pkg1 $PROPATH/pkg1/__init__.py >result.log 2>&1

$PYARMOR obfuscate -O $PROPATH/dist/pkg2 --advanced 1 $PROPATH/pkg2/__init__.py >result.log 2>&1

$PYARMOR obfuscate -O $PROPATH/dist --exact $PROPATH/main.py >result.log 2>&1

(cd $PROPATH/dist; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/dist/result.log 'This is package 1'

check_file_content $PROPATH/dist/result.log 'This is package 2'

check_file_content $PROPATH/dist/result.log 'This is main script'

csih_inform "16. Import the package obfuscated by others with advanced mode"

$PYARMOR obfuscate -O $PROPATH/dist --advanced 1 --exact $PROPATH/main.py >result.log 2>&1

(cd $PROPATH/dist; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/dist/result.log 'This is package 1'

check_file_content $PROPATH/dist/result.log 'This is package 2'

check_file_content $PROPATH/dist/result.log 'This is main script'

csih_inform "17. Obfuscate big script"

ascript="big_script.py"

$PYTHON -c"

with open('$ascript', 'w') as f:

for i in range(200):

f.write('def test_{0}(n):\n'.format(i))

f.write(' return n + {0}\n'.format(i))"

$PYARMOR obfuscate --exact -O dist-big-script $ascript >result.log 2>&1

check_file_content result.log 'Too big code object, the limitation is'

csih_inform "18. Get issuer of the obfuscate scripts"

(cd dist;

$PYTHON -c"

from pytransform import pyarmor_init, get_license_info

pyarmor_init(is_runtime=1)

licinfo = get_license_info()

print('License code is %s' % licinfo['CODE'])

print('Issuer is %s' % licinfo['ISSUER'])" > result.log 2>&1)

check_return_value

check_file_content dist/result.log 'License code is Joker'

check_file_content dist/result.log 'Issuer is trial'

csih_inform "19. Protect data file"

dist=test-data-file

$PYTHON -m helper.build_data_module protect_code2.pt > protect_data.py

cat <<EOF > safedata.py

import protect_data

with protect_data.Safestr() as text:

print('Got data: %s' % text[:32])

from pytransform import clean_str

from sys import version_info as ver

data = ('a' * 30) if ver[0] == 2 else (b'a' * 30).decode()

clean_str(data)

if (ver[0] * 10 + ver[1]) not in (30, 31, 32):

print(data)

EOF

$PYARMOR obfuscate --exact -O $dist safedata.py >result.log 2>&1

$PYARMOR obfuscate --exact -O $dist --no-runtime --no-bootstrap \

--restrict 4 protect_data.py >result.log 2>&1

(cd $dist; $PYTHON safedata.py >result.log 2>&1)

check_return_value

check_file_content $dist/result.log 'Got data: def protect_pytransform'

check_file_content $dist/result.log 'aaaaaaaaaa' not

csih_inform "20. Obfuscate scripts with --runtime"

dist=test-with-runtime

$PYARMOR obfuscate --runtime test-runtime-suffix -O $dist \

examples/simple/queens.py >result.log 2>&1

(cd $dist; $PYTHON queens.py >result.log 2>&1)

check_return_value

check_file_content $dist/result.log 'Found 92 solutions'

# ======================================================================

#

# Start test with normal version.

#

# ======================================================================

echo ""

echo "-------------------- Test Normal Version ------------------------"

echo ""

test_suffix="_unk_000001"

csih_inform "0. Register keyfile"

$PYARMOR register data/pyarmor-test-0001.zip >result.log 2>&1

check_return_value

check_file_exists $workhome/license.lic

$PYARMOR register >result.log 2>&1

check_file_content result.log "PyArmor Trial" not

csih_inform "1. Show version information"

$PYARMOR --version >result.log 2>&1

check_file_content result.log "pyarmor-test-0001"

check_file_content result.log "PyArmor Trial" not

csih_inform "2. Obfuscate foo.py"

$PYARMOR obfuscate examples/helloworld/foo.py >result.log 2>&1

check_return_value

check_file_exists $workhome/.pyarmor_capsule.zip

csih_inform "3. Run obfuscated foo.py"

(cd dist; $PYTHON foo.py >result.log 2>&1)

check_return_value

check_file_content dist/result.log "is never expired"

check_file_content dist/result.log "Hello world"

check_file_content dist/result.log "1 + 1 = 2"

csih_inform "4. Generate expired license"

$PYARMOR licenses -e $(next_month) Joker >result.log 2>&1

check_return_value

check_file_exists licenses/Joker/license.lic

csih_inform "5. Run obfuscated foo.py with expired license"

$PYARMOR obfuscate --with-license licenses/Joker/license.lic \

examples/helloworld/foo.py >result.log 2>&1

check_return_value

(cd dist; $PYTHON foo.py >result.log 2>&1)

check_return_value

check_file_content dist/result.log "This license for Joker will be expired in"

check_file_content dist/result.log "Hello world"

check_file_content dist/result.log "1 + 1 = 2"

csih_inform "6. Import obfuscated package"

$PYARMOR obfuscate -O dist/mypkg examples/testpkg/mypkg/__init__.py >result.log 2>&1

(cd dist; $PYTHON -c "from mypkg import foo

foo.hello('pyarmor')" >result.log 2>&1)

check_return_value

check_file_content dist/result.log "Hello!"

csih_inform "7. Import many obfuscated packages"

PROPATH=test-many-packages

mkdir -p $PROPATH/pkg1

mkdir -p $PROPATH/pkg2

echo "name = 'This is first package'" > $PROPATH/pkg1/__init__.py

echo "name = 'This is second package'" > $PROPATH/pkg2/__init__.py

$PYARMOR init --src=$PROPATH/pkg1 --entry=__init__.py $PROPATH/pkg1 >result.log 2>&1

$PYARMOR init --src=$PROPATH/pkg2 --entry=__init__.py $PROPATH/pkg2 >result.log 2>&1

$PYARMOR build --output $PROPATH/obf --no-runtime $PROPATH/pkg1 >result.log 2>&1

$PYARMOR build --output $PROPATH/obf --no-runtime $PROPATH/pkg2 >result.log 2>&1

$PYARMOR build --output $PROPATH/obf --only-runtime $PROPATH/pkg1 >result.log 2>&1

cat <<EOF > $PROPATH/obf/main.py

from pkg1 import name as pkg_name1

from pkg2 import name as pkg_name2

print(pkg_name1)

print(pkg_name2)

EOF

(cd $PROPATH/obf; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/obf/result.log 'This is first package'

check_file_content $PROPATH/obf/result.log 'This is second package'

csih_inform "8. Obfuscate scripts with advanced mode"

$PYARMOR obfuscate --advanced 1 --output dist-advanced examples/simple/queens.py >result.log 2>&1

check_return_value

check_file_exists dist-advanced/queens.py

(cd dist-advanced; $PYTHON queens.py >result.log 2>&1)

check_return_value

check_file_content dist-advanced/result.log 'Found 92 solutions'

csih_inform "9. Obfuscate scripts with advanced mode but more than 30 functions"

$PYARMOR obfuscate --advanced 1 --output dist-advanced-2 --exact t32.py >result.log 2>&1

check_return_value

check_file_exists dist-advanced-2/t32.py

(cd dist-advanced-2; $PYTHON t32.py >result.log 2>&1)

check_return_value

check_file_content dist-advanced-2/result.log 'Hello world'

csih_inform "10. Run big array scripts"

ascript="big_array.py"

$PYTHON -c"

with open('$ascript', 'w') as f:

for i in range(100):

f.write('a{0} = {1}\n'.format(i, [1] * 1000))"

$PYARMOR obfuscate --exact -O dist-big-array $ascript >result.log 2>&1

check_return_value

csih_inform "11. Obfuscate big code object without wrap mode"

PROPATH=projects/test_big_code_object

$PYTHON -c"

with open('big_array.py', 'w') as f:

for i in range(100):

f.write('a{0} = {1}\n'.format(i, [1] * 1000))

f.write('print(\"a99 = %s\" % a99)')"

$PYARMOR init --src=. --entry=big_array.py -t app $PROPATH >result.log 2>&1

$PYARMOR config --wrap-mode=0 --manifest="include big_array.py" $PROPATH >result.log 2>&1

$PYARMOR build $PROPATH >$PROPATH/result.log 2>&1

check_file_exists $PROPATH/dist/big_array.py

check_file_content $PROPATH/dist/big_array.py 'pyarmor_runtime'

check_file_content $PROPATH/dist/big_array.py '__pyarmor__(__name__'

(cd $PROPATH/dist; $PYTHON big_array.py >result.log 2>&1 )

check_return_value

check_file_content $PROPATH/dist/result.log 'a99 ='

csih_inform "12. Obfuscate the scripts with option --enable-suffix"

$PYARMOR obfuscate -O dist-suffix --enable-suffix \

examples/simple/queens.py >result.log 2>&1

check_return_value

check_file_exists dist-suffix/pytransform${test_suffix}/__init__.py

check_file_exists dist-suffix/queens.py

check_file_content dist-suffix/queens.py "from pytransform${test_suffix} import"

check_file_content dist-suffix/queens.py "pyarmor_runtime(suffix="

(cd dist-suffix; $PYTHON queens.py >result.log 2>&1)

check_return_value

check_file_content dist-suffix/result.log 'Found 92 solutions'

csih_inform "13. Generate runtime package with option --enable-suffix"

$PYARMOR runtime -O test-runtime-suffix --enable-suffix >result.log 2>&1

check_return_value

check_file_exists test-runtime-suffix/pytransform${test_suffix}/__init__.py

check_file_content test-runtime-suffix/pytransform_bootstrap.py \

"from pytransform${test_suffix} import pyarmor_runtime"

check_file_content test-runtime-suffix/pytransform_bootstrap.py \

"pyarmor_runtime(suffix="

$PYARMOR runtime -O test-runtime-suffix-2 --no-package --enable-suffix >result.log 2>&1

check_return_value

check_file_exists test-runtime-suffix-2/pytransform${test_suffix}.py

check_file_content test-runtime-suffix-2/pytransform_bootstrap.py \

"from pytransform${test_suffix} import pyarmor_runtime"

check_file_content test-runtime-suffix-2/pytransform_bootstrap.py \

"pyarmor_runtime(suffix="

csih_inform "14. Build project with option --enable-suffix"

PROPATH=projects/test_enable_suffix

[[ -d "$PROPATH" ]] && rm -rf $PROPATH

mkdir $PROPATH

echo "print('Hello this is a project with suffix')" > $PROPATH/main.py

$PYARMOR init --src=$PROPATH --entry=main.py $PROPATH >result.log 2>&1

$PYARMOR config --enable-suffix 1 $PROPATH >result.log 2>&1

check_return_value

$PYARMOR build $PROPATH >$PROPATH/result.log 2>&1

check_return_value

check_file_exists $PROPATH/dist/pytransform${test_suffix}/__init__.py

check_file_content $PROPATH/dist/main.py \

"from pytransform${test_suffix} import pyarmor_runtime"

check_file_content $PROPATH/dist/main.py \

"pyarmor_runtime(suffix="

(cd $PROPATH/dist; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/dist/result.log 'Hello this is a project with suffix'

csih_inform "15. Import the package obfuscated by others"

PROPATH=test-import-other-package

$PYARMOR obfuscate -O $PROPATH/dist --enable-suffix --exact $PROPATH/main.py >result.log 2>&1

(cd $PROPATH/dist; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/dist/result.log 'This is package 1'

check_file_content $PROPATH/dist/result.log 'This is package 2'

check_file_content $PROPATH/dist/result.log 'This is main script'

csih_inform "16. Import the package obfuscated by others with advanced mode"

$PYARMOR obfuscate -O $PROPATH/dist --enable-suffix --advanced 1 --exact \

$PROPATH/main.py >result.log 2>&1

(cd $PROPATH/dist; $PYTHON main.py >result.log 2>&1)

check_return_value

check_file_content $PROPATH/dist/result.log 'This is package 1'

check_file_content $PROPATH/dist/result.log 'This is package 2'

check_file_content $PROPATH/dist/result.log 'This is main script'

csih_inform "17. Obfuscate big script"

ascript="big_script.py"

$PYTHON -c"

with open('$ascript', 'w') as f:

for i in range(200):

f.write('def test_{0}(n):\n'.format(i))

f.write(' return n + {0}\n'.format(i))"

$PYARMOR obfuscate --exact -O dist-big-script $ascript >result.log 2>&1

check_return_value

check_file_content result.log 'Check license failed' not

check_file_content result.log 'Too big code object, the limitation is' not

csih_inform "18. Get issuer of the obfuscate scripts"

(cd dist;

$PYTHON -c"

from pytransform import pyarmor_init, get_license_info

pyarmor_init(is_runtime=1)

licinfo = get_license_info()

print('License code is %s' % licinfo['CODE'])

print('Issuer is %s' % licinfo['ISSUER'])" > result.log 2>&1)

check_return_value

check_file_content dist/result.log 'License code is Joker'

check_file_content dist/result.log 'Issuer is pyarmor-test-0001'

csih_inform "19. Protect data file"

dist=test-data-file

$PYTHON -m helper.build_data_module protect_code2.pt > protect_data.py

cat <<EOF > safedata.py

import protect_data

with protect_data.Safestr() as text:

print('Got data: %s' % text[:32])

from pytransform import clean_str

from sys import version_info as ver

data = ('a' * 30) if ver[0] == 2 else (b'a' * 30).decode()

clean_str(data)

if (ver[0] * 10 + ver[1]) not in (30, 31, 32):

print(data)

EOF

$PYARMOR obfuscate --exact -O $dist safedata.py >result.log 2>&1

$PYARMOR obfuscate --exact -O $dist --no-runtime --no-bootstrap \

--restrict 4 protect_data.py >result.log 2>&1

(cd $dist; $PYTHON safedata.py >result.log 2>&1)

check_return_value

check_file_content $dist/result.log 'Got data: def protect_pytransform'

check_file_content $dist/result.log 'aaaaaaaaaa' not

csih_inform "20. Obfuscate scripts with --runtime"

dist=test-with-runtime

$PYARMOR obfuscate --runtime test-runtime-suffix -O $dist \

examples/simple/queens.py >result.log 2>&1

(cd $dist; $PYTHON queens.py >result.log 2>&1)

check_return_value

check_file_exists $dist/pytransform${test_suffix}/__init__.py

check_file_content $dist/result.log 'Found 92 solutions'

# ======================================================================

#

# Finished and cleanup.

#

# ======================================================================

echo "" && csih_inform "Clean pyarmor data"

rm -rf ~/.pyarmor_capsule.zip ~/.pyarmor

[[ -n "$USERPROFILE" ]] && rm -rf "$USERPROFILE\\.pyarmor*"

# Return test root

cd ../..

echo "----------------------------------------------------------------"

echo ""

csih_inform "Test finished for ${PYTHON}"

(( ${_bug_counter} == 0 )) || csih_error "${_bug_counter} bugs found"

echo "" \

&& csih_inform "Remove workpath ${workpath}" \

&& echo "" \

&& rm -rf ${workpath} \

&& csih_inform "Congratulations, there is no bug found"