/*
 *  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);
	union
	{
		u64 off64;
		u32 off32[2];
	} off;
	off.off64 = 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.off32[0]) cmd->diReg[1] = 1;
	cmd->diReg[2] = off.off32[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;

}