//2002.1.30 Ver1.1 by optimize
#include <windows.h>
#include <stdio.h>
#include <process.h>
#include <winioctl.h>

#include "cusb.h"

s32 usb_open(s32 n,HANDLE *h){
	s8 dev_name[256];
	sprintf((char *)dev_name,(char *)"\\\\.\\ezusb-%d",n);
	*h = CreateFile((char *)dev_name,
		GENERIC_WRITE,
		FILE_SHARE_WRITE,
		NULL,
		OPEN_EXISTING,
		0,
		NULL);
	if(*h == INVALID_HANDLE_VALUE) {
		return(-1);
	}
	return(0);
}

s32 usb_close(HANDLE *h){
   CloseHandle(*h);
   return(0);
}

s32 usb_halt(HANDLE *h){
	unsigned long nbyte;
	BOOLEAN ret = FALSE;
	VENDOR_REQUEST_IN vreq;

	vreq.bRequest = 0xA0;
	vreq.wValue = 0x7F92;
	vreq.wIndex = 0x00;
	vreq.wLength = 0x01;
	vreq.bData = 1;
	vreq.direction = 0x00;
	ret = DeviceIoControl (*h,
							IOCTL_Ezusb_VENDOR_REQUEST,
							&vreq,
							sizeof(VENDOR_REQUEST_IN),
							NULL,
							0,
							&nbyte,
							NULL);
	if(ret==FALSE){
		printf("i8051 halt err.\n");
		return(-1);
	}
	return(0);
}

s32 usb_run(HANDLE *h){
	unsigned long nbyte;
	BOOLEAN ret = FALSE;
	VENDOR_REQUEST_IN vreq;

	vreq.bRequest = 0xA0;
	vreq.wValue = 0x7F92;
	vreq.wIndex = 0x00;
	vreq.wLength = 0x01;
	vreq.bData = 0;
	vreq.direction = 0x00;
	ret = DeviceIoControl (*h,
							IOCTL_Ezusb_VENDOR_REQUEST,
							&vreq,
							sizeof(VENDOR_REQUEST_IN),
							NULL,
							0,
							&nbyte,
							NULL);
	if(ret==FALSE){
		printf("i8051 run err.\n");
		return(-1);
	}
	return(0);
}

s32 usb_dwnload(HANDLE *h,u8 *image,s32 len){
	unsigned long nbyte;
	BOOLEAN ret = FALSE;
	
	ret = DeviceIoControl (*h,
							IOCTL_Ezusb_ANCHOR_DOWNLOAD,
							image,
							len,
							NULL,
							0,
							&nbyte,
							NULL);
	if(ret==FALSE){
		printf("usb dwnload err.\n");
		return(-1);
	}
	return(0);
}

s32 usb_resetpipe(HANDLE *h,ULONG p){
	unsigned long nbyte;
	BOOLEAN ret = FALSE;
	
	ret = DeviceIoControl (*h,
							IOCTL_Ezusb_RESETPIPE,
							&p,
							sizeof(ULONG),
							NULL,
							0,
							&nbyte,
							NULL);
	if(ret==FALSE){
		return(-1);
	}
	return(0);
}

s32 usb_get_string(HANDLE *h,s32 idx,s8 *s){
	unsigned long nbyte;
	s32 i;
	u8  pvbuf[2+128];
	GET_STRING_DESCRIPTOR_IN sin;
	BOOLEAN ret = FALSE;

	sin.Index=idx;
	sin.LanguageId=27;
	ret = DeviceIoControl (*h,
							IOCTL_Ezusb_GET_STRING_DESCRIPTOR,
							&sin,
							sizeof(GET_STRING_DESCRIPTOR_IN),
							pvbuf,
							sizeof (pvbuf),
							&nbyte,
							NULL);
	if(ret==FALSE){
		return(-1);
	}
	for(i=0;i<pvbuf[0]/2-1;i++){
		*(s++)=(s8)pvbuf[2+i*2];
	}
	*s=0;
	return(0);
}

s32 usb_bulk_write(HANDLE *h,s32 pipe,u8 *buf,s32 len){
	unsigned long nbyte;
	s32 i,l;
	BOOLEAN ret = FALSE;
	BULK_TRANSFER_CONTROL bulk_control;

	bulk_control.pipeNum = pipe;
	for(i=0;len>0;){
		if(len>0x8000){
			l=0x8000;
		}
		else{
			l=len;
		}
		ret = DeviceIoControl (*h,
							IOCTL_EZUSB_BULK_WRITE,
							&bulk_control,
							sizeof(BULK_TRANSFER_CONTROL),
							buf+i,
							l,
							&nbyte,
							NULL);
		if(ret==FALSE){
			return(-1);
		}
		i+=l;
		len-=l;
	}
	return(0);
}

s32 usb_bulk_read(HANDLE *h,s32 pipe,u8 *buf,s32 len){
	unsigned long nbyte;
	s32 i,l,cnt;
	BOOLEAN ret = FALSE;
	BULK_TRANSFER_CONTROL bulk_control;

	bulk_control.pipeNum = pipe;
	for(i=cnt=0;len>0;){
		if(len>0x8000){
			l=0x8000;
		}
		else{
			l=len;
		}
		ret = DeviceIoControl (*h,
							IOCTL_EZUSB_BULK_READ,
							&bulk_control,
							sizeof(BULK_TRANSFER_CONTROL),
							buf+i,
							l,
							&nbyte,
							NULL);
		if(ret==FALSE){
			return(-1);
		}
		i+=l;
		len-=l;
		cnt+=nbyte;
	}
	return(cnt);
}

#if CUSB_DEBUG==1
void tty_thread(void *p){
	HANDLE tty_h;
	u8 buf[1024];
	usb_open(0,&tty_h);
	for(;;){
		usb_bulk_read(&tty_h,13,buf,64);
		printf("recv %s\n",buf);
	}
	usb_close(&tty_h);
}
#endif

s32 cusb_init(s32 n,HANDLE *h,u8 *fw,s8 *str1,s8 *str2){
	s8 s1[128],s2[128];

	if(usb_open(n,h)) return(-1);
	if(usb_get_string(h,1,s1)) return(-1);
	if(usb_get_string(h,2,s2)) return(-1);
	if(strcmp((const char *)str1,(const char *)s1)||strcmp((const char *)str2,(const char *)s2)){
		if(usb_halt(h)) return(-1);
		if(usb_dwnload(h,fw,CUSB_DWLSIZE)) return(-1);
		if(usb_run(h)) return(-1);
		usb_close(h);
		for(;;){
			s32 err=0;
			Sleep(2000);
			if(usb_open(n,h)) err=1;
			if(err==0){
				break;
			}
		}
	}
#if CUSB_DEBUG==1
	{
		s32 i;
		i=_beginthread(tty_thread,0,0);
		if(i<0){
			printf("Can't Create Thread.\n");
			return(-1);
		}
	}
#endif
	return(0);
}