mmap.c

Go to the documentation of this file.

#include "ctl/ctldef.h"
#include "ctl/dir.h"
#ifdef WIN32
#include <Windows.h>
#include <io.h>
#include <fcntl.h>
#else
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <unistd.h>
#include <dirent.h>
#endif
#include <fcntl.h>
#include <sys/stat.h>
#include <stdio.h>
#include "ctl/mmap.h"

#ifndef CTL_UNIT
#ifdef WIN32

const void* ctl_mmap_open( ctl_mmap* map, const char* szPath )
{
    assertobjptr(map);
    assertconst(szPath,1);
    map->base = NULL;
    map->size = 0;
    map->hMap = INVALID_HANDLE_VALUE;
    map->hFile = CreateFile( szPath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
    if ( map->hFile != INVALID_HANDLE_VALUE ) 
    { 
        map->size = (size_t)GetFileSize( map->hFile, NULL );
        map->hMap = CreateFileMapping( map->hFile, NULL, PAGE_READONLY, 0, 0, NULL );
        if ( map->hMap != INVALID_HANDLE_VALUE ) 
        {
            map->base = MapViewOfFile( map->hMap, FILE_MAP_READ, 0, 0, 0 );
            if( map->base )
                return map->base;
            if( map->hMap != INVALID_HANDLE_VALUE )
                CloseHandle( map->hMap );
        }
        CloseHandle( map->hFile );
        map->hMap = INVALID_HANDLE_VALUE;
        map->hFile = INVALID_HANDLE_VALUE;
    }
    return NULL;
}

void ctl_mmap_close( ctl_mmap* map )
{
    assertobjptr(map);
    if( map->base )
        UnmapViewOfFile( map->base );
    if( map->hMap != INVALID_HANDLE_VALUE )
        CloseHandle( map->hMap );
    if( map->hFile != INVALID_HANDLE_VALUE )
        CloseHandle( map->hFile );
    map->hMap = INVALID_HANDLE_VALUE;
    map->hFile = INVALID_HANDLE_VALUE;
    map->base = NULL;
    map->size = 0;
}


#else /* !WIN32 */

const void* ctl_mmap_open( ctl_mmap* map, const char* szPath )
{
    assertobjptr(map);
    assertconst(szPath,1);
    map->base = NULL;
    map->size = 0;
    map->fd = open(szPath, O_RDONLY);
    if(map->fd != -1) 
    { 
        struct stat buf;
        if ( !fstat(map->fd, &buf) )
        { 
            map->size = buf.st_size;
            map->base = mmap(0, map->size, PROT_READ, MAP_SHARED, map->fd, 0);
            if( map->base == MAP_FAILED )
            {
                map->base = NULL;
                map->size = 0;
                close(map->fd);
                map->fd = -1;
            }
            return map->base;
        }
    }
    return NULL;
}

void ctl_mmap_close( ctl_mmap* map )
{
    assertobjptr(map);
    if( map->base )
        munmap( (void*)map->base, map->size );
    if( map->fd != -1 )
        close( map->fd );
    map->fd = -1;
    map->base = NULL;
    map->size = 0;
}


#endif /* IS_MSVC */

#else /* CTL_UNIT */

#include "unit/unit.h"

void Test_MMAp(void)
{
    const char szTesting[] = "Now is the time for all good men to come to the aid of their country.";
    FILE* fp = fopen( "Data.txt", "wb" );
    if( fp )
    {
        if( 1 != fwrite( szTesting, sizeof(szTesting), 1, fp ) )
        {
            fclose(fp);
            return;
        }
        else
        {
            fclose(fp);
        }
        {
            ctl_mmap map;
            ctl_mmap_open( &map, "Data.txt" );
            Test_Error( ctl_mmap_ready(&map) );
            Test_Error( !strcmp( (const char*)ctl_mmap_ptr(&map), szTesting ) );
            ctl_mmap_close( &map );
            unlink("Data.txt");
        }
    }
}

#endif /* CTL_UNIT */