def ReadPipeMessage(hPipe):
data = ""
fSuccess = 0
chBuf = create_string_buffer(BUFSIZE)
cbRead = DWORD(0)
while not fSuccess: # repeat loop if ERROR_MORE_DATA
fSuccess = ReadFile(hPipe, chBuf, BUFSIZE, byref(cbRead), None)
if fSuccess == 1:
data += chBuf.value
break
elif GetLastError() != ERROR_MORE_DATA:
break
data += chBuf.value
return loads(data)
def RunAs(filePath, asAdministrator, *args):
sei = SHELLEXECUTEINFO()
sei.cbSize = sizeof(SHELLEXECUTEINFO)
sei.fMask = (SEE_MASK_FLAG_DDEWAIT | SEE_MASK_FLAG_NO_UI
| SEE_MASK_NOCLOSEPROCESS)
if asAdministrator:
sei.lpVerb = u"runas"
else:
sei.lpVerb = u""
sei.lpFile = GetUncPathOf(filePath)
sei.lpParameters = " ".join(
['"%s"' % arg.replace('"', '""') for arg in args])
sei.nShow = SW_SHOWNORMAL
if not ctypes.windll.shell32.ShellExecuteExW(byref(sei)):
err = GetLastError()
raise WindowsError(err, "ShellExecuteEx: %s" % FormatError(err))
return sei.hProcess
def IsAdmin():
"""
Find out if the user (the owner of the current process) is a member of
the administrators group on the local computer (not on the domain!).
"""
# First we must open a handle to the access token for this thread.
hThread = HANDLE()
if not OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, 0, byref(hThread)):
err = GetLastError()
if err == ERROR_NO_TOKEN:
# If the thread does not have an access token, we'll examine the
# access token associated with the process.
if not OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY,
byref(hThread)):
raise WinError()
else:
raise WinError(err)
# Then we must query the size of the group information associated with
# the token. Note that we expect a FALSE result from GetTokenInformation
# because we've given it a NULL buffer. On exit cbTokenGroups will tell
# the size of the group information.
cbTokenGroups = DWORD()
if GetTokenInformation(hThread, TokenGroups, None, 0,
byref(cbTokenGroups)):
raise WinError()
# Here we verify that GetTokenInformation failed for lack of a large
# enough buffer.
err = GetLastError()
if err != ERROR_INSUFFICIENT_BUFFER:
raise WinError(err)
# Now we allocate a buffer for the group information.
ptg = create_string_buffer(cbTokenGroups.value)
# Now we ask for the group information again.
# This may fail if an administrator has added this account to an additional
# group between our first call to GetTokenInformation and this one.
if not GetTokenInformation(hThread, TokenGroups, ptg, cbTokenGroups,
byref(cbTokenGroups)):
raise WinError()
# Now we must create a System Identifier for the Admin group.
systemSidAuthority = SID_IDENTIFIER_AUTHORITY()
systemSidAuthority.Value[5] = SECURITY_NT_AUTHORITY
psidAdmin = PSID()
if not AllocateAndInitializeSid(
byref(systemSidAuthority), 2, SECURITY_BUILTIN_DOMAIN_RID,
DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, byref(psidAdmin)):
raise WinError()
# Finally we'll iterate through the list of groups for this access
# token looking for a match against the SID we created above.
ptg = cast(ptg, POINTER(TOKEN_GROUPS))
groups = cast(ptg.contents.Groups, POINTER(SID_AND_ATTRIBUTES))
isAdmin = False
for i in range(ptg.contents.GroupCount):
if EqualSid(groups[i].Sid, psidAdmin.value):
isAdmin = True
break
# Before we exit we must explicitly deallocate the SID we created.
FreeSid(psidAdmin)
return isAdmin
def GetComPorts(availableOnly=True):
"""
Scans the registry for serial ports and return a list of (port, desc, hwid)
tuples.
If availableOnly is true only return currently existing ports.
"""
result = []
stringBuffer = create_unicode_buffer(256)
flags = DIGCF_DEVICEINTERFACE
if availableOnly:
flags |= DIGCF_PRESENT
hdi = SetupDiGetClassDevs(byref(GUID_CLASS_COMPORT), None, 0, flags)
if hdi == INVALID_HANDLE_VALUE:
raise WinError()
dwRequiredSize = DWORD()
dwIndex = 0
while True:
did = SP_DEVICE_INTERFACE_DATA()
did.cbSize = sizeof(did)
if not SetupDiEnumDeviceInterfaces(
hdi, None, byref(GUID_CLASS_COMPORT), dwIndex, byref(did)):
err = GetLastError()
if err != ERROR_NO_MORE_ITEMS:
raise WinError(err)
break
# get the size
if not SetupDiGetDeviceInterfaceDetail(hdi, byref(did), None, 0,
byref(dwRequiredSize), None):
# Ignore ERROR_INSUFFICIENT_BUFFER
err = GetLastError()
if err != ERROR_INSUFFICIENT_BUFFER:
raise WinError(err)
# allocate buffer
class _SP_DEVICE_INTERFACE_DETAIL_DATA(Structure):
_fields_ = [
('cbSize', DWORD),
('DevicePath', TCHAR * (dwRequiredSize.value - sizeof(DWORD))),
]
idd = _SP_DEVICE_INTERFACE_DETAIL_DATA()
idd.cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA)
devinfo = SP_DEVINFO_DATA()
devinfo.cbSize = sizeof(devinfo)
if not SetupDiGetDeviceInterfaceDetail(
hdi, byref(did),
cast(byref(idd), PSP_DEVICE_INTERFACE_DETAIL_DATA),
dwRequiredSize, None, byref(devinfo)):
raise WinError()
# hardware ID
if not SetupDiGetDeviceRegistryProperty(
hdi, byref(devinfo), SPDRP_HARDWAREID, None,
cast(stringBuffer, PBYTE),
sizeof(stringBuffer) - 1, None):
# Ignore ERROR_INSUFFICIENT_BUFFER
err = GetLastError()
if err != ERROR_INSUFFICIENT_BUFFER:
raise WinError(err)
szHardwareID = stringBuffer.value
# friendly name
if not SetupDiGetDeviceRegistryProperty(
hdi, byref(devinfo), SPDRP_FRIENDLYNAME, None,
cast(stringBuffer, PBYTE),
sizeof(stringBuffer) - 1, None):
# Ignore ERROR_INSUFFICIENT_BUFFER
err = GetLastError()
if err != ERROR_INSUFFICIENT_BUFFER:
raise WinError(err)
szFriendlyName = stringBuffer.value
portName = re.search(r"\((.*)\)", szFriendlyName).group(1)
result.append((portName, szFriendlyName, szHardwareID))
dwIndex += 1
SetupDiDestroyDeviceInfoList(hdi)
return result
