mirror of
https://github.com/hakril/PythonForWindows
synced 2026-06-08 14:31:45 +00:00
hakril
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import collections
|
|
import os
|
|
|
|
import windows
|
|
import windows.generated_def as gdef
|
|
|
|
from windows.winobject.event_trace import EventTraceProperties
|
|
|
|
|
|
TASK_SCHEDULER_PROVIDER = "047311A9-FA52-4A68-A1E4-4E289FBB8D17"
|
|
|
|
EVENT_ID_COUNT = collections.Counter()
|
|
|
|
def show(event):
|
|
EVENT_ID_COUNT.update([event.id])
|
|
|
|
|
|
def test_etw_trace_open_with_guid():
|
|
trace = windows.system.etw.open_trace("PFW_test_etw_1", guid="42424242-4242-4242-4242-000000001234")
|
|
|
|
def test_etw_trace_registration_and_processing():
|
|
EVENT_ID_COUNT.clear()
|
|
# RealTime Test
|
|
|
|
trace = windows.system.etw.open_trace("PFW_test_etw_2", logfile="pfw_test_trace.etl")
|
|
assert not trace.exists()
|
|
trace.start()
|
|
assert trace.exists()
|
|
trace.enable(TASK_SCHEDULER_PROVIDER, 0xff, 0xff)
|
|
# Scheduler code that generate event
|
|
windows.system.task_scheduler(r"\Microsoft\Windows\Chkdsk")["SyspartRepair"]
|
|
# End of scheduler code
|
|
trace.stop()
|
|
assert not trace.exists()
|
|
trace.process(show)
|
|
# Task scheduler generate event id 10,11,12
|
|
assert EVENT_ID_COUNT[10] >= 1
|
|
assert EVENT_ID_COUNT[11] >= 1
|
|
assert EVENT_ID_COUNT[12] >= 1
|
|
os.unlink("pfw_test_trace.etl")
|
|
|
|
def test_etw_EventTraceProperties_struct():
|
|
traceprop = EventTraceProperties.create()
|
|
traceprop.set_logger_name(u"中国银行网银助手")
|
|
assert traceprop.get_logger_name() == u"中国银行网银助手"
|
|
|
|
traceprop.set_logfilename(u"中国银行网银助手_中国银行网银助手")
|
|
assert traceprop.get_logfilename() == u"中国银行网银助手_中国银行网银助手"
|
|
|
|
def test_etw_trace_unicode():
|
|
trace = windows.system.etw.open_trace(u"中国银行网银助手", logfile="pfw_test_trace_unicode.etl")
|
|
try:
|
|
assert not trace.exists()
|
|
trace.start()
|
|
assert trace.exists()
|
|
trace.flush()
|
|
# Test session listing for unicode string as well
|
|
assert u"中国银行网银助手" in [session.name for session in windows.system.etw.sessions]
|
|
trace.stop()
|
|
assert not trace.exists()
|
|
finally:
|
|
if trace.exists():
|
|
trace.stop()
|
|
if os.path.exists("pfw_test_trace_unicode.etl"):
|
|
os.unlink("pfw_test_trace_unicode.etl")
|
|
|
|
def test_etw_trace_sessions():
|
|
assert any(u"Kernel" in session.name for session in windows.system.etw.sessions) |