/*
 *  Copyright (C) 2008 Nuke (wiinuke@gmail.com)
 *
 *  this file is part of GeckoOS for USB Gecko
 *  http://www.usbgecko.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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <malloc.h>
#include <time.h>
#include <gcutil.h>
#include <ogc/lwp_queue.h>
#include <gccore.h>	
 
#include "dvd_broadway.h"
 
#define DI_CMDCTX_CNT				4
 
#define DVD_DISKIDSIZE				0x20
#define DVD_DRVINFSIZE				0x20
 
#define IOCTL_DI_INQUIRY			0x12
#define IOCTL_DI_READID				0x70
#define IOCTL_DI_READ				0x71
#define IOCTL_DI_WAITCVRCLOSE		0x79
#define IOCTL_DI_COVER				0x7A
#define IOCTL_DI_RESETNOTIFY		0x7E
#define IOCTL_DI_RESET				0x8A
#define IOCTL_DI_OPENPART			0x8B
#define IOCTL_DI_CLOSEPART			0x8C
#define IOCTL_DI_UNENCREAD			0x8D
#define IOCTL_DI_ENABLE_DVD			0x8E
#define IOCTL_DI_SEEK				0xAB
#define IOCTL_DI_READ_DVDVIDEO		0xD0
#define IOCTL_DI_STOPLASER			0xD2
#define IOCTL_DI_OFFSET				0xD9
#define IOCTL_DI_REQERROR			0xE0
#define IOCTL_DI_STOPMOTOR			0xE3
#define IOCTL_DI_SETOFFBASE			0xF0
#define IOCTL_DI_GETOFFBASE			0xF1
#define IOCTL_DI_SETCRYPTMODE		0xF2
#define IOCTL_DI_GETCRYPTMODE		0xF3
#define IOCTL_DI_SETDVDROMMODE		0xF4
#define IOCTL_DI_GETDVDROMMODE		0xF5
 
#define _SHIFTL(v, s, w)	\
    ((u32) (((u32)(v) & ((0x01 << (w)) - 1)) << (s)))
#define _SHIFTR(v, s, w)	\
    ((u32)(((u32)(v) >> (s)) & ((0x01 << (w)) - 1)))
 
struct dicommand
{
	u32 diReg[8];
};
 
struct dicontext
{
	lwp_node node;
	dvdcallbacklow cb;
	struct dicommand *cmd;
};
 
static s32 __dvd_fd = -1;
static u32 __dvd_spinupval = 1;
static lwp_queue __di_contextq;
static u32 __dvd_readlength = 0;
static u32 __dvd_cbinprogress = 0;
static u32 __dvd_reqinprogress = 0;
static u32 __dvd_lowinitcalled = 0;
static struct dicommand *__di_commands = NULL;
static struct dicontext __di_contexts[DI_CMDCTX_CNT];
static u32 __di_regbuffer[0x08] ATTRIBUTE_ALIGN(32);
static u32 __di_regvalcache[0x08] ATTRIBUTE_ALIGN(32);
static u32 __di_lastticketerror[0x08] ATTRIBUTE_ALIGN(32);
static ioctlv __di_iovector[0x08] ATTRIBUTE_ALIGN(32);
static char __di_fs[] ATTRIBUTE_ALIGN(32) = "/dev/di";
 
extern u32 __IPC_ClntInit();
 
static __inline__ lwp_node* __lwp_queue_head(lwp_queue *queue)
{
	return (lwp_node*)queue;
}
 
static __inline__ lwp_node* __lwp_queue_tail(lwp_queue *queue)
{
	return (lwp_node*)&queue->perm_null;
}
 
 
static __inline__ void __lwp_queue_init_empty(lwp_queue *queue)
{
	queue->first = __lwp_queue_tail(queue);
	queue->perm_null = NULL;
	queue->last = __lwp_queue_head(queue);
}
 
static struct dicontext* __dvd_getcontext(dvdcallbacklow cb)
{
	struct dicontext *ctx;
 
	ctx = (struct dicontext*)__lwp_queue_get(&__di_contextq);
	if(ctx!=NULL) ctx->cb = cb;
 
	return ctx;
}
 
static s32 __dvd_iostransactionCB(s32 result,void *usrdata)
{
	struct dicontext *ctx = (struct dicontext*)usrdata;
 
	__dvd_reqinprogress = 0;
 
	if(ctx->cb!=NULL) {
		__dvd_cbinprogress = 1;
		if(result!=0) __dvd_readlength = 0;
		ctx->cb(result);
		__dvd_cbinprogress = 0;
	}
	__lwp_queue_append(&__di_contextq,&ctx->node);
 
	return 0;
}
 
static s32 __dvd_ioscoverregisterCB(s32 result,void *usrdata)
{
	struct dicontext *ctx = (struct dicontext*)usrdata;
 
	__dvd_reqinprogress = 0;
	__di_regvalcache[1] = __di_regbuffer[0];
 
	if(ctx->cb!=NULL) {
		__dvd_cbinprogress = 1;
		ctx->cb(result);
		__dvd_cbinprogress = 0;
	}
	__lwp_queue_append(&__di_contextq,&ctx->node);
 
	return 0;
}
 
static s32 __dvd_ioscovercloseCB(s32 result,void *usrdata)
{
	struct dicontext *ctx = (struct dicontext*)usrdata;
 
	__dvd_reqinprogress = 0;
 
	if(ctx->cb!=NULL) {
		__dvd_cbinprogress = 1;
		ctx->cb(result);
		__dvd_cbinprogress = 0;
	}
	__lwp_queue_append(&__di_contextq,&ctx->node);
 
	return 0;
}
 
s32 bwDVD_LowInit()
{
	s32 i,ret = 0;
	u32 ipclo,ipchi;
	lwp_queue inactives;
	struct dicontext *ctx;
 
	if(__dvd_lowinitcalled==0) {
		ret = __IPC_ClntInit();
		if(ret<0) return ret;
 
		ipclo = (((u32)IPC_GetBufferLo()+0x1f)&~0x1f);
		ipchi = (u32)IPC_GetBufferHi();
		if(ipchi>=(ipclo+(sizeof(struct dicommand)*DI_CMDCTX_CNT))) {
			__di_commands = (struct dicommand*)ipclo;
			IPC_SetBufferLo((void*)(ipclo+(sizeof(struct dicommand)*DI_CMDCTX_CNT)));
 
			memset(__di_commands,0,(sizeof(struct dicommand)*DI_CMDCTX_CNT));
 
			i = 0;
			__lwp_queue_init_empty(&__di_contextq);
			__lwp_queue_initialize(&inactives,__di_contexts,DI_CMDCTX_CNT,sizeof(struct dicontext));
			while((ctx=(struct dicontext*)__lwp_queue_get(&inactives))!=NULL) {
				ctx->cmd = &__di_commands[i];
				ctx->cb = NULL;
				__lwp_queue_append(&__di_contextq,&ctx->node);
 
				i++;
			}
		}
 
		ret = IOS_Open(__di_fs,0);
		if(ret<0) return ret;
 
		__dvd_fd = ret;
//		__dvd_lowinitcalled = 1;
 
	//	printf("DVD_LowInit(%d)\n",ret);
	}
	return 0;
}
 
s32 bwDVD_LowInquiry(dvddrvinfo *info,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_INQUIRY<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_INQUIRY,cmd->diReg,sizeof(struct dicommand),info,DVD_DRVINFSIZE,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowReadID(dvddiskid *diskID,dvdcallbacklow cb)
{
	s32 ret = 0;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
//	printf("DVD_LowReadID()\n");
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_READID<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_READID,cmd->diReg,sizeof(struct dicommand),diskID,DVD_DISKIDSIZE,__dvd_iostransactionCB,ctx);
 
//	printf("DVD_LowReadID(%d)\n",ret);
	return ret;
}

 
s32 bwDVD_LowRead(void *buf,u32 len,u32 offset,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	if(buf==NULL || ((u32)buf%32)!=0) return -1;
 
	__dvd_reqinprogress = 1;
	__dvd_readlength = len;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_READ<<24);
	cmd->diReg[1] = len;
	cmd->diReg[2] = offset;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_READ,cmd->diReg,sizeof(struct dicommand),buf,len,__dvd_iostransactionCB,ctx);
 
	return ret;
}

// never got this function working, probably removed from wii
s32 bwDVD_LowReadVideo(void *buf,u32 len,u32 offset,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
	__dvd_readlength = len;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_READ_DVDVIDEO<<24);
	cmd->diReg[1] = len;
	cmd->diReg[2] = offset;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_READ_DVDVIDEO,cmd->diReg,sizeof(struct dicommand),buf,len,__dvd_iostransactionCB,ctx);
 
	return ret;
}

 

s32 bwDVD_LowStopLaser(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_STOPLASER<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_STOPLASER,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
}

// never got this function working, probably removed from wii
s32 bwDVD_EnableVideo(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_ENABLE_DVD<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_ENABLE_DVD,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowSeek(u32 offset,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_SEEK<<24);
	cmd->diReg[1] = offset;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_SEEK,cmd->diReg,sizeof(struct dicommand),NULL,0,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowOffset(u64 offset,dvdcallbacklow cb)
{
	s32 ret;
	u32 *off = (u32*)(void*)(&offset);
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_OFFSET<<24);
	cmd->diReg[1] = 0;
	if(off[0]) cmd->diReg[1] = 1;
	cmd->diReg[2] = off[1];
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_OFFSET,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowPrepareCoverRegister(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_COVER<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_COVER,cmd->diReg,sizeof(struct dicommand),__di_regbuffer,0x20,__dvd_ioscoverregisterCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowOpenPartition(u32 offset,void *eticket,u32 certin_len,void *certificate_in,void *certificate_out,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	if(eticket!=NULL && ((u32)eticket%32)!=0) return -1;
	if(certificate_in!=NULL && ((u32)certificate_in%32)!=0) return -1;
	if(certificate_out!=NULL && ((u32)certificate_out%32)!=0) return -1;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_OPENPART<<24);
	cmd->diReg[1] = offset;
 
	__di_iovector[0].data = cmd;
	__di_iovector[0].len = sizeof(struct dicommand);
 
	__di_iovector[1].data = eticket;
	if(eticket==NULL) __di_iovector[1].len = 0;
	else __di_iovector[1].len = 676;
 
	__di_iovector[2].data = certificate_in;
	if(certificate_in==NULL) __di_iovector[2].len = 0;
	else __di_iovector[2].len = certin_len;
 
	__di_iovector[3].data = certificate_out;
	__di_iovector[3].len = 18916;
	__di_iovector[4].data = __di_lastticketerror;
	__di_iovector[4].len = 0x20;
	ret = IOS_IoctlvAsync(__dvd_fd,IOCTL_DI_OPENPART,3,2,__di_iovector,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowClosePartition(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_CLOSEPART<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_CLOSEPART,cmd->diReg,sizeof(struct dicommand),NULL,0,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowUnencryptedRead(void *buf,u32 len,u32 offset,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
	__dvd_readlength = len;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_UNENCREAD<<24);
	cmd->diReg[1] = len;
	cmd->diReg[2] = offset;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_UNENCREAD,cmd->diReg,sizeof(struct dicommand),buf,len,__dvd_iostransactionCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowWaitCoverClose(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_WAITCVRCLOSE<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_WAITCVRCLOSE,cmd->diReg,sizeof(struct dicommand),NULL,0,__dvd_ioscovercloseCB,ctx);
 
	return ret;
}
 
s32 bwDVD_LowResetNotify()
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	if(__dvd_cbinprogress==1) return -1;
 
	ctx = __dvd_getcontext(NULL);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_RESETNOTIFY<<24);
	ret = IOS_Ioctl(__dvd_fd,IOCTL_DI_RESETNOTIFY,cmd->diReg,sizeof(struct dicommand),NULL,0);
 
	return ret;
}
 
s32 bwDVD_LowReset(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
//	printf("DVD_LowReset()\n");
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_RESET<<24);
	cmd->diReg[1] = __dvd_spinupval;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_RESET,cmd->diReg,sizeof(struct dicommand),NULL,0,__dvd_iostransactionCB,ctx);
 
//	printf("DVD_LowReset(%d)\n",ret);
	return ret;
}
 
s32 bwDVD_LowStopMotor(u8 stop1,u8 stop2,dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_STOPMOTOR<<24);
	cmd->diReg[1] = (stop1<<24);
	cmd->diReg[2] = (stop2<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_STOPMOTOR,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
 
}
 
s32 bwDVD_LowRequestError(dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_REQERROR<<24);
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_REQERROR,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
}


s32 bwDVD_SetDecryption(s32 mode, dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_SETCRYPTMODE<<24);
	cmd->diReg[1] = mode;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_SETCRYPTMODE,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
 
}

s32 bwDVD_SetOffset(u32 offset, dvdcallbacklow cb)
{
	s32 ret;
	struct dicontext *ctx;
	struct dicommand *cmd;
 
	__dvd_reqinprogress = 1;
 
	ctx = __dvd_getcontext(cb);
	if(ctx==NULL) return IPC_ENOMEM;
 
	cmd = ctx->cmd;
	cmd->diReg[0] = (IOCTL_DI_SETOFFBASE<<24);
	cmd->diReg[1] = offset;
	ret = IOS_IoctlAsync(__dvd_fd,IOCTL_DI_SETOFFBASE,cmd->diReg,sizeof(struct dicommand),__di_regvalcache,0x20,__dvd_iostransactionCB,ctx);
 
	return ret;
 
}