#include <exec/ports.h>
#include <exec/io.h>
#include <exec/memory.h>
#include <devices/trackdisk.h>
#include <stdio.h>
#include <fcntl.h>

#define FILE_SIG 0x4449534b
#define NUMTRKS 10
#define TD_BUFSIZE (11 * 1024 * NUMTRKS)
#define NUMBLOCKS (NUMTRKS * 22)

typedef unsigned long ulong;
typedef unsigned char uchar;

struct fhb {
    ulong sig1;
    ulong sig2;
    uchar name[504];
};

struct fhb header;

struct MsgPort *td_mp;
struct IOStdReq *td_iob;

char fname[255];

extern struct MsgPort *CreatePort();
extern struct IOStdReq *CreateStdIO();
extern long OpenDevice();
extern void *AllocMem();
extern int open();
extern int write();

extern int Enable_Abort;

uchar *diskbuf;

main()
{
    int fd;
    long i;
    Enable_Abort = 0;

    td_mp = CreatePort(0L,0L);
    if(!td_mp) exit(99);
    td_iob = CreateStdIO(td_mp);
    if(!td_iob) {
        DeletePort(td_mp);
        exit(99);
    }
    if(OpenDevice("trackdisk.device",0L,td_iob,0L)) {
        printf("Unable to open floppy device driver.\n");
        DeleteStdIO(td_iob);
        DeletePort(td_mp);
        exit(99);
    }

    diskbuf = AllocMem(TD_BUFSIZE,MEMF_PUBLIC|MEMF_CHIP);
    if(!diskbuf) {
        printf("Unable to obtain %ld byte buffer in chip memory.\n",TD_BUFSIZE);
        CloseDevice(td_iob);
        DeleteStdIO(td_iob);
        DeletePort(td_mp);
        exit(99);
    }

    printf("Input file name? ");
    gets(fname);

    fd = open(fname,O_RDONLY);
    if(fd<0) printf("Unable to open input file.\n");
    else {
        if(sizeof(header)!=read(fd,&header,sizeof(header)))
            printf("Error reading header block.\n");
        else if(header.sig1 != FILE_SIG || header.sig2 != ~FILE_SIG)
            printf("Not a disk dump.\n");
        else {
            printf("Extracting: %s\n",header.name);

            for(i=0;i<80;i+=NUMTRKS) {
                printf("Reading...              \r");
                fflush(stdout);
                if(TD_BUFSIZE != read(fd,diskbuf,TD_BUFSIZE)) {
                    printf("Error reading from file.\n");
                    break;
                } else {
                    printf("Writing tracks %ld to %ld.\r",i,i+NUMTRKS-1);
                    fflush(stdout);
                    td_iob->io_Command = TD_FORMAT;
                    td_iob->io_Data = (APTR) diskbuf;
                    td_iob->io_Length = TD_BUFSIZE;
                    td_iob->io_Offset = 11 * 1024 * i;
                    DoIO(td_iob);
                    if(td_iob->io_Error) {
                        printf("Floppy write failed, error number %d.\n",
                                td_iob->io_Error);
                        break;
                    }
                }
            }
        }
        close(fd);
    }
    printf("                             \r");
    fflush(stdout);

    FreeMem(diskbuf,TD_BUFSIZE);

    td_iob->io_Command = CMD_UPDATE;
    DoIO(td_iob);
    if(td_iob->io_Error) printf("Floppy update failed, error number %d.\n",
                                td_iob->io_Error);

    td_iob->io_Command = TD_MOTOR;
    td_iob->io_Length = 0;
    DoIO(td_iob);

    CloseDevice(td_iob);
    DeleteStdIO(td_iob);
    DeletePort(td_mp);
}