/* KICKDATE - Copyright 1990 Joe Porkka
               FREELY DISTRIBUTABLE

       Purpose: Keeps the current date on the Kickstart disk
                for A1000 owners with autoboot harddrives.

Version 1.0

*/


#include <exec/types.h>
#include <exec/io.h>
#include <devices/trackdisk.h>
#include <exec/memory.h>

char *buffer;

struct IOStdReq *diskreq, *CreateStdIO();
struct MsgPort *rport=0,*CreatePort();
int dev=0, ioreqs=0, ports=0;
int limit;
int outcount=0;

DeleteStdIO(req) struct IOStdReq *req;
{
    if(req==0) {
       /* printf("Delete a null message-ignored\n"); */
    } else {
       req->io_Message.mn_Node.ln_Type=0xff;
       req->io_Device=(struct Device *)-1;
       req->io_Unit=(struct Unit *)-1;
       /* printf("Freeing a stdio\n"); */
       FreeMem(req,sizeof(struct IOStdReq));
    }
}

void
main(av,ac)
    int av;
    char **ac;
{

    int status;
    int fetchtime=0;
    if(av!=2) {
       printf("Usage: kickdate set | get | getset\n");
       printf("       set -> write date to disk. get -> get date from disk\n");
       printf("       getset -> get, then periodically write date\n");
       return;
    }
    if( stricmp(ac[1],"set")==0 ) fetchtime=0;
    if( stricmp(ac[1],"get")==0 ) fetchtime = 1;
    if( stricmp(ac[1],"getset")==0 ) fetchtime = 2;
    buffer=0;
    rport=CreatePort("disk.port",0);
    if(rport==0) {
       printf("port open failed\n");
       goto portfail;
    }
    ports=1;
    /* printf("GEE!\n"); */
    diskreq=CreateStdIO(rport);
    if(diskreq==0) {
       printf("CreateStdIO failed\n");
       goto iofail;
    }
    ioreqs=1;
    /* printf("Attempt OpenDevice\n"); */
    status=OpenDevice("trackdisk.device",0,diskreq,0);
    if(status!=0) {
       printf("Opendevice failed\n");
       goto iofail;
    }
    dev=1;

    buffer = (char *)AllocMem(512, MEMF_CHIP);
    if(buffer==0) goto memfail;
    getsector(0,0,0);
    if(!(diskreq->io_Error || diskreq->io_Actual!=512)) {
       if( ((long *)buffer)[0]=='KICK') {
           if(fetchtime) {
               SetNewTime( ((long *)buffer)[1]);
           } else {
               ((long*)buffer)[1]=GetSysTime();
               writesector(0,0,0);
               if(diskreq->io_Error || diskreq->io_Actual!=512) {
                   fetchtime=0;
               }
           }
       }
       if(fetchtime==2) {  /* Stay resident */
           while(1) {
               Delay(5*60*50); /* Sleep for 5 minutes */
               /* printf("Done sleep %ld\n", SetSignal(0,0)); */
               if(SetSignal(0,0)&(1<<12)) break;  /* CTRL_C */
               getsector(0,0,0);
               /* printf("Got it!\n"); */
               if(!(diskreq->io_Error || diskreq->io_Actual!=512)) {
                   if( ((long *)buffer)[0]=='KICK') {
                       ((long*)buffer)[1]=GetSysTime();
                       writesector(0,0,0);
                       /* printf("Wrote it!\n"); */
                       if(diskreq->io_Error || diskreq->io_Actual!=512) {
                           /* break; */
                       }
                       MotorOff();
                   }
                   MotorOff();
               } else break;
           }
       }
    }


    MotorOff();


memfail:
    if(buffer) FreeMem(buffer, 512);
    CloseDevice(diskreq);
iofail:
    DeleteStdIO(diskreq);
portfail:
    if(rport) {
       RemPort(rport);
       FreeMem(rport,sizeof(*rport));
    }
}

CXBRK() {
    MotorOff();

    if(dev) {CloseDevice(diskreq);}
    if(ioreqs) {
       DeleteStdIO(diskreq);
    }
    if(ports) {
       if(rport) {
           RemPort(rport);
           FreeMem(rport,sizeof(*rport));
       }
    }
    exit(0);
}

getsector(cyl,sec,head)
    int cyl,sec,head;
{
    diskreq->io_Offset = TD_SECTOR * (sec+NUMSECS*head+NUMSECS*2*cyl);
    diskreq->io_Length=TD_SECTOR;
    diskreq->io_Data=(APTR)buffer;
    diskreq->io_Command=CMD_READ;
    diskreq->io_Flags=0;
    DoIO(diskreq);
}

writesector(cyl,sec,head)
    int cyl,sec,head;
{
    diskreq->io_Offset = TD_SECTOR * (sec+NUMSECS*head+NUMSECS*2*cyl);
    diskreq->io_Length=TD_SECTOR;
    diskreq->io_Data=(APTR)buffer;
    diskreq->io_Command=CMD_WRITE;
    diskreq->io_Flags=0;
    DoIO(diskreq);
    diskreq->io_Command=CMD_UPDATE;
    diskreq->io_Flags=0;
    DoIO(diskreq);
}

MotorOff()
{
    diskreq->io_Command=TD_MOTOR;
    diskreq->io_Length=0;
    DoIO(diskreq);
}