# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. from winappdbg import Table # Instance a Table object. table = Table() # Add a few rows. table.addRow( "Right justified column text", "Left justified column text" ) table.addRow( "---------------------------", "--------------------------" ) table.addRow( "example", "text" ) table.addRow( "jabberwocky", "snark" ) table.addRow( "Trillian", "Zaphod", "Arthur Dent" ) # one extra! table.addRow( "Dalek", "Cyberman" ) # By default all columns are left justified. Let's change that. table.justify( 0, 1 ) # column 0 is now right justified # Let's find out how wide the table is. print "Table width: %d" % table.getWidth() # Let's find out how many bytes would it be if written to a file. print "Text size in characters: %d" % len( table.getOutput() ) # Show the table contents on screen. print table.show(),
table.addRow(' %d' % pid, fileName, caplist[0])
for i in xrange(1, len(caplist)):
table.addRow('', '', caplist[i])
else:
table.addRow(' %d' % pid, fileName, '')
elif options.services:
if len(srvlist):
table.addRow(' %d' % pid, fileName, srvlist[0])
for i in xrange(1, len(srvlist)):
table.addRow('', '', srvlist[i])
else:
table.addRow(' %d' % pid, fileName, '')
else:
table.addRow(' %d' % pid, fileName)
table.justify(0, 1)
if options.format == "auto" and table.getWidth() >= 80:
options.format = "long"
else:
table.show()
if options.format == "long":
# If it doesn't fit, build a new table of only two rows. The first row
# contains the headers and the second row the data. Insert an empty row
# between each process.
need_empty_row = False
table = Table()
for pid in pid_list:
if pid in filenames:
if need_empty_row:
table.addRow()
else:
table.addRow(' %d' % pid, fileName, caplist[0])
for i in xrange(1, len(caplist)):
table.addRow('', '', caplist[i])
else:
table.addRow(' %d' % pid, fileName, '')
elif options.services:
if len(srvlist):
table.addRow(' %d' % pid, fileName, srvlist[0])
for i in xrange(1, len(srvlist)):
table.addRow('', '', srvlist[i])
else:
table.addRow(' %d' % pid, fileName, '')
else:
table.addRow(' %d' % pid, fileName)
table.justify(0, 1)
if options.format == "auto" and table.getWidth() >= 80:
options.format = "long"
else:
table.show()
if options.format == "long":
# If it doesn't fit, build a new table of only two rows. The first row
# contains the headers and the second row the data. Insert an empty row
# between each process.
need_empty_row = False
table = Table()
for pid in pid_list:
if pid in filenames:
if need_empty_row:
table.addRow()
else:
def main(argv):
'Main function.'
# Print the banner.
print("Process enumerator")
print("by Mario Vilas (mvilas at gmail.com)")
print()
# Parse the command line options.
(options, argv) = parse_cmdline(argv)
showFilenameOnly = not options.full_path
searchString = options.search
# Windows filenames are case insensitive.
if searchString:
searchString = searchString.lower()
# Take a snapshot of the running processes.
s = System()
s.request_debug_privileges()
try:
s.scan_processes()
if not showFilenameOnly:
s.scan_process_filenames()
except WindowsError:
s.scan_processes_fast()
pid_list = s.get_process_ids()
pid_list.sort()
if not pid_list:
print("Unknown error enumerating processes!")
return
# Get the filename of each process.
filenames = dict()
for pid in pid_list:
p = s.get_process(pid)
fileName = p.get_filename()
# Special process IDs.
# PID 0: System Idle Process. Also has a special meaning to the
# toolhelp APIs (current process).
# PID 4: System Integrity Group. See this forum post for more info:
# http://tinyurl.com/ycza8jo
# (points to social.technet.microsoft.com)
# Only on XP and above
# PID 8: System (?) only in Windows 2000 and below AFAIK.
# It's probably the same as PID 4 in XP and above.
if pid == 0:
fileName = "[System Idle Process]"
elif pid == 4:
fileName = "[System Integrity Group]"
elif pid == 8:
fileName = "[System]"
# Filename not available.
elif not fileName:
fileName = ""
# Get the process pathname instead, if requested.
elif showFilenameOnly:
fileName = PathOperations.pathname_to_filename(fileName)
# Filter the output with the search string.
if searchString and searchString not in fileName.lower():
continue
# Remember the filename.
filenames[pid] = fileName
# Get the window captions if requested.
# TODO: show window handles too if possible
captions = dict()
if options.windows:
for w in s.get_windows():
try:
pid = w.get_pid()
text = w.get_text()
except WindowsError:
continue
try:
captions[pid].add(text)
except KeyError:
capset = set()
capset.add(text)
captions[pid] = capset
# Get the services if requested.
services = dict()
if options.services:
try:
for descriptor in s.get_services():
try:
services[descriptor.ProcessId].add(descriptor.ServiceName)
except KeyError:
srvset = set()
srvset.add(descriptor.ServiceName)
services[descriptor.ProcessId] = srvset
except WindowsError as e:
print("Error getting the list of services: %s" % str(e))
return
if options.format == "auto":
if options.windows or options.services:
options.format = "long"
if options.format != "long":
headers = [" PID", "Filename"]
if options.windows:
headers.append("Windows")
if options.services:
headers.append("Services")
table = Table()
table.addRow(*headers)
for pid in pid_list:
if pid in filenames:
fileName = filenames[pid]
caplist = sorted(captions.get(pid, set()))
srvlist = sorted(services.get(pid, set()))
if options.windows and options.services:
if len(caplist) < len(srvlist):
caplist.extend([''] * (len(srvlist) - len(caplist)))
elif len(srvlist) < len(caplist):
srvlist.extend([''] * (len(caplist) - len(srvlist)))
if len(caplist):
table.addRow(' %d' % pid, fileName, caplist[0],
srvlist[0])
for i in range(1, len(caplist)):
table.addRow('', '', caplist[i], srvlist[i])
else:
table.addRow(' %d' % pid, fileName, '', '')
elif options.windows:
if len(caplist):
table.addRow(' %d' % pid, fileName, caplist[0])
for i in range(1, len(caplist)):
table.addRow('', '', caplist[i])
else:
table.addRow(' %d' % pid, fileName, '')
elif options.services:
if len(srvlist):
table.addRow(' %d' % pid, fileName, srvlist[0])
for i in range(1, len(srvlist)):
table.addRow('', '', srvlist[i])
else:
table.addRow(' %d' % pid, fileName, '')
else:
table.addRow(' %d' % pid, fileName)
table.justify(0, 1)
if options.format == "auto" and table.getWidth() >= 80:
options.format = "long"
else:
table.show()
if options.format == "long":
# If it doesn't fit, build a new table of only two rows. The first row
# contains the headers and the second row the data. Insert an empty row
# between each process.
need_empty_row = False
table = Table()
for pid in pid_list:
if pid in filenames:
if need_empty_row:
table.addRow()
else:
need_empty_row = True
table.addRow("PID:", pid)
fileName = filenames[pid]
if fileName:
table.addRow("Filename:", fileName)
caplist = sorted(captions.get(pid, set()))
if caplist:
caption = caplist.pop(0)
table.addRow("Windows:", caption)
for caption in caplist:
table.addRow('', caption)
srvlist = sorted(services.get(pid, set()))
if srvlist:
srvname = srvlist.pop(0)
table.addRow("Services:", srvname)
for srvname in srvlist:
table.addRow('', srvname)
table.justify(0, 1)
table.show()
