def _obfuscate(args):
'''Obfuscate scripts without project'''
path = args.src
logging.info('Obfuscate scripts in path "%s" ...', path)
capsule = os.path.join(path, capsule_filename)
if not os.path.exists(capsule):
logging.info('Generate capsule %s', capsule)
make_capsule(capsule)
output = args.output
files = Project.build_globfiles(args.patterns, path)
filepairs = [(os.path.join(path, x), os.path.join(output, x))
for x in files]
mode = Project.map_obfuscate_mode(default_obf_module_mode,
default_obf_code_mode)
logging.info('Obfuscate scripts with mode %s', mode)
logging.info('Save obfuscated scripts to "%s"', output)
for a, b in filepairs:
logging.info('\t%s -> %s', a, b)
obfuscate_scripts(filepairs, mode, capsule, output)
logging.info('Make runtime files')
make_runtime(capsule, output)
for entry in args.entry.split(','):
filename = os.path.join(output, entry.strip())
if not os.path.exists(filename):
shutil.copy(os.path.join(path, entry.strip()), filename)
logging.info('Update entry script %s', filename)
make_entry(filename)
logging.info('Obfuscate %d scripts OK.', len(files))
def _runtime(args):
'''Generate runtime package separately.'''
capsule = DEFAULT_CAPSULE
output = args.output
package = not args.no_package
platform = args.platform
make_runtime(capsule, output, licfile=args.with_license, platform=platform,
package=package)
def _obfuscate(args):
'''Obfuscate scripts without project'''
if args.src is None and args.entry is None and not args.scripts:
raise RuntimeError('No entry script')
entry = args.entry or args.scripts[0]
path = os.path.abspath(
os.path.dirname(entry) if args.src is None else args.src)
logging.info('Obfuscate scripts in path "%s" ...', path)
capsule = args.capsule if args.capsule else DEFAULT_CAPSULE
if os.path.exists(capsule) and check_capsule(capsule):
logging.info('Use cached capsule %s', capsule)
else:
logging.info('Generate capsule %s', capsule)
make_capsule(capsule)
output = args.output
if args.recursive:
pats = ['global-include *.py', 'prune build', 'prune dist']
if hasattr('', 'decode'):
pats = [p.decode() for p in pats]
files = Project.build_manifest(pats, path)
else:
files = Project.build_globfiles(['*.py'], path)
filepairs = [(os.path.join(path, x), os.path.join(output, x))
for x in files]
if args.restrict:
logging.info('Restrict mode is eanbled')
mode = Project.map_obfuscate_mode(default_obf_module_mode,
default_obf_code_mode)
else:
logging.info('Restrict mode is disabled')
mode = Project.map_obfuscate_mode(default_obf_module_mode, 'wrap')
logging.info('Obfuscate scripts with mode %s', mode)
logging.info('Save obfuscated scripts to "%s"', output)
for a, b in filepairs:
logging.info('\t%s -> %s', a, b)
obfuscate_scripts(filepairs, mode, capsule, output)
logging.info('Make runtime files')
make_runtime(capsule, output)
if not args.restrict:
licode = '*FLAGS:%c*CODE:Pyarmor-Project' % chr(1)
licfile = os.path.join(output, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
make_entry(os.path.basename(entry), path, output)
for script in args.scripts[1:]:
make_entry(os.path.basename(script), path, output)
logging.info('Obfuscate %d scripts OK.', len(files))
def _runtime(args):
'''Generate runtime package separately.'''
capsule = DEFAULT_CAPSULE
output = args.output
package = not args.no_package
platforms = compatible_platform_names(args.platforms)
make_runtime(capsule,
output,
licfile=args.with_license,
platforms=platforms,
package=package)
def _build(args):
'''Build project, obfuscate all files in the project'''
project = Project()
project.open(args.project)
logging.info('Build project %s ...', args.project)
capsule = build_path(project.capsule, args.project)
if not args.only_runtime:
output = project.output
mode = project.get_obfuscate_mode()
files = project.get_build_files(args.force)
src = project.src
filepairs = [(os.path.join(src, x), os.path.join(output, x))
for x in files]
logging.info('%s increment build',
'Disable' if args.force else 'Enable')
logging.info('Search scripts from %s', src)
logging.info('Obfuscate %d scripts with mode %s', len(files), mode)
for x in files:
logging.info('\t%s', x)
logging.info('Save obfuscated scripts to %s', output)
obfuscate_scripts(filepairs, mode, capsule, output)
# for x in targets:
# output = os.path.join(project.output, x)
# pairs = [(os.path.join(src, x), os.path.join(output, x))
# for x in files]
# for src, dst in pairs:
# try:
# shutil.copy2(src, dst)
# except Exception:
# os.makedirs(os.path.dirname(dst))
# shutil.copy2(src, dst)
project['build_time'] = time.time()
project.save(args.project)
if not args.no_runtime:
logging.info('Make runtime files')
make_runtime(capsule, output)
if project.entry:
for x in project.entry.split(','):
filename = os.path.join(output, x.strip())
logging.info('Update entry script %s', filename)
make_entry(filename, project.runtime_path)
else:
logging.info('\tIn order to import obfuscated scripts, insert ')
logging.info('\t2 lines in entry script:')
logging.info('\t\tfrom pytransfrom import pyarmor_runtime')
logging.info('\t\tpyarmor_runtime()')
logging.info('Build project OK.')
def _runtime(args):
'''Generate runtime package separately.'''
capsule = DEFAULT_CAPSULE
name = 'pytransform_bootstrap'
output = os.path.join(args.output, name) if args.inside else args.output
package = not args.no_package
platforms = compatible_platform_names(args.platforms)
make_runtime(capsule, output, licfile=args.with_license,
platforms=platforms, package=package)
filename = os.path.join(output, '__init__.py') if args.inside else \
os.path.join(args.output, name + '.py')
logging.info('Generating bootstrap script ...')
make_bootstrap_script(filename, capsule=capsule)
logging.info('Generating bootstrap script %s OK', filename)
def _obfuscate(args):
'''Obfuscate scripts without project'''
path = args.src
logging.info('Obfuscate scripts in path "%s" ...', path)
capsule = args.capsule if args.capsule else capsule_filename
if os.path.exists(capsule):
logging.info('Use cached capsule %s', capsule)
else:
logging.info('Generate capsule %s', capsule)
make_capsule(capsule)
output = args.output
if args.recursive:
pat = 'global-include *.py'.decode() if hasattr('', 'decode') \
else 'global-include *.py'
files = Project.build_manifest([pat], path)
else:
files = Project.build_globfiles(args.patterns, path)
filepairs = [(os.path.join(path, x), os.path.join(output, x))
for x in files]
if args.no_restrict:
logging.info('Restrict mode is disabled')
mode = Project.map_obfuscate_mode(default_obf_module_mode, 'wrap')
else:
mode = Project.map_obfuscate_mode(default_obf_module_mode,
default_obf_code_mode)
logging.info('Obfuscate scripts with mode %s', mode)
logging.info('Save obfuscated scripts to "%s"', output)
for a, b in filepairs:
logging.info('\t%s -> %s', a, b)
obfuscate_scripts(filepairs, mode, capsule, output)
logging.info('Make runtime files')
make_runtime(capsule, output)
if args.no_restrict:
licode = '*FLAGS:%c*CODE:Pyarmor-Project' % chr(1)
licfile = os.path.join(output, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
if args.entry:
make_entry(args.entry, path, output)
logging.info('Obfuscate %d scripts OK.', len(files))
def _obfuscate(args):
'''Obfuscate scripts without project.'''
check_cross_platform(args.platform)
for x in ('entry', 'cross-protection'):
if getattr(args, x.replace('-', '_')) is not None:
logging.warning('Option --%s has been deprecated', x)
if args.src is None and not args.scripts:
args.src = '.'
if args.src is None:
if args.scripts[0].lower().endswith('.py'):
path = os.path.abspath(os.path.dirname(args.scripts[0]))
else:
path = os.path.abspath(args.scripts[0])
args.src = path
if len(args.scripts) > 1:
raise RuntimeError('Only one path is allowed')
args.scripts = []
else:
path = os.path.abspath(args.src)
if not os.path.exists(path):
raise RuntimeError('Not found source path: %s' % path)
logging.info('Source path is "%s"', path)
entry = args.entry or (args.scripts and args.scripts[0])
logging.info('Entry script is %s', entry)
capsule = args.capsule if args.capsule else DEFAULT_CAPSULE
if os.path.exists(capsule):
logging.info('Use cached capsule %s', capsule)
else:
logging.info('Generate capsule %s', capsule)
make_capsule(capsule)
output = args.output
if os.path.abspath(output) == path:
raise RuntimeError('Output path can not be same as src')
if args.recursive:
logging.info('Recursive mode is on')
pats = ['global-include *.py']
if args.exclude:
for item in args.exclude:
for x in item.split(','):
logging.info('Exclude path "%s"', x)
pats.append('prune %s' % x)
if os.path.abspath(output).startswith(path):
x = os.path.abspath(output)[len(path):].strip('/\\')
pats.append('prune %s' % x)
logging.info('Auto exclude output path "%s"', x)
if hasattr('', 'decode'):
pats = [p.decode() for p in pats]
files = Project.build_manifest(pats, path)
elif args.exact:
logging.info('Exact mode is on')
files = [os.path.abspath(x) for x in args.scripts]
else:
logging.info('Normal mode is on')
files = Project.build_globfiles(['*.py'], path)
logging.info('Save obfuscated scripts to "%s"', output)
if not os.path.exists(output):
os.makedirs(output)
logging.info('Read public key from capsule')
prokey = get_product_key(capsule)
logging.info('Obfuscate scripts with default mode')
cross_protection = 0 if args.no_cross_protection else \
1 if args.cross_protection is None else args.cross_protection
if args.platform:
logging.info('Target platform is %s', args.platform)
if cross_protection == 1:
cross_protection = args.platform
elif isinstance(cross_protection, str):
cross_protection = ','.join([cross_protection, args.platform])
advanced = 1 if args.advanced else 0
logging.info('Advanced mode is %d', advanced)
restrict = args.restrict
logging.info('Restrict mode is %d', restrict)
for x in files:
if os.path.isabs(x):
a, b = x, os.path.join(output, os.path.basename(x))
else:
a, b = os.path.join(path, x), os.path.join(output, x)
logging.info('\t%s -> %s', x, b)
is_entry = entry and (os.path.abspath(a) == os.path.abspath(entry))
protection = is_entry and cross_protection
plugins = protection and args.plugins
d = os.path.dirname(b)
if not os.path.exists(d):
os.makedirs(d)
vmode = advanced | (8 if is_entry else 0)
encrypt_script(prokey, a, b, adv_mode=vmode, rest_mode=restrict,
protection=protection, plugins=plugins)
logging.info('%d scripts have been obfuscated', len(files))
if (not args.no_bootstrap) and entry and os.path.exists(entry):
inner = args.package_runtime != 2
entryname = entry if args.src else os.path.basename(entry)
if os.path.exists(os.path.join(output, entryname)):
make_entry(entryname, path, output, inner=inner)
else:
logging.info('Use outer entry script "%s"', entry)
make_entry(entry, path, output, inner=inner)
if args.no_runtime:
logging.info('Obfuscate %d scripts OK.', len(files))
return
make_runtime(capsule, output, platform=args.platform,
package=args.package_runtime)
logging.info('Obfuscate scripts with restrict mode %s',
'on' if args.restrict else 'off')
if not args.restrict:
licode = '*FLAGS:%c*CODE:PyArmor-Project' % chr(1)
licpath = os.path.join(output, 'pytransform') if args.package_runtime \
else output
licfile = os.path.join(licpath, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
logging.info('Obfuscate %d scripts OK.', len(files))
def _build(args):
'''Build project, obfuscate all scripts in the project.'''
project = Project()
project.open(args.project)
logging.info('Build project %s ...', args.project)
logging.info('Check project')
project._check(args.project)
pro_path = args.project \
if args.project == '' or os.path.exists(args.project) \
else os.path.dirname(args.project)
capsule = build_path(project.capsule, pro_path)
logging.info('Use capsule: %s', capsule)
output = build_path(project.output, pro_path) \
if args.output is None else os.path.normpath(args.output)
logging.info('Output path is: %s', output)
platform = args.platform if args.platform else project.get('platform')
if platform:
logging.info('Taget platform is: %s', platform)
check_cross_platform(platform)
restrict = project.get('restrict_mode',
0 if project.get('disable_restrict_mode') else 1)
if not args.only_runtime:
src = project.src
if os.path.abspath(output).startswith(src):
excludes = ['prune %s' % os.path.abspath(output)[len(src)+1:]]
else:
excludes = []
files = project.get_build_files(args.force, excludes=excludes)
soutput = os.path.join(output, os.path.basename(src)) \
if project.get('is_package') else output
logging.info('Save obfuscated scripts to "%s"', soutput)
if not os.path.exists(soutput):
os.makedirs(soutput)
logging.info('Read public key from capsule')
prokey = get_product_key(capsule)
logging.info('%s increment build',
'Disable' if args.force else 'Enable')
logging.info('Search scripts from %s', src)
logging.info('Obfuscate scripts with mode:')
if hasattr(project, 'obf_mod'):
obf_mod = project.obf_mod
else:
obf_mod = project.obf_module_mode == 'des'
if hasattr(project, 'wrap_mode'):
wrap_mode = project.wrap_mode
obf_code = project.obf_code
elif project.obf_code_mode == 'wrap':
wrap_mode = 1
obf_code = 1
else:
wrap_mode = 0
obf_code = 0 if project.obf_code_mode == 'none' else 1
adv_mode = (1 if project.advanced_mode else 0) \
if hasattr(project, 'advanced_mode') else 0
def v(t):
return 'on' if t else 'off'
logging.info('Obfuscating the whole module is %s', v(obf_mod))
logging.info('Obfuscating each function is %s', v(obf_code))
logging.info('Autowrap each code object mode is %s', v(wrap_mode))
logging.info('Advanced mode is %s', v(adv_mode))
logging.info('Restrict mode is %s', restrict)
entries = [build_path(s.strip(), project.src)
for s in project.entry.split(',')] if project.entry else []
protection = project.cross_protection \
if hasattr(project, 'cross_protection') else 1
if platform:
if protection == 1:
protection = platform
elif not isinstance(protection, int):
protection = ','.join([protection, platform])
for x in files:
a, b = os.path.join(src, x), os.path.join(soutput, x)
logging.info('\t%s -> %s', x, b)
d = os.path.dirname(b)
if not os.path.exists(d):
os.makedirs(d)
if entries and (os.path.abspath(a) in entries):
vmode = adv_mode | 8
pcode = protection
if hasattr(project, 'plugins'):
plugins = project.plugins
else:
vmode = adv_mode
pcode = 0
plugins = None
encrypt_script(prokey, a, b, obf_code=obf_code, obf_mod=obf_mod,
wrap_mode=wrap_mode, adv_mode=vmode,
rest_mode=restrict, protection=pcode,
plugins=plugins, rpath=project.runtime_path)
logging.info('%d scripts has been obfuscated', len(files))
project['build_time'] = time.time()
project.save(args.project)
if project.entry:
soutput = os.path.join(output, os.path.basename(project.src)) \
if project.get('is_package') else output
package_runtime = project.get('package_runtime', 0) \
if args.package_runtime is None else args.package_runtime
make_entry(project.entry, project.src, soutput,
rpath=project.runtime_path,
inner=(package_runtime != 2) and (not args.no_runtime))
if not args.no_runtime:
routput = output if args.output is not None and args.only_runtime \
else os.path.join(output, os.path.basename(project.src)) \
if project.get('is_package') else output
if not os.path.exists(routput):
logging.info('Make path: %s', routput)
os.mkdir(routput)
package = project.get('package_runtime', 0) \
if args.package_runtime is None else args.package_runtime
make_runtime(capsule, routput, platform=platform, package=package)
if not restrict:
licode = '*FLAGS:%c*CODE:PyArmor-Project' % chr(1)
licpath = os.path.join(routput, 'pytransform') if package \
else routput
licfile = os.path.join(licpath, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
logging.info('Build project OK.')
def _build(args):
'''Build project, obfuscate all scripts in the project.'''
project = Project()
project.open(args.project)
logging.info('Build project %s ...', args.project)
capsule = build_path(project.capsule, args.project)
logging.info('Use capsule: %s', capsule)
output = build_path(project.output, args.project) \
if args.output is None else args.output
logging.info('Output path is: %s', output)
if not args.only_runtime:
mode = project.get_obfuscate_mode()
files = project.get_build_files(args.force)
src = project.src
soutput = os.path.join(output, os.path.basename(src)) \
if project.get('is_package') else output
filepairs = [(os.path.join(src, x), os.path.join(soutput, x))
for x in files]
logging.info('%s increment build',
'Disable' if args.force else 'Enable')
logging.info('Search scripts from %s', src)
logging.info('Obfuscate %d scripts with mode %s', len(files), mode)
for x in files:
logging.info('\t%s', x)
logging.info('Save obfuscated scripts to %s', soutput)
obfuscate_scripts(filepairs, mode, capsule, soutput)
# for x in targets:
# output = os.path.join(project.output, x)
# pairs = [(os.path.join(src, x), os.path.join(output, x))
# for x in files]
# for src, dst in pairs:
# try:
# shutil.copy2(src, dst)
# except Exception:
# os.makedirs(os.path.dirname(dst))
# shutil.copy2(src, dst)
project['build_time'] = time.time()
project.save(args.project)
if project.entry:
make_entry(project.entry,
project.src,
output,
rpath=project.runtime_path,
ispackage=project.get('is_package'))
if not args.no_runtime:
routput = os.path.join(output, os.path.basename(project.src)) \
if project.get('is_package') else output
if not os.path.exists(routput):
logging.info('Make path: %s', routput)
os.mkdir(routput)
logging.info('Make runtime files to %s', routput)
make_runtime(capsule, routput)
if project.get('disable_restrict_mode'):
licode = '*FLAGS:%c*CODE:Pyarmor-Project' % chr(1)
licfile = os.path.join(routput, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
else:
logging.info('\tIn order to import obfuscated scripts, insert ')
logging.info('\t2 lines in entry script:')
logging.info('\t\tfrom pytransfrom import pyarmor_runtime')
logging.info('\t\tpyarmor_runtime()')
logging.info('Build project OK.')
def _obfuscate(args):
'''Obfuscate scripts without project'''
if args.src is None and args.entry is None and not args.scripts:
raise RuntimeError('No entry script')
entry = args.entry or (args.scripts and args.scripts[0])
path = os.path.abspath(
os.path.dirname(entry) if args.src is None else args.src)
logging.info('Obfuscate scripts in path "%s"', path)
capsule = args.capsule if args.capsule else DEFAULT_CAPSULE
if os.path.exists(capsule) and check_capsule(capsule):
logging.info('Use cached capsule %s', capsule)
else:
logging.info('Generate capsule %s', capsule)
make_capsule(capsule)
output = args.output
if os.path.abspath(output) == path:
raise RuntimeError('Output path can not be same as src')
if args.recursive:
pats = ['global-include *.py', 'prune build', 'prune dist']
if os.path.abspath(output).startswith(path):
x = os.path.abspath(output)[len(path):].strip('/\\')
pats.append('prune %s' % x)
if hasattr('', 'decode'):
pats = [p.decode() for p in pats]
files = Project.build_manifest(pats, path)
else:
files = Project.build_globfiles(['*.py'], path)
logging.info('Save obfuscated scripts to "%s"', output)
if not os.path.exists(output):
os.makedirs(output)
logging.info('Read public key from capsule')
prokey = get_product_key(capsule)
logging.info('Obfuscate scripts with default mode')
for x in files:
a, b = os.path.join(path, x), os.path.join(output, x)
logging.info('\t%s -> %s', x, b)
protection = args.cross_protection and entry \
and (os.path.abspath(a) == os.path.abspath(entry))
d = os.path.dirname(b)
if not os.path.exists(d):
os.makedirs(d)
encrypt_script(prokey, a, b, protection=protection)
logging.info('%d scripts have been obfuscated', len(files))
make_runtime(capsule, output)
if entry and entry.endswith('__init__.py') and args.restrict is None:
logging.info('Disable restrict mode for package by default')
restrict = 0
else:
restrict = 1 if args.restrict is None else args.restrict
logging.info('Obfuscate scripts with restrict mode %s',
'on' if restrict else 'off')
if not restrict:
licode = '*FLAGS:%c*CODE:PyArmor-Project' % chr(1)
licfile = os.path.join(output, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
if entry:
make_entry(os.path.basename(entry), path, output)
for script in args.scripts[1:]:
make_entry(os.path.basename(script), path, output)
logging.info('Obfuscate %d scripts OK.', len(files))
def _build(args):
'''Build project, obfuscate all scripts in the project.'''
project = Project()
project.open(args.project)
logging.info('Build project %s ...', args.project)
capsule = build_path(project.capsule, args.project)
logging.info('Use capsule: %s', capsule)
output = build_path(project.output, args.project) \
if args.output is None else args.output
logging.info('Output path is: %s', output)
if not args.only_runtime:
files = project.get_build_files(args.force)
src = project.src
soutput = os.path.join(output, os.path.basename(src)) \
if project.get('is_package') else output
logging.info('Save obfuscated scripts to "%s"', soutput)
if not os.path.exists(soutput):
os.makedirs(soutput)
logging.info('Read public key from capsule')
prokey = get_product_key(capsule)
logging.info('%s increment build',
'Disable' if args.force else 'Enable')
logging.info('Search scripts from %s', src)
logging.info('Obfuscate scripts with mode:')
if hasattr(project, 'obf_mod'):
obf_mod = project.obf_mod
else:
obf_mod = project.obf_module_mode == 'des'
if hasattr(project, 'wrap_mode'):
wrap_mode = project.wrap_mode
obf_code = project.obf_code
elif project.obf_code_mode == 'wrap':
wrap_mode = 1
obf_code = 1
else:
wrap_mode = 0
obf_code = 0 if project.obf_code_mode == 'none' else 1
def v(t):
return 'on' if t else 'off'
logging.info('Obfuscating the whole module is %s', v(obf_mod))
logging.info('Obfuscating each function is %s', v(obf_code))
logging.info('Autowrap each code object mode is %s', v(wrap_mode))
entry = os.path.abspath(project.entry) if project.entry else None
protection = project.cross_protection \
if hasattr(project, 'cross_protection') else 1
for x in files:
a, b = os.path.join(src, x), os.path.join(soutput, x)
logging.info('\t%s -> %s', x, b)
d = os.path.dirname(b)
if not os.path.exists(d):
os.makedirs(d)
pcode = entry and (os.path.abspath(a) == entry) and protection
encrypt_script(prokey,
a,
b,
obf_code=obf_code,
obf_mod=obf_mod,
wrap_mode=wrap_mode,
protection=pcode,
rpath=project.runtime_path)
logging.info('%d scripts has been obfuscated', len(files))
project['build_time'] = time.time()
project.save(args.project)
if project.entry:
make_entry(project.entry,
project.src,
output,
rpath=project.runtime_path,
ispackage=project.get('is_package'))
if not args.no_runtime:
routput = os.path.join(output, os.path.basename(project.src)) \
if project.get('is_package') else output
if not os.path.exists(routput):
logging.info('Make path: %s', routput)
os.mkdir(routput)
logging.info('Make runtime files to %s', routput)
make_runtime(capsule, routput)
if project.get('disable_restrict_mode'):
licode = '*FLAGS:%c*CODE:PyArmor-Project' % chr(1)
licfile = os.path.join(routput, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_project_license(capsule, licode, licfile)
else:
logging.info('\tIn order to import obfuscated scripts, insert ')
logging.info('\t2 lines in entry script:')
logging.info('\t\tfrom pytransform import pyarmor_runtime')
logging.info('\t\tpyarmor_runtime()')
logging.info('Build project OK.')
def _obfuscate(args):
'''Obfuscate scripts without project.'''
platforms = compatible_platform_names(args.platforms)
if platforms:
logging.info('Target platforms: %s', platforms)
if check_cross_platform(platforms) is not False:
return
for x in ('entry', 'cross-protection'):
if getattr(args, x.replace('-', '_')) is not None:
logging.warning('Option --%s has been deprecated', x)
if args.src is None:
if args.scripts[0].lower().endswith('.py'):
path = os.path.abspath(os.path.dirname(args.scripts[0]))
else:
path = os.path.abspath(args.scripts[0])
args.src = path
if len(args.scripts) > 1:
raise RuntimeError('Only one path is allowed')
args.scripts = []
else:
for s in args.scripts:
if not s.lower().endswith('.py'):
raise RuntimeError('Only one path is allowed')
path = os.path.abspath(args.src)
if not args.exact and len(args.scripts) > 1:
raise RuntimeError('Two many entry scripts, only one is allowed')
if not os.path.exists(path):
raise RuntimeError('Not found source path: %s' % path)
logging.info('Source path is "%s"', path)
entries = [args.entry] if args.entry else args.scripts
logging.info('Entry scripts are %s', entries)
capsule = args.capsule if args.capsule else DEFAULT_CAPSULE
if os.path.exists(capsule):
logging.info('Use cached capsule %s', capsule)
else:
logging.info('Generate capsule %s', capsule)
make_capsule(capsule)
output = args.output
if os.path.abspath(output) == path:
raise RuntimeError('Output path can not be same as src')
suffix = get_name_suffix() if args.enable_suffix else ''
if args.recursive:
logging.info('Recursive mode is on')
pats = ['global-include *.py']
if args.exclude:
for item in args.exclude:
for x in item.split(','):
if x.endswith('.py'):
logging.info('Exclude pattern "%s"', x)
pats.append('exclude %s' % x)
else:
logging.info('Exclude path "%s"', x)
pats.append('prune %s' % x)
if os.path.abspath(output).startswith(path):
x = os.path.abspath(output)[len(path):].strip('/\\')
pats.append('prune %s' % x)
logging.info('Auto exclude output path "%s"', x)
if hasattr('', 'decode'):
pats = [p.decode() for p in pats]
files = Project.build_manifest(pats, path)
elif args.exact:
logging.info('Exact mode is on')
files = [os.path.abspath(x) for x in args.scripts]
else:
logging.info('Normal mode is on')
files = Project.build_globfiles(['*.py'], path)
logging.info('Save obfuscated scripts to "%s"', output)
if not os.path.exists(output):
os.makedirs(output)
logging.info('Read public key from capsule')
prokey = get_product_key(capsule)
logging.info('Obfuscate scripts with default mode')
cross_protection = 0 if args.no_cross_protection else \
1 if args.cross_protection is None else args.cross_protection
advanced = 1 if args.advanced else 0
logging.info('Advanced mode is %d', advanced)
restrict = args.restrict
logging.info('Restrict mode is %d', restrict)
n = args.bootstrap_code
relative = True if n == 3 else False if n == 2 else None
bootstrap = (not args.no_bootstrap) and n
elist = [os.path.abspath(x) for x in entries]
for x in sorted(files):
if os.path.isabs(x):
a, b = x, os.path.join(output, os.path.basename(x))
else:
a, b = os.path.join(path, x), os.path.join(output, x)
logging.info('\t%s -> %s', x, relpath(b))
is_entry = os.path.abspath(a) in elist
protection = is_entry and cross_protection
plugins = search_plugins(args.plugins)
d = os.path.dirname(b)
if not os.path.exists(d):
os.makedirs(d)
vmode = advanced | (8 if is_entry else 0)
encrypt_script(prokey, a, b, adv_mode=vmode, rest_mode=restrict,
protection=protection, platforms=platforms,
plugins=plugins, suffix=suffix)
if is_entry and bootstrap:
name = os.path.abspath(a)[len(path)+1:]
make_entry(name, path, output, relative=relative, suffix=suffix)
logging.info('%d scripts have been obfuscated', len(files))
if args.no_runtime:
logging.info('Obfuscate %d scripts OK.', len(files))
return
package = args.package_runtime
make_runtime(capsule, output, platforms=platforms,
package=package, suffix=suffix)
logging.info('Obfuscate scripts with restrict mode %s',
'on' if args.restrict else 'off')
if not args.restrict:
licode = '*FLAGS:%c*CODE:PyArmor-Project' % chr(1)
licpath = (os.path.join(output, 'pytransform' + suffix) if package
else output)
licfile = os.path.join(licpath, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_license_key(capsule, licode, licfile)
logging.info('Obfuscate %d scripts OK.', len(files))
def _build(args):
'''Build project, obfuscate all scripts in the project.'''
project = Project()
project.open(args.project)
logging.info('Build project %s ...', args.project)
logging.info('Check project')
project.check()
suffix = get_name_suffix() if project.get('enable_suffix', 0) else ''
capsule = project.get('capsule', DEFAULT_CAPSULE)
logging.info('Use capsule: %s', capsule)
output = project.output if args.output is None \
else os.path.normpath(args.output)
logging.info('Output path is: %s', output)
if args.platforms:
platforms = [] if '' in args.platforms else args.platforms
elif project.get('platform'):
platforms = project.get('platform').split(',')
else:
platforms = []
if platforms:
platforms = compatible_platform_names(platforms)
logging.info('Taget platforms: %s', platforms)
if check_cross_platform(platforms) is not False:
return
restrict = project.get('restrict_mode',
0 if project.get('disable_restrict_mode') else 1)
if not args.only_runtime:
src = project.src
if os.path.abspath(output).startswith(src):
excludes = ['prune %s' % os.path.abspath(output)[len(src)+1:]]
else:
excludes = []
files = project.get_build_files(args.force, excludes=excludes)
soutput = os.path.join(output, os.path.basename(src)) \
if project.get('is_package') else output
logging.info('Save obfuscated scripts to "%s"', soutput)
if not os.path.exists(soutput):
os.makedirs(soutput)
logging.info('Read public key from capsule')
prokey = get_product_key(capsule)
logging.info('%s increment build',
'Disable' if args.force else 'Enable')
logging.info('Search scripts from %s', src)
logging.info('Obfuscate scripts with mode:')
if hasattr(project, 'obf_mod'):
obf_mod = project.obf_mod
else:
obf_mod = project.obf_module_mode == 'des'
if hasattr(project, 'wrap_mode'):
wrap_mode = project.wrap_mode
obf_code = project.obf_code
elif project.obf_code_mode == 'wrap':
wrap_mode = 1
obf_code = 1
else:
wrap_mode = 0
obf_code = 0 if project.obf_code_mode == 'none' else 1
adv_mode = (1 if project.advanced_mode else 0) \
if hasattr(project, 'advanced_mode') else 0
def v(t):
return 'on' if t else 'off'
logging.info('Obfuscating the whole module is %s', v(obf_mod))
logging.info('Obfuscating each function is %s', v(obf_code))
logging.info('Autowrap each code object mode is %s', v(wrap_mode))
logging.info('Advanced mode is %s', v(adv_mode))
logging.info('Restrict mode is %s', restrict)
entries = [build_path(s.strip(), project.src)
for s in project.entry.split(',')] if project.entry else []
protection = project.cross_protection \
if hasattr(project, 'cross_protection') else 1
for x in sorted(files):
a, b = os.path.join(src, x), os.path.join(soutput, x)
logging.info('\t%s -> %s', x, relpath(b))
d = os.path.dirname(b)
if not os.path.exists(d):
os.makedirs(d)
if hasattr(project, 'plugins'):
plugins = search_plugins(project.plugins)
else:
plugins = None
if entries and (os.path.abspath(a) in entries):
vmode = adv_mode | 8
pcode = protection
else:
vmode = adv_mode
pcode = 0
encrypt_script(prokey, a, b, obf_code=obf_code, obf_mod=obf_mod,
wrap_mode=wrap_mode, adv_mode=vmode,
rest_mode=restrict, protection=pcode,
platforms=platforms, plugins=plugins,
rpath=project.runtime_path, suffix=suffix)
logging.info('%d scripts has been obfuscated', len(files))
project['build_time'] = time.time()
project.save(args.project)
if project.entry and project.get('bootstrap_code', 1):
soutput = os.path.join(output, os.path.basename(project.src)) \
if project.get('is_package') else output
n = project.get('bootstrap_code', 1)
relative = True if n == 3 else \
False if (n == 2 or (args.no_runtime and n == 1)) else None
make_entry(project.entry, project.src, soutput,
rpath=project.runtime_path, relative=relative,
suffix=suffix)
if not args.no_runtime:
routput = output if args.output is not None and args.only_runtime \
else os.path.join(output, os.path.basename(project.src)) \
if project.get('is_package') else output
if not os.path.exists(routput):
logging.info('Make path: %s', routput)
os.mkdir(routput)
package = project.get('package_runtime', 0) \
if args.package_runtime is None else args.package_runtime
make_runtime(capsule, routput, platforms=platforms, package=package,
suffix=suffix)
licfile = project.license_file
if licfile:
logging.info('Project has customized license file: %s', licfile)
licpath = os.path.join(routput, 'pytransform' + suffix) \
if package else routput
logging.info('Copy project license file to %s', licpath)
shutil.copy(licfile, licpath)
elif not restrict:
licode = '*FLAGS:%c*CODE:PyArmor-Project' % chr(1)
licpath = os.path.join(routput, 'pytransform' + suffix) \
if package else routput
licfile = os.path.join(licpath, license_filename)
logging.info('Generate no restrict mode license file: %s', licfile)
make_license_key(capsule, licode, licfile)
logging.info('Build project OK.')
