#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "ata.h"

#define BLOCK_SIZE 512

static FILE* file;

int ata_read_sectors(unsigned long start, unsigned char count, void* buf)
{
    if(fseek(file,start*BLOCK_SIZE,SEEK_SET)) {
        perror("fseek");
        return -1;
    }
    if(!fread(buf,BLOCK_SIZE,count,file)) {
        printf("Failed reading %d blocks starting at block %ld\n",count,start); 
        perror("fread");
        return -1;
    }
    return 0;
}

int ata_write_sectors(unsigned long start, unsigned char count, void* buf)
{
    if(fseek(file,start*BLOCK_SIZE,SEEK_SET)) {
        perror("fseek");
        return -1;
    }
    if(!fwrite(buf,BLOCK_SIZE,count,file)) {
        perror("fwrite");
        return -1;
    }
    return 0;
}

int ata_init(void)
{
    /* check disk size */
    file=fopen("disk.img","r+");
    if(!file) {
        fprintf(stderr, "read_disk() - Could not find \"disk.img\"\n");
        return -1;
    }
    return 0;
}

void ata_exit(void)
{
    fclose(file);
}