2016-07-24 62 views
1

我正在使用ctypes模块来调用GetTcpTable2。 我一直在慢慢地将C++中的here示例转换为Python;但在实地访问期间会崩溃。Python在现场访问ctypes时崩溃

if __name__ == "__main__": 
    ptcp_table = POINTER(MIB_TCPTABLE2)() 
    ptcp_table = cast(create_string_buffer(sizeof(MIB_TCPTABLE2)), 
         POINTER(MIB_TCPTABLE2)) 
    ip_addr = in_addr() 
    size = c_ulong(sizeof(MIB_TCPTABLE2)) 
    retval = GetTcpTable2(ptcp_table, byref(size), TRUE) 
    if retval == ERROR_INSUFFICIENT_BUFFER: 
     ptcp_table = cast(create_string_buffer(size.value), 
          POINTER(MIB_TCPTABLE2)) 
     if not ptcp_table: 
      #throw error 
      pass 

    retval = GetTcpTable2(ptcp_table, byref(size), TRUE) 
    if retval == NO_ERROR: 
     print("Entries %d" % ptcp_table[0].dwNumEntries) 
     for i in range(0, ptcp_table[0].dwNumEntries): 
      print(ptcp_table[0].table[i]) 
      #ip_addr.S_un.S_addr = ptcp_table[0].table[i].dwLocalAddr 
      #ip_addr_string = inet_nota(ip_addr) 
      #print(ip_addr_string) 
      #print(string_at(ip_addr_string)) 

它试图访问dwLocalAddr分开的table[i]时崩溃。

ptcp_table[0].table[i].dwLocalAddr 

但是,它只是在打印ptcp_table[0].table[i]时不会崩溃。 我试过打印和访问其他字段;但Python只是崩溃。

这里是我的结构定义:

定义的 GetTcpTable2
class MIB_TCPROW2(Structure): 
    _fields_ = [ 
     ("dwState", c_ulong), 
     ("dwLocalAddr", c_ulong), 
     ("dwLocalPort", c_ulong), 
     ("dwRemoteAddr", c_ulong), 
     ("dwRemotePort", c_ulong), 
     ("dwOwningPid", c_ulong), 
     ("dwOffloadState", c_int) 
    ] 


class MIB_TCPTABLE2(Structure): 
    _fields_ = [ 
     ("dwNumEntries", c_ulong), 
     ("table", POINTER(MIB_TCPROW2)) 
    ] 

GetTcpTable2 = windll.iphlpapi.GetTcpTable2 
GetTcpTable2.argtypes = [POINTER(MIB_TCPTABLE2), POINTER(c_ulong), c_char] 
GetTcpTable2.restype = c_ulong 

我有一个小的预感,在MIB_TCPTABLE2结构的定义;该文件说tableMIB_TCPROW2大小ANY_SIZE的数组;并进一步检查是从检查iphlpapi.h文件中得到的。而且我知道POINTER(MIB_TCPROW2)的尺寸不等于MIB_TCPROW2的尺寸。

+1

'MIB_TCPTABLE2'是一个可变大小的结构; 'dwNumEntries'告诉你'table'数组实际上有多大。你如何将它表示为Python我不知道。 –

+0

@JonathanPotter谢谢你指出这一点。我已经做了一些更多的搜索,并着眼于为'MIB_TCPTABLE2'创建一个类工厂方法来生成它的变量加长版本。 – jacob

+0

@ J.J.Hakala必须等2天才能做到这一点。 (从现在起3个小时) – jacob

回答

1

我查看了围绕结构内变长字段的其他ctypes问题,并提出了一个answer,它建议使用工厂方法来生成类定义。

def MIB_TCPTABLE2_FACTORY(size): 
    class MIB_TCPTABLE2(Structure): 
     _fields_ = [ 
      ("dwNumEntries", c_ulong), 
      ("table", MIB_TCPROW2 * size) 
     ] 
    return MIB_TCPTABLE2 

我可以用这个知道sizeGetTcpTable2返回到创建一个新的类型。然后,我所要做的就是将的argtypes更改为接受void *

GetTcpTable2.argtypes = [c_void_p, POINTER(c_ulong), c_char]