def ioctl(self, path, cmd, arg, fh, flags, data):
M_IOWR = IOWR(ord('M'), 1, ctypes.c_uint32)
if cmd == M_IOWR:
inbuf = ctypes.create_string_buffer(4)
ctypes.memmove(inbuf, data, 4)
data_in = struct.unpack('<I', inbuf)[0]
data_out = data_in + 1
outbuf = struct.pack('<I', data_out)
ctypes.memmove(data, outbuf, 4)
else:
raise FuseOSError(ENOTTY)
return 0
def __init__(self, device_name="/dev/skunk"):
self.device_name = device_name
self.func_call_ioctl_num = IOWR(0xEE, 0, ctypes.c_char_p)
self.mock_create_ioctl_num = IOW(0xEE, 1, ctypes.c_char_p)
self.mock_destroy_ioctl_num = IO(0xEE, 2)
self.mock_return_values = skunk_pb2.ReturnValueMock()
class DRM_V3D(object):
def __init__(self, path='/dev/dri/by-path/platform-fec00000.v3d-card'):
self.fd = os.open(path, os.O_RDWR)
def close(self):
if self.fd is not None:
os.close(self.fd)
self.fd = None
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, traceback):
self.close()
return exc_value is None
# Derived from linux/include/uapi/drm/drm.h
DRM_IOCTL_BASE = ord('d')
DRM_COMMAND_BASE = 0x40
DRM_GEM_CLOSE = 0x09
# Derived from linux/include/uapi/drm/v3d_drm.h
DRM_V3D_WAIT_BO = DRM_COMMAND_BASE + 0x01
DRM_V3D_CREATE_BO = DRM_COMMAND_BASE + 0x02
DRM_V3D_MMAP_BO = DRM_COMMAND_BASE + 0x03
DRM_V3D_GET_PARAM = DRM_COMMAND_BASE + 0x04
DRM_V3D_SUBMIT_CSD = DRM_COMMAND_BASE + 0x07
V3D_PARAM_V3D_UIFCFG = 0
V3D_PARAM_V3D_HUB_IDENT1 = 1
V3D_PARAM_V3D_HUB_IDENT2 = 2
V3D_PARAM_V3D_HUB_IDENT3 = 3
V3D_PARAM_V3D_CORE0_IDENT0 = 4
V3D_PARAM_V3D_CORE0_IDENT1 = 5
V3D_PARAM_V3D_CORE0_IDENT2 = 6
V3D_PARAM_SUPPORTS_TFU = 7
V3D_PARAM_SUPPORTS_CSD = 8
class st_gem_close(Structure):
_fields_ = [
('handle', c_uint32),
('pad', c_uint32),
]
class st_v3d_wait_bo(Structure):
_fields_ = [
('handle', c_uint32),
('pad', c_uint32),
('timeout_ns', c_uint64),
]
class st_v3d_create_bo(Structure):
_fields_ = [
('size', c_uint32),
('flags', c_uint32),
('handle', c_uint32),
('offset', c_uint32),
]
class st_v3d_mmap_bo(Structure):
_fields_ = [
('handle', c_uint32),
('flags', c_uint32),
('offset', c_uint64),
]
class st_v3d_get_param(Structure):
_fields_ = [
('param', c_uint32),
('pad', c_uint32),
('value', c_uint64),
]
class st_v3d_submit_csd(Structure):
_fields_ = [
('cfg', c_uint32 * 7),
('coef', c_uint32 * 4),
('bo_handles', c_uint64),
('bo_handle_count', c_uint32),
('in_sync', c_uint32),
('out_sync', c_uint32),
]
IOCTL_GEM_CLOSE = IOW(DRM_IOCTL_BASE, DRM_GEM_CLOSE, st_gem_close)
IOCTL_V3D_WAIT_BO = IOWR(DRM_IOCTL_BASE, DRM_V3D_WAIT_BO, st_v3d_wait_bo)
IOCTL_V3D_CREATE_BO = IOWR(DRM_IOCTL_BASE, DRM_V3D_CREATE_BO,
st_v3d_create_bo)
IOCTL_V3D_MMAP_BO = IOWR(DRM_IOCTL_BASE, DRM_V3D_MMAP_BO, st_v3d_mmap_bo)
IOCTL_V3D_GET_PARAM = IOWR(DRM_IOCTL_BASE, DRM_V3D_GET_PARAM,
st_v3d_get_param)
IOCTL_V3D_SUBMIT_CSD = IOW(DRM_IOCTL_BASE, DRM_V3D_SUBMIT_CSD,
st_v3d_submit_csd)
def gem_close(self, handle):
st = self.st_gem_close(
handle=handle,
pad=0,
)
ioctl(self.fd, self.IOCTL_GEM_CLOSE, st)
def v3d_wait_bo(self, handle, timeout_ns):
st = self.st_v3d_wait_bo(
handle=handle,
pad=0,
timeout_ns=timeout_ns,
)
ioctl(self.fd, self.IOCTL_V3D_WAIT_BO, st)
def v3d_create_bo(self, size, flags=0):
st = self.st_v3d_create_bo(
size=size,
flags=flags,
handle=0,
offset=0,
)
ioctl(self.fd, self.IOCTL_V3D_CREATE_BO, st)
return st.handle, st.offset
def v3d_mmap_bo(self, handle, flags=0):
st = self.st_v3d_mmap_bo(
handle=handle,
flags=flags,
offset=0,
)
ioctl(self.fd, self.IOCTL_V3D_MMAP_BO, st)
return st.offset
def v3d_get_param(self, param):
st = self.st_v3d_get_param(
param=param,
pad=0,
value=0,
)
ioctl(self.fd, self.IOCTL_V3D_GET_PARAM, st)
return st.value
def v3d_submit_csd(self, cfg, coef, bo_handles, bo_handle_count, in_sync,
out_sync):
st = self.st_v3d_submit_csd(
# XXX: Dirty hack!
cfg=(c_uint32 * 7)(*cfg),
coef=(c_uint32 * 4)(*coef),
bo_handles=bo_handles,
bo_handle_count=bo_handle_count,
in_sync=in_sync,
out_sync=out_sync,
)
ioctl(self.fd, self.IOCTL_V3D_SUBMIT_CSD, st)
def cmd(num):
return IOWR(ord('h'), num, hevd_io)