refactor memory mapping and memory buffering methods into separate file
This commit is contained in:
Aaron Boxer
parent
8c1c6560b0
commit
908c045cb1
|
|
@ -37,6 +37,8 @@ set(OPENJPEG_SRCS
|
|||
${CMAKE_CURRENT_SOURCE_DIR}/j2k.h
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/jp2.c
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/jp2.h
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/mem_stream.c
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/mem_stream.h
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/mct.c
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/mct.h
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/mqc.c
|
||||
|
|
|
|||
|
|
@ -0,0 +1,387 @@
|
|||
/*
|
||||
* Copyright (c) 2016, Aaron Boxer
|
||||
* This Source Code Form is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
*/
|
||||
|
||||
#include "opj_includes.h"
|
||||
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <windows.h>
|
||||
#else /* _WIN32 */
|
||||
|
||||
#include <errno.h>
|
||||
|
||||
#include <stdarg.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/stat.h>
|
||||
# include <unistd.h>
|
||||
# include <sys/mman.h>
|
||||
|
||||
#endif
|
||||
|
||||
#include <fcntl.h>
|
||||
|
||||
|
||||
#ifdef _WIN32
|
||||
typedef void* opj_handle_t;
|
||||
#else
|
||||
typedef OPJ_INT32 opj_handle_t;
|
||||
#endif
|
||||
|
||||
|
||||
typedef struct opj_buf_info
|
||||
{
|
||||
OPJ_BYTE *buf;
|
||||
OPJ_OFF_T off;
|
||||
OPJ_SIZE_T len;
|
||||
opj_handle_t fd; // for file mapping
|
||||
} opj_buf_info_t;
|
||||
|
||||
static void opj_free_buffer_info(void* user_data) {
|
||||
if (user_data)
|
||||
opj_free(user_data);
|
||||
}
|
||||
|
||||
static OPJ_SIZE_T opj_zero_copy_read_from_buffer(void ** p_buffer,
|
||||
OPJ_SIZE_T p_nb_bytes,
|
||||
opj_buf_info_t* p_source_buffer)
|
||||
{
|
||||
OPJ_SIZE_T l_nb_read = 0;
|
||||
|
||||
if (((OPJ_SIZE_T)p_source_buffer->off + p_nb_bytes) < p_source_buffer->len) {
|
||||
l_nb_read = p_nb_bytes;
|
||||
}
|
||||
|
||||
*p_buffer = p_source_buffer->buf + p_source_buffer->off;
|
||||
p_source_buffer->off += (OPJ_OFF_T)l_nb_read;
|
||||
|
||||
return l_nb_read ? l_nb_read : ((OPJ_SIZE_T)-1);
|
||||
}
|
||||
|
||||
static OPJ_SIZE_T opj_read_from_buffer(void * p_buffer,
|
||||
OPJ_SIZE_T p_nb_bytes,
|
||||
opj_buf_info_t* p_source_buffer)
|
||||
{
|
||||
OPJ_SIZE_T l_nb_read;
|
||||
|
||||
if ((OPJ_SIZE_T)p_source_buffer->off + p_nb_bytes < p_source_buffer->len) {
|
||||
l_nb_read = p_nb_bytes;
|
||||
}
|
||||
else {
|
||||
l_nb_read = (p_source_buffer->len - (OPJ_SIZE_T)p_source_buffer->off);
|
||||
}
|
||||
memcpy(p_buffer, p_source_buffer->buf + p_source_buffer->off, l_nb_read);
|
||||
p_source_buffer->off += (OPJ_OFF_T)l_nb_read;
|
||||
|
||||
return l_nb_read ? l_nb_read : ((OPJ_SIZE_T)-1);
|
||||
}
|
||||
|
||||
static OPJ_SIZE_T opj_write_from_buffer(void * p_buffer,
|
||||
OPJ_SIZE_T p_nb_bytes,
|
||||
opj_buf_info_t* p_source_buffer)
|
||||
{
|
||||
memcpy(p_source_buffer->buf + (OPJ_SIZE_T)p_source_buffer->off, p_buffer, p_nb_bytes);
|
||||
p_source_buffer->off += (OPJ_OFF_T)p_nb_bytes;
|
||||
p_source_buffer->len += p_nb_bytes;
|
||||
|
||||
return p_nb_bytes;
|
||||
}
|
||||
|
||||
static OPJ_OFF_T opj_skip_from_buffer(OPJ_OFF_T p_nb_bytes,
|
||||
opj_buf_info_t * p_source_buffer)
|
||||
{
|
||||
if (p_source_buffer->off + p_nb_bytes < (OPJ_OFF_T)p_source_buffer->len) {
|
||||
p_source_buffer->off += p_nb_bytes;
|
||||
}
|
||||
else {
|
||||
p_source_buffer->off = (OPJ_OFF_T)p_source_buffer->len;
|
||||
}
|
||||
return p_nb_bytes;
|
||||
}
|
||||
|
||||
static OPJ_BOOL opj_seek_from_buffer(OPJ_OFF_T p_nb_bytes,
|
||||
opj_buf_info_t * p_source_buffer)
|
||||
{
|
||||
if (p_nb_bytes < (OPJ_OFF_T)p_source_buffer->len) {
|
||||
p_source_buffer->off = p_nb_bytes;
|
||||
}
|
||||
else {
|
||||
p_source_buffer->off = (OPJ_OFF_T)p_source_buffer->len;
|
||||
}
|
||||
return OPJ_TRUE;
|
||||
}
|
||||
|
||||
|
||||
static void opj_set_up_buffer_stream(opj_stream_t* l_stream, OPJ_SIZE_T len, OPJ_BOOL p_is_read_stream) {
|
||||
opj_stream_set_user_data_length(l_stream, len);
|
||||
|
||||
if (p_is_read_stream) {
|
||||
opj_stream_set_read_function(l_stream, (opj_stream_read_fn)opj_read_from_buffer);
|
||||
opj_stream_set_zero_copy_read_function(l_stream, (opj_stream_zero_copy_read_fn)opj_zero_copy_read_from_buffer);
|
||||
}
|
||||
else
|
||||
opj_stream_set_write_function(l_stream, (opj_stream_write_fn)opj_write_from_buffer);
|
||||
opj_stream_set_skip_function(l_stream, (opj_stream_skip_fn)opj_skip_from_buffer);
|
||||
opj_stream_set_seek_function(l_stream, (opj_stream_seek_fn)opj_seek_from_buffer);
|
||||
}
|
||||
|
||||
opj_stream_t* opj_create_buffer_stream(OPJ_BYTE *buf,
|
||||
OPJ_SIZE_T len,
|
||||
OPJ_BOOL p_is_read_stream)
|
||||
{
|
||||
opj_stream_t* l_stream;
|
||||
opj_buf_info_t* p_source_buffer = NULL;
|
||||
|
||||
if (!buf || !len)
|
||||
return NULL;
|
||||
|
||||
p_source_buffer = (opj_buf_info_t*)opj_malloc(sizeof(opj_buf_info_t));
|
||||
if (!p_source_buffer)
|
||||
return NULL;
|
||||
|
||||
l_stream = opj_stream_create(0, p_is_read_stream);
|
||||
if (!l_stream) {
|
||||
opj_free(p_source_buffer);
|
||||
return NULL;
|
||||
|
||||
}
|
||||
|
||||
memset(p_source_buffer, 0, sizeof(opj_buf_info_t));
|
||||
p_source_buffer->buf = buf;
|
||||
p_source_buffer->off = 0;
|
||||
p_source_buffer->len = len;
|
||||
|
||||
opj_stream_set_user_data(l_stream, p_source_buffer, opj_free_buffer_info);
|
||||
opj_set_up_buffer_stream(l_stream, p_source_buffer->len, p_is_read_stream);
|
||||
return l_stream;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
OPJ_INT32 opj_get_file_open_mode(const char* mode)
|
||||
{
|
||||
OPJ_INT32 m = -1;
|
||||
switch (mode[0]) {
|
||||
case 'r':
|
||||
m = O_RDONLY;
|
||||
if (mode[1] == '+')
|
||||
m = O_RDWR;
|
||||
break;
|
||||
case 'w':
|
||||
case 'a':
|
||||
m = O_RDWR | O_CREAT;
|
||||
if (mode[0] == 'w')
|
||||
m |= O_TRUNC;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return m;
|
||||
}
|
||||
|
||||
|
||||
#ifdef _WIN32
|
||||
|
||||
static OPJ_UINT64 opj_size_proc(opj_handle_t fd)
|
||||
{
|
||||
ULARGE_INTEGER m;
|
||||
m.LowPart = GetFileSize(fd, &m.HighPart);
|
||||
return(m.QuadPart);
|
||||
}
|
||||
|
||||
|
||||
static void* opj_map(opj_handle_t fd, OPJ_SIZE_T len) {
|
||||
void* ptr = NULL;
|
||||
HANDLE hMapFile = NULL;
|
||||
|
||||
if (!fd || !len)
|
||||
return NULL;
|
||||
|
||||
/* Passing in 0 for the maximum file size indicates that we
|
||||
would like to create a file mapping object for the full file size */
|
||||
hMapFile = CreateFileMapping(fd, NULL, PAGE_READONLY, 0, 0, NULL);
|
||||
if (hMapFile == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
ptr = MapViewOfFile(hMapFile, FILE_MAP_READ, 0, 0, 0);
|
||||
CloseHandle(hMapFile);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_unmap(void* ptr, OPJ_SIZE_T len) {
|
||||
OPJ_INT32 rc = -1;
|
||||
(void)len;
|
||||
if (ptr)
|
||||
rc = UnmapViewOfFile(ptr) ? 0 : -1;
|
||||
return rc;
|
||||
}
|
||||
|
||||
static opj_handle_t opj_open_fd(const char* fname, const char* mode) {
|
||||
void* fd = NULL;
|
||||
OPJ_INT32 m = -1;
|
||||
DWORD dwMode = 0;
|
||||
|
||||
if (!fname)
|
||||
return (opj_handle_t)-1;
|
||||
|
||||
|
||||
m = opj_get_file_open_mode(mode);
|
||||
switch (m) {
|
||||
case O_RDONLY:
|
||||
dwMode = OPEN_EXISTING;
|
||||
break;
|
||||
case O_RDWR:
|
||||
dwMode = OPEN_ALWAYS;
|
||||
break;
|
||||
case O_RDWR | O_CREAT:
|
||||
dwMode = OPEN_ALWAYS;
|
||||
break;
|
||||
case O_RDWR | O_TRUNC:
|
||||
dwMode = CREATE_ALWAYS;
|
||||
break;
|
||||
case O_RDWR | O_CREAT | O_TRUNC:
|
||||
dwMode = CREATE_ALWAYS;
|
||||
break;
|
||||
default:
|
||||
return NULL;
|
||||
}
|
||||
|
||||
fd = (opj_handle_t)CreateFileA(fname,
|
||||
(m == O_RDONLY) ? GENERIC_READ : (GENERIC_READ | GENERIC_WRITE),
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, dwMode,
|
||||
(m == O_RDONLY) ? FILE_ATTRIBUTE_READONLY : FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
if (fd == INVALID_HANDLE_VALUE) {
|
||||
return (opj_handle_t)-1;
|
||||
}
|
||||
return fd;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_close_fd(opj_handle_t fd) {
|
||||
OPJ_INT32 rc = -1;
|
||||
if (fd) {
|
||||
rc = CloseHandle(fd) ? 0 : -1;
|
||||
}
|
||||
return rc;
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
static OPJ_UINT64 opj_size_proc(opj_handle_t fd)
|
||||
{
|
||||
struct stat sb;
|
||||
if (!fd)
|
||||
return 0;
|
||||
|
||||
if (fstat(fd, &sb)<0)
|
||||
return(0);
|
||||
else
|
||||
return((OPJ_UINT64)sb.st_size);
|
||||
}
|
||||
|
||||
static void* opj_map(opj_handle_t fd, OPJ_SIZE_T len) {
|
||||
void* ptr = NULL;
|
||||
OPJ_UINT64 size64 = 0;
|
||||
|
||||
if (!fd)
|
||||
return NULL;
|
||||
|
||||
size64 = opj_size_proc(fd);
|
||||
ptr = (void*)mmap(0, (size_t)size64, PROT_READ, MAP_SHARED, fd, 0);
|
||||
return ptr == (void*)-1 ? NULL : ptr;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_unmap(void* ptr, OPJ_SIZE_T len) {
|
||||
if (ptr)
|
||||
munmap(ptr, len);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static opj_handle_t opj_open_fd(const char* fname, const char* mode) {
|
||||
opj_handle_t fd = 0;
|
||||
OPJ_INT32 m = -1;
|
||||
if (!fname) {
|
||||
return (opj_handle_t)-1;
|
||||
}
|
||||
m = opj_get_file_open_mode(mode);
|
||||
fd = open(fname, m, 0666);
|
||||
if (fd < 0) {
|
||||
#ifdef DEBUG_ERRNO
|
||||
if (errno > 0 && strerror(errno) != NULL) {
|
||||
printf("%s: %s", fname, strerror(errno));
|
||||
}
|
||||
else {
|
||||
printf("%s: Cannot open", fname);
|
||||
}
|
||||
#endif
|
||||
return (opj_handle_t)-1;
|
||||
}
|
||||
return fd;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_close_fd(opj_handle_t fd) {
|
||||
if (!fd)
|
||||
return 0;
|
||||
return (close(fd));
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
static void opj_mem_map_free(void* user_data) {
|
||||
if (user_data) {
|
||||
opj_buf_info_t* buffer_info = (opj_buf_info_t*)user_data;
|
||||
opj_unmap(buffer_info->buf, buffer_info->len);
|
||||
opj_close_fd(buffer_info->fd);
|
||||
opj_free(buffer_info);
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
Currently, only read streams are supported for memory mapped files.
|
||||
*/
|
||||
opj_stream_t* opj_create_mapped_file_read_stream(const char *fname)
|
||||
{
|
||||
opj_stream_t* l_stream = NULL;
|
||||
opj_buf_info_t* buffer_info = NULL;
|
||||
void* mapped_view = NULL;
|
||||
OPJ_BOOL p_is_read_stream = OPJ_TRUE;
|
||||
|
||||
opj_handle_t fd = opj_open_fd(fname, p_is_read_stream ? "r" : "w");
|
||||
if (fd == (opj_handle_t)-1)
|
||||
return NULL;
|
||||
|
||||
buffer_info = (opj_buf_info_t*)opj_malloc(sizeof(opj_buf_info_t));
|
||||
memset(buffer_info, 0, sizeof(opj_buf_info_t));
|
||||
buffer_info->fd = fd;
|
||||
buffer_info->len = (OPJ_SIZE_T)opj_size_proc(fd);
|
||||
|
||||
l_stream = opj_stream_create(0, p_is_read_stream);
|
||||
if (!l_stream) {
|
||||
opj_mem_map_free(buffer_info);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
mapped_view = opj_map(fd, buffer_info->len);
|
||||
if (!mapped_view) {
|
||||
opj_stream_destroy(l_stream);
|
||||
opj_mem_map_free(buffer_info);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
buffer_info->buf = mapped_view;
|
||||
buffer_info->off = 0;
|
||||
|
||||
opj_stream_set_user_data(l_stream, buffer_info, (opj_stream_free_user_data_fn)opj_mem_map_free);
|
||||
opj_set_up_buffer_stream(l_stream, buffer_info->len, p_is_read_stream);
|
||||
|
||||
|
||||
return l_stream;
|
||||
}
|
||||
|
|
@ -0,0 +1,17 @@
|
|||
/*
|
||||
* Copyright (c) 2016, Aaron Boxer
|
||||
* This Source Code Form is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
*/
|
||||
#ifndef __MEM_STREAM_H
|
||||
#define __MEM_STREAM_H
|
||||
|
||||
opj_stream_t* opj_create_buffer_stream(OPJ_BYTE *buf,
|
||||
OPJ_SIZE_T len,
|
||||
OPJ_BOOL p_is_read_stream);
|
||||
|
||||
opj_stream_t* opj_create_mapped_file_read_stream(const char *fname);
|
||||
|
||||
#endif
|
||||
|
|
@ -157,95 +157,6 @@ static OPJ_BOOL opj_seek_from_file (OPJ_OFF_T p_nb_bytes, FILE * p_user_data)
|
|||
|
||||
/* ---------------------------------------------------------------------- */
|
||||
|
||||
#ifdef _WIN32
|
||||
typedef void* opj_handle_t;
|
||||
#else
|
||||
typedef OPJ_INT32 opj_handle_t;
|
||||
#endif
|
||||
|
||||
|
||||
typedef struct opj_buf_info
|
||||
{
|
||||
OPJ_BYTE *buf;
|
||||
OPJ_OFF_T off;
|
||||
OPJ_SIZE_T len;
|
||||
opj_handle_t fd; // for file mapping
|
||||
} opj_buf_info_t;
|
||||
|
||||
static void opj_free_buffer_info(void* user_data) {
|
||||
if (user_data)
|
||||
opj_free(user_data);
|
||||
}
|
||||
|
||||
static OPJ_SIZE_T opj_zero_copy_read_from_buffer(void ** p_buffer,
|
||||
OPJ_SIZE_T p_nb_bytes,
|
||||
opj_buf_info_t* p_source_buffer)
|
||||
{
|
||||
OPJ_SIZE_T l_nb_read = 0;
|
||||
|
||||
if ( ((OPJ_SIZE_T)p_source_buffer->off + p_nb_bytes) < p_source_buffer->len){
|
||||
l_nb_read = p_nb_bytes;
|
||||
}
|
||||
|
||||
*p_buffer = p_source_buffer->buf + p_source_buffer->off;
|
||||
p_source_buffer->off += (OPJ_OFF_T)l_nb_read;
|
||||
|
||||
return l_nb_read ? l_nb_read : ((OPJ_SIZE_T)-1);
|
||||
}
|
||||
|
||||
static OPJ_SIZE_T opj_read_from_buffer(void * p_buffer,
|
||||
OPJ_SIZE_T p_nb_bytes,
|
||||
opj_buf_info_t* p_source_buffer)
|
||||
{
|
||||
OPJ_SIZE_T l_nb_read;
|
||||
|
||||
if ((OPJ_SIZE_T)p_source_buffer->off + p_nb_bytes < p_source_buffer->len){
|
||||
l_nb_read = p_nb_bytes;
|
||||
}
|
||||
else{
|
||||
l_nb_read = (p_source_buffer->len - (OPJ_SIZE_T)p_source_buffer->off);
|
||||
}
|
||||
memcpy(p_buffer, p_source_buffer->buf + p_source_buffer->off, l_nb_read);
|
||||
p_source_buffer->off += (OPJ_OFF_T)l_nb_read;
|
||||
|
||||
return l_nb_read ? l_nb_read : ((OPJ_SIZE_T)-1);
|
||||
}
|
||||
|
||||
static OPJ_SIZE_T opj_write_from_buffer(void * p_buffer,
|
||||
OPJ_SIZE_T p_nb_bytes,
|
||||
opj_buf_info_t* p_source_buffer)
|
||||
{
|
||||
memcpy(p_source_buffer->buf + (OPJ_SIZE_T)p_source_buffer->off, p_buffer, p_nb_bytes);
|
||||
p_source_buffer->off += (OPJ_OFF_T)p_nb_bytes;
|
||||
p_source_buffer->len += p_nb_bytes;
|
||||
|
||||
return p_nb_bytes;
|
||||
}
|
||||
|
||||
static OPJ_OFF_T opj_skip_from_buffer(OPJ_OFF_T p_nb_bytes,
|
||||
opj_buf_info_t * p_source_buffer)
|
||||
{
|
||||
if (p_source_buffer->off + p_nb_bytes < (OPJ_OFF_T)p_source_buffer->len) {
|
||||
p_source_buffer->off += p_nb_bytes;
|
||||
}
|
||||
else {
|
||||
p_source_buffer->off = (OPJ_OFF_T)p_source_buffer->len;
|
||||
}
|
||||
return p_nb_bytes;
|
||||
}
|
||||
|
||||
static OPJ_BOOL opj_seek_from_buffer(OPJ_OFF_T p_nb_bytes,
|
||||
opj_buf_info_t * p_source_buffer)
|
||||
{
|
||||
if (p_nb_bytes < (OPJ_OFF_T)p_source_buffer->len) {
|
||||
p_source_buffer->off = p_nb_bytes;
|
||||
}
|
||||
else {
|
||||
p_source_buffer->off = (OPJ_OFF_T)p_source_buffer->len;
|
||||
}
|
||||
return OPJ_TRUE;
|
||||
}
|
||||
|
||||
/* ---------------------------------------------------------------------- */
|
||||
|
||||
|
||||
|
|
@ -1089,275 +1000,21 @@ opj_stream_t* OPJ_CALLCONV opj_stream_create_file_stream (
|
|||
}
|
||||
|
||||
/* ---------------------------------------------------------------------- */
|
||||
|
||||
static void opj_set_up_buffer_stream(opj_stream_t* l_stream, OPJ_SIZE_T len, OPJ_BOOL p_is_read_stream) {
|
||||
opj_stream_set_user_data_length(l_stream, len);
|
||||
|
||||
if (p_is_read_stream) {
|
||||
opj_stream_set_read_function(l_stream, (opj_stream_read_fn)opj_read_from_buffer);
|
||||
opj_stream_set_zero_copy_read_function(l_stream, (opj_stream_zero_copy_read_fn)opj_zero_copy_read_from_buffer);
|
||||
}
|
||||
else
|
||||
opj_stream_set_write_function(l_stream, (opj_stream_write_fn)opj_write_from_buffer);
|
||||
opj_stream_set_skip_function(l_stream, (opj_stream_skip_fn)opj_skip_from_buffer);
|
||||
opj_stream_set_seek_function(l_stream, (opj_stream_seek_fn)opj_seek_from_buffer);
|
||||
}
|
||||
|
||||
opj_stream_t* OPJ_CALLCONV opj_stream_create_buffer_stream(OPJ_BYTE *buf,
|
||||
OPJ_SIZE_T len,
|
||||
OPJ_BOOL p_is_read_stream)
|
||||
{
|
||||
opj_stream_t* l_stream;
|
||||
opj_buf_info_t* p_source_buffer = NULL;
|
||||
OPJ_SIZE_T len,
|
||||
OPJ_BOOL p_is_read_stream) {
|
||||
return opj_create_buffer_stream(buf, len, p_is_read_stream);
|
||||
|
||||
if (!buf || !len)
|
||||
return NULL;
|
||||
|
||||
p_source_buffer = (opj_buf_info_t*)opj_malloc(sizeof(opj_buf_info_t));
|
||||
if (!p_source_buffer)
|
||||
return NULL;
|
||||
|
||||
l_stream = opj_stream_create(0,p_is_read_stream);
|
||||
if (!l_stream) {
|
||||
opj_free(p_source_buffer);
|
||||
return NULL;
|
||||
|
||||
}
|
||||
|
||||
memset(p_source_buffer, 0, sizeof(opj_buf_info_t));
|
||||
p_source_buffer->buf = buf;
|
||||
p_source_buffer->off = 0;
|
||||
p_source_buffer->len = len;
|
||||
|
||||
opj_stream_set_user_data(l_stream, p_source_buffer, opj_free_buffer_info);
|
||||
opj_set_up_buffer_stream(l_stream, p_source_buffer->len, p_is_read_stream);
|
||||
return l_stream;
|
||||
}
|
||||
|
||||
/* ---------------------------------------------------------------------- */
|
||||
|
||||
OPJ_INT32 opj_get_file_open_mode(const char* mode)
|
||||
{
|
||||
OPJ_INT32 m = -1;
|
||||
switch (mode[0]) {
|
||||
case 'r':
|
||||
m = O_RDONLY;
|
||||
if (mode[1] == '+')
|
||||
m = O_RDWR;
|
||||
break;
|
||||
case 'w':
|
||||
case 'a':
|
||||
m = O_RDWR | O_CREAT;
|
||||
if (mode[0] == 'w')
|
||||
m |= O_TRUNC;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return m;
|
||||
}
|
||||
|
||||
|
||||
#ifdef _WIN32
|
||||
|
||||
static OPJ_UINT64 opj_size_proc(opj_handle_t fd)
|
||||
{
|
||||
ULARGE_INTEGER m;
|
||||
m.LowPart = GetFileSize(fd, &m.HighPart);
|
||||
return(m.QuadPart);
|
||||
}
|
||||
|
||||
|
||||
static void* opj_map(opj_handle_t fd, OPJ_SIZE_T len) {
|
||||
void* ptr = NULL;
|
||||
HANDLE hMapFile = NULL;
|
||||
|
||||
if (!fd || !len)
|
||||
return NULL;
|
||||
|
||||
/* Passing in 0 for the maximum file size indicates that we
|
||||
would like to create a file mapping object for the full file size */
|
||||
hMapFile = CreateFileMapping(fd, NULL, PAGE_READONLY, 0, 0, NULL);
|
||||
if (hMapFile == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
ptr = MapViewOfFile(hMapFile, FILE_MAP_READ, 0, 0, 0);
|
||||
CloseHandle(hMapFile);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_unmap(void* ptr, OPJ_SIZE_T len){
|
||||
OPJ_INT32 rc = -1;
|
||||
(void)len;
|
||||
if (ptr)
|
||||
rc = UnmapViewOfFile(ptr) ? 0 : -1;
|
||||
return rc;
|
||||
}
|
||||
|
||||
static opj_handle_t opj_open_fd(const char* fname, const char* mode) {
|
||||
void* fd = NULL;
|
||||
OPJ_INT32 m = -1;
|
||||
DWORD dwMode = 0;
|
||||
|
||||
if (!fname)
|
||||
return (opj_handle_t)-1;
|
||||
|
||||
|
||||
m = opj_get_file_open_mode(mode);
|
||||
switch (m) {
|
||||
case O_RDONLY:
|
||||
dwMode = OPEN_EXISTING;
|
||||
break;
|
||||
case O_RDWR:
|
||||
dwMode = OPEN_ALWAYS;
|
||||
break;
|
||||
case O_RDWR | O_CREAT:
|
||||
dwMode = OPEN_ALWAYS;
|
||||
break;
|
||||
case O_RDWR | O_TRUNC:
|
||||
dwMode = CREATE_ALWAYS;
|
||||
break;
|
||||
case O_RDWR | O_CREAT | O_TRUNC:
|
||||
dwMode = CREATE_ALWAYS;
|
||||
break;
|
||||
default:
|
||||
return NULL;
|
||||
}
|
||||
|
||||
fd = (opj_handle_t)CreateFileA(fname,
|
||||
(m == O_RDONLY) ? GENERIC_READ : (GENERIC_READ | GENERIC_WRITE),
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, dwMode,
|
||||
(m == O_RDONLY) ? FILE_ATTRIBUTE_READONLY : FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
if (fd == INVALID_HANDLE_VALUE) {
|
||||
return (opj_handle_t)-1;
|
||||
}
|
||||
return fd;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_close_fd(opj_handle_t fd) {
|
||||
OPJ_INT32 rc = -1;
|
||||
if (fd) {
|
||||
rc = CloseHandle(fd) ? 0 : -1;
|
||||
}
|
||||
return rc;
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
static OPJ_UINT64 opj_size_proc(opj_handle_t fd)
|
||||
{
|
||||
struct stat sb;
|
||||
if (!fd)
|
||||
return 0;
|
||||
|
||||
if (fstat(fd, &sb)<0)
|
||||
return(0);
|
||||
else
|
||||
return((OPJ_UINT64)sb.st_size);
|
||||
}
|
||||
|
||||
static void* opj_map(opj_handle_t fd, OPJ_SIZE_T len) {
|
||||
void* ptr = NULL;
|
||||
OPJ_UINT64 size64 = 0;
|
||||
|
||||
if (!fd)
|
||||
return NULL;
|
||||
|
||||
size64 = opj_size_proc(fd);
|
||||
ptr = (void*)mmap(0, (size_t)size64, PROT_READ, MAP_SHARED, fd, 0);
|
||||
return ptr == (void*)-1 ? NULL : ptr;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_unmap(void* ptr, OPJ_SIZE_T len) {
|
||||
if (ptr)
|
||||
munmap(ptr, len);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static opj_handle_t opj_open_fd(const char* fname, const char* mode) {
|
||||
opj_handle_t fd = 0;
|
||||
OPJ_INT32 m = -1;
|
||||
if (!fname) {
|
||||
return (opj_handle_t )-1;
|
||||
}
|
||||
m = opj_get_file_open_mode(mode);
|
||||
fd = open(fname, m, 0666);
|
||||
if (fd < 0) {
|
||||
#ifdef DEBUG_ERRNO
|
||||
if (errno > 0 && strerror(errno) != NULL) {
|
||||
printf("%s: %s", fname, strerror(errno));
|
||||
}
|
||||
else {
|
||||
printf("%s: Cannot open", fname);
|
||||
}
|
||||
#endif
|
||||
return (opj_handle_t )-1;
|
||||
}
|
||||
return fd;
|
||||
}
|
||||
|
||||
static OPJ_INT32 opj_close_fd(opj_handle_t fd) {
|
||||
if (!fd)
|
||||
return 0;
|
||||
return (close(fd));
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
static void opj_mem_map_free(void* user_data) {
|
||||
if (user_data) {
|
||||
opj_buf_info_t* buffer_info = (opj_buf_info_t*)user_data;
|
||||
opj_unmap(buffer_info->buf, buffer_info->len);
|
||||
opj_close_fd(buffer_info->fd);
|
||||
opj_free(buffer_info);
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
Currently, only read streams are supported for memory mapped files.
|
||||
*/
|
||||
opj_stream_t* OPJ_CALLCONV opj_stream_create_mapped_file_read_stream(const char *fname)
|
||||
{
|
||||
opj_stream_t* l_stream = NULL;
|
||||
opj_buf_info_t* buffer_info = NULL;
|
||||
void* mapped_view = NULL;
|
||||
OPJ_BOOL p_is_read_stream = OPJ_TRUE;
|
||||
|
||||
opj_handle_t fd = opj_open_fd(fname, p_is_read_stream ? "r" : "w");
|
||||
if (fd == (opj_handle_t)-1)
|
||||
return NULL;
|
||||
|
||||
buffer_info = (opj_buf_info_t*)opj_malloc(sizeof(opj_buf_info_t));
|
||||
memset(buffer_info, 0, sizeof(opj_buf_info_t));
|
||||
buffer_info->fd = fd;
|
||||
buffer_info->len = (OPJ_SIZE_T)opj_size_proc(fd);
|
||||
|
||||
l_stream = opj_stream_create(0, p_is_read_stream);
|
||||
if (!l_stream) {
|
||||
opj_mem_map_free(buffer_info);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
mapped_view = opj_map(fd, buffer_info->len);
|
||||
if (!mapped_view) {
|
||||
opj_stream_destroy(l_stream);
|
||||
opj_mem_map_free(buffer_info);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
buffer_info->buf = mapped_view;
|
||||
buffer_info->off = 0;
|
||||
|
||||
opj_stream_set_user_data(l_stream, buffer_info, (opj_stream_free_user_data_fn)opj_mem_map_free);
|
||||
opj_set_up_buffer_stream(l_stream, buffer_info->len, p_is_read_stream);
|
||||
|
||||
|
||||
return l_stream;
|
||||
return opj_create_mapped_file_read_stream(fname);
|
||||
}
|
||||
|
||||
|
||||
/* ---------------------------------------------------------------------- */
|
||||
|
||||
void OPJ_CALLCONV opj_image_all_components_data_free(opj_image_t* image) {
|
||||
OPJ_UINT32 i;
|
||||
if (!image || !image->comps)
|
||||
|
|
|
|||
|
|
@ -172,6 +172,7 @@ static INLINE long opj_lrintf(float f) {
|
|||
# pragma intrinsic(__emul)
|
||||
#endif
|
||||
|
||||
#include "mem_stream.h"
|
||||
#include "opj_inttypes.h"
|
||||
#include "opj_clock.h"
|
||||
#include "opj_malloc.h"
|
||||
|
|
|
|||
Loading…
Reference in New Issue