#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "mystdlib.h"

MAP map_file( char *filename, int readonly )
{
  struct stat fstatus;
  MAP    map = (MAP)malloc( sizeof( *map ));

  if( map )
  {
    memset( map, 0, sizeof( *map ));

    if( ( map->fh = open( filename, readonly ? O_RDONLY : O_RDWR ) ) >= 0 )
    {
      fstat( map->fh, &fstatus );
      if( ( map->addr = mmap( NULL, map->size = (size_t)fstatus.st_size,
         PROT_READ | ( readonly ? 0 : PROT_WRITE), (readonly ? MAP_PRIVATE : MAP_SHARED), map->fh, 0) ) == MAP_FAILED )
      {
        fprintf( stderr, "Mapping file '%s' failed\n", filename );
        close( map->fh ); free( map ); map = NULL;
      }
    } else {
      fprintf( stderr, "Couldn't open file: '%s'\n", filename );
      free( map ); map = NULL;
    }
  } else {
    fputs( "Couldn't allocate memory", stderr );
  }

  return map;
}

void unmap_file ( MAP *pMap )
{
  if( !pMap || !*pMap ) return;
  munmap( (*pMap)->addr, (*pMap)->size);
  close( (*pMap)->fh);
  free( *pMap ); *pMap = NULL;
}

int getfilesize( int fd, unsigned long *size)
{
  struct stat sb;
  if( fstat( fd, &sb )) return -1;
  *size = sb.st_size;
  return 0;
}