usbloadergx/source/patches/dvd_broadway.c

/*
 *  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;

}