Page Menu
Home
Phabricator
Search
Configure Global Search
Log In
Files
F4324355
No One
Temporary
Actions
View File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Size
210 KB
Referenced Files
None
Subscribers
None
View Options
diff --git a/codelite/xmount.project b/codelite/xmount.project
index 584d48e..2628180 100644
--- a/codelite/xmount.project
+++ b/codelite/xmount.project
@@ -1,251 +1,259 @@
<?xml version="1.0" encoding="UTF-8"?>
<CodeLite_Project Name="xmount" InternalType="Console">
<Description/>
<Dependencies/>
<VirtualDirectory Name="src">
<File Name="../src/CMakeLists.txt"/>
<File Name="../src/macros.h"/>
<File Name="../src/md5.c"/>
<File Name="../src/md5.h"/>
<File Name="../src/xmount.c"/>
<File Name="../src/xmount.h"/>
<File Name="../src/xmount_fuse.c"/>
<File Name="../src/xmount_fuse.h"/>
+ <File Name="../src/xmount_input.c"/>
+ <File Name="../src/xmount_input.h"/>
+ <File Name="../src/xmount_morphing.c"/>
+ <File Name="../src/xmount_morphing.h"/>
+ <File Name="../src/xmount_output.c"/>
+ <File Name="../src/xmount_output.h"/>
+ <File Name="../src/xmount_cache.c"/>
+ <File Name="../src/xmount_cache.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount">
<File Name="../libxmount/endianness.h"/>
<File Name="../libxmount/libxmount.c"/>
<File Name="../libxmount/libxmount.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_input">
<File Name="../libxmount_input/CMakeLists.txt"/>
<File Name="../libxmount_input/libxmount_input.h"/>
<VirtualDirectory Name="libxmount_input_aaff">
<File Name="../libxmount_input/libxmount_input_aaff/CMakeLists.txt"/>
<File Name="../libxmount_input/libxmount_input_aaff/libxmount_input_aaff.c"/>
<File Name="../libxmount_input/libxmount_input_aaff/libxmount_input_aaff.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_input_aewf">
<File Name="../libxmount_input/libxmount_input_aewf/CMakeLists.txt"/>
<File Name="../libxmount_input/libxmount_input_aewf/libxmount_input_aewf.c"/>
<File Name="../libxmount_input/libxmount_input_aewf/libxmount_input_aewf.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_input_aff">
<File Name="../libxmount_input/libxmount_input_aff/CMakeLists.txt"/>
<File Name="../libxmount_input/libxmount_input_aff/libxmount_input_aff.c"/>
<File Name="../libxmount_input/libxmount_input_aff/libxmount_input_aff.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_input_ewf">
<File Name="../libxmount_input/libxmount_input_ewf/CMakeLists.txt"/>
<File Name="../libxmount_input/libxmount_input_ewf/libxmount_input_ewf.c"/>
<File Name="../libxmount_input/libxmount_input_ewf/libxmount_input_ewf.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_input_raw">
<File Name="../libxmount_input/libxmount_input_raw/CMakeLists.txt"/>
<File Name="../libxmount_input/libxmount_input_raw/libxmount_input_raw.c"/>
<File Name="../libxmount_input/libxmount_input_raw/libxmount_input_raw.h"/>
</VirtualDirectory>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_morphing">
<VirtualDirectory Name="libxmount_morphing_combine">
<File Name="../libxmount_morphing/libxmount_morphing_combine/CMakeLists.txt"/>
<File Name="../libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.c"/>
<File Name="../libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_morphing_raid">
<File Name="../libxmount_morphing/libxmount_morphing_raid/CMakeLists.txt"/>
<File Name="../libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.c"/>
<File Name="../libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.h"/>
</VirtualDirectory>
<VirtualDirectory Name="libxmount_morphing_unallocated">
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/CMakeLists.txt"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/fat_functions.c"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/fat_functions.h"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.c"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.h"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.c"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.h"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated_retvalues.h"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.c"/>
<File Name="../libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.h"/>
</VirtualDirectory>
<File Name="../libxmount_morphing/CMakeLists.txt"/>
<File Name="../libxmount_morphing/libxmount_morphing.h"/>
</VirtualDirectory>
<Settings Type="Executable">
<GlobalSettings>
<Compiler Options="" C_Options="" Assembler="">
<IncludePath Value="."/>
</Compiler>
<Linker Options="">
<LibraryPath Value="."/>
</Linker>
<ResourceCompiler Options=""/>
</GlobalSettings>
<Configuration Name="Debug_Linux" CompilerType="gnu gcc" DebuggerType="GNU gdb debugger" Type="Executable" BuildCmpWithGlobalSettings="append" BuildLnkWithGlobalSettings="append" BuildResWithGlobalSettings="append">
<Compiler Options="-g;-O0;-Wall" C_Options="-g;-O0;-Wall" Assembler="" Required="yes" PreCompiledHeader="" PCHInCommandLine="no" PCHFlags="" PCHFlagsPolicy="0">
<IncludePath Value="."/>
</Compiler>
<Linker Options="" Required="yes"/>
<ResourceCompiler Options="" Required="no"/>
<General OutputFile="$(IntermediateDirectory)/$(ProjectName)" IntermediateDirectory="./Debug" Command="./build-debug/HelloWorld" CommandArguments="" UseSeparateDebugArgs="no" DebugArguments="" WorkingDirectory="" PauseExecWhenProcTerminates="yes" IsGUIProgram="no" IsEnabled="yes"/>
<Environment EnvVarSetName="<Use Defaults>" DbgSetName="<Use Defaults>">
<![CDATA[]]>
</Environment>
<Debugger IsRemote="no" RemoteHostName="" RemoteHostPort="" DebuggerPath="" IsExtended="no">
<DebuggerSearchPaths/>
<PostConnectCommands/>
<StartupCommands/>
</Debugger>
<PreBuild/>
<PostBuild/>
<CustomBuild Enabled="yes">
<Target Name="cmake">cmake ../.. -DCMAKE_BUILD_TYPE=Debug -DCMAKE_EXPORT_COMPILE_COMMANDS=1</Target>
<RebuildCommand>make clean && make -j4</RebuildCommand>
<CleanCommand>make clean</CleanCommand>
<BuildCommand>make -j4</BuildCommand>
<PreprocessFileCommand/>
<SingleFileCommand/>
<MakefileGenerationCommand/>
<ThirdPartyToolName>None</ThirdPartyToolName>
<WorkingDirectory>$(WorkspacePath)/build-debug</WorkingDirectory>
</CustomBuild>
<AdditionalRules>
<CustomPostBuild/>
<CustomPreBuild/>
</AdditionalRules>
<Completion EnableCpp11="no" EnableCpp14="no">
<ClangCmpFlagsC/>
<ClangCmpFlags/>
<ClangPP/>
<SearchPaths/>
</Completion>
</Configuration>
<Configuration Name="Debug_Windows" CompilerType="gnu gcc" DebuggerType="GNU gdb debugger" Type="Executable" BuildCmpWithGlobalSettings="append" BuildLnkWithGlobalSettings="append" BuildResWithGlobalSettings="append">
<Compiler Options="-g;-O0;-Wall" C_Options="-g;-O0;-Wall" Assembler="" Required="yes" PreCompiledHeader="" PCHInCommandLine="no" PCHFlags="" PCHFlagsPolicy="0">
<IncludePath Value="."/>
</Compiler>
<Linker Options="" Required="yes"/>
<ResourceCompiler Options="" Required="no"/>
<General OutputFile="$(IntermediateDirectory)/$(ProjectName)" IntermediateDirectory="./Debug" Command="./build-debug/HelloWorld" CommandArguments="" UseSeparateDebugArgs="no" DebugArguments="" WorkingDirectory="" PauseExecWhenProcTerminates="yes" IsGUIProgram="no" IsEnabled="yes"/>
<Environment EnvVarSetName="<Use Defaults>" DbgSetName="<Use Defaults>">
<![CDATA[]]>
</Environment>
<Debugger IsRemote="no" RemoteHostName="" RemoteHostPort="" DebuggerPath="" IsExtended="no">
<DebuggerSearchPaths/>
<PostConnectCommands/>
<StartupCommands/>
</Debugger>
<PreBuild/>
<PostBuild/>
<CustomBuild Enabled="yes">
<Target Name="cmake">cmake .. -G "MinGW Makefiles" -DCMAKE_BUILD_TYPE=Debug -DCMAKE_EXPORT_COMPILE_COMMANDS=1</Target>
<RebuildCommand>mingw32-make clean && mingw32-make -j4</RebuildCommand>
<CleanCommand>mingw32-make clean</CleanCommand>
<BuildCommand>mingw32-make -j4</BuildCommand>
<PreprocessFileCommand/>
<SingleFileCommand/>
<MakefileGenerationCommand/>
<ThirdPartyToolName>None</ThirdPartyToolName>
<WorkingDirectory>$(WorkspacePath)/build-debug</WorkingDirectory>
</CustomBuild>
<AdditionalRules>
<CustomPostBuild/>
<CustomPreBuild/>
</AdditionalRules>
<Completion EnableCpp11="no" EnableCpp14="no">
<ClangCmpFlagsC/>
<ClangCmpFlags/>
<ClangPP/>
<SearchPaths/>
</Completion>
</Configuration>
<Configuration Name="Release_Linux" CompilerType="gnu gcc" DebuggerType="GNU gdb debugger" Type="Executable" BuildCmpWithGlobalSettings="append" BuildLnkWithGlobalSettings="append" BuildResWithGlobalSettings="append">
<Compiler Options="-g;-O0;-Wall" C_Options="-g;-O0;-Wall" Assembler="" Required="yes" PreCompiledHeader="" PCHInCommandLine="no" PCHFlags="" PCHFlagsPolicy="0">
<IncludePath Value="."/>
</Compiler>
<Linker Options="" Required="yes"/>
<ResourceCompiler Options="" Required="no"/>
<General OutputFile="$(IntermediateDirectory)/$(ProjectName)" IntermediateDirectory="./Release" Command="./build-release/HelloWorld" CommandArguments="" UseSeparateDebugArgs="no" DebugArguments="" WorkingDirectory="" PauseExecWhenProcTerminates="yes" IsGUIProgram="no" IsEnabled="yes"/>
<Environment EnvVarSetName="<Use Defaults>" DbgSetName="<Use Defaults>">
<![CDATA[]]>
</Environment>
<Debugger IsRemote="no" RemoteHostName="" RemoteHostPort="" DebuggerPath="" IsExtended="no">
<DebuggerSearchPaths/>
<PostConnectCommands/>
<StartupCommands/>
</Debugger>
<PreBuild/>
<PostBuild/>
<CustomBuild Enabled="yes">
<Target Name="cmake">cmake ../.. -DCMAKE_EXPORT_COMPILE_COMMANDS=1</Target>
<RebuildCommand>make clean && make -j4</RebuildCommand>
<CleanCommand>make clean</CleanCommand>
<BuildCommand>make -j4</BuildCommand>
<PreprocessFileCommand/>
<SingleFileCommand/>
<MakefileGenerationCommand/>
<ThirdPartyToolName>None</ThirdPartyToolName>
<WorkingDirectory>$(WorkspacePath)/build-release</WorkingDirectory>
</CustomBuild>
<AdditionalRules>
<CustomPostBuild/>
<CustomPreBuild/>
</AdditionalRules>
<Completion EnableCpp11="no" EnableCpp14="no">
<ClangCmpFlagsC/>
<ClangCmpFlags/>
<ClangPP/>
<SearchPaths/>
</Completion>
</Configuration>
<Configuration Name="Release_Windows" CompilerType="gnu gcc" DebuggerType="GNU gdb debugger" Type="Executable" BuildCmpWithGlobalSettings="append" BuildLnkWithGlobalSettings="append" BuildResWithGlobalSettings="append">
<Compiler Options="-g;-O0;-Wall" C_Options="-g;-O0;-Wall" Assembler="" Required="yes" PreCompiledHeader="" PCHInCommandLine="no" PCHFlags="" PCHFlagsPolicy="0">
<IncludePath Value="."/>
</Compiler>
<Linker Options="" Required="yes"/>
<ResourceCompiler Options="" Required="no"/>
<General OutputFile="$(IntermediateDirectory)/$(ProjectName)" IntermediateDirectory="./Release" Command="./build-release/HelloWorld" CommandArguments="" UseSeparateDebugArgs="no" DebugArguments="" WorkingDirectory="" PauseExecWhenProcTerminates="yes" IsGUIProgram="no" IsEnabled="yes"/>
<Environment EnvVarSetName="<Use Defaults>" DbgSetName="<Use Defaults>">
<![CDATA[]]>
</Environment>
<Debugger IsRemote="no" RemoteHostName="" RemoteHostPort="" DebuggerPath="" IsExtended="no">
<DebuggerSearchPaths/>
<PostConnectCommands/>
<StartupCommands/>
</Debugger>
<PreBuild/>
<PostBuild/>
<CustomBuild Enabled="yes">
<Target Name="cmake">cmake .. -G "MinGW Makefiles" -DCMAKE_EXPORT_COMPILE_COMMANDS=1</Target>
<RebuildCommand>mingw32-make clean && mingw32-make -j4</RebuildCommand>
<CleanCommand>mingw32-make clean</CleanCommand>
<BuildCommand>mingw32-make -j4</BuildCommand>
<PreprocessFileCommand/>
<SingleFileCommand/>
<MakefileGenerationCommand/>
<ThirdPartyToolName>None</ThirdPartyToolName>
<WorkingDirectory>$(WorkspacePath)/build-release</WorkingDirectory>
</CustomBuild>
<AdditionalRules>
<CustomPostBuild/>
<CustomPreBuild/>
</AdditionalRules>
<Completion EnableCpp11="no" EnableCpp14="no">
<ClangCmpFlagsC/>
<ClangCmpFlags/>
<ClangPP/>
<SearchPaths/>
</Completion>
</Configuration>
</Settings>
<VirtualDirectory Name="libxmount_output">
<File Name="../libxmount_output/CMakeLists.txt"/>
<File Name="../libxmount_output/libxmount_output.h"/>
<VirtualDirectory Name="libxmount_output_raw">
<File Name="../libxmount_output/libxmount_output_raw/CMakeLists.txt"/>
<File Name="../libxmount_output/libxmount_output_raw/libxmount_output_raw.c"/>
<File Name="../libxmount_output/libxmount_output_raw/libxmount_output_raw.h"/>
</VirtualDirectory>
</VirtualDirectory>
</CodeLite_Project>
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index a1445d1..e649374 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -1,22 +1,22 @@
if(NOT APPLE)
include_directories(${LIBFUSE_INCLUDE_DIRS})
set(LIBS ${LIBS} ${LIBFUSE_LIBRARIES})
else(NOT APPLE)
include_directories(${LIBOSXFUSE_INCLUDE_DIRS})
set(LIBS ${LIBS} ${LIBOSXFUSE_LIBRARIES})
link_directories(${LIBOSXFUSE_LIBRARY_DIRS})
endif(NOT APPLE)
set(LIBS ${LIBS} ${libgidafs_LIBRARIES})
include_directories(${libgidafs_INCLUDE_DIRS})
set(LIBS ${LIBS} "dl")
add_definitions(-DXMOUNT_LIBRARY_PATH="${CMAKE_INSTALL_PREFIX}/lib/xmount")
-add_executable(xmount xmount.c xmount_fuse.c md5.c ../libxmount/libxmount.c)
+add_executable(xmount xmount.c xmount_input.c xmount_morphing.c xmount_cache.c xmount_output.c xmount_fuse.c md5.c ../libxmount/libxmount.c)
target_link_libraries(xmount ${LIBS})
install(TARGETS xmount DESTINATION bin)
diff --git a/src/xmount.c b/src/xmount.c
index a60b4e1..cae2851 100755
--- a/src/xmount.c
+++ b/src/xmount.c
@@ -1,2930 +1,2170 @@
/*******************************************************************************
* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
* *
* xmount is a small tool to "fuse mount" various harddisk image formats as dd, *
* vdi, vhd or vmdk files and enable virtual write access to them. *
* *
* This program is free software: you can redistribute it and/or modify it *
* under the terms of the GNU General Public License as published by the Free *
* Software Foundation, either version 3 of the License, or (at your option) *
* any later version. *
* *
* This program is distributed in the hope that it will be useful, but WITHOUT *
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
* more details. *
* *
* You should have received a copy of the GNU General Public License along with *
* this program. If not, see <http://www.gnu.org/licenses/>. *
*******************************************************************************/
#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <inttypes.h> // For PRI*
#include <errno.h>
#include <dlfcn.h> // For dlopen, dlclose, dlsym
#include <dirent.h> // For opendir, readdir, closedir
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/stat.h> // For fstat
#include <sys/types.h>
#ifdef HAVE_LINUX_FS_H
#include <linux/fs.h> // For SEEK_* ??
#endif
#if !defined(__APPLE__) && defined(HAVE_GRP_H) && defined(HAVE_PWD_H)
#include <grp.h> // For getgrnam, struct group
#include <pwd.h> // For getpwuid, struct passwd
#endif
#include <pthread.h>
#include <time.h> // For time
#include "xmount.h"
#include "xmount_fuse.h"
#include "md5.h"
#include "macros.h"
#include "../libxmount/libxmount.h"
#define XMOUNT_COPYRIGHT_NOTICE \
"xmount v%s Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu>"
#define LOG_WARNING(...) { \
LIBXMOUNT_LOG_WARNING(__VA_ARGS__); \
}
#define LOG_ERROR(...) { \
LIBXMOUNT_LOG_ERROR(__VA_ARGS__); \
}
#define LOG_DEBUG(...) { \
LIBXMOUNT_LOG_DEBUG(glob_xmount.debug,__VA_ARGS__); \
}
+#define HASH_AMOUNT (1024*1024)*10 // Amount of data used to construct a
+ // "unique" hash for every input image
+ // (10MByte)
/*******************************************************************************
* Forward declarations
******************************************************************************/
/*
* Misc
*/
static void InitResources();
static void FreeResources();
static void PrintUsage(char*);
static void CheckFuseSettings();
static int ParseCmdLine(const int, char**);
static int ExtractOutputFileNames(char*);
static int CalculateInputImageHash(uint64_t*, uint64_t*);
-/*
- * Input
- */
-static int ReadInputImageData(pts_InputImage, char*, off_t, size_t, size_t*);
-/*
- * Morphing
- */
-static int GetMorphedImageSize(uint64_t*);
-static int ReadMorphedImageData(char*, off_t, size_t, size_t*);
-static int WriteMorphedImageData(const char*, off_t, size_t, size_t*);
-/*
- * Cache
- */
-static int InitCacheFile();
-static int UpdateBlockCacheIndex(uint64_t, t_CacheFileBlockIndex);
-
/*
* Info file
*/
static int InitInfoFile();
/*
* Lib related
*/
static int LoadLibs();
static int FindInputLib(pts_InputImage);
static int FindMorphingLib();
static int FindOutputLib();
static int SplitLibraryParameters(char*, uint32_t*, pts_LibXmountOptions**);
/*
* Functions exported to LibXmount_Morphing
*/
static int LibXmount_Morphing_ImageCount(uint64_t*);
static int LibXmount_Morphing_Size(uint64_t, uint64_t*);
static int LibXmount_Morphing_Read(uint64_t, char*, off_t, size_t, size_t*);
/*
* Functions exported to LibXmount_Output
*/
static int LibXmount_Output_Size(uint64_t*);
static int LibXmount_Output_Read(char*, off_t, size_t, size_t*);
static int LibXmount_Output_Write(char*, off_t, size_t, size_t*);
/*******************************************************************************
* Helper functions
******************************************************************************/
//! Print usage instructions (cmdline options etc..)
/*!
* \param p_prog_name Program name (argv[0])
*/
static void PrintUsage(char *p_prog_name) {
char *p_buf;
int first;
int ret;
printf("\n" XMOUNT_COPYRIGHT_NOTICE "\n",XMOUNT_VERSION);
printf("\nUsage:\n");
printf(" %s [fopts] <xopts> <mntp>\n\n",p_prog_name);
printf("Options:\n");
printf(" fopts:\n");
printf(" -d : Enable FUSE's and xmount's debug mode.\n");
printf(" -h : Display this help message.\n");
printf(" -s : Run single threaded.\n");
printf(" -o no_allow_other : Disable automatic addition of FUSE's "
"allow_other option.\n");
printf(" -o <fopts> : Specify fuse mount options. Will also disable "
"automatic addition of FUSE's allow_other option!\n");
printf("\n");
printf(" xopts:\n");
printf(" --cache <cfile> : Enable virtual write support.\n");
printf(" <cfile> specifies the cache file to use.\n");
printf(" --in <itype> <ifile> : Input image format and source file(s). "
"May be specified multiple times.\n");
printf(" <itype> can be ");
// List supported input formats
first=1;
for(uint32_t i=0;i<glob_xmount.input.libs_count;i++) {
p_buf=glob_xmount.input.pp_libs[i]->p_supported_input_types;
while(*p_buf!='\0') {
if(first==1) {
printf("\"%s\"",p_buf);
first=0;
} else printf(", \"%s\"",p_buf);
p_buf+=(strlen(p_buf)+1);
}
}
printf(".\n");
printf(" <ifile> specifies the source file. If your image is split into "
"multiple files, you have to specify them all!\n");
printf(" --inopts <iopts> : Specify input library specific options.\n");
printf(" <iopts> specifies a comma separated list of key=value options. "
"See below for details.\n");
printf(" --info : Print out infos about used compiler and libraries.\n");
printf(" --morph <mtype> : Morphing function to apply to input image(s). "
"If not specified, defaults to \"combine\".\n");
printf(" <mtype> can be ");
// List supported morphing functions
first=1;
for(uint32_t i=0;i<glob_xmount.morphing.libs_count;i++) {
p_buf=glob_xmount.morphing.pp_libs[i]->p_supported_morphing_types;
while(*p_buf!='\0') {
if(first==1) {
printf("\"%s\"",p_buf);
first=0;
} else printf(", \"%s\"",p_buf);
p_buf+=(strlen(p_buf)+1);
}
}
printf(".\n");
printf(" --morphopts <mopts> : Specify morphing library specific "
"options.\n");
printf(" <mopts> specifies a comma separated list of key=value options. "
"See below for details.\n");
printf(" --offset <off> : Move the output image data start <off> bytes "
"into the input image(s).\n");
printf(" --out <otype> : Output image format. If not specified, "
"defaults to ");
#ifdef __APPLE__
printf("\"dmg\".\n");
#else
printf("\"raw\".\n");
#endif
printf(" <otype> can be ");
// List supported output formats
first=1;
for(uint32_t i=0;i<glob_xmount.output.libs_count;i++) {
p_buf=glob_xmount.output.pp_libs[i]->p_supported_output_formats;
while(*p_buf!='\0') {
if(first==1) {
printf("\"%s\"",p_buf);
first=0;
} else printf(", \"%s\"",p_buf);
p_buf+=(strlen(p_buf)+1);
}
}
printf(".\n");
printf(" --outopts <oopts> : Specify output library specific "
"options.\n");
printf(" --owcache <file> : Same as --cache <file> but overwrites "
"existing cache file.\n");
printf(" --sizelimit <size> : The data end of input image(s) is set to no "
"more than <size> bytes after the data start.\n");
printf(" --version : Same as --info.\n");
printf("\n");
printf(" mntp:\n");
printf(" Mount point where output image should be located.\n");
printf("\n");
printf("Infos:\n");
printf(" * One --in option and a mount point are mandatory!\n");
printf(" * If you specify --in multiple times, data from all images is "
"morphed into one output image using the specified morphing "
"function.\n");
printf(" * For VMDK emulation, you have to uncomment \"user_allow_other\" "
"in /etc/fuse.conf or run xmount as root.\n");
printf("\n");
printf("Input / Morphing library specific options:\n");
printf(" Input / Morphing libraries might support an own set of "
"options to configure / tune their behaviour.\n");
printf(" Libraries supporting this feature (if any) and their "
"options are listed below.\n");
printf("\n");
// List input, morphing and output lib options
for(uint32_t i=0;i<glob_xmount.input.libs_count;i++) {
ret=glob_xmount.input.pp_libs[i]->
lib_functions.OptionsHelp((const char**)&p_buf);
if(ret!=0) {
LOG_ERROR("Unable to get options help for library '%s': %s!\n",
glob_xmount.input.pp_libs[i]->p_name,
glob_xmount.input.pp_libs[i]->
lib_functions.GetErrorMessage(ret));
}
if(p_buf==NULL) continue;
printf(" - %s\n",glob_xmount.input.pp_libs[i]->p_name);
printf("%s",p_buf);
printf("\n");
ret=glob_xmount.input.pp_libs[i]->lib_functions.FreeBuffer(p_buf);
if(ret!=0) {
LOG_ERROR("Unable to free options help text from library '%s': %s!\n",
glob_xmount.input.pp_libs[i]->p_name,
glob_xmount.input.pp_libs[i]->
lib_functions.GetErrorMessage(ret));
}
}
for(uint32_t i=0;i<glob_xmount.morphing.libs_count;i++) {
ret=glob_xmount.morphing.pp_libs[i]->
lib_functions.OptionsHelp((const char**)&p_buf);
if(ret!=0) {
LOG_ERROR("Unable to get options help for library '%s': %s!\n",
glob_xmount.morphing.pp_libs[i]->p_name,
glob_xmount.morphing.pp_libs[i]->
lib_functions.GetErrorMessage(ret));
}
if(p_buf==NULL) continue;
printf(" - %s\n",glob_xmount.morphing.pp_libs[i]->p_name);
printf("%s",p_buf);
printf("\n");
}
for(uint32_t i=0;i<glob_xmount.output.libs_count;i++) {
ret=glob_xmount.output.pp_libs[i]->
lib_functions.OptionsHelp((const char**)&p_buf);
if(ret!=0) {
LOG_ERROR("Unable to get options help for library '%s': %s!\n",
glob_xmount.output.pp_libs[i]->p_name,
glob_xmount.output.pp_libs[i]->
lib_functions.GetErrorMessage(ret));
}
if(p_buf==NULL) continue;
printf(" - %s\n",glob_xmount.output.pp_libs[i]->p_name);
printf("%s",p_buf);
printf("\n");
}
}
//! Check fuse settings
/*!
* Check if FUSE allows us to pass the -o allow_other parameter. This only works
* if we are root or user_allow_other is set in /etc/fuse.conf.
*
* In addition, this function also checks if the user is member of the fuse
* group which is generally needed to use fuse at all.
*/
static void CheckFuseSettings() {
#if !defined(__APPLE__) && defined(HAVE_GRP_H) && defined(HAVE_PWD_H)
struct group *p_group;
struct passwd *p_passwd;
#endif
int found;
FILE *h_fuse_conf;
char line[256];
glob_xmount.may_set_fuse_allow_other=FALSE;
if(geteuid()==0) {
// Running as root, there should be no problems
glob_xmount.may_set_fuse_allow_other=TRUE;
return;
}
#if !defined(__APPLE__) && defined(HAVE_GRP_H) && defined(HAVE_PWD_H)
// Check if a fuse group exists and if so, make sure user is a member of it.
// Makes only sense on Linux because as far as I know osxfuse has no own group
p_group=getgrnam("fuse");
if(p_group!=NULL) {
// Get effective user name
p_passwd=getpwuid(geteuid());
if(p_passwd==NULL) {
printf("\nWARNING: Unable to determine your effective user name. If "
"mounting works, you can ignore this message.\n\n");
return;
}
// Check if user is member of fuse group
found=FALSE;
while(*(p_group->gr_mem)!=NULL) {
if(strcmp(*(p_group->gr_mem),p_passwd->pw_name)==0) {
found=TRUE;
break;
}
p_group->gr_mem++;
}
if(found==FALSE) {
printf("\nWARNING: You are not a member of the \"fuse\" group. This will "
"prevent you from mounting images using xmount. Please add "
"yourself to the \"fuse\" group using the command "
"\"sudo usermod -a -G fuse %s\" and reboot your system or "
"execute xmount as root.\n\n",
p_passwd->pw_name);
return;
}
} else {
printf("\nWARNING: Your system does not seem to have a \"fuse\" group. If "
"mounting works, you can ignore this message.\n\n");
}
#endif
// Read FUSE's config file /etc/fuse.conf and check for set user_allow_other
h_fuse_conf=(FILE*)FOPEN("/etc/fuse.conf","r");
if(h_fuse_conf!=NULL) {
// Search conf file for set user_allow_others
found=FALSE;
while(fgets(line,sizeof(line),h_fuse_conf)!=NULL) {
// TODO: This works as long as there is no other parameter beginning with
// "user_allow_other" :)
if(strncmp(line,"user_allow_other",16)==0) {
found=TRUE;
break;
}
}
fclose(h_fuse_conf);
if(found==TRUE) {
glob_xmount.may_set_fuse_allow_other=TRUE;
} else {
printf("\nWARNING: FUSE will not allow other users nor root to access "
"your virtual harddisk image. To change this behavior, please "
"add \"user_allow_other\" to /etc/fuse.conf or execute xmount "
"as root.\n\n");
}
} else {
printf("\nWARNING: Unable to open /etc/fuse.conf. If mounting works, you "
"can ignore this message. If you encounter issues, please create "
"the file and add a single line containing the string "
"\"user_allow_other\" or execute xmount as root.\n\n");
return;
}
}
//! Parse command line options
/*!
* \param argc Number of cmdline params
* \param pp_argv Array containing cmdline params
* \return TRUE on success, FALSE on error
*/
static int ParseCmdLine(const int argc, char **pp_argv) {
int i=1;
int FuseMinusOControl=TRUE;
int FuseAllowOther=TRUE;
int first;
char *p_buf;
pts_InputImage p_input_image=NULL;
int ret;
// add pp_argv[0] to FUSE's argv
XMOUNT_MALLOC(glob_xmount.pp_fuse_argv,char**,sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[0],pp_argv[0]);
glob_xmount.fuse_argc=1;
// Parse options
while(i<argc && *pp_argv[i]=='-') {
if(strlen(pp_argv[i])>1 && *(pp_argv[i]+1)!='-') {
// Options beginning with one - are mostly FUSE specific
if(strcmp(pp_argv[i],"-d")==0) {
// Enable FUSE's and xmount's debug mode
XMOUNT_REALLOC(glob_xmount.pp_fuse_argv,
char**,
(glob_xmount.fuse_argc+1)*sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc],
pp_argv[i])
glob_xmount.fuse_argc++;
glob_xmount.debug=TRUE;
} else if(strcmp(pp_argv[i],"-h")==0) {
// Print help message
PrintUsage(pp_argv[0]);
exit(0);
} else if(strcmp(pp_argv[i],"-o")==0) {
// Next parameter specifies fuse mount options
if((i+1)<argc) {
i++;
// As the user specified the -o option, we assume he knows what he is
// doing. We won't append allow_other automatically. And we allow him
// to disable allow_other by passing a single "-o no_allow_other"
// which won't be passed to FUSE as it is xmount specific.
if(strcmp(pp_argv[i],"no_allow_other")!=0) {
glob_xmount.fuse_argc+=2;
XMOUNT_REALLOC(glob_xmount.pp_fuse_argv,
char**,
glob_xmount.fuse_argc*sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-2],
pp_argv[i-1]);
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1],
pp_argv[i]);
FuseMinusOControl=FALSE;
} else FuseAllowOther=FALSE;
} else {
LOG_ERROR("Couldn't parse fuse mount options!\n")
return FALSE;
}
} else if(strcmp(pp_argv[i],"-s")==0) {
// Enable FUSE's single threaded mode
XMOUNT_REALLOC(glob_xmount.pp_fuse_argv,
char**,
(glob_xmount.fuse_argc+1)*sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc],
pp_argv[i]);
glob_xmount.fuse_argc++;
} else if(strcmp(pp_argv[i],"-V")==0) {
// Display FUSE version info
XMOUNT_REALLOC(glob_xmount.pp_fuse_argv,
char**,
(glob_xmount.fuse_argc+1)*sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc],
pp_argv[i]);
glob_xmount.fuse_argc++;
} else {
LOG_ERROR("Unknown command line option \"%s\"\n",pp_argv[i]);
return FALSE;
}
} else {
// Options beginning with -- are xmount specific
if(strcmp(pp_argv[i],"--cache")==0 /*|| strcmp(pp_argv[i],"--rw")==0*/) {
// Emulate writable access to mounted image
// Next parameter must be cache file to read/write changes from/to
if((i+1)<argc) {
i++;
XMOUNT_STRSET(glob_xmount.cache.p_cache_file,pp_argv[i])
glob_xmount.output.writable=TRUE;
} else {
LOG_ERROR("You must specify a cache file!\n")
return FALSE;
}
LOG_DEBUG("Enabling virtual write support using cache file \"%s\"\n",
glob_xmount.cache.p_cache_file)
} else if(strcmp(pp_argv[i],"--in")==0) {
// Specify input image type and source files
if((i+2)<argc) {
i++;
// Alloc and init new ts_InputImage struct
XMOUNT_MALLOC(p_input_image,pts_InputImage,sizeof(ts_InputImage));
XMOUNT_STRSET(p_input_image->p_type,pp_argv[i]);
p_input_image->pp_files=NULL;
p_input_image->p_functions=NULL;
p_input_image->p_handle=NULL;
// Parse input image filename(s) and add to p_input_image->pp_files
i++;
p_input_image->files_count=0;
while(i<(argc-1) && strncmp(pp_argv[i],"--",2)!=0) {
p_input_image->files_count++;
XMOUNT_REALLOC(p_input_image->pp_files,
char**,
p_input_image->files_count*sizeof(char*));
XMOUNT_STRSET(p_input_image->pp_files[p_input_image->files_count-1],
pp_argv[i]);
i++;
}
i--;
if(p_input_image->files_count==0) {
LOG_ERROR("No input files specified for \"--in %s\"!\n",
p_input_image->p_type)
free(p_input_image->p_type);
free(p_input_image);
return FALSE;
}
// Add input image struct to input image array
glob_xmount.input.images_count++;
XMOUNT_REALLOC(glob_xmount.input.pp_images,
pts_InputImage*,
glob_xmount.input.images_count*
sizeof(pts_InputImage));
glob_xmount.input.pp_images[glob_xmount.input.images_count-1]=
p_input_image;
} else {
LOG_ERROR("You must specify an input image type and source file!\n");
return FALSE;
}
} else if(strcmp(pp_argv[i],"--inopts")==0) {
// Set input lib options
if((i+1)<argc) {
i++;
if(glob_xmount.input.pp_lib_params==NULL) {
if(SplitLibraryParameters(pp_argv[i],
&(glob_xmount.input.lib_params_count),
&(glob_xmount.input.pp_lib_params)
)==FALSE)
{
LOG_ERROR("Unable to parse input library options '%s'!\n",
pp_argv[i]);
return FALSE;
}
} else {
LOG_ERROR("You can only specify --inopts once!")
return FALSE;
}
} else {
LOG_ERROR("You must specify special options!\n");
return FALSE;
}
} else if(strcmp(pp_argv[i],"--morph")==0) {
// Set morphing lib to use
if((i+1)<argc) {
i++;
if(glob_xmount.morphing.p_morph_type==NULL) {
XMOUNT_STRSET(glob_xmount.morphing.p_morph_type,pp_argv[i]);
} else {
LOG_ERROR("You can only specify --morph once!")
return FALSE;
}
} else {
LOG_ERROR("You must specify morphing type!\n");
return FALSE;
}
} else if(strcmp(pp_argv[i],"--morphopts")==0) {
// Set morphing lib options
if((i+1)<argc) {
i++;
if(glob_xmount.morphing.pp_lib_params==NULL) {
if(SplitLibraryParameters(pp_argv[i],
&(glob_xmount.morphing.lib_params_count),
&(glob_xmount.morphing.pp_lib_params)
)==FALSE)
{
LOG_ERROR("Unable to parse morphing library options '%s'!\n",
pp_argv[i]);
return FALSE;
}
} else {
LOG_ERROR("You can only specify --morphopts once!")
return FALSE;
}
} else {
LOG_ERROR("You must specify special morphing lib params!\n");
return FALSE;
}
} else if(strcmp(pp_argv[i],"--offset")==0) {
// Set input image offset
if((i+1)<argc) {
i++;
glob_xmount.input.image_offset=StrToUint64(pp_argv[i],&ret);
if(ret==0) {
LOG_ERROR("Unable to convert '%s' to a number!\n",pp_argv[i])
return FALSE;
}
} else {
LOG_ERROR("You must specify an offset!\n")
return FALSE;
}
LOG_DEBUG("Setting input image offset to \"%" PRIu64 "\"\n",
glob_xmount.input.image_offset)
} else if(strcmp(pp_argv[i],"--out")==0) {
// Set output lib to use
if((i+1)<argc) {
i++;
if(glob_xmount.output.p_output_format==NULL) {
XMOUNT_STRSET(glob_xmount.output.p_output_format,pp_argv[i]);
} else {
LOG_ERROR("You can only specify --out once!")
return FALSE;
}
} else {
LOG_ERROR("You must specify an output format!\n");
return FALSE;
}
} else if(strcmp(pp_argv[i],"--outopts")==0) {
// Set output lib options
if((i+1)<argc) {
i++;
if(glob_xmount.output.pp_lib_params==NULL) {
if(SplitLibraryParameters(pp_argv[i],
&(glob_xmount.output.lib_params_count),
&(glob_xmount.output.pp_lib_params)
)==FALSE)
{
LOG_ERROR("Unable to parse output library options '%s'!\n",
pp_argv[i]);
return FALSE;
}
} else {
LOG_ERROR("You can only specify --outopts once!")
return FALSE;
}
} else {
LOG_ERROR("You must specify special output lib params!\n");
return FALSE;
}
} else if(strcmp(pp_argv[i],"--owcache")==0) {
// Enable writable access to mounted image and overwrite existing cache
// Next parameter must be cache file to read/write changes from/to
if((i+1)<argc) {
i++;
XMOUNT_STRSET(glob_xmount.cache.p_cache_file,pp_argv[i])
glob_xmount.output.writable=TRUE;
glob_xmount.cache.overwrite_cache=TRUE;
} else {
LOG_ERROR("You must specify a cache file!\n")
return FALSE;
}
LOG_DEBUG("Enabling virtual write support overwriting cache file %s\n",
glob_xmount.cache.p_cache_file)
} else if(strcmp(pp_argv[i],"--sizelimit")==0) {
// Set input image size limit
if((i+1)<argc) {
i++;
glob_xmount.input.image_size_limit=StrToUint64(pp_argv[i],&ret);
if(ret==0) {
LOG_ERROR("Unable to convert '%s' to a number!\n",pp_argv[i])
return FALSE;
}
} else {
LOG_ERROR("You must specify a size limit!\n")
return FALSE;
}
LOG_DEBUG("Setting input image size limit to \"%" PRIu64 "\"\n",
glob_xmount.input.image_size_limit)
} else if(strcmp(pp_argv[i],"--version")==0 ||
strcmp(pp_argv[i],"--info")==0)
{
// Print xmount info
printf(XMOUNT_COPYRIGHT_NOTICE "\n\n",XMOUNT_VERSION);
#ifdef __GNUC__
printf(" compile timestamp: %s %s\n",__DATE__,__TIME__);
printf(" gcc version: %s\n",__VERSION__);
#endif
printf(" loaded input libraries:\n");
for(uint32_t ii=0;ii<glob_xmount.input.libs_count;ii++) {
printf(" - %s supporting ",glob_xmount.input.pp_libs[ii]->p_name);
p_buf=glob_xmount.input.pp_libs[ii]->p_supported_input_types;
first=TRUE;
while(*p_buf!='\0') {
if(first) {
printf("\"%s\"",p_buf);
first=FALSE;
} else printf(", \"%s\"",p_buf);
p_buf+=(strlen(p_buf)+1);
}
printf("\n");
}
printf(" loaded morphing libraries:\n");
for(uint32_t ii=0;ii<glob_xmount.morphing.libs_count;ii++) {
printf(" - %s supporting ",
glob_xmount.morphing.pp_libs[ii]->p_name);
p_buf=glob_xmount.morphing.pp_libs[ii]->p_supported_morphing_types;
first=TRUE;
while(*p_buf!='\0') {
if(first) {
printf("\"%s\"",p_buf);
first=FALSE;
} else printf(", \"%s\"",p_buf);
p_buf+=(strlen(p_buf)+1);
}
printf("\n");
}
printf(" loaded output libraries:\n");
for(uint32_t ii=0;ii<glob_xmount.output.libs_count;ii++) {
printf(" - %s supporting ",
glob_xmount.output.pp_libs[ii]->p_name);
p_buf=glob_xmount.output.pp_libs[ii]->p_supported_output_formats;
first=TRUE;
while(*p_buf!='\0') {
if(first) {
printf("\"%s\"",p_buf);
first=FALSE;
} else printf(", \"%s\"",p_buf);
p_buf+=(strlen(p_buf)+1);
}
printf("\n");
}
printf("\n");
exit(0);
} else {
LOG_ERROR("Unknown command line option \"%s\"\n",pp_argv[i]);
return FALSE;
}
}
i++;
}
// Extract mountpoint
if(i==(argc-1)) {
XMOUNT_STRSET(glob_xmount.p_mountpoint,pp_argv[argc-1])
XMOUNT_REALLOC(glob_xmount.pp_fuse_argv,
char**,
(glob_xmount.fuse_argc+1)*sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc],
glob_xmount.p_mountpoint);
glob_xmount.fuse_argc++;
} else {
LOG_ERROR("No mountpoint specified!\n")
return FALSE;
}
if(FuseMinusOControl==TRUE) {
// We control the -o flag, set subtype, fsname and allow_other options
glob_xmount.fuse_argc+=2;
XMOUNT_REALLOC(glob_xmount.pp_fuse_argv,
char**,
glob_xmount.fuse_argc*sizeof(char*));
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-2],"-o");
XMOUNT_STRSET(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1],
"subtype=xmount");
if(glob_xmount.input.images_count!=0) {
// Set name of first source file as fsname
XMOUNT_STRAPP(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1],
",fsname='");
// If possible, use full path
p_buf=realpath(glob_xmount.input.pp_images[0]->pp_files[0],NULL);
if(p_buf==NULL) {
XMOUNT_STRSET(p_buf,glob_xmount.input.pp_images[0]->pp_files[0]);
}
// Make sure fsname does not include some forbidden chars
for(uint32_t i=0;i<strlen(p_buf);i++) {
if(p_buf[i]=='\'') p_buf[i]='_';
}
// Set fsname
XMOUNT_STRAPP(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1],
p_buf);
XMOUNT_STRAPP(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1],
"'");
free(p_buf);
}
if(FuseAllowOther==TRUE) {
// Add "allow_other" option if allowed
if(glob_xmount.may_set_fuse_allow_other) {
XMOUNT_STRAPP(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1],
",allow_other");
}
}
}
return TRUE;
}
//! Extract output file name from input image name
/*!
* \param p_orig_name Name of input image (may include a path)
* \return TRUE on success, FALSE on error
*/
static int ExtractOutputFileNames(char *p_orig_name) {
char *tmp;
// Truncate any leading path
tmp=strrchr(p_orig_name,'/');
if(tmp!=NULL) p_orig_name=tmp+1;
// Extract file extension
tmp=strrchr(p_orig_name,'.');
// Set leading '/'
XMOUNT_STRSET(glob_xmount.output.p_virtual_image_path,"/");
XMOUNT_STRSET(glob_xmount.output.p_info_path,"/");
// Copy filename
if(tmp==NULL) {
// Input image filename has no extension
XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,p_orig_name);
XMOUNT_STRAPP(glob_xmount.output.p_info_path,p_orig_name);
XMOUNT_STRAPP(glob_xmount.output.p_info_path,".info");
} else {
XMOUNT_STRNAPP(glob_xmount.output.p_virtual_image_path,p_orig_name,
strlen(p_orig_name)-strlen(tmp));
XMOUNT_STRNAPP(glob_xmount.output.p_info_path,p_orig_name,
strlen(p_orig_name)-strlen(tmp));
XMOUNT_STRAPP(glob_xmount.output.p_info_path,".info");
}
// Add virtual file extensions
// TODO: Get from output lib and add
//XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,".dd");
LOG_DEBUG("Set virtual image name to \"%s\"\n",
glob_xmount.output.p_virtual_image_path);
LOG_DEBUG("Set virtual image info name to \"%s\"\n",
glob_xmount.output.p_info_path);
return TRUE;
}
-//! Get size of morphed image
-/*!
- * \param p_size Buf to save size to
- * \return TRUE on success, FALSE on error
- */
-static int GetMorphedImageSize(uint64_t *p_size) {
- int ret;
-
- ret=glob_xmount.morphing.p_functions->Size(glob_xmount.morphing.p_handle,
- p_size);
- if(ret!=0) {
- LOG_ERROR("Unable to get morphed image size: %s!\n",
- glob_xmount.morphing.p_functions->GetErrorMessage(ret));
- return FALSE;
- }
-
- return TRUE;
-}
-
-//! Get size of output image
-/*!
- * \param p_size Pointer to an uint64_t to which the size will be written to
- * \return TRUE on success, FALSE on error
- */
-int GetOutputImageSize(uint64_t *p_size) {
- int ret;
- uint64_t output_image_size=0;
-
- if(glob_xmount.output.image_size!=0) {
- *p_size=glob_xmount.output.image_size;
- return TRUE;
- }
-
- ret=glob_xmount.output.p_functions->Size(glob_xmount.output.p_handle,
- &output_image_size);
- if(ret!=0) {
- LOG_ERROR("Couldn't get output image size!\n")
- return FALSE;
- }
-
- glob_xmount.output.image_size=output_image_size;
- *p_size=output_image_size;
- return TRUE;
-}
-
-//! Read data from input image
-/*!
- * \param p_image Image from which to read data
- * \param p_buf Pointer to buffer to write read data to (must be preallocated!)
- * \param offset Offset at which data should be read
- * \param size Size of data which should be read (size of buffer)
- * \param p_read Number of read bytes on success
- * \return 0 on success, negated error code on error
- */
-static int ReadInputImageData(pts_InputImage p_image,
- char *p_buf,
- off_t offset,
- size_t size,
- size_t *p_read)
-{
- int ret;
- size_t to_read=0;
- int read_errno=0;
-
- LOG_DEBUG("Reading %zu bytes at offset %zu from input image '%s'\n",
- size,
- offset,
- p_image->pp_files[0]);
-
- // Make sure we aren't reading past EOF of image file
- if(offset>=p_image->size) {
- // Offset is beyond image size
- LOG_DEBUG("Offset %zu is at / beyond size of input image '%s'\n",
- offset,
- p_image->pp_files[0]);
- *p_read=0;
- return 0;
- }
- if(offset+size>p_image->size) {
- // Attempt to read data past EOF of image file
- to_read=p_image->size-offset;
- LOG_DEBUG("Attempt to read data past EOF of input image '%s'. "
- "Correcting size from %zu to %zu\n",
- p_image->pp_files[0],
- size,
- to_read);
- } else to_read=size;
-
- // Read data from image file (adding input image offset if one was specified)
- ret=p_image->p_functions->Read(p_image->p_handle,
- p_buf,
- offset+glob_xmount.input.image_offset,
- to_read,
- p_read,
- &read_errno);
- if(ret!=0) {
- LOG_ERROR("Couldn't read %zu bytes at offset %zu from input image "
- "'%s': %s!\n",
- to_read,
- offset,
- p_image->pp_files[0],
- p_image->p_functions->GetErrorMessage(ret));
- if(read_errno==0) return -EIO;
- else return (read_errno*(-1));
- }
-
- return 0;
-}
-
-//! Read data from morphed image
-/*!
- * \param p_buf Pointer to buffer to write read data to (must be preallocated!)
- * \param offset Offset at which data should be read
- * \param size Size of data which should be read (size of buffer)
- * \param p_read Number of read bytes on success
- * \return TRUE on success, negated error code on error
- */
-static int ReadMorphedImageData(char *p_buf,
- off_t offset,
- size_t size,
- size_t *p_read)
-{
- uint64_t block_off=0;
- uint64_t cur_block=0;
- uint64_t cur_to_read=0;
- uint64_t image_size=0;
- size_t read=0;
- size_t to_read=0;
- int ret;
- teGidaFsError gidafs_ret=eGidaFsError_None;
-
- // Make sure we aren't reading past EOF of image file
- if(GetMorphedImageSize(&image_size)!=TRUE) {
- LOG_ERROR("Couldn't get size of morphed image!\n");
- return -EIO;
- }
- if(offset>=image_size) {
- // Offset is beyond image size
- LOG_DEBUG("Offset %zu is at / beyond size of morphed image.\n",offset);
- *p_read=0;
- return FALSE;
- }
- if(offset+size>image_size) {
- // Attempt to read data past EOF of morphed image file
- to_read=image_size-offset;
- LOG_DEBUG("Attempt to read data past EOF of morphed image. Corrected size "
- "from %zu to %" PRIu64 ".\n",
- size,
- to_read);
- } else to_read=size;
-
- // Calculate block to start reading data from
- cur_block=offset/CACHE_BLOCK_SIZE;
- block_off=offset%CACHE_BLOCK_SIZE;
-
- // Read image data
- while(to_read!=0) {
- // Calculate how many bytes we have to read from this block
- if(block_off+to_read>CACHE_BLOCK_SIZE) {
- cur_to_read=CACHE_BLOCK_SIZE-block_off;
- } else cur_to_read=to_read;
-
- // Check if block is cached
- if(glob_xmount.output.writable==TRUE &&
- glob_xmount.cache.p_block_cache_index[cur_block]!=CACHE_BLOCK_FREE)
- {
- // Write support enabled and need to read altered data from cachefile
- LOG_DEBUG("Reading %zu bytes at offset %" PRIu64
- " from block cache file\n",
- cur_to_read,
- glob_xmount.cache.p_block_cache_index[cur_block]+block_off)
-
- gidafs_ret=GidaFsLib_ReadFile(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache,
- glob_xmount.cache.
- p_block_cache_index[cur_block]+block_off,
- cur_to_read,
- p_buf,
- &read);
- if(gidafs_ret!=eGidaFsError_None || read!=cur_to_read) {
- LOG_ERROR("Unable to read cached data from block %" PRIu64
- ": Error code %u!\n",
- cur_block,
- gidafs_ret);
- return -EIO;
- }
- } else {
- // No write support or data not cached
- ret=glob_xmount.morphing.p_functions->Read(glob_xmount.morphing.p_handle,
- p_buf,
- (cur_block*CACHE_BLOCK_SIZE)+
- block_off,
- cur_to_read,
- &read);
- if(ret!=0 || read!=cur_to_read) {
- LOG_ERROR("Couldn't read %zu bytes at offset %zu from morphed image: "
- "%s!\n",
- cur_to_read,
- offset,
- glob_xmount.morphing.p_functions->GetErrorMessage(ret));
- return -EIO;
- }
- LOG_DEBUG("Read %" PRIu64 " bytes at offset %" PRIu64
- " from morphed image file\n",
- cur_to_read,
- (cur_block*CACHE_BLOCK_SIZE)+block_off);
- }
-
- cur_block++;
- block_off=0;
- p_buf+=cur_to_read;
- to_read-=cur_to_read;
- }
-
- *p_read=to_read;
- return TRUE;
-}
-
-//! Write data to morphed image
-/*!
- * \param p_buf Buffer with data to write
- * \param offset Offset to start writing at
- * \param count Amount of bytes to write
- * \param p_written Amount of successfully written bytes
- * \return TRUE on success, negated error code on error
- */
-static int WriteMorphedImageData(const char *p_buf,
- off_t offset,
- size_t count,
- size_t *p_written)
-{
- uint64_t block_off=0;
- uint64_t cur_block=0;
- uint64_t cur_to_read=0;
- uint64_t cur_to_write=0;
- uint64_t image_size=0;
- size_t written=0;
- size_t to_write=0;
- int ret;
- teGidaFsError gidafs_ret=eGidaFsError_None;
- char *p_buf2=NULL;
-
- // Make sure we aren't writing past EOF of image file
- if(GetMorphedImageSize(&image_size)!=TRUE) {
- LOG_ERROR("Couldn't get size of morphed image!\n");
- return -EIO;
- }
- if(offset>=image_size) {
- // Offset is beyond image size
- LOG_DEBUG("Offset %zu is at / beyond size of morphed image.\n",offset);
- *p_written=0;
- return 0;
- }
- if(offset+count>image_size) {
- // Attempt to write data past EOF of morphed image file
- to_write=image_size-offset;
- LOG_DEBUG("Attempt to write data past EOF of morphed image. Corrected size "
- "from %zu to %" PRIu64 ".\n",
- count,
- to_write);
- } else to_write=count;
-
- // Calculate block to start writing data to
- cur_block=offset/CACHE_BLOCK_SIZE;
- block_off=offset%CACHE_BLOCK_SIZE;
-
- while(to_write!=0) {
- // Calculate how many bytes we have to write to this block
- if(block_off+to_write>CACHE_BLOCK_SIZE) {
- cur_to_write=CACHE_BLOCK_SIZE-block_off;
- } else cur_to_write=to_write;
-
- // Check if block is cached
- if(glob_xmount.cache.p_block_cache_index[cur_block]!=CACHE_BLOCK_FREE) {
- // Block is cached
- gidafs_ret=GidaFsLib_WriteFile(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache,
- glob_xmount.cache.
- p_block_cache_index[cur_block]+block_off,
- cur_to_write,
- p_buf,
- &written);
- if(gidafs_ret!=eGidaFsError_None || written!=cur_to_write) {
- LOG_ERROR("Unable to write data to cached block %" PRIu64
- ": Error code %u!\n",
- cur_block,
- gidafs_ret);
- return -EIO;
- }
-
- LOG_DEBUG("Wrote %" PRIu64 " bytes at offset %" PRIu64
- " to block cache file\n",
- cur_to_write,
- glob_xmount.cache.p_block_cache_index[cur_block]+block_off);
- } else {
- // Uncached block. Need to cache entire new block
- // Prepare new write buffer
- XMOUNT_MALLOC(p_buf2,char*,CACHE_BLOCK_SIZE);
- memset(p_buf2,0x00,CACHE_BLOCK_SIZE);
-
- // Read full block from morphed image
- cur_to_read=CACHE_BLOCK_SIZE;
- if((cur_block*CACHE_BLOCK_SIZE)+CACHE_BLOCK_SIZE>image_size) {
- cur_to_read=CACHE_BLOCK_SIZE-(((cur_block*CACHE_BLOCK_SIZE)+
- CACHE_BLOCK_SIZE)-image_size);
- }
- ret=glob_xmount.morphing.p_functions->Read(glob_xmount.morphing.p_handle,
- p_buf2,
- cur_block*CACHE_BLOCK_SIZE,
- cur_to_read,
- &read);
- if(ret!=0 || read!=cur_to_read) {
- LOG_ERROR("Couldn't read %" PRIu64 " bytes at offset %zu "
- "from morphed image: %s!\n",
- cur_to_read,
- offset,
- glob_xmount.morphing.p_functions->GetErrorMessage(ret));
- XMOUNT_FREE(p_buf2);
- return -EIO;
- }
-
- // Set changed data
- memcpy(p_buf2+block_off,p_buf,cur_to_write);
-
- // Write new block to block cache
- // Get current block cache size
- gidafs_ret=GidaFsLib_GetFileSize(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache,
- &(glob_xmount.cache.
- p_block_cache_index[cur_block]));
- if(gidafs_ret!=eGidaFsError_None) {
- LOG_ERROR("Unable to get current block cache size: Error code %u!\n",
- gidafs_ret);
- XMOUNT_FREE(p_buf2);
- return -EIO;
- }
- // Append new block
- gidafs_ret=GidaFsLib_WriteFile(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache,
- glob_xmount.cache.
- p_block_cache_index[cur_block],
- CACHE_BLOCK_SIZE,
- p_buf2,
- &written);
- if(gidafs_ret!=eGidaFsError_None || written!=cur_to_write) {
- LOG_ERROR("Unable to write data to cached block %" PRIu64
- ": Error code %u!\n",
- cur_block,
- gidafs_ret);
- XMOUNT_FREE(p_buf2);
- return -EIO;
- }
- XMOUNT_FREE(p_buf2);
- // Update on-disk block cache index
- ret=UpdateBlockCacheIndex(cur_block,
- glob_xmount.cache.p_block_cache_index[
- cur_block]);
- if(ret!=TRUE) {
- LOG_ERROR("Unable to update block cache index %" PRIu64
- ": Error code %u!\n",
- cur_block,
- gidafs_ret);
- return -EIO;
- }
-
- LOG_DEBUG("Updated cache file block index: Number=%" PRIu64
- ", Data offset=%" PRIu64 "\n",
- cur_block,
- glob_xmount.cache.p_block_cache_index[cur_block]);
- }
-
- block_off=0;
- cur_block++;
- p_buf+=cur_to_write;
- to_write-=cur_to_write;
- }
-
- *p_written=to_write;
- return TRUE;
-}
-
-//! Read data from output image
-/*!
- * \param p_buf Pointer to buffer to write read data to
- * \param offset Offset at which data should be read
- * \param size Size of data which should be read
- * \return Number of read bytes on success or negated error code on error
- */
-int ReadOutputImageData(char *p_buf, off_t offset, size_t size) {
- uint64_t output_image_size;
- size_t read=0;
- int ret;
-
- // Get output image size
- if(GetOutputImageSize(&output_image_size)!=TRUE) {
- LOG_ERROR("Couldn't get size of output image!\n")
- return -EIO;
- }
-
- // Make sure request is within output image
- if(offset>=output_image_size) {
- LOG_DEBUG("Offset %zu is at / beyond size of output image.\n",offset);
- return 0;
- }
- if(offset+size>output_image_size) {
- LOG_DEBUG("Attempt to read data past EOF of output image. Correcting size "
- "from %zu to %zu.\n",
- size,
- output_image_size-offset);
- size=output_image_size-offset;
- }
-
- // Read data
- ret=glob_xmount.output.p_functions->Read(glob_xmount.output.p_handle,
- p_buf,
- offset,
- size,
- &read);
- if(ret!=0) {
- LOG_ERROR("Unable to read %zu bytes at offset %zu from output image!\n",
- size,
- offset)
- return ret;
- } else if(read!=size) {
- LOG_WARNING("Unable to read all requested data from output image!\n")
- return read;
- }
-
- return size;
-}
-
-//! Write data to output image
-/*!
- * \param p_buf Buffer with data to write
- * \param offset Offset to write to
- * \param size Amount of bytes to write
- * \return Number of written bytes on success or "-1" on error
- */
-int WriteOutputImageData(const char *p_buf, off_t offset, size_t size) {
- uint64_t output_image_size;
- int ret;
- size_t written;
-
- // Get output image size
- if(!GetOutputImageSize(&output_image_size)) {
- LOG_ERROR("Couldn't get output image size!\n")
- return -1;
- }
-
- // Make sure write is within output image
- if(offset>=output_image_size) {
- LOG_ERROR("Attempt to write beyond EOF of output image file!\n")
- return -1;
- }
- if(offset+size>output_image_size) {
- LOG_DEBUG("Attempt to write past EOF of output image file. Correcting size "
- "from %zu to %zu.\n",
- size,
- output_image_size-offset);
- size=output_image_size-offset;
- }
-
- ret=glob_xmount.output.p_functions->Write(glob_xmount.output.p_handle,
- p_buf,
- offset,
- size,
- &written);
- if(ret!=0) {
- LOG_ERROR("Unable to write %zu bytes at offset %zu to output image!\n",
- offset,
- size)
- return ret;
- } else if(written!=size) {
- LOG_WARNING("Unable to write all requested data to output image!\n")
- }
-
- return size;
-}
-
//! Calculates an MD5 hash of the first HASH_AMOUNT bytes of the input image
/*!
* \param p_hash_low Pointer to the lower 64 bit of the hash
* \param p_hash_high Pointer to the higher 64 bit of the hash
* \return TRUE on success, FALSE on error
*/
static int CalculateInputImageHash(uint64_t *p_hash_low,
uint64_t *p_hash_high)
{
char hash[16];
md5_state_t md5_state;
char *p_buf;
int ret;
size_t read_data;
XMOUNT_MALLOC(p_buf,char*,HASH_AMOUNT*sizeof(char));
ret=ReadMorphedImageData(p_buf,0,HASH_AMOUNT,&read_data);
if(ret!=TRUE || read_data==0) {
LOG_ERROR("Couldn't read data from morphed image file!\n")
free(p_buf);
return FALSE;
}
// Calculate MD5 hash
md5_init(&md5_state);
md5_append(&md5_state,(const md5_byte_t*)p_buf,read_data);
md5_finish(&md5_state,(md5_byte_t*)hash);
// Convert MD5 hash into two 64bit integers
*p_hash_low=*((uint64_t*)hash);
*p_hash_high=*((uint64_t*)(hash+8));
free(p_buf);
return TRUE;
}
//! Create info file
/*!
* \return TRUE on success, FALSE on error
*/
static int InitInfoFile() {
int ret;
char *p_buf;
// Start with static input header
XMOUNT_MALLOC(glob_xmount.output.p_info_file,
char*,
strlen(IMAGE_INFO_INPUT_HEADER)+1);
strncpy(glob_xmount.output.p_info_file,
IMAGE_INFO_INPUT_HEADER,
strlen(IMAGE_INFO_INPUT_HEADER)+1);
// Get and add infos from input lib(s)
for(uint64_t i=0;i<glob_xmount.input.images_count;i++) {
ret=glob_xmount.input.pp_images[i]->p_functions->
GetInfofileContent(glob_xmount.input.pp_images[i]->p_handle,(const char**)&p_buf);
if(ret!=0) {
LOG_ERROR("Unable to get info file content for image '%s': %s!\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret));
return FALSE;
}
// Add infos to main buffer and free p_buf
XMOUNT_STRAPP(glob_xmount.output.p_info_file,"\n--> ");
XMOUNT_STRAPP(glob_xmount.output.p_info_file,
glob_xmount.input.pp_images[i]->pp_files[0]);
XMOUNT_STRAPP(glob_xmount.output.p_info_file," <--\n");
if(p_buf!=NULL) {
XMOUNT_STRAPP(glob_xmount.output.p_info_file,p_buf);
glob_xmount.input.pp_images[i]->p_functions->FreeBuffer(p_buf);
} else {
XMOUNT_STRAPP(glob_xmount.output.p_info_file,"None\n");
}
}
// Add static morphing header
XMOUNT_STRAPP(glob_xmount.output.p_info_file,IMAGE_INFO_MORPHING_HEADER);
// Get and add infos from morphing lib
ret=glob_xmount.morphing.p_functions->
GetInfofileContent(glob_xmount.morphing.p_handle,(const char**)&p_buf);
if(ret!=0) {
LOG_ERROR("Unable to get info file content from morphing lib: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret));
return FALSE;
}
if(p_buf!=NULL) {
XMOUNT_STRAPP(glob_xmount.output.p_info_file,p_buf);
glob_xmount.morphing.p_functions->FreeBuffer(p_buf);
} else {
XMOUNT_STRAPP(glob_xmount.output.p_info_file,"None\n");
}
return TRUE;
}
-//! Create / load cache file to enable virtual write support
-/*!
- * \return TRUE on success, FALSE on error
- */
-static int InitCacheFile() {
- uint64_t blockindex_size=0;
- uint64_t image_size=0;
- uint64_t read=0;
- uint8_t is_new_cache_file=0;
- teGidaFsError gidafs_ret=eGidaFsError_None;
-
- // Get input image size for later use
- if(!GetMorphedImageSize(&image_size)) {
- LOG_ERROR("Couldn't get morphed image size!\n")
- return FALSE;
- }
-
- if(!glob_xmount.cache.overwrite_cache) {
- // Try to open an existing cache file or create a new one
- gidafs_ret=GidaFsLib_OpenFs(&(glob_xmount.cache.h_cache_file),
- glob_xmount.cache.p_cache_file);
- if(gidafs_ret!=eGidaFsError_None &&
- gidafs_ret!=eGidaFsError_FailedOpeningFsFile)
- {
- // TODO: Check for old cache file type and inform user it isn't supported
- // anymore!
- LOG_ERROR("Couldn't open cache file '%s': Error code %u!\n",
- glob_xmount.cache.p_cache_file,
- gidafs_ret)
- return FALSE;
- } else if(gidafs_ret==eGidaFsError_FailedOpeningFsFile) {
- // Unable to open cache file. It might simply not exist.
- LOG_DEBUG("Cache file '%s' does not exist. Creating new one\n",
- glob_xmount.cache.p_cache_file)
- gidafs_ret=GidaFsLib_NewFs(&(glob_xmount.cache.h_cache_file),
- glob_xmount.cache.p_cache_file,
- 0);
- if(gidafs_ret!=eGidaFsError_None) {
- // There is really a problem opening/creating the file
- LOG_ERROR("Couldn't open cache file '%s': Error code %u!\n",
- glob_xmount.cache.p_cache_file,
- gidafs_ret)
- return FALSE;
- }
- is_new_cache_file=1;
- }
- } else {
- // Overwrite existing cache file or create a new one
- gidafs_ret=GidaFsLib_NewFs(&(glob_xmount.cache.h_cache_file),
- glob_xmount.cache.p_cache_file,
- 0);
- if(gidafs_ret!=eGidaFsError_None) {
- // There is really a problem opening/creating the file
- LOG_ERROR("Couldn't open cache file '%s': Error code %u!\n",
- glob_xmount.cache.p_cache_file,
- gidafs_ret)
- return FALSE;
- }
- is_new_cache_file=1;
- }
-
-#define INITCACHEFILE__CLOSE_CACHE do { \
- gidafs_ret=GidaFsLib_CloseFs(&(glob_xmount.cache.h_cache_file)); \
- if(gidafs_ret!=eGidaFsError_None) { \
- LOG_ERROR("Unable to close cache file: Error code %u: Ignoring!\n", \
- gidafs_ret) \
- } \
-} while(0)
-
-#define INITCACHEFILE__CLOSE_BLOCK_CACHE do { \
- gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file, \
- &(glob_xmount.cache.h_block_cache)); \
- if(gidafs_ret!=eGidaFsError_None) { \
- LOG_ERROR("Unable to close block cache file: Error code %u: Ignoring!\n", \
- gidafs_ret) \
- } \
-} while(0)
-
-#define INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX do { \
- gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file, \
- &(glob_xmount.cache.h_block_cache_index)); \
- if(gidafs_ret!=eGidaFsError_None) { \
- LOG_ERROR("Unable to close block cache index file: Error code %u: " \
- "Ignoring!\n", \
- gidafs_ret) \
- } \
-} while(0)
-
- // TODO: Check if cache file uses same block size as we do
-
- if(is_new_cache_file==1) {
- // New cache file, create needed xmount subdirectory
- gidafs_ret=GidaFsLib_CreateDir(glob_xmount.cache.h_cache_file,
- XMOUNT_CACHE_FOLDER,
- eGidaFsNodeFlag_RWXu);
- if(gidafs_ret!=eGidaFsError_None) {
- LOG_ERROR("Unable to create cache file directory '%s': Error code %u!\n",
- XMOUNT_CACHE_FOLDER,
- gidafs_ret)
- INITCACHEFILE__CLOSE_CACHE;
- return FALSE;
- }
- }
-
- // Open / Create block cache file
- gidafs_ret=GidaFsLib_OpenFile(glob_xmount.cache.h_cache_file,
- XMOUNT_CACHE_BLOCK_FILE,
- &(glob_xmount.cache.h_block_cache),
- eGidaFsOpenFileFlag_ReadWrite |
- (is_new_cache_file==1 ?
- eGidaFsOpenFileFlag_CreateAlways : 0),
- eGidaFsNodeFlag_Rall |
- eGidaFsNodeFlag_Wusr);
- if(gidafs_ret!=eGidaFsError_None) {
- LOG_ERROR("Unable to open / create block cache file '%s': Error code %u!\n",
- XMOUNT_CACHE_BLOCK_FILE,
- gidafs_ret)
- INITCACHEFILE__CLOSE_CACHE;
- return FALSE;
- }
-
- // Open / Create block cache index file
- gidafs_ret=GidaFsLib_OpenFile(glob_xmount.cache.h_cache_file,
- XMOUNT_CACHE_BLOCK_INDEX_FILE,
- &(glob_xmount.cache.h_block_cache_index),
- eGidaFsOpenFileFlag_ReadWrite |
- (is_new_cache_file==1 ?
- eGidaFsOpenFileFlag_CreateAlways : 0),
- eGidaFsNodeFlag_Rall |
- eGidaFsNodeFlag_Wusr);
- if(gidafs_ret!=eGidaFsError_None) {
- LOG_ERROR("Unable to open / create block cache index file '%s': "
- "Error code %u!\n",
- XMOUNT_CACHE_BLOCK_FILE,
- gidafs_ret)
- INITCACHEFILE__CLOSE_BLOCK_CACHE;
- INITCACHEFILE__CLOSE_CACHE;
- return FALSE;
- }
-
- // Calculate how many cache blocks are needed and how big the cache block
- // index must be
- glob_xmount.cache.block_cache_index_len=image_size/CACHE_BLOCK_SIZE;
- if((image_size%CACHE_BLOCK_SIZE)!=0) {
- glob_xmount.cache.block_cache_index_len++;
- }
-
- LOG_DEBUG("Cache blocks: %u (0x%04X) entries using %zd (0x%08zX) bytes\n",
- glob_xmount.cache.block_cache_index_len,
- glob_xmount.cache.block_cache_index_len,
- glob_xmount.cache.block_cache_index_len*
- sizeof(t_CacheFileBlockIndex),
- glob_xmount.cache.block_cache_index_len*
- sizeof(t_CacheFileBlockIndex))
-
- // Prepare in-memory buffer for block cache index
- XMOUNT_MALLOC(glob_xmount.cache.p_block_cache_index,
- t_CacheFileBlockIndex*,
- glob_xmount.cache.block_cache_index_len*
- sizeof(t_CacheFileBlockIndex))
-
- if(is_new_cache_file==1) {
- // Generate initial block cache index
- for(uint64_t i=0;i<glob_xmount.cache.block_cache_index_len;i++) {
- glob_xmount.cache.p_block_cache_index[i]=CACHE_BLOCK_FREE;
- }
- // Write initial block cache index to cache file
- if(UpdateBlockCacheIndex(XMOUNT_BLOCK_CACHE_INVALID_INDEX,0)!=TRUE) {
- LOG_ERROR("Unable to generate initial block cache index file '%s': "
- "Error code %u!\n",
- XMOUNT_CACHE_BLOCK_FILE,
- gidafs_ret)
- INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
- INITCACHEFILE__CLOSE_BLOCK_CACHE;
- INITCACHEFILE__CLOSE_CACHE;
- XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
- return FALSE;
- }
- } else {
- // Existing cache file, make sure block cache index has correct size
- gidafs_ret=GidaFsLib_GetFileSize(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache_index,
- &blockindex_size);
- if(gidafs_ret!=eGidaFsError_None) {
- LOG_ERROR("Unable to get block cache index file size: Error code %u!\n",
- gidafs_ret)
- INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
- INITCACHEFILE__CLOSE_BLOCK_CACHE;
- INITCACHEFILE__CLOSE_CACHE;
- XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
- return FALSE;
- }
- if(blockindex_size%sizeof(t_CacheFileBlockIndex)!=0 ||
- (blockindex_size/sizeof(t_CacheFileBlockIndex))!=
- glob_xmount.cache.block_cache_index_len)
- {
- // TODO: Be more helpfull in error message
- LOG_ERROR("Block cache index size is incorrect for given input image!\n")
- INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
- INITCACHEFILE__CLOSE_BLOCK_CACHE;
- INITCACHEFILE__CLOSE_CACHE;
- XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
- return FALSE;
- }
- // Read block cache index into memory
- gidafs_ret=GidaFsLib_ReadFile(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache_index,
- 0,
- blockindex_size,
- glob_xmount.cache.p_block_cache_index,
- &read);
- if(gidafs_ret!=eGidaFsError_None || read!=blockindex_size) {
- LOG_ERROR("Unable to read block cache index: Error code %u!\n",
- gidafs_ret);
- INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
- INITCACHEFILE__CLOSE_BLOCK_CACHE;
- INITCACHEFILE__CLOSE_CACHE;
- XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
- return FALSE;
- }
- }
-
-#undef INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX
-#undef INITCACHEFILE__CLOSE_BLOCK_CACHE
-#undef INITCACHEFILE__CLOSE_CACHE
-
- return TRUE;
-}
-
-//! Update block cache index
-/*!
- * \return TRUE on success, FALSE on error
- */
-static int UpdateBlockCacheIndex(uint64_t index, t_CacheFileBlockIndex value) {
- uint64_t update_size=0;
- uint64_t written=0;
- teGidaFsError gidafs_ret=eGidaFsError_None;
- t_CacheFileBlockIndex *p_buf;
-
- if(index!=XMOUNT_BLOCK_CACHE_INVALID_INDEX) {
- // Update specific index element in cache file
- p_buf=glob_xmount.cache.p_block_cache_index+index;
- update_size=sizeof(t_CacheFileBlockIndex);
- } else {
- // Dump whole block cache index to cache file
- p_buf=glob_xmount.cache.p_block_cache_index;
- update_size=glob_xmount.cache.block_cache_index_len*
- sizeof(t_CacheFileBlockIndex);
- }
-
- // Update cache file
- gidafs_ret=GidaFsLib_WriteFile(glob_xmount.cache.h_cache_file,
- glob_xmount.cache.h_block_cache_index,
- 0,
- update_size,
- p_buf,
- &written);
- if(gidafs_ret!=eGidaFsError_None || written!=update_size) {
- LOG_ERROR("Unable to update block cache index file '%s': "
- "Error code %u!\n",
- XMOUNT_CACHE_BLOCK_INDEX_FILE,
- gidafs_ret)
- return FALSE;
- }
-
- return TRUE;
-}
-
//! Load input / morphing libs
/*!
* \return TRUE on success, FALSE on error
*/
static int LoadLibs() {
DIR *p_dir=NULL;
struct dirent *p_dirent=NULL;
int base_library_path_len=0;
char *p_library_path=NULL;
void *p_libxmount=NULL;
t_LibXmount_Input_GetApiVersion pfun_input_GetApiVersion;
t_LibXmount_Input_GetSupportedFormats pfun_input_GetSupportedFormats;
t_LibXmount_Input_GetFunctions pfun_input_GetFunctions;
t_LibXmount_Morphing_GetApiVersion pfun_morphing_GetApiVersion;
t_LibXmount_Morphing_GetSupportedTypes pfun_morphing_GetSupportedTypes;
t_LibXmount_Morphing_GetFunctions pfun_morphing_GetFunctions;
t_LibXmount_Output_GetApiVersion pfun_output_GetApiVersion;
t_LibXmount_Output_GetSupportedFormats pfun_output_GetSupportedFormats;
t_LibXmount_Output_GetFunctions pfun_output_GetFunctions;
const char *p_supported_formats=NULL;
const char *p_buf;
uint32_t supported_formats_len=0;
pts_InputLib p_input_lib=NULL;
pts_MorphingLib p_morphing_lib=NULL;
pts_OutputLib p_output_lib=NULL;
LOG_DEBUG("Searching for xmount libraries in '%s'.\n",
XMOUNT_LIBRARY_PATH);
// Open lib dir
p_dir=opendir(XMOUNT_LIBRARY_PATH);
if(p_dir==NULL) {
LOG_ERROR("Unable to access xmount library directory '%s'!\n",
XMOUNT_LIBRARY_PATH);
return FALSE;
}
// Construct base library path
base_library_path_len=strlen(XMOUNT_LIBRARY_PATH);
XMOUNT_STRSET(p_library_path,XMOUNT_LIBRARY_PATH);
if(XMOUNT_LIBRARY_PATH[base_library_path_len]!='/') {
base_library_path_len++;
XMOUNT_STRAPP(p_library_path,"/");
}
#define LIBXMOUNT_LOAD(path) { \
p_libxmount=dlopen(path,RTLD_NOW); \
if(p_libxmount==NULL) { \
LOG_ERROR("Unable to load input library '%s': %s!\n", \
path, \
dlerror()); \
continue; \
} \
}
#define LIBXMOUNT_LOAD_SYMBOL(name,pfun) { \
if((pfun=dlsym(p_libxmount,name))==NULL) { \
LOG_ERROR("Unable to load symbol '%s' from library '%s'!\n", \
name, \
p_library_path); \
dlclose(p_libxmount); \
p_libxmount=NULL; \
continue; \
} \
}
// Loop over lib dir
while((p_dirent=readdir(p_dir))!=NULL) {
LOG_DEBUG("Trying to load '%s'\n",p_dirent->d_name);
// Construct full path to found object
p_library_path=realloc(p_library_path,
base_library_path_len+strlen(p_dirent->d_name)+1);
if(p_library_path==NULL) {
LOG_ERROR("Couldn't allocate memory!\n");
exit(1);
}
strcpy(p_library_path+base_library_path_len,p_dirent->d_name);
if(strncmp(p_dirent->d_name,"libxmount_input_",16)==0) {
// Found possible input lib. Try to load it
LIBXMOUNT_LOAD(p_library_path);
// Load library symbols
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Input_GetApiVersion",
pfun_input_GetApiVersion);
// Check library's API version
if(pfun_input_GetApiVersion()!=LIBXMOUNT_INPUT_API_VERSION) {
LOG_DEBUG("Failed! Wrong API version.\n");
LOG_ERROR("Unable to load input library '%s'. Wrong API version\n",
p_library_path);
dlclose(p_libxmount);
continue;
}
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Input_GetSupportedFormats",
pfun_input_GetSupportedFormats);
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Input_GetFunctions",
pfun_input_GetFunctions);
// Construct new entry for our library list
XMOUNT_MALLOC(p_input_lib,pts_InputLib,sizeof(ts_InputLib));
// Initialize lib_functions structure to NULL
memset(&(p_input_lib->lib_functions),
0,
sizeof(ts_LibXmountInputFunctions));
// Set name and handle
XMOUNT_STRSET(p_input_lib->p_name,p_dirent->d_name);
p_input_lib->p_lib=p_libxmount;
// Get and set supported formats
p_supported_formats=pfun_input_GetSupportedFormats();
supported_formats_len=0;
p_buf=p_supported_formats;
while(*p_buf!='\0') {
supported_formats_len+=(strlen(p_buf)+1);
p_buf+=(strlen(p_buf)+1);
}
supported_formats_len++;
XMOUNT_MALLOC(p_input_lib->p_supported_input_types,
char*,
supported_formats_len);
memcpy(p_input_lib->p_supported_input_types,
p_supported_formats,
supported_formats_len);
// Get, set and check lib_functions
pfun_input_GetFunctions(&(p_input_lib->lib_functions));
if(p_input_lib->lib_functions.CreateHandle==NULL ||
p_input_lib->lib_functions.DestroyHandle==NULL ||
p_input_lib->lib_functions.Open==NULL ||
p_input_lib->lib_functions.Close==NULL ||
p_input_lib->lib_functions.Size==NULL ||
p_input_lib->lib_functions.Read==NULL ||
p_input_lib->lib_functions.OptionsHelp==NULL ||
p_input_lib->lib_functions.OptionsParse==NULL ||
p_input_lib->lib_functions.GetInfofileContent==NULL ||
p_input_lib->lib_functions.GetErrorMessage==NULL ||
p_input_lib->lib_functions.FreeBuffer==NULL)
{
LOG_DEBUG("Missing implemention of one or more functions in lib %s!\n",
p_dirent->d_name);
free(p_input_lib->p_supported_input_types);
free(p_input_lib->p_name);
free(p_input_lib);
dlclose(p_libxmount);
continue;
}
// Add entry to the input library list
XMOUNT_REALLOC(glob_xmount.input.pp_libs,
pts_InputLib*,
sizeof(pts_InputLib)*(glob_xmount.input.libs_count+1));
glob_xmount.input.pp_libs[glob_xmount.input.libs_count++]=p_input_lib;
LOG_DEBUG("Input library '%s' loaded successfully\n",p_dirent->d_name);
} if(strncmp(p_dirent->d_name,"libxmount_morphing_",19)==0) {
// Found possible morphing lib. Try to load it
LIBXMOUNT_LOAD(p_library_path);
// Load library symbols
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Morphing_GetApiVersion",
pfun_morphing_GetApiVersion);
// Check library's API version
if(pfun_morphing_GetApiVersion()!=LIBXMOUNT_MORPHING_API_VERSION) {
LOG_DEBUG("Failed! Wrong API version.\n");
LOG_ERROR("Unable to load morphing library '%s'. Wrong API version\n",
p_library_path);
dlclose(p_libxmount);
continue;
}
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Morphing_GetSupportedTypes",
pfun_morphing_GetSupportedTypes);
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Morphing_GetFunctions",
pfun_morphing_GetFunctions);
// Construct new entry for our library list
XMOUNT_MALLOC(p_morphing_lib,pts_MorphingLib,sizeof(ts_MorphingLib));
// Initialize lib_functions structure to NULL
memset(&(p_morphing_lib->lib_functions),
0,
sizeof(ts_LibXmountMorphingFunctions));
// Set name and handle
XMOUNT_STRSET(p_morphing_lib->p_name,p_dirent->d_name);
p_morphing_lib->p_lib=p_libxmount;
// Get and set supported types
p_supported_formats=pfun_morphing_GetSupportedTypes();
supported_formats_len=0;
p_buf=p_supported_formats;
while(*p_buf!='\0') {
supported_formats_len+=(strlen(p_buf)+1);
p_buf+=(strlen(p_buf)+1);
}
supported_formats_len++;
XMOUNT_MALLOC(p_morphing_lib->p_supported_morphing_types,
char*,
supported_formats_len);
memcpy(p_morphing_lib->p_supported_morphing_types,
p_supported_formats,
supported_formats_len);
// Get, set and check lib_functions
pfun_morphing_GetFunctions(&(p_morphing_lib->lib_functions));
if(p_morphing_lib->lib_functions.CreateHandle==NULL ||
p_morphing_lib->lib_functions.DestroyHandle==NULL ||
p_morphing_lib->lib_functions.Morph==NULL ||
p_morphing_lib->lib_functions.Size==NULL ||
p_morphing_lib->lib_functions.Read==NULL ||
p_morphing_lib->lib_functions.OptionsHelp==NULL ||
p_morphing_lib->lib_functions.OptionsParse==NULL ||
p_morphing_lib->lib_functions.GetInfofileContent==NULL ||
p_morphing_lib->lib_functions.GetErrorMessage==NULL ||
p_morphing_lib->lib_functions.FreeBuffer==NULL)
{
LOG_DEBUG("Missing implemention of one or more functions in lib %s!\n",
p_dirent->d_name);
free(p_morphing_lib->p_supported_morphing_types);
free(p_morphing_lib->p_name);
free(p_morphing_lib);
dlclose(p_libxmount);
continue;
}
// Add entry to the input library list
XMOUNT_REALLOC(glob_xmount.morphing.pp_libs,
pts_MorphingLib*,
sizeof(pts_MorphingLib)*
(glob_xmount.morphing.libs_count+1));
glob_xmount.morphing.pp_libs[glob_xmount.morphing.libs_count++]=
p_morphing_lib;
LOG_DEBUG("Morphing library '%s' loaded successfully\n",p_dirent->d_name);
} if(strncmp(p_dirent->d_name,"libxmount_output_",17)==0) {
// Found possible output lib. Try to load it
LIBXMOUNT_LOAD(p_library_path);
// Load library symbols
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Output_GetApiVersion",
pfun_output_GetApiVersion);
// Check library's API version
if(pfun_output_GetApiVersion()!=LIBXMOUNT_OUTPUT_API_VERSION) {
LOG_DEBUG("Failed! Wrong API version.\n");
LOG_ERROR("Unable to load output library '%s'. Wrong API version\n",
p_library_path);
dlclose(p_libxmount);
continue;
}
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Output_GetSupportedFormats",
pfun_output_GetSupportedFormats);
LIBXMOUNT_LOAD_SYMBOL("LibXmount_Output_GetFunctions",
pfun_output_GetFunctions);
// Construct new entry for our library list
XMOUNT_MALLOC(p_output_lib,pts_OutputLib,sizeof(ts_OutputLib));
// Initialize lib_functions structure to NULL
memset(&(p_output_lib->lib_functions),
0,
sizeof(ts_LibXmountOutput_Functions));
// Set name and handle
XMOUNT_STRSET(p_output_lib->p_name,p_dirent->d_name);
p_output_lib->p_lib=p_libxmount;
// Get and set supported types
p_supported_formats=pfun_output_GetSupportedFormats();
supported_formats_len=0;
p_buf=p_supported_formats;
while(*p_buf!='\0') {
supported_formats_len+=(strlen(p_buf)+1);
p_buf+=(strlen(p_buf)+1);
}
supported_formats_len++;
XMOUNT_MALLOC(p_output_lib->p_supported_output_formats,
char*,
supported_formats_len);
memcpy(p_output_lib->p_supported_output_formats,
p_supported_formats,
supported_formats_len);
// Get, set and check lib_functions
pfun_output_GetFunctions(&(p_output_lib->lib_functions));
if(p_output_lib->lib_functions.CreateHandle==NULL ||
p_output_lib->lib_functions.DestroyHandle==NULL ||
p_output_lib->lib_functions.Transform==NULL ||
p_output_lib->lib_functions.Size==NULL ||
p_output_lib->lib_functions.Read==NULL ||
p_output_lib->lib_functions.Write==NULL ||
p_output_lib->lib_functions.OptionsHelp==NULL ||
p_output_lib->lib_functions.OptionsParse==NULL ||
p_output_lib->lib_functions.GetInfofileContent==NULL ||
p_output_lib->lib_functions.GetErrorMessage==NULL ||
p_output_lib->lib_functions.FreeBuffer==NULL)
{
LOG_DEBUG("Missing implemention of one or more functions in lib %s!\n",
p_dirent->d_name);
free(p_output_lib->p_supported_output_formats);
free(p_output_lib->p_name);
free(p_output_lib);
dlclose(p_libxmount);
continue;
}
// Add entry to the input library list
XMOUNT_REALLOC(glob_xmount.output.pp_libs,
pts_OutputLib*,
sizeof(pts_OutputLib)*
(glob_xmount.output.libs_count+1));
glob_xmount.output.pp_libs[glob_xmount.output.libs_count++]=
p_output_lib;
LOG_DEBUG("Output library '%s' loaded successfully\n",p_dirent->d_name);
} else {
LOG_DEBUG("Ignoring '%s'.\n",p_dirent->d_name);
continue;
}
}
#undef LIBXMOUNT_LOAD_SYMBOL
#undef LIBXMOUNT_LOAD
LOG_DEBUG("A total of %u input libs, %u morphing libs and %u output libs "
"were loaded.\n",
glob_xmount.input.libs_count,
glob_xmount.morphing.libs_count,
glob_xmount.output.libs_count);
free(p_library_path);
closedir(p_dir);
return ((glob_xmount.input.libs_count>0 &&
glob_xmount.morphing.libs_count>0 &&
glob_xmount.output.libs_count>0) ? TRUE : FALSE);
}
//! Search an appropriate input lib for specified input type
/*!
* \param p_input_image Input image to search input lib for
* \return TRUE on success, FALSE on error
*/
static int FindInputLib(pts_InputImage p_input_image) {
char *p_buf;
LOG_DEBUG("Trying to find suitable library for input type '%s'.\n",
p_input_image->p_type);
// Loop over all loaded libs
for(uint32_t i=0;i<glob_xmount.input.libs_count;i++) {
LOG_DEBUG("Checking input library %s\n",
glob_xmount.input.pp_libs[i]->p_name);
p_buf=glob_xmount.input.pp_libs[i]->p_supported_input_types;
while(*p_buf!='\0') {
if(strcmp(p_buf,p_input_image->p_type)==0) {
// Library supports input type, set lib functions
LOG_DEBUG("Input library '%s' pretends to handle that input type.\n",
glob_xmount.input.pp_libs[i]->p_name);
p_input_image->p_functions=
&(glob_xmount.input.pp_libs[i]->lib_functions);
return TRUE;
}
p_buf+=(strlen(p_buf)+1);
}
}
LOG_DEBUG("Couldn't find any suitable library.\n");
// No library supporting input type found
return FALSE;
}
//! Search an appropriate morphing lib for the specified morph type
/*!
* \return TRUE on success, FALSE on error
*/
static int FindMorphingLib() {
char *p_buf;
LOG_DEBUG("Trying to find suitable library for morph type '%s'.\n",
glob_xmount.morphing.p_morph_type);
// Loop over all loaded libs
for(uint32_t i=0;i<glob_xmount.morphing.libs_count;i++) {
LOG_DEBUG("Checking morphing library %s\n",
glob_xmount.morphing.pp_libs[i]->p_name);
p_buf=glob_xmount.morphing.pp_libs[i]->p_supported_morphing_types;
while(*p_buf!='\0') {
if(strcmp(p_buf,glob_xmount.morphing.p_morph_type)==0) {
// Library supports morph type, set lib functions
LOG_DEBUG("Morphing library '%s' pretends to handle that morph type.\n",
glob_xmount.morphing.pp_libs[i]->p_name);
glob_xmount.morphing.p_functions=
&(glob_xmount.morphing.pp_libs[i]->lib_functions);
return TRUE;
}
p_buf+=(strlen(p_buf)+1);
}
}
LOG_DEBUG("Couldn't find any suitable library.\n");
// No library supporting morph type found
return FALSE;
}
//! Search an appropriate output lib for the specified output format
/*!
* \return TRUE on success, FALSE on error
*/
static int FindOutputLib() {
char *p_buf;
LOG_DEBUG("Trying to find suitable library for output format '%s'.\n",
glob_xmount.output.p_output_format);
// Loop over all loaded output libs
for(uint32_t i=0;i<glob_xmount.output.libs_count;i++) {
LOG_DEBUG("Checking output library %s\n",
glob_xmount.output.pp_libs[i]->p_name);
p_buf=glob_xmount.output.pp_libs[i]->p_supported_output_formats;
while(*p_buf!='\0') {
if(strcmp(p_buf,glob_xmount.output.p_output_format)==0) {
// Library supports output type, set lib functions
LOG_DEBUG("Output library '%s' pretends to handle that output format.\n",
glob_xmount.output.pp_libs[i]->p_name);
glob_xmount.output.p_functions=
&(glob_xmount.output.pp_libs[i]->lib_functions);
return TRUE;
}
p_buf+=(strlen(p_buf)+1);
}
}
LOG_DEBUG("Couldn't find any suitable library.\n");
// No library supporting output format found
return FALSE;
}
static void InitResources() {
// Input
glob_xmount.input.libs_count=0;
glob_xmount.input.pp_libs=NULL;
glob_xmount.input.lib_params_count=0;
glob_xmount.input.pp_lib_params=NULL;
glob_xmount.input.images_count=0;
glob_xmount.input.pp_images=NULL;
glob_xmount.input.image_offset=0;
glob_xmount.input.image_size_limit=0;
glob_xmount.input.image_hash_lo=0;
glob_xmount.input.image_hash_hi=0;
// Morphing
glob_xmount.morphing.libs_count=0;
glob_xmount.morphing.pp_libs=NULL;
glob_xmount.morphing.p_morph_type=NULL;
glob_xmount.morphing.lib_params_count=0;
glob_xmount.morphing.pp_lib_params=NULL;
glob_xmount.morphing.p_handle=NULL;
glob_xmount.morphing.p_functions=NULL;
glob_xmount.morphing.input_image_functions.ImageCount=
&LibXmount_Morphing_ImageCount;
glob_xmount.morphing.input_image_functions.Size=&LibXmount_Morphing_Size;
glob_xmount.morphing.input_image_functions.Read=&LibXmount_Morphing_Read;
// Cache
glob_xmount.cache.overwrite_cache=FALSE;
glob_xmount.cache.p_cache_file=NULL;
glob_xmount.cache.h_cache_file=NULL;
glob_xmount.cache.h_block_cache=NULL;
glob_xmount.cache.h_block_cache_index=NULL;
glob_xmount.cache.p_block_cache_index=NULL;
glob_xmount.cache.block_cache_index_len=0;
// Output
glob_xmount.output.libs_count=0;
glob_xmount.output.pp_libs=NULL;
glob_xmount.output.p_output_format=NULL;
glob_xmount.output.lib_params_count=0;
glob_xmount.output.pp_lib_params=NULL;
glob_xmount.output.p_handle=NULL;
glob_xmount.output.p_functions=NULL;
glob_xmount.output.input_functions.Size=&LibXmount_Output_Size;
glob_xmount.output.input_functions.Read=&LibXmount_Output_Read;
glob_xmount.output.input_functions.Write=&LibXmount_Output_Write;
glob_xmount.output.image_size=0;
glob_xmount.output.writable=FALSE;
glob_xmount.output.p_virtual_image_path=NULL;
glob_xmount.output.p_info_path=NULL;
glob_xmount.output.p_info_file=NULL;
// Misc data
glob_xmount.debug=FALSE;
glob_xmount.may_set_fuse_allow_other=FALSE;
glob_xmount.fuse_argc=0;
glob_xmount.pp_fuse_argv=NULL;
glob_xmount.p_mountpoint=NULL;
}
/*
* FreeResources
*/
static void FreeResources() {
int ret;
teGidaFsError gidafs_ret=eGidaFsError_None;
LOG_DEBUG("Freeing all resources\n");
// Misc
if(glob_xmount.pp_fuse_argv!=NULL) {
for(int i=0;i<glob_xmount.fuse_argc;i++) free(glob_xmount.pp_fuse_argv[i]);
free(glob_xmount.pp_fuse_argv);
}
if(glob_xmount.p_mountpoint!=NULL) free(glob_xmount.p_mountpoint);
// Output
if(glob_xmount.output.p_functions!=NULL) {
if(glob_xmount.output.p_handle!=NULL) {
// Destroy output handle
ret=glob_xmount.output.p_functions->
DestroyHandle(&(glob_xmount.output.p_handle));
if(ret!=0) {
LOG_ERROR("Unable to destroy output handle: %s!\n",
glob_xmount.output.p_functions->GetErrorMessage(ret));
}
}
}
if(glob_xmount.output.pp_lib_params!=NULL) {
for(uint32_t i=0;i<glob_xmount.output.lib_params_count;i++)
free(glob_xmount.output.pp_lib_params[i]);
free(glob_xmount.output.pp_lib_params);
}
if(glob_xmount.output.p_output_format!=NULL)
free(glob_xmount.output.p_output_format);
if(glob_xmount.output.pp_libs!=NULL) {
// Unload output libs
for(uint32_t i=0;i<glob_xmount.output.libs_count;i++) {
if(glob_xmount.output.pp_libs[i]==NULL) continue;
if(glob_xmount.output.pp_libs[i]->p_supported_output_formats!=NULL)
free(glob_xmount.output.pp_libs[i]->p_supported_output_formats);
if(glob_xmount.output.pp_libs[i]->p_lib!=NULL)
dlclose(glob_xmount.output.pp_libs[i]->p_lib);
if(glob_xmount.output.pp_libs[i]->p_name!=NULL)
free(glob_xmount.output.pp_libs[i]->p_name);
free(glob_xmount.output.pp_libs[i]);
}
free(glob_xmount.output.pp_libs);
}
if(glob_xmount.output.p_info_path!=NULL)
free(glob_xmount.output.p_info_path);
if(glob_xmount.output.p_info_file!=NULL)
free(glob_xmount.output.p_info_file);
if(glob_xmount.output.p_virtual_image_path!=NULL)
free(glob_xmount.output.p_virtual_image_path);
// Cache
if(glob_xmount.cache.h_cache_file!=NULL) {
if(glob_xmount.cache.p_block_cache_index!=NULL)
free(glob_xmount.cache.p_block_cache_index);
if(glob_xmount.cache.h_block_cache_index!=NULL) {
gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file,
&(glob_xmount.cache.h_block_cache_index));
if(gidafs_ret!=eGidaFsError_None) {
LOG_ERROR("Unable to close block cache index file: Error code %u: "
"Ignoring!\n",
gidafs_ret)
}
}
if(glob_xmount.cache.h_block_cache!=NULL) {
gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file,
&(glob_xmount.cache.h_block_cache));
if(gidafs_ret!=eGidaFsError_None) {
LOG_ERROR("Unable to close block cache file: Error code %u: "
"Ignoring!\n",
gidafs_ret)
}
}
gidafs_ret=GidaFsLib_CloseFs(&(glob_xmount.cache.h_cache_file));
if(gidafs_ret!=eGidaFsError_None) {
LOG_ERROR("Unable to close cache file: Error code %u: Ignoring!\n",
gidafs_ret)
}
}
if(glob_xmount.cache.p_cache_file!=NULL) free(glob_xmount.cache.p_cache_file);
// Morphing
if(glob_xmount.morphing.p_functions!=NULL) {
if(glob_xmount.morphing.p_handle!=NULL) {
// Destroy morphing handle
ret=glob_xmount.morphing.p_functions->
DestroyHandle(&(glob_xmount.morphing.p_handle));
if(ret!=0) {
LOG_ERROR("Unable to destroy morphing handle: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret));
}
}
}
if(glob_xmount.morphing.pp_lib_params!=NULL) {
for(uint32_t i=0;i<glob_xmount.morphing.lib_params_count;i++)
free(glob_xmount.morphing.pp_lib_params[i]);
free(glob_xmount.morphing.pp_lib_params);
}
if(glob_xmount.morphing.p_morph_type!=NULL)
free(glob_xmount.morphing.p_morph_type);
if(glob_xmount.morphing.pp_libs!=NULL) {
// Unload morphing libs
for(uint32_t i=0;i<glob_xmount.morphing.libs_count;i++) {
if(glob_xmount.morphing.pp_libs[i]==NULL) continue;
if(glob_xmount.morphing.pp_libs[i]->p_supported_morphing_types!=NULL)
free(glob_xmount.morphing.pp_libs[i]->p_supported_morphing_types);
if(glob_xmount.morphing.pp_libs[i]->p_lib!=NULL)
dlclose(glob_xmount.morphing.pp_libs[i]->p_lib);
if(glob_xmount.morphing.pp_libs[i]->p_name!=NULL)
free(glob_xmount.morphing.pp_libs[i]->p_name);
free(glob_xmount.morphing.pp_libs[i]);
}
free(glob_xmount.morphing.pp_libs);
}
// Input
if(glob_xmount.input.pp_images!=NULL) {
// Close all input images
for(uint64_t i=0;i<glob_xmount.input.images_count;i++) {
if(glob_xmount.input.pp_images[i]==NULL) continue;
if(glob_xmount.input.pp_images[i]->p_functions!=NULL) {
if(glob_xmount.input.pp_images[i]->p_handle!=NULL) {
ret=glob_xmount.input.pp_images[i]->p_functions->
Close(glob_xmount.input.pp_images[i]->p_handle);
if(ret!=0) {
LOG_ERROR("Unable to close input image: %s\n",
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret));
}
ret=glob_xmount.input.pp_images[i]->p_functions->
DestroyHandle(&(glob_xmount.input.pp_images[i]->p_handle));
if(ret!=0) {
LOG_ERROR("Unable to destroy input image handle: %s\n",
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret));
}
}
}
if(glob_xmount.input.pp_images[i]->pp_files!=NULL) {
for(uint64_t ii=0;ii<glob_xmount.input.pp_images[i]->files_count;ii++) {
if(glob_xmount.input.pp_images[i]->pp_files[ii]!=NULL)
free(glob_xmount.input.pp_images[i]->pp_files[ii]);
}
free(glob_xmount.input.pp_images[i]->pp_files);
}
if(glob_xmount.input.pp_images[i]->p_type!=NULL)
free(glob_xmount.input.pp_images[i]->p_type);
free(glob_xmount.input.pp_images[i]);
}
free(glob_xmount.input.pp_images);
}
if(glob_xmount.input.pp_lib_params!=NULL) {
for(uint32_t i=0;i<glob_xmount.input.lib_params_count;i++)
free(glob_xmount.input.pp_lib_params[i]);
free(glob_xmount.input.pp_lib_params);
}
if(glob_xmount.input.pp_libs!=NULL) {
// Unload all input libs
for(uint32_t i=0;i<glob_xmount.input.libs_count;i++) {
if(glob_xmount.input.pp_libs[i]->p_supported_input_types!=NULL)
free(glob_xmount.input.pp_libs[i]->p_supported_input_types);
if(glob_xmount.input.pp_libs[i]->p_lib!=NULL)
dlclose(glob_xmount.input.pp_libs[i]->p_lib);
if(glob_xmount.input.pp_libs[i]->p_name!=NULL)
free(glob_xmount.input.pp_libs[i]->p_name);
free(glob_xmount.input.pp_libs[i]);
}
free(glob_xmount.input.pp_libs);
}
// Before we return, initialize everything in case ReleaseResources would be
// called again.
InitResources();
}
//! Function to split given library options
static int SplitLibraryParameters(char *p_params,
uint32_t *p_ret_opts_count,
pts_LibXmountOptions **ppp_ret_opt)
{
pts_LibXmountOptions p_opts=NULL;
pts_LibXmountOptions *pp_opts=NULL;
uint32_t params_len;
uint32_t opts_count=0;
uint32_t sep_pos=0;
char *p_buf=p_params;
if(p_params==NULL) return FALSE;
// Get params length
params_len=strlen(p_params);
// Return if no params specified
if(params_len==0) {
*ppp_ret_opt=NULL;
p_ret_opts_count=0;
return TRUE;
}
// Split params
while(*p_buf!='\0') {
XMOUNT_MALLOC(p_opts,pts_LibXmountOptions,sizeof(ts_LibXmountOptions));
p_opts->valid=0;
#define FREE_PP_OPTS() { \
if(pp_opts!=NULL) { \
for(uint32_t i=0;i<opts_count;i++) free(pp_opts[i]); \
free(pp_opts); \
} \
}
// Search next assignment operator
sep_pos=0;
while(p_buf[sep_pos]!='\0' && p_buf[sep_pos]!='=') sep_pos++;
if(sep_pos==0 || p_buf[sep_pos]=='\0') {
LOG_ERROR("Library parameter '%s' is missing an assignment operator!\n",
p_buf);
free(p_opts);
FREE_PP_OPTS();
return FALSE;
}
// Save option key
XMOUNT_STRNSET(p_opts->p_key,p_buf,sep_pos);
p_buf+=(sep_pos+1);
// Search next separator
sep_pos=0;
while(p_buf[sep_pos]!='\0' && p_buf[sep_pos]!=',') sep_pos++;
if(sep_pos==0) {
LOG_ERROR("Library parameter '%s' is not of format key=value!\n",
p_opts->p_key);
free(p_opts->p_key);
free(p_opts);
FREE_PP_OPTS();
return FALSE;
}
// Save option value
XMOUNT_STRNSET(p_opts->p_value,p_buf,sep_pos);
p_buf+=sep_pos;
LOG_DEBUG("Extracted library option: '%s' = '%s'\n",
p_opts->p_key,
p_opts->p_value);
#undef FREE_PP_OPTS
// Add current option to return array
XMOUNT_REALLOC(pp_opts,
pts_LibXmountOptions*,
sizeof(pts_LibXmountOptions)*(opts_count+1));
pp_opts[opts_count++]=p_opts;
// If we're not at the end of p_params, skip over separator for next run
if(*p_buf!='\0') p_buf++;
}
LOG_DEBUG("Extracted a total of %" PRIu32 " library options\n",opts_count);
*p_ret_opts_count=opts_count;
*ppp_ret_opt=pp_opts;
return TRUE;
}
/*******************************************************************************
* LibXmount_Morphing function implementation
******************************************************************************/
//! Function to get the amount of input images
/*!
* \param p_count Count of input images
* \return 0 on success
*/
static int LibXmount_Morphing_ImageCount(uint64_t *p_count) {
*p_count=glob_xmount.input.images_count;
return 0;
}
//! Function to get the size of the morphed data
/*!
* \param image Image number
* \param p_size Pointer to store input image's size to
* \return 0 on success
*/
static int LibXmount_Morphing_Size(uint64_t image, uint64_t *p_size) {
if(image>=glob_xmount.input.images_count) return -1;
*p_size=glob_xmount.input.pp_images[image]->size;
return 0;
}
//! Function to read data from input image
/*!
* \param image Image number
* \param p_buf Buffer to store read data to
* \param offset Position at which to start reading
* \param count Amount of bytes to read
* \param p_read Number of read bytes on success
* \return 0 on success or negated error code on error
*/
static int LibXmount_Morphing_Read(uint64_t image,
char *p_buf,
off_t offset,
size_t count,
size_t *p_read)
{
if(image>=glob_xmount.input.images_count) return -EIO;
return ReadInputImageData(glob_xmount.input.pp_images[image],
p_buf,
offset,
count,
p_read);
}
/*******************************************************************************
* LibXmount_Output function implementation
******************************************************************************/
//! Function to get the size of the morphed image
/*!
* \param p_size Pointer to store morphed image's size to
* \return 0 on success
*/
static int LibXmount_Output_Size(uint64_t *p_size) {
int ret=0;
ret=GetMorphedImageSize(p_size);
return ret==TRUE ? 0 : ret;
}
//! Function to read data from the morphed image
/*!
* \param p_buf Buffer to store read data to
* \param offset Position at which to start reading
* \param count Amount of bytes to read
* \param p_read Number of read bytes on success
* \return 0 on success or negated error code on error
*/
static int LibXmount_Output_Read(char *p_buf,
off_t offset,
size_t count,
size_t *p_read)
{
return ReadMorphedImageData(p_buf,offset,count,p_read);
}
//! Function to write data to the morphed image
/*!
* \param p_buf Buffer with data to write
* \param offset Position at which to start writing
* \param count Amount of bytes to write
* \param p_written Number of written bytes on success
* \return 0 on success or negated error code on error
*/
static int LibXmount_Output_Write(char *p_buf,
off_t offset,
size_t count,
size_t *p_written)
{
return WriteMorphedImageData(p_buf,offset,count,p_written);
}
/*******************************************************************************
* Main
******************************************************************************/
int main(int argc, char *argv[]) {
struct stat file_stat;
int ret;
int fuse_ret;
char *p_err_msg;
// Set implemented FUSE functions
struct fuse_operations xmount_operations = {
//.access=FuseAccess,
.getattr=Xmount_FuseGetAttr,
.mkdir=Xmount_FuseMkDir,
.mknod=Xmount_FuseMkNod,
.open=Xmount_FuseOpen,
.readdir=Xmount_FuseReadDir,
.read=Xmount_FuseRead,
.rename=Xmount_FuseRename,
.rmdir=Xmount_FuseRmDir,
//.statfs=FuseStatFs,
.unlink=Xmount_FuseUnlink,
.write=Xmount_FuseWrite
};
// Disable std output / input buffering
setbuf(stdout,NULL);
setbuf(stderr,NULL);
// Init glob_xmount
InitResources();
// Load input and morphing libs
if(!LoadLibs()) {
LOG_ERROR("Unable to load any libraries!\n")
return 1;
}
// Check FUSE settings
CheckFuseSettings();
// Parse command line options
if(ParseCmdLine(argc,argv)!=TRUE) {
PrintUsage(argv[0]);
FreeResources();
return 1;
}
// Check command line options
if(glob_xmount.input.images_count==0) {
LOG_ERROR("No --in command line option specified!\n")
PrintUsage(argv[0]);
FreeResources();
return 1;
}
if(glob_xmount.fuse_argc<2) {
LOG_ERROR("Couldn't parse command line options!\n")
PrintUsage(argv[0]);
FreeResources();
return 1;
}
if(glob_xmount.morphing.p_morph_type==NULL) {
XMOUNT_STRSET(glob_xmount.morphing.p_morph_type,"combine");
}
// Check if mountpoint is a valid dir
if(stat(glob_xmount.p_mountpoint,&file_stat)!=0) {
LOG_ERROR("Unable to stat mount point '%s'!\n",glob_xmount.p_mountpoint);
PrintUsage(argv[0]);
FreeResources();
return 1;
}
if(!S_ISDIR(file_stat.st_mode)) {
LOG_ERROR("Mount point '%s' is not a directory!\n",
glob_xmount.p_mountpoint);
PrintUsage(argv[0]);
FreeResources();
return 1;
}
if(glob_xmount.debug==TRUE) {
LOG_DEBUG("Options passed to FUSE: ")
for(int i=0;i<glob_xmount.fuse_argc;i++) {
printf("%s ",glob_xmount.pp_fuse_argv[i]);
}
printf("\n");
}
// Init mutexes
pthread_mutex_init(&(glob_xmount.mutex_image_rw),NULL);
pthread_mutex_init(&(glob_xmount.mutex_info_read),NULL);
// Load input images
for(uint64_t i=0;i<glob_xmount.input.images_count;i++) {
if(glob_xmount.debug==TRUE) {
if(glob_xmount.input.pp_images[i]->files_count==1) {
LOG_DEBUG("Loading image file \"%s\"...\n",
glob_xmount.input.pp_images[i]->pp_files[0])
} else {
LOG_DEBUG("Loading image files \"%s .. %s\"...\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->
pp_files[glob_xmount.input.pp_images[i]->files_count-1])
}
}
// Find input lib
if(!FindInputLib(glob_xmount.input.pp_images[i])) {
LOG_ERROR("Unknown input image type '%s' for input image '%s'!\n",
glob_xmount.input.pp_images[i]->p_type,
glob_xmount.input.pp_images[i]->pp_files[0])
PrintUsage(argv[0]);
FreeResources();
return 1;
}
// Init input image handle
ret=glob_xmount.input.pp_images[i]->p_functions->
CreateHandle(&(glob_xmount.input.pp_images[i]->p_handle),
glob_xmount.input.pp_images[i]->p_type,
glob_xmount.debug);
if(ret!=0) {
LOG_ERROR("Unable to init input handle for input image '%s': %s!\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret));
FreeResources();
return 1;
}
// Parse input lib specific options
if(glob_xmount.input.pp_lib_params!=NULL) {
ret=glob_xmount.input.pp_images[i]->p_functions->
OptionsParse(glob_xmount.input.pp_images[i]->p_handle,
glob_xmount.input.lib_params_count,
glob_xmount.input.pp_lib_params,
(const char**)&p_err_msg);
if(ret!=0) {
if(p_err_msg!=NULL) {
LOG_ERROR("Unable to parse input library specific options for image "
"'%s': %s: %s!\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret),
p_err_msg);
glob_xmount.input.pp_images[i]->p_functions->FreeBuffer(p_err_msg);
FreeResources();
return 1;
} else {
LOG_ERROR("Unable to parse input library specific options for image "
"'%s': %s!\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret));
FreeResources();
return 1;
}
}
}
// Open input image
ret=
glob_xmount.input.pp_images[i]->
p_functions->
Open(glob_xmount.input.pp_images[i]->p_handle,
(const char**)(glob_xmount.input.pp_images[i]->pp_files),
glob_xmount.input.pp_images[i]->files_count);
if(ret!=0) {
LOG_ERROR("Unable to open input image file '%s': %s!\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->p_functions->
GetErrorMessage(ret));
FreeResources();
return 1;
}
// Determine input image size
ret=glob_xmount.input.pp_images[i]->
p_functions->
Size(glob_xmount.input.pp_images[i]->p_handle,
&(glob_xmount.input.pp_images[i]->size));
if(ret!=0) {
LOG_ERROR("Unable to determine size of input image '%s': %s!\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.pp_images[i]->
p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
// If an offset was specified, check it against offset and change size
if(glob_xmount.input.image_offset!=0) {
if(glob_xmount.input.image_offset>glob_xmount.input.pp_images[i]->size) {
LOG_ERROR("The specified offset is larger than the size of the input "
"image '%s'! (%" PRIu64 " > %" PRIu64 ")\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.image_offset,
glob_xmount.input.pp_images[i]->size);
FreeResources();
return 1;
}
glob_xmount.input.pp_images[i]->size-=glob_xmount.input.image_offset;
}
// If a size limit was specified, check it and change size
if(glob_xmount.input.image_size_limit!=0) {
if(glob_xmount.input.pp_images[i]->size<
glob_xmount.input.image_size_limit)
{
LOG_ERROR("The specified size limit is larger than the size of the "
"input image '%s'! (%" PRIu64 " > %" PRIu64 ")\n",
glob_xmount.input.pp_images[i]->pp_files[0],
glob_xmount.input.image_size_limit,
glob_xmount.input.pp_images[i]->size);
FreeResources();
return 1;
}
glob_xmount.input.pp_images[i]->size=glob_xmount.input.image_size_limit;
}
LOG_DEBUG("Input image loaded successfully\n")
}
// Find morphing lib
if(FindMorphingLib()!=TRUE) {
LOG_ERROR("Unable to find a library supporting the morphing type '%s'!\n",
glob_xmount.morphing.p_morph_type);
FreeResources();
return 1;
}
// Init morphing
ret=glob_xmount.morphing.p_functions->
CreateHandle(&glob_xmount.morphing.p_handle,
glob_xmount.morphing.p_morph_type,
glob_xmount.debug);
if(ret!=0) {
LOG_ERROR("Unable to create morphing handle: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
// Parse morphing lib specific options
if(glob_xmount.morphing.pp_lib_params!=NULL) {
p_err_msg=NULL;
ret=glob_xmount.morphing.p_functions->
OptionsParse(glob_xmount.morphing.p_handle,
glob_xmount.morphing.lib_params_count,
glob_xmount.morphing.pp_lib_params,
(const char**)&p_err_msg);
if(ret!=0) {
if(p_err_msg!=NULL) {
LOG_ERROR("Unable to parse morphing library specific options: %s: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret),
p_err_msg);
glob_xmount.morphing.p_functions->FreeBuffer(p_err_msg);
FreeResources();
return 1;
} else {
LOG_ERROR("Unable to parse morphing library specific options: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
}
}
// Morph image
ret=glob_xmount.morphing.p_functions->
Morph(glob_xmount.morphing.p_handle,
&(glob_xmount.morphing.input_image_functions));
if(ret!=0) {
LOG_ERROR("Unable to start morphing: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
// Init random generator
srand(time(NULL));
// Calculate partial MD5 hash of input image file
if(CalculateInputImageHash(&(glob_xmount.input.image_hash_lo),
&(glob_xmount.input.image_hash_hi))==FALSE)
{
LOG_ERROR("Couldn't calculate partial hash of morphed image!\n")
return 1;
}
if(glob_xmount.debug==TRUE) {
LOG_DEBUG("Partial MD5 hash of morphed image: ")
for(int i=0;i<8;i++)
printf("%02hhx",*(((char*)(&(glob_xmount.input.image_hash_lo)))+i));
for(int i=0;i<8;i++)
printf("%02hhx",*(((char*)(&(glob_xmount.input.image_hash_hi)))+i));
printf("\n");
}
if(!ExtractOutputFileNames(glob_xmount.input.pp_images[0]->pp_files[0])) {
LOG_ERROR("Couldn't extract virtual file names!\n");
FreeResources();
return 1;
}
LOG_DEBUG("Virtual file names extracted successfully\n")
// Find output lib
if(FindOutputLib()!=TRUE) {
LOG_ERROR("Unable to find a library supporting the output format '%s'!\n",
glob_xmount.output.p_output_format);
FreeResources();
return 1;
}
// Init output
ret=glob_xmount.output.p_functions->
CreateHandle(&glob_xmount.output.p_handle,
glob_xmount.output.p_output_format,
glob_xmount.debug);
if(ret!=0) {
LOG_ERROR("Unable to create output handle: %s!\n",
glob_xmount.output.p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
// Parse output lib specific options
if(glob_xmount.output.pp_lib_params!=NULL) {
p_err_msg=NULL;
ret=glob_xmount.output.p_functions->
OptionsParse(glob_xmount.output.p_handle,
glob_xmount.output.lib_params_count,
glob_xmount.output.pp_lib_params,
(const char**)&p_err_msg);
if(ret!=0) {
if(p_err_msg!=NULL) {
LOG_ERROR("Unable to parse output library specific options: %s: %s!\n",
glob_xmount.output.p_functions->GetErrorMessage(ret),
p_err_msg);
glob_xmount.output.p_functions->FreeBuffer(p_err_msg);
FreeResources();
return 1;
} else {
LOG_ERROR("Unable to parse output library specific options: %s!\n",
glob_xmount.output.p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
}
}
// Morph image
ret=glob_xmount.morphing.p_functions->
Morph(glob_xmount.morphing.p_handle,
&(glob_xmount.morphing.input_image_functions));
if(ret!=0) {
LOG_ERROR("Unable to start morphing: %s!\n",
glob_xmount.morphing.p_functions->GetErrorMessage(ret));
FreeResources();
return 1;
}
// Gather infos for info file
if(!InitInfoFile()) {
LOG_ERROR("Couldn't gather infos for virtual image info file!\n")
FreeResources();
return 1;
}
LOG_DEBUG("Virtual image info file build successfully\n")
if(glob_xmount.output.writable) {
// Init cache file and cache file block index
if(!InitCacheFile()) {
LOG_ERROR("Couldn't initialize cache file!\n")
FreeResources();
return 1;
}
LOG_DEBUG("Cache file initialized successfully\n")
}
// Call fuse_main to do the fuse magic
fuse_ret=fuse_main(glob_xmount.fuse_argc,
glob_xmount.pp_fuse_argv,
&xmount_operations,
NULL);
// Destroy mutexes
pthread_mutex_destroy(&(glob_xmount.mutex_image_rw));
pthread_mutex_destroy(&(glob_xmount.mutex_info_read));
// Free allocated memory
FreeResources();
return fuse_ret;
}
diff --git a/src/xmount.h b/src/xmount.h
index 72e993c..58bfb58 100755
--- a/src/xmount.h
+++ b/src/xmount.h
@@ -1,312 +1,83 @@
/*******************************************************************************
* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
* *
* xmount is a small tool to "fuse mount" various image formats and enable *
* virtual write access. *
* *
* This program is free software: you can redistribute it and/or modify it *
* under the terms of the GNU General Public License as published by the Free *
* Software Foundation, either version 3 of the License, or (at your option) *
* any later version. *
* *
* This program is distributed in the hope that it will be useful, but WITHOUT *
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
* more details. *
* *
* You should have received a copy of the GNU General Public License along with *
* this program. If not, see <http://www.gnu.org/licenses/>. *
*******************************************************************************/
#ifndef XMOUNT_H
#define XMOUNT_H
-#include <gidafs.h>
-
-#include "../libxmount_input/libxmount_input.h"
-#include "../libxmount_morphing/libxmount_morphing.h"
-#include "../libxmount_output/libxmount_output.h"
+#include "xmount_input.h"
+#include "xmount_morphing.h"
+#include "xmount_cache.h"
+#include "xmount_output.h"
#undef FALSE
#undef TRUE
#define FALSE 0
#define TRUE 1
/*
* Constants
*/
#define IMAGE_INFO_INPUT_HEADER \
"------> The following values are supplied by the used input library(ies) " \
"<------\n"
#define IMAGE_INFO_MORPHING_HEADER \
"\n------> The following values are supplied by the used morphing library " \
"<------\n\n"
/*******************************************************************************
* Xmount specific structures
******************************************************************************/
-
-#ifdef __LP64__
- #define CACHE_BLOCK_FREE 0xFFFFFFFFFFFFFFFF
-#else
- #define CACHE_BLOCK_FREE 0xFFFFFFFFFFFFFFFFLL
-#endif
-#ifdef __LP64__
- #define XMOUNT_BLOCK_CACHE_INVALID_INDEX 0xFFFFFFFFFFFFFFFF
-#else
- #define XMOUNT_BLOCK_CACHE_INVALID_INDEX 0xFFFFFFFFFFFFFFFFLL
-#endif
-//! Cache file block index array element
-typedef uint64_t t_CacheFileBlockIndex;
-// TODO: Remove
-typedef struct s_CacheFileBlockIndex {
- //! Set to 1 if block is assigned (this block has data in cache file)
- uint32_t Assigned;
- //! Offset to data in cache file
- uint64_t off_data;
-} __attribute__ ((packed)) ts_CacheFileBlockIndex, *pts_CacheFileBlockIndex;
-
-#define CACHE_BLOCK_SIZE (1024*1024) // 1 megabyte
-#ifdef __LP64__
- #define CACHE_FILE_SIGNATURE 0xFFFF746E756F6D78 // "xmount\xFF\xFF"
-#else
- #define CACHE_FILE_SIGNATURE 0xFFFF746E756F6D78LL
-#endif
-#define CUR_CACHE_FILE_VERSION 0x00000002 // Current cache file version
-#define HASH_AMOUNT (1024*1024)*10 // Amount of data used to construct a
- // "unique" hash for every input image
- // (10MByte)
-//! Cache file header structure
-typedef struct s_CacheFileHeader {
- //! Simple signature to identify cache files
- uint64_t FileSignature;
- //! Cache file version
- uint32_t CacheFileVersion;
- //! Cache block size
- uint64_t BlockSize;
- //! Total amount of cache blocks
- uint64_t BlockCount;
- //! Offset to the first block index array element
- uint64_t pBlockIndex;
- //! Set to 1 if VDI file header is cached
- uint32_t VdiFileHeaderCached;
- //! Offset to cached VDI file header
- uint64_t pVdiFileHeader;
- //! Set to 1 if VMDK file is cached
- uint32_t VmdkFileCached;
- //! Size of VMDK file
- uint64_t VmdkFileSize;
- //! Offset to cached VMDK file
- uint64_t pVmdkFile;
- //! Set to 1 if VHD header is cached
- uint32_t VhdFileHeaderCached;
- //! Offset to cached VHD header
- uint64_t pVhdFileHeader;
- //! Padding to get 512 byte alignment and ease further additions
- char HeaderPadding[432];
-} __attribute__ ((packed)) ts_CacheFileHeader, *pts_CacheFileHeader;
-
-//! Cache file header structure - Old v1 header
-typedef struct s_CacheFileHeader_v1 {
- //! Simple signature to identify cache files
- uint64_t FileSignature;
- //! Cache file version
- uint32_t CacheFileVersion;
- //! Total amount of cache blocks
- uint64_t BlockCount;
- //! Offset to the first block index array element
- uint64_t pBlockIndex;
- //! Set to 1 if VDI file header is cached
- uint32_t VdiFileHeaderCached;
- //! Offset to cached VDI file header
- uint64_t pVdiFileHeader;
- //! Set to 1 if VMDK file is cached
-} ts_CacheFileHeader_v1, *pts_CacheFileHeader_v1;
-
-//! Structure containing infos about input libs
-typedef struct s_InputLib {
- //! Filename of lib (without path)
- char *p_name;
- //! Handle to the loaded lib
- void *p_lib;
- //! Array of supported input types
- char *p_supported_input_types;
- //! Struct containing lib functions
- ts_LibXmountInputFunctions lib_functions;
-} ts_InputLib, *pts_InputLib;
-
-//! Structure containing infos about input images
-typedef struct s_InputImage {
- //! Image type
- char *p_type;
- //! Image source file count
- uint64_t files_count;
- //! Image source files
- char **pp_files;
- //! Input lib functions for this image
- pts_LibXmountInputFunctions p_functions;
- //! Image handle
- void *p_handle;
- //! Image size
- uint64_t size;
-} ts_InputImage, *pts_InputImage;
-
-typedef struct s_InputData {
- //! Loaded input lib count
- uint32_t libs_count;
- //! Array containing infos about loaded input libs
- pts_InputLib *pp_libs;
- //! Amount of input lib params (--inopts)
- uint32_t lib_params_count;
- //! Input lib params (--inopts)
- pts_LibXmountOptions *pp_lib_params;
- //! Input image count
- uint64_t images_count;
- //! Input images
- pts_InputImage *pp_images;
- //! Input image offset (--offset)
- uint64_t image_offset;
- //! Input image size limit (--sizelimit)
- uint64_t image_size_limit;
- //! MD5 hash of partial input image (lower 64 bit) (after morph)
- uint64_t image_hash_lo;
- //! MD5 hash of partial input image (higher 64 bit) (after morph)
- uint64_t image_hash_hi;
-} ts_InputData;
-
-//! Structure containing infos about morphing libs
-typedef struct s_MorphingLib {
- //! Filename of lib (without path)
- char *p_name;
- //! Handle to the loaded lib
- void *p_lib;
- //! Array of supported morphing types
- char *p_supported_morphing_types;
- //! Struct containing lib functions
- ts_LibXmountMorphingFunctions lib_functions;
-} ts_MorphingLib, *pts_MorphingLib;
-
-//! Structures and vars needed for morph support
-typedef struct s_MorphingData {
- //! Loaded morphing lib count
- uint32_t libs_count;
- //! Array containing infos about loaded morphing libs
- pts_MorphingLib *pp_libs;
- //! Specified morphing type (--morph)
- char *p_morph_type;
- //! Amount of specified morphing lib params (--morphopts)
- uint32_t lib_params_count;
- //! Specified morphing lib params (--morphopts)
- pts_LibXmountOptions *pp_lib_params;
- //! Handle to initialized morphing lib
- void *p_handle;
- //! Morphing functions of initialized lib
- pts_LibXmountMorphingFunctions p_functions;
- //! Input image functions passed to morphing lib
- ts_LibXmountMorphingInputFunctions input_image_functions;
-} ts_MorphingData;
-
-//! Structures and vars needed for write access
-#define XMOUNT_CACHE_FOLDER "/.xmount"
-#define XMOUNT_CACHE_BLOCK_FILE XMOUNT_CACHE_FOLDER "/blocks.data"
-#define XMOUNT_CACHE_BLOCK_INDEX_FILE XMOUNT_CACHE_FOLDER "/blocks.index"
-typedef struct s_CacheData {
- //! Cache file to save changes to
- char *p_cache_file;
- //! Handle to cache file
- hGidaFs h_cache_file;
- //! Handle to block cache
- hGidaFsFile h_block_cache;
- //! Handle to block cache index
- hGidaFsFile h_block_cache_index;
- //! In-memory copy of cache index
- t_CacheFileBlockIndex *p_block_cache_index;
- //! Length (in elements) of in-memory block cache index
- uint64_t block_cache_index_len;
- // TODO: Move to s_XmountData
- //! Overwrite existing cache
- uint8_t overwrite_cache;
-} ts_CacheData;
-
-//! Structure containing infos about output libs
-typedef struct s_OutputLib {
- //! Filename of lib (without path)
- char *p_name;
- //! Handle to the loaded lib
- void *p_lib;
- //! Array of supported output formats
- char *p_supported_output_formats;
- //! Struct containing lib functions
- ts_LibXmountOutput_Functions lib_functions;
-} ts_OutputLib, *pts_OutputLib;
-
-//! Structure containing infos about output image
-typedef struct s_OutputData {
- //! Loaded output lib count
- uint32_t libs_count;
- //! Array containing infos about loaded output libs
- pts_OutputLib *pp_libs;
- //! Specified output format (--out)
- char *p_output_format;
- //! Amount of specified output lib params
- uint32_t lib_params_count;
- //! Specified output lib params (--outopts)
- pts_LibXmountOptions *pp_lib_params;
- //! Handle to initialized output lib
- void *p_handle;
- //! Transformation functions of initialized lib
- pts_LibXmountOutput_Functions p_functions;
- //! Input image functions passed to output lib
- ts_LibXmountOutput_InputFunctions input_functions;
- //! Size
- uint64_t image_size;
- //! Writable? (Set to 1 if --cache was specified)
- uint8_t writable;
- //! Path of virtual image file
- char *p_virtual_image_path;
- //! Path of virtual image info file
- char *p_info_path;
- //! Pointer to virtual info file
- char *p_info_file;
-} ts_OutputData;
-
//! Structure containing global xmount runtime infos
typedef struct s_XmountData {
//! Input image related data
ts_InputData input;
//! Morphing related data
ts_MorphingData morphing;
//! Cache file related data
ts_CacheData cache;
//! Output image related data
ts_OutputData output;
//! Enable debug output
uint8_t debug;
//! Set if we are allowed to set fuse's allow_other option
uint8_t may_set_fuse_allow_other;
//! Argv for FUSE
int fuse_argc;
//! Argv for FUSE
char **pp_fuse_argv;
//! Mount point
char *p_mountpoint;
//! Mutex to control concurrent read & write access on output image
pthread_mutex_t mutex_image_rw;
//! Mutex to control concurrent read access on info file
pthread_mutex_t mutex_info_read;
} ts_XmountData;
/*******************************************************************************
* Global vars
******************************************************************************/
//! Struct that contains various runtime configuration options
ts_XmountData glob_xmount;
/*******************************************************************************
* Xmount functions
******************************************************************************/
-int GetOutputImageSize(uint64_t*);
-int ReadOutputImageData(char*, off_t, size_t);
-int WriteOutputImageData(const char*, off_t, size_t);
#endif // XMOUNT_H
diff --git a/src/xmount_cache.c b/src/xmount_cache.c
new file mode 100644
index 0000000..576b3ef
--- /dev/null
+++ b/src/xmount_cache.c
@@ -0,0 +1,957 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+
+#include "xmount_cache.h"
+#include "xmount.h"
+#include "macros.h"
+
+/*******************************************************************************
+ * Private definitions / macros
+ ******************************************************************************/
+#define LOG_WARNING(...) { \
+ LIBXMOUNT_LOG_WARNING(__VA_ARGS__); \
+}
+#define LOG_ERROR(...) { \
+ LIBXMOUNT_LOG_ERROR(__VA_ARGS__); \
+}
+#define LOG_DEBUG(...) { \
+ LIBXMOUNT_LOG_DEBUG(glob_xmount.debug,__VA_ARGS__); \
+}
+
+#ifndef __LP64__
+ //! Value used to indicate an uncached block entry
+ #define XMOUNT_CACHE_INVALID_INDEX 0xFFFFFFFFFFFFFFFFLL
+#else
+ #define XMOUNT_CACHE_INVALID_INDEX 0xFFFFFFFFFFFFFFFF
+#endif
+
+//! Structures and vars needed for write access
+#define XMOUNT_CACHE_FOLDER "/.xmount"
+#define XMOUNT_CACHE_BLOCK_FILE XMOUNT_CACHE_FOLDER "/blocks.data"
+#define XMOUNT_CACHE_BLOCK_INDEX_FILE XMOUNT_CACHE_FOLDER "/blocks.index"
+
+/*******************************************************************************
+ * Private types / structures / enums
+ ******************************************************************************/
+typedef struct s_XmountCacheHandle {
+ //! Cache file to save changes to
+ char *p_cache_file;
+ //! Handle to cache file
+ hGidaFs h_cache_file;
+ //! Handle to block cache
+ hGidaFsFile h_block_cache;
+ //! Handle to block cache index
+ hGidaFsFile h_block_cache_index;
+ //! In-memory copy of cache index
+ uint64_t *p_block_cache_index;
+ //! Length (in elements) of in-memory block cache index
+ uint64_t block_cache_index_len;
+ // TODO: Move to s_XmountData
+ //! Overwrite existing cache
+ uint8_t overwrite_cache;
+} ts_XmountCacheHandle, *pts_XmountCacheHandle;
+
+/*******************************************************************************
+ * Private functions declarations
+ ******************************************************************************/
+/*!
+ * \brief Create a new xmount cache handle
+ *
+ * \param pp_h Pointer to an xmount cache handle
+ * \param p_file Cache file path / name
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_CreateHandle(pts_XmountCacheHandle *pp_h,
+ const char *p_file);
+
+/*!
+ * \brief Destroy an xmount cache handle
+ *
+ * \param pp_h Pointer to an xmount cache handle
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_DestroyHandle(pts_XmountCacheHandle *pp_h);
+
+/*!
+ * \brief Check if a file exists
+ *
+ * Checks if the given file p_file exists.
+ *
+ * \param p_file File to check for
+ * \return e_XmountCache_Error_None if file exists
+ */
+te_XmountCache_Error XmountCache_FileExists(const char *p_file);
+
+/*******************************************************************************
+ * Public functions implementations
+ ******************************************************************************/
+/*
+ * XmountCache_Create
+ */
+te_XmountCache_Error XmountCache_Create(pts_XmountCacheHandle *pp_h,
+ const char *p_file,
+ uint64_t image_size,
+ uint8_t overwrite)
+{
+ pts_XmountCacheHandle p_h=NULL;
+ teGidaFsError gidafs_ret=eGidaFsError_None;
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+
+ // Check params
+ if(pp_h==NULL) return e_XmountCache_Error_InvalidHandlePointer;
+ if(p_file==NULL) return e_XmountCache_Error_InvalidString;
+ if(strlen(p_file)==0) return e_XmountCache_Error_InvalidFile;
+
+ // Make sure file does not exist when overwrite was not specified
+ if(overwrite==0 && XmountCache_FileExists(p_file)==e_XmountCache_Error_None) {
+ // Given file exists and overwrite was not specified. This is fatal!
+ return e_XmountCache_Error_ExistingFile;
+ }
+
+ // Create new handle
+ ret=XmountCache_CreateHandle(&p_h,p_file);
+ if(ret!=e_XmountCache_Error_None) return ret;
+
+#define XMOUNTCACHE_CREATE__DESTROY_HANDLE do { \
+ ret=XmountCache_DestroyHandle(&p_h); \
+ if(ret!=e_XmountCache_Error_None) { \
+ LOG_ERROR("Unable to destroy cache handle: Error code %u: Ignoring!\n", \
+ ret); \
+ } \
+ *pp_h=NULL; \
+} while(0)
+
+ // Create new cache file
+ gidafs_ret=GidaFsLib_NewFs(&(p_h->h_cache_file),
+ p_h->p_cache_file,
+ 0);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to create new xmount cache file '%s': Error code %u!\n",
+ p_h->p_cache_file,
+ gidafs_ret);
+ XMOUNTCACHE_CREATE__DESTROY_HANDLE;
+ return FALSE;
+ }
+
+#define XMOUNTCACHE_CREATE__CLOSE_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFs(&(p_h->h_cache_file)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close xmount cache file '%s': " \
+ "Error code %u: Ignoring!\n", \
+ p_h->p_cache_file, \
+ gidafs_ret); \
+ } \
+} while(0)
+
+ // TODO: Check if cache file uses same block size as we do
+
+ // Create needed xmount subdirectory
+ gidafs_ret=GidaFsLib_CreateDir(p_h->h_cache_file,
+ XMOUNT_CACHE_FOLDER,
+ eGidaFsNodeFlag_RWXu);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to create cache file directory '%s': Error code %u!\n",
+ XMOUNT_CACHE_FOLDER,
+ gidafs_ret);
+ XMOUNTCACHE_CREATE__CLOSE_CACHE;
+ XMOUNTCACHE_CREATE__DESTROY_HANDLE;
+ return e_XmountCache_Error_FailedCacheInit;
+ }
+
+ // Create block cache file
+ gidafs_ret=GidaFsLib_OpenFile(p_h->h_cache_file,
+ XMOUNT_CACHE_BLOCK_FILE,
+ &(p_h->h_block_cache),
+ eGidaFsOpenFileFlag_ReadWrite |
+ eGidaFsOpenFileFlag_CreateAlways,
+ eGidaFsNodeFlag_Rall | eGidaFsNodeFlag_Wusr);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to create block cache file '%s': Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret);
+ XMOUNTCACHE_CREATE__CLOSE_CACHE;
+ XMOUNTCACHE_CREATE__DESTROY_HANDLE;
+ return e_XmountCache_Error_FailedCacheInit;
+ }
+
+#define XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFile(p_h->h_cache_file, \
+ &(p_h->h_block_cache)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache file: Error code %u: " \
+ "Ignoring!\n", \
+ gidafs_ret); \
+ } \
+} while(0)
+
+ // Create block cache index file
+ gidafs_ret=GidaFsLib_OpenFile(p_h->h_cache_file,
+ XMOUNT_CACHE_BLOCK_INDEX_FILE,
+ &(p_h->h_block_cache_index),
+ eGidaFsOpenFileFlag_ReadWrite |
+ eGidaFsOpenFileFlag_CreateAlways,
+ eGidaFsNodeFlag_Rall | eGidaFsNodeFlag_Wusr);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to create block cache index file '%s': Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret);
+ XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE;
+ XMOUNTCACHE_CREATE__CLOSE_CACHE;
+ XMOUNTCACHE_CREATE__DESTROY_HANDLE;
+ return e_XmountCache_Error_FailedCacheInit;
+ }
+
+#define XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE_INDEX do { \
+ gidafs_ret=GidaFsLib_CloseFile(p_h->h_cache_file, \
+ &(p_h->h_block_cache_index)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache index file: Error code %u: " \
+ "Ignoring!\n", \
+ gidafs_ret); \
+ } \
+} while(0)
+
+ // Calculate how many cache blocks are needed and how big the cache block
+ // index must be
+ p_h->block_cache_index_len=image_size/XMOUNT_CACHE_BLOCK_SIZE;
+ if((image_size%XMOUNT_CACHE_BLOCK_SIZE)!=0) p_h->block_cache_index_len++;
+
+ LOG_DEBUG("Cache blocks: %" PRIu64 " entries using %" PRIu64 " bytes\n",
+ p_h->block_cache_index_len,
+ p_h->block_cache_index_len*sizeof(uint8_t));
+
+ // Prepare in-memory buffer for block cache index
+ p_h->p_block_cache_index=
+ (uint64_t*)calloc(1,p_h->block_cache_index_len*sizeof(uint64_t));
+ if(p_h->p_block_cache_index==NULL) {
+ XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE_INDEX;
+ XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE;
+ XMOUNTCACHE_CREATE__CLOSE_CACHE;
+ XMOUNTCACHE_CREATE__DESTROY_HANDLE;
+ return e_XmountCache_Error_Alloc;
+ }
+
+ // Generate initial block cache index
+ for(uint64_t i=0;i<p_h->block_cache_index_len;i++) {
+ p_h->p_block_cache_index[i]=XMOUNT_CACHE_INVALID_INDEX;
+ }
+
+ // Write initial block cache index to cache file
+ if(UpdateBlockCacheIndex(XMOUNT_BLOCK_CACHE_INVALID_INDEX,0)!=TRUE) {
+ LOG_ERROR("Unable to generate initial block cache index file '%s': "
+ "Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret);
+ XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE_INDEX;
+ XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE;
+ XMOUNTCACHE_CREATE__CLOSE_CACHE;
+ XMOUNTCACHE_CREATE__DESTROY_HANDLE;
+ return e_XmountCache_Error_FailedCacheInit;
+ }
+
+#undef XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE_INDEX
+#undef XMOUNTCACHE_CREATE__CLOSE_BLOCK_CACHE
+#undef XMOUNTCACHE_CREATE__CLOSE_CACHE
+#undef XMOUNTCACHE_CREATE__DESTROY_HANDLE
+
+ *pp_h=p_h;
+ return e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_Open
+ */
+te_XmountCache_Error XmountCache_Open(pts_XmountCacheHandle *pp_h,
+ const char *p_file,
+ uint64_t image_size)
+{
+ pts_XmountCacheHandle p_h=NULL;
+ teGidaFsError gidafs_ret=eGidaFsError_None;
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+
+ // Check params
+ if(pp_h==NULL) return e_XmountCache_Error_InvalidHandlePointer;
+ if(p_file==NULL) return e_XmountCache_Error_InvalidString;
+ if(strlen(p_file)==0) return e_XmountCache_Error_InvalidFile;
+
+ // Make sure file exists
+ if(XmountCache_FileExists(p_file)!=e_XmountCache_Error_None) {
+ // Given file does not exist. This is fatal!
+ return e_XmountCache_Error_InexistingFile;
+ }
+
+ // Create new handle
+ ret=XmountCache_CreateHandle(&p_h,p_file);
+ if(ret!=e_XmountCache_Error_None) return ret;
+
+#define XMOUNTCACHE_OPEN__DESTROY_HANDLE do { \
+ ret=XmountCache_DestroyHandle(&p_h); \
+ if(ret!=e_XmountCache_Error_None) { \
+ LOG_ERROR("Unable to destroy cache handle: Error code %u: Ignoring!\n", \
+ ret); \
+ } \
+ *pp_h=NULL; \
+} while(0)
+
+ // Open cache file
+ gidafs_ret=GidaFsLib_OpenFs(&(p_h->h_cache_file),p_h->p_cache_file);
+ if(gidafs_ret!=eGidaFsError_None) {
+ // TODO: Check for old cache file type and inform user it isn't supported
+ // anymore!
+ LOG_ERROR("Couldn't open xmount cache file '%s': Error code %u!\n",
+ p_h->p_cache_file,
+ gidafs_ret)
+ return e_XmountCache_Error_FailedOpeningCache;
+ }
+
+#define XMOUNTCACHE_OPEN__CLOSE_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFs(&(p_h->h_cache_file)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close xmount cache file '%s': " \
+ "Error code %u: Ignoring!\n", \
+ p_h->p_cache_file, \
+ gidafs_ret); \
+ } \
+} while(0)
+
+#define XMOUNTCACHE_OPEN__CLOSE_BLOCK_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFile(p_h->h_cache_file, \
+ &(p_h->h_block_cache)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache file: Error code %u: " \
+ "Ignoring!\n", \
+ gidafs_ret); \
+ } \
+} while(0)
+
+#define XMOUNTCACHE_OPEN__CLOSE_BLOCK_CACHE_INDEX do { \
+ gidafs_ret=GidaFsLib_CloseFile(p_h->h_cache_file, \
+ &(p_h->h_block_cache_index)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache index file: Error code %u: " \
+ "Ignoring!\n", \
+ gidafs_ret); \
+ } \
+} while(0)
+
+
+
+
+
+
+
+
+
+/*
+
+
+#define INITCACHEFILE__CLOSE_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFs(&(glob_xmount.cache.h_cache_file)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close cache file: Error code %u: Ignoring!\n", \
+ gidafs_ret) \
+ } \
+} while(0)
+
+#define INITCACHEFILE__CLOSE_BLOCK_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file, \
+ &(glob_xmount.cache.h_block_cache)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache file: Error code %u: Ignoring!\n", \
+ gidafs_ret) \
+ } \
+} while(0)
+
+#define INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX do { \
+ gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file, \
+ &(glob_xmount.cache.h_block_cache_index)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache index file: Error code %u: " \
+ "Ignoring!\n", \
+ gidafs_ret) \
+ } \
+} while(0)
+
+ // TODO: Check if cache file uses same block size as we do
+
+ if(is_new_cache_file==1) {
+ // New cache file, create needed xmount subdirectory
+ gidafs_ret=GidaFsLib_CreateDir(glob_xmount.cache.h_cache_file,
+ XMOUNT_CACHE_FOLDER,
+ eGidaFsNodeFlag_RWXu);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to create cache file directory '%s': Error code %u!\n",
+ XMOUNT_CACHE_FOLDER,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_CACHE;
+ return FALSE;
+ }
+ }
+
+ // Open / Create block cache file
+ gidafs_ret=GidaFsLib_OpenFile(glob_xmount.cache.h_cache_file,
+ XMOUNT_CACHE_BLOCK_FILE,
+ &(glob_xmount.cache.h_block_cache),
+ eGidaFsOpenFileFlag_ReadWrite |
+ (is_new_cache_file==1 ?
+ eGidaFsOpenFileFlag_CreateAlways : 0),
+ eGidaFsNodeFlag_Rall |
+ eGidaFsNodeFlag_Wusr);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to open / create block cache file '%s': Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_CACHE;
+ return FALSE;
+ }
+
+ // Open / Create block cache index file
+ gidafs_ret=GidaFsLib_OpenFile(glob_xmount.cache.h_cache_file,
+ XMOUNT_CACHE_BLOCK_INDEX_FILE,
+ &(glob_xmount.cache.h_block_cache_index),
+ eGidaFsOpenFileFlag_ReadWrite |
+ (is_new_cache_file==1 ?
+ eGidaFsOpenFileFlag_CreateAlways : 0),
+ eGidaFsNodeFlag_Rall |
+ eGidaFsNodeFlag_Wusr);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to open / create block cache index file '%s': "
+ "Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ return FALSE;
+ }
+
+ // Calculate how many cache blocks are needed and how big the cache block
+ // index must be
+ glob_xmount.cache.block_cache_index_len=image_size/CACHE_BLOCK_SIZE;
+ if((image_size%CACHE_BLOCK_SIZE)!=0) {
+ glob_xmount.cache.block_cache_index_len++;
+ }
+
+ LOG_DEBUG("Cache blocks: %u (0x%04X) entries using %zd (0x%08zX) bytes\n",
+ glob_xmount.cache.block_cache_index_len,
+ glob_xmount.cache.block_cache_index_len,
+ glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex),
+ glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex))
+
+ // Prepare in-memory buffer for block cache index
+ XMOUNT_MALLOC(glob_xmount.cache.p_block_cache_index,
+ t_CacheFileBlockIndex*,
+ glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex))
+
+ if(is_new_cache_file==1) {
+ // Generate initial block cache index
+ for(uint64_t i=0;i<glob_xmount.cache.block_cache_index_len;i++) {
+ glob_xmount.cache.p_block_cache_index[i]=CACHE_BLOCK_FREE;
+ }
+ // Write initial block cache index to cache file
+ if(UpdateBlockCacheIndex(XMOUNT_BLOCK_CACHE_INVALID_INDEX,0)!=TRUE) {
+ LOG_ERROR("Unable to generate initial block cache index file '%s': "
+ "Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ } else {
+ // Existing cache file, make sure block cache index has correct size
+ gidafs_ret=GidaFsLib_GetFileSize(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache_index,
+ &blockindex_size);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to get block cache index file size: Error code %u!\n",
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ if(blockindex_size%sizeof(t_CacheFileBlockIndex)!=0 ||
+ (blockindex_size/sizeof(t_CacheFileBlockIndex))!=
+ glob_xmount.cache.block_cache_index_len)
+ {
+ // TODO: Be more helpfull in error message
+ LOG_ERROR("Block cache index size is incorrect for given input image!\n")
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ // Read block cache index into memory
+ gidafs_ret=GidaFsLib_ReadFile(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache_index,
+ 0,
+ blockindex_size,
+ glob_xmount.cache.p_block_cache_index,
+ &read);
+ if(gidafs_ret!=eGidaFsError_None || read!=blockindex_size) {
+ LOG_ERROR("Unable to read block cache index: Error code %u!\n",
+ gidafs_ret);
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ }
+
+#undef INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX
+#undef INITCACHEFILE__CLOSE_BLOCK_CACHE
+#undef INITCACHEFILE__CLOSE_CACHE
+*/
+
+#undef XMOUNTCACHE_OPEN__CLOSE_BLOCK_CACHE_INDEX
+#undef XMOUNTCACHE_OPEN__CLOSE_BLOCK_CACHE
+#undef XMOUNTCACHE_OPEN__CLOSE_CACHE
+#undef XMOUNTCACHE_OPEN__DESTROY_HANDLE
+
+ *pp_h=p_h;
+ return e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_Close
+ */
+te_XmountCache_Error XmountCache_Close(pts_XmountCacheHandle *pp_h) {
+ pts_XmountCacheHandle p_h=NULL;
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+
+
+
+ *pp_h=NULL;
+ return e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_BlockCacheRead
+ */
+te_XmountCache_Error XmountCache_BlockCacheRead(pts_XmountCacheHandle p_h,
+ char *p_buf,
+ uint64_t block,
+ uint64_t block_offset,
+ uint64_t count)
+{
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_BlockCacheWrite
+ */
+te_XmountCache_Error XmountCache_BlockCacheWrite(pts_XmountCacheHandle p_h,
+ char *p_buf,
+ uint64_t block,
+ uint64_t block_offset,
+ uint64_t count)
+{
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_BlockCacheAppend
+ */
+te_XmountCache_Error XmountCache_BlockCacheAppend(pts_XmountCacheHandle p_h,
+ char *p_buf,
+ uint64_t block)
+{
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_IsBlockCached
+ */
+te_XmountCache_Error XmountCache_IsBlockCached(pts_XmountCacheHandle p_h,
+ uint64_t block)
+{
+ te_XmountCache_Error ret=e_XmountCache_Error_None;
+}
+
+/*******************************************************************************
+ * Private functions implementations
+ ******************************************************************************/
+/*
+ * XmountCache_CreateHandle
+ */
+te_XmountCache_Error XmountCache_CreateHandle(pts_XmountCacheHandle *pp_h,
+ const char *p_file)
+{
+ pts_XmountCacheHandle p_h=NULL;
+
+ // Check given handle pointer
+ if(pp_h==NULL) return e_XmountCache_Error_InvalidHandlePointer;
+
+ // Alloc memory for handle
+ p_h=(pts_XmountCacheHandle)calloc(1,sizeof(ts_XmountCacheHandle));
+ if(p_h==NULL) {
+ *pp_h=NULL;
+ return e_XmountCache_Error_Alloc;
+ }
+
+ // Init values. The p_cache_file alloc and memcpy works corrently as strlen()
+ // counts the amount of bytes, not chars. No UTF8-issue here.
+ p_h->p_cache_file=(char*)calloc(1,strlen(p_file)+1);
+ if(p_h->p_cache_file==NULL) {
+ free(p_h);
+ *pp_h=NULL;
+ return e_XmountCache_Error_Alloc;
+ }
+ memcpy(p_h->p_cache_file,p_file,strlen(p_file));
+ p_h->h_cache_file=NULL;
+ p_h->h_block_cache=NULL;
+ p_h->h_block_cache_index=NULL;
+ p_h->p_block_cache_index=NULL;
+
+ // Return new handle
+ *pp_h=p_h;
+ return e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_DestroyHandle
+ */
+te_XmountCache_Error XmountCache_DestroyHandle(pts_XmountCacheHandle *pp_h) {
+ pts_XmountCacheHandle p_h=NULL;
+
+ // Check given handle pointer
+ if(pp_h==NULL) return e_XmountCache_Error_InvalidHandlePointer;
+ if(*pp_h==NULL) return e_XmountCache_Error_InvalidHandle;
+ p_h=*pp_h;
+
+ // Free handle
+ if(p_h->p_cache_file!=NULL) free(p_h->p_cache_file);
+ if(p_h->p_block_cache_index!=NULL) free(p_h->p_block_cache_index);
+ free(p_h);
+
+ // Return destroyed handle
+ *pp_h=NULL;
+ return e_XmountCache_Error_None;
+}
+
+/*
+ * XmountCache_FileExists
+ */
+te_XmountCache_Error XmountCache_FileExists(const char *p_file) {
+ struct stat statbuf;
+
+ if(lstat(p_file,&statbuf)==0) return e_XmountCache_Error_None;
+ else return e_XmountCache_Error_InexistingFile;
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+//! Create / load cache file to enable virtual write support
+/*!
+ * \return TRUE on success, FALSE on error
+ */
+int InitCacheFile() {
+ uint64_t blockindex_size=0;
+ uint64_t image_size=0;
+ uint64_t read=0;
+ uint8_t is_new_cache_file=0;
+ teGidaFsError gidafs_ret=eGidaFsError_None;
+
+ // Get input image size for later use
+ if(!GetMorphedImageSize(&image_size)) {
+ LOG_ERROR("Couldn't get morphed image size!\n")
+ return FALSE;
+ }
+
+ if(!glob_xmount.cache.overwrite_cache) {
+ // Try to open an existing cache file or create a new one
+ gidafs_ret=GidaFsLib_OpenFs(&(glob_xmount.cache.h_cache_file),
+ glob_xmount.cache.p_cache_file);
+ if(gidafs_ret!=eGidaFsError_None &&
+ gidafs_ret!=eGidaFsError_FailedOpeningFsFile)
+ {
+ // TODO: Check for old cache file type and inform user it isn't supported
+ // anymore!
+ LOG_ERROR("Couldn't open cache file '%s': Error code %u!\n",
+ glob_xmount.cache.p_cache_file,
+ gidafs_ret)
+ return FALSE;
+ } else if(gidafs_ret==eGidaFsError_FailedOpeningFsFile) {
+ // Unable to open cache file. It might simply not exist.
+ LOG_DEBUG("Cache file '%s' does not exist. Creating new one\n",
+ glob_xmount.cache.p_cache_file)
+ gidafs_ret=GidaFsLib_NewFs(&(glob_xmount.cache.h_cache_file),
+ glob_xmount.cache.p_cache_file,
+ 0);
+ if(gidafs_ret!=eGidaFsError_None) {
+ // There is really a problem opening/creating the file
+ LOG_ERROR("Couldn't open cache file '%s': Error code %u!\n",
+ glob_xmount.cache.p_cache_file,
+ gidafs_ret)
+ return FALSE;
+ }
+ is_new_cache_file=1;
+ }
+ } else {
+ // Overwrite existing cache file or create a new one
+ gidafs_ret=GidaFsLib_NewFs(&(glob_xmount.cache.h_cache_file),
+ glob_xmount.cache.p_cache_file,
+ 0);
+ if(gidafs_ret!=eGidaFsError_None) {
+ // There is really a problem opening/creating the file
+ LOG_ERROR("Couldn't open cache file '%s': Error code %u!\n",
+ glob_xmount.cache.p_cache_file,
+ gidafs_ret)
+ return FALSE;
+ }
+ is_new_cache_file=1;
+ }
+
+#define INITCACHEFILE__CLOSE_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFs(&(glob_xmount.cache.h_cache_file)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close cache file: Error code %u: Ignoring!\n", \
+ gidafs_ret) \
+ } \
+} while(0)
+
+#define INITCACHEFILE__CLOSE_BLOCK_CACHE do { \
+ gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file, \
+ &(glob_xmount.cache.h_block_cache)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache file: Error code %u: Ignoring!\n", \
+ gidafs_ret) \
+ } \
+} while(0)
+
+#define INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX do { \
+ gidafs_ret=GidaFsLib_CloseFile(glob_xmount.cache.h_cache_file, \
+ &(glob_xmount.cache.h_block_cache_index)); \
+ if(gidafs_ret!=eGidaFsError_None) { \
+ LOG_ERROR("Unable to close block cache index file: Error code %u: " \
+ "Ignoring!\n", \
+ gidafs_ret) \
+ } \
+} while(0)
+
+ // TODO: Check if cache file uses same block size as we do
+
+ if(is_new_cache_file==1) {
+ // New cache file, create needed xmount subdirectory
+ gidafs_ret=GidaFsLib_CreateDir(glob_xmount.cache.h_cache_file,
+ XMOUNT_CACHE_FOLDER,
+ eGidaFsNodeFlag_RWXu);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to create cache file directory '%s': Error code %u!\n",
+ XMOUNT_CACHE_FOLDER,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_CACHE;
+ return FALSE;
+ }
+ }
+
+ // Open / Create block cache file
+ gidafs_ret=GidaFsLib_OpenFile(glob_xmount.cache.h_cache_file,
+ XMOUNT_CACHE_BLOCK_FILE,
+ &(glob_xmount.cache.h_block_cache),
+ eGidaFsOpenFileFlag_ReadWrite |
+ (is_new_cache_file==1 ?
+ eGidaFsOpenFileFlag_CreateAlways : 0),
+ eGidaFsNodeFlag_Rall |
+ eGidaFsNodeFlag_Wusr);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to open / create block cache file '%s': Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_CACHE;
+ return FALSE;
+ }
+
+ // Open / Create block cache index file
+ gidafs_ret=GidaFsLib_OpenFile(glob_xmount.cache.h_cache_file,
+ XMOUNT_CACHE_BLOCK_INDEX_FILE,
+ &(glob_xmount.cache.h_block_cache_index),
+ eGidaFsOpenFileFlag_ReadWrite |
+ (is_new_cache_file==1 ?
+ eGidaFsOpenFileFlag_CreateAlways : 0),
+ eGidaFsNodeFlag_Rall |
+ eGidaFsNodeFlag_Wusr);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to open / create block cache index file '%s': "
+ "Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ return FALSE;
+ }
+
+ // Calculate how many cache blocks are needed and how big the cache block
+ // index must be
+ glob_xmount.cache.block_cache_index_len=image_size/CACHE_BLOCK_SIZE;
+ if((image_size%CACHE_BLOCK_SIZE)!=0) {
+ glob_xmount.cache.block_cache_index_len++;
+ }
+
+ LOG_DEBUG("Cache blocks: %u (0x%04X) entries using %zd (0x%08zX) bytes\n",
+ glob_xmount.cache.block_cache_index_len,
+ glob_xmount.cache.block_cache_index_len,
+ glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex),
+ glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex))
+
+ // Prepare in-memory buffer for block cache index
+ XMOUNT_MALLOC(glob_xmount.cache.p_block_cache_index,
+ t_CacheFileBlockIndex*,
+ glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex))
+
+ if(is_new_cache_file==1) {
+ // Generate initial block cache index
+ for(uint64_t i=0;i<glob_xmount.cache.block_cache_index_len;i++) {
+ glob_xmount.cache.p_block_cache_index[i]=CACHE_BLOCK_FREE;
+ }
+ // Write initial block cache index to cache file
+ if(UpdateBlockCacheIndex(XMOUNT_BLOCK_CACHE_INVALID_INDEX,0)!=TRUE) {
+ LOG_ERROR("Unable to generate initial block cache index file '%s': "
+ "Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_FILE,
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ } else {
+ // Existing cache file, make sure block cache index has correct size
+ gidafs_ret=GidaFsLib_GetFileSize(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache_index,
+ &blockindex_size);
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to get block cache index file size: Error code %u!\n",
+ gidafs_ret)
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ if(blockindex_size%sizeof(t_CacheFileBlockIndex)!=0 ||
+ (blockindex_size/sizeof(t_CacheFileBlockIndex))!=
+ glob_xmount.cache.block_cache_index_len)
+ {
+ // TODO: Be more helpfull in error message
+ LOG_ERROR("Block cache index size is incorrect for given input image!\n")
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ // Read block cache index into memory
+ gidafs_ret=GidaFsLib_ReadFile(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache_index,
+ 0,
+ blockindex_size,
+ glob_xmount.cache.p_block_cache_index,
+ &read);
+ if(gidafs_ret!=eGidaFsError_None || read!=blockindex_size) {
+ LOG_ERROR("Unable to read block cache index: Error code %u!\n",
+ gidafs_ret);
+ INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX;
+ INITCACHEFILE__CLOSE_BLOCK_CACHE;
+ INITCACHEFILE__CLOSE_CACHE;
+ XMOUNT_FREE(glob_xmount.cache.p_block_cache_index);
+ return FALSE;
+ }
+ }
+
+#undef INITCACHEFILE__CLOSE_BLOCK_CACHE_INDEX
+#undef INITCACHEFILE__CLOSE_BLOCK_CACHE
+#undef INITCACHEFILE__CLOSE_CACHE
+
+ return TRUE;
+}
+
+//! Update block cache index
+/*!
+ * \return TRUE on success, FALSE on error
+ */
+int UpdateBlockCacheIndex(uint64_t index, t_CacheFileBlockIndex value) {
+ uint64_t update_size=0;
+ uint64_t written=0;
+ teGidaFsError gidafs_ret=eGidaFsError_None;
+ t_CacheFileBlockIndex *p_buf;
+
+ if(index!=XMOUNT_BLOCK_CACHE_INVALID_INDEX) {
+ // Update specific index element in cache file
+ p_buf=glob_xmount.cache.p_block_cache_index+index;
+ update_size=sizeof(t_CacheFileBlockIndex);
+ } else {
+ // Dump whole block cache index to cache file
+ p_buf=glob_xmount.cache.p_block_cache_index;
+ update_size=glob_xmount.cache.block_cache_index_len*
+ sizeof(t_CacheFileBlockIndex);
+ }
+
+ // Update cache file
+ gidafs_ret=GidaFsLib_WriteFile(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache_index,
+ 0,
+ update_size,
+ p_buf,
+ &written);
+ if(gidafs_ret!=eGidaFsError_None || written!=update_size) {
+ LOG_ERROR("Unable to update block cache index file '%s': "
+ "Error code %u!\n",
+ XMOUNT_CACHE_BLOCK_INDEX_FILE,
+ gidafs_ret)
+ return FALSE;
+ }
+
+ return TRUE;
+}
diff --git a/src/xmount_cache.h b/src/xmount_cache.h
new file mode 100644
index 0000000..cfc318a
--- /dev/null
+++ b/src/xmount_cache.h
@@ -0,0 +1,176 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#ifndef XMOUNT_CACHE_H
+#define XMOUNT_CACHE_H
+
+#include <gidafs.h>
+
+/*******************************************************************************
+ * Public definitions / macros
+ ******************************************************************************/
+//! Default block cache block size to use (1 megabyte)
+#define XMOUNT_CACHE_BLOCK_SIZE (1024*1024)
+
+/*******************************************************************************
+ * Public types / structures / enums
+ ******************************************************************************/
+typedef struct s_XmountCacheHandle *pts_XmountCacheHandle;
+
+typedef enum e_XmountCache_Error {
+ //! No error
+ e_XmountCache_Error_None=0,
+ //! Error to allocate memory
+ e_XmountCache_Error_Alloc,
+ //! Invalid cache handle
+ e_XmountCache_Error_InvalidHandle,
+ //! Invalid pointer to a cache handle
+ e_XmountCache_Error_InvalidHandlePointer,
+ //! A given string is invalid
+ e_XmountCache_Error_InvalidString,
+ //! A given file path / name is invalid
+ e_XmountCache_Error_InvalidFile,
+ //! A given file does not exist
+ e_XmountCache_Error_InexistingFile,
+ //! A given file exists
+ e_XmountCache_Error_ExistingFile,
+ //! Unable to create needed xmount structures inside cache file
+ e_XmountCache_Error_FailedCacheInit,
+ //! Unable to open xmount cache file
+ e_XmountCache_Error_FailedOpeningCache,
+} te_XmountCache_Error;
+
+/*******************************************************************************
+ * Public functions declarations
+ ******************************************************************************/
+/*!
+ * \brief Create new xmount cache file
+ *
+ * Creates a new xmount cache file in the given file. If the given file already
+ * exists, this function will fail except if overwrite is set to 1.
+ *
+ * \param pp_handle Pointer to an xmount cache handle
+ * \param p_file File to use as cache file
+ * \param image_size Size of image in bytes for which this cache will be used
+ * \param overwrite If set to 1, overwrites existig cache file
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_Create(pts_XmountCacheHandle *pp_h,
+ const char *p_file,
+ uint64_t image_size,
+ uint8_t overwrite);
+
+/*!
+ * \brief Open an existing xmount cache file
+ *
+ * Opens the given xmount cache file.
+ *
+ * \param pp_handle Pointer to an xmount cache handle
+ * \param p_file File to use as cache file
+ * \param image_size Size of image in bytes for which this cache will be used
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_Open(pts_XmountCacheHandle *pp_h,
+ const char *p_file,
+ uint64_t image_size);
+
+/*!
+ * \brief Closes a previously openend xmount cache file
+ *
+ * Closes the given xmount cache file and frees any used resources.
+ *
+ * \param pp_handle Pointer to an xmount cache handle
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_Close(pts_XmountCacheHandle *pp_h);
+
+/*!
+ * \brief Read data from block cache
+ *
+ * Reads count bytes at offset block_offset from block number block and writes
+ * the read data into the pre-allocated buffer p_buf. The given block has to
+ * have been previously cached by a call to XmountCache_BlockCacheAppend().
+ *
+ * WARNING: This function does only work on single blocks. It is not possible to
+ * read beyond a block end.
+ *
+ * \param p_handle Xmount cache handle
+ * \param p_buf Buffer to store read data into
+ * \param block Number of block to read data from
+ * \param block_offset Offset inside block to start reading from
+ * \param count Amount of bytes to read
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_BlockCacheRead(pts_XmountCacheHandle p_h,
+ char *p_buf,
+ uint64_t block,
+ uint64_t block_offset,
+ uint64_t count);
+
+/*!
+ * \brief Write data to block cache
+ *
+ * Writes count bytes from buffer p_buf to block number block at offset
+ * block_offset. The given block has to have been previously cached by a call to
+ * XmountCache_BlockCacheAppend().
+ *
+ * WARNING: This function does only work on single blocks. It is not possible to
+ * write beyond a block end.
+ *
+ * \param p_handle Xmount cache handle
+ * \param p_buf Buffer with data to write
+ * \param block Number of block to write data to
+ * \param block_offset Offset inside block to start writing from
+ * \param count Amount of bytes to write
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_BlockCacheWrite(pts_XmountCacheHandle p_h,
+ char *p_buf,
+ uint64_t block,
+ uint64_t block_offset,
+ uint64_t count);
+
+/*!
+ * \brief Add a new block to the cache
+ *
+ * Adds the data inside p_buf to the cache file and saves it under the block
+ * number block. Every block must contain exactly XMOUNT_CACHE_BLOCK_SIZE bytes.
+ * Every block can only be cached once. Appending the same block twice will
+ * fail.
+ *
+ * \param p_handle Xmount cache handle
+ * \param p_buf Buffer with block data
+ * \param block Number of block under which to save given data
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_BlockCacheAppend(pts_XmountCacheHandle p_h,
+ char *p_buf,
+ uint64_t block);
+
+/*!
+ * \brief Chech if a block has previously been chached
+ *
+ * Checks if the given block has previously been cached.
+ *
+ * \param p_handle Xmount cache handle
+ * \param block Number of block to check
+ * \return e_XmountCache_Error_None on success
+ */
+te_XmountCache_Error XmountCache_IsBlockCached(pts_XmountCacheHandle p_h,
+ uint64_t block);
+
+#endif // XMOUNT_CACHE_H
diff --git a/src/xmount_input.c b/src/xmount_input.c
new file mode 100644
index 0000000..66fa81f
--- /dev/null
+++ b/src/xmount_input.c
@@ -0,0 +1,95 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#include <errno.h>
+
+#include "xmount_input.h"
+#include "xmount.h"
+
+#define LOG_WARNING(...) { \
+ LIBXMOUNT_LOG_WARNING(__VA_ARGS__); \
+}
+#define LOG_ERROR(...) { \
+ LIBXMOUNT_LOG_ERROR(__VA_ARGS__); \
+}
+#define LOG_DEBUG(...) { \
+ LIBXMOUNT_LOG_DEBUG(glob_xmount.debug,__VA_ARGS__); \
+}
+
+//! Read data from input image
+/*!
+ * \param p_image Image from which to read data
+ * \param p_buf Pointer to buffer to write read data to (must be preallocated!)
+ * \param offset Offset at which data should be read
+ * \param size Size of data which should be read (size of buffer)
+ * \param p_read Number of read bytes on success
+ * \return 0 on success, negated error code on error
+ */
+int ReadInputImageData(pts_InputImage p_image,
+ char *p_buf,
+ off_t offset,
+ size_t size,
+ size_t *p_read)
+{
+ int ret;
+ size_t to_read=0;
+ int read_errno=0;
+
+ LOG_DEBUG("Reading %zu bytes at offset %zu from input image '%s'\n",
+ size,
+ offset,
+ p_image->pp_files[0]);
+
+ // Make sure we aren't reading past EOF of image file
+ if(offset>=p_image->size) {
+ // Offset is beyond image size
+ LOG_DEBUG("Offset %zu is at / beyond size of input image '%s'\n",
+ offset,
+ p_image->pp_files[0]);
+ *p_read=0;
+ return 0;
+ }
+ if(offset+size>p_image->size) {
+ // Attempt to read data past EOF of image file
+ to_read=p_image->size-offset;
+ LOG_DEBUG("Attempt to read data past EOF of input image '%s'. "
+ "Correcting size from %zu to %zu\n",
+ p_image->pp_files[0],
+ size,
+ to_read);
+ } else to_read=size;
+
+ // Read data from image file (adding input image offset if one was specified)
+ ret=p_image->p_functions->Read(p_image->p_handle,
+ p_buf,
+ offset+glob_xmount.input.image_offset,
+ to_read,
+ p_read,
+ &read_errno);
+ if(ret!=0) {
+ LOG_ERROR("Couldn't read %zu bytes at offset %zu from input image "
+ "'%s': %s!\n",
+ to_read,
+ offset,
+ p_image->pp_files[0],
+ p_image->p_functions->GetErrorMessage(ret));
+ if(read_errno==0) return -EIO;
+ else return (read_errno*(-1));
+ }
+
+ return 0;
+}
diff --git a/src/xmount_input.h b/src/xmount_input.h
new file mode 100644
index 0000000..1d23d43
--- /dev/null
+++ b/src/xmount_input.h
@@ -0,0 +1,76 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#ifndef XMOUNT_INPUT_H
+#define XMOUNT_INPUT_H
+
+#include "../libxmount_input/libxmount_input.h"
+
+//! Structure containing infos about input libs
+typedef struct s_InputLib {
+ //! Filename of lib (without path)
+ char *p_name;
+ //! Handle to the loaded lib
+ void *p_lib;
+ //! Array of supported input types
+ char *p_supported_input_types;
+ //! Struct containing lib functions
+ ts_LibXmountInputFunctions lib_functions;
+} ts_InputLib, *pts_InputLib;
+
+//! Structure containing infos about input images
+typedef struct s_InputImage {
+ //! Image type
+ char *p_type;
+ //! Image source file count
+ uint64_t files_count;
+ //! Image source files
+ char **pp_files;
+ //! Input lib functions for this image
+ pts_LibXmountInputFunctions p_functions;
+ //! Image handle
+ void *p_handle;
+ //! Image size
+ uint64_t size;
+} ts_InputImage, *pts_InputImage;
+
+typedef struct s_InputData {
+ //! Loaded input lib count
+ uint32_t libs_count;
+ //! Array containing infos about loaded input libs
+ pts_InputLib *pp_libs;
+ //! Amount of input lib params (--inopts)
+ uint32_t lib_params_count;
+ //! Input lib params (--inopts)
+ pts_LibXmountOptions *pp_lib_params;
+ //! Input image count
+ uint64_t images_count;
+ //! Input images
+ pts_InputImage *pp_images;
+ //! Input image offset (--offset)
+ uint64_t image_offset;
+ //! Input image size limit (--sizelimit)
+ uint64_t image_size_limit;
+ //! MD5 hash of partial input image (lower 64 bit) (after morph)
+ uint64_t image_hash_lo;
+ //! MD5 hash of partial input image (higher 64 bit) (after morph)
+ uint64_t image_hash_hi;
+} ts_InputData;
+
+int ReadInputImageData(pts_InputImage, char*, off_t, size_t, size_t*);
+
+#endif // XMOUNT_INPUT_H
diff --git a/src/xmount_morphing.c b/src/xmount_morphing.c
new file mode 100644
index 0000000..bcc5a44
--- /dev/null
+++ b/src/xmount_morphing.c
@@ -0,0 +1,324 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#include <errno.h>
+
+#include "xmount_morphing.h"
+#include "xmount.h"
+#include "macros.h"
+
+#define LOG_WARNING(...) { \
+ LIBXMOUNT_LOG_WARNING(__VA_ARGS__); \
+}
+#define LOG_ERROR(...) { \
+ LIBXMOUNT_LOG_ERROR(__VA_ARGS__); \
+}
+#define LOG_DEBUG(...) { \
+ LIBXMOUNT_LOG_DEBUG(glob_xmount.debug,__VA_ARGS__); \
+}
+
+//! Get size of morphed image
+/*!
+ * \param p_size Buf to save size to
+ * \return TRUE on success, FALSE on error
+ */
+int GetMorphedImageSize(uint64_t *p_size) {
+ int ret;
+
+ ret=glob_xmount.morphing.p_functions->Size(glob_xmount.morphing.p_handle,
+ p_size);
+ if(ret!=0) {
+ LOG_ERROR("Unable to get morphed image size: %s!\n",
+ glob_xmount.morphing.p_functions->GetErrorMessage(ret));
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+//! Read data from morphed image
+/*!
+ * \param p_buf Pointer to buffer to write read data to (must be preallocated!)
+ * \param offset Offset at which data should be read
+ * \param size Size of data which should be read (size of buffer)
+ * \param p_read Number of read bytes on success
+ * \return TRUE on success, negated error code on error
+ */
+int ReadMorphedImageData(char *p_buf,
+ off_t offset,
+ size_t size,
+ size_t *p_read)
+{
+ uint64_t block_off=0;
+ uint64_t cur_block=0;
+ uint64_t cur_to_read=0;
+ uint64_t image_size=0;
+ size_t read=0;
+ size_t to_read=0;
+ int ret;
+ teGidaFsError gidafs_ret=eGidaFsError_None;
+
+ // Make sure we aren't reading past EOF of image file
+ if(GetMorphedImageSize(&image_size)!=TRUE) {
+ LOG_ERROR("Couldn't get size of morphed image!\n");
+ return -EIO;
+ }
+ if(offset>=image_size) {
+ // Offset is beyond image size
+ LOG_DEBUG("Offset %zu is at / beyond size of morphed image.\n",offset);
+ *p_read=0;
+ return FALSE;
+ }
+ if(offset+size>image_size) {
+ // Attempt to read data past EOF of morphed image file
+ to_read=image_size-offset;
+ LOG_DEBUG("Attempt to read data past EOF of morphed image. Corrected size "
+ "from %zu to %" PRIu64 ".\n",
+ size,
+ to_read);
+ } else to_read=size;
+
+ // Calculate block to start reading data from
+ cur_block=offset/CACHE_BLOCK_SIZE;
+ block_off=offset%CACHE_BLOCK_SIZE;
+
+ // Read image data
+ while(to_read!=0) {
+ // Calculate how many bytes we have to read from this block
+ if(block_off+to_read>CACHE_BLOCK_SIZE) {
+ cur_to_read=CACHE_BLOCK_SIZE-block_off;
+ } else cur_to_read=to_read;
+
+ // Check if block is cached
+ if(glob_xmount.output.writable==TRUE &&
+ glob_xmount.cache.p_block_cache_index[cur_block]!=CACHE_BLOCK_FREE)
+ {
+ // Write support enabled and need to read altered data from cachefile
+ LOG_DEBUG("Reading %zu bytes at offset %" PRIu64
+ " from block cache file\n",
+ cur_to_read,
+ glob_xmount.cache.p_block_cache_index[cur_block]+block_off)
+
+ gidafs_ret=GidaFsLib_ReadFile(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache,
+ glob_xmount.cache.
+ p_block_cache_index[cur_block]+block_off,
+ cur_to_read,
+ p_buf,
+ &read);
+ if(gidafs_ret!=eGidaFsError_None || read!=cur_to_read) {
+ LOG_ERROR("Unable to read cached data from block %" PRIu64
+ ": Error code %u!\n",
+ cur_block,
+ gidafs_ret);
+ return -EIO;
+ }
+ } else {
+ // No write support or data not cached
+ ret=glob_xmount.morphing.p_functions->Read(glob_xmount.morphing.p_handle,
+ p_buf,
+ (cur_block*CACHE_BLOCK_SIZE)+
+ block_off,
+ cur_to_read,
+ &read);
+ if(ret!=0 || read!=cur_to_read) {
+ LOG_ERROR("Couldn't read %zu bytes at offset %zu from morphed image: "
+ "%s!\n",
+ cur_to_read,
+ offset,
+ glob_xmount.morphing.p_functions->GetErrorMessage(ret));
+ return -EIO;
+ }
+ LOG_DEBUG("Read %" PRIu64 " bytes at offset %" PRIu64
+ " from morphed image file\n",
+ cur_to_read,
+ (cur_block*CACHE_BLOCK_SIZE)+block_off);
+ }
+
+ cur_block++;
+ block_off=0;
+ p_buf+=cur_to_read;
+ to_read-=cur_to_read;
+ }
+
+ *p_read=to_read;
+ return TRUE;
+}
+
+//! Write data to morphed image
+/*!
+ * \param p_buf Buffer with data to write
+ * \param offset Offset to start writing at
+ * \param count Amount of bytes to write
+ * \param p_written Amount of successfully written bytes
+ * \return TRUE on success, negated error code on error
+ */
+int WriteMorphedImageData(const char *p_buf,
+ off_t offset,
+ size_t count,
+ size_t *p_written)
+{
+ uint64_t block_off=0;
+ uint64_t cur_block=0;
+ uint64_t cur_to_read=0;
+ uint64_t cur_to_write=0;
+ uint64_t image_size=0;
+ uint64_t read=0;
+ size_t written=0;
+ size_t to_write=0;
+ int ret;
+ teGidaFsError gidafs_ret=eGidaFsError_None;
+ char *p_buf2=NULL;
+
+ // Make sure we aren't writing past EOF of image file
+ if(GetMorphedImageSize(&image_size)!=TRUE) {
+ LOG_ERROR("Couldn't get size of morphed image!\n");
+ return -EIO;
+ }
+ if(offset>=image_size) {
+ // Offset is beyond image size
+ LOG_DEBUG("Offset %zu is at / beyond size of morphed image.\n",offset);
+ *p_written=0;
+ return 0;
+ }
+ if(offset+count>image_size) {
+ // Attempt to write data past EOF of morphed image file
+ to_write=image_size-offset;
+ LOG_DEBUG("Attempt to write data past EOF of morphed image. Corrected size "
+ "from %zu to %" PRIu64 ".\n",
+ count,
+ to_write);
+ } else to_write=count;
+
+ // Calculate block to start writing data to
+ cur_block=offset/CACHE_BLOCK_SIZE;
+ block_off=offset%CACHE_BLOCK_SIZE;
+
+ while(to_write!=0) {
+ // Calculate how many bytes we have to write to this block
+ if(block_off+to_write>CACHE_BLOCK_SIZE) {
+ cur_to_write=CACHE_BLOCK_SIZE-block_off;
+ } else cur_to_write=to_write;
+
+ // Check if block is cached
+ if(glob_xmount.cache.p_block_cache_index[cur_block]!=CACHE_BLOCK_FREE) {
+ // Block is cached
+ gidafs_ret=GidaFsLib_WriteFile(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache,
+ glob_xmount.cache.
+ p_block_cache_index[cur_block]+block_off,
+ cur_to_write,
+ p_buf,
+ &written);
+ if(gidafs_ret!=eGidaFsError_None || written!=cur_to_write) {
+ LOG_ERROR("Unable to write data to cached block %" PRIu64
+ ": Error code %u!\n",
+ cur_block,
+ gidafs_ret);
+ return -EIO;
+ }
+
+ LOG_DEBUG("Wrote %" PRIu64 " bytes at offset %" PRIu64
+ " to block cache file\n",
+ cur_to_write,
+ glob_xmount.cache.p_block_cache_index[cur_block]+block_off);
+ } else {
+ // Uncached block. Need to cache entire new block
+ // Prepare new write buffer
+ XMOUNT_MALLOC(p_buf2,char*,CACHE_BLOCK_SIZE);
+ memset(p_buf2,0x00,CACHE_BLOCK_SIZE);
+
+ // Read full block from morphed image
+ cur_to_read=CACHE_BLOCK_SIZE;
+ if((cur_block*CACHE_BLOCK_SIZE)+CACHE_BLOCK_SIZE>image_size) {
+ cur_to_read=CACHE_BLOCK_SIZE-(((cur_block*CACHE_BLOCK_SIZE)+
+ CACHE_BLOCK_SIZE)-image_size);
+ }
+ ret=glob_xmount.morphing.p_functions->Read(glob_xmount.morphing.p_handle,
+ p_buf2,
+ cur_block*CACHE_BLOCK_SIZE,
+ cur_to_read,
+ &read);
+ if(ret!=0 || read!=cur_to_read) {
+ LOG_ERROR("Couldn't read %" PRIu64 " bytes at offset %zu "
+ "from morphed image: %s!\n",
+ cur_to_read,
+ offset,
+ glob_xmount.morphing.p_functions->GetErrorMessage(ret));
+ XMOUNT_FREE(p_buf2);
+ return -EIO;
+ }
+
+ // Set changed data
+ memcpy(p_buf2+block_off,p_buf,cur_to_write);
+
+ // Write new block to block cache
+ // Get current block cache size
+ gidafs_ret=GidaFsLib_GetFileSize(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache,
+ &(glob_xmount.cache.
+ p_block_cache_index[cur_block]));
+ if(gidafs_ret!=eGidaFsError_None) {
+ LOG_ERROR("Unable to get current block cache size: Error code %u!\n",
+ gidafs_ret);
+ XMOUNT_FREE(p_buf2);
+ return -EIO;
+ }
+ // Append new block
+ gidafs_ret=GidaFsLib_WriteFile(glob_xmount.cache.h_cache_file,
+ glob_xmount.cache.h_block_cache,
+ glob_xmount.cache.
+ p_block_cache_index[cur_block],
+ CACHE_BLOCK_SIZE,
+ p_buf2,
+ &written);
+ if(gidafs_ret!=eGidaFsError_None || written!=cur_to_write) {
+ LOG_ERROR("Unable to write data to cached block %" PRIu64
+ ": Error code %u!\n",
+ cur_block,
+ gidafs_ret);
+ XMOUNT_FREE(p_buf2);
+ return -EIO;
+ }
+ XMOUNT_FREE(p_buf2);
+ // Update on-disk block cache index
+ ret=UpdateBlockCacheIndex(cur_block,
+ glob_xmount.cache.p_block_cache_index[
+ cur_block]);
+ if(ret!=TRUE) {
+ LOG_ERROR("Unable to update block cache index %" PRIu64
+ ": Error code %u!\n",
+ cur_block,
+ gidafs_ret);
+ return -EIO;
+ }
+
+ LOG_DEBUG("Updated cache file block index: Number=%" PRIu64
+ ", Data offset=%" PRIu64 "\n",
+ cur_block,
+ glob_xmount.cache.p_block_cache_index[cur_block]);
+ }
+
+ block_off=0;
+ cur_block++;
+ p_buf+=cur_to_write;
+ to_write-=cur_to_write;
+ }
+
+ *p_written=to_write;
+ return TRUE;
+}
diff --git a/src/xmount_morphing.h b/src/xmount_morphing.h
new file mode 100644
index 0000000..d0c0e10
--- /dev/null
+++ b/src/xmount_morphing.h
@@ -0,0 +1,59 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#ifndef XMOUNT_MORPHING_H
+#define XMOUNT_MORPHING_H
+
+#include "../libxmount_morphing/libxmount_morphing.h"
+
+//! Structure containing infos about morphing libs
+typedef struct s_MorphingLib {
+ //! Filename of lib (without path)
+ char *p_name;
+ //! Handle to the loaded lib
+ void *p_lib;
+ //! Array of supported morphing types
+ char *p_supported_morphing_types;
+ //! Struct containing lib functions
+ ts_LibXmountMorphingFunctions lib_functions;
+} ts_MorphingLib, *pts_MorphingLib;
+
+//! Structures and vars needed for morph support
+typedef struct s_MorphingData {
+ //! Loaded morphing lib count
+ uint32_t libs_count;
+ //! Array containing infos about loaded morphing libs
+ pts_MorphingLib *pp_libs;
+ //! Specified morphing type (--morph)
+ char *p_morph_type;
+ //! Amount of specified morphing lib params (--morphopts)
+ uint32_t lib_params_count;
+ //! Specified morphing lib params (--morphopts)
+ pts_LibXmountOptions *pp_lib_params;
+ //! Handle to initialized morphing lib
+ void *p_handle;
+ //! Morphing functions of initialized lib
+ pts_LibXmountMorphingFunctions p_functions;
+ //! Input image functions passed to morphing lib
+ ts_LibXmountMorphingInputFunctions input_image_functions;
+} ts_MorphingData;
+
+int GetMorphedImageSize(uint64_t*);
+int ReadMorphedImageData(char*, off_t, size_t, size_t*);
+int WriteMorphedImageData(const char*, off_t, size_t, size_t*);
+
+#endif // XMOUNT_MORPHING_H
diff --git a/src/xmount_output.c b/src/xmount_output.c
new file mode 100644
index 0000000..b389258
--- /dev/null
+++ b/src/xmount_output.c
@@ -0,0 +1,155 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#include <errno.h>
+
+#include "xmount_output.h"
+#include "xmount.h"
+
+#define LOG_WARNING(...) { \
+ LIBXMOUNT_LOG_WARNING(__VA_ARGS__); \
+}
+#define LOG_ERROR(...) { \
+ LIBXMOUNT_LOG_ERROR(__VA_ARGS__); \
+}
+#define LOG_DEBUG(...) { \
+ LIBXMOUNT_LOG_DEBUG(glob_xmount.debug,__VA_ARGS__); \
+}
+
+//! Get size of output image
+/*!
+ * \param p_size Pointer to an uint64_t to which the size will be written to
+ * \return TRUE on success, FALSE on error
+ */
+int GetOutputImageSize(uint64_t *p_size) {
+ int ret;
+ uint64_t output_image_size=0;
+
+ if(glob_xmount.output.image_size!=0) {
+ *p_size=glob_xmount.output.image_size;
+ return TRUE;
+ }
+
+ ret=glob_xmount.output.p_functions->Size(glob_xmount.output.p_handle,
+ &output_image_size);
+ if(ret!=0) {
+ LOG_ERROR("Couldn't get output image size!\n")
+ return FALSE;
+ }
+
+ glob_xmount.output.image_size=output_image_size;
+ *p_size=output_image_size;
+ return TRUE;
+}
+
+//! Read data from output image
+/*!
+ * \param p_buf Pointer to buffer to write read data to
+ * \param offset Offset at which data should be read
+ * \param size Size of data which should be read
+ * \return Number of read bytes on success or negated error code on error
+ */
+int ReadOutputImageData(char *p_buf, off_t offset, size_t size) {
+ uint64_t output_image_size;
+ size_t read=0;
+ int ret;
+
+ // Get output image size
+ if(GetOutputImageSize(&output_image_size)!=TRUE) {
+ LOG_ERROR("Couldn't get size of output image!\n")
+ return -EIO;
+ }
+
+ // Make sure request is within output image
+ if(offset>=output_image_size) {
+ LOG_DEBUG("Offset %zu is at / beyond size of output image.\n",offset);
+ return 0;
+ }
+ if(offset+size>output_image_size) {
+ LOG_DEBUG("Attempt to read data past EOF of output image. Correcting size "
+ "from %zu to %zu.\n",
+ size,
+ output_image_size-offset);
+ size=output_image_size-offset;
+ }
+
+ // Read data
+ ret=glob_xmount.output.p_functions->Read(glob_xmount.output.p_handle,
+ p_buf,
+ offset,
+ size,
+ &read);
+ if(ret!=0) {
+ LOG_ERROR("Unable to read %zu bytes at offset %zu from output image!\n",
+ size,
+ offset)
+ return ret;
+ } else if(read!=size) {
+ LOG_WARNING("Unable to read all requested data from output image!\n")
+ return read;
+ }
+
+ return size;
+}
+
+//! Write data to output image
+/*!
+ * \param p_buf Buffer with data to write
+ * \param offset Offset to write to
+ * \param size Amount of bytes to write
+ * \return Number of written bytes on success or "-1" on error
+ */
+int WriteOutputImageData(const char *p_buf, off_t offset, size_t size) {
+ uint64_t output_image_size;
+ int ret;
+ size_t written;
+
+ // Get output image size
+ if(!GetOutputImageSize(&output_image_size)) {
+ LOG_ERROR("Couldn't get output image size!\n")
+ return -1;
+ }
+
+ // Make sure write is within output image
+ if(offset>=output_image_size) {
+ LOG_ERROR("Attempt to write beyond EOF of output image file!\n")
+ return -1;
+ }
+ if(offset+size>output_image_size) {
+ LOG_DEBUG("Attempt to write past EOF of output image file. Correcting size "
+ "from %zu to %zu.\n",
+ size,
+ output_image_size-offset);
+ size=output_image_size-offset;
+ }
+
+ ret=glob_xmount.output.p_functions->Write(glob_xmount.output.p_handle,
+ p_buf,
+ offset,
+ size,
+ &written);
+ if(ret!=0) {
+ LOG_ERROR("Unable to write %zu bytes at offset %zu to output image!\n",
+ offset,
+ size)
+ return ret;
+ } else if(written!=size) {
+ LOG_WARNING("Unable to write all requested data to output image!\n")
+ }
+
+ return size;
+}
diff --git a/src/xmount_output.h b/src/xmount_output.h
new file mode 100644
index 0000000..ca4f67e
--- /dev/null
+++ b/src/xmount_output.h
@@ -0,0 +1,69 @@
+/*******************************************************************************
+* xmount Copyright (c) 2008-2016 by Gillen Daniel <gillen.dan@pinguin.lu> *
+* *
+* This program is free software: you can redistribute it and/or modify it *
+* under the terms of the GNU General Public License as published by the Free *
+* Software Foundation, either version 3 of the License, or (at your option) *
+* any later version. *
+* *
+* This program is distributed in the hope that it will be useful, but WITHOUT *
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or *
+* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for *
+* more details. *
+* *
+* You should have received a copy of the GNU General Public License along with *
+* this program. If not, see <http://www.gnu.org/licenses/>. *
+*******************************************************************************/
+
+#ifndef XMOUNT_OUTPUT_H
+#define XMOUNT_OUTPUT_H
+
+#include "../libxmount_output/libxmount_output.h"
+
+//! Structure containing infos about output libs
+typedef struct s_OutputLib {
+ //! Filename of lib (without path)
+ char *p_name;
+ //! Handle to the loaded lib
+ void *p_lib;
+ //! Array of supported output formats
+ char *p_supported_output_formats;
+ //! Struct containing lib functions
+ ts_LibXmountOutput_Functions lib_functions;
+} ts_OutputLib, *pts_OutputLib;
+
+//! Structure containing infos about output image
+typedef struct s_OutputData {
+ //! Loaded output lib count
+ uint32_t libs_count;
+ //! Array containing infos about loaded output libs
+ pts_OutputLib *pp_libs;
+ //! Specified output format (--out)
+ char *p_output_format;
+ //! Amount of specified output lib params
+ uint32_t lib_params_count;
+ //! Specified output lib params (--outopts)
+ pts_LibXmountOptions *pp_lib_params;
+ //! Handle to initialized output lib
+ void *p_handle;
+ //! Transformation functions of initialized lib
+ pts_LibXmountOutput_Functions p_functions;
+ //! Input image functions passed to output lib
+ ts_LibXmountOutput_InputFunctions input_functions;
+ //! Size
+ uint64_t image_size;
+ //! Writable? (Set to 1 if --cache was specified)
+ uint8_t writable;
+ //! Path of virtual image file
+ char *p_virtual_image_path;
+ //! Path of virtual image info file
+ char *p_info_path;
+ //! Pointer to virtual info file
+ char *p_info_file;
+} ts_OutputData;
+
+int GetOutputImageSize(uint64_t*);
+int ReadOutputImageData(char*, off_t, size_t);
+int WriteOutputImageData(const char*, off_t, size_t);
+
+#endif // XMOUNT_OUTPUT_H
File Metadata
Details
Attached
Mime Type
text/x-diff
Expires
Mon, Dec 23, 10:56 AM (6 h, 24 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
1176900
Default Alt Text
(210 KB)
Attached To
Mode
rXMOUNT xmount
Attached
Detach File
Event Timeline
Log In to Comment