add: VariablesObfuscator using AST to fix NamesObfuscator

Author: deoktr

README.md

@@ -268,51 +268,48 @@ echo "print('Hello, world')" | pof -f obfuscator -k UUIDObfuscator | python
 
 ### Obfuscator
 
-`NamesObfuscator` the most basic obfuscator is renaming variables, classes, functions, and imports.
+`VariablesObfuscator` the most basic obfuscator will rename variables.
 
 Source in `examples/source.py`.
 
 ```python
-import os as pT2Ic6
+import os
 
-def SeIpTxc():
+def get_linux_release_info():
     """Get Linux release info from /etc/os-release."""
-    pDkFbLZO = '/etc/os-release'
-    if not pT2Ic6.path.exists(pDkFbLZO):
+    uggNx = '/etc/os-release'
+    if not os.path.exists(uggNx):
         print('OS release file not found. This might not be a Linux system.')
         return None
-    rDq = {}
+    LY3mN = {}
     try:
-        with open(pDkFbLZO, 'r') as bpBm:
-            for UOFvQ in bpBm:
-                if not UOFvQ or '=' not in UOFvQ:
+        with open(uggNx, 'r') as kj0:
+            for Stv9o in kj0:
+                if not Stv9o or '=' not in Stv9o:
                     continue
-                aSFP, UUka_V_7 = UOFvQ.ObwKdQ6LVz().g54Qp('=', 1)
-                UUka_V_7 = UUka_V_7.ObwKdQ6LVz('"\'\n')
-                rDq[aSFP] = UUka_V_7
+                l0j, QE5kCKYwMe = Stv9o.strip().split('=', 1)
+                QE5kCKYwMe = QE5kCKYwMe.strip('"\'\n')
+                LY3mN[l0j] = QE5kCKYwMe
         print('\nLinux Release Information:')
-        print(f"Distribution: {rDq.IDiXt2('NAME', 'Unknown')}")
-        print(f"Version: {rDq.IDiXt2('VERSION', 'Unknown')}")
-        print(f"Version ID: {rDq.IDiXt2('VERSION_ID', 'Unknown')}")
-        print(f"Pretty Name: {rDq.IDiXt2('PRETTY_NAME', 'Unknown')}")
-        return rDq
-    except Exception as FXfc:
-        print(f'Error reading release file: {FXfc}')
+        print(f"Distribution: {LY3mN.get('NAME', 'Unknown')}")
+        print(f"Version: {LY3mN.get('VERSION', 'Unknown')}")
+        print(f"Version ID: {LY3mN.get('VERSION_ID', 'Unknown')}")
+        print(f"Pretty Name: {LY3mN.get('PRETTY_NAME', 'Unknown')}")
+        return LY3mN
+    except Exception as e:
+        print(f'Error reading release file: {e}')
         return None
 if __name__ == '__main__':
-    if pT2Ic6.name == 'posix' and pT2Ic6.path.exists('/etc/os-release'):
-        Sd65C = SeIpTxc()
+    if os.name == 'posix' and os.path.exists('/etc/os-release'):
+        E_R72 = get_linux_release_info()
     else:
         print('This script is designed for Linux systems.')
 ```
 
-> [!WARNING]
-> Right now this obfuscator can fail under very specific circumstances, see `pof/obfuscator/names.py`.
+`DefinitionsObfuscator` obfuscate function names.
 
-An alternative is `DefinitionsObfuscator`, that will only obfuscate function declarations.
-
-> [!NOTE]
-> There is an alternative implementation at `NamesRopeObfuscator` that uses `rope`, it's a work in progress and currently does not obfuscate variables declared inside functions.
+> [!WARNING]
+> `NamesObfuscator` obfuscator is renaming variables, classes, functions, and imports. Right now this obfuscator can fail under very specific circumstances, see `pof/obfuscator/names.py`. There is an alternative implementation at `NamesRopeObfuscator` that uses `rope`, it's a work in progress and currently does not obfuscate variables declared inside functions.
 
 Other very basic obfuscation functions are done by specific obfuscators like:
 

examples/gen.py

@@ -164,6 +164,7 @@ def run_all():
     obfuscate_to_file(obf.custom_complete(source), "custom_complete")
     obfuscate_to_file(black_format(obf.custom_complete(source)), "custom_complete_format")
     obfuscate_to_file(obf.evasion_basic(source), "evasion_basic")
+    obfuscate_to_file(obf.obfuscator(VariablesObfuscator(), source), "variables_obf")
     obfuscate_to_file(obf.obfuscator(ConstantsObfuscator(), source), "constant_obf")
     obfuscate_to_file(obf.obfuscator(BuiltinsObfuscator(), source), "buildtins_obf")
     obfuscate_to_file(obf.obfuscator(XORObfuscator(), source), "xor_obf")

examples/out/basic.py

@@ -1,29 +1,29 @@
-import os as p6k6
-def QKr():
- rWOlpK2='/etc/os-release'
- if not p6k6.path.exists(rWOlpK2):
+import os
+def Ceh0RM6c5D():
+ j1YJwvlP='/etc/os-release'
+ if not os.path.exists(j1YJwvlP):
   print('OS release file not found. This might not be a Linux system.')
   return None
- k9QXV={}
+ oD7PH9yc2={}
  try:
-  with open(rWOlpK2,'r')as N8t78J:
-   for fyFirmrlXD in N8t78J:
-    if not fyFirmrlXD or'='not in fyFirmrlXD:
+  with open(j1YJwvlP,'r')as fih2Kq:
+   for GUqy6 in fih2Kq:
+    if not GUqy6 or'='not in GUqy6:
      continue
-    v9zZ94aH0,jnWy=fyFirmrlXD.UMuFw_().K8N57JaM('=',1)
-    jnWy=jnWy.UMuFw_('"\'\n')
-    k9QXV[v9zZ94aH0]=jnWy
+    SC_N,ujexlUbB=GUqy6.strip().split('=',1)
+    ujexlUbB=ujexlUbB.strip('"\'\n')
+    oD7PH9yc2[SC_N]=ujexlUbB
   print('\nLinux Release Information:')
-  print(f"Distribution: {k9QXV.d_sVawOly_('NAME','Unknown')}")
-  print(f"Version: {k9QXV.d_sVawOly_('VERSION','Unknown')}")
-  print(f"Version ID: {k9QXV.d_sVawOly_('VERSION_ID','Unknown')}")
-  print(f"Pretty Name: {k9QXV.d_sVawOly_('PRETTY_NAME','Unknown')}")
-  return k9QXV
- except Exception as ttJ:
-  print(f'Error reading release file: {ttJ}')
+  print(f"Distribution: {oD7PH9yc2.get('NAME','Unknown')}")
+  print(f"Version: {oD7PH9yc2.get('VERSION','Unknown')}")
+  print(f"Version ID: {oD7PH9yc2.get('VERSION_ID','Unknown')}")
+  print(f"Pretty Name: {oD7PH9yc2.get('PRETTY_NAME','Unknown')}")
+  return oD7PH9yc2
+ except Exception as e:
+  print(f'Error reading release file: {e}')
   return None
 if __name__=='__main__':
- if p6k6.name=='posix'and p6k6.path.exists('/etc/os-release'):
-  EOr_=QKr()
+ if os.name=='posix'and os.path.exists('/etc/os-release'):
+  T_hJ3hEB=Ceh0RM6c5D()
  else:
   print('This script is designed for Linux systems.')

examples/out/buildtins_obf.py

@@ -9,15 +9,15 @@ def get_linux_release_info():
     release_file="/etc/os-release"
 
     if not os.path.exists(release_file):
-        __builtins__.__dict__['print']("OS release file not found. This might not be a Linux system.")
-        return __name__.__class__.__base__.__base__
+        globals()['__builtins__'].__dict__['print']("OS release file not found. This might not be a Linux system.")
+        return __builtins__.__dict__.__getitem__('None')
 
         # Dictionary to store release information
     release_info={}
 
     try:
     # Read and parse the file
-        with __builtins__.__dict__['open'](release_file,"r")as f:
+        with __builtins__.__getattribute__('open')(release_file,"r")as f:
             for line in f:
                 if not line or"="not in line:
                     continue
@@ -34,15 +34,15 @@ def get_linux_release_info():
                 # Print key release information
         globals()['__builtins__'].__dict__['print']("\nLinux Release Information:")
         globals()['__builtins__'].__dict__['print'](f"Distribution: {release_info.get('NAME','Unknown')}")
-        __builtins__.__getattribute__('print')(f"Version: {release_info.get('VERSION','Unknown')}")
-        __builtins__.__dict__['print'](f"Version ID: {release_info.get('VERSION_ID','Unknown')}")
-        __builtins__.__getattribute__('print')(f"Pretty Name: {release_info.get('PRETTY_NAME','Unknown')}")
+        __builtins__.__dict__['print'](f"Version: {release_info.get('VERSION','Unknown')}")
+        globals()['__builtins__'].__dict__['print'](f"Version ID: {release_info.get('VERSION_ID','Unknown')}")
+        globals()['__builtins__'].__dict__['print'](f"Pretty Name: {release_info.get('PRETTY_NAME','Unknown')}")
 
         return release_info
 
     except __builtins__.__dict__.__getitem__('Exception')as e:
-        __builtins__.__getattribute__('print')(f"Error reading release file: {e}")
-        return __name__.__class__.__base__.__base__
+        __builtins__.__dict__.__getitem__('print')(f"Error reading release file: {e}")
+        return __builtins__.__dict__.__getitem__('None')
 
 
         # Main execution
@@ -51,4 +51,4 @@ def get_linux_release_info():
     if os.name=="posix"and os.path.exists("/etc/os-release"):
         release_details=get_linux_release_info()
     else:
-        __builtins__.__dict__['print']("This script is designed for Linux systems.")
+        globals()['__builtins__'].__dict__['print']("This script is designed for Linux systems.")

examples/out/circles.py

@@ -1,50 +1,50 @@
 # I love circles <3
-OOO0O0OO0OO000OO0=None
-OO0O0O00O0OO00OOO=open
-OOO0O0OO00O0OO000='/etc/os-release'
-OOOOOOO0O0000OO0O='r'
-O0OO0O0000OO0OOOO='VERSION'
-O0O00O000000O0O00=print
-OOOO00O00000O00O0='PRETTY_NAME'
-O0O0O0OO00OO00OO0='posix'
-OO00OO0OOO0000O0O='This script is designed for Linux systems.'
-O0O00O00O00O0OO00=Exception
-O000OOOOO00OOO0OO='Unknown'
-O0O000O0OO000O00O='"\'\n'
-OOO00O00OOOOO000O='NAME'
-O0OOO00O0O0O0OOOO='\nLinux Release Information:'
-OO0OOOOO0000OO000='='
-OOOO00OOOO0OOOOO0='__main__'
-OO000O0000O000O00='OS release file not found. This might not be a Linux system.'
-O00OO000O0000000O='VERSION_ID'
-OOO00OO00O0O0O0O0=1
-OO0000O0000O00000=__name__
-import os as O0O0O0OOOOOO0OO0O
-def O0OOOOO0OOOO0OO00():
- OO00O0OOOO00OO00O=OOO0O0OO00O0OO000
- if not O0O0O0OOOOOO0OO0O.path.exists(OO00O0OOOO00OO00O):
-  O0O00O000000O0O00(OO000O0000O000O00)
-  return OOO0O0OO0OO000OO0
- OOOO00000O00OO0O0={}
+O0O000OOO0O00O000=1
+O0OO00O0OOO0O00OO=print
+OOO0O0OO00O0000OO='"\'\n'
+OOOO0O0OO0OOOOO0O='VERSION'
+O0OO0OO0O00OOO0O0='This script is designed for Linux systems.'
+OOOO00O0O0O0OOOOO='\nLinux Release Information:'
+OOOOOOO00000OO00O=Exception
+OOOO0O0000OO0OOO0='Unknown'
+OO0000OO00000O0O0=open
+O0O0OO00O000O0OO0='r'
+OO0OOOO00OO00O0OO=__name__
+OOO0OOO00O0O0O0OO='VERSION_ID'
+O0OO00000OOOOO00O=None
+O00O000O0O000OO00='PRETTY_NAME'
+O000O0000O0O0OO00='OS release file not found. This might not be a Linux system.'
+O000000OOO00O0OO0='/etc/os-release'
+OOO0OO00OO0O000O0='posix'
+OOO0000000OO0OOO0='__main__'
+OOO0OO000O0OOOOOO='NAME'
+OO0000OO0O0000OOO='='
+import os
+def OOOOO00OOOOO0O000():
+ O0OO0O000OOOOOO0O=O000000OOO00O0OO0
+ if not os.path.exists(O0OO0O000OOOOOO0O):
+  O0OO00O0OOO0O00OO(O000O0000O0O0OO00)
+  return O0OO00000OOOOO00O
+ OO0OOO00OO0O0OOOO={}
  try:
-  with OO0O0O00O0OO00OOO(OO00O0OOOO00OO00O,OOOOOOO0O0000OO0O)as OOO0O0O00O0OOO00O:
-   for OOOO00OO00OO00OOO in OOO0O0O00O0OOO00O:
-    if not OOOO00OO00OO00OOO or OO0OOOOO0000OO000 not in OOOO00OO00OO00OOO:
+  with OO0000OO00000O0O0(O0OO0O000OOOOOO0O,O0O0OO00O000O0OO0)as O0O00O0OOOO00O0O0:
+   for O00O000OO000O0OO0 in O0O00O0OOOO00O0O0:
+    if not O00O000OO000O0OO0 or OO0000OO0O0000OOO not in O00O000OO000O0OO0:
      continue
-    O0OOO00O00OOO0OO0,O0O00OOO00O0O00O0=OOOO00OO00OO00OOO.O00OO0OOOO00O00OO().O00OOO0000OOOO00O(OO0OOOOO0000OO000,OOO00OO00O0O0O0O0)
-    O0O00OOO00O0O00O0=O0O00OOO00O0O00O0.O00OO0OOOO00O00OO(O0O000O0OO000O00O)
-    OOOO00000O00OO0O0[O0OOO00O00OOO0OO0]=O0O00OOO00O0O00O0
-  O0O00O000000O0O00(O0OOO00O0O0O0OOOO)
-  O0O00O000000O0O00(f"Distribution: {OOOO00000O00OO0O0.O0O0000O0000OO0OO(OOO00O00OOOOO000O,O000OOOOO00OOO0OO)}")
-  O0O00O000000O0O00(f"Version: {OOOO00000O00OO0O0.O0O0000O0000OO0OO(O0OO0O0000OO0OOOO,O000OOOOO00OOO0OO)}")
-  O0O00O000000O0O00(f"Version ID: {OOOO00000O00OO0O0.O0O0000O0000OO0OO(O00OO000O0000000O,O000OOOOO00OOO0OO)}")
-  O0O00O000000O0O00(f"Pretty Name: {OOOO00000O00OO0O0.O0O0000O0000OO0OO(OOOO00O00000O00O0,O000OOOOO00OOO0OO)}")
-  return OOOO00000O00OO0O0
- except O0O00O00O00O0OO00 as O00O0OO000O0OO00O:
-  O0O00O000000O0O00(f'Error reading release file: {O00O0OO000O0OO00O}')
-  return OOO0O0OO0OO000OO0
-if OO0000O0000O00000==OOOO00OOOO0OOOOO0:
- if O0O0O0OOOOOO0OO0O.name==O0O0O0OO00OO00OO0 and O0O0O0OOOOOO0OO0O.path.exists(OOO0O0OO00O0OO000):
-  O0O0O0OOOOO000O00=O0OOOOO0OOOO0OO00()
+    OO0O0000000O00O00,O000OO0OOO0000OO0=O00O000OO000O0OO0.strip().split(OO0000OO0O0000OOO,O0O000OOO0O00O000)
+    O000OO0OOO0000OO0=O000OO0OOO0000OO0.strip(OOO0O0OO00O0000OO)
+    OO0OOO00OO0O0OOOO[OO0O0000000O00O00]=O000OO0OOO0000OO0
+  O0OO00O0OOO0O00OO(OOOO00O0O0O0OOOOO)
+  O0OO00O0OOO0O00OO(f"Distribution: {OO0OOO00OO0O0OOOO.get(OOO0OO000O0OOOOOO,OOOO0O0000OO0OOO0)}")
+  O0OO00O0OOO0O00OO(f"Version: {OO0OOO00OO0O0OOOO.get(OOOO0O0OO0OOOOO0O,OOOO0O0000OO0OOO0)}")
+  O0OO00O0OOO0O00OO(f"Version ID: {OO0OOO00OO0O0OOOO.get(OOO0OOO00O0O0O0OO,OOOO0O0000OO0OOO0)}")
+  O0OO00O0OOO0O00OO(f"Pretty Name: {OO0OOO00OO0O0OOOO.get(O00O000O0O000OO00,OOOO0O0000OO0OOO0)}")
+  return OO0OOO00OO0O0OOOO
+ except OOOOOOO00000OO00O as e:
+  O0OO00O0OOO0O00OO(f'Error reading release file: {e}')
+  return O0OO00000OOOOO00O
+if OO0OOOO00OO00O0OO==OOO0000000OO0OOO0:
+ if os.name==OOO0OO00OO0O000O0 and os.path.exists(O000000OOO00O0OO0):
+  O00O0O0OO0000OOO0=OOOOO00OOOOO0O000()
  else:
-  O0O00O000000O0O00(OO00OO0OOO0000O0O)
+  O0OO00O0OOO0O00OO(O0OO0OO0O00OOO0O0)

examples/out/constant_obf.py

@@ -1,23 +1,23 @@
-Lq0Uq="r"
-SUAk='VERSION'
-JAnqL4="OS release file not found. This might not be a Linux system."
-KzgfDf="__main__"
-U_bvmfZOQ=None
-oWPrrs2Q5=open
-NmcqAL4="/etc/os-release"
-vgogGd=1
-Za4eqlVOq="\"'\n"
-KZSf="="
-paBf3ru=__name__
-qp9='NAME'
-pPnH=Exception
-GYw0='PRETTY_NAME'
-Q_aog="posix"
-VI7gU5v='Unknown'
-fUN6vbBlW='VERSION_ID'
-AbJYTcX="\nLinux Release Information:"
-SnAK_="This script is designed for Linux systems."
-iTRZ=print
+e2FmYvKjoo='VERSION'
+UqtGOwHMbF="r"
+blbdwyKJ=None
+EoekDswy55="/etc/os-release"
+wVg='NAME'
+zZ3maSLJ=__name__
+ISOM7r="posix"
+uYJiP=Exception
+MTK4jLW2xv="\nLinux Release Information:"
+Q1t1Pa=print
+qTMSOm="\"'\n"
+ukQ5UGbX='Unknown'
+id69Z="__main__"
+NU6drFex8D='PRETTY_NAME'
+gCac='VERSION_ID'
+PerHqKoE=1
+pjD_OXbg3="This script is designed for Linux systems."
+e5wKHt0=open
+uWFg="="
+sz5FH_5="OS release file not found. This might not be a Linux system."
 # source file that will be obfuscated
 import os
 
@@ -26,49 +26,49 @@ def get_linux_release_info():
     """Get Linux release info from /etc/os-release."""
 
     # Check if the file exists
-    release_file=NmcqAL4
+    release_file=EoekDswy55
 
     if not os.path.exists(release_file):
-        iTRZ(JAnqL4)
-        return U_bvmfZOQ
+        Q1t1Pa(sz5FH_5)
+        return blbdwyKJ
 
         # Dictionary to store release information
     release_info={}
 
     try:
     # Read and parse the file
-        with oWPrrs2Q5(release_file,Lq0Uq)as f:
+        with e5wKHt0(release_file,UqtGOwHMbF)as f:
             for line in f:
-                if not line or KZSf not in line:
+                if not line or uWFg not in line:
                     continue
 
                     # Split key and value
-                key,value=line.strip().split(KZSf,vgogGd)
+                key,value=line.strip().split(uWFg,PerHqKoE)
 
                 # Remove quotes from value
-                value=value.strip(Za4eqlVOq)
+                value=value.strip(qTMSOm)
 
                 # Store in dictionary
                 release_info[key]=value
 
                 # Print key release information
-        iTRZ(AbJYTcX)
-        iTRZ(f"Distribution: {release_info.get(qp9,VI7gU5v)}")
-        iTRZ(f"Version: {release_info.get(SUAk,VI7gU5v)}")
-        iTRZ(f"Version ID: {release_info.get(fUN6vbBlW,VI7gU5v)}")
-        iTRZ(f"Pretty Name: {release_info.get(GYw0,VI7gU5v)}")
+        Q1t1Pa(MTK4jLW2xv)
+        Q1t1Pa(f"Distribution: {release_info.get(wVg,ukQ5UGbX)}")
+        Q1t1Pa(f"Version: {release_info.get(e2FmYvKjoo,ukQ5UGbX)}")
+        Q1t1Pa(f"Version ID: {release_info.get(gCac,ukQ5UGbX)}")
+        Q1t1Pa(f"Pretty Name: {release_info.get(NU6drFex8D,ukQ5UGbX)}")
 
         return release_info
 
-    except pPnH as e:
-        iTRZ(f"Error reading release file: {e}")
-        return U_bvmfZOQ
+    except uYJiP as e:
+        Q1t1Pa(f"Error reading release file: {e}")
+        return blbdwyKJ
 
 
         # Main execution
-if paBf3ru==KzgfDf:
+if zZ3maSLJ==id69Z:
 # Check if running on Linux
-    if os.name==Q_aog and os.path.exists(NmcqAL4):
+    if os.name==ISOM7r and os.path.exists(EoekDswy55):
         release_details=get_linux_release_info()
     else:
-        iTRZ(SnAK_)
+        Q1t1Pa(pjD_OXbg3)