Класс Monkey Patching, полученный из ctypes. Union не работает

1

Я пытаюсь "патч обезьяны" класс, полученный из Python ctypes "Union", но я не могу этого сделать - получить странные ошибки и иногда seg-faults. То же самое хорошо работает при построении ctypes "Structure".

Я сузил это до самого простого возможного тестового примера, который я публикую ниже. Я использую Python 3.6.4. Мне интересно, что я делаю что-то неправильно (или есть проблема с реализацией ctypes "Union"?). См. Код ниже и соответствующий вывод.

import ctypes

def display(self):
  """ A new kind of display """
  return f'Print Type #2: ( {self.y1}, {self.y2} )'

class MyStruct(ctypes.Structure):
  _fields_ = [ 
      ('y1', ctypes.c_uint32),
      ('y2', ctypes.c_uint32)
      ] 

  def __str__(self):
    return f'Print Type #1: [ {self.y1}, {self.y2} ]'

class MyUnion(ctypes.Union):
  _fields_ = [ 
      ('y1', ctypes.c_uint32),
      ('y2', ctypes.c_uint32)
      ] 

  def __str__(self):
    return f'Print Type #1: [ {self.y1}, {self.y2} ]'

if __name__ == '__main__':

  a = MyStruct()
  a.y1 = 10
  a.y2 = 20

  print('Using Structure:')
  print('----------------')
  print(a)
  print('Original :', MyStruct.__str__)
  # monkey patch  __str__ with a different function.
  MyStruct.__str__ = display
  print('Patched :', MyStruct.__str__)
  print('Patched (dict) :', MyStruct.__dict__['__str__'])
  print(a)

  a = MyUnion()
  a.y1 = 10
  a.y2 = 20

  print('Using Union:')
  print('------------')
  print(a)
  print('Original :', MyUnion.__str__)
  # monkey patch  __str__ with a different function.
  MyUnion.__str__ = display
  print('Patched :', MyUnion.__str__)
  print('Patched (dict) :', MyUnion.__dict__['__str__'])
  print(a)

Здесь вывод, когда я запускаю программу.

Using Structure:
----------------
Print Type #1: [ 10, 20 ]
Original : <function MyStruct.__str__ at 0x7fdf89d02e18>
Patched : <function display at 0x7fdf8b0ebe18>
Patched (dict) : <function display at 0x7fdf8b0ebe18>
Print Type #2: ( 10, 20 )
Using Union:
------------
Print Type #1: [ 20, 20 ]
Original : <function MyUnion.__str__ at 0x7fdf89d02f28>
Patched : <function MyUnion.__str__ at 0x7fdf89d02f28>
Patched (dict) : <function display at 0x7fdf8b0ebe18>
Traceback (most recent call last):
  File "ctypes_bug.py", line 54, in <module>
    print(a)
TypeError: 'managedbuffer' object is not callable

Ясно, что я могу "запланировать" __str__ когда соответствующий объект Python был получен из "Structure", но я не могу "патчить" __str__ когда соответствующий объект Python был получен из "Союза".

Интересно, что MyUnion.__dict__[__str__] и MyUnion.__str__ показывают разные результаты - что тоже странно.

Что-то я здесь делаю неправильно? Я высоко ценю любую помощь или прозрение!

Теги:
ctypes
monkeypatching

1 ответ

1

Я думаю, что здесь есть реальная ошибка CPython. Реализация __setattr__ для объектов типа для типов struct использует PyType_Type.tp_setattro:

static int
PyCStructType_setattro(PyObject *self, PyObject *key, PyObject *value)
{
    /* XXX Should we disallow deleting _fields_? */
    if (-1 == PyType_Type.tp_setattro(self, key, value))
        return -1;

    if (value && PyUnicode_Check(key) &&
        _PyUnicode_EqualToASCIIString(key, "_fields_"))
        return PyCStructUnionType_update_stgdict(self, value, 1);
    return 0;
}

но для объектов типа для типов union используется PyObject_GenericSetAttr:

static int
UnionType_setattro(PyObject *self, PyObject *key, PyObject *value)
{
    /* XXX Should we disallow deleting _fields_? */
    if (-1 == PyObject_GenericSetAttr(self, key, value))
        return -1;

    if (PyUnicode_Check(key) &&
        _PyUnicode_EqualToASCIIString(key, "_fields_"))
        return PyCStructUnionType_update_stgdict(self, value, 0);
    return 0;
}

Использование PyType_Type.tp_setattro необходимо для обновления слотов типа и аннулирования кэша атрибутов внутреннего типа. PyObject_GenericSetAttr не знает, что он должен делать любую из этих вещей, вызывая потенциальное повреждение памяти из-за кэшированных атрибутов "зомби". Похоже, что одна и та же ошибка была исправлена в начале 2008 года для структур, но они забыли обработать профсоюзы.

  • 0
    Я просто открывал редактор ответов, чтобы заявить то же самое. UnionType является метаклассом, поэтому он должен использовать PyType_Type.tp_setattro .
  • 0
    Кроме того, лучшая ссылка для набора изменений: github.com/python/cpython/commit/…
Показать ещё 7 комментариев

Ещё вопросы

Сообщество Overcoder
Наверх
Меню