#pragma rtl

#include <stddef.h>
#include <locator.h>
#include <misctool.h>
#include <tcpip.h>
#include <memory.h>
#include <gsos.h>
#include <orca.h>
#include "driver.h"
#include "installdriver.h"
#include "asmglue.h"
#include "mounturl.h"
#include "version.h"

#define ICON_HEIGHT 20
#define ICON_WIDTH 28

struct {
    Word iconType;
    Word iconSize;
    Word iconHeight;
    Word iconWidth;
    Byte iconImage[ICON_WIDTH * ICON_HEIGHT / 2];
    Byte iconMask[ICON_WIDTH * ICON_HEIGHT / 2];
} icon = {
    0x8000,                             /* color icon */
    ICON_WIDTH * ICON_HEIGHT / 2,
    ICON_HEIGHT,
    ICON_WIDTH,
    {
        0xFF,0xFF,0xF0,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0xF3,0x33,0x33,0xF3,0x33,0xFF,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0xF3,0x33,0xF3,0x33,0x33,0x3F,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xF0,0x00,0x00,0x00,0x00,0x00,0x00,0x0F,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xFF,0xFF,0xCC,0xCC,0xCC,0xCC,0xFF,0xFF,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xF0,0xFF,0xFF,0xCC,0xFC,0xCC,0xCC,0xFF,0xFF,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xFF,0x0F,0xFF,0xCC,0xFC,0xCC,0xCC,0xFF,0xFF,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xFF,0xF0,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x0F,0xFF,0xFF,
        0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0xF0,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
        0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,
        0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0xFF,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,
        0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,
        0xF0,0xF0,0x00,0x00,0x00,0x00,0x00,0xF0,0x00,0x00,0x00,0x00,0x0F,0x0F,
        0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
        0xF0,0xF0,0x00,0x00,0x00,0x00,0x00,0xF0,0x00,0x00,0x00,0x00,0x0F,0x0F
    },
    {
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x00,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x00,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0x00,0x00,
        0x00,0x00,0x00,0x00,0x00,0x00,0x0F,0xFF,0x00,0x00,0x00,0x00,0x00,0x00,
        0x00,0x00,0x00,0x00,0x00,0x00,0xFF,0xFF,0xF0,0x00,0x00,0x00,0x00,0x00,
        0x00,0x00,0x00,0x00,0x00,0x00,0xFF,0xFF,0xF0,0x00,0x00,0x00,0x00,0x00,
        0x00,0x00,0x00,0x00,0x00,0x00,0xFF,0xFF,0xF0,0x00,0x00,0x00,0x00,0x00,
        0x0F,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0xF0,
        0x0F,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xF0,0xF0,
        0x0F,0x0F,0xFF,0xFF,0xFF,0xFF,0xFF,0x0F,0xFF,0xFF,0xFF,0xFF,0xF0,0xF0
    }
};

#define TCPIP_REQUEST_NAME "\pTCP/IP~STH~NetDisk~"

const char bootInfoString[] = "NetDisk               " BOOT_INFO_VERSION;

Word *unloadFlagPtr;

static struct NotificationProcRec {
    Long reserved1;
    Word reserved2;
    Word Signature;
    Long Event_flags;
    Long Event_code;
    Byte jml;
    void (*proc)(void);
} notificationProcRec;

#define NOTIFY_SHUTDOWN    0x20
#define NOTIFY_GSOS_SWITCH 0x04

static void notificationProc(void);
static pascal Word mountRequestProc(Word reqCode, void *dataIn, void *dataOut);
static pascal Word tcpipRequestProc(Word reqCode, void *dataIn, void *dataOut);

static void ejectAll(void);

#define JML 0x5C

/* Custom DControl code for switching to DOS order */
#define SwitchToDOSOrder 0x8081


static void setUnloadFlag(void) {
    if (unloadFlagPtr != NULL && *unloadFlagPtr == 0)
        *unloadFlagPtr = 1;
}

int main(void) {
    /*
     * Load Marinetti.
     * We may get an error if the TCPIP init isn't loaded yet, but we ignore it.
     * The tool stub is still loaded in that case, which is enough for now.
     */
    LoadOneTool(54, 0x0200);
    if (toolerror() && toolerror() != terrINITNOTFOUND)
        goto error;

    /* Initialize the DIBs for our driver */
    InitDIBs();

    /* Install our driver */
    if (InstallDriver() != 0) {
        UnloadOneTool(54);
        goto error;
    }

    /* We're not going to error out, so show boot info. */
    ShowBootInfo(bootInfoString, (Pointer)&icon);

    /* Add notification proc to be called on shutdown and switches to GS/OS */
    notificationProcRec.Signature = 0xA55A;
    notificationProcRec.Event_flags = NOTIFY_SHUTDOWN | NOTIFY_GSOS_SWITCH;
    notificationProcRec.jml = JML;
    notificationProcRec.proc = notificationProc;
    NotifyProcRecGS addNotifyProcRec = {1, (ProcPtr)&notificationProcRec};
    AddNotifyProcGS(&addNotifyProcRec);

    /*
     * Put Marinetti in the default TPT so its tool stub won't be unloaded,
     * even if UnloadOneTool is called on it.  Programs may still call
     * TCPIPStartUp and TCPIPShutDown, but those don't actually do
     * anything, so the practical effect is that Marinetti will always
     * be available once its init has loaded (which may not have happened
     * yet when this init loads).
     */
    SetDefaultTPT();
    
    /* Accept requests to mount URLs */
    AcceptRequests(NETDISK_REQUEST_NAME, userid(), &mountRequestProc);

    /* Accept requests (notifications) from Marinetti */
    AcceptRequests(TCPIP_REQUEST_NAME, userid(), &tcpipRequestProc);

    return 0;
    
error:
    setUnloadFlag();
}

/*
 * Notification procedure called at shutdown time or when switching to GS/OS.
 */
#pragma databank 1
static void notificationProc(void) {
    Word stateReg = ForceRomIn();

    if (notificationProcRec.Event_code & NOTIFY_GSOS_SWITCH) {
        /* Reinstall our driver on switch back to GS/OS from P8 */
        InstallDriver();
    } else if (notificationProcRec.Event_code & NOTIFY_SHUTDOWN) {
        //TODO eject disks
    }

    RestoreStateReg(stateReg);
}
#pragma databank 0

static void doMountURL(struct MountURLRec *mountURLRec) {
    if (!TCPIPStatus() || toolerror()) {
        mountURLRec->result = MARINETTI_NOT_PRESENT;
        return;
    }

    DAccessRecGS controlRec = {5};
    controlRec.code = MountURL;
    controlRec.list = (pointer)mountURLRec;
    controlRec.requestCount = mountURLRec->byteCount;

    for (unsigned i = 0; i < NDIBS; i++) {
        Word devNum = dibs[i].DIBDevNum;
        if (devNum == 0)
            continue;
    
        controlRec.devNum = devNum;
        DControl(&controlRec);
        
        if (mountURLRec->result == OPERATION_SUCCESSFUL) {
            if (mountURLRec->format != formatAutoDetect)
                return;

            /* Current operating assumption is that it's ProDOS-order (raw) */
            mountURLRec->format = formatRaw;

            /*
             * Auto-detect DOS vs ProDOS order by seeing if
             * GS/OS can recognize the volume.
             */
            static ResultBuf32 devName = {35};
            static DInfoRecGS dInfoRec = {2, 0, &devName};
            dInfoRec.devNum = devNum;
            DInfo(&dInfoRec);
            if (toolerror())
                return;
        
            static ResultBuf255 volumeName = {255+2+2};
            static VolumeRecGS volumeRec = {2, &devName.bufString, &volumeName};
            Volume(&volumeRec);
            
            if (toolerror() == unknownVol) {
                controlRec.code = SwitchToDOSOrder;
                DControl(&controlRec);
            }
            
            return;
        } else if (mountURLRec->result != DISK_ALREADY_MOUNTED) {
            return;
        }
    }
    
    mountURLRec->result = NO_DIBS_AVAILABLE;
}

/*
 * Request procedure called to mount a disk image by URL.
 */
#pragma databank 1
#pragma toolparms 1
static pascal Word mountRequestProc(Word reqCode, void *dataIn, void *dataOut) {
    if (reqCode == MountURL) {
        doMountURL((struct MountURLRec *) dataIn);
        return 0x8000;
    }
    
    return 0;
}
#pragma toolparms 0
#pragma databank 0


/*
 * Procedure to "eject" all our disks (called when network goes down).
 */
static void ejectAll(void) {
    DAccessRecGS controlRec = {5};
    controlRec.code = eject;
    controlRec.list = NULL;
    controlRec.requestCount = 0;
        
    for (unsigned i = 0; i < NDIBS; i++) {
        controlRec.devNum = dibs[i].DIBDevNum;
        if (controlRec.devNum == 0)
            continue;

        DControl(&controlRec);
    }
}

/*
 * Request procedure called by Marinetti with its notifications.
 * If the network has gone down, we unmount all the disks.
 */
#pragma databank 1
#pragma toolparms 1
static pascal Word tcpipRequestProc(Word reqCode, void *dataIn, void *dataOut) {
    if (reqCode == TCPIPSaysNetworkDown) {
        ejectAll();
    }
    
    return 0;
}
#pragma toolparms 0
#pragma databank 0