python中使用ctypes調(diào)用so傳參設(shè)置遇到的問(wèn)題及解決方法

問(wèn)題

近日在做一組聲紋聚類時(shí)，使用了另一團(tuán)隊(duì)同學(xué)開發(fā)的聲紋距離算法。該算法對(duì)外提供的是一組so包，需要使用方自己去使用。在python中調(diào)用純so包一般使用ctypes類庫(kù)，用起來(lái)看起來(lái)簡(jiǎn)單但也有不少細(xì)節(jié)容易犯錯(cuò)。本次使用過(guò)程中，就遇到傳參的問(wèn)題。

目標(biāo)so庫(kù)中對(duì)外export的函數(shù)是大致如下的三個(gè)函數(shù):

void* create_handler();
  int extract_feature(void* hander);
  bool destroy(void* handler);

這三個(gè)函數(shù)使用起來(lái)倒也簡(jiǎn)單，順序使用就可以了。但發(fā)現(xiàn)寫成如下形式的python代碼后，執(zhí)行會(huì)直接segment fault。

import sys
  import ctypes
  so = ctypes.CDLL("./lib/libbase.so")
  p = so.create_handler()
  feature = so.extract_feature(p)
  so.destroy(p)

解決

這段代碼中p是int類型，由void*自動(dòng)轉(zhuǎn)來(lái)，在ctyeps中這種轉(zhuǎn)型本身是沒(méi)問(wèn)題的。segment fault發(fā)生在extract_feature函數(shù)調(diào)用中，問(wèn)題應(yīng)當(dāng)出在參數(shù)上，回傳的handler已經(jīng)不是原來(lái)的pointer了，導(dǎo)致訪問(wèn)指針出錯(cuò)。

查閱ctypes的文檔后，發(fā)現(xiàn)ctypes可以聲明so庫(kù)中函數(shù)的參數(shù)，返回類型。試了試，顯示聲明后問(wèn)題得到了解決，證明我們的猜想是對(duì)的，確實(shí)指針發(fā)生了變化。修改后代碼如下:

import sys
  import ctypes
  so = ctypes.CDLL("./lib/libbase.so")
  so.create_handler.restype=ctypes.c_void_p
  so.extract_feature.argtypes=[ctypes.c_void_p]
  so.destroy.argtypes=[ctypes.c_void_p]
  p = so.create_handler()
  feature = so.extract_feature(p)
  so.destroy(p)

結(jié)論:

ctypes中傳遞指針類型參數(shù)需要顯示聲明c函數(shù)的參數(shù)，返回類型。

總結(jié)

以上所述是小編給大家介紹的python中使用ctypes調(diào)用so傳參設(shè)置遇到的問(wèn)題及解決方法,希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)億速云網(wǎng)站的支持！
如果你覺(jué)得本文對(duì)你有幫助，歡迎轉(zhuǎn)載，煩請(qǐng)注明出處，謝謝！