/*!
 * @file libshut.c
 * @brief HID Library - Serial SHUT backend for Generic HID Access (using MGE HIDParser)
 *
 * @author Copyright (C) 2006
 *	Arnaud Quette <aquette.dev@gmail.com>
 *
 *      This program is free software; you can redistribute it and/or modify
 *      it under the terms of the GNU General Public License as published by
 *      the Free Software Foundation; either version 2 of the License, or
 *      (at your option) any later version.
 *
 *      This program is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *      GNU General Public License for more details.
 *
 *      You should have received a copy of the GNU General Public License
 *      along with this program; if not, write to the Free Software
 *      Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 * -------------------------------------------------------------------------- */

/* FIXME: remainder
 * - complete shut_set_report
 * - complete libshut_get_interrupt
 * - complete shut_strerror
 * - complete commands and data table in mge-hid from mge-shut
 */

#include <stdio.h>
#include <stdlib.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <string.h>

#ifdef HAVE_STDINT_H
#include <stdint.h> /* for uint8_t, uint16_t */
#endif

#include "serial.h"
#include "libshut.h"
#include "libhid.h"

#include "common.h" /* for xmalloc, upsdebugx prototypes */

#define SHUT_DRIVER_NAME	"MGE SHUT communication driver"
#define SHUT_DRIVER_VERSION	"0.67"
#define MAX_TRY 4

/*!
 * HID descriptor, completed with desc{type,len}
 */
struct my_hid_descriptor {
        uint8_t  bLength;
        uint8_t  bDescriptorType;
        uint16_t bcdHID;
        uint8_t  bCountryCode;
        uint8_t  bNumDescriptors;
        uint8_t  bReportDescriptorType;
        uint16_t wDescriptorLength;
};

/*!********************************************************
 * USB spec information, synchronised with (ripped from) libusb
 */

/*
 * Device and/or Interface Class codes
 */
#define USB_CLASS_PER_INTERFACE		0	/* for DeviceClass */
#define USB_CLASS_AUDIO			1
#define USB_CLASS_COMM			2
#define USB_CLASS_HID			3
#define USB_CLASS_PRINTER		7
#define USB_CLASS_MASS_STORAGE		8
#define USB_CLASS_HUB			9
#define USB_CLASS_DATA			10
#define USB_CLASS_VENDOR_SPEC		0xff

/*
 * Descriptor types
 */
#define USB_DT_DEVICE			0x01
#define USB_DT_CONFIG			0x02
#define USB_DT_STRING			0x03
#define USB_DT_INTERFACE		0x04
#define USB_DT_ENDPOINT			0x05

#define USB_DT_HID			0x21
#define USB_DT_REPORT			0x22
#define USB_DT_PHYSICAL			0x23
#define USB_DT_HUB			0x29

/*
 * Descriptor sizes per descriptor type
 */
#define USB_DT_DEVICE_SIZE		18
#define USB_DT_CONFIG_SIZE		9
#define USB_DT_INTERFACE_SIZE		9
#define USB_DT_ENDPOINT_SIZE		7
#define USB_DT_ENDPOINT_AUDIO_SIZE	9	/* Audio extension */
#define USB_DT_HUB_NONVAR_SIZE		7

/*
 * Standard requests
 */
#define USB_REQ_GET_STATUS              0x00
#define USB_REQ_CLEAR_FEATURE           0x01
/* 0x02 is reserved */
#define USB_REQ_SET_FEATURE             0x03
/* 0x04 is reserved */
#define USB_REQ_SET_ADDRESS             0x05
#define USB_REQ_GET_DESCRIPTOR          0x06
#define USB_REQ_SET_DESCRIPTOR          0x07
#define USB_REQ_GET_CONFIGURATION       0x08
#define USB_REQ_SET_CONFIGURATION       0x09
#define USB_REQ_GET_INTERFACE           0x0A
#define USB_REQ_SET_INTERFACE           0x0B
#define USB_REQ_SYNCH_FRAME             0x0C

#define USB_TYPE_STANDARD               (0x00 << 5)
#define USB_TYPE_CLASS                  (0x01 << 5)
#define USB_TYPE_VENDOR                 (0x02 << 5)
#define USB_TYPE_RESERVED               (0x03 << 5)

#define USB_RECIP_DEVICE                0x00
#define USB_RECIP_INTERFACE             0x01
#define USB_RECIP_ENDPOINT              0x02
#define USB_RECIP_OTHER                 0x03

/*
 * Various libusb API related stuff
 */
#define USB_ENDPOINT_IN                 0x80
#define USB_ENDPOINT_OUT                0x00
/*!
 * end of USB spec information
 *********************************************************/

/*!
 * HID definitions
 */
#define HID_REPORT_TYPE_INPUT	0x01
#define HID_REPORT_TYPE_OUTPUT	0x02
#define HID_REPORT_TYPE_FEATURE	0x03

#define REQUEST_TYPE_USB        0x80
#define REQUEST_TYPE_HID        0x81
#define REQUEST_TYPE_GET_REPORT 0xa1
#define REQUEST_TYPE_SET_REPORT 0x21

#define MAX_REPORT_SIZE         0x1800

/*!
 * SHUT definitions - From Simplified SHUT spec
 */

#define SHUT_TYPE_REQUEST       0x01
#define SHUT_TYPE_RESPONSE      0x04
#define SHUT_TYPE_NOTIFY        0x05
#define SHUT_OK                 0x06
#define SHUT_NOK                0x15
/* sync signals are also used to set the notification level */
#define SHUT_SYNC               0x16 /* complete notifications - not yet managed */
				     /* but needed for some early Ellipse models */
#define SHUT_SYNC_LIGHT         0x17 /* partial notifications */
#define SHUT_SYNC_OFF           0x18 /* disable notifications - only do polling */
#define SHUT_PKT_LAST           0x80

#define SHUT_TIMEOUT 3000

/*!
 * SHUT functions for HID marshalling
 */
int shut_get_descriptor(shut_dev_handle_t *sdev, unsigned char type,
			unsigned char index, void *buf, int size);
int shut_get_string_simple(shut_dev_handle_t *dev, int index,
			   unsigned char *buf, size_t buflen);
int libshut_get_report(shut_dev_handle_t *devp, int ReportId,
		       unsigned char *raw_buf, int ReportSize );
int shut_set_report(shut_dev_handle_t *sdev, int id, unsigned char *pkt, int reportlen);
int libshut_get_interrupt(shut_dev_handle_t *devp, unsigned char *buf,
			  int bufsize, int timeout);
int shut_control_msg(shut_dev_handle_t *dev, int requesttype, int request,
		     int value, int index, unsigned char *bytes, int size, int timeout);

/* FIXME */
char * shut_strerror() { return ""; }

/*!
 * From SHUT specifications
 * sync'ed with libusb
 */

typedef struct shut_ctrltransfer_s {
	uint8_t  bRequestType;
	uint8_t  bRequest;
	uint16_t wValue;
	uint16_t wIndex;
	uint16_t wLength;
	
	uint32_t timeout;      /* in milliseconds */

	/* pointer to data */
	void *data;
	/*   uint8_t  padding[8]; for use with shut_set_report?! */
} shut_ctrltransfer_t;


typedef union hid_data_t {
	shut_ctrltransfer_t  hid_pkt;
	uint8_t raw_pkt[8]; /* max report lengh, was 8 */
} hid_data_t;

typedef struct shut_packet_s {
	uint8_t bType;
	uint8_t bLength;
	hid_data_t    data;
	uint8_t bChecksum;
} shut_packet_t;

typedef union shut_data_t {
	shut_packet_t shut_pkt;
	uint8_t raw_pkt[11];
} shut_data_t;

typedef union hid_desc_data_t {
	struct my_hid_descriptor hid_desc;
	uint8_t    raw_desc[9]; /* max report lengh, aws 9 */
} hid_desc_data_t;

/* Device descriptor */
typedef struct device_descriptor_s {
	uint8_t  bLength;
	uint8_t  bDescriptorType;
	uint16_t bcdUSB;
	uint8_t  bDeviceClass;
	uint8_t  bDeviceSubClass;
	uint8_t  bDeviceProtocol;
	uint8_t  bMaxPacketSize0;
	uint16_t idVendor;
	uint16_t idProduct;
	uint16_t bcdDevice;
	uint8_t  iManufacturer;
	uint8_t  iProduct;
	uint8_t  iSerialNumber;
	uint8_t  bNumConfigurations;
} device_descriptor_t;
#if 0
typedef union device_desc_data_t {
	device_descriptor_t dev_desc;
	uint8_t       raw_desc[18];
} device_desc_data_t;
#endif
/* Low level SHUT (Serial HID UPS Transfer) routines  */
void setline (int fd, int set);
bool_t shut_synchronise (int fd);
int shut_wait_ack (int fd);
int shut_interrupt_read(shut_dev_handle_t *dev, int ep, unsigned char *bytes,
			int size, int timeout);
int shut_control_msg(shut_dev_handle_t *dev, int requesttype, int request, int value,
		     int index, unsigned char *bytes, int size, int timeout);


/* Data portability */
/* realign packet data according to Endianess */
#define BYTESWAP(in) (((in & 0xFF) << 8) + ((in & 0xFF00) >> 8))
static void align_request(struct shut_ctrltransfer_s *ctrl )
{
#if WORDS_BIGENDIAN
        /* Sparc/Mips/... are big endian, USB/SHUT little endian */
	(*ctrl).wValue    = BYTESWAP((*ctrl).wValue);
	(*ctrl).wIndex    = BYTESWAP((*ctrl).wIndex);
	(*ctrl).wLength   = BYTESWAP((*ctrl).wLength);
#endif
}

/* return report descriptor on success, NULL otherwise */
/* mode: MODE_OPEN for the 1rst time, MODE_REOPEN to skip getting
    report descriptor (the longer part). On success, fill in the
    curDevice structure and return the report descriptor length. On
    failure, return -1. Note: ReportDesc must point to a large enough
    buffer. There's no way to know the size ahead of time. Matcher is
    a linked list of matchers (see libhid.h), and the opened device
    must match all of them. */
int libshut_open(shut_dev_handle_t **sdevp, HIDDevice_t *curDevice,
		 HIDDeviceMatcher_t *matcher, unsigned char *ReportDesc, int mode)
{
	int ret, res; 
	unsigned char buf[20];
	char string[256];
	struct my_hid_descriptor *desc;
	struct device_descriptor_s *dev_descriptor;
	shut_dev_handle_t *devp = *sdevp;
	
	upsdebugx(2, "libshut_open: using port %s", devp->device_path);
	
	/* initialize serial port */
	/* FIXME: add variable baudrate detection */
	devp->upsfd = ser_open(devp->device_path);
	ser_set_speed(devp->upsfd, devp->device_path, B2400);
	setline (devp->upsfd, 1);

	/* initialise communication */
	if (!shut_synchronise (devp->upsfd))
	{
		upsdebugx(2, "No communication with UPS");
		return -1;
	}
	else
		upsdebugx(2, "Communication with UPS established");

	/* Get DEVICE descriptor */
	dev_descriptor = (struct device_descriptor_s *)buf;
	res = shut_get_descriptor(devp, USB_DT_DEVICE, 0, buf, USB_DT_DEVICE_SIZE);
	/* res = shut_control_msg(devp, USB_ENDPOINT_IN+1, USB_REQ_GET_DESCRIPTOR,
	(USB_DT_DEVICE << 8) + 0, 0, buf, 0x9, SHUT_TIMEOUT); */
	if (res < 0)
	{
		upsdebugx(2, "Unable to get DEVICE descriptor (%s)", shut_strerror());
		return -1;
	}
	else if (res < 9)
	{
		upsdebugx(2, "DEVICE descriptor too short (expected %d, got %d)",
		      USB_DT_DEVICE_SIZE, res);
		return -1;
	} 

	/* collect the identifying information of this
		device. Note that this is safe, because
		there's no need to claim an interface for
		this (and therefore we do not yet need to
		detach any kernel drivers). */

	curDevice->VendorID = dev_descriptor->idVendor;
	curDevice->ProductID = dev_descriptor->idProduct;
	curDevice->Vendor = strdup("MGE UPS SYSTEMS");
	curDevice->Product = strdup("unknown");
	curDevice->Serial = strdup("unknown");
	curDevice->Bus = strdup("shut");

	if (dev_descriptor->iProduct)
	{
		ret = shut_get_string_simple(devp, dev_descriptor->iProduct,
					     string, 0x25);
		if (ret > 0)
			curDevice->Product = strdup(string);
	}

	if (dev_descriptor->iSerialNumber)
	{
		ret = shut_get_string_simple(devp, dev_descriptor->iSerialNumber,
					     string, 0x25);
		if (ret > 0)
			curDevice->Serial = strdup(string);
	}

	upsdebugx(2, "- VendorID: %04x", curDevice->VendorID);
	upsdebugx(2, "- ProductID: %04x", curDevice->ProductID);
	upsdebugx(2, "- Manufacturer: %s", curDevice->Vendor ? curDevice->Vendor : "unknown");
	upsdebugx(2, "- Product: %s", curDevice->Product ? curDevice->Product : "unknown");
	upsdebugx(2, "- Serial Number: %s", curDevice->Serial ? curDevice->Serial : "unknown");
	upsdebugx(2, "- Bus: %s", curDevice->Bus ? curDevice->Bus : "unknown");
	upsdebugx(2, "Device matches");

	if (mode == MODE_REOPEN)
		return 1;

	/* Get HID descriptor */
	desc = (struct my_hid_descriptor *)buf;
	res = shut_get_descriptor(devp, USB_DT_HID, 0, buf, 0x9);
	/* res = shut_control_msg(devp, USB_ENDPOINT_IN+1, USB_REQ_GET_DESCRIPTOR,
			(USB_DT_HID << 8) + 0, 0, buf, 0x9, SHUT_TIMEOUT); */

	if (res < 0)
	{
		upsdebugx(2, "Unable to get HID descriptor (%s)", shut_strerror());
		return -1;
	}
	else if (res < 9)
	{
		upsdebugx(2, "HID descriptor too short (expected %d, got %d)", 8, res);
		return -1;
	} 

	/* USB_LE16_TO_CPU(desc->wDescriptorLength); */
	desc->wDescriptorLength = buf[7] | (buf[8] << 8);
	upsdebugx(2, "HID descriptor retrieved (Reportlen = %u)", desc->wDescriptorLength);

/*	if (!dev->config) {
		upsdebugx(2, "  Couldn't retrieve descriptors");
		return -1;
	}*/

	/* Get REPORT descriptor */
	res = shut_get_descriptor(devp, USB_DT_REPORT, 0, ReportDesc,
				   desc->wDescriptorLength);
	/* res = shut_control_msg(devp, USB_ENDPOINT_IN+1, USB_REQ_GET_DESCRIPTOR,
				(USB_DT_REPORT << 8) + 0, 0, ReportDesc, 
			desc->wDescriptorLength, SHUT_TIMEOUT); */
	if (res >= desc->wDescriptorLength) 
	{
		upsdebugx(2, "Report descriptor retrieved (Reportlen = %u)",
		      desc->wDescriptorLength);
		upsdebugx(2, "Found HID device");
		fflush(stdout);

		return desc->wDescriptorLength;
	}
	else if (res < 0)
	{
		upsdebugx(2, "Unable to get Report descriptor (%d)", res);
	}
	else
	{
		upsdebugx(2, "Report descriptor too short (expected %d, got %d)",
		      desc->wDescriptorLength, res);
	}
	
	upsdebugx(2, "No appropriate HID device found");
	fflush(stdout);

	return -1;
}

void libshut_close(shut_dev_handle_t *sdev)
{
	ser_close(sdev->upsfd, sdev->device_path);
}

/* return the report of ID=type in report 
 * return -1 on failure, report length on success
 */
int libshut_get_report(shut_dev_handle_t *devp, int ReportId,
		       unsigned char *raw_buf, int ReportSize )
{
	upsdebugx(4, "Entering libshut_get_report");

	if (devp != NULL)
	{
		return shut_control_msg(devp,
			REQUEST_TYPE_GET_REPORT,
			// == USB_ENDPOINT_IN + USB_TYPE_CLASS + USB_RECIP_INTERFACE,
			 0x01,
			 ReportId+(0x03<<8), /* HID_REPORT_TYPE_FEATURE */
			 0, raw_buf, ReportSize, SHUT_TIMEOUT);
	}
	else
		return 0;
}

int libshut_set_report(shut_dev_handle_t *devp, int ReportId,
		       unsigned char *raw_buf, int ReportSize )
{
	int ret = 0;
	
	upsdebugx(1, "Entering libshut_set_report");

	if (devp != NULL)
	{
		ret = shut_control_msg(devp, 
			REQUEST_TYPE_SET_REPORT,
			// == USB_ENDPOINT_OUT + USB_TYPE_CLASS + USB_RECIP_INTERFACE,
			0x09,
			ReportId+(0x03<<8), /* HID_REPORT_TYPE_FEATURE */
			0, raw_buf, ReportSize, SHUT_TIMEOUT);
	}
exit(0);
	return ret;
}

int libshut_get_string(shut_dev_handle_t *devp, int StringIdx, char *buf)
{
	int ret = -1;
	
	if (devp != NULL)
	{
		ret = shut_get_string_simple(devp, StringIdx, buf, 20);
		 /* sizeof(buf)); */
		if (ret > 0)
			upsdebugx(2, "-> String: %s (len = %i/%i)",
			      buf, ret, sizeof(buf));
		else
			upsdebugx(2, "- Unable to fetch buf");
	}
	return ret;
}

int libshut_get_interrupt(shut_dev_handle_t *devp, unsigned char *buf,
			   int bufsize, int timeout)
{
  int ret = -1;

  if (devp != NULL)
	{
	  /* FIXME: hardcoded interrupt EP => need to get EP descr for IF descr */
	  ret = shut_interrupt_read(devp, 0x81, buf, bufsize, timeout);
	  if (ret > 0)
		upsdebugx(6, " ok");
	  else
		upsdebugx(6, " none (%i)", ret);
	}

  return ret;
}

communication_subdriver_t shut_subdriver = {
	SHUT_DRIVER_NAME,
	SHUT_DRIVER_VERSION,
	libshut_open,
	libshut_close,
	libshut_get_report,
	libshut_set_report,
	libshut_get_string,
	libshut_get_interrupt
};

/***********************************************************************/
/********** Low level SHUT (Serial HID UPS Transfer) routines **********/
/***********************************************************************/

/*
 * set RTS to on and DTR to off
 * 
 * set : 1 to set comm
 * set : 0 to stop commupsh.
 */
void setline (int fd, int set)
{
	int dtr = TIOCM_DTR;
	int rts = TIOCM_RTS;
	
	upsdebugx(3, "entering setline(%i)", set);

	if (set == 1) {
		ioctl (fd, TIOCMBIC, &dtr);
		ioctl (fd, TIOCMBIS, &rts);
	}
	else
	{
		ioctl (fd, TIOCMBIS, &dtr);
		ioctl (fd, TIOCMBIC, &rts);
	}
}

/*****************************************************************************
 * shut_synchronise ()
 * 
 * initiate communication with the UPS
 *
 * return TRUE on success, FALSE on failure
 *
 *****************************************************************************/
bool_t shut_synchronise (int fd)
{
	bool_t retCode = FALSE;
	u_char c = SHUT_SYNC, reply;
	int try;
		
	upsdebugx (2, "entering shut_synchronise()");
	reply = '\0';

	/* FIXME: re enable notification support?
	switch (notification)
	{
		case OFF_NOTIFICATION:
			c = SHUT_SYNC_OFF;
			break;
		case LIGHT_NOTIFICATION:
			c = SHUT_SYNC_LIGHT;
			break;
		default:
		case COMPLETE_NOTIFICATION:
			c = SHUT_SYNC;
			break;
	}
	*/

	/* Sync with the UPS according to notification */
	for (try = 0; try < MAX_TRY; try++)
	{
		upsdebugx (3, "Syncing communication (try %i)", try);

		if ((ser_send_char(fd, c)) == -1)
		{
			upsdebugx (3, "Communication error while writing to port");
			retCode = FALSE;
		}
		else
		{
			ser_get_char(fd, &reply, 1, 0);
			if (reply == c)
			{
				upsdebugx (3, "Syncing and notification setting done");
				retCode = TRUE;
			}
		}
	}
	return retCode;
}

/*!
 * Compute a SHUT checksum for the packet "buf"
 */
u_char shut_checksum(const u_char *buf, int bufsize)
{
	int i;
	u_char chk=0;
	
	for(i=0; i<bufsize; i++)
		chk^=buf[i];

	upsdebugx (4, "shut_checksum: %02x", chk);
	return chk;
}


int shut_packet_recv (int fd, u_char *Buf, int datalen)
{
	u_char   Start[2];
	u_char   Frame[8];
	u_char   Chk[1];
	u_short  Size=8;
	u_short  Pos=0;
	u_char   Retry=0;
	int recv;
	shut_data_t   sdata;
	
	upsdebugx (4, "entering shut_packet_recv (%i)", datalen);
	
	while(datalen>0 && Retry<3)
	{
		/// if(serial_read (SHUT_TIMEOUT, &Start[0]) >= 0)
		if(ser_get_char(fd, &Start[0], SHUT_TIMEOUT/1000, 0) >= 0)
		{
			sdata.shut_pkt.bType = Start[0];
			if(Start[0]==SHUT_SYNC)
			{
				upsdebugx (4, "received SYNC token");
				memcpy(Buf, Start, 1);
				return 1;
			}
			else 
			{
				/// if((serial_read (SHUT_TIMEOUT, &Start[1]) >= 0) &&
				if( (ser_get_char(fd, &Start[1], SHUT_TIMEOUT/1000, 0) >= 0) &&
							((Start[1]>>4)==(Start[1]&0x0F)))
				{
					upsdebug_hex(4, "Receive", Start, 2);
					Size=Start[1]&0x0F;
					sdata.shut_pkt.bLength = Size;
					for(recv=0;recv<Size;recv++)
					{
						///if(serial_read (SHUT_TIMEOUT, &Frame[recv]) < 0)
						if(ser_get_char(fd, &Frame[recv], SHUT_TIMEOUT/1000, 0) < 0)
							break;
					}
					upsdebug_hex(4, "Receive", Frame, Size);

					///serial_read (SHUT_TIMEOUT, &Chk[0]);
					ser_get_char(fd, &Chk[0], SHUT_TIMEOUT/1000, 0);
					if(Chk[0]==shut_checksum(Frame, Size))
					{
						upsdebugx (4, "shut_checksum: %02x => OK", Chk[0]);
						memcpy(Buf, Frame, Size);
						datalen-=Size;
						Buf+=Size;
						Pos+=Size;
						Retry=0;
					
						ser_send_char(fd, SHUT_OK);
						//shut_token_send(SHUT_OK);

						if(Start[0]&SHUT_PKT_LAST)
						{
							if ((Start[0]&SHUT_PKT_LAST) == SHUT_TYPE_NOTIFY)
							{
								/* TODO: process notification (dropped for now) */
								upsdebugx (4, "=> notification");
								datalen+=Pos;
								Pos=0;
							}
							else
						                return Pos;
						}
						else
							upsdebugx (4, "need more data (%i)!", datalen);
					}
					else
					{
						upsdebugx (4, "shut_checksum: %02x => NOK", Chk[0]);
						ser_send_char(fd, SHUT_NOK);
						//shut_token_send(SHUT_NOK);
						Retry++;
					}
				}
				else
					return 0;
			}
		}
		else
			Retry++;
	} /* while */
	
	return 0;
}

/**********************************************************************/
int shut_interrupt_read(shut_dev_handle_t *dev, int ep, unsigned char *bytes, int size,
		       int timeout)
{
	/* FIXME: to be written */
	return 0;
}

/**********************************************************************/
int shut_get_string_simple(shut_dev_handle_t *dev, int index,
			   unsigned char *buf, size_t buflen)
{
	char tbuf[255];       /* Some devices choke on size > 255 */
	int ret, si, di;
	
	ret = shut_control_msg(dev, USB_ENDPOINT_IN, USB_REQ_GET_DESCRIPTOR,
			(USB_DT_STRING << 8) + index, 0x0, tbuf, buflen, 1000);
	if (ret < 0)
		return ret;

	if (tbuf[1] != USB_DT_STRING)
		return -EIO;

	if (tbuf[0] > ret)
		return -EFBIG;

	/* skip the zero'ed high bytes */
	for (di = 0, si = 2; si < tbuf[0]; si += 2)
	{
		if (di >= (buflen - 1))
			break;

		if (tbuf[si + 1])   /* high byte */
			buf[di++] = '?';
		else
			buf[di++] = tbuf[si];
	}

	buf[di] = 0;
	return di;
}
		
/* 
 * Human Interface Device (HID) functions
 *********************************************************************/

/**********************************************************************
 * shut_get_descriptor(int desctype, u_char *pkt)
 * 
 * get descriptor specified by DescType and return it in Buf
 *
 * desctype  - from shutdataType
 * pkt       - where to store the report received
 *
 * return 0 on success, -1 on failure, -2 on NACK
 *
 *********************************************************************/
int shut_get_descriptor(shut_dev_handle_t *sdev, unsigned char type,
			unsigned char index, void *buf, int size)
{
	memset(buf, 0, size);

	upsdebugx (2, "entering shut_get_descriptor(n %02x, %i)", type, size);

	return shut_control_msg(sdev, USB_ENDPOINT_IN+(type>=USB_DT_HID?1:0),
				 USB_REQ_GET_DESCRIPTOR, (type << 8) + index, 0, buf, size, 1000);
}

/* Take care of a SHUT transfer (sending and receiving data) */
int shut_control_msg(shut_dev_handle_t *sdev, int requesttype, int request,
		    int value, int index, unsigned char *bytes, int size, int timeout)
{
	unsigned char shut_pkt[11];
	short Retry=1;
	short data_size, remaining_size = size; //Size;
	int i;
	struct shut_ctrltransfer_s ctrl;
	int ret = 0;
	
	upsdebugx (3, "entering shut_control_msg");
		
	/* build the control request */
	ctrl.bRequestType = requesttype;
	ctrl.bRequest = request;
	ctrl.wValue = value;
	ctrl.wIndex = index;
	ctrl.wLength = size;
	ctrl.data = bytes;
	ctrl.timeout = timeout;

	align_request(&ctrl);

	/* Send all data */
	while(remaining_size > 0 && Retry > 0)
        {
		data_size = (size >= 8) ? 8 : remaining_size;
		
		/* Forge the SHUT Frame */
		/* shut_pkt[0] = SHUT_TYPE_REQUEST + SHUT_PKT_LAST; */ //( (remaining_size>8)? 0 :
		shut_pkt[0] = SHUT_TYPE_REQUEST + ( ((requesttype == REQUEST_TYPE_SET_REPORT) && (remaining_size>8))? 0 : SHUT_PKT_LAST);
		shut_pkt[1] = (data_size<<4) + data_size;
		if ( (requesttype == REQUEST_TYPE_SET_REPORT) && (remaining_size < 8) )
			memcpy(&shut_pkt[2], &bytes, remaining_size); // we need to send ctrl.data 
		else
			memcpy(&shut_pkt[2], &ctrl, 8);
		shut_pkt[(data_size+3) - 1] = shut_checksum(&shut_pkt[2], data_size);
		
		/* Packets need only to be sent once
		* NACK handling should take care of the rest */
		if (Retry == 1)
		{
			ser_send_buf(sdev->upsfd, shut_pkt, data_size+3);
			upsdebug_hex(3, "shut_control_msg", shut_pkt, data_size+3);
			//serial_send (shut_pkt, data_size+3);
		}

		i = shut_wait_ack (sdev->upsfd);
		switch (i)
		{
			case 0:
				remaining_size = 0; //remaining_size-=data_size;
				Retry=1;
				break;
			case -1:
				if (Retry >= MAX_TRY)
				{
					upsdebugx(2, "Max tries reached while waiting for ACK, still getting errors");
					
					/* try to resync, and give one more try */
					Retry--;
					shut_synchronise(sdev->upsfd);
					return i;
				}
				else
				{
					upsdebugx(4, "Retry = %i", Retry);
					/* Send a NACK to get a resend from the UPS */
					ser_send_char(sdev->upsfd, SHUT_NOK);
					Retry++;
				}
				break;
			case -3:
				/* FIXME: notification caught => to be processed */
			default:
				;
		}
	}
	if (remaining_size != 0)
		return -1;
	
	
	
	/* now receive data, except for SET_REPORT */
	if (requesttype != REQUEST_TYPE_SET_REPORT)
		ret = shut_packet_recv (sdev->upsfd, bytes, size);
	
	
	
	
	
	
	
//	ret = shut_transfer(sdev->upsfd, &ctrl);
	//if (ret < 0)
	//	USB_ERROR_STR(-errno, "error sending control message: %s",
	//		       strerror(errno));
#if 0
	// just shut_set_report : SHUT_PKT_LAST => 0 + 8 => sizeof(data)
	if((ret = shut_packet_send (sdev, &ctrl, 8, SHUT_PKT_LAST)) > 0)
	{
		if((ret = shut_packet_recv (sdev, ctrl, reportlen)) > 0)
		{
			//upsdebug_hex(3, "shut_get_report", pkt, retcode)
			;
		}
	}
#endif
	/* first packet to instruct a set command */
// 	if((retcode = shut_packet_send (&data, sizeof(data), 0x0)) > 0)
// 	{
// 		/* second packet to give the actual data */
// 		memcpy(&data.raw_pkt, pkt, reportlen);
// 		upsdebug_hex(3, "Set2", pkt, reportlen);
// 
// 		retcode = shut_packet_send (&data, reportlen, SHUT_PKT_LAST);
// 	}

	// ret = ioctl(dev->fd, IOCTL_USB_CONTROL, &ctrl);
	//if (ret < 0)
	//	USB_ERROR_STR(-errno, "error sending control message: %s",
	//		       strerror(errno));

	return ret;
}

/**********************************************************************
 * shut_wait_ack()
 *
 * wait for an ACK packet
 *
 * returns 0 on success, -1 on error, -2 on NACK, -3 on NOTIFICATION
 * 
 *********************************************************************/
int shut_wait_ack (int fd)
{
	int retCode = -1;
	u_char c = '\0';

	ser_get_char(fd, &c, SHUT_TIMEOUT/1000, 0);
	if (c == SHUT_OK)
	{
		upsdebugx (2, "shut_wait_ack(): ACK received");
		retCode = 0;
	}
	else if (c == SHUT_NOK)
	{
		upsdebugx (2, "shut_wait_ack(): NACK received");
		retCode = -2;
	}
	else if ((c & SHUT_PKT_LAST) == SHUT_TYPE_NOTIFY)
	{
		upsdebugx (2, "shut_wait_ack(): NOTIFY received");
		retCode = -3;
	}
	else if (c == '\0')
		upsdebugx (2, "shut_wait_ack(): Nothing received");
	
	return retCode;
}