/*
  data source for libisoburn.

  Copyright 2007 Vreixo Formoso Lopes <metalpain2002@yahoo.es>
*/

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

#include <stdio.h>


#ifndef Xorriso_standalonE

#include <libburn/libburn.h>

#include <libisofs/libisofs.h>

#else /* ! Xorriso_standalonE */

#include "../libisofs/libisofs.h"
#include "../libburn/libburn.h"

#endif /* Xorriso_standalonE */


#include "isoburn.h"


/* Cached reading of image tree data */
/* Current implementation : single tile 128 kB */

/* powers of 2 only ! */
#define Libisoburn_cache_blockS 64

struct isoburn_cached_drive {
 struct burn_drive *drive;

 char cache_data[Libisoburn_cache_blockS * 2048];
 uint32_t cache_lba;
 uint32_t last_error_lba;
 uint32_t last_aligned_error_lba;
 int cache_hits;
};


/* Debugging only: This reports cache loads on stderr.
#define Libisoburn_read_cache_reporT 1
*/
#define Libisoburn_use_read_cachE

#ifdef Libisoburn_use_read_cachE

int 
ds_read_block(IsoDataSource *src, uint32_t lba, uint8_t *buffer)
{
 int ret;
 struct burn_drive *d;
 off_t count;
 uint32_t aligned_lba;
 char msg[80];

 struct isoburn_cached_drive *icd;

 if(src == NULL || buffer == NULL)
   return -1;

 icd = (struct isoburn_cached_drive *) src->data;
 d = (struct burn_drive*) icd->drive;

 aligned_lba= lba & ~(Libisoburn_cache_blockS - 1);
 if(aligned_lba == icd->cache_lba && icd->cache_lba != 0xffffffff) {
   (icd->cache_hits)++;
   memcpy(buffer, icd->cache_data + (lba - aligned_lba) * 2048, 2048);
   count= 2048;
   return 1;
 } 

 icd->cache_lba= 0xffffffff; /* invalidate cache */
 if(icd->last_aligned_error_lba == aligned_lba) {
   ret = 0;
 } else {
   ret = burn_read_data(d, (off_t) aligned_lba * (off_t) 2048,
                        (char *) icd->cache_data,
                        Libisoburn_cache_blockS * 2048, &count, 0);
 }
 if (ret <= 0 ) {
   icd->last_aligned_error_lba = aligned_lba;

   /* Read-ahead failure ? Try to read 2048 directly. */
   if(icd->last_error_lba == lba)
     ret = 0;
   else
     ret = burn_read_data(d, (off_t) lba * (off_t) 2048, (char *) buffer,
                        2048, &count, 0);
   if (ret > 0)
     return 1;
   icd->last_error_lba = lba;
   sprintf(msg, "ds_read_block(%lu) returns -1", (unsigned long) lba);
   burn_msgs_submit(0x00060000, msg, 0, "DEBUG", NULL);
   return -1;
 }

#ifdef Libisoburn_read_cache_reporT
 fprintf(stderr, "After %3d hits, new load from %8x , count= %d\n",
         icd->cache_hits, aligned_lba, (int) count);
#endif

 icd->cache_lba= aligned_lba;
 icd->cache_hits= 1;
 memcpy(buffer, icd->cache_data + (lba - aligned_lba) * 2048, 2048);
 count= 2048;

 return 1;
}

#else /* Libisoburn_use_read_cachE */

static int 
ds_read_block(IsoDataSource *src, uint32_t lba, uint8_t *buffer)
{
 int ret;
 struct burn_drive *d;
 off_t count;

 if(src == NULL || buffer == NULL)
   return -1;

 d = (struct burn_drive*) ((struct isoburn_cached_drive *) src->data)->drive;

 ret = burn_read_data(d, (off_t) lba * (off_t) 2048, (char *) buffer,
                      2048, &count, 0);
 if (ret <= 0 )
   return -1;

 return 1;
}

#endif /* ! Libisoburn_use_read_cachE */

	
static int
ds_open(IsoDataSource *src)
{
 /* nothing to do, device is always grabbed */
 return 1;
}

static int
ds_close(IsoDataSource *src)
{
 /* nothing to do, device is always grabbed */
 return 1;
}

static void 
ds_free_data(IsoDataSource *src)
{
 /* nothing to do */;
 if(src->data != NULL)
   free(src->data);
 src->data= NULL;
}

IsoDataSource *
isoburn_data_source_new(struct burn_drive *d)
{
 IsoDataSource *ret;
 struct isoburn_cached_drive *icd= NULL;

 if (d==NULL)
   return NULL;
 ret = malloc(sizeof(IsoDataSource));
 icd = calloc(1,sizeof(struct isoburn_cached_drive));
 if (ret == NULL || icd == NULL)
   return NULL;
 ret->refcount = 1;
 ret->read_block = ds_read_block;
 ret->open = ds_open;
 ret->close = ds_close;
 ret->free_data = ds_free_data;
 ret->data = icd;
 icd->drive = d;
 icd->cache_lba = 0xffffffff;
 icd->cache_hits = 0;
 icd->last_error_lba = 0xffffffff;
 icd->last_aligned_error_lba = 0xffffffff;
 return ret;
}