pyarmor 8.5.3__zip → 8.5.5__zip

{pyarmor-8.5.3 → pyarmor-8.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: pyarmor
-Version: 8.5.3
+Version: 8.5.5
 Summary: A tool used to obfuscate python scripts, bind obfuscated scripts to fixed machine or expire obfuscated scripts.
 Home-page: https://github.com/dashingsoft/pyarmor
 Author: Jondy Zhao

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/cli/__init__.py

@@ -1,6 +1,6 @@
 import logging
 
-__VERSION__ = '8.5.3'
+__VERSION__ = '8.5.5'
 
 logger = logging.getLogger('cli')
 

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/cli/__main__.py

@@ -135,8 +135,8 @@ def format_gen_args(ctx, args):
         options['user_data'] = args.bind_data
 
     if args.pack:
-        dist_path = os.path.join(ctx.repack_path, 'dist')
-        logger.info('implicitly set output to "%s"', dist_path)
+        dist_path = ctx.pack_obfpath
+        logger.info('implicitly save obfuscated scripts to "%s"', dist_path)
         options['output'] = dist_path
 
     return options
@@ -209,8 +209,10 @@ def check_gen_context(ctx, args):
         raise CliError('--outer conflicts with any -e, --period, -b')
 
     if args.pack:
-        if not os.path.isfile(args.pack):
-            raise CliError('--pack must be an executable file')
+        choices = 'auto', 'onefile', 'onedir', 'F', 'D', 'FC', 'DC'
+        if args.pack not in choices and not os.path.isfile(args.pack):
+            raise CliError('--pack must be an executable file, '
+                           '"auto", "onefile" or "onedir"')
         if args.no_runtime:
             raise CliError('--pack conficts with --no-runtime, --use-runtime')
         if ctx.import_prefix:
@@ -230,10 +232,16 @@ def cmd_gen(ctx, args):
         return _cmd_gen_key(builder, options)
     elif args.inputs[0].lower() in ('runtime', 'run', 'r'):
         _cmd_gen_runtime(builder, options)
+    elif args.pack in ('auto', 'onefile', 'onedir', 'F', 'D', 'FC', 'DC'):
+        from .repack import Repacker6
+        packer = Repacker6(ctx, args.pack, options['inputs'], args.output)
+        packer.check()
+        builder.process(options, packer)
+        Plugin.post_build(ctx, pack=args.pack)
     elif args.pack:
         from .repack import Repacker
         codesign = ctx.cfg['pack'].get('codesign_identify', None)
-        packer = Repacker(args.pack, ctx.repack_path, codesign=codesign)
+        packer = Repacker(args.pack, ctx.pack_basepath, codesign=codesign)
         packer.check()
         builder.process(options, packer=packer)
         Plugin.post_build(ctx, pack=args.pack)
@@ -412,8 +420,8 @@ https://pyarmor.readthedocs.io/en/stable/reference/man.html#pyarmor-gen
         'action arguments'
     ).add_mutually_exclusive_group()
     group.add_argument(
-        '--pack', metavar='BUNDLE',
-        help='repack bundle with obfuscated scripts'
+        '--pack', metavar='MODE',
+        help='specify pack mode, onefile or onedir'
     )
     group.add_argument(
         '--no-runtime', action='store_true', help=argparse.SUPPRESS

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/cli/context.py

@@ -366,10 +366,6 @@ class Context(object):
         return self._check_logpath(
             self.cfg['logging'].get('trace_logfile', 'pyarmor.trace.log'))
 
-    @property
-    def repack_path(self):
-        return os.path.join(self.local_path, 'pack')
-
     def _optb(self, section, name):
         return self.cfg.getboolean(section, name, vars=self.cmd_options)
 
@@ -608,6 +604,10 @@ class Context(object):
     def runtime_obf_key_mode(self):
         return self._opti('runtime', 'obf_key_mode')
 
+    @property
+    def runtime_patch_extension(self):
+        return self._opti('runtime', 'patch_extension')
+
     #
     # RFT settings
     #
@@ -717,3 +717,30 @@ class Context(object):
 
         with get_response('pyarmor.dashingsoft.com') as res:
             return res.read()
+
+    #
+    # Pack options for auto/onefile/onefolder mode
+    #
+    @property
+    def pyi_options(self):
+        from json import loads as json_loads
+        from shlex import split as split_opt
+        pyi_opts = []
+        jsmarker = 'json::'
+        for line in self.cfg['pack'].get('pyi_options', '').splitlines():
+            line = line.strip()
+            if line.startswith(jsmarker):
+                pyi_opts.extend(json_loads(line[len(jsmarker):]))
+            elif line.startswith('-'):
+                pyi_opts.extend(split_opt(line))
+            elif line:
+                pyi_opts.append(line)
+        return pyi_opts
+
+    @property
+    def pack_basepath(self):
+        return os.path.join(self.local_path, 'pack')
+
+    @property
+    def pack_obfpath(self):
+        return os.path.join(self.pack_basepath, 'dist')

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/cli/default.cfg

@@ -3,7 +3,7 @@
 ;; Pyarmor version
 major = 8
 minor = 5
-patch = 3
+patch = 5
 
 ;; Compatible core version
 cli.core = 6.5.1
@@ -223,6 +223,19 @@ package_name_format = pyarmor_runtime_{suffix}
 ;;     pyarmor_runtime.so
 simple_extension_name = 1
 
+;; Whether patching extension to embedded runtime key
+;;
+;;   0: do not patch extension, but store runtime key to one file
+;;   1: runtime key is embedded into extension
+;;
+;; The purpose is to solve code sign issue for Darwin platform
+;; Do not patch extension will keep code sign, so need not re-sign
+;; But it need an extra file.
+;;
+;; Note: it's different from outer runtime key.
+;;
+patch_extension = 1
+
 ;; Obfuscate runtime key
 ;;   0: no obfuscation
 ;;   1: simple obfuscation
@@ -342,6 +355,20 @@ strip = 0
 ;;    append, append it to archive
 no_matched_pyc = error
 
+;; Extra options to call PyInstaller when packing the obfuscated scripts
+;; One option one line, name and value separted by one whitespace
+;; For example
+;;
+;;   pyi_options = -w
+;;                 --add-data default.cfg:.
+;;                 --icon myapp.ico
+;;
+;; The following options could not be used here:
+;;
+;;   --distpath, --workpath, --specpath
+;;
+; pyi_options =
+
 [bcc]
 unsupported_functions = exec eval super locals __assert_armored__
 unsupported_nodes = AsyncFunctionDef AsyncFor AsyncWith Await Yield YieldFrom GeneratorExp NamedExpr MatchValue MatchSingleton MatchSequence MatchMapping MatchClass MatchStar MatchAs MatchOr

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/cli/generate.py

@@ -62,6 +62,13 @@ class Finder(object):
                 self.ctx.obfuscated_modules.add(res.pkgname)
                 self.ctx.extra_resources.append(res)
 
+    def append(self, resources):
+        resnames = [x.pkgname for x in self.ctx.resources]
+        for res in self._build_resource(resources):
+            if res.pkgname not in resnames:
+                self.ctx.obfuscated_modules.add(res.pkgname)
+                self.ctx.extra_resources.append(res)
+
     def process(self):
         logger.info('search inputs ...')
         self.prepare(self.ctx.input_paths)
@@ -149,8 +156,13 @@ class Builder(object):
         finder = Finder(self.ctx)
         finder.process()
 
-        if packer and options.get('self_contained'):
-            finder.process_extra(packer.contents)
+        if packer:
+            if options.get('self_contained'):
+                finder.process_extra(packer.contents)
+            if hasattr(packer, 'analysis'):
+                auto_resources = packer.analysis()
+                logger.info('find extra resources: %s', auto_resources)
+                finder.append(auto_resources)
 
         Pytransform3.pre_build(self.ctx)
 

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/cli/repack.py

@@ -3,6 +3,7 @@ import marshal
 import os
 import shutil
 import struct
+import sys
 import tempfile
 
 from importlib._bootstrap_external import _code_to_timestamp_pyc
@@ -435,3 +436,181 @@ class Repacker:
             check_call(cmdlist, stdout=DEVNULL, stderr=DEVNULL)
         except Exception as e:
             logger.warning('%s', e)
+
+
+# This patch is used to modify the specfile generate by PyInstaller
+#
+# It has 2 goals:
+#
+# 1. Find all the imported modules and packages
+#
+#    Save them to hook script
+#
+# 2. Search scripts and packages in the path of entry script
+#
+#    All of them will be obfuscated automatically
+#
+spec_patch_code = '''
+from PyInstaller.compat import base_prefix
+import marshal
+import os
+
+hiddenimports = set([])
+plist = set([])
+mlist = []
+src = {src}
+sn = len(src) + 1
+prefix = os.path.relpath(src)
+prefix = '' if prefix == '.' else prefix
+
+for name, path, kind in a.pure:
+    if name.startswith('pyi_rth'):
+        continue
+    if path.startswith(src) and not path.startswith(base_prefix):
+        hiddenimports.add(name)
+        if name.find('.') == -1 and os.path.basename(path) != '__init__.py':
+            mlist.append(os.path.join(prefix, path[sn:]))
+        else:
+            pkgname = os.path.dirname(path[sn:]).split(os.sep)[0]
+            plist.add(os.path.join(prefix, pkgname))
+    else:
+        hiddenimports.add(name.split('.')[0])
+
+with open({resfile}, 'wb') as f:
+    marshal.dump(mlist + list(plist), f)
+with open({hookscript}, 'w') as f:
+    f.write("hiddenimports=[%s]" % ", ".join([repr(x) for x in hiddenimports]))
+'''
+
+
+class Repacker6:
+    """New repacker to support PyInstaller 6.0+
+
+    Args:
+        ctx: build context
+        mode: onefile or onefolder
+        inputs: main script to pack
+    """
+
+    def __init__(self, ctx, mode, inputs, output):
+        self.ctx = ctx
+        self.inputs = inputs
+        self.output = 'dist' if output is None else os.path.normpath(output)
+
+        self.script = self.inputs[0]
+        self.obfpath = os.path.normpath(self.ctx.pack_obfpath)
+        self.packpath = os.path.normpath(self.ctx.pack_basepath)
+        self.workpath = os.path.join(self.packpath, 'build')
+        self.pyicmd = [sys.executable, '-m', 'PyInstaller']
+        self.pyiopts = self.init_opts(mode)
+        self.autoclean = mode in ('FC', 'DC')
+
+    def init_opts(self, mode):
+        opts = []
+        exopts = '--noconfirm', '-y'
+        if mode == 'auto':
+            exopts += ('--onefile', '-F', '--onefolder', '-D')
+        else:
+            opts.append('-F' if mode in ('onefile', 'F', 'FC') else '-D')
+        opts.extend(self.ctx.pyi_options)
+
+        exvalues = '--distpath', '--specpath', '--workpath'
+        return self.filter_opts(opts, exvalues, exopts)
+
+    def filter_opts(self, opts, exvalues, exopts=None):
+        result = []
+        isvalue = False
+        for x in opts:
+            if isvalue:
+                isvalue = False
+            elif x in exvalues:
+                isvalue = True
+            elif exopts is None or x not in exopts:
+                result.append(x)
+        return result
+
+    def analysis(self):
+        """Got imported modules and packages by PyInstaller
+
+        Generate hook script
+        Return file/dir list need to be obfuscated
+        """
+        cmdspec = [
+            sys.executable, '-m', 'PyInstaller.utils.cliutils.makespec',
+            '--specpath', self.packpath
+        ]
+        cmdspec.extend(self.filter_opts(self.pyiopts, ('--name', '-N')))
+        cmdspec.append(self.script)
+        logger.debug('%s', ' '.join(cmdspec))
+        logger.info('call PyInstaller to generate specfile')
+        check_call(cmdspec, stdout=DEVNULL, stderr=DEVNULL)
+
+        name = os.path.splitext(os.path.basename(self.script))[0]
+        rtname = self.ctx.runtime_package_name
+        specfile = os.path.join(self.packpath, name + '.spec')
+        resfile = os.path.join(self.packpath, 'resources.list')
+        hookscript = os.path.join(self.packpath, 'hook-%s.py' % rtname)
+        self.patch_specfile(specfile, hookscript, resfile)
+
+        cmdlist = self.pyicmd + ['--clean', '--workpath', self.workpath]
+        cmdlist.extend(self.pyiopts)
+        cmdlist.append(specfile)
+        logger.debug('%s', ' '.join(cmdlist))
+        logger.info('call PyInstaller to analysis, '
+                    'it may take several minutes ...')
+        check_call(cmdlist, stdout=DEVNULL, stderr=DEVNULL)
+
+        with open(resfile, 'rb') as f:
+            return marshal.load(f)
+
+    def build(self):
+        """Generate final bundle to output"""
+        script = os.path.join(self.obfpath, os.path.basename(self.inputs[0]))
+        cmdlist = self.pyicmd + [
+            '--clean',
+            '--distpath', self.output,
+            '--workpath', self.workpath,
+            '--specpath', self.packpath,
+            '--additional-hooks-dir', self.packpath,
+        ]
+        cmdlist.extend(self.pyiopts)
+        cmdlist.append(script)
+
+        logger.debug('%s', ' '.join(cmdlist))
+        logger.info('call PyInstaller to generate final bundle ...\n')
+        check_call(cmdlist)
+        logger.info('')
+        logger.info('the final bundle "%s" has been generate successfully',
+                    self.output)
+
+    def repack(self, *unused):
+        """Only for compatible with Repacker"""
+        self.build()
+
+    def patch_specfile(self, specfile, hookscript, resfile):
+        # TODO: non-ascii need specify encoding to open file
+        lines = []
+        with open(specfile, 'r') as f:
+            for line in f:
+                if line.startswith('pyz = PYZ'):
+                    break
+                lines.append(line)
+
+        lines.append(spec_patch_code.format(
+            src=repr(os.path.abspath(os.path.dirname(self.script))),
+            hookscript=repr(os.path.abspath(hookscript)),
+            resfile=repr(os.path.abspath(resfile))))
+
+        with open(specfile, 'w') as f:
+            f.write(''.join(lines))
+
+    def check(self):
+        if os.path.exists(self.output) and self.autoclean:
+            n = len(os.path.abspath(self.output).split(os.sep))
+            if n < 3:
+                prompt = 'Are you sure to remove path "%s" (y/n)? '
+                choice = input(prompt % self.output).lower()[:1]
+                if not choice == 'y':
+                    return
+            logger.info('clean output path "%s"', self.output)
+            shutil.rmtree(self.output)

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor/config.py

@@ -1,6 +1,6 @@
 from sys import platform
 
-version = '8.5.3'
+version = '8.5.5'
 
 # The corresponding version of pytransform.so
 core_version = 'r52.6'

{pyarmor-8.5.3 → pyarmor-8.5.5}/pyarmor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: pyarmor
-Version: 8.5.3
+Version: 8.5.5
 Summary: A tool used to obfuscate python scripts, bind obfuscated scripts to fixed machine or expire obfuscated scripts.
 Home-page: https://github.com/dashingsoft/pyarmor
 Author: Jondy Zhao