/*

Simple DirectMedia Layer

Copyright (C) 1997-2011 Sam Lantinga <slouken@libsdl.org>

This software is provided 'as-is', without any express or implied

warranty. In no event will the authors be held liable for any damages

arising from the use of this software.

Permission is granted to anyone to use this software for any purpose,

including commercial applications, and to alter it and redistribute it

freely, subject to the following restrictions:

1. The origin of this software must not be misrepresented; you must not

claim that you wrote the original software. If you use this software

in a product, an acknowledgment in the product documentation would be

appreciated but is not required.

2. Altered source versions must be plainly marked as such, and must not be

misrepresented as being the original software.

3. This notice may not be removed or altered from any source distribution.

*/

#include "SDL_config.h"

#ifdef SDL_JOYSTICK_USBHID

/*

* Joystick driver for the uhid(4) interface found in OpenBSD,

* NetBSD and FreeBSD.

*

* Maintainer: <vedge at csoft.org>

*/

#include <sys/param.h>

#include <unistd.h>

#include <fcntl.h>

#include <errno.h>

#ifndef __FreeBSD_kernel_version

#define __FreeBSD_kernel_version __FreeBSD_version

#endif

#if defined(HAVE_USB_H)

#include <usb.h>

#endif

#ifdef __DragonFly__

#include <bus/usb/usb.h>

#include <bus/usb/usbhid.h>

#else

#include <dev/usb/usb.h>

#include <dev/usb/usbhid.h>

#endif

#if defined(HAVE_USBHID_H)

#include <usbhid.h>

#elif defined(HAVE_LIBUSB_H)

#include <libusb.h>

#elif defined(HAVE_LIBUSBHID_H)

#include <libusbhid.h>

#endif

#if defined(__FREEBSD__) || defined(__FreeBSD_kernel__)

#ifndef __DragonFly__

#include <osreldate.h>

#endif

#if __FreeBSD_kernel_version > 800063

#include <dev/usb/usb_ioctl.h>

#endif

#include <sys/joystick.h>

#endif

#if SDL_JOYSTICK_USBHID_MACHINE_JOYSTICK_H

#include <machine/joystick.h>

#endif

#include "SDL_joystick.h"

#include "../SDL_sysjoystick.h"

#include "../SDL_joystick_c.h"

#define MAX_UHID_JOYS 4

#define MAX_JOY_JOYS 2

#define MAX_JOYS (MAX_UHID_JOYS + MAX_JOY_JOYS)

struct report

{

#if defined(__FREEBSD__) && (__FreeBSD_kernel_version > 800063)

struct usb_gen_descriptor *buf; /* Buffer */

#else

size_t size; /* Buffer size */

int rid; /* Report ID */

enum

{

SREPORT_UNINIT,

SREPORT_CLEAN,

SREPORT_DIRTY

} status;

};

static struct

{

int uhid_report;

hid_kind_t kind;

const char *name;

} const repinfo[] = {

{UHID_INPUT_REPORT, hid_input, "input"},

{UHID_OUTPUT_REPORT, hid_output, "output"},

{UHID_FEATURE_REPORT, hid_feature, "feature"}

};

enum

{

REPORT_INPUT = 0,

REPORT_OUTPUT = 1,

REPORT_FEATURE = 2

};

enum

{

JOYAXE_X,

JOYAXE_Y,

JOYAXE_Z,

JOYAXE_SLIDER,

JOYAXE_WHEEL,

JOYAXE_RX,

JOYAXE_RY,

JOYAXE_RZ,

JOYAXE_count

};

struct joystick_hwdata

{

int fd;

char *path;

enum

{

BSDJOY_UHID, /* uhid(4) */

BSDJOY_JOY /* joy(4) */

} type;

struct report_desc *repdesc;

struct report inreport;

int axis_map[JOYAXE_count]; /* map present JOYAXE_* to 0,1,.. */

};

static char *joynames[MAX_JOYS];

static char *joydevnames[MAX_JOYS];

static int report_alloc(struct report *, struct report_desc *, int);

static void report_free(struct report *);

#elif (defined(__FREEBSD__) && (__FreeBSD_kernel_version > 800063))

#define REP_BUF_DATA(rep) ((rep)->buf->ugd_data)

#else

#define REP_BUF_DATA(rep) ((rep)->buf->data)

#endif

int

SDL_SYS_JoystickInit(void)

{

char s[16];

int i, fd;

SDL_numjoysticks = 0;

SDL_memset(joynames, 0, sizeof(joynames));

SDL_memset(joydevnames, 0, sizeof(joydevnames));

for (i = 0; i < MAX_UHID_JOYS; i++) {

SDL_Joystick nj;

SDL_snprintf(s, SDL_arraysize(s), "/dev/uhid%d", i);

nj.index = SDL_numjoysticks;

joynames[nj.index] = strdup(s);

if (SDL_SYS_JoystickOpen(&nj) == 0) {

SDL_SYS_JoystickClose(&nj);

SDL_numjoysticks++;

} else {

SDL_free(joynames[nj.index]);

joynames[nj.index] = NULL;

}

}

for (i = 0; i < MAX_JOY_JOYS; i++) {

SDL_snprintf(s, SDL_arraysize(s), "/dev/joy%d", i);

fd = open(s, O_RDONLY);

if (fd != -1) {

joynames[SDL_numjoysticks++] = strdup(s);

close(fd);

}

}

/* Read the default USB HID usage table. */

hid_init(NULL);

return (SDL_numjoysticks);

}

const char *

SDL_SYS_JoystickName(int index)

{

if (joydevnames[index] != NULL) {

return (joydevnames[index]);

}

return (joynames[index]);

}

static int

usage_to_joyaxe(unsigned usage)

{

int joyaxe;

switch (usage) {

case HUG_X:

joyaxe = JOYAXE_X;

break;

case HUG_Y:

joyaxe = JOYAXE_Y;

break;

case HUG_Z:

joyaxe = JOYAXE_Z;

break;

case HUG_SLIDER:

joyaxe = JOYAXE_SLIDER;

break;

case HUG_WHEEL:

joyaxe = JOYAXE_WHEEL;

break;

case HUG_RX:

joyaxe = JOYAXE_RX;

break;

case HUG_RY:

joyaxe = JOYAXE_RY;

break;

case HUG_RZ:

joyaxe = JOYAXE_RZ;

break;

default:

joyaxe = -1;

}

return joyaxe;

}

static unsigned

hatval_to_sdl(Sint32 hatval)

{

static const unsigned hat_dir_map[8] = {

SDL_HAT_UP, SDL_HAT_RIGHTUP, SDL_HAT_RIGHT, SDL_HAT_RIGHTDOWN,

SDL_HAT_DOWN, SDL_HAT_LEFTDOWN, SDL_HAT_LEFT, SDL_HAT_LEFTUP

};

unsigned result;

if ((hatval & 7) == hatval)

result = hat_dir_map[hatval];

else

result = SDL_HAT_CENTERED;

return result;

}

int

SDL_SYS_JoystickOpen(SDL_Joystick * joy)

{

char *path = joynames[joy->index];

struct joystick_hwdata *hw;

struct hid_item hitem;

struct hid_data *hdata;

struct report *rep;

int fd;

int i;

fd = open(path, O_RDONLY);

if (fd == -1) {

SDL_SetError("%s: %s", path, strerror(errno));

return (-1);

}

hw = (struct joystick_hwdata *)

SDL_malloc(sizeof(struct joystick_hwdata));

if (hw == NULL) {

SDL_OutOfMemory();

close(fd);

return (-1);

}

joy->hwdata = hw;

hw->fd = fd;

hw->path = strdup(path);

if (!SDL_strncmp(path, "/dev/joy", 8)) {

hw->type = BSDJOY_JOY;

joy->naxes = 2;

joy->nbuttons = 2;

joy->nhats = 0;

joy->nballs = 0;

joydevnames[joy->index] = strdup("Gameport joystick");

goto usbend;

} else {

hw->type = BSDJOY_UHID;

}

{

int ax;

for (ax = 0; ax < JOYAXE_count; ax++)

hw->axis_map[ax] = -1;

}

hw->repdesc = hid_get_report_desc(fd);

if (hw->repdesc == NULL) {

SDL_SetError("%s: USB_GET_REPORT_DESC: %s", hw->path,

strerror(errno));

goto usberr;

}

rep = &hw->inreport;

#if defined(__FREEBSD__) && (__FreeBSD_kernel_version > 800063) || defined(__FreeBSD_kernel__)

rep->rid = hid_get_report_id(fd);

if (rep->rid < 0) {

#else

if (ioctl(fd, USB_GET_REPORT_ID, &rep->rid) < 0) {

#endif

rep->rid = -1; /* XXX */

}

if (report_alloc(rep, hw->repdesc, REPORT_INPUT) < 0) {

goto usberr;

}

if (rep->size <= 0) {

SDL_SetError("%s: Input report descriptor has invalid length",

hw->path);

goto usberr;

}

#if defined(USBHID_NEW) || (defined(__FREEBSD__) && __FreeBSD_kernel_version >= 500111) || defined(__FreeBSD_kernel__)

hdata = hid_start_parse(hw->repdesc, 1 << hid_input, rep->rid);

#else

hdata = hid_start_parse(hw->repdesc, 1 << hid_input);

#endif

if (hdata == NULL) {

SDL_SetError("%s: Cannot start HID parser", hw->path);

goto usberr;

}

joy->naxes = 0;

joy->nbuttons = 0;

joy->nhats = 0;

joy->nballs = 0;

for (i = 0; i < JOYAXE_count; i++)

hw->axis_map[i] = -1;

while (hid_get_item(hdata, &hitem) > 0) {

char *sp;

const char *s;

switch (hitem.kind) {

case hid_collection:

switch (HID_PAGE(hitem.usage)) {

case HUP_GENERIC_DESKTOP:

switch (HID_USAGE(hitem.usage)) {

case HUG_JOYSTICK:

case HUG_GAME_PAD:

s = hid_usage_in_page(hitem.usage);

sp = SDL_malloc(SDL_strlen(s) + 5);

SDL_snprintf(sp, SDL_strlen(s) + 5, "%s (%d)",

s, joy->index);

joydevnames[joy->index] = sp;

}

}

break;

case hid_input:

switch (HID_PAGE(hitem.usage)) {

case HUP_GENERIC_DESKTOP:

{

unsigned usage = HID_USAGE(hitem.usage);

int joyaxe = usage_to_joyaxe(usage);

if (joyaxe >= 0) {

hw->axis_map[joyaxe] = 1;

} else if (usage == HUG_HAT_SWITCH) {

joy->nhats++;

}

break;

}

case HUP_BUTTON:

joy->nbuttons++;

break;

default:

break;

}

break;

default:

break;

}

}

hid_end_parse(hdata);

for (i = 0; i < JOYAXE_count; i++)

if (hw->axis_map[i] > 0)

hw->axis_map[i] = joy->naxes++;

usbend:

/* The poll blocks the event thread. */

fcntl(fd, F_SETFL, O_NONBLOCK);

return (0);

usberr:

close(hw->fd);

SDL_free(hw->path);

SDL_free(hw);

return (-1);

}

void

SDL_SYS_JoystickUpdate(SDL_Joystick * joy)

{

struct hid_item hitem;

struct hid_data *hdata;

struct report *rep;

int nbutton, naxe = -1;

Sint32 v;

#if defined(__FREEBSD__) || SDL_JOYSTICK_USBHID_MACHINE_JOYSTICK_H || defined(__FreeBSD_kernel__)

struct joystick gameport;

static int x, y, xmin = 0xffff, ymin = 0xffff, xmax = 0, ymax = 0;

if (joy->hwdata->type == BSDJOY_JOY) {

if (read(joy->hwdata->fd, &gameport, sizeof gameport) !=

sizeof gameport)

return;

if (abs(x - gameport.x) > 8) {

x = gameport.x;

if (x < xmin) {

xmin = x;

}

if (x > xmax) {

xmax = x;

}

if (xmin == xmax) {

xmin--;

xmax++;

}

v = (Sint32) x;

v -= (xmax + xmin + 1) / 2;

v *= 32768 / ((xmax - xmin + 1) / 2);

SDL_PrivateJoystickAxis(joy, 0, v);

}

if (abs(y - gameport.y) > 8) {

y = gameport.y;

if (y < ymin) {

ymin = y;

}

if (y > ymax) {

ymax = y;

}

if (ymin == ymax) {

ymin--;

ymax++;

}

v = (Sint32) y;

v -= (ymax + ymin + 1) / 2;

v *= 32768 / ((ymax - ymin + 1) / 2);

SDL_PrivateJoystickAxis(joy, 1, v);

}

if (gameport.b1 != joy->buttons[0]) {

SDL_PrivateJoystickButton(joy, 0, gameport.b1);

}

if (gameport.b2 != joy->buttons[1]) {

SDL_PrivateJoystickButton(joy, 1, gameport.b2);

}

return;

}

#endif /* defined(__FREEBSD__) || SDL_JOYSTICK_USBHID_MACHINE_JOYSTICK_H */

rep = &joy->hwdata->inreport;

if (read(joy->hwdata->fd, REP_BUF_DATA(rep), rep->size) != rep->size) {

return;

}

#if defined(USBHID_NEW) || (defined(__FREEBSD__) && __FreeBSD_kernel_version >= 500111) || defined(__FreeBSD_kernel__)

hdata = hid_start_parse(joy->hwdata->repdesc, 1 << hid_input, rep->rid);

#else

hdata = hid_start_parse(joy->hwdata->repdesc, 1 << hid_input);

#endif

if (hdata == NULL) {

fprintf(stderr, "%s: Cannot start HID parser\n", joy->hwdata->path);

return;

}

for (nbutton = 0; hid_get_item(hdata, &hitem) > 0;) {

switch (hitem.kind) {

case hid_input:

switch (HID_PAGE(hitem.usage)) {

case HUP_GENERIC_DESKTOP:

{

unsigned usage = HID_USAGE(hitem.usage);

int joyaxe = usage_to_joyaxe(usage);

if (joyaxe >= 0) {

naxe = joy->hwdata->axis_map[joyaxe];

/* scaleaxe */

v = (Sint32) hid_get_data(REP_BUF_DATA(rep), &hitem);

v -= (hitem.logical_maximum +

hitem.logical_minimum + 1) / 2;

v *= 32768 /

((hitem.logical_maximum -

hitem.logical_minimum + 1) / 2);

if (v != joy->axes[naxe]) {

SDL_PrivateJoystickAxis(joy, naxe, v);

}

} else if (usage == HUG_HAT_SWITCH) {

v = (Sint32) hid_get_data(REP_BUF_DATA(rep), &hitem);

SDL_PrivateJoystickHat(joy, 0,

hatval_to_sdl(v) -

hitem.logical_minimum);

}

break;

}

case HUP_BUTTON:

v = (Sint32) hid_get_data(REP_BUF_DATA(rep), &hitem);

if (joy->buttons[nbutton] != v) {

SDL_PrivateJoystickButton(joy, nbutton, v);

}

nbutton++;

break;

default:

continue;

}

break;

default:

break;

}

}

hid_end_parse(hdata);

return;

}

/* Function to close a joystick after use */

void

SDL_SYS_JoystickClose(SDL_Joystick * joy)

{

if (SDL_strncmp(joy->hwdata->path, "/dev/joy", 8)) {

report_free(&joy->hwdata->inreport);

hid_dispose_report_desc(joy->hwdata->repdesc);

}

close(joy->hwdata->fd);

SDL_free(joy->hwdata->path);

SDL_free(joy->hwdata);

return;

}

void

SDL_SYS_JoystickQuit(void)

{

int i;

for (i = 0; i < MAX_JOYS; i++) {

if (joynames[i] != NULL)

SDL_free(joynames[i]);

if (joydevnames[i] != NULL)

SDL_free(joydevnames[i]);

}

return;

}

static int

report_alloc(struct report *r, struct report_desc *rd, int repind)

{

int len;

#ifdef __DragonFly__

len = hid_report_size(rd, r->rid, repinfo[repind].kind);

#elif __FREEBSD__

# if (__FreeBSD_kernel_version >= 460000) || defined(__FreeBSD_kernel__)

# if (__FreeBSD_kernel_version <= 500111)

len = hid_report_size(rd, r->rid, repinfo[repind].kind);

# else

len = hid_report_size(rd, repinfo[repind].kind, r->rid);

# endif

# else

len = hid_report_size(rd, repinfo[repind].kind, &r->rid);

# endif

#else

# ifdef USBHID_NEW

len = hid_report_size(rd, repinfo[repind].kind, r->rid);

# else

len = hid_report_size(rd, repinfo[repind].kind, &r->rid);

# endif

#endif

if (len < 0) {

SDL_SetError("Negative HID report size");

return (-1);

}

r->size = len;

if (r->size > 0) {

r->buf = SDL_malloc(sizeof(*r->buf) - sizeof(REP_BUF_DATA(r)) +

r->size);

if (r->buf == NULL) {

SDL_OutOfMemory();

return (-1);

}

} else {

r->buf = NULL;

}

r->status = SREPORT_CLEAN;

return (0);

}

static void

report_free(struct report *r)

{

if (r->buf != NULL) {

SDL_free(r->buf);

}

r->status = SREPORT_UNINIT;

}

#endif /* SDL_JOYSTICK_USBHID */

/* vi: set ts=4 sw=4 expandtab: */