def click_button(key):
"""Simulate key press
"""
log.debug("click_button (%d)", key)
USER32.keybd_event(key, 0, 0, 0)
KERNEL32.Sleep(60)
USER32.keybd_event(key, 0, win32con.KEYEVENTF_KEYUP, 0)
def foreach_child(hwnd, lparam):
# List of buttons labels to click.
buttons = [
"yes",
"ok",
"accept",
"next",
"install",
"run",
"agree",
"enable",
"don't send",
"don't save",
"continue",
"unzip",
"open",
"close the program",
"save",
"later",
"finish",
"end",
"allow access",
"remind me later",
]
# List of buttons labels to not click.
dontclick = [
"check online for a solution",
"don't run",
"do not ask again until the next update is available",
"cancel",
"do not accept the agreement",
"i would like to help make reader even better"
]
classname = create_unicode_buffer(128)
USER32.GetClassNameW(hwnd, classname, 128)
# Check if the class of the child is button.
if "button" in classname.value.lower() or classname.value == "NUIDialog" or classname.value == "bosa_sdm_msword":
# Get the text of the button.
length = USER32.SendMessageW(hwnd, WM_GETTEXTLENGTH, 0, 0)
if not length:
return True
text = create_unicode_buffer(length + 1)
USER32.SendMessageW(hwnd, WM_GETTEXT, length + 1, text)
textval = text.value.replace('&','')
if "Microsoft" in textval and (classname.value == "NUIDialog" or classname.value == "bosa_sdm_msword"):
log.info("Issuing keypress on Office dialog")
USER32.SetForegroundWindow(hwnd)
# enter key down/up
USER32.keybd_event(0x0d, 0x1c, 0, 0)
USER32.keybd_event(0x0d, 0x1c, 2, 0)
return False
# we don't want to bother clicking any non-visible child elements, as they
# generally won't respond and will cause us to fixate on them for the
# rest of the analysis, preventing progress with visible elements
if not USER32.IsWindowVisible(hwnd):
return True
# Check if the button is set as "clickable" and click it.
for button in buttons:
if button in textval.lower():
dontclickb = False
for btn in dontclick:
if btn in textval.lower():
dontclickb = True
if not dontclickb:
log.info("Found button \"%s\", clicking it" % text.value)
USER32.SetForegroundWindow(hwnd)
KERNEL32.Sleep(1000)
USER32.SendMessageW(hwnd, BM_CLICK, 0, 0)
# only stop searching when we click a button
return False
return True
def foreach_child(hwnd, lparam):
# List of buttons labels to click.
buttons = [
# english
"yes",
"ok",
"accept",
"next",
"install",
"run",
"agree",
"enable",
"don't send",
"don't save",
"continue",
"unzip",
"open",
"close the program",
"save",
"later",
"finish",
"end",
"allow access",
"remind me later",
# german
"ja",
"weiter",
"akzeptieren",
"ende",
"starten",
"jetzt starten",
"neustarten",
"neu starten",
"jetzt neu starten",
"beenden",
"oeffnen",
"schliessen",
"installation weiterfuhren",
"fertig",
"beenden",
"fortsetzen",
"fortfahren",
"stimme zu",
"zustimmen",
"senden",
"nicht senden",
"speichern",
"nicht speichern",
"ausfuehren",
"spaeter",
"einverstanden"
]
# List of buttons labels to not click.
dontclick = [
# english
"check online for a solution",
"don't run",
"do not ask again until the next update is available",
"cancel",
"do not accept the agreement",
"i would like to help make reader even better",
# german
"abbrechen",
"online nach losung suchen",
"abbruch",
"nicht ausfuehren",
"hilfe",
"stimme nicht zu"
]
classname = create_unicode_buffer(128)
USER32.GetClassNameW(hwnd, classname, 128)
# Check if the class of the child is button.
if "button" in classname.value.lower(
) or classname.value == "NUIDialog" or classname.value == "bosa_sdm_msword":
# Get the text of the button.
length = USER32.SendMessageW(hwnd, WM_GETTEXTLENGTH, 0, 0)
if not length:
return True
text = create_unicode_buffer(length + 1)
USER32.SendMessageW(hwnd, WM_GETTEXT, length + 1, text)
textval = text.value.replace('&', '')
if "Microsoft" in textval and (classname.value == "NUIDialog" or
classname.value == "bosa_sdm_msword"):
log.info("Issuing keypress on Office dialog")
USER32.SetForegroundWindow(hwnd)
# enter key down/up
USER32.keybd_event(0x0d, 0x1c, 0, 0)
USER32.keybd_event(0x0d, 0x1c, 2, 0)
return False
# we don't want to bother clicking any non-visible child elements, as they
# generally won't respond and will cause us to fixate on them for the
# rest of the analysis, preventing progress with visible elements
if not USER32.IsWindowVisible(hwnd):
return True
# Check if the button is set as "clickable" and click it.
for button in buttons:
if button in textval.lower():
dontclickb = False
for btn in dontclick:
if btn in textval.lower():
dontclickb = True
if not dontclickb:
log.info("Found button \"%s\", clicking it" % text.value)
USER32.SetForegroundWindow(hwnd)
KERNEL32.Sleep(1000)
USER32.SendMessageW(hwnd, BM_CLICK, 0, 0)
# only stop searching when we click a button
return False
return True
def foreach_child(hwnd, lparam):
# List of buttons labels to click.
buttons = [
"yes",
"ok",
"accept",
"next",
"install",
"run",
"agree",
"enable",
"don't send",
"don't save",
"continue",
"unzip",
"open",
"close the program",
"save",
"later",
"finish",
"end",
"allow access",
]
# List of buttons labels to not click.
dontclick = [
"don't run",
"do not ask again until the next update is available",
]
classname = create_unicode_buffer(128)
USER32.GetClassNameW(hwnd, classname, 128)
# Check if the class of the child is button.
if "button" in classname.value.lower() or classname.value == "NUIDialog":
# Get the text of the button.
length = USER32.SendMessageW(hwnd, WM_GETTEXTLENGTH, 0, 0)
if not length:
return True
text = create_unicode_buffer(length + 1)
USER32.SendMessageW(hwnd, WM_GETTEXT, length + 1, text)
textval = text.value.replace('&','')
if classname.value == "NUIDialog" and "Microsoft" in textval:
log.info("Issuing keypress on Office dialog")
USER32.SetForegroundWindow(hwnd)
# enter key down/up
USER32.keybd_event(0x0d, 0x1c, 0, 0)
USER32.keybd_event(0x0d, 0x1c, 2, 0)
return False
# Check if the button is set as "clickable" and click it.
for button in buttons:
if button in textval.lower():
dontclickb = False
for btn in dontclick:
if btn in textval.lower():
dontclickb = True
if not dontclickb:
log.info("Found button \"%s\", clicking it" % text.value)
USER32.SetForegroundWindow(hwnd)
KERNEL32.Sleep(1000)
USER32.SendMessageW(hwnd, BM_CLICK, 0, 0)
# only stop searching when we click a button
return False
return True
def foreach_child(hwnd, lparam):
# List of buttons labels to click.
buttons = [
"yes",
"ok",
"accept",
"next",
"install",
"run",
"agree",
"enable",
"don't send",
"don't save",
"continue",
"unzip",
"open",
"close the program",
"save",
"later",
"finish",
"end",
"allow access",
]
# List of buttons labels to not click.
dontclick = [
"don't run",
"do not ask again until the next update is available",
]
classname = create_unicode_buffer(128)
USER32.GetClassNameW(hwnd, classname, 128)
# Check if the class of the child is button.
if "button" in classname.value.lower() or classname.value == "NUIDialog":
# Get the text of the button.
length = USER32.SendMessageW(hwnd, WM_GETTEXTLENGTH, 0, 0)
if not length:
return True
text = create_unicode_buffer(length + 1)
USER32.SendMessageW(hwnd, WM_GETTEXT, length + 1, text)
textval = text.value.replace('&', '')
if classname.value == "NUIDialog" and "Microsoft" in textval:
log.info("Issuing keypress on Office dialog")
USER32.SetForegroundWindow(hwnd)
# enter key down/up
USER32.keybd_event(0x0d, 0x1c, 0, 0)
USER32.keybd_event(0x0d, 0x1c, 2, 0)
return False
# Check if the button is set as "clickable" and click it.
for button in buttons:
if button in textval.lower():
dontclickb = False
for btn in dontclick:
if btn in textval.lower():
dontclickb = True
if not dontclickb:
log.info("Found button \"%s\", clicking it" % text.value)
USER32.SetForegroundWindow(hwnd)
KERNEL32.Sleep(1000)
USER32.SendMessageW(hwnd, BM_CLICK, 0, 0)
# only stop searching when we click a button
return False
return True
def type_keyboard(key_value, scan_value):
USER32.keybd_event(key_value, scan_value, 0, 0)
USER32.keybd_event(key_value, scan_value, 0x0002, 0)
KERNEL32.Sleep(80)
