270 lines
6.6 KiB
C
270 lines
6.6 KiB
C
|
/*
|
||
|
* Copyright (c) 2021 Yubico AB. All rights reserved.
|
||
|
* Use of this source code is governed by a BSD-style
|
||
|
* license that can be found in the LICENSE file.
|
||
|
*/
|
||
|
|
||
|
#include <sys/types.h>
|
||
|
|
||
|
#include <linux/hidraw.h>
|
||
|
#include <linux/input.h>
|
||
|
|
||
|
#include <assert.h>
|
||
|
#include <errno.h>
|
||
|
#include <libudev.h>
|
||
|
#include <stdlib.h>
|
||
|
|
||
|
#include "mutator_aux.h"
|
||
|
|
||
|
struct udev {
|
||
|
int magic;
|
||
|
};
|
||
|
|
||
|
struct udev_enumerate {
|
||
|
int magic;
|
||
|
struct udev_list_entry *list_entry;
|
||
|
};
|
||
|
|
||
|
struct udev_list_entry {
|
||
|
int magic;
|
||
|
};
|
||
|
|
||
|
struct udev_device {
|
||
|
int magic;
|
||
|
struct udev_device *parent;
|
||
|
};
|
||
|
|
||
|
#define UDEV_MAGIC 0x584492cc
|
||
|
#define UDEV_DEVICE_MAGIC 0x569180dd
|
||
|
#define UDEV_LIST_ENTRY_MAGIC 0x497422ee
|
||
|
#define UDEV_ENUM_MAGIC 0x583570ff
|
||
|
|
||
|
#define ASSERT_TYPE(x, m) assert((x) != NULL && (x)->magic == (m))
|
||
|
#define ASSERT_UDEV(x) ASSERT_TYPE((x), UDEV_MAGIC)
|
||
|
#define ASSERT_UDEV_ENUM(x) ASSERT_TYPE((x), UDEV_ENUM_MAGIC)
|
||
|
#define ASSERT_UDEV_LIST_ENTRY(x) ASSERT_TYPE((x), UDEV_LIST_ENTRY_MAGIC)
|
||
|
#define ASSERT_UDEV_DEVICE(x) ASSERT_TYPE((x), UDEV_DEVICE_MAGIC)
|
||
|
|
||
|
static const char *uevent;
|
||
|
static const struct blob *report_descriptor;
|
||
|
|
||
|
struct udev *__wrap_udev_new(void);
|
||
|
struct udev_device *__wrap_udev_device_get_parent_with_subsystem_devtype(
|
||
|
struct udev_device *, const char *, const char *);
|
||
|
struct udev_device *__wrap_udev_device_new_from_syspath(struct udev *,
|
||
|
const char *);
|
||
|
struct udev_enumerate *__wrap_udev_enumerate_new(struct udev *);
|
||
|
struct udev_list_entry *__wrap_udev_enumerate_get_list_entry(
|
||
|
struct udev_enumerate *);
|
||
|
struct udev_list_entry *__wrap_udev_list_entry_get_next(
|
||
|
struct udev_list_entry *);
|
||
|
const char *__wrap_udev_device_get_sysattr_value(struct udev_device *,
|
||
|
const char *);
|
||
|
const char *__wrap_udev_list_entry_get_name(struct udev_list_entry *);
|
||
|
const char *__wrap_udev_device_get_devnode(struct udev_device *);
|
||
|
const char *__wrap_udev_device_get_sysnum(struct udev_device *);
|
||
|
int __wrap_udev_enumerate_add_match_subsystem(struct udev_enumerate *,
|
||
|
const char *);
|
||
|
int __wrap_udev_enumerate_scan_devices(struct udev_enumerate *);
|
||
|
int __wrap_ioctl(int, unsigned long , ...);
|
||
|
void __wrap_udev_device_unref(struct udev_device *);
|
||
|
void __wrap_udev_enumerate_unref(struct udev_enumerate *);
|
||
|
void __wrap_udev_unref(struct udev *);
|
||
|
void set_udev_parameters(const char *, const struct blob *);
|
||
|
|
||
|
struct udev_device *
|
||
|
__wrap_udev_device_get_parent_with_subsystem_devtype(struct udev_device *child,
|
||
|
const char *subsystem, const char *devtype)
|
||
|
{
|
||
|
ASSERT_UDEV_DEVICE(child);
|
||
|
fido_log_debug("%s", subsystem); /* XXX consume */
|
||
|
fido_log_debug("%s", devtype); /* XXX consume */
|
||
|
if (child->parent != NULL)
|
||
|
return child->parent;
|
||
|
if ((child->parent = calloc(1, sizeof(*child->parent))) == NULL)
|
||
|
return NULL;
|
||
|
child->parent->magic = UDEV_DEVICE_MAGIC;
|
||
|
|
||
|
return child->parent;
|
||
|
}
|
||
|
|
||
|
const char *
|
||
|
__wrap_udev_device_get_sysattr_value(struct udev_device *udev_device,
|
||
|
const char *sysattr)
|
||
|
{
|
||
|
ASSERT_UDEV_DEVICE(udev_device);
|
||
|
if (uniform_random(400) < 1)
|
||
|
return NULL;
|
||
|
if (!strcmp(sysattr, "manufacturer") || !strcmp(sysattr, "product"))
|
||
|
return "product info"; /* XXX randomise? */
|
||
|
else if (!strcmp(sysattr, "uevent"))
|
||
|
return uevent;
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
const char *
|
||
|
__wrap_udev_list_entry_get_name(struct udev_list_entry *entry)
|
||
|
{
|
||
|
ASSERT_UDEV_LIST_ENTRY(entry);
|
||
|
return uniform_random(400) < 1 ? NULL : "name"; /* XXX randomise? */
|
||
|
}
|
||
|
|
||
|
struct udev_device *
|
||
|
__wrap_udev_device_new_from_syspath(struct udev *udev, const char *syspath)
|
||
|
{
|
||
|
struct udev_device *udev_device;
|
||
|
|
||
|
ASSERT_UDEV(udev);
|
||
|
fido_log_debug("%s", syspath);
|
||
|
if ((udev_device = calloc(1, sizeof(*udev_device))) == NULL)
|
||
|
return NULL;
|
||
|
udev_device->magic = UDEV_DEVICE_MAGIC;
|
||
|
|
||
|
return udev_device;
|
||
|
}
|
||
|
|
||
|
const char *
|
||
|
__wrap_udev_device_get_devnode(struct udev_device *udev_device)
|
||
|
{
|
||
|
ASSERT_UDEV_DEVICE(udev_device);
|
||
|
return uniform_random(400) < 1 ? NULL : "/dev/zero";
|
||
|
}
|
||
|
|
||
|
const char *
|
||
|
__wrap_udev_device_get_sysnum(struct udev_device *udev_device)
|
||
|
{
|
||
|
ASSERT_UDEV_DEVICE(udev_device);
|
||
|
return uniform_random(400) < 1 ? NULL : "101010"; /* XXX randomise? */
|
||
|
}
|
||
|
|
||
|
void
|
||
|
__wrap_udev_device_unref(struct udev_device *udev_device)
|
||
|
{
|
||
|
ASSERT_UDEV_DEVICE(udev_device);
|
||
|
if (udev_device->parent) {
|
||
|
ASSERT_UDEV_DEVICE(udev_device->parent);
|
||
|
free(udev_device->parent);
|
||
|
}
|
||
|
free(udev_device);
|
||
|
}
|
||
|
|
||
|
struct udev *
|
||
|
__wrap_udev_new(void)
|
||
|
{
|
||
|
struct udev *udev;
|
||
|
|
||
|
if ((udev = calloc(1, sizeof(*udev))) == NULL)
|
||
|
return NULL;
|
||
|
udev->magic = UDEV_MAGIC;
|
||
|
|
||
|
return udev;
|
||
|
}
|
||
|
|
||
|
struct udev_enumerate *
|
||
|
__wrap_udev_enumerate_new(struct udev *udev)
|
||
|
{
|
||
|
struct udev_enumerate *udev_enum;
|
||
|
|
||
|
ASSERT_UDEV(udev);
|
||
|
if ((udev_enum = calloc(1, sizeof(*udev_enum))) == NULL)
|
||
|
return NULL;
|
||
|
udev_enum->magic = UDEV_ENUM_MAGIC;
|
||
|
|
||
|
return udev_enum;
|
||
|
}
|
||
|
|
||
|
int
|
||
|
__wrap_udev_enumerate_add_match_subsystem(struct udev_enumerate *udev_enum,
|
||
|
const char *subsystem)
|
||
|
{
|
||
|
ASSERT_UDEV_ENUM(udev_enum);
|
||
|
fido_log_debug("%s:", subsystem);
|
||
|
return uniform_random(400) < 1 ? -EINVAL : 0;
|
||
|
}
|
||
|
|
||
|
int
|
||
|
__wrap_udev_enumerate_scan_devices(struct udev_enumerate *udev_enum)
|
||
|
{
|
||
|
ASSERT_UDEV_ENUM(udev_enum);
|
||
|
return uniform_random(400) < 1 ? -EINVAL : 0;
|
||
|
}
|
||
|
|
||
|
struct udev_list_entry *
|
||
|
__wrap_udev_enumerate_get_list_entry(struct udev_enumerate *udev_enum)
|
||
|
{
|
||
|
ASSERT_UDEV_ENUM(udev_enum);
|
||
|
if ((udev_enum->list_entry = calloc(1,
|
||
|
sizeof(*udev_enum->list_entry))) == NULL)
|
||
|
return NULL;
|
||
|
udev_enum->list_entry->magic = UDEV_LIST_ENTRY_MAGIC;
|
||
|
|
||
|
return udev_enum->list_entry;
|
||
|
}
|
||
|
|
||
|
struct udev_list_entry *
|
||
|
__wrap_udev_list_entry_get_next(struct udev_list_entry *udev_list_entry)
|
||
|
{
|
||
|
ASSERT_UDEV_LIST_ENTRY(udev_list_entry);
|
||
|
return uniform_random(400) < 1 ? NULL : udev_list_entry;
|
||
|
}
|
||
|
|
||
|
void
|
||
|
__wrap_udev_enumerate_unref(struct udev_enumerate *udev_enum)
|
||
|
{
|
||
|
ASSERT_UDEV_ENUM(udev_enum);
|
||
|
if (udev_enum->list_entry)
|
||
|
ASSERT_UDEV_LIST_ENTRY(udev_enum->list_entry);
|
||
|
free(udev_enum->list_entry);
|
||
|
free(udev_enum);
|
||
|
}
|
||
|
|
||
|
void
|
||
|
__wrap_udev_unref(struct udev *udev)
|
||
|
{
|
||
|
ASSERT_UDEV(udev);
|
||
|
free(udev);
|
||
|
}
|
||
|
|
||
|
int
|
||
|
__wrap_ioctl(int fd, unsigned long request, ...)
|
||
|
{
|
||
|
va_list ap;
|
||
|
struct hidraw_report_descriptor *hrd;
|
||
|
|
||
|
(void)fd;
|
||
|
|
||
|
if (uniform_random(400) < 1) {
|
||
|
errno = EINVAL;
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
va_start(ap, request);
|
||
|
|
||
|
switch (request) {
|
||
|
case IOCTL_REQ(HIDIOCGRDESCSIZE):
|
||
|
*va_arg(ap, int *) = (int)report_descriptor->len;
|
||
|
break;
|
||
|
case IOCTL_REQ(HIDIOCGRDESC):
|
||
|
hrd = va_arg(ap, struct hidraw_report_descriptor *);
|
||
|
assert(hrd->size == report_descriptor->len);
|
||
|
memcpy(hrd->value, report_descriptor->body, hrd->size);
|
||
|
break;
|
||
|
default:
|
||
|
warnx("%s: unknown request 0x%lx", __func__, request);
|
||
|
abort();
|
||
|
}
|
||
|
|
||
|
va_end(ap);
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
void
|
||
|
set_udev_parameters(const char *uevent_ptr,
|
||
|
const struct blob *report_descriptor_ptr)
|
||
|
{
|
||
|
uevent = uevent_ptr;
|
||
|
report_descriptor = report_descriptor_ptr;
|
||
|
}
|