diff --git a/trunk/README b/trunk/README index cbd679f..ae1957c 100644 --- a/trunk/README +++ b/trunk/README @@ -1,224 +1,224 @@ -XMOUNT README FILE v0.7.0 +XMOUNT README FILE v0.7.x Table of contents 0.0 Author and license stuff 1.0 What is xmount? - A short description 2.0 Input image support 2.1 libxmount_input_raw 2.2 libxmount_input_ewf 2.3 libxmount_input_aewf 2.4 libxmount_input_aff 2.5 libxmount_input_aaff 3.0 Morphing support 3.1 libxmount_morphing_combine 3.2 libxmount_morphing_raid 3.3 libxmount_morphing_unallocated 4.0 Output image 4.1 Emulated RAW file 4.2 Emulated DMG file 4.3 Emulated VDI file 4.4 Emulated VMDK file 4.5 Emulated VHD file 5.0 Virtual write access 6.0 Installation instructions 6.1 Prerequisits 6.1 Install from a package 6.2 Install from source 7.0 xmount command line parameters 7.1 xmount usage examples 0.0 Author and license stuff - xmount Copyright (c) 2008-2014 by Gillen Daniel + xmount Copyright (c) 2008-2015 by Gillen Daniel 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 . --- Due to AFFLIB, I have to mention that: This product includes software developed by Simson L. Garfinkel and Basis Technology Corp. 1.0 What is xmount? - A short description xmount allows you to convert on-the-fly between multiple input and output harddisk image formats. xmount creates a virtual file system using FUSE (Filesystem in Userspace) that contains a virtual representation of the input image. The virtual representation can be in raw DD, Apple's Disk Image format (DMG), VirtualBox's virtual disk file format (VDI), Microsoft's Virtual Hard Disk Image format (VHD) or in VmWare's VMDK file format. Input images can be raw DD, EWF (Expert Witness Compression Format) or AFF (Advanced Forensic Format) files. In addition, xmount also supports virtual write access to the output files that is redirected to a cache file. This makes it for example possible to boot acquired harddisk images using QEMU, KVM, VirtualBox, VmWare or alike. 2.0 Input image support Starting with version 0.7.0, input image support has been moved into external libraries. The following chapters list the officially supported ones. They might not all be available on your system though. 2.1 libxmount_input_raw Supports raw DD images ("--in raw" or "--in dd"). 2.2 libxmount_input_ewf Supports EWF (Expert Witness Compression Format) images ("--in ewf") using Joachim Metz's libewf (https://code.google.com/p/libewf/). 2.3 libxmount_input_aewf Supports EWF (Expert Witness Compression Format) images ("--in aewf") generated with Guymager (http://guymager.sourceforge.net/). This library uses an EWF implementation written by Guy Voncken. It consumes much less memory than libewf when mounting big (>1TB) images. 2.4 libxmount_input_aff Supports AFF (Advanced Forensic Format) images ("--in aff") using Simson Garfinkel's afflib (https://github.com/simsong/AFFLIBv3). 2.5 libxmount_input_aaff Supports AFF (Advanced Forensic Format) images ("--in aaff") using an AFF implementation written by Guy Voncken. In essence, it is a lot faster than afflib. 3.0 Morphing support Also starting with xmount version 0.7.0, a new concept of input image morphing has been added. Morphing is a process which is applied to the data of all specified input images and produces a single virtual image which can then be converted into an output image. The following chapters list the currently officially supported morphing libraries. As with the input image libs, not all might be available on your system. 3.1 libxmount_morphing_combine Using "--morph combine", data from all input images is concatenated into a single image. Images are appended in the same order they are specified on the command line. This morphing type is also the default if no "--morph" option is specified. 3.2 libxmount_morphing_raid This morphing library supports emulation of hardware / software raid. Currently, only RAID0 ("--morph raid0") is supported. The used chunk / stripe size in bytes can be specified with "--morphopts raid_chunksize=XXX". All input images are treated as member disks of the original raid. 3.3 libxmount_morphing_unallocated Using "--morph unallocated" it is possible to extract unallocated sectors from a filesystem. For this to work, your input image must start with the volume you want to extract unallocated sectors from. If you have an image of a whole disk, you might need to use xmounts "--offset" parameter to move the start of the data to the start of the volume. Currently, unallocated sectors from HFS and FAT can be extracted. 4.0 Output image 4.1 Emulated RAW file This is the standard data representation used by xmount. It is a file containing the raw data contained in the input image file. Except for OSx (see 2.2) this is the default output format when no --out option is specified. 4.2 Emulated DMG file This is also a file containing the raw data contained in the input image file but with a .dmg file extension. On OSx, this is the default output format when no --out option is specified. 4.3 Emulated VDI file When using the "--out vdi" command line parameter, a VirtualBox disk file is emulated. This means that xmount builds a virtual VDI header and prepends it to the raw data contained in the input image. The VDI file can be used in VirtualBox as a virtual disk file. 4.4 Emulated VMDK file When using the "--out vmdk(s)" command line paramter, a VMWare virtual disk file will be emulated. This mainly consists in building a VMDK descriptor file that will access an emulated RAW file. You can use this VMDK file to add your input image as virtual disk to a VMWare virtual machine. "vmdk" is a VMWare disk attached to the IDE bus and "vmdks" is attached to the SCSI bus. When emulating VMDK files to be used in VMWare, you have to enable "user_allow_other" in /etc/fuse.conf or run xmount and VmWare as root. WARNING: VMDK support should be considered experimental and currently the VMDK descriptor file isn't cached. 4.5 Emulated VHD file When using the "--out vhd" command line parameter, a Virtual Hard Disk Image file is emulated by appending to the raw data a valid VHD footer. Valid here means as described in Microsoft's Virtual Hard Disk Image Format specs v1. 5.0 Virtual write access By using the "--cache " command line parameter, xmount allows you to change data in the emulated image files. All changes are written to a separate cache file. No data will ever be written to the original input image files! Write access is limited to change existing data. It isn't possible to change the emulated image's file size (no append or truncate)! The same cache file can be used with different output image formats without loosing changed data. 6.0 Installation instructions To install xmount, you can use a prebuild package for your distribution or you can build xmount from source. The two methods are described in the following sections. 6.1 Prerequisits SYSTEM: - Linux: Some sort of Linux with a recent kernel (2.6.x or above). - OSx: OSx 10.6 or above FUSE: Your os must support FUSE and have the appropriate libraries installed. (See http://fuse.sourceforge.net/ for more informations). To be able to install from source, you also need the FUSE header files. On OSx, you can use osxfuse instead. LIBEWF: To enable EWF input image support. Get it from http://sourceforge.net/projects/libewf/ LIBAFF: To enable AFF input image support. Get it from http://www.afflib.org/ 6.2 Install from a package Chances are I provide prebuild binary packages for Debian / Ubuntu. In this case, you only have to fire up your distribution's package manager and install xmount. See https://www.pinguin.lu/xmount for more information and download links. 6.3 Install from source After having checked the prerequisits, you should be able to compile and install xmount by simply issuing the following commands: $ mkdir build $ cd build $ cmake -DCMAKE_BUILD_TYPE=Release .. $ make $ sudo make install 7.1 xmount usage examples To xmount an EWF harddisk image from your acquired disk as a raw DD image under /mnt, use the following command: xmount --in ewf ./acquired_disk.E?? /mnt To xmount the same ewf image as vdi file, you would use a command like this: xmount --in ewf ./acquired_disk.E?? --out vdi /mnt And to enable virtual write support on a raw DD input image xmounted as VDI file: xmount --in raw ./acquired_disk.dd --out vdi --cache ./disk.cache /mnt EOF diff --git a/trunk/config.h.in b/trunk/config.h.in index dc1d9fe..1afdbb4 100644 --- a/trunk/config.h.in +++ b/trunk/config.h.in @@ -1,47 +1,47 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef CONFIG_H #define CONFIG_H #cmakedefine HAVE_STDLIB_H 1 #cmakedefine HAVE_STDIO_H 1 #cmakedefine HAVE_STDINT_H 1 #cmakedefine HAVE_STDARG_H 1 #cmakedefine HAVE_STRING_H 1 #cmakedefine HAVE_ERRNO_H 1 #cmakedefine HAVE_FCNTL_H 1 #cmakedefine HAVE_DLFCN_H 1 #cmakedefine HAVE_DIRENT_H 1 #cmakedefine HAVE_UNISTD_H 1 #cmakedefine HAVE_SYS_IOCTL_H 1 #cmakedefine HAVE_SYS_TYPES_H 1 #cmakedefine HAVE_LINUX_FS_H 1 #cmakedefine HAVE_GRP_H 1 #cmakedefine HAVE_PWD_H 1 #cmakedefine HAVE_PTHREAD_H 1 #cmakedefine HAVE_TIME_H 1 #cmakedefine HAVE_INTTYPES_H 1 #cmakedefine HAVE_BYTESWAP_H 1 #cmakedefine HAVE_ENDIAN_H 1 #cmakedefine HAVE_LIBKERN_OSBYTEORDER_H 1 #endif // CONFIG_H diff --git a/trunk/libxmount/endianness.h b/trunk/libxmount/endianness.h index c69c717..824423b 100644 --- a/trunk/libxmount/endianness.h +++ b/trunk/libxmount/endianness.h @@ -1,178 +1,178 @@ /******************************************************************************* -* Copyright (c) 2008-2014 by Gillen Daniel * +* Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ /* * This file aims at making endianness conversions using the following * functions portable. * * Conversion from big endian to host / from host to big endian * be16toh(uint16_t) * be32toh(uint32_t) * be64toh(uint64_t) * htobe16(uint16_t) * htobe32(uint32_t) * htobe64(uint64_t) * * Conversion from little endian to host / from host to little endian * le16toh(uint16_t) * le32toh(uint32_t) * le64toh(uint64_t) * htole16(uint16_t) * htole32(uint32_t) * htole64(uint64_t) * * In addition, including this file will define HOST_BYTEORDER_IS_BE or * HOST_BYTEORDER_IS_LE according to the current byte order of the host. */ #ifndef ENDIANNESS_H #define ENDIANNESS_H #include #ifdef HAVE_STDINT_H #include #else #error "Including this file requires you to have stdint.h" #endif #ifdef HAVE_ENDIAN_H #include #endif // First we need to have the bswap functions #if defined(HAVE_BYTESWAP_H) #include #elif defined(__APPLE__) && defined(HAVE_LIBKERN_OSBYTEORDER_H) #include #define bswap_16 OSSwapInt16 #define bswap_32 OSSwapInt32 #define bswap_64 OSSwapInt64 #else #define bswap_16(value) { \ ((((value) & 0xff) << 8) | ((value) >> 8)) \ } #define bswap_32(value) { \ (((uint32_t)bswap_16((uint16_t)((value) & 0xffff)) << 16) | \ (uint32_t)bswap_16((uint16_t)((value) >> 16))) \ } #define bswap_64(value) { \ (((uint64_t)bswap_32((uint32_t)((value) & 0xffffffff)) << 32) | \ (uint64_t)bswap_32((uint32_t)((value) >> 32))) \ } #endif // Next we need to know what endianness is used #if defined(__LITTLE_ENDIAN__) #define HOST_BYTEORDER_IS_LE #elif defined(__BIG_ENDIAN__) #define HOST_BYTEORDER_IS_BE #elif defined(__BYTE_ORDER) #if __BYTE_ORDER == __LITTLE_ENDIAN #define HOST_BYTEORDER_IS_LE #else #define HOST_BYTEORDER_IS_BE #endif #elif defined(__BYTE_ORDER__) #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ #define HOST_BYTEORDER_IS_LE #else #define HOST_BYTEORDER_IS_BE #endif #else #error "Unable to determine host byteorder" #endif // And finally we can define the endianness conversion macros #ifdef HOST_BYTEORDER_IS_LE #ifndef be16toh #define be16toh(x) bswap_16(x) #endif #ifndef htobe16 #define htobe16(x) bswap_16(x) #endif #ifndef be32toh #define be32toh(x) bswap_32(x) #endif #ifndef htobe32 #define htobe32(x) bswap_32(x) #endif #ifndef be64toh #define be64toh(x) bswap_64(x) #endif #ifndef htobe64 #define htobe64(x) bswap_64(x) #endif #ifndef le16toh #define le16toh(x) (x) #endif #ifndef htole16 #define htole16(x) (x) #endif #ifndef le32toh #define le32toh(x) (x) #endif #ifndef htole32 #define htole32(x) (x) #endif #ifndef le64toh #define le64toh(x) (x) #endif #ifndef htole64 #define htole64(x) (x) #endif #else #ifndef be16toh #define be16toh(x) (x) #endif #ifndef htobe16 #define htobe16(x) (x) #endif #ifndef be32toh #define be32toh(x) (x) #endif #ifndef htobe32 #define htobe32(x) (x) #endif #ifndef be64toh #define be64toh(x) (x) #endif #ifndef htobe64 #define htobe64(x) (x) #endif #ifndef le16toh #define le16toh(x) bswap_16(x) #endif #ifndef htole16 #define htole16(x) bswap_16(x) #endif #ifndef le32toh #define le32toh(x) bswap_32(x) #endif #ifndef htole32 #define htole32(x) bswap_32(x) #endif #ifndef le64toh #define le64toh(x) bswap_64(x) #endif #ifndef htole64 #define htole64(x) bswap_64(x) #endif #endif #endif //ENDIANNESS_H diff --git a/trunk/libxmount/libxmount.c b/trunk/libxmount/libxmount.c index 3da9b8e..2fb48f5 100644 --- a/trunk/libxmount/libxmount.c +++ b/trunk/libxmount/libxmount.c @@ -1,126 +1,126 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include #include #include #include #include "libxmount.h" //! Print error and debug messages to stdout /*! * \param p_msg_type "ERROR" or "DEBUG" * \param p_calling_fun Name of calling function * \param line Line number of call * \param p_msg Message string * \param ... Variable params with values to include in message string */ /* * LogMessage */ void LogMessage(char *p_msg_type, char *p_calling_fun, int line, char *p_msg, ...) { va_list var_list; // Print message "header" printf("%s: %s@%u : ",p_msg_type,p_calling_fun,line); // Print message with variable parameters va_start(var_list,p_msg); vprintf(p_msg,var_list); va_end(var_list); } /* * StrToInt32 */ int32_t StrToInt32(const char *p_value, int *p_ok) { long int num; char *p_tail; errno=0; num=strtol(p_value,&p_tail,0); if(errno==ERANGE || *p_tail!='\0' || numINT32_MAX) { *p_ok=0; return 0; } *p_ok=1; return (int32_t)num; } /* * StrToUint32 */ uint32_t StrToUint32(const char *p_value, int *p_ok) { unsigned long int num; char *p_tail; errno=0; num=strtoul(p_value,&p_tail,0); if(errno==ERANGE || *p_tail!='\0' || num>UINT32_MAX) { *p_ok=0; return 0; } *p_ok=1; return (uint32_t)num; } /* * StrToInt64 */ int64_t StrToInt64(const char *p_value, int *p_ok) { long long int num; char *p_tail; errno=0; num=strtoll(p_value,&p_tail,0); if(errno==ERANGE || *p_tail!='\0' || numINT64_MAX) { *p_ok=0; return 0; } *p_ok=1; return (int64_t)num; } /* * StrToUint64 */ uint64_t StrToUint64(const char *p_value, int *p_ok) { unsigned long long int num; char *p_tail; errno=0; num=strtoull(p_value,&p_tail,0); if(errno==ERANGE || *p_tail!='\0' || num>UINT64_MAX) { *p_ok=0; return 0; } *p_ok=1; return (uint64_t)num; } diff --git a/trunk/libxmount/libxmount.h b/trunk/libxmount/libxmount.h index ba41213..8daadfa 100644 --- a/trunk/libxmount/libxmount.h +++ b/trunk/libxmount/libxmount.h @@ -1,74 +1,74 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_H #define LIBXMOUNT_H #include "endianness.h" /* * Under OSx, fopen handles 64bit I/O too */ #ifdef __APPLE__ #define fopen64 fopen #define fseeko64 fseeko #endif /* * Macros to ease debugging and error reporting */ #define LIBXMOUNT_LOG_ERROR(...) { \ LogMessage("ERROR",(char*)__FUNCTION__,__LINE__,__VA_ARGS__); \ } #define LIBXMOUNT_LOG_WARNING(...) { \ LogMessage("WARNING",(char*)__FUNCTION__,__LINE__,__VA_ARGS__); \ } #define LIBXMOUNT_LOG_DEBUG(debug,...) { \ if(debug) LogMessage("DEBUG",(char*)__FUNCTION__,__LINE__,__VA_ARGS__); \ } //! Struct containing lib options typedef struct s_LibXmountOptions { //! Option name char *p_key; //! Option value char *p_value; //! Set to 1 if key/value has been parsed and is valid uint8_t valid; } ts_LibXmountOptions, *pts_LibXmountOptions; //! Log messages /*! * \param p_msg_type "ERROR", "DEBUG", etc... * \param p_calling_fun Name of calling function * \param line Line number of call * \param p_msg Message string * \param ... Variable params with values to include in message string */ void LogMessage(char *p_msg_type, char *p_calling_fun, int line, char *p_msg, ...); int32_t StrToInt32(const char *p_value, int *p_ok); uint32_t StrToUint32(const char *p_value, int *p_ok); int64_t StrToInt64(const char *p_value, int *p_ok); uint64_t StrToUint64(const char *p_value, int *p_ok); #endif // LIBXMOUNT_H diff --git a/trunk/libxmount_input/libxmount_input.h b/trunk/libxmount_input/libxmount_input.h index c04893a..791eab1 100644 --- a/trunk/libxmount_input/libxmount_input.h +++ b/trunk/libxmount_input/libxmount_input.h @@ -1,226 +1,226 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_INPUT_H #define LIBXMOUNT_INPUT_H #define LIBXMOUNT_INPUT_API_VERSION 1 #include #include // For int*_t and uint*_t #include // For PRI* #include // For off_t #include "../libxmount/libxmount.h" //! Structure containing pointers to the lib's functions typedef struct s_LibXmountInputFunctions { //! Function to initialize handle /*! * This function is called once to allow the lib to alloc any needed * structures before other functions that rely upon a valid handle are called * (for ex. OptionsParse or Open). * * The p_format parameter specifies one of the formats returned by * LibXmount_Input_GetSupportedFormats() which should be used for this handle. * * \param pp_handle Pointer to handle * \param p_format Input image format * \param debug If set to 1, print debugging infos to stdout * \return 0 on success or error code */ int (*CreateHandle)(void **pp_handle, const char *p_format, uint8_t debug); //! Function to destroy handle /*! * In this function, any structures allocated with CreateHandle should be * freed. It is generally the last function called before unloading of lib * happens. * * By convention, after this function has been called, *pp_handle must be * NULL. * * \param pp_handle Pointer to handle * \return 0 on success or error code */ int (*DestroyHandle)(void **pp_handle); //! Function to open input image /*! * Opens the specified image for reading. * * \param p_handle Handle * \param pp_filename_arr Array containing all specified input images * \param filename_arr_len Length of pp_filename_arr * \return 0 on success or error code */ int (*Open)(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len); //! Function to close opened input image /*! * Closes input image and frees any memory allocaed during opening but does * not invalidate the main handle. Further calls to for ex. Open() must still * be possible without first calling CreateHandle again! * * \param p_handle Handle * \return 0 on success or error code */ int (*Close)(void *p_handle); //! Function to get the input image's size /*! * Returns the real size of the input image. Real means the size of the * uncompressed or otherwise made available data contained inside the input * image. * * \param p_handle Handle * \param p_size Pointer to store input image's size to * \return 0 on success or error code */ int (*Size)(void *p_handle, uint64_t *p_size); //! Function to read data from input image /*! * Reads count bytes at offset from input image and copies them into memory * starting at the address of p_buf. Memory is pre-allocated to as much bytes * as should be read. * * \param p_handle Handle * \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 Amount of bytes read * \param p_errno errno in case of an error * \return 0 on success or error code */ int (*Read)(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read, int *p_errno); //! Function to get a help message for any supported lib-specific options /*! * Calling this function should return a string containing help messages for * any supported lib-specific options. Lines should be formated as follows: * * " option : description\n" * * Returned string will be freed by the caller using FreeBuffer(). * * If there is no help text, this function must return NULL in pp_help. * * \param Pointer to a string to return help text * \return 0 on success or error code on error */ int (*OptionsHelp)(const char **pp_help); //! Function to parse any lib-specific options /*! * This function is called with the options given with the --inopts parameter. * For all options that are valid for the lib, the option's valid member * must be set to 1. If errors are encountered, this function should fail and * return an error message in pp_error. pp_error will be freed by the caller * by using FreeBuffer. * * \param p_handle Handle * \param options_count Count of elements in pp_options * \param pp_options Input library options * \param pp_error Pointer to a string with error message * \return 0 on success or error code and error message on error */ int (*OptionsParse)(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); //! Function to get content to add to the info file /*! * The returned string is added to xmount's info file. This function is only * called once when the info file is generated. The returned string is then * freed with a call to FreeBuffer(). * * \param p_handle Handle * \param pp_info_buf Pointer to store the null-terminated content * \return 0 on success or error code */ int (*GetInfofileContent)(void *p_handle, const char **pp_info_buf); //! Function to get an error message /*! * This function should translate an error code that was previously returned * by one of the library functions into a human readable error message. * * By convention, this function must always return a valid pointer to a * NULL-terminated string! * * \param err_num Error code as returned by lib */ const char* (*GetErrorMessage)(int err_num); //! Function to free buffers that were allocated by lib /*! * \param p_buf Buffer to free * \return 0 on success or error code */ int (*FreeBuffer)(void *p_buf); } ts_LibXmountInputFunctions, *pts_LibXmountInputFunctions; //! Get library API version /*! * This function should return the value of LIBXMOUNT_INPUT_API_VERSION * * \return Supported version */ uint8_t LibXmount_Input_GetApiVersion(); typedef uint8_t (*t_LibXmount_Input_GetApiVersion)(); //! Get a list of supported formats /*! * Gets a list of supported input image formats. These are the strings * specified with xmount's --in command line option. The returned * string must be a constant vector of image formats split by \0 chars. To mark * the end of the vector, a single \0 must be used. * * As an example, "first\0second\0\0" would be a correct vector to return for * a lib supporting two input image formats. * * \return Vector containing supported format strings */ const char* LibXmount_Input_GetSupportedFormats(); typedef const char* (*t_LibXmount_Input_GetSupportedFormats)(); //! Get the lib's s_LibXmountInputFunctions structure /*! * This function should set the members of the given s_LibXmountInputFunctions * structure to the internal lib functions. All members have to be set. * * \param p_functions s_LibXmountInputFunctions structure to fill */ void LibXmount_Input_GetFunctions(pts_LibXmountInputFunctions p_functions); typedef void (*t_LibXmount_Input_GetFunctions)(pts_LibXmountInputFunctions); #endif // LIBXMOUNT_INPUT_H diff --git a/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.c b/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.c index c81c1ef..096d493 100644 --- a/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.c +++ b/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.c @@ -1,1046 +1,1046 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * This module has been written by Guy Voncken. It contains the functions for * * accessing simple AFF images created by Guymager. * * * * 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 . * *******************************************************************************/ // Please don't touch source code formatting! #ifdef LINTING // #define _LARGEFILE_SOURCE // #define _FILE_OFFSET_BITS 64 #define AAFF_STANDALONE #endif #ifdef AAFF_STANDALONE #define LOG_STDOUT TRUE #endif #include #include #include #include #include #include #include #include #include #include #include #include "../libxmount_input.h" #include "libxmount_input_aaff.h" static int AaffClose (void *pHandle); static const char* AaffGetErrorMessage (int ErrNum); #define AAFF_OPTION_MAXPAGEARRMEM "aaffmaxmem" #define AAFF_OPTION_LOG "aafflog" // ---------------------------- // Logging and error handling // ---------------------------- #define LOG_HEADER_LEN 80 int LogvEntry (const char *pLogFileName, uint8_t LogStdout, const char *pFileName, const char *pFunctionName, int LineNr, const char *pFormat, va_list pArguments) { time_t NowT; struct tm *pNowTM; FILE *pFile; int wr; char *pFullLogFileName = NULL; const char *pBase; char LogLineHeader[1024]; pid_t OwnPID; if (!LogStdout && (pLogFileName==NULL)) return AAFF_OK; time (&NowT); pNowTM = localtime (&NowT); OwnPID = getpid(); // pthread_self() wr = (int) strftime (&LogLineHeader[0] , sizeof(LogLineHeader) , "%a %d.%b.%Y %H:%M:%S ", pNowTM); //lint !e713 wr += snprintf (&LogLineHeader[wr], sizeof(LogLineHeader)-wr, "%5d ", OwnPID); //lint !e737 if (pFileName && pFunctionName) { pBase = strrchr(pFileName, '/'); if (pBase) pFileName = pBase+1; wr += snprintf (&LogLineHeader[wr], sizeof(LogLineHeader)-wr, "%s %s %d ", pFileName, pFunctionName, LineNr); //lint !e737 } // while (wr < LOG_HEADER_LEN) // LogLineHeader[wr++] = ' '; if (pLogFileName) { wr = asprintf (&pFullLogFileName, "%s_%d", pLogFileName, OwnPID); if ((wr <= 0) || (pFullLogFileName == NULL)) { if (LogStdout) printf ("\nLog file error: Can't build filename"); return AAFF_MEMALLOC_FAILED; } else { pFile = fopen64 (pFullLogFileName, "a"); if (pFile == NULL) { if (LogStdout) printf ("\nLog file error: Can't be opened"); return AAFF_CANNOT_OPEN_LOGFILE; } else { fprintf (pFile, "%-*s", LOG_HEADER_LEN, &LogLineHeader[0]); vfprintf (pFile, pFormat, pArguments); fprintf (pFile, "\n"); fclose (pFile); } free (pFullLogFileName); } } if (LogStdout) { printf ("%s", &LogLineHeader[0]); vprintf (pFormat, pArguments); printf ("\n"); } return AAFF_OK; } int LogEntry (const char *pLogFileName, uint8_t LogStdout, const char *pFileName, const char *pFunctionName, int LineNr, const char *pFormat, ...) { va_list VaList; int rc; if (!LogStdout && (pLogFileName==NULL)) return AAFF_OK; va_start (VaList, pFormat); //lint !e530 Symbol 'VaList' not initialized rc = LogvEntry (pLogFileName, LogStdout, pFileName, pFunctionName, LineNr, pFormat, VaList); va_end(VaList); return rc; } // CHK requires existance of pAaff handle #ifdef AAFF_STANDALONE #define LOG_ERRORS_ON_STDOUT TRUE #else #define LOG_ERRORS_ON_STDOUT pAaff->LogStdout #endif #define CHK(ChkVal) \ { \ int ChkValRc; \ if ((ChkValRc=(ChkVal)) != AAFF_OK) \ { \ const char *pErr = AaffGetErrorMessage (ChkValRc); \ LogEntry (pAaff->pLogFilename, LOG_ERRORS_ON_STDOUT, __FILE__, __FUNCTION__, __LINE__, "Error %d (%s) occured", ChkValRc, pErr); \ return ChkValRc; \ } \ } #define LOG(...) \ LogEntry (pAaff->pLogFilename, pAaff->LogStdout, __FILE__, __FUNCTION__, __LINE__, __VA_ARGS__); // AaffCheckError is called before exiting AaffRead. It should not // be called elsewehere or else the statistics would become wrong. static void AaffCheckError (t_pAaff pAaff, int Ret, int *pErrno) { *pErrno = 0; if (Ret != AAFF_OK) { if ((Ret >= AAFF_ERROR_ENOMEM_START) && (Ret <= AAFF_ERROR_ENOMEM_END)) *pErrno = ENOMEM; else if ((Ret >= AAFF_ERROR_EINVAL_START) && (Ret <= AAFF_ERROR_EINVAL_END)) *pErrno = EINVAL; else *pErrno = EIO; // all other errors } } // ------------------------------------ // Internal functions // ------------------------------------ uint64_t AaffU64 (char *pData) { uint64_t Val=0; int i; for (i=4; i<8; i++) Val = (Val << 8) | pData[i]; for (i=0; i<4; i++) Val = (Val << 8) | pData[i]; return Val; } static int AaffPageNumberFromSegmentName (char *pSegmentName, uint64_t *pPageNumber) { char *pSegmentNamePageNumber; char *pTail; pSegmentNamePageNumber = &pSegmentName[strlen(AFF_SEGNAME_PAGE)]; *pPageNumber = strtoull (pSegmentNamePageNumber, &pTail, 10); if (*pTail != '\0') return AAFF_INVALID_PAGE_NUMBER; // There should be no extra chars after the number return AAFF_OK; } static inline uint64_t AaffGetCurrentSeekPos (t_Aaff *pAaff) { return ftello (pAaff->pFile); } static inline uint64_t AaffSetCurrentSeekPos (t_Aaff *pAaff, uint64_t Val, int Whence) { if (fseeko (pAaff->pFile, Val, Whence) != 0) return AAFF_CANNOT_SEEK; return AAFF_OK; } static int AaffReadFile (t_Aaff *pAaff, void *pData, uint32_t DataLen) { if (fread (pData, DataLen, 1, pAaff->pFile) != 1) return AAFF_CANNOT_READ_DATA; return AAFF_OK; } static int AaffRealloc (void **ppBuff, uint32_t *pCurrentLen, uint32_t NewLen) { if (NewLen > *pCurrentLen) { *ppBuff = realloc (*ppBuff, NewLen); if (*ppBuff == NULL) return AAFF_MEMALLOC_FAILED; *pCurrentLen = NewLen; } return AAFF_OK; } static int AaffReadSegment (t_pAaff pAaff, char **ppName, uint32_t *pArg, char **ppData, uint32_t *pDataLen) { t_AffSegmentHeader Header; t_AffSegmentFooter Footer; CHK (AaffReadFile (pAaff, &Header, offsetof(t_AffSegmentHeader, Name))) if (strcmp (&Header.Magic[0], AFF_SEGMENT_HEADER_MAGIC) != 0) return AAFF_INVALID_HEADER; Header.NameLen = ntohl (Header.NameLen ); Header.DataLen = ntohl (Header.DataLen ); Header.Argument = ntohl (Header.Argument); CHK (AaffRealloc ((void**)&pAaff->pNameBuff, &pAaff->NameBuffLen, Header.NameLen+1)) // alloc +1, as is might be a string which can be more CHK (AaffRealloc ((void**)&pAaff->pDataBuff, &pAaff->DataBuffLen, Header.DataLen+1)) // easily handled by the calling fn when adding a \0 CHK (AaffReadFile (pAaff, pAaff->pNameBuff, Header.NameLen)) if (Header.DataLen) CHK (AaffReadFile (pAaff, pAaff->pDataBuff, Header.DataLen)) pAaff->pNameBuff[Header.NameLen] = '\0'; pAaff->pDataBuff[Header.DataLen] = '\0'; if (ppName) *ppName = pAaff->pNameBuff; if (pArg ) *pArg = Header.Argument; if (ppData) *ppData = pAaff->pDataBuff; if (pDataLen) *pDataLen = Header.DataLen; // Read footer and position file pointer to next segemnt at the same time // ---------------------------------------------------------------------- CHK (AaffReadFile (pAaff, &Footer, sizeof(Footer))) if (strcmp (&Footer.Magic[0], AFF_SEGMENT_FOOTER_MAGIC) != 0) return AAFF_INVALID_FOOTER; return AAFF_OK; } static int AaffReadSegmentPage (t_pAaff pAaff, uint64_t SearchPage, uint64_t *pFoundPage, char **ppData, uint32_t *pDataLen) { t_AffSegmentHeader Header; t_AffSegmentFooter Footer; char SearchPageStr[128]; int rc = AAFF_OK; *ppData = NULL; *pDataLen = 0; sprintf (SearchPageStr, "page%" PRIu64, SearchPage); CHK (AaffReadFile (pAaff, &Header, offsetof(t_AffSegmentHeader, Name))) if (strcmp (&Header.Magic[0], AFF_SEGMENT_HEADER_MAGIC) != 0) return AAFF_INVALID_HEADER; Header.NameLen = ntohl (Header.NameLen ); Header.DataLen = ntohl (Header.DataLen ); Header.Argument = ntohl (Header.Argument); CHK (AaffRealloc ((void**)&pAaff->pNameBuff, &pAaff->NameBuffLen, Header.NameLen+1)) CHK (AaffReadFile (pAaff, pAaff->pNameBuff, Header.NameLen)) pAaff->pNameBuff[Header.NameLen] = '\0'; if (strncmp (pAaff->pNameBuff, AFF_SEGNAME_PAGE, strlen(AFF_SEGNAME_PAGE)) != 0) return AAFF_WRONG_SEGMENT; CHK (AaffPageNumberFromSegmentName (pAaff->pNameBuff, pFoundPage)) if (*pFoundPage == SearchPage) { unsigned int Len; uLongf ZLen; int zrc; switch (Header.Argument) { case AFF_PAGEFLAGS_UNCOMPRESSED: CHK (AaffReadFile (pAaff, pAaff->pPageBuff, Header.DataLen)) pAaff->PageBuffDataLen = Header.DataLen; break; case AFF_PAGEFLAGS_COMPRESSED_ZERO: CHK (AaffReadFile (pAaff, &Len, sizeof(Len))) Len = ntohl (Len); memset (pAaff->pPageBuff, 0, Len); pAaff->PageBuffDataLen = Len; break; case AFF_PAGEFLAGS_COMPRESSED_ZLIB: CHK (AaffRealloc ((void**)&pAaff->pDataBuff, &pAaff->DataBuffLen, Header.DataLen)); CHK (AaffReadFile (pAaff, pAaff->pDataBuff, Header.DataLen)) // read into pDataBuff ZLen = pAaff->PageSize; // size of pPageBuff zrc = uncompress ((unsigned char*)(pAaff->pPageBuff), &ZLen, (unsigned char*)(pAaff->pDataBuff), Header.DataLen); // uncompress into pPageBuff pAaff->PageBuffDataLen = ZLen; if (zrc != Z_OK) return AAFF_UNCOMPRESS_FAILED; break; default: return AAFF_INVALID_PAGE_ARGUMENT; } *ppData = pAaff->pPageBuff; *pDataLen = pAaff->PageBuffDataLen; pAaff->CurrentPage = *pFoundPage; rc = AAFF_FOUND; } else { CHK (AaffSetCurrentSeekPos (pAaff, Header.DataLen, SEEK_CUR)) } // Read footer and position file pointer to next segemnt at the same time // ---------------------------------------------------------------------- CHK (AaffReadFile (pAaff, &Footer, sizeof(Footer))) if (strcmp (&Footer.Magic[0], AFF_SEGMENT_FOOTER_MAGIC) != 0) return AAFF_INVALID_FOOTER; return rc; } static int AaffReadPage (t_pAaff pAaff, uint64_t Page, char **ppBuffer, uint32_t *pLen) { if (Page >= pAaff->TotalPages) return AAFF_READ_BEYOND_LAST_PAGE; // Check if it's the current page // ------------------------------ if (Page == pAaff->CurrentPage) { *ppBuffer = pAaff->pPageBuff; *pLen = pAaff->PageBuffDataLen; return AAFF_OK; } // Set the seek position for starting the search // --------------------------------------------- int MaxHops; if ((pAaff->CurrentPage != AAFF_CURRENTPAGE_NOTSET) && (pAaff->CurrentPage+1 == Page)) // The current seek pos already is the correct one { MaxHops = 1; } else // Find the closest entry in PageSeekArr { int64_t Entry; Entry = Page / pAaff->Interleave; while (pAaff->pPageSeekArr[Entry] == 0) { Entry--; if (Entry<0) return AAFF_SEEKARR_CORRUPT; } AaffSetCurrentSeekPos (pAaff, pAaff->pPageSeekArr[Entry], SEEK_SET); MaxHops = Page - (Entry * pAaff->Interleave) +1; } // Run through segment list until page is found // -------------------------------------------- uint64_t Seek; uint64_t FoundPage=0; int rc; LOG ("Searching for page %" PRIu64 ", MaxHops=%d", Page, MaxHops); while (MaxHops--) { Seek = AaffGetCurrentSeekPos (pAaff); rc = AaffReadSegmentPage (pAaff, Page, &FoundPage, ppBuffer, pLen); if (rc != AAFF_FOUND) CHK (rc) LOG (" %" PRIu64 " (%d)", FoundPage, rc); if ((FoundPage % pAaff->Interleave) == 0) pAaff->pPageSeekArr[FoundPage/pAaff->Interleave] = Seek; if (rc == AAFF_FOUND) break; } if (MaxHops < 0) return AAFF_PAGE_NOT_FOUND; return AAFF_OK; } // --------------- // API functions // --------------- static int AaffCreateHandle (void **ppHandle, const char *pFormat, uint8_t Debug) { t_pAaff pAaff; *ppHandle = NULL; pAaff = (t_pAaff) malloc (sizeof(t_Aaff)); if (pAaff == NULL) return AAFF_MEMALLOC_FAILED; memset (pAaff, 0, sizeof(t_Aaff)); pAaff->MaxPageArrMem = AAFF_DEFAULT_MAX_PAGE_ARR_MEM; pAaff->LogStdout = Debug; *ppHandle = (void*) pAaff; return AAFF_OK; } static int AaffDestroyHandle (void **ppHandle) { t_pAaff pAaff = (t_pAaff) *ppHandle; if (pAaff->pFilename) free (pAaff->pFilename); if (pAaff->pPageSeekArr) free (pAaff->pPageSeekArr); if (pAaff->pLibVersion) free (pAaff->pLibVersion); if (pAaff->pFileType) free (pAaff->pFileType); if (pAaff->pNameBuff) free (pAaff->pNameBuff); if (pAaff->pDataBuff) free (pAaff->pDataBuff); if (pAaff->pPageBuff) free (pAaff->pPageBuff); if (pAaff->pInfoBuffConst) free (pAaff->pInfoBuffConst); if (pAaff->pInfoBuff) free (pAaff->pInfoBuff); memset (pAaff, 0, sizeof(t_Aaff)); free (pAaff); *ppHandle = NULL; return AAFF_OK; } int AaffOpen (void *pHandle, const char **ppFilenameArr, uint64_t FilenameArrLen) { t_pAaff pAaff = (t_pAaff) pHandle; char Signature[strlen(AFF_HEADER)+1]; uint64_t Seek; LOG ("Called - Files=%" PRIu64, FilenameArrLen); if (FilenameArrLen != 1) CHK (AAFF_SPLIT_IMAGES_NOT_SUPPORTED) pAaff->pFilename = strdup (ppFilenameArr[0]); pAaff->pFile = fopen (ppFilenameArr[0],"r"); if(pAaff->pFile==NULL) { (void) AaffDestroyHandle ((void**) &pAaff); CHK (AAFF_FILE_OPEN_FAILED) } // Check signature // --------------- CHK (AaffReadFile (pAaff, &Signature, sizeof(Signature))) if (memcmp (Signature, AFF_HEADER, sizeof(Signature)) !=0) { (void)AaffClose((void**)&pAaff); CHK (AAFF_INVALID_SIGNATURE) } // Read header segments // -------------------- char *pName; uint32_t Arg; char *pData; uint32_t DataLen; const int MAX_HEADER_SEGMENTS = 100; int Seg; unsigned int i; int wr; int Pos = 0; const unsigned HexStrLen = 32; char HexStr[HexStrLen+1]; #define REM (AaffInfoBuffLen-Pos) pAaff->pInfoBuffConst = (char *) malloc (AaffInfoBuffLen); pAaff->pInfoBuff = (char *) malloc (AaffInfoBuffLen*2); // Search for known segments at the image start for (Seg=0; SegPageSize = Arg; else if (strcmp (pName, AFF_SEGNAME_SECTORSIZE) == 0 ) pAaff->SectorSize = Arg; else if (strcmp (pName, AFF_SEGNAME_SECTORS) == 0 ) pAaff->Sectors = AaffU64(pData); else if (strcmp (pName, AFF_SEGNAME_IMAGESIZE) == 0 ) pAaff->ImageSize = AaffU64(pData); else if (strcmp (pName, AFF_SEGNAME_AFFLIB_VERSION) == 0 ) pAaff->pLibVersion = strdup((char*)pData); else if (strcmp (pName, AFF_SEGNAME_FILETYPE) == 0 ) pAaff->pFileType = strdup((char*)pData); else if ((strcmp(pName, AFF_SEGNAME_GID) == 0 ) || (strcmp(pName, AFF_SEGNAME_BADFLAG) == 0 )) { wr=0; for (i=0; ipInfoBuffConst[Pos]), REM,"%-25s %s", pName, HexStr); if (ipInfoBuffConst[Pos]), REM,"..."); Pos += snprintf (&(pAaff->pInfoBuffConst[Pos]), REM,"\n"); } else if (strncmp(pName,AFF_SEGNAME_PAGE,strlen(AFF_SEGNAME_PAGE))==0) { break; } else { if ((Arg == 0) && DataLen) Pos += snprintf (&(pAaff->pInfoBuffConst[Pos]), REM,"%-25s %s\n", pName, pData); } } #undef REM if (Seg >= MAX_HEADER_SEGMENTS) { (void) AaffClose ((void**)&pAaff); CHK (AAFF_TOO_MANY_HEADER_SEGEMENTS) } if (strstr (pAaff->pLibVersion, "Guymager") == NULL) { (void) AaffClose ((void**)&pAaff); CHK (AAFF_NOT_CREATED_BY_GUYMAGER) } // Prepare page seek array // ----------------------- uint64_t MaxEntries; int ArrBytes; pAaff->TotalPages = pAaff->ImageSize / pAaff->PageSize; if (pAaff->ImageSize % pAaff->PageSize) pAaff->TotalPages++; MaxEntries = (pAaff->MaxPageArrMem*1024*1024) / (sizeof (unsigned long long *) + 1); // +1 in order not to risk a result of 0 MaxEntries = GETMIN (MaxEntries, pAaff->TotalPages); pAaff->Interleave = pAaff->TotalPages / MaxEntries; if (pAaff->TotalPages % MaxEntries) pAaff->Interleave++; pAaff->PageSeekArrLen = pAaff->TotalPages / pAaff->Interleave; ArrBytes = pAaff->PageSeekArrLen * sizeof(uint64_t *); pAaff->pPageSeekArr = (uint64_t*)malloc (ArrBytes); memset (pAaff->pPageSeekArr, 0, ArrBytes); CHK (AaffPageNumberFromSegmentName (pName, &pAaff->CurrentPage)); if (pAaff->CurrentPage != 0) { (void) AaffClose ((void**)&pAaff); CHK (AAFF_UNEXPECTED_PAGE_NUMBER) } pAaff->pPageSeekArr[0] = Seek; // Alloc Buffers // ------------- pAaff->pPageBuff = (char *) malloc (pAaff->PageSize); pAaff->CurrentPage = AAFF_CURRENTPAGE_NOTSET; LOG ("Ret"); return AAFF_OK; } static int AaffClose (void *pHandle) { t_pAaff pAaff = (t_pAaff) pHandle; int rc = AAFF_OK; LOG ("Called"); if (fclose (pAaff->pFile)) rc = AAFF_CANNOT_CLOSE_FILE; LOG ("Ret"); return rc; } static int AaffSize (void *pHandle, uint64_t *pSize) { t_pAaff pAaff = (t_pAaff) pHandle; LOG ("Called"); *pSize = pAaff->ImageSize; LOG ("Ret - Size=%" PRIu64, *pSize); return AAFF_OK; } static int AaffRead (void *pHandle, char *pBuf, off_t Seek, size_t Count, size_t *pRead, int *pErrno) { t_pAaff pAaff = (t_pAaff) pHandle; char *pPageBuffer=NULL; uint64_t Page; uint64_t Seek64; uint64_t Remaining; uint32_t PageLen=0, Ofs, ToCopy; int Ret = AAFF_OK; LOG ("Called - Seek=%'" PRIu64 ",Count=%'" PRIu64, Seek, Count); *pRead = 0; *pErrno = 0; if (Seek < 0) { Ret = AAFF_NEGATIVE_SEEK; goto Leave; } Seek64 = Seek; if (Seek64 >= pAaff->ImageSize) // If calling function asks goto Leave; // for data beyond end of if ((Seek64+Count) > pAaff->ImageSize) // image simply return what Count = pAaff->ImageSize - Seek64; // is possible. Page = Seek64 / pAaff->PageSize; Ofs = Seek64 % pAaff->PageSize; Remaining = Count; while (Count) { Ret = AaffReadPage (pAaff, Page, &pPageBuffer, &PageLen); if (Ret) goto Leave; if (PageLen == 0) { Ret = AAFF_PAGE_LENGTH_ZERO; goto Leave; } ToCopy = GETMIN (PageLen-Ofs, Remaining); memcpy (pBuf, pPageBuffer+Ofs, ToCopy); Remaining -= ToCopy; pBuf += ToCopy; *pRead += ToCopy; Ofs=0; Page++; } Leave: AaffCheckError (pAaff, Ret, pErrno); LOG ("Ret %d - Read=%" PRIu32, Ret, *pRead); return Ret; } static int AaffOptionsHelp (const char **ppHelp) { char *pHelp=NULL; int wr; wr = asprintf (&pHelp, " %-12s : Maximum amount of RAM cache, in MiB, for image seek offsets. Default: %"PRIu64" MiB\n" " %-12s : Log file name.\n" " Specify full path for %s. The given file name is extended by _.\n", AAFF_OPTION_MAXPAGEARRMEM, AAFF_DEFAULT_MAX_PAGE_ARR_MEM, AAFF_OPTION_LOG, AAFF_OPTION_LOG); if ((pHelp == NULL) || (wr<=0)) return AAFF_MEMALLOC_FAILED; *ppHelp = pHelp; return AAFF_OK; } static int AaffOptionsParse (void *pHandle, uint32_t OptionCount, const pts_LibXmountOptions *ppOptions, const char **ppError) { pts_LibXmountOptions pOption; t_pAaff pAaff = (t_pAaff) pHandle; const char *pError = NULL; int rc = AAFF_OK; int Ok; LOG ("Called - OptionCount=%" PRIu32, OptionCount); *ppError = NULL; #define TEST_OPTION_UINT64(Opt,DestField) \ if (strcmp (pOption->p_key, Opt) == 0) \ { \ pAaff->DestField = StrToUint64 (pOption->p_value, &Ok); \ if (!Ok) \ { \ pError = "Error in option %s: Invalid value"; \ break; \ } \ LOG ("Option %s set to %" PRIu64, Opt, pAaff->DestField) \ } for (uint32_t i=0; ip_key, AAFF_OPTION_LOG) == 0) { pAaff->pLogFilename = strdup (pOption->p_value); rc = LOG ("Logging for libxmount_input_aaff started") if (rc != AAFF_OK) { pError = "Write test to log file failed"; break; } pOption->valid = TRUE; LOG ("Option %s set to %s", AAFF_OPTION_LOG, pAaff->pLogFilename); } else TEST_OPTION_UINT64 (AAFF_OPTION_MAXPAGEARRMEM, MaxPageArrMem) } #undef TEST_OPTION_UINT64 if (pError) *ppError = strdup (pError); LOG ("Ret - rc=%d,Error=%s", rc, *ppError); return rc; } static int AaffGetInfofileContent (void *pHandle, const char **ppInfoBuf) { t_pAaff pAaff = (t_pAaff) pHandle; uint64_t i; uint64_t Entries = 0; int Pos = 0; LOG ("Called"); #define REM (AaffInfoBuffLen-Pos) Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "AFF IMAGE INFORMATION"); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\n---------------------"); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nAFF file %s" , pAaff->pFilename ); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nPage size %u" , pAaff->PageSize ); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nSector size %d" , pAaff->SectorSize ); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nSectors %" PRIu64, pAaff->Sectors); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nImage size %" PRIu64 " (%0.1f GiB)", pAaff->ImageSize, pAaff->ImageSize/(1024.0*1024.0*1024.0)); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nTotal pages %" PRIu64, pAaff->TotalPages); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\n"); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\n%s", pAaff->pInfoBuffConst); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\n"); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nCurrent page "); if (pAaff->CurrentPage == AAFF_CURRENTPAGE_NOTSET) Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "not set"); else Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "%" PRIu64, pAaff->CurrentPage); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nSeek array length %" PRIu64, pAaff->PageSeekArrLen); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nSeek interleave %" PRIu64, pAaff->Interleave); for (i=0; iPageSeekArrLen; i++) { if (pAaff->pPageSeekArr[i]) Entries++; } Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\nSeek array entries %" PRIu64, Entries); Pos += snprintf (&pAaff->pInfoBuff[Pos], REM, "\n"); #undef REM *ppInfoBuf = strdup (pAaff->pInfoBuff); if (*ppInfoBuf == NULL) CHK (AAFF_MEMALLOC_FAILED) LOG ("Ret - %d bytes of info", strlen(*ppInfoBuf)+1); return AAFF_OK; } static const char* AaffGetErrorMessage (int ErrNum) { const char *pMsg; #define ADD_ERR(ErrCode) \ case ErrCode: pMsg = #ErrCode; \ break; switch (ErrNum) { ADD_ERR (AAFF_OK) ADD_ERR (AAFF_FOUND) ADD_ERR (AAFF_MEMALLOC_FAILED) ADD_ERR (AAFF_OPTIONS_ERROR) ADD_ERR (AAFF_SPLIT_IMAGES_NOT_SUPPORTED) ADD_ERR (AAFF_INVALID_SIGNATURE) ADD_ERR (AAFF_NOT_CREATED_BY_GUYMAGER) ADD_ERR (AAFF_CANNOT_OPEN_LOGFILE) ADD_ERR (AAFF_FILE_OPEN_FAILED) ADD_ERR (AAFF_CANNOT_READ_DATA) ADD_ERR (AAFF_INVALID_HEADER) ADD_ERR (AAFF_INVALID_FOOTER) ADD_ERR (AAFF_TOO_MANY_HEADER_SEGEMENTS) ADD_ERR (AAFF_INVALID_PAGE_NUMBER) ADD_ERR (AAFF_UNEXPECTED_PAGE_NUMBER) ADD_ERR (AAFF_CANNOT_CLOSE_FILE) ADD_ERR (AAFF_CANNOT_SEEK) ADD_ERR (AAFF_WRONG_SEGMENT) ADD_ERR (AAFF_UNCOMPRESS_FAILED) ADD_ERR (AAFF_INVALID_PAGE_ARGUMENT) ADD_ERR (AAFF_SEEKARR_CORRUPT) ADD_ERR (AAFF_PAGE_NOT_FOUND) ADD_ERR (AAFF_READ_BEYOND_IMAGE_LENGTH) ADD_ERR (AAFF_READ_BEYOND_LAST_PAGE) ADD_ERR (AAFF_PAGE_LENGTH_ZERO) ADD_ERR (AAFF_NEGATIVE_SEEK) default: pMsg = "Unknown error"; } #undef ARR_ERR return pMsg; } static int AaffFreeBuffer (void *pBuf) { free (pBuf); return AAFF_OK; } // ------------------------------------ // LibXmount_Input API implementation // ------------------------------------ uint8_t LibXmount_Input_GetApiVersion () { return LIBXMOUNT_INPUT_API_VERSION; } const char* LibXmount_Input_GetSupportedFormats () { return "aaff\0\0"; } void LibXmount_Input_GetFunctions(ts_LibXmountInputFunctions *pFunctions) { pFunctions->CreateHandle = &AaffCreateHandle; pFunctions->DestroyHandle = &AaffDestroyHandle; pFunctions->Open = &AaffOpen; pFunctions->Close = &AaffClose; pFunctions->Size = &AaffSize; pFunctions->Read = &AaffRead; pFunctions->OptionsHelp = &AaffOptionsHelp; pFunctions->OptionsParse = &AaffOptionsParse; pFunctions->GetInfofileContent = &AaffGetInfofileContent; pFunctions->GetErrorMessage = &AaffGetErrorMessage; pFunctions->FreeBuffer = &AaffFreeBuffer; } // ----------------------------------------------------- // Small main routine for testing // It converts an aff file to dd // ----------------------------------------------------- #ifdef AAFF_STANDALONE #define PRINT_ERROR_AND_EXIT(...) \ { \ printf (__VA_ARGS__); \ exit (1); \ } int ParseOptions (t_pAaff pAaff, char *pOptions) { pts_LibXmountOptions pOptionArr; pts_LibXmountOptions *ppOptionArr; int OptionCount; char *pSep; char *pEqual; char *pTmp; const char *pError; char *pOpt; int rc; if (pOptions == NULL) return AAFF_OK; if (*pOptions == '\0') return AAFF_OK; if (*pOptions == ',') return AAFF_OPTIONS_ERROR; if (pOptions[strlen(pOptions)-1] == ',') return AAFF_OPTIONS_ERROR; pOpt = strdup (pOptions); // Count number of comma separated options OptionCount = 1; pTmp = pOpt; while ((pTmp = strchr (pTmp, ',')) != NULL) { OptionCount++; pTmp++; } // Create and fill option array pOptionArr = (pts_LibXmountOptions) malloc (OptionCount * sizeof(ts_LibXmountOptions)); if (pOptionArr == NULL) PRINT_ERROR_AND_EXIT ("Cannot allocate pOptionArr"); memset (pOptionArr, 0, OptionCount * sizeof(ts_LibXmountOptions)); pTmp = pOpt; for (int i=0; i <...> [-comma_separated_options]\n", argv[0]); printf ("Possible options:\n%s\n", pHelp); printf ("The output file will be named dd.\n"); CHK (AaffFreeBuffer ((void*) pHelp)) exit (1); } if (argv[argc-1][0] == '-') { pOptions = strdup (&(argv[argc-1][1])); argc--; } rc = AaffCreateHandle ((void**) &pAaff, "aaff", LOG_STDOUT); if (rc != AAFF_OK) PRINT_ERROR_AND_EXIT ("Cannot create handle, rc=%d\n", rc) if (pOptions) CHK (ParseOptions(pAaff, pOptions)) rc = AaffOpen (pAaff, &argv[1], 1); if (rc) { printf ("Error %d while opening file %s\n", rc, argv[1]); exit (2); } CHK (AaffGetInfofileContent ((void*) pAaff, &pInfoBuff)) if (pInfoBuff) printf ("Contents of info buffer:\n%s\n", pInfoBuff); CHK (AaffFreeBuffer ((void*)pInfoBuff)) // Create destination file and fill it with data from aff // ------------------------------------------------------ FILE *pFile; pFile = fopen (argv[2], "w"); // const unsigned BuffSize = 13; // weird Buffsize for testing const unsigned BuffSize = 65536; char *pBuff; uint64_t ToRead; uint64_t Read; Remaining = pAaff->ImageSize; pBuff = (char *) malloc (BuffSize); CurrentPos=0; Errno = 0; PercentOld = -1; while (Remaining) { ToRead = GETMIN (Remaining, BuffSize); rc = AaffRead ((void*) pAaff, pBuff, CurrentPos, ToRead, &Read, &Errno); if ((rc != AAFF_OK) || (Errno != 0)) PRINT_ERROR_AND_EXIT("Error %d while calling AewfRead (Errno %d)\n", rc, Errno); if (Read != ToRead) PRINT_ERROR_AND_EXIT("Only %" PRIu64 " out of %" PRIu64 " bytes read\n", Read, ToRead); if (fwrite (pBuff, Read, 1, pFile) != 1) { printf ("Could not write to destinationfile\n"); exit (2); } CurrentPos += Read; Remaining -= Read; Percent = (100*CurrentPos) / pAaff->ImageSize; if (Percent != PercentOld) { printf ("\r%d%% done...", Percent); PercentOld = Percent; } } free (pBuff); fclose (pFile); return 0; } #endif diff --git a/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.h b/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.h index 95b7d87..4fc5a50 100644 --- a/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.h +++ b/trunk/libxmount_input/libxmount_input_aaff/libxmount_input_aaff.h @@ -1,180 +1,180 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * This module has been written by Guy Voncken. It contains the functions for * * accessing simple AFF images created by Guymager. * * * * 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 . * *******************************************************************************/ // Please don't touch source code formatting! #ifndef AAFF_H #define AAFF_H typedef struct _t_Aaff *t_pAaff; // ---------------------- // Constant definitions // ---------------------- #define GETMAX(a,b) ((a)>(b)?(a):(b)) #define GETMIN(a,b) ((a)<(b)?(a):(b)) #define FALSE 0 #define TRUE 1 const uint64_t AAFF_DEFAULT_MAX_PAGE_ARR_MEM = 10; // Default max. memory for caching seek points for fast page access (MiB) const uint64_t AAFF_CURRENTPAGE_NOTSET = UINT64_MAX; // ----------------- // AFF definitions // ----------------- #define AFF_GID_LENGTH 16 #define AFF_SEGARG_U64 2 // Used as argument for segments that contain a 64 bit unsigned in the data field #define AFF_HEADER "AFF10\r\n" #define AFF_SEGMENT_HEADER_MAGIC "AFF" #define AFF_SEGMENT_FOOTER_MAGIC "ATT" #define AFF_BADSECTOR_HEADER "BAD SECTOR" #define AFF_FILE_TYPE "AFF" #define AFF_SEGNAME_BADFLAG "badflag" #define AFF_SEGNAME_AFFLIB_VERSION "afflib_version" #define AFF_SEGNAME_FILETYPE "aff_file_type" #define AFF_SEGNAME_GID "image_gid" #define AFF_SEGNAME_SECTORS "devicesectors" #define AFF_SEGNAME_SECTORSIZE "sectorsize" #define AFF_SEGNAME_IMAGESIZE "imagesize" #define AFF_SEGNAME_PAGESIZE "pagesize" #define AFF_SEGNAME_BADSECTORS "badsectors" #define AFF_SEGNAME_MD5 "md5" #define AFF_SEGNAME_SHA256 "sha256" #define AFF_SEGNAME_DURATION "acquisition_seconds" #define AFF_SEGNAME_PAGE "page" #define AAFF_SEGNAME_COMMAND_LINE "acquisition_commandline" #define AAFF_SEGNAME_MACADDR "acquisition_macaddr" #define AAFF_SEGNAME_DATE "acquisition_date" // Format: YYYY-MM-DD HH:MM:SS TZT #define AAFF_SEGNAME_DEVICE "acquisition_device" #define AAFF_SEGNAME_MODEL "device_model" #define AAFF_SEGNAME_SN "device_sn" #define AFF_PAGEFLAGS_UNCOMPRESSED 0x0000 #define AFF_PAGEFLAGS_COMPRESSED_ZLIB 0x0001 #define AFF_PAGEFLAGS_COMPRESSED_ZERO 0x0033 #define AAFF_MD5_LEN 16 #define AAFF_SHA256_LEN 32 #define AAFF_BADSECTORMARKER_MAXLEN 65536 typedef struct { char Magic[4]; unsigned int NameLen; unsigned int DataLen; unsigned int Argument; // Named "flags" in original aff source, named "arg" in afinfo output. char Name[]; //lint !e1501 } __attribute__ ((packed)) t_AffSegmentHeader; typedef t_AffSegmentHeader *t_pAffSegmentHeader; // Between header and footer lie the segment name and the data typedef struct { char Magic[4]; unsigned int SegmentLen; } __attribute__ ((packed)) t_AffSegmentFooter; const int AaffInfoBuffLen = 1024*1024; typedef struct _t_Aaff { char *pFilename; FILE *pFile; char *pLibVersion; // AFF File Header info char *pFileType; unsigned int PageSize; unsigned int SectorSize; uint64_t Sectors; uint64_t ImageSize; uint64_t TotalPages; char *pNameBuff; // Buffers char *pDataBuff; unsigned int NameBuffLen; unsigned int DataBuffLen; uint64_t CurrentPage; char *pPageBuff; // Length is PageSize, contains data of CurrentPage unsigned int PageBuffDataLen; // Length of current data in PageBuff (the same for all pages, but the last one might contain less data) char *pInfoBuff; char *pInfoBuffConst; uint64_t *pPageSeekArr; uint64_t PageSeekArrLen; uint64_t Interleave; // The number of pages lying between 2 entries in the PageSeekArr // Options char *pLogFilename; uint64_t MaxPageArrMem; // Maximum amount of memory (in MiB) for storing page pointers uint8_t LogStdout; } t_Aaff; // Possible error codes enum { AAFF_OK = 0, AAFF_FOUND, AAFF_ERROR_ENOMEM_START=1000, AAFF_MEMALLOC_FAILED, AAFF_ERROR_ENOMEM_END, AAFF_ERROR_EINVAL_START=2000, AAFF_OPTIONS_ERROR, AAFF_SPLIT_IMAGES_NOT_SUPPORTED, AAFF_INVALID_SIGNATURE, AAFF_NOT_CREATED_BY_GUYMAGER, AAFF_CANNOT_OPEN_LOGFILE, AAFF_ERROR_EINVAL_END, AAFF_ERROR_EIO_START=3000, AAFF_FILE_OPEN_FAILED, AAFF_CANNOT_READ_DATA, AAFF_INVALID_HEADER, AAFF_INVALID_FOOTER, AAFF_TOO_MANY_HEADER_SEGEMENTS, AAFF_INVALID_PAGE_NUMBER, AAFF_UNEXPECTED_PAGE_NUMBER, AAFF_CANNOT_CLOSE_FILE, AAFF_CANNOT_SEEK, AAFF_WRONG_SEGMENT, AAFF_UNCOMPRESS_FAILED, AAFF_INVALID_PAGE_ARGUMENT, AAFF_SEEKARR_CORRUPT, AAFF_PAGE_NOT_FOUND, AAFF_READ_BEYOND_IMAGE_LENGTH, AAFF_READ_BEYOND_LAST_PAGE, AAFF_PAGE_LENGTH_ZERO, AAFF_NEGATIVE_SEEK, AAFF_ERROR_EIO_END, }; #endif diff --git a/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.c b/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.c index f0fb198..1d10304 100644 --- a/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.c +++ b/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.c @@ -1,260 +1,260 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include // For O_RDONLY #include "../libxmount_input.h" #ifndef HAVE_LIBAFF_STATIC #include #else #include "libaff/lib/afflib.h" #endif #include "libxmount_input_aff.h" /******************************************************************************* * LibXmount_Input API implementation ******************************************************************************/ /* * LibXmount_Input_GetApiVersion */ uint8_t LibXmount_Input_GetApiVersion() { return LIBXMOUNT_INPUT_API_VERSION; } /* * LibXmount_Input_GetSupportedFormats */ const char* LibXmount_Input_GetSupportedFormats() { return "aff\0\0"; } /* * LibXmount_Input_GetFunctions */ void LibXmount_Input_GetFunctions(ts_LibXmountInputFunctions *p_functions) { p_functions->CreateHandle=&AffCreateHandle; p_functions->DestroyHandle=&AffDestroyHandle; p_functions->Open=&AffOpen; p_functions->Close=&AffClose; p_functions->Size=&AffSize; p_functions->Read=&AffRead; p_functions->OptionsHelp=&AffOptionsHelp; p_functions->OptionsParse=&AffOptionsParse; p_functions->GetInfofileContent=&AffGetInfofileContent; p_functions->GetErrorMessage=&AffGetErrorMessage; p_functions->FreeBuffer=&AffFreeBuffer; } /******************************************************************************* * Private ******************************************************************************/ /* * AffCreateHandle */ static int AffCreateHandle(void **pp_handle, const char *p_format, uint8_t debug) { (void)p_format; pts_AffHandle p_aff_handle; // Alloc new lib handle p_aff_handle=(pts_AffHandle)malloc(sizeof(ts_AffHandle)); if(p_aff_handle==NULL) return AFF_MEMALLOC_FAILED; // Init lib handle p_aff_handle->h_aff=NULL; *pp_handle=p_aff_handle; return AFF_OK; } /* * AffDestroyHandle */ static int AffDestroyHandle(void **pp_handle) { pts_AffHandle p_aff_handle=(pts_AffHandle)*pp_handle; // Free lib handle if(p_aff_handle!=NULL) free(p_aff_handle); *pp_handle=NULL; return AFF_OK; } /* * AffOpen */ static int AffOpen(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len) { pts_AffHandle p_aff_handle=(pts_AffHandle)p_handle; // We need exactly one file if(filename_arr_len==0) return AFF_NO_INPUT_FILES; if(filename_arr_len>1) return AFF_TOO_MANY_INPUT_FILES; // Open AFF file p_aff_handle->h_aff=af_open(pp_filename_arr[0],O_RDONLY,0); if(p_aff_handle->h_aff==NULL) { // LOG_ERROR("Couldn't open AFF file!\n") return AFF_OPEN_FAILED; } // Encrypted images aren't supported for now // TODO: Add support trough lib params, f. ex. aff_password=xxxx if(af_cannot_decrypt(p_aff_handle->h_aff)) { af_close(p_aff_handle->h_aff); return AFF_ENCRYPTION_UNSUPPORTED; } return AFF_OK; } /* * AffClose */ static int AffClose(void *p_handle) { pts_AffHandle p_aff_handle=(pts_AffHandle)p_handle; // Close AFF handle if(af_close(p_aff_handle->h_aff)!=0) return AFF_CLOSE_FAILED; return AFF_OK; } /* * AffSize */ static int AffSize(void *p_handle, uint64_t *p_size) { pts_AffHandle p_aff_handle=(pts_AffHandle)p_handle; // TODO: Check for error. Unfortunately, I don't know how :( *p_size=af_seek(p_aff_handle->h_aff,0,SEEK_END); return AFF_OK; } /* * AffRead */ static int AffRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read, int *p_errno) { pts_AffHandle p_aff_handle=(pts_AffHandle)p_handle; size_t bytes_read; // Seek to requested position if(af_seek(p_aff_handle->h_aff,offset,SEEK_SET)!=offset) { return AFF_SEEK_FAILED; } // Read data // TODO: Check for errors bytes_read=af_read(p_aff_handle->h_aff,(unsigned char*)p_buf,count); if(bytes_read!=count) return AFF_READ_FAILED; *p_read=bytes_read; return AFF_OK; } /* * AffOptionsHelp */ static int AffOptionsHelp(const char **pp_help) { *pp_help=NULL; return AFF_OK; } /* * AffOptionsParse */ static int AffOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error) { return AFF_OK; } /* * AffGetInfofileContent */ static int AffGetInfofileContent(void *p_handle, const char **pp_info_buf) { // TODO *pp_info_buf=NULL; return AFF_OK; } /* * AffGetErrorMessage */ static const char* AffGetErrorMessage(int err_num) { switch(err_num) { case AFF_MEMALLOC_FAILED: return "Unable to allocate memory"; break; case AFF_NO_INPUT_FILES: return "No input file(s) specified"; break; case AFF_TOO_MANY_INPUT_FILES: return "Too many input files specified"; break; case AFF_OPEN_FAILED: return "Unable to open EWF file(s)"; break; case AFF_CLOSE_FAILED: return "Unable to close EWF file(s)"; break; case AFF_ENCRYPTION_UNSUPPORTED: return "Encrypted AFF files are currently not supported"; break; case AFF_SEEK_FAILED: return "Unable to seek into EWF data"; break; case AFF_READ_FAILED: return "Unable to read EWF data"; break; default: return "Unknown error"; } } /* * AffFreeBuffer */ static int AffFreeBuffer(void *p_buf) { free(p_buf); return AFF_OK; } /* ----- Change log ----- 20140724: * Initial version implementing AffOpen, AffSize, AffRead, AffClose, AffOptionsHelp, AffOptionsParse and AffFreeBuffer 20140804: * Added error handling and AffGetErrorMessage */ diff --git a/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.h b/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.h index 608a0e6..92e34f3 100644 --- a/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.h +++ b/trunk/libxmount_input/libxmount_input_aff/libxmount_input_aff.h @@ -1,73 +1,73 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_INPUT_AFF_H #define LIBXMOUNT_INPUT_AFF_H /******************************************************************************* * Enums, Typedefs, etc... ******************************************************************************/ //! Possible error return codes enum { AFF_OK=0, AFF_MEMALLOC_FAILED, AFF_NO_INPUT_FILES, AFF_TOO_MANY_INPUT_FILES, AFF_OPEN_FAILED, AFF_CLOSE_FAILED, AFF_ENCRYPTION_UNSUPPORTED, AFF_SEEK_FAILED, AFF_READ_FAILED }; //! Library handle typedef struct s_AffHandle { //! AFF handle AFFILE *h_aff; } ts_AffHandle, *pts_AffHandle; /******************************************************************************* * Forward declarations ******************************************************************************/ static int AffCreateHandle(void **pp_handle, const char *p_format, uint8_t debug); static int AffDestroyHandle(void **pp_handle); static int AffOpen(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len); static int AffClose(void *p_handle); static int AffSize(void *p_handle, uint64_t *p_size); static int AffRead(void *p_handle, char *p_buf, off_t seek, size_t count, size_t *p_read, int *p_errno); static int AffOptionsHelp(const char **pp_help); static int AffOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); static int AffGetInfofileContent(void *p_handle, const char **pp_info_buf); static const char* AffGetErrorMessage(int err_num); static int AffFreeBuffer(void *p_buf); #endif // LIBXMOUNT_INPUT_AFF_H diff --git a/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.c b/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.c index 7ae1c36..385c375 100644 --- a/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.c +++ b/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.c @@ -1,487 +1,487 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include "../libxmount_input.h" #ifndef HAVE_LIBEWF_STATIC #include #else #include "libewf/include/libewf.h" #endif #ifndef LIBEWF_HANDLE // libewf version 2 no longer defines LIBEWF_HANDLE #define HAVE_LIBEWF_V2_API #endif #include "libxmount_input_ewf.h" /******************************************************************************* * LibXmount_Input API implementation ******************************************************************************/ /* * LibXmount_Input_GetApiVersion */ uint8_t LibXmount_Input_GetApiVersion() { return LIBXMOUNT_INPUT_API_VERSION; } /* * LibXmount_Input_GetSupportedFormats */ const char* LibXmount_Input_GetSupportedFormats() { return "ewf\0\0"; } /* * LibXmount_Input_GetFunctions */ void LibXmount_Input_GetFunctions(ts_LibXmountInputFunctions *p_functions) { p_functions->CreateHandle=&EwfCreateHandle; p_functions->DestroyHandle=&EwfDestroyHandle; p_functions->Open=&EwfOpen; p_functions->Size=&EwfSize; p_functions->Read=&EwfRead; p_functions->Close=&EwfClose; p_functions->OptionsHelp=&EwfOptionsHelp; p_functions->OptionsParse=&EwfOptionsParse; p_functions->GetInfofileContent=&EwfGetInfofileContent; p_functions->GetErrorMessage=&EwfGetErrorMessage; p_functions->FreeBuffer=&EwfFreeBuffer; } /******************************************************************************* * Private ******************************************************************************/ /* * EwfCreateHandle */ static int EwfCreateHandle(void **pp_handle, const char *p_format, uint8_t debug) { (void)p_format; pts_EwfHandle p_ewf_handle; // Alloc new lib handle p_ewf_handle=(pts_EwfHandle)malloc(sizeof(ts_EwfHandle)); if(p_ewf_handle==NULL) return EWF_MEMALLOC_FAILED; // Init handle values p_ewf_handle->h_ewf=NULL; // Init lib handle #ifdef HAVE_LIBEWF_V2_API if(libewf_handle_initialize(&(p_ewf_handle->h_ewf),NULL)!=1) { return EWF_HANDLE_CREATION_FAILED; } #endif *pp_handle=p_ewf_handle; return EWF_OK; } /* * EwfDestroyHandle */ static int EwfDestroyHandle(void **pp_handle) { pts_EwfHandle p_ewf_handle=(pts_EwfHandle)*pp_handle; int ret=EWF_OK; if(p_ewf_handle!=NULL) { // Free EWF handle #ifdef HAVE_LIBEWF_V2_API if(libewf_handle_free(&(p_ewf_handle->h_ewf),NULL)!=1) { ret=EWF_HANDLE_DESTRUCTION_FAILED; } #endif // Free lib handle free(p_ewf_handle); p_ewf_handle=NULL; } *pp_handle=p_ewf_handle; return ret; } /* * EwfOpen */ static int EwfOpen(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len) { pts_EwfHandle p_ewf_handle=(pts_EwfHandle)p_handle; // We need at least one file if(filename_arr_len==0) return EWF_NO_INPUT_FILES; // Make sure all files are EWF files for(uint64_t i=0;ih_ewf, (char* const*)pp_filename_arr, filename_arr_len, libewf_get_access_flags_read(), NULL)!=1) #else p_ewf_handle->h_ewf=libewf_open((char* const*)pp_filename_arr, filename_arr_len, libewf_get_flags_read()); if(p_ewf_handle->h_ewf==NULL) #endif { return EWF_OPEN_FAILED; } #ifndef HAVE_LIBEWF_V2_API // Parse EWF header if(libewf_parse_header_values(p_ewf_handle->h_ewf, LIBEWF_DATE_FORMAT_ISO8601)!=1) { return EWF_HEADER_PARSING_FAILED; } #endif return EWF_OK; } /* * EwfClose */ static int EwfClose(void *p_handle) { pts_EwfHandle p_ewf_handle=(pts_EwfHandle)p_handle; // Close EWF handle #ifdef HAVE_LIBEWF_V2_API if(libewf_handle_close(p_ewf_handle->h_ewf,NULL)!=0) #else if(libewf_close(p_ewf_handle->h_ewf)!=0) #endif { return EWF_CLOSE_FAILED; } return EWF_OK; } /* * EwfSize */ static int EwfSize(void *p_handle, uint64_t *p_size) { pts_EwfHandle p_ewf_handle=(pts_EwfHandle)p_handle; #ifdef HAVE_LIBEWF_V2_API if(libewf_handle_get_media_size(p_ewf_handle->h_ewf,p_size,NULL)!=1) { #else if(libewf_get_media_size(p_ewf_handle->h_ewf,p_size)!=1) { #endif return EWF_GET_SIZE_FAILED; } return EWF_OK; } /* * EwfRead */ static int EwfRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read, int *p_errno) { pts_EwfHandle p_ewf_handle=(pts_EwfHandle)p_handle; // TODO: Return value of libewf_handle_read_buffer is ssize_t with -1 on error size_t bytes_read; #ifdef HAVE_LIBEWF_V2_API if(libewf_handle_seek_offset(p_ewf_handle->h_ewf, offset, SEEK_SET, NULL)!=-1) #else if(libewf_seek_offset(p_ewf_handle->h_ewf,offset)!=-1) #endif { #ifdef HAVE_LIBEWF_V2_API bytes_read=libewf_handle_read_buffer(p_ewf_handle->h_ewf, p_buf, count, NULL); #else bytes_read=libewf_read_buffer(p_ewf_handle->h_ewf,p_buf,count); #endif if(bytes_read!=count) return EWF_READ_FAILED; } else { return EWF_SEEK_FAILED; } *p_read=bytes_read; return EWF_OK; } /* * EwfOptionsHelp */ static int EwfOptionsHelp(const char **pp_help) { *pp_help=NULL; return EWF_OK; } /* * EwfOptionsParse */ static int EwfOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error) { return EWF_OK; } /* * EwfGetInfofileContent */ static int EwfGetInfofileContent(void *p_handle, const char **pp_info_buf) { pts_EwfHandle p_ewf_handle=(pts_EwfHandle)p_handle; char *p_infobuf=NULL; int ret; char buf[512]; uint8_t uint8value; uint32_t uint32value; uint64_t uint64value; #define EWF_INFOBUF_REALLOC(size) { \ p_infobuf=(char*)realloc(p_infobuf,size); \ if(p_infobuf==NULL) return EWF_MEMALLOC_FAILED; \ } #define EWF_INFOBUF_APPEND_STR(str) { \ if(p_infobuf!=NULL) { \ EWF_INFOBUF_REALLOC(strlen(p_infobuf)+strlen(str)+1); \ strcpy(p_infobuf+strlen(p_infobuf),str); \ } else { \ EWF_INFOBUF_REALLOC(strlen(str)+1); \ strcpy(p_infobuf,str); \ } \ } #define EWF_INFOBUF_APPEND_VALUE(desc) { \ if(ret==1) { \ EWF_INFOBUF_APPEND_STR(desc); \ EWF_INFOBUF_APPEND_STR(buf); \ EWF_INFOBUF_APPEND_STR("\n"); \ } \ } EWF_INFOBUF_APPEND_STR("_Acquiry information_\n"); #ifdef HAVE_LIBEWF_V2_API #define EWF_GET_HEADER_VALUE(fun) { \ ret=fun(p_ewf_handle->h_ewf,(uint8_t*)buf,sizeof(buf),NULL); \ } EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_case_number); EWF_INFOBUF_APPEND_VALUE("Case number: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_description); EWF_INFOBUF_APPEND_VALUE("Description: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_examiner_name); EWF_INFOBUF_APPEND_VALUE("Examiner: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_evidence_number); EWF_INFOBUF_APPEND_VALUE("Evidence number: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_notes); EWF_INFOBUF_APPEND_VALUE("Notes: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_acquiry_date); EWF_INFOBUF_APPEND_VALUE("Acquiry date: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_system_date); EWF_INFOBUF_APPEND_VALUE("System date: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_acquiry_operating_system); EWF_INFOBUF_APPEND_VALUE("Acquiry os: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_acquiry_software_version); EWF_INFOBUF_APPEND_VALUE("Acquiry sw version: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_model); EWF_INFOBUF_APPEND_VALUE("Model: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_header_value_serial_number); EWF_INFOBUF_APPEND_VALUE("Serial number: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_hash_value_md5); EWF_INFOBUF_APPEND_VALUE("MD5 hash: "); EWF_GET_HEADER_VALUE(libewf_handle_get_utf8_hash_value_sha1); EWF_INFOBUF_APPEND_VALUE("SHA1 hash: "); #undef EWF_GET_HEADER_VALUE #else #define EWF_GET_HEADER_VALUE(fun) { \ ret=fun(p_ewf_handle->h_ewf,buf,sizeof(buf)); \ } EWF_GET_HEADER_VALUE(libewf_get_header_value_case_number); EWF_INFOBUF_APPEND_VALUE("Case number: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_description); EWF_INFOBUF_APPEND_VALUE("Description: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_examiner_name); EWF_INFOBUF_APPEND_VALUE("Examiner: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_evidence_number); EWF_INFOBUF_APPEND_VALUE("Evidence number: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_notes); EWF_INFOBUF_APPEND_VALUE("Notes: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_acquiry_date); EWF_INFOBUF_APPEND_VALUE("Acquiry date: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_system_date); EWF_INFOBUF_APPEND_VALUE("System date: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_acquiry_operating_system); EWF_INFOBUF_APPEND_VALUE("Acquiry os: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_acquiry_software_version); EWF_INFOBUF_APPEND_VALUE("Acquiry sw version: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_model); EWF_INFOBUF_APPEND_VALUE("Model: "); EWF_GET_HEADER_VALUE(libewf_get_header_value_serial_number); EWF_INFOBUF_APPEND_VALUE("Serial number: "); EWF_GET_HEADER_VALUE(libewf_get_hash_value_md5); EWF_INFOBUF_APPEND_VALUE("MD5 hash: "); EWF_GET_HEADER_VALUE(libewf_get_hash_value_sha1); EWF_INFOBUF_APPEND_VALUE("SHA1 hash: "); #undef EWF_GET_HEADER_VALUE #endif EWF_INFOBUF_APPEND_STR("\n_Media information_\n"); #ifdef HAVE_LIBEWF_V2_API ret=libewf_handle_get_media_type(p_ewf_handle->h_ewf,&uint8value,NULL); #else ret=libewf_get_media_type(p_ewf_handle->h_ewf,&uint8value); #endif if(ret==1) { EWF_INFOBUF_APPEND_STR("Media type: "); switch(uint8value) { case LIBEWF_MEDIA_TYPE_REMOVABLE: EWF_INFOBUF_APPEND_STR("removable disk\n"); break; case LIBEWF_MEDIA_TYPE_FIXED: EWF_INFOBUF_APPEND_STR("fixed disk\n"); break; case LIBEWF_MEDIA_TYPE_OPTICAL: EWF_INFOBUF_APPEND_STR("optical\n"); break; case LIBEWF_MEDIA_TYPE_SINGLE_FILES: EWF_INFOBUF_APPEND_STR("single files\n"); break; case LIBEWF_MEDIA_TYPE_MEMORY: EWF_INFOBUF_APPEND_STR("memory\n"); break; default: EWF_INFOBUF_APPEND_STR("unknown\n"); } } #ifdef HAVE_LIBEWF_V2_API ret=libewf_handle_get_bytes_per_sector(p_ewf_handle->h_ewf,&uint32value,NULL); sprintf(buf,"%" PRIu32,uint32value); EWF_INFOBUF_APPEND_VALUE("Bytes per sector: "); ret=libewf_handle_get_number_of_sectors(p_ewf_handle->h_ewf,&uint64value,NULL); sprintf(buf,"%" PRIu64,uint64value); EWF_INFOBUF_APPEND_VALUE("Number of sectors: "); #else ret=libewf_get_bytes_per_sector(p_ewf_handle->h_ewf,&uint32value); sprintf(buf,"%" PRIu32,uint32value); EWF_INFOBUF_APPEND_VALUE("Bytes per sector: "); ret=libewf_handle_get_amount_of_sectors(p_ewf_handle->h_ewf,&uint64value); sprintf(buf,"%" PRIu64,uint64value); EWF_INFOBUF_APPEND_VALUE("Number of sectors: "); #endif #undef EWF_INFOBUF_APPEND_VALUE #undef EWF_INFOBUF_APPEND_STR #undef EWF_INFOBUF_REALLOC *pp_info_buf=p_infobuf; return EWF_OK; } /* * EwfGetErrorMessage */ static const char* EwfGetErrorMessage(int err_num) { switch(err_num) { case EWF_MEMALLOC_FAILED: return "Unable to allocate memory"; break; case EWF_HANDLE_CREATION_FAILED: return "Unable to create EWF handle"; break; case EWF_HANDLE_DESTRUCTION_FAILED: return "Unable to destroy EWF handle"; break; case EWF_NO_INPUT_FILES: return "No input file(s) specified"; break; case EWF_INVALID_INPUT_FILES: return "The specified input file(s) are not valid EWF files"; break; case EWF_OPEN_FAILED: return "Unable to open EWF file(s)"; break; case EWF_HEADER_PARSING_FAILED: return "Unable to parse EWF header values"; break; case EWF_CLOSE_FAILED: return "Unable to close EWF file(s)"; break; case EWF_GET_SIZE_FAILED: return "Unable to get EWF data size"; break; case EWF_SEEK_FAILED: return "Unable to seek into EWF data"; break; case EWF_READ_FAILED: return "Unable to read EWF data"; break; default: return "Unknown error"; } } /* * EwfFreeBuffer */ static int EwfFreeBuffer(void *p_buf) { free(p_buf); return EWF_OK; } /* ----- Change log ----- 20140724: * Initial version implementing EwfOpen, EwfSize, EwfRead, EwfClose, EwfOptionsHelp, EwfOptionsParse and EwfFreeBuffer. 20140731: * Added support for libewf v1 API. 20140803: * Added EwfCreateHandle, EwfDestroyHandle and EwfGetInfofileContent. */ diff --git a/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.h b/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.h index e06c545..405de47 100644 --- a/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.h +++ b/trunk/libxmount_input/libxmount_input_ewf/libxmount_input_ewf.h @@ -1,81 +1,81 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_INPUT_EWF_H #define LIBXMOUNT_INPUT_EWF_H /******************************************************************************* * Enums, Typedefs, etc... ******************************************************************************/ //! Possible error return codes enum { EWF_OK=0, EWF_MEMALLOC_FAILED, EWF_HANDLE_CREATION_FAILED, EWF_HANDLE_DESTRUCTION_FAILED, EWF_NO_INPUT_FILES, EWF_INVALID_INPUT_FILES, EWF_OPEN_FAILED, EWF_HEADER_PARSING_FAILED, EWF_CLOSE_FAILED, EWF_GET_SIZE_FAILED, EWF_SEEK_FAILED, EWF_READ_FAILED }; //! Library handle typedef struct s_EwfHandle { #ifdef HAVE_LIBEWF_V2_API //! EWF handle libewf_handle_t *h_ewf; #else //! EWF handle LIBEWF_HANDLE *h_ewf; #endif } ts_EwfHandle, *pts_EwfHandle; /******************************************************************************* * Forward declarations ******************************************************************************/ static int EwfCreateHandle(void **pp_handle, const char *p_format, uint8_t debug); static int EwfDestroyHandle(void **pp_handle); static int EwfOpen(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len); static int EwfClose(void *p_handle); static int EwfSize(void *p_handle, uint64_t *p_size); static int EwfRead(void *p_handle, char *p_buf, off_t seek, size_t count, size_t *p_read, int *p_errno); static int EwfOptionsHelp(const char **pp_help); static int EwfOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); static int EwfGetInfofileContent(void *p_handle, const char **pp_info_buf); static const char* EwfGetErrorMessage(int err_num); static int EwfFreeBuffer(void *p_buf); #endif // LIBXMOUNT_INPUT_EWF_H diff --git a/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.c b/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.c index ccacbad..7ba1b64 100644 --- a/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.c +++ b/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.c @@ -1,333 +1,333 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * Mostly based upon code written and copyright 2014 by Guy Voncken. * * * * 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 . * *******************************************************************************/ #include #include #include #include "../libxmount_input.h" #include "libxmount_input_raw.h" #define LOG_WARNING(...) { \ LIBXMOUNT_LOG_WARNING(__VA_ARGS__); \ } /******************************************************************************* * LibXmount_Input API implementation ******************************************************************************/ /* * LibXmount_Input_GetApiVersion */ uint8_t LibXmount_Input_GetApiVersion() { return LIBXMOUNT_INPUT_API_VERSION; } /* * LibXmount_Input_GetSupportedFormats */ const char* LibXmount_Input_GetSupportedFormats() { return "raw\0dd\0\0"; } /* * LibXmount_Input_GetFunctions */ void LibXmount_Input_GetFunctions(ts_LibXmountInputFunctions *p_functions) { p_functions->CreateHandle=&RawCreateHandle; p_functions->DestroyHandle=&RawDestroyHandle; p_functions->Open=&RawOpen; p_functions->Close=&RawClose; p_functions->Size=&RawSize; p_functions->Read=&RawRead; p_functions->OptionsHelp=&RawOptionsHelp; p_functions->OptionsParse=&RawOptionsParse; p_functions->GetInfofileContent=&RawGetInfofileContent; p_functions->GetErrorMessage=&RawGetErrorMessage; p_functions->FreeBuffer=&RawFreeBuffer; } /******************************************************************************* * Private ******************************************************************************/ // --------------------------- // Internal static functions // --------------------------- static inline uint64_t RawGetCurrentSeekPos(t_pPiece pPiece) { return ftello(pPiece->pFile); } static inline int RawSetCurrentSeekPos(t_pPiece pPiece, uint64_t Val, int Whence) { if (fseeko (pPiece->pFile, Val, Whence) != 0) return RAW_CANNOT_SEEK; return RAW_OK; } static int RawRead0(t_praw praw, char *pBuffer, uint64_t Seek, uint32_t *pCount) { t_pPiece pPiece; uint64_t i; // Find correct piece to read from // ------------------------------- for (i=0; iPieces; i++) { pPiece = &praw->pPieceArr[i]; if (Seek < pPiece->FileSize) break; Seek -= pPiece->FileSize; } if (i >= praw->Pieces) return RAW_READ_BEYOND_END_OF_IMAGE; // Read from this piece // -------------------- CHK (RawSetCurrentSeekPos (pPiece, Seek, SEEK_SET)) *pCount = GETMIN (*pCount, pPiece->FileSize - Seek); if (fread (pBuffer, *pCount, 1, pPiece->pFile) != 1) { return RAW_CANNOT_READ_DATA; } return RAW_OK; } // --------------- // API functions // --------------- /* * RawCreateHandle */ static int RawCreateHandle(void **pp_handle, const char *p_format, uint8_t debug) { (void)p_format; t_praw p_raw=NULL; p_raw=(t_praw)malloc(sizeof(t_raw)); if(p_raw==NULL) return RAW_MEMALLOC_FAILED; memset(p_raw,0,sizeof(t_raw)); if(strcmp(p_format,"dd")==0) { LOG_WARNING("Using '--in dd' is deprecated and will be removed in the next " "release. Please use '--in raw' instead.\n"); } *pp_handle=p_raw; return RAW_OK; } /* * RawDestroyHandle */ static int RawDestroyHandle(void **pp_handle) { free(*pp_handle); *pp_handle=NULL; return RAW_OK; } /* * RawOpen */ static int RawOpen(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len) { t_praw praw=(t_praw)p_handle; t_pPiece pPiece; praw->Pieces = filename_arr_len; praw->pPieceArr = (t_pPiece) malloc (praw->Pieces * sizeof(t_Piece)); if (praw->pPieceArr == NULL) return RAW_MEMALLOC_FAILED; // Need to set everything to 0 in case an error occurs later and RawClose is // called memset(praw->pPieceArr,0,praw->Pieces * sizeof(t_Piece)); praw->TotalSize = 0; for (uint64_t i=0; i < praw->Pieces; i++) { pPiece = &praw->pPieceArr[i]; pPiece->pFilename = strdup (pp_filename_arr[i]); if (pPiece->pFilename == NULL) { RawClose(p_handle); return RAW_MEMALLOC_FAILED; } pPiece->pFile = fopen (pPiece->pFilename, "r"); if (pPiece->pFile == NULL) { RawClose(p_handle); return RAW_FILE_OPEN_FAILED; } CHK(RawSetCurrentSeekPos(pPiece, 0, SEEK_END)) pPiece->FileSize = RawGetCurrentSeekPos (pPiece); praw->TotalSize += pPiece->FileSize; } return RAW_OK; } /* * RawClose */ static int RawClose(void *p_handle) { t_praw praw = (t_praw)p_handle; t_pPiece pPiece; int CloseErrors = 0; if (praw->pPieceArr) { for (uint64_t i=0; i < praw->Pieces; i++) { pPiece = &praw->pPieceArr[i]; if (pPiece->pFile) { if (fclose (pPiece->pFile)) CloseErrors=1; } if (pPiece->pFilename) free (pPiece->pFilename); } free (praw->pPieceArr); } if (CloseErrors) return RAW_CANNOT_CLOSE_FILE; return RAW_OK; } /* * RawSize */ static int RawSize(void *p_handle, uint64_t *p_size) { t_praw p_raw_handle=(t_praw)p_handle; *p_size=p_raw_handle->TotalSize; return RAW_OK; } /* * RawRead */ static int RawRead(void *p_handle, char *p_buf, off_t seek, size_t count, size_t *p_read, int *p_errno) { t_praw p_raw_handle=(t_praw)p_handle; uint32_t remaining=count; uint32_t to_read; if((seek+count)>p_raw_handle->TotalSize) { return RAW_READ_BEYOND_END_OF_IMAGE; } do { to_read=remaining; CHK(RawRead0(p_raw_handle,p_buf,seek,&to_read)) remaining-=to_read; p_buf+=to_read; seek+=to_read; } while(remaining); *p_read=count; return RAW_OK; } /* * RawOptionsHelp */ static int RawOptionsHelp(const char **pp_help) { *pp_help=NULL; return RAW_OK; } /* * RawOptionsParse */ static int RawOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error) { return RAW_OK; } /* * RawGetInfofileContent */ static int RawGetInfofileContent(void *p_handle, const char **pp_info_buf) { t_praw p_raw_handle=(t_praw)p_handle; int ret; char *p_info_buf; ret=asprintf(&p_info_buf, "RAW image assembled of %" PRIu64 " piece(s)\n" "%" PRIu64 " bytes in total (%0.3f GiB)\n", p_raw_handle->Pieces, p_raw_handle->TotalSize, p_raw_handle->TotalSize/(1024.0*1024.0*1024.0)); if(ret<0 || *pp_info_buf==NULL) return RAW_MEMALLOC_FAILED; *pp_info_buf=p_info_buf; return RAW_OK; } /* * RawGetErrorMessage */ static const char* RawGetErrorMessage(int err_num) { switch(err_num) { case RAW_MEMALLOC_FAILED: return "Unable to allocate memory"; break; case RAW_FILE_OPEN_FAILED: return "Unable to open raw file(s)"; break; case RAW_CANNOT_READ_DATA: return "Unable to read raw data"; break; case RAW_CANNOT_CLOSE_FILE: return "Unable to close raw file(s)"; break; case RAW_CANNOT_SEEK: return "Unable to seek into raw data"; break; case RAW_READ_BEYOND_END_OF_IMAGE: return "Unable to read raw data: Attempt to read past EOF"; break; default: return "Unknown error"; } } /* * RawFreeBuffer */ static int RawFreeBuffer(void *p_buf) { free(p_buf); return RAW_OK; } diff --git a/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.h b/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.h index ce69ea8..337e3ff 100644 --- a/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.h +++ b/trunk/libxmount_input/libxmount_input_raw/libxmount_input_raw.h @@ -1,115 +1,115 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * Mostly based upon code written and copyright 2014 by Guy Voncken. * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_INPUT_RAW_H #define LIBXMOUNT_INPUT_RAW_H /******************************************************************************* * Error codes etc... ******************************************************************************/ enum { RAW_OK=0, RAW_MEMALLOC_FAILED, RAW_FILE_OPEN_FAILED, RAW_CANNOT_READ_DATA, RAW_CANNOT_CLOSE_FILE, RAW_CANNOT_SEEK, RAW_READ_BEYOND_END_OF_IMAGE }; // ---------------------- // Constant definitions // ---------------------- #define GETMAX(a,b) ((a)>(b)?(a):(b)) #define GETMIN(a,b) ((a)<(b)?(a):(b)) // --------------------- // Types and strutures // --------------------- typedef struct { char *pFilename; uint64_t FileSize; FILE *pFile; } t_Piece, *t_pPiece; typedef struct { t_pPiece pPieceArr; uint64_t Pieces; uint64_t TotalSize; } t_raw, *t_praw; // ---------------- // Error handling // ---------------- #ifdef RAW_DEBUG #define CHK(ChkVal) \ { \ int ChkValRc; \ if ((ChkValRc=(ChkVal)) != RAW_OK) \ { \ printf ("Err %d in %s, %d\n", ChkValRc, __FILE__, __LINE__); \ return ChkValRc; \ } \ } #define DEBUG_PRINTF(pFormat, ...) \ printf (pFormat, ##__VA_ARGS__); #else #define CHK(ChkVal) \ { \ int ChkValRc; \ if ((ChkValRc=(ChkVal)) != RAW_OK) \ return ChkValRc; \ } #define DEBUG_PRINTF(...) #endif /******************************************************************************* * Forward declarations ******************************************************************************/ static int RawCreateHandle(void **pp_handle, const char *p_format, uint8_t debug); static int RawDestroyHandle(void **pp_handle); static int RawOpen(void *p_handle, const char **pp_filename_arr, uint64_t filename_arr_len); static int RawClose(void *p_handle); static int RawSize(void *p_handle, uint64_t *p_size); static int RawRead(void *p_handle, char *p_buf, off_t seek, size_t count, size_t *p_read, int *p_errno); static int RawOptionsHelp(const char **pp_help); static int RawOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); static int RawGetInfofileContent(void *p_handle, const char **pp_info_buf); static const char* RawGetErrorMessage(int err_num); static int RawFreeBuffer(void *p_buf); #endif // LIBXMOUNT_INPUT_RAW_H diff --git a/trunk/libxmount_morphing/libxmount_morphing.h b/trunk/libxmount_morphing/libxmount_morphing.h index 7c90343..36a9b06 100644 --- a/trunk/libxmount_morphing/libxmount_morphing.h +++ b/trunk/libxmount_morphing/libxmount_morphing.h @@ -1,248 +1,248 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_MORPHING_H #define LIBXMOUNT_MORPHING_H #define LIBXMOUNT_MORPHING_API_VERSION 1 #include #include // For alloc, calloc, free #include // For printf #include // For int*_t and uint*_t #include // For va_*, vprintf #include // For PRI* #include "../libxmount/libxmount.h" /******************************************************************************* * Type defs ******************************************************************************/ //! Struct containing pointers to input image functions typedef struct s_LibXmountMorphingInputFunctions { //! Function to get the amount of input images /*! * \param p_count Count of input images * \return 0 on success */ int (*ImageCount)(uint64_t *p_count); //! 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 */ int (*Size)(uint64_t image, uint64_t *p_size); //! 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 */ int (*Read)(uint64_t image, char *p_buf, off_t offset, size_t count, size_t *p_read); } ts_LibXmountMorphingInputFunctions, *pts_LibXmountMorphingInputFunctions; //! Structure containing pointers to the lib's functions typedef struct s_LibXmountMorphingFunctions { //! Function to initialize handle /*! * This function is called once to allow the lib to alloc any needed * structures before other functions that rely upon a valid handle are called * (for ex. OptionsParse or Morph). * * The p_format parameter specifies one of the morphing functions returned by * LibXmount_Morphing_GetSupportedMorphFunctions() which should be used for * this handle. * * \param pp_handle Pointer to store handle to * \param p_type Morph type to use * \param debug If set to 1, print debugging infos to stdout * \return 0 on success or error code */ int (*CreateHandle)(void **pp_handle, const char *p_type, uint8_t debug); //! Function to destroy handle /*! * In this function, any structures allocated with CreateHandle should be * freed. It is generally the last function called before unloading of lib * happens. * * By convention, after this function has been called, *pp_handle must be * NULL. * * \param pp_handle Pointer to store handle to * \return 0 on success or error code */ int (*DestroyHandle)(void **pp_handle); //! Function to start morphing /*! * Begins to morph input image * * \param p_handle Handle * \param p_input_functions ts_LibXmountInputFunctions structure * \return 0 on success or error code */ int (*Morph)(void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions); //! Function to get the size of the morphed data /*! * \param p_handle Handle to the opened image * \param p_size Pointer to store input image's size to * \return 0 on success or error code */ int (*Size)(void *p_handle, uint64_t *p_size); //! Function to read data from an input image /*! * Reads count bytes at offset from input image and copies them into memory * starting at the address of p_buf. Memory is pre-allocated to as much bytes * as should be read. * * \param p_handle Handle to the opened 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 */ int (*Read)(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read); //! Function to get a help message for any supported lib-specific options /*! * Calling this function should return a string containing help messages for * any supported lib-specific options. Lines should be formated as follows: * * " option : description\n" * * Returned string will be freed by the caller using FreeBuffer(). * * If there is no help text, this function must return NULL in pp_help. * * \param Pointer to a string to return help text * \return 0 on success or error code on error */ int (*OptionsHelp)(const char **pp_help); //! Function to parse any lib-specific options /*! * This function is called with the options given with the --inopts parameter. * All contained options are for the lib. If errors or unknown options are * found, this function should fail and return an error message in pp_error. * pp_error will be freed by the caller by using FreeBuffer. * * \param p_handle Handle to the opened image * \param p_options String with specified options * \param pp_error Pointer to a string with error message * \return 0 on success or error code and error message */ int (*OptionsParse)(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); //! Function to get content to add to the info file /*! * The returned string is added to xmount's info file. This function is only * called once when the info file is generated. The returned string is then * freed with a call to FreeBuffer. * * \param p_handle Handle to the opened image * \param pp_info_buf Pointer to store the null-terminated content * \return 0 on success or error code */ int (*GetInfofileContent)(void *p_handle, const char **pp_info_buf); //! Function to get an error message /*! * This function should translate an error code that was previously returned * by one of the library functions into a human readable error message. * * By convention, this function must always return a valid pointer to a * NULL-terminated string! * * \param err_num Error code as returned by lib */ const char* (*GetErrorMessage)(int err_num); //! Function to free buffers that were allocated by lib /*! * \param p_buf Buffer to free */ void (*FreeBuffer)(void *p_buf); } ts_LibXmountMorphingFunctions, *pts_LibXmountMorphingFunctions; /******************************************************************************* * API functions ******************************************************************************/ //! Get library API version /*! * This function should return the value of LIBXMOUNT_MORPHING_API_VERSION * * \return Supported version */ uint8_t LibXmount_Morphing_GetApiVersion(); typedef uint8_t (*t_LibXmount_Morphing_GetApiVersion)(); //! Get a list of supported morphing functions /*! * Gets a list of supported morphing functions. These is the string * specified with xmount's --morph command line option. The returned * string must be a constant vector of morphing functions split by \0 chars. To * mark the end of the vector, a single \0 must be used. * * As an example, "first\0second\0\0" would be a correct string to return for * a lib supporting two morphing functions. * * \return Vector containing supported morphing functions */ const char* LibXmount_Morphing_GetSupportedTypes(); typedef const char* (*t_LibXmount_Morphing_GetSupportedTypes)(); //! Get the lib's s_LibXmountMorphingFunctions structure /*! * This function should set the members of the given * s_LibXmountMorphingFunctions structure to the internal lib functions. All * members have to be set. * * \param p_functions s_LibXmountMorphingFunctions structure to fill */ void LibXmount_Morphing_GetFunctions(pts_LibXmountMorphingFunctions p_functions); typedef void (*t_LibXmount_Morphing_GetFunctions)(pts_LibXmountMorphingFunctions); #endif // LIBXMOUNT_MORPHING_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.c b/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.c index 6813d67..f3dfd09 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.c +++ b/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.c @@ -1,297 +1,297 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include #include "../libxmount_morphing.h" #include "libxmount_morphing_combine.h" #define LOG_DEBUG(...) { \ LIBXMOUNT_LOG_DEBUG(p_combine_handle->debug,__VA_ARGS__); \ } /******************************************************************************* * LibXmount_Morphing API implementation ******************************************************************************/ /* * LibXmount_Morphing_GetApiVersion */ uint8_t LibXmount_Morphing_GetApiVersion() { return LIBXMOUNT_MORPHING_API_VERSION; } /* * LibXmount_Morphing_GetSupportedFormats */ const char* LibXmount_Morphing_GetSupportedTypes() { return "combine\0\0"; } /* * LibXmount_Morphing_GetFunctions */ void LibXmount_Morphing_GetFunctions(ts_LibXmountMorphingFunctions *p_functions) { p_functions->CreateHandle=&CombineCreateHandle; p_functions->DestroyHandle=&CombineDestroyHandle; p_functions->Morph=&CombineMorph; p_functions->Size=&CombineSize; p_functions->Read=&CombineRead; p_functions->OptionsHelp=&CombineOptionsHelp; p_functions->OptionsParse=&CombineOptionsParse; p_functions->GetInfofileContent=&CombineGetInfofileContent; p_functions->GetErrorMessage=&CombineGetErrorMessage; p_functions->FreeBuffer=&CombineFreeBuffer; } /******************************************************************************* * Private ******************************************************************************/ /* * CombineCreateHandle */ static int CombineCreateHandle(void **pp_handle, const char *p_format, uint8_t debug) { pts_CombineHandle p_combine_handle; // Alloc new handle p_combine_handle=malloc(sizeof(ts_CombineHandle)); if(p_combine_handle==NULL) return COMBINE_MEMALLOC_FAILED; // Init handle values p_combine_handle->debug=debug; p_combine_handle->input_images_count=0; p_combine_handle->p_input_functions=NULL; p_combine_handle->morphed_image_size=0; LOG_DEBUG("Created new LibXmount_Morphing_Combine handle\n"); // Return new handle *pp_handle=p_combine_handle; return COMBINE_OK; } /* * CombineDestroyHandle */ static int CombineDestroyHandle(void **pp_handle) { pts_CombineHandle p_combine_handle=(pts_CombineHandle)*pp_handle; LOG_DEBUG("Destroying LibXmount_Morphing_Combine handle\n"); // Free handle free(p_combine_handle); *pp_handle=NULL; return COMBINE_OK; } /* * CombineMorph */ static int CombineMorph(void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions) { pts_CombineHandle p_combine_handle=(pts_CombineHandle)p_handle; int ret; uint64_t input_image_size; LOG_DEBUG("Initializing LibXmount_Morphing_Combine\n"); // Set input functions and get image count p_combine_handle->p_input_functions=p_input_functions; if(p_combine_handle-> p_input_functions-> ImageCount(&p_combine_handle->input_images_count)!=0) { return COMBINE_CANNOT_GET_IMAGECOUNT; } // Calculate morphed image size for(uint64_t i=0;iinput_images_count;i++) { ret=p_combine_handle-> p_input_functions-> Size(i,&input_image_size); if(ret!=0) return COMBINE_CANNOT_GET_IMAGESIZE; LOG_DEBUG("Adding %" PRIu64 " bytes from image %" PRIu64 "\n", input_image_size, i); p_combine_handle->morphed_image_size+=input_image_size; } LOG_DEBUG("Total morphed image size is %" PRIu64 " bytes\n", p_combine_handle->morphed_image_size); return COMBINE_OK; } /* * CombineSize */ static int CombineSize(void *p_handle, uint64_t *p_size) { *p_size=((pts_CombineHandle)(p_handle))->morphed_image_size; return COMBINE_OK; } /* * CombineRead */ static int CombineRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read) { pts_CombineHandle p_combine_handle=(pts_CombineHandle)p_handle; uint64_t cur_input_image=0; uint64_t cur_input_image_size=0; off_t cur_offset=offset; int ret; size_t cur_count; size_t read; LOG_DEBUG("Reading %zu bytes at offset %zu from morphed image\n", count, offset); // Make sure read parameters are within morphed image bounds if(offset>=p_combine_handle->morphed_image_size || offset+count>p_combine_handle->morphed_image_size) { return COMBINE_READ_BEYOND_END_OF_IMAGE; } // Search starting image to read from ret=p_combine_handle->p_input_functions->Size(cur_input_image, &cur_input_image_size); while(ret==0 && cur_offset>=cur_input_image_size) { cur_offset-=cur_input_image_size; cur_input_image++; ret=p_combine_handle->p_input_functions->Size(cur_input_image, &cur_input_image_size); } if(ret!=0) return COMBINE_CANNOT_GET_IMAGESIZE; // Init p_read *p_read=0; // Read data while(cur_input_imageinput_images_count && count!=0) { // Get current input image size ret=p_combine_handle->p_input_functions->Size(cur_input_image, &cur_input_image_size); if(ret!=0) return COMBINE_CANNOT_GET_IMAGESIZE; // Calculate how many bytes to read from current input image if(cur_offset+count>cur_input_image_size) { cur_count=cur_input_image_size-cur_offset; } else { cur_count=count; } LOG_DEBUG("Reading %zu bytes at offset %zu from input image %" PRIu64 "\n", cur_count, cur_offset, cur_input_image); // Read bytes ret=p_combine_handle->p_input_functions-> Read(cur_input_image, p_buf, cur_offset, cur_count, &read); if(ret!=0 || read!=cur_count) return COMBINE_CANNOT_READ_DATA; p_buf+=cur_count; cur_offset=0; count-=cur_count; cur_input_image++; (*p_read)+=cur_count; } if(count!=0) return COMBINE_CANNOT_READ_DATA; return COMBINE_OK; } /* * CombineOptionsHelp */ static int CombineOptionsHelp(const char **pp_help) { *pp_help=NULL; return COMBINE_OK; } /* * CombineOptionsParse */ static int CombineOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error) { return COMBINE_OK; } /* * CombineGetInfofileContent */ static int CombineGetInfofileContent(void *p_handle, const char **pp_info_buf) { *pp_info_buf=NULL; return COMBINE_OK; } /* * CombineGetErrorMessage */ static const char* CombineGetErrorMessage(int err_num) { switch(err_num) { case COMBINE_MEMALLOC_FAILED: return "Unable to allocate memory"; break; case COMBINE_CANNOT_GET_IMAGECOUNT: return "Unable to get input image count"; break; case COMBINE_CANNOT_GET_IMAGESIZE: return "Unable to get input image size"; break; case COMBINE_READ_BEYOND_END_OF_IMAGE: return "Unable to read data: Attempt to read past EOF"; break; case COMBINE_CANNOT_READ_DATA: return "Unable to read data"; break; default: return "Unknown error"; } } /* * CombineFreeBuffer */ static void CombineFreeBuffer(void *p_buf) { free(p_buf); } diff --git a/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.h b/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.h index f359824..41cb760 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.h +++ b/trunk/libxmount_morphing/libxmount_morphing_combine/libxmount_morphing_combine.h @@ -1,67 +1,67 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_MORPHING_COMBINE_H #define LIBXMOUNT_MORPHING_COMBINE_H /******************************************************************************* * Enums, type defs, etc... ******************************************************************************/ enum { COMBINE_OK=0, COMBINE_MEMALLOC_FAILED, COMBINE_CANNOT_GET_IMAGECOUNT, COMBINE_CANNOT_GET_IMAGESIZE, COMBINE_READ_BEYOND_END_OF_IMAGE, COMBINE_CANNOT_READ_DATA }; typedef struct s_CombineHandle { uint8_t debug; uint64_t input_images_count; pts_LibXmountMorphingInputFunctions p_input_functions; uint64_t morphed_image_size; } ts_CombineHandle, *pts_CombineHandle; /******************************************************************************* * Forward declarations ******************************************************************************/ static int CombineCreateHandle(void **pp_handle, const char *p_format, uint8_t debug); static int CombineDestroyHandle(void **pp_handle); static int CombineMorph(void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions); static int CombineSize(void *p_handle, uint64_t *p_size); static int CombineRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read); static int CombineOptionsHelp(const char **pp_help); static int CombineOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); static int CombineGetInfofileContent(void *p_handle, const char **pp_info_buf); static const char* CombineGetErrorMessage(int err_num); static void CombineFreeBuffer(void *p_buf); #endif // LIBXMOUNT_MORPHING_COMBINE_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.c b/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.c index 0d468ba..3ced503 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.c +++ b/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.c @@ -1,374 +1,374 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include #include "../libxmount_morphing.h" #include "libxmount_morphing_raid.h" #define LOG_DEBUG(...) { \ LIBXMOUNT_LOG_DEBUG(p_raid_handle->debug,__VA_ARGS__); \ } /******************************************************************************* * LibXmount_Morphing API implementation ******************************************************************************/ /* * LibXmount_Morphing_GetApiVersion */ uint8_t LibXmount_Morphing_GetApiVersion() { return LIBXMOUNT_MORPHING_API_VERSION; } /* * LibXmount_Morphing_GetSupportedFormats */ const char* LibXmount_Morphing_GetSupportedTypes() { return "raid0\0\0"; } /* * LibXmount_Morphing_GetFunctions */ void LibXmount_Morphing_GetFunctions(ts_LibXmountMorphingFunctions *p_functions) { p_functions->CreateHandle=&RaidCreateHandle; p_functions->DestroyHandle=&RaidDestroyHandle; p_functions->Morph=&RaidMorph; p_functions->Size=&RaidSize; p_functions->Read=&RaidRead; p_functions->OptionsHelp=&RaidOptionsHelp; p_functions->OptionsParse=&RaidOptionsParse; p_functions->GetInfofileContent=&RaidGetInfofileContent; p_functions->GetErrorMessage=&RaidGetErrorMessage; p_functions->FreeBuffer=&RaidFreeBuffer; } /******************************************************************************* * Private ******************************************************************************/ /* * RaidCreateHandle */ static int RaidCreateHandle(void **pp_handle, const char *p_format, uint8_t debug) { pts_RaidHandle p_raid_handle; // Alloc new handle p_raid_handle=malloc(sizeof(ts_RaidHandle)); if(p_raid_handle==NULL) return RAID_MEMALLOC_FAILED; // Init handle values p_raid_handle->debug=debug; p_raid_handle->input_images_count=0; p_raid_handle->chunk_size=RAID_DEFAULT_CHUNKSIZE; p_raid_handle->chunks_per_image=0; p_raid_handle->p_input_functions=NULL; p_raid_handle->morphed_image_size=0; LOG_DEBUG("Created new LibXmount_Morphing_Raid handle\n"); // Return new handle *pp_handle=p_raid_handle; return RAID_OK; } /* * RaidDestroyHandle */ static int RaidDestroyHandle(void **pp_handle) { pts_RaidHandle p_raid_handle=(pts_RaidHandle)*pp_handle; LOG_DEBUG("Destroying LibXmount_Morphing_Raid handle\n"); // Free handle free(p_raid_handle); *pp_handle=NULL; return RAID_OK; } /* * RaidMorph */ static int RaidMorph(void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions) { pts_RaidHandle p_raid_handle=(pts_RaidHandle)p_handle; int ret; uint64_t input_image_size; uint64_t chunks_per_image; LOG_DEBUG("Initializing LibXmount_Morphing_Raid\n"); // Set input functions and get image count p_raid_handle->p_input_functions=p_input_functions; if(p_raid_handle-> p_input_functions-> ImageCount(&p_raid_handle->input_images_count)!=0) { return RAID_CANNOT_GET_IMAGECOUNT; } // Calculate chunks per image for(uint64_t i=0;iinput_images_count;i++) { ret=p_raid_handle-> p_input_functions-> Size(i,&input_image_size); if(ret!=0) return RAID_CANNOT_GET_IMAGESIZE; chunks_per_image=input_image_size/p_raid_handle->chunk_size; LOG_DEBUG("Image %" PRIu64 " can hold a maximum of %" PRIu64 " chunks of %" PRIu32 " bytes\n", i, chunks_per_image, p_raid_handle->chunk_size); // The smallest image determines how many chunks are availbale on all images if(p_raid_handle->chunks_per_image==0) { p_raid_handle->chunks_per_image=chunks_per_image; } else if(chunks_per_imagechunks_per_image) { p_raid_handle->chunks_per_image=chunks_per_image; } } LOG_DEBUG("Smallest image holds %" PRIu64 " chunks of %" PRIu32 " bytes\n", p_raid_handle->chunks_per_image, p_raid_handle->chunk_size); // Calculate total raid capacity based on smallest disk p_raid_handle->morphed_image_size= p_raid_handle->chunks_per_image* p_raid_handle->chunk_size*p_raid_handle->input_images_count; LOG_DEBUG("Total raid capacity is %" PRIu64 " bytes\n", p_raid_handle->morphed_image_size); return RAID_OK; } /* * RaidSize */ static int RaidSize(void *p_handle, uint64_t *p_size) { *p_size=((pts_RaidHandle)(p_handle))->morphed_image_size; return RAID_OK; } /* * RaidRead */ static int RaidRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read) { pts_RaidHandle p_raid_handle=(pts_RaidHandle)p_handle; uint64_t cur_chunk; uint64_t cur_image; off_t cur_chunk_offset; off_t cur_image_offset; size_t cur_count; int ret; size_t read; LOG_DEBUG("Reading %zu bytes at offset %zu from morphed image\n", count, offset); // Make sure read parameters are within morphed image bounds if(offset>=p_raid_handle->morphed_image_size || offset+count>p_raid_handle->morphed_image_size) { return RAID_READ_BEYOND_END_OF_IMAGE; } // Calculate starting chunk, and chunk offset cur_chunk=offset/p_raid_handle->chunk_size; cur_chunk_offset=offset-(cur_chunk*p_raid_handle->chunk_size); // Init p_read *p_read=0; while(count!=0) { // Calculate image and image offset to read from cur_image=cur_chunk%p_raid_handle->input_images_count; cur_image_offset= (cur_chunk/p_raid_handle->input_images_count)*p_raid_handle->chunk_size; // Calculate how many bytes to read from current chunk if(cur_chunk_offset+count>p_raid_handle->chunk_size) { cur_count=p_raid_handle->chunk_size-cur_chunk_offset; } else { cur_count=count; } LOG_DEBUG("Reading %zu bytes at offset %zu from image %" PRIu64 " (chunk %" PRIu64 ")\n", cur_count, cur_image_offset+cur_chunk_offset, cur_image, cur_chunk); // Read bytes ret=p_raid_handle->p_input_functions-> Read(cur_image, p_buf, cur_image_offset+cur_chunk_offset, cur_count, &read); if(ret!=0 || read!=cur_count) return RAID_CANNOT_READ_DATA; p_buf+=cur_count; cur_chunk_offset=0; count-=cur_count; cur_chunk++; (*p_read)+=cur_count; } return RAID_OK; } /* * RaidOptionsHelp */ static int RaidOptionsHelp(const char **pp_help) { int ok; char *p_buf; ok=asprintf(&p_buf, " raid_chunksize : Specify the chunk size to use in bytes. " "Defaults to 524288 (512k).\n"); if(ok<0 || p_buf==NULL) { *pp_help=NULL; return RAID_MEMALLOC_FAILED; } *pp_help=p_buf; return RAID_OK; } /* * RaidOptionsParse */ static int RaidOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error) { pts_RaidHandle p_raid_handle=(pts_RaidHandle)p_handle; int ok; uint32_t uint32value; char *p_buf; for(uint32_t i=0;ip_key,"raid_chunksize")) { // Convert value to uint32 uint32value=StrToUint32(pp_options[i]->p_value,&ok); if(ok==0 || uint32value==0) { // Conversion failed, generate error message and return ok=asprintf(&p_buf, "Unable to parse value '%s' of '%s' as valid 32bit number", pp_options[i]->p_value, pp_options[i]->p_key); if(ok<0 || *pp_error==NULL) { *pp_error=NULL; return RAID_MEMALLOC_FAILED; } *pp_error=p_buf; return RAID_CANNOT_PARSE_OPTION; } LOG_DEBUG("Setting chunk size to %" PRIu32 " bytes\n",uint32value); // Conversion ok, save value and mark option as valid p_raid_handle->chunk_size=uint32value; pp_options[i]->valid=1; continue; } } return RAID_OK; } /* * RaidGetInfofileContent */ static int RaidGetInfofileContent(void *p_handle, const char **pp_info_buf) { pts_RaidHandle p_raid_handle=(pts_RaidHandle)p_handle; int ret; char *p_buf; ret=asprintf(&p_buf, "Simulating RAID level 0 over %" PRIu64 " disks.\n" "Chunk size: %" PRIu32 " bytes\n" "Chunks per disk: %" PRIu64 "\n" "Total capacity: %" PRIu64 " bytes (%0.3f GiB)\n", p_raid_handle->input_images_count, p_raid_handle->chunk_size, p_raid_handle->chunks_per_image, p_raid_handle->morphed_image_size, p_raid_handle->morphed_image_size/(1024.0*1024.0*1024.0)); if(ret<0 || *pp_info_buf==NULL) return RAID_MEMALLOC_FAILED; *pp_info_buf=p_buf; return RAID_OK; } /* * RaidGetErrorMessage */ static const char* RaidGetErrorMessage(int err_num) { switch(err_num) { case RAID_MEMALLOC_FAILED: return "Unable to allocate memory"; break; case RAID_CANNOT_GET_IMAGECOUNT: return "Unable to get input image count"; break; case RAID_CANNOT_GET_IMAGESIZE: return "Unable to get input image size"; break; case RAID_READ_BEYOND_END_OF_IMAGE: return "Unable to read data: Attempt to read past EOF"; break; case RAID_CANNOT_READ_DATA: return "Unable to read data"; break; case RAID_CANNOT_PARSE_OPTION: return "Unable to parse library option"; break; default: return "Unknown error"; } } /* * RaidFreeBuffer */ static void RaidFreeBuffer(void *p_buf) { free(p_buf); } diff --git a/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.h b/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.h index 4c6dea9..0baf85e 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.h +++ b/trunk/libxmount_morphing/libxmount_morphing_raid/libxmount_morphing_raid.h @@ -1,71 +1,71 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_MORPHING_RAID_H #define LIBXMOUNT_MORPHING_RAID_H /******************************************************************************* * Enums, type defs, etc... ******************************************************************************/ enum { RAID_OK=0, RAID_MEMALLOC_FAILED, RAID_CANNOT_GET_IMAGECOUNT, RAID_CANNOT_GET_IMAGESIZE, RAID_READ_BEYOND_END_OF_IMAGE, RAID_CANNOT_READ_DATA, RAID_CANNOT_PARSE_OPTION }; #define RAID_DEFAULT_CHUNKSIZE 512*1024 typedef struct s_RaidHandle { uint8_t debug; uint64_t input_images_count; uint32_t chunk_size; uint64_t chunks_per_image; pts_LibXmountMorphingInputFunctions p_input_functions; uint64_t morphed_image_size; } ts_RaidHandle, *pts_RaidHandle; /******************************************************************************* * Forward declarations ******************************************************************************/ static int RaidCreateHandle(void **pp_handle, const char *p_format, uint8_t debug); static int RaidDestroyHandle(void **pp_handle); static int RaidMorph(void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions); static int RaidSize(void *p_handle, uint64_t *p_size); static int RaidRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read); static int RaidOptionsHelp(const char **pp_help); static int RaidOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); static int RaidGetInfofileContent(void *p_handle, const char **pp_info_buf); static const char* RaidGetErrorMessage(int err_num); static void RaidFreeBuffer(void *p_buf); #endif // LIBXMOUNT_MORPHING_RAID_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.c b/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.c index 698e1e4..2e153c9 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.c +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.c @@ -1,406 +1,406 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include "fat_functions.h" #include "libxmount_morphing_unallocated_retvalues.h" #include // For memset #define LOG_DEBUG(...) { \ LIBXMOUNT_LOG_DEBUG(p_fat_handle->debug,__VA_ARGS__); \ } /* * ReadFatHeader */ int ReadFatHeader(pts_FatHandle p_fat_handle, pts_LibXmountMorphingInputFunctions p_input_functions, uint8_t debug) { pts_FatVH p_fat_vh; int ret; size_t bytes_read; uint32_t root_dir_sectors; uint32_t fat_size; uint32_t total_sectors; uint32_t data_sectors; uint32_t cluster_count; // Init FAT handle memset(p_fat_handle,0,sizeof(ts_FatHandle)); p_fat_handle->fat_type=FatType_Unknown; p_fat_handle->debug=debug; LOG_DEBUG("Trying to read FAT volume header\n"); // Alloc buffer for header p_fat_vh=calloc(1,sizeof(ts_FatVH)); if(p_fat_vh==NULL) return UNALLOCATED_MEMALLOC_FAILED; // Read VH from input image ret=p_input_functions->Read(0, (char*)(p_fat_vh), 0, sizeof(ts_FatVH), &bytes_read); if(ret!=0 || bytes_read!=sizeof(ts_FatVH)) { free(p_fat_vh); return UNALLOCATED_FAT_CANNOT_READ_HEADER; } // Convert values to host endianness (FAT values are always stored in little // endian) p_fat_vh->bytes_per_sector=le16toh(p_fat_vh->bytes_per_sector); p_fat_vh->reserved_sectors=le16toh(p_fat_vh->reserved_sectors); p_fat_vh->root_entry_count=le16toh(p_fat_vh->root_entry_count); p_fat_vh->total_sectors_16=le16toh(p_fat_vh->total_sectors_16); p_fat_vh->fat16_sectors=le16toh(p_fat_vh->fat16_sectors); p_fat_vh->total_sectors_32=le32toh(p_fat_vh->total_sectors_32); p_fat_vh->fat32_sectors=le32toh(p_fat_vh->fat32_sectors); LOG_DEBUG("FAT VH jump instruction 1: 0x%02X\n",p_fat_vh->jump_inst[0]); LOG_DEBUG("FAT bytes per sector: %" PRIu16 "\n", p_fat_vh->bytes_per_sector); LOG_DEBUG("FAT sectors per cluster: %" PRIu8 "\n", p_fat_vh->sectors_per_cluster); LOG_DEBUG("FAT reserved sectors: %" PRIu16 "\n", p_fat_vh->reserved_sectors); LOG_DEBUG("FAT count: %" PRIu8 "\n",p_fat_vh->fat_count); LOG_DEBUG("FAT root entry count: %" PRIu16 "\n", p_fat_vh->root_entry_count); LOG_DEBUG("FAT media type: %02X\n",p_fat_vh->media_type); LOG_DEBUG("FAT total sector count (16bit): %" PRIu16 "\n", p_fat_vh->total_sectors_16); LOG_DEBUG("FAT sectors per FAT (16bit): %" PRIu16 "\n", p_fat_vh->fat16_sectors); LOG_DEBUG("FAT total sector count (32bit): %" PRIu32 "\n", p_fat_vh->total_sectors_32); LOG_DEBUG("FAT sectors per FAT (32bit): %" PRIu32 "\n", p_fat_vh->fat32_sectors); // Check header values if((p_fat_vh->jump_inst[0]!=0xEB && p_fat_vh->jump_inst[0]!=0xE9) || p_fat_vh->bytes_per_sector==0 || p_fat_vh->bytes_per_sector%512!=0 || p_fat_vh->sectors_per_cluster==0 || p_fat_vh->sectors_per_cluster%2!=0 || p_fat_vh->reserved_sectors==0 || p_fat_vh->fat_count==0 || (p_fat_vh->total_sectors_16==0 && p_fat_vh->total_sectors_32==0) || (p_fat_vh->total_sectors_16!=0 && p_fat_vh->total_sectors_32!=0)) { free(p_fat_vh); return UNALLOCATED_FAT_INVALID_HEADER; } LOG_DEBUG("Determining FAT type\n"); // Determine the count of sectors occupied by the root directory root_dir_sectors=((p_fat_vh->root_entry_count*32)+ (p_fat_vh->bytes_per_sector-1))/p_fat_vh->bytes_per_sector; // Determine the count of sectors in the data region if(p_fat_vh->fat16_sectors!=0) fat_size=p_fat_vh->fat16_sectors; else fat_size=p_fat_vh->fat32_sectors; if(p_fat_vh->total_sectors_16!=0) total_sectors=p_fat_vh->total_sectors_16; else total_sectors=p_fat_vh->total_sectors_32; data_sectors=total_sectors-(p_fat_vh->reserved_sectors+ (p_fat_vh->fat_count*fat_size)+root_dir_sectors); // Determine the count of clusters cluster_count=data_sectors/p_fat_vh->sectors_per_cluster; // Determine FAT type if(cluster_count<4085) { LOG_DEBUG("FAT is of unsupported type FAT12\n"); free(p_fat_vh); return UNALLOCATED_FAT_UNSUPPORTED_FS_TYPE; } else if(cluster_count<65525) { LOG_DEBUG("FAT is of type FAT16\n"); p_fat_handle->fat_type=FatType_Fat16; } else { LOG_DEBUG("FAT is of type FAT32\n"); p_fat_handle->fat_type=FatType_Fat32; } // TODO: What about newer version of FAT like exFAT etc... ?? p_fat_handle->p_fat_vh=p_fat_vh; return UNALLOCATED_OK; } /* * FreeFatHeader */ void FreeFatHeader(pts_FatHandle p_fat_handle) { if(p_fat_handle==NULL) return; if(p_fat_handle->p_fat_vh!=NULL) free(p_fat_handle->p_fat_vh); switch(p_fat_handle->fat_type) { case FatType_Fat16: { if(p_fat_handle->u_fat.p_fat16!=NULL) free(p_fat_handle->u_fat.p_fat16); break; } case FatType_Fat32: { if(p_fat_handle->u_fat.p_fat32!=NULL) free(p_fat_handle->u_fat.p_fat32); break; } case FatType_Unknown: default: break; } } /* * ReadFat */ int ReadFat(pts_FatHandle p_fat_handle, pts_LibXmountMorphingInputFunctions p_input_functions) { pts_FatVH p_fat_vh=p_fat_handle->p_fat_vh; int ret; size_t fat_size; off_t fat_offset; size_t bytes_read; LOG_DEBUG("Trying to read FAT\n"); // Determine FAT size if(p_fat_vh->fat16_sectors!=0) fat_size=p_fat_vh->fat16_sectors; else fat_size=p_fat_vh->fat32_sectors; fat_size*=p_fat_vh->bytes_per_sector; // Calculate FAT offset fat_offset=p_fat_vh->reserved_sectors*p_fat_vh->bytes_per_sector; LOG_DEBUG("FAT consists of %zu bytes starting at offset %zu\n", fat_size, fat_offset); // Alloc buffer and Read FAT if(p_fat_handle->fat_type==FatType_Fat32) { // Alloc buffer p_fat_handle->u_fat.p_fat32=(uint32_t*)calloc(1,fat_size); if(p_fat_handle->u_fat.p_fat32==NULL) return UNALLOCATED_MEMALLOC_FAILED; // Read FAT ret=p_input_functions->Read(0, (char*)(p_fat_handle->u_fat.p_fat32), fat_offset, fat_size, &bytes_read); if(ret!=0 || bytes_read!=fat_size) { free(p_fat_handle->u_fat.p_fat32); p_fat_handle->u_fat.p_fat32=NULL; return UNALLOCATED_FAT_CANNOT_READ_FAT; } // Convert FAT to host endianness for(uint64_t i=0;iu_fat.p_fat32[i]=le32toh(p_fat_handle->u_fat.p_fat32[i]); } } else { // Alloc buffer p_fat_handle->u_fat.p_fat16=(uint16_t*)calloc(1,fat_size); if(p_fat_handle->u_fat.p_fat16==NULL) return UNALLOCATED_MEMALLOC_FAILED; // Read FAT ret=p_input_functions->Read(0, (char*)(p_fat_handle->u_fat.p_fat16), fat_offset, fat_size, &bytes_read); if(ret!=0 || bytes_read!=fat_size) { free(p_fat_handle->u_fat.p_fat16); p_fat_handle->u_fat.p_fat16=NULL; return UNALLOCATED_FAT_CANNOT_READ_FAT; } // Convert FAT to host endianness for(uint64_t i=0;iu_fat.p_fat16[i]=le16toh(p_fat_handle->u_fat.p_fat16[i]); } } LOG_DEBUG("FAT read successfully\n"); return UNALLOCATED_OK; } /* * BuildFatBlockMap */ int BuildFatBlockMap(pts_FatHandle p_fat_handle, uint64_t **pp_free_block_map, uint64_t *p_free_block_map_size, uint64_t *p_block_size) { pts_FatVH p_fat_vh=p_fat_handle->p_fat_vh; uint64_t data_offset; uint64_t total_clusters; uint64_t *p_free_block_map=NULL; uint64_t free_block_map_size=0; LOG_DEBUG("Searching unallocated FAT clusters\n"); // Calculate offset of first data cluster data_offset=p_fat_vh->reserved_sectors+ (((p_fat_vh->root_entry_count*32)+(p_fat_vh->bytes_per_sector-1))/ p_fat_vh->bytes_per_sector); if(p_fat_vh->fat16_sectors!=0) { data_offset+=(p_fat_vh->fat_count*p_fat_vh->fat16_sectors); } else { data_offset+=(p_fat_vh->fat_count*p_fat_vh->fat32_sectors); } data_offset*=p_fat_vh->bytes_per_sector; // Calculate total amount of data clusters if(p_fat_vh->total_sectors_16!=0) total_clusters=p_fat_vh->total_sectors_16; else total_clusters=p_fat_vh->total_sectors_32; total_clusters-=(data_offset/p_fat_vh->bytes_per_sector); total_clusters/=p_fat_vh->sectors_per_cluster; // Add 2 clusters as clusters 0 and 1 do not exist total_clusters+=2; LOG_DEBUG("Filesystem contains a total of %" PRIu64 " (2-%" PRIu64 ") " " data clusters starting at offset %" PRIu64 "\n", total_clusters-2, total_clusters-1, data_offset); // Save offset of every unallocated cluster in block map // Clusters 0 and 1 can not hold data in a FAT fs if(p_fat_handle->fat_type==FatType_Fat32) { for(uint64_t cur_cluster=2;cur_clusteru_fat.p_fat32[cur_cluster] & 0x0FFFFFFF)==0 || (p_fat_handle->u_fat.p_fat32[cur_cluster] & 0x0FFFFFFF)==0x0FFFFFF7) { p_free_block_map=realloc(p_free_block_map, (free_block_map_size+1)*sizeof(uint64_t)); if(p_free_block_map==NULL) return UNALLOCATED_MEMALLOC_FAILED; p_free_block_map[free_block_map_size]=data_offset+((cur_cluster-2)* p_fat_vh->bytes_per_sector*p_fat_vh->sectors_per_cluster); LOG_DEBUG("Cluster %" PRIu64 " is unallocated " "(FAT value 0x%04X, Image offset %" PRIu64 ")\n", cur_cluster, p_fat_handle->u_fat.p_fat32[cur_cluster], p_free_block_map[free_block_map_size]); free_block_map_size++; } else { LOG_DEBUG("Cluster %" PRIu64 " is allocated (FAT value 0x%08X)\n", cur_cluster, p_fat_handle->u_fat.p_fat32[cur_cluster]); } } } else { for(uint64_t cur_cluster=2;cur_clusteru_fat.p_fat16[cur_cluster] & 0x0FFF)==0 || (p_fat_handle->u_fat.p_fat16[cur_cluster] & 0x0FFF)==0x0FF7) { p_free_block_map=realloc(p_free_block_map, (free_block_map_size+1)*sizeof(uint64_t)); if(p_free_block_map==NULL) return UNALLOCATED_MEMALLOC_FAILED; p_free_block_map[free_block_map_size]=data_offset+((cur_cluster-2)* p_fat_vh->sectors_per_cluster*p_fat_vh->bytes_per_sector); LOG_DEBUG("Cluster %" PRIu64 " is unallocated " "(FAT value 0x%04X, Image offset %" PRIu64 ")\n", cur_cluster, p_fat_handle->u_fat.p_fat16[cur_cluster], p_free_block_map[free_block_map_size]); free_block_map_size++; } else { LOG_DEBUG("Cluster %" PRIu64 " is allocated (FAT value 0x%04X)\n", cur_cluster, p_fat_handle->u_fat.p_fat16[cur_cluster]); } } } LOG_DEBUG("Found %" PRIu64 " unallocated FAT clusters\n", free_block_map_size); // Free FAT as it is no longer needed if(p_fat_handle->fat_type==FatType_Fat32) { free(p_fat_handle->u_fat.p_fat32); p_fat_handle->u_fat.p_fat32=NULL; } else { free(p_fat_handle->u_fat.p_fat16); p_fat_handle->u_fat.p_fat16=NULL; } // TODO: There may be trailing unclustered sectors. Add them? *pp_free_block_map=p_free_block_map; *p_free_block_map_size=free_block_map_size; *p_block_size=p_fat_vh->bytes_per_sector*p_fat_vh->sectors_per_cluster; return UNALLOCATED_OK; } /* * GetFatInfos */ int GetFatInfos(pts_FatHandle p_fat_handle, char **pp_buf) { pts_FatVH p_fat_vh=p_fat_handle->p_fat_vh; int ret; char *p_buf=NULL; char *p_fat_type; switch(p_fat_handle->fat_type) { case FatType_Fat16: p_fat_type="FAT16"; break; case FatType_Fat32: p_fat_type="FAT32"; break; case FatType_Unknown: default: p_fat_type="Unknown"; } ret=asprintf(&p_buf, "FAT filesystem type: %s\n" "FAT bytes per sector: %" PRIu16 "\n" "FAT sectors per cluster: %" PRIu8 "\n" "FAT reserved sectors: %" PRIu16 "\n" "FAT count: %" PRIu8 "\n" "FAT root entry count: %" PRIu16 "\n" "FAT media type: 0x%02X\n" "FAT total sector count (16bit): %" PRIu16 "\n" "FAT sectors per FAT (16bit): %" PRIu16 "\n" "FAT total sector count (32bit): %" PRIu32 "\n" "FAT sectors per FAT (32bit): %" PRIu32, p_fat_type, p_fat_vh->bytes_per_sector, p_fat_vh->sectors_per_cluster, p_fat_vh->reserved_sectors, p_fat_vh->fat_count, p_fat_vh->root_entry_count, p_fat_vh->media_type, p_fat_vh->total_sectors_16, p_fat_vh->fat16_sectors, p_fat_vh->total_sectors_32, p_fat_vh->fat32_sectors); // Check if asprintf worked if(ret<0 || p_buf==NULL) return UNALLOCATED_MEMALLOC_FAILED; *pp_buf=p_buf; return UNALLOCATED_OK; } diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.h b/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.h index c908268..0522080 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.h +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/fat_functions.h @@ -1,72 +1,72 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef FAT_FUNCTIONS_H #define FAT_FUNCTIONS_H #include "../libxmount_morphing.h" // Supported FAT types typedef enum e_FatType { FatType_Unknown=0, FatType_Fat16, FatType_Fat32 } te_FatType; // Needed parts of the FAT volume header typedef struct s_FatVH { uint8_t jump_inst[3]; uint8_t oem_name[8]; uint16_t bytes_per_sector; uint8_t sectors_per_cluster; uint16_t reserved_sectors; uint8_t fat_count; uint16_t root_entry_count; uint16_t total_sectors_16; uint8_t media_type; uint16_t fat16_sectors; uint64_t unused01; uint32_t total_sectors_32; // Following value is only valid for FAT32 uint32_t fat32_sectors; } __attribute__ ((packed)) ts_FatVH, *pts_FatVH; // FAT handle typedef struct s_FatHandle { te_FatType fat_type; pts_FatVH p_fat_vh; union { uint16_t *p_fat16; uint32_t *p_fat32; } u_fat; uint8_t debug; } ts_FatHandle, *pts_FatHandle; int ReadFatHeader(pts_FatHandle p_fat_handle, pts_LibXmountMorphingInputFunctions p_input_functions, uint8_t debug); void FreeFatHeader(pts_FatHandle p_fat_handle); int ReadFat(pts_FatHandle p_fat_handle, pts_LibXmountMorphingInputFunctions p_input_functions); int BuildFatBlockMap(pts_FatHandle p_fat_handle, uint64_t **pp_free_block_map, uint64_t *p_free_block_map_size, uint64_t *p_block_size); int GetFatInfos(pts_FatHandle p_fat_handle, char **pp_buf); #endif // FAT_FUNCTIONS_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.c b/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.c index 167764b..a1d3025 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.c +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.c @@ -1,260 +1,260 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include "hfs_functions.h" #include "libxmount_morphing_unallocated_retvalues.h" #define LOG_DEBUG(...) { \ LIBXMOUNT_LOG_DEBUG(p_hfs_handle->debug,__VA_ARGS__); \ } /* * ReadHfsHeader */ int ReadHfsHeader(pts_HfsHandle p_hfs_handle, pts_LibXmountMorphingInputFunctions p_input_functions, uint8_t debug) { pts_HfsVH p_hfs_vh; int ret; size_t bytes_read; pts_HfsExtend p_extend; // Init HFS handle p_hfs_handle->p_hfs_vh=NULL; p_hfs_handle->p_alloc_file=NULL; p_hfs_handle->debug=debug; LOG_DEBUG("Trying to read HFS volume header\n"); // Alloc buffer for header p_hfs_vh=calloc(1,sizeof(ts_HfsVH)); if(p_hfs_vh==NULL) return UNALLOCATED_MEMALLOC_FAILED; // Read VH from input image ret=p_input_functions->Read(0, (char*)(p_hfs_vh), HFS_VH_OFFSET, sizeof(ts_HfsVH), &bytes_read); if(ret!=0 || bytes_read!=sizeof(ts_HfsVH)) { free(p_hfs_vh); p_hfs_vh=NULL; return UNALLOCATED_HFS_CANNOT_READ_HEADER; } // Convert VH to host endianness (HFS values are always stored in big endian) p_hfs_vh->signature=be16toh(p_hfs_vh->signature); p_hfs_vh->version=be16toh(p_hfs_vh->version); p_hfs_vh->block_size=be32toh(p_hfs_vh->block_size); p_hfs_vh->total_blocks=be32toh(p_hfs_vh->total_blocks); p_hfs_vh->free_blocks=be32toh(p_hfs_vh->free_blocks); p_hfs_vh->alloc_file_size=be64toh(p_hfs_vh->alloc_file_size); p_hfs_vh->alloc_file_clump_size=be32toh(p_hfs_vh->alloc_file_clump_size); p_hfs_vh->alloc_file_total_blocks=be32toh(p_hfs_vh->alloc_file_total_blocks); for(int i=0;i<8;i++) { p_extend=&(p_hfs_vh->alloc_file_extends[i]); p_extend->start_block=be32toh(p_extend->start_block); p_extend->block_count=be32toh(p_extend->block_count); } LOG_DEBUG("HFS VH signature: 0x%04X\n",p_hfs_vh->signature); LOG_DEBUG("HFS VH version: %" PRIu16 "\n",p_hfs_vh->version); LOG_DEBUG("HFS block size: %" PRIu32 " bytes\n",p_hfs_vh->block_size); LOG_DEBUG("HFS total blocks: %" PRIu32 "\n",p_hfs_vh->total_blocks); LOG_DEBUG("HFS free blocks: %" PRIu32 "\n",p_hfs_vh->free_blocks); LOG_DEBUG("HFS allocation file size: %" PRIu64 " bytes\n", p_hfs_vh->alloc_file_size); LOG_DEBUG("HFS allocation file blocks: %" PRIu32 "\n", p_hfs_vh->alloc_file_total_blocks); // Check header signature and version if(p_hfs_vh->signature!=HFS_VH_SIGNATURE || p_hfs_vh->version!=HFS_VH_VERSION) { free(p_hfs_vh); p_hfs_vh=NULL; return UNALLOCATED_HFS_INVALID_HEADER; } // We currently only support HFS+ p_hfs_handle->hfs_type=HfsType_HfsPlus; LOG_DEBUG("HFS volume header read successfully\n"); p_hfs_handle->p_hfs_vh=p_hfs_vh; return UNALLOCATED_OK; } /* * FreeHfsHeader */ void FreeHfsHeader(pts_HfsHandle p_hfs_handle) { if(p_hfs_handle==NULL) return; if(p_hfs_handle->p_hfs_vh!=NULL) free(p_hfs_handle->p_hfs_vh); if(p_hfs_handle->p_alloc_file!=NULL) free(p_hfs_handle->p_alloc_file); } /* * ReadHfsAllocFile */ int ReadHfsAllocFile(pts_HfsHandle p_hfs_handle, pts_LibXmountMorphingInputFunctions p_input_functions) { uint8_t *p_alloc_file; pts_HfsExtend p_extend; int ret; char *p_buf; size_t bytes_read; uint64_t total_bytes_read=0; LOG_DEBUG("Trying to read HFS allocation file\n"); // Alloc buffer for file p_alloc_file=calloc(1,p_hfs_handle->p_hfs_vh->alloc_file_size); if(p_alloc_file==NULL) return UNALLOCATED_MEMALLOC_FAILED; // Loop over extends and read data p_buf=(char*)(p_alloc_file); for(int i=0;i<8;i++) { p_extend=&(p_hfs_handle->p_hfs_vh->alloc_file_extends[i]); // If start_block and block_count are zero, we parsed last extend if(p_extend->start_block==0 && p_extend->block_count==0) break; LOG_DEBUG("Extend %d contains %" PRIu32 " block(s) starting with block %" PRIu32 "\n", i, p_extend->block_count, p_extend->start_block); // Read data for(uint32_t ii=0;iiblock_count;ii++) { LOG_DEBUG("Reading %" PRIu32 " bytes from block %" PRIu32 " at offset %" PRIu64 "\n", p_hfs_handle->p_hfs_vh->block_size, p_extend->start_block+ii, (uint64_t)((p_extend->start_block+ii)* p_hfs_handle->p_hfs_vh->block_size)); ret=p_input_functions->Read(0, p_buf, (p_extend->start_block+ii)* p_hfs_handle->p_hfs_vh->block_size, p_hfs_handle->p_hfs_vh->block_size, &bytes_read); if(ret!=0 || bytes_read!=p_hfs_handle->p_hfs_vh->block_size) { free(p_alloc_file); return UNALLOCATED_HFS_CANNOT_READ_ALLOC_FILE; } p_buf+=p_hfs_handle->p_hfs_vh->block_size; total_bytes_read+=p_hfs_handle->p_hfs_vh->block_size; } } // Alloc files with more than 8 extends aren't supported yet if(total_bytes_read!=p_hfs_handle->p_hfs_vh->alloc_file_size) { free(p_alloc_file); return UNALLOCATED_HFS_ALLOC_FILE_HAS_TOO_MUCH_EXTENDS; } LOG_DEBUG("HFS allocation file read successfully\n"); p_hfs_handle->p_alloc_file=p_alloc_file; return UNALLOCATED_OK; } /* * BuildHfsBlockMap */ int BuildHfsBlockMap(pts_HfsHandle p_hfs_handle, uint64_t **pp_free_block_map, uint64_t *p_free_block_map_size, uint64_t *p_block_size) { uint64_t *p_free_block_map=NULL; uint64_t free_block_map_size=0; LOG_DEBUG("Searching unallocated HFS blocks\n"); // Save offset of every unallocated block in block map for(uint32_t cur_block=0; cur_blockp_hfs_vh->total_blocks; cur_block++) { if((p_hfs_handle->p_alloc_file[cur_block/8] & (1<<(7-(cur_block%8))))==0) { p_free_block_map=realloc(p_free_block_map, (free_block_map_size+1)*sizeof(uint64_t)); if(p_free_block_map==NULL) return UNALLOCATED_MEMALLOC_FAILED; p_free_block_map[free_block_map_size]= cur_block*p_hfs_handle->p_hfs_vh->block_size; free_block_map_size++; } } LOG_DEBUG("Found %" PRIu64 " unallocated HFS blocks\n", free_block_map_size); if(p_hfs_handle->p_hfs_vh->free_blocks!=free_block_map_size) { LIBXMOUNT_LOG_WARNING("According to VH, there should be %" PRIu64 " unallocated blocks but I found %" PRIu64 "\n", p_hfs_handle->p_hfs_vh->free_blocks, free_block_map_size); } // Free alloc file as it is no longer needed free(p_hfs_handle->p_alloc_file); p_hfs_handle->p_alloc_file=NULL; *pp_free_block_map=p_free_block_map; *p_free_block_map_size=free_block_map_size; *p_block_size=p_hfs_handle->p_hfs_vh->block_size; return UNALLOCATED_OK; } /* * GetHfsInfos */ int GetHfsInfos(pts_HfsHandle p_hfs_handle, char **pp_buf) { pts_HfsVH p_hfs_vh=p_hfs_handle->p_hfs_vh; char *p_buf=NULL; int ret; ret=asprintf(&p_buf, "HFS filesystem type: HFS+\n" "HFS VH signature: 0x%04X\n" "HFS VH version: %" PRIu16 "\n" "HFS block size: %" PRIu32 " bytes\n" "HFS total blocks: %" PRIu32 "\n" "HFS free blocks: %" PRIu32 "\n" "HFS allocation file size: %" PRIu64 " bytes\n" "HFS allocation file blocks: %" PRIu32, p_hfs_vh->signature, p_hfs_vh->version, p_hfs_vh->block_size, p_hfs_vh->total_blocks, p_hfs_vh->free_blocks, p_hfs_vh->alloc_file_size, p_hfs_vh->alloc_file_total_blocks); // Check if asprintf worked if(ret<0 || p_buf==NULL) return UNALLOCATED_MEMALLOC_FAILED; *pp_buf=p_buf; return UNALLOCATED_OK; } diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.h b/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.h index 6e25652..4bf6e0a 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.h +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/hfs_functions.h @@ -1,73 +1,73 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef HFS_FUNCTIONS_H #define HFS_FUNCTIONS_H #include "../libxmount_morphing.h" // Supported HFS types typedef enum e_HfsType { HfsType_HfsPlus=0 } te_HfsType; // HFS extend typedef struct s_HfsExtend { uint32_t start_block; uint32_t block_count; } __attribute__ ((packed)) ts_HfsExtend, *pts_HfsExtend; // Needed parts of the HFS volume header #define HFS_VH_OFFSET 1024 #define HFS_VH_SIGNATURE 0x482b //"H+" #define HFS_VH_VERSION 4 typedef struct s_HfsVH { uint16_t signature; // "H+" uint16_t version; // Currently 4 for HFS+ uint32_t unused01[9]; uint32_t block_size; uint32_t total_blocks; uint32_t free_blocks; uint32_t unused02[15]; uint64_t alloc_file_size; uint32_t alloc_file_clump_size; uint32_t alloc_file_total_blocks; ts_HfsExtend alloc_file_extends[8]; } __attribute__ ((packed)) ts_HfsVH, *pts_HfsVH; // HFS handle typedef struct s_HfsHandle { te_HfsType hfs_type; pts_HfsVH p_hfs_vh; uint8_t *p_alloc_file; uint8_t debug; } ts_HfsHandle, *pts_HfsHandle; int ReadHfsHeader(pts_HfsHandle p_hfs_handle, pts_LibXmountMorphingInputFunctions p_input_functions, uint8_t debug); void FreeHfsHeader(pts_HfsHandle p_hfs_handle); int ReadHfsAllocFile(pts_HfsHandle p_hfs_handle, pts_LibXmountMorphingInputFunctions p_input_functions); int BuildHfsBlockMap(pts_HfsHandle p_hfs_handle, uint64_t **pp_free_block_map, uint64_t *p_free_block_map_size, uint64_t *p_block_size); int GetHfsInfos(pts_HfsHandle p_hfs_handle, char **pp_buf); #endif // HFS_FUNCTIONS_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.c b/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.c index 35fb6ae..fe24778 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.c +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.c @@ -1,511 +1,511 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include #include "libxmount_morphing_unallocated.h" #include "libxmount_morphing_unallocated_retvalues.h" #define LOG_DEBUG(...) { \ LIBXMOUNT_LOG_DEBUG(p_unallocated_handle->debug,__VA_ARGS__); \ } /******************************************************************************* * LibXmount_Morphing API implementation ******************************************************************************/ /* * LibXmount_Morphing_GetApiVersion */ uint8_t LibXmount_Morphing_GetApiVersion() { return LIBXMOUNT_MORPHING_API_VERSION; } /* * LibXmount_Morphing_GetSupportedFormats */ const char* LibXmount_Morphing_GetSupportedTypes() { return "unallocated\0\0"; } /* * LibXmount_Morphing_GetFunctions */ void LibXmount_Morphing_GetFunctions(ts_LibXmountMorphingFunctions *p_functions) { p_functions->CreateHandle=&UnallocatedCreateHandle; p_functions->DestroyHandle=&UnallocatedDestroyHandle; p_functions->Morph=&UnallocatedMorph; p_functions->Size=&UnallocatedSize; p_functions->Read=&UnallocatedRead; p_functions->OptionsHelp=&UnallocatedOptionsHelp; p_functions->OptionsParse=&UnallocatedOptionsParse; p_functions->GetInfofileContent=&UnallocatedGetInfofileContent; p_functions->GetErrorMessage=&UnallocatedGetErrorMessage; p_functions->FreeBuffer=&UnallocatedFreeBuffer; } /******************************************************************************* * Private ******************************************************************************/ /* * UnallocatedCreateHandle */ static int UnallocatedCreateHandle(void **pp_handle, const char *p_format, uint8_t debug) { pts_UnallocatedHandle p_unallocated_handle; // Alloc new handle. Using calloc in order to set everything to 0x00 p_unallocated_handle=calloc(1,sizeof(ts_UnallocatedHandle)); if(p_unallocated_handle==NULL) return UNALLOCATED_MEMALLOC_FAILED; // Init handle values p_unallocated_handle->debug=debug; p_unallocated_handle->fs_type=UnallocatedFsType_Unknown; LOG_DEBUG("Created new LibXmount_Morphing_Unallocated handle\n"); // Return new handle *pp_handle=p_unallocated_handle; return UNALLOCATED_OK; } /* * UnallocatedDestroyHandle */ static int UnallocatedDestroyHandle(void **pp_handle) { pts_UnallocatedHandle p_unallocated_handle=(pts_UnallocatedHandle)*pp_handle; LOG_DEBUG("Destroying LibXmount_Morphing_Unallocated handle\n"); // TODO: Return if p_unallocated_handle==NULL // Free fs handle switch(p_unallocated_handle->fs_type) { case UnallocatedFsType_Hfs: { FreeHfsHeader(&(p_unallocated_handle->u_fs.hfs_handle)); break; } case UnallocatedFsType_Fat: { FreeFatHeader(&(p_unallocated_handle->u_fs.fat_handle)); break; } case UnallocatedFsType_Unknown: default: break; } // Free handle values and handle if(p_unallocated_handle->p_free_block_map!=NULL) free(p_unallocated_handle->p_free_block_map); free(p_unallocated_handle); *pp_handle=NULL; return UNALLOCATED_OK; } /* * UnallocatedMorph */ static int UnallocatedMorph( void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions) { pts_UnallocatedHandle p_unallocated_handle=(pts_UnallocatedHandle)p_handle; uint64_t input_images_count; int ret; LOG_DEBUG("Initializing LibXmount_Morphing_Unallocated\n"); // Set input functions and get image count p_unallocated_handle->p_input_functions=p_input_functions; if(p_unallocated_handle-> p_input_functions-> ImageCount(&input_images_count)!=0) { return UNALLOCATED_CANNOT_GET_IMAGECOUNT; } // Make sure there is exactly one input image if(input_images_count==0 || input_images_count>1) { return UNALLOCATED_WRONG_INPUT_IMAGE_COUNT; } // Read filesystem header switch(p_unallocated_handle->fs_type) { case UnallocatedFsType_Hfs: { // Read HFS VH ret=ReadHfsHeader(&(p_unallocated_handle->u_fs.hfs_handle), p_unallocated_handle->p_input_functions, p_unallocated_handle->debug); if(ret!=UNALLOCATED_OK) return ret; break; } case UnallocatedFsType_Fat: { // Read FAT VH ret=ReadFatHeader(&(p_unallocated_handle->u_fs.fat_handle), p_unallocated_handle->p_input_functions, p_unallocated_handle->debug); if(ret!=UNALLOCATED_OK) return ret; break; } case UnallocatedFsType_Unknown: { // Filesystem wasn't specified. Try to autodetect it by reading all // available fs headers LOG_DEBUG("Autodetecting filesystem\n"); LOG_DEBUG("Trying HFS\n"); ret=ReadHfsHeader(&(p_unallocated_handle->u_fs.hfs_handle), p_unallocated_handle->p_input_functions, p_unallocated_handle->debug); if(ret==UNALLOCATED_OK) { LOG_DEBUG("Detected HFS fs\n"); p_unallocated_handle->fs_type=UnallocatedFsType_Hfs; break; } LOG_DEBUG("Trying FAT\n"); ret=ReadFatHeader(&(p_unallocated_handle->u_fs.fat_handle), p_unallocated_handle->p_input_functions, p_unallocated_handle->debug); if(ret==UNALLOCATED_OK) { LOG_DEBUG("Detected FAT fs\n"); p_unallocated_handle->fs_type=UnallocatedFsType_Fat; break; } LOG_DEBUG("Unable to autodetect fs\n"); return UNALLOCATED_NO_SUPPORTED_FS_DETECTED; } default: { return UNALLOCATED_INTERNAL_ERROR; } } // Extract unallocated blocks from input image switch(p_unallocated_handle->fs_type) { case UnallocatedFsType_Hfs: { // Read HFS alloc file ret=ReadHfsAllocFile(&(p_unallocated_handle->u_fs.hfs_handle), p_unallocated_handle->p_input_functions); if(ret!=UNALLOCATED_OK) return ret; // Build free block map ret=BuildHfsBlockMap(&(p_unallocated_handle->u_fs.hfs_handle), &(p_unallocated_handle->p_free_block_map), &(p_unallocated_handle->free_block_map_size), &(p_unallocated_handle->block_size)); if(ret!=UNALLOCATED_OK) return ret; break; } case UnallocatedFsType_Fat: { // Read FAT ret=ReadFat(&(p_unallocated_handle->u_fs.fat_handle), p_unallocated_handle->p_input_functions); if(ret!=UNALLOCATED_OK) return ret; // Build free block map ret=BuildFatBlockMap(&(p_unallocated_handle->u_fs.fat_handle), &(p_unallocated_handle->p_free_block_map), &(p_unallocated_handle->free_block_map_size), &(p_unallocated_handle->block_size)); if(ret!=UNALLOCATED_OK) return ret; break; } case UnallocatedFsType_Unknown: default: return UNALLOCATED_INTERNAL_ERROR; } // Calculate morphed image size p_unallocated_handle->morphed_image_size=p_unallocated_handle->block_size* p_unallocated_handle->free_block_map_size; LOG_DEBUG("Total size of unallocated blocks is %" PRIu64 " bytes\n", p_unallocated_handle->morphed_image_size); return UNALLOCATED_OK; } /* * UnallocatedSize */ static int UnallocatedSize(void *p_handle, uint64_t *p_size) { *p_size=((pts_UnallocatedHandle)(p_handle))->morphed_image_size; return UNALLOCATED_OK; } /* * UnallocatedRead */ static int UnallocatedRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read) { pts_UnallocatedHandle p_unallocated_handle=(pts_UnallocatedHandle)p_handle; uint64_t cur_block; off_t cur_block_offset; off_t cur_image_offset; size_t cur_count; int ret; size_t bytes_read; LOG_DEBUG("Reading %zu bytes at offset %zu from morphed image\n", count, offset); // Make sure read parameters are within morphed image bounds if(offset>=p_unallocated_handle->morphed_image_size || offset+count>p_unallocated_handle->morphed_image_size) { return UNALLOCATED_READ_BEYOND_END_OF_IMAGE; } // Calculate starting block and block offset cur_block=offset/p_unallocated_handle->block_size; cur_block_offset=offset-(cur_block*p_unallocated_handle->block_size); // Init p_read *p_read=0; while(count!=0) { // Calculate input image offset to read from cur_image_offset= p_unallocated_handle->p_free_block_map[cur_block]+cur_block_offset; // Calculate how many bytes to read from current block if(cur_block_offset+count>p_unallocated_handle->block_size) { cur_count=p_unallocated_handle->block_size-cur_block_offset; } else { cur_count=count; } LOG_DEBUG("Reading %zu bytes at offset %zu (block %" PRIu64 ")\n", cur_count, cur_image_offset+cur_block_offset, cur_block); // Read bytes ret=p_unallocated_handle->p_input_functions-> Read(0, p_buf, cur_image_offset+cur_block_offset, cur_count, &bytes_read); if(ret!=0 || bytes_read!=cur_count) return UNALLOCATED_CANNOT_READ_DATA; p_buf+=cur_count; cur_block_offset=0; count-=cur_count; cur_block++; (*p_read)+=cur_count; } return UNALLOCATED_OK; } /* * UnallocatedOptionsHelp */ static int UnallocatedOptionsHelp(const char **pp_help) { int ok; char *p_buf; ok=asprintf(&p_buf, " unallocated_fs : Specify the filesystem to extract " "unallocated blocks from. Supported filesystems are: " "'hfs', 'fat'. " "Default: autodetect.\n"); if(ok<0 || p_buf==NULL) { *pp_help=NULL; return UNALLOCATED_MEMALLOC_FAILED; } *pp_help=p_buf; return UNALLOCATED_OK; } /* * UnallocatedOptionsParse */ static int UnallocatedOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error) { pts_UnallocatedHandle p_unallocated_handle=(pts_UnallocatedHandle)p_handle; int ok; char *p_buf; for(uint32_t i=0;ip_key,"unallocated_fs")==0) { if(strcmp(pp_options[i]->p_value,"hfs")==0) { p_unallocated_handle->fs_type=UnallocatedFsType_Hfs; } else if(strcmp(pp_options[i]->p_value,"fat")==0) { p_unallocated_handle->fs_type=UnallocatedFsType_Fat; } else { ok=asprintf(&p_buf, "Unsupported filesystem '%s' specified", pp_options[i]->p_value); if(ok<0 || p_buf==NULL) { *pp_error=NULL; return UNALLOCATED_MEMALLOC_FAILED; } *pp_error=p_buf; return UNALLOCATED_UNSUPPORTED_FS_SPECIFIED; } LOG_DEBUG("Setting fs to %s\n",pp_options[i]->p_value); pp_options[i]->valid=1; } } return UNALLOCATED_OK; } /* * UnallocatedGetInfofileContent */ static int UnallocatedGetInfofileContent(void *p_handle, const char **pp_info_buf) { pts_UnallocatedHandle p_unallocated_handle=(pts_UnallocatedHandle)p_handle; int ret=-1; char *p_fs_buf=NULL; char *p_buf=NULL; switch(p_unallocated_handle->fs_type) { case UnallocatedFsType_Hfs: { ret=GetHfsInfos(&(p_unallocated_handle->u_fs.hfs_handle),&p_fs_buf); break; } case UnallocatedFsType_Fat: { ret=GetFatInfos(&(p_unallocated_handle->u_fs.fat_handle),&p_fs_buf); break; } case UnallocatedFsType_Unknown: default: return UNALLOCATED_INTERNAL_ERROR; } if(ret!=UNALLOCATED_OK) return ret; if(p_fs_buf!=NULL) { ret=asprintf(&p_buf, "%s\n" "Discovered free blocks: %" PRIu64 "\n" "Total unallocated size: %" PRIu64 " bytes (%0.3f GiB)\n", p_fs_buf, p_unallocated_handle->free_block_map_size, p_unallocated_handle->free_block_map_size* p_unallocated_handle->block_size, (p_unallocated_handle->free_block_map_size* p_unallocated_handle->block_size)/(1024.0*1024.0*1024.0)); free(p_fs_buf); } else { ret=asprintf(&p_buf, "Discovered free blocks: %" PRIu64 "\n" "Total unallocated size: %" PRIu64 " bytes (%0.3f GiB)\n", p_unallocated_handle->free_block_map_size, p_unallocated_handle->free_block_map_size* p_unallocated_handle->block_size, (p_unallocated_handle->free_block_map_size* p_unallocated_handle->block_size)/(1024.0*1024.0*1024.0)); } // Check if asprintf worked if(ret<0 || p_buf==NULL) return UNALLOCATED_MEMALLOC_FAILED; *pp_info_buf=p_buf; return UNALLOCATED_OK; } /* * UnallocatedGetErrorMessage */ static const char* UnallocatedGetErrorMessage(int err_num) { switch(err_num) { case UNALLOCATED_MEMALLOC_FAILED: return "Unable to allocate memory"; break; case UNALLOCATED_NO_SUPPORTED_FS_DETECTED: return "Unable to detect a supported file system"; break; case UNALLOCATED_UNSUPPORTED_FS_SPECIFIED: return "Unsupported fs specified"; break; case UNALLOCATED_INTERNAL_ERROR: return "Internal error"; break; case UNALLOCATED_CANNOT_GET_IMAGECOUNT: return "Unable to get input image count"; break; case UNALLOCATED_WRONG_INPUT_IMAGE_COUNT: return "Only 1 input image is supported"; break; case UNALLOCATED_CANNOT_GET_IMAGESIZE: return "Unable to get input image size"; break; case UNALLOCATED_READ_BEYOND_END_OF_IMAGE: return "Unable to read data: Attempt to read past EOF"; break; case UNALLOCATED_CANNOT_READ_DATA: return "Unable to read data"; break; case UNALLOCATED_CANNOT_PARSE_OPTION: return "Unable to parse library option"; break; // HFS errors case UNALLOCATED_HFS_CANNOT_READ_HEADER: return "Unable to read HFS volume header"; break; case UNALLOCATED_HFS_INVALID_HEADER: return "Found invalid HFS volume header"; break; case UNALLOCATED_HFS_CANNOT_READ_ALLOC_FILE: return "Unable to read HFS allocation file"; break; case UNALLOCATED_HFS_ALLOC_FILE_HAS_TOO_MUCH_EXTENDS: return "HFS allocation file has more than 8 extends. " "This is unsupported"; break; // FAT errors case UNALLOCATED_FAT_CANNOT_READ_HEADER: return "Unable to read FAT volume header"; break; case UNALLOCATED_FAT_INVALID_HEADER: return "Found invalid FAT volume header"; break; case UNALLOCATED_FAT_UNSUPPORTED_FS_TYPE: return "Found unsupported FAT type"; break; case UNALLOCATED_FAT_CANNOT_READ_FAT: return "Unable to read FAT"; break; default: return "Unknown error"; } } /* * UnallocatedFreeBuffer */ static void UnallocatedFreeBuffer(void *p_buf) { free(p_buf); } diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.h b/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.h index e0dcae5..d508e59 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.h +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated.h @@ -1,83 +1,83 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_MORPHING_UNALLOCATED_H #define LIBXMOUNT_MORPHING_UNALLOCATED_H #include "../libxmount_morphing.h" #include "hfs_functions.h" #include "fat_functions.h" #include "ntfs_functions.h" /******************************************************************************* * Enums, type defs, etc... ******************************************************************************/ // Supported fs types typedef enum e_UnallocatedFsType { UnallocatedFsType_Unknown=0, UnallocatedFsType_Hfs, UnallocatedFsType_Fat } te_UnallocatedFsType; // Handle typedef struct s_UnallocatedHandle { uint8_t debug; te_UnallocatedFsType fs_type; pts_LibXmountMorphingInputFunctions p_input_functions; uint64_t block_size; uint64_t free_block_map_size; uint64_t *p_free_block_map; uint64_t morphed_image_size; union { ts_HfsHandle hfs_handle; ts_FatHandle fat_handle; ts_NtfsHandle ntfs_handle; } u_fs; } ts_UnallocatedHandle, *pts_UnallocatedHandle; /******************************************************************************* * Forward declarations ******************************************************************************/ static int UnallocatedCreateHandle(void **pp_handle, const char *p_format, uint8_t debug); static int UnallocatedDestroyHandle(void **pp_handle); static int UnallocatedMorph( void *p_handle, pts_LibXmountMorphingInputFunctions p_input_functions); static int UnallocatedSize(void *p_handle, uint64_t *p_size); static int UnallocatedRead(void *p_handle, char *p_buf, off_t offset, size_t count, size_t *p_read); static int UnallocatedOptionsHelp(const char **pp_help); static int UnallocatedOptionsParse(void *p_handle, uint32_t options_count, const pts_LibXmountOptions *pp_options, const char **pp_error); static int UnallocatedGetInfofileContent(void *p_handle, const char **pp_info_buf); static const char* UnallocatedGetErrorMessage(int err_num); static void UnallocatedFreeBuffer(void *p_buf); #endif // LIBXMOUNT_MORPHING_UNALLOCATED_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated_retvalues.h b/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated_retvalues.h index af24799..f65f75d 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated_retvalues.h +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/libxmount_morphing_unallocated_retvalues.h @@ -1,47 +1,47 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef LIBXMOUNT_MORPHING_UNALLOCATED_RETVALUES_H #define LIBXMOUNT_MORPHING_UNALLOCATED_RETVALUES_H // Error codes enum { UNALLOCATED_OK=0, UNALLOCATED_MEMALLOC_FAILED, UNALLOCATED_NO_SUPPORTED_FS_DETECTED, UNALLOCATED_UNSUPPORTED_FS_SPECIFIED, UNALLOCATED_INTERNAL_ERROR, UNALLOCATED_CANNOT_GET_IMAGECOUNT, UNALLOCATED_WRONG_INPUT_IMAGE_COUNT, UNALLOCATED_CANNOT_GET_IMAGESIZE, UNALLOCATED_READ_BEYOND_END_OF_IMAGE, UNALLOCATED_CANNOT_READ_DATA, UNALLOCATED_CANNOT_PARSE_OPTION, // HFS return values UNALLOCATED_HFS_CANNOT_READ_HEADER, UNALLOCATED_HFS_INVALID_HEADER, UNALLOCATED_HFS_CANNOT_READ_ALLOC_FILE, UNALLOCATED_HFS_ALLOC_FILE_HAS_TOO_MUCH_EXTENDS, // FAT return values UNALLOCATED_FAT_CANNOT_READ_HEADER, UNALLOCATED_FAT_INVALID_HEADER, UNALLOCATED_FAT_UNSUPPORTED_FS_TYPE, UNALLOCATED_FAT_CANNOT_READ_FAT }; #endif // LIBXMOUNT_MORPHING_UNALLOCATED_RETVALUES_H diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.c b/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.c index 78a6d7e..8401c38 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.c +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.c @@ -1,34 +1,34 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include "ntfs_functions.h" #include "libxmount_morphing_unallocated_retvalues.h" #define LOG_DEBUG(...) { \ LIBXMOUNT_LOG_DEBUG(p_ntfs_handle->debug,__VA_ARGS__); \ } /* * ReadNtfsHeader */ int ReadNtfsHeader(pts_NtfsHandle p_ntfs_handle, pts_LibXmountMorphingInputFunctions p_input_functions, uint8_t debug) { } diff --git a/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.h b/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.h index 66f9e38..c5ee7c6 100644 --- a/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.h +++ b/trunk/libxmount_morphing/libxmount_morphing_unallocated/ntfs_functions.h @@ -1,55 +1,55 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef NTFS_FUNCTIONS_H #define NTFS_FUNCTIONS_H #include "../libxmount_morphing.h" // Needed parts of the FAT volume header typedef struct s_NtfsVH { uint8_t jump_inst[3]; uint8_t oem_name[8]; uint16_t bytes_per_sector; uint8_t sectors_per_cluster; uint16_t reserved_sectors; uint8_t null01; uint8_t null02; uint8_t null03; uint16_t unused01; uint8_t media_type; uint16_t null04; uint8_t unused02[14]; uint64_t total_sectors; uint64_t mft_cluster; uint64_t mftmirr_cluster; int16_t clusters_per_file_rec; int8_t clusters_per_index_buf; } __attribute__ ((packed)) ts_NtfsVH, *pts_NtfsVH; // NTFS handle typedef struct s_NtfsHandle { pts_NtfsVH p_ntfs_vh; uint8_t debug; } ts_NtfsHandle, *pts_NtfsHandle; int ReadNtfsHeader(pts_NtfsHandle p_ntfs_handle, pts_LibXmountMorphingInputFunctions p_input_functions, uint8_t debug); #endif // NTFS_FUNCTIONS_H diff --git a/trunk/src/macros.h b/trunk/src/macros.h index 179c3b0..80afafc 100644 --- a/trunk/src/macros.h +++ b/trunk/src/macros.h @@ -1,69 +1,69 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #ifndef MACROS_H #define MACROS_H #ifndef __APPLE__ #define FOPEN fopen64 #else // Apple does use fopen for fopen64 too #define FOPEN fopen #endif /* * Macros to alloc or realloc memory and check whether it worked or not */ #define XMOUNT_MALLOC(var,var_type,size) { \ (var)=(var_type)malloc(size); \ if((var)==NULL) { \ LOG_ERROR("Couldn't allocate memmory!\n"); \ exit(1); \ } \ } #define XMOUNT_REALLOC(var,var_type,size) { \ (var)=(var_type)realloc((var),size); \ if((var)==NULL) { \ LOG_ERROR("Couldn't allocate memmory!\n"); \ exit(1); \ } \ } /* * Macros for some often used string functions */ #define XMOUNT_STRSET(var1,var2) { \ XMOUNT_MALLOC(var1,char*,strlen(var2)+1) \ strcpy(var1,var2); \ } #define XMOUNT_STRNSET(var1,var2,size) { \ XMOUNT_MALLOC(var1,char*,(size)+1) \ strncpy(var1,var2,size); \ (var1)[size]='\0'; \ } #define XMOUNT_STRAPP(var1,var2) { \ XMOUNT_REALLOC(var1,char*,strlen(var1)+strlen(var2)+1) \ strcpy((var1)+strlen(var1),var2); \ } #define XMOUNT_STRNAPP(var1,var2,size) { \ XMOUNT_REALLOC(var1,char*,strlen(var1)+(size)+1) \ (var1)[strlen(var1)+(size)]='\0'; \ strncpy((var1)+strlen(var1),var2,size); \ } #endif // MACROS_H diff --git a/trunk/src/xmount.c b/trunk/src/xmount.c index 7c7439f..e658079 100755 --- a/trunk/src/xmount.c +++ b/trunk/src/xmount.c @@ -1,4121 +1,4121 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include #include #include #include #include #include // For PRI* #include #include // For dlopen, dlclose, dlsym #include // For opendir, readdir, closedir #include #include #include // For fstat #include #ifdef HAVE_LINUX_FS_H #include // For SEEK_* ?? #endif #if !defined(__APPLE__) && defined(HAVE_GRP_H) && defined(HAVE_PWD_H) #include // For getgrnam, struct group #include // For getpwuid, struct passwd #endif #include #include // For time #define FUSE_USE_VERSION 26 #include #include "xmount.h" #include "md5.h" #include "macros.h" #include "../libxmount/libxmount.h" #define XMOUNT_COPYRIGHT_NOTICE \ - "xmount v%s Copyright (c) 2008-2014 by Gillen Daniel " + "xmount v%s Copyright (c) 2008-2015 by Gillen Daniel " #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__); \ } /******************************************************************************* * Global vars ******************************************************************************/ //! Struct that contains various runtime configuration options static ts_XmountData glob_xmount; /******************************************************************************* * Forward declarations ******************************************************************************/ // Helper functions static void PrintUsage(char*); static void CheckFuseSettings(); static int ParseCmdLine(const int, char**); static int ExtractVirtFileNames(char*); static int GetMorphedImageSize(uint64_t*); static int GetVirtImageSize(uint64_t*); static int GetInputImageData(pts_InputImage, char*, off_t, size_t, size_t*); static int GetMorphedImageData(char*, off_t, size_t, size_t*); static int GetVirtImageData(char*, off_t, size_t); static int SetVdiFileHeaderData(char*, off_t, size_t); static int SetVhdFileHeaderData(char*, off_t, size_t); static int SetVirtImageData(const char*, off_t, size_t); static int CalculateInputImageHash(uint64_t*, uint64_t*); static int InitVirtVdiHeader(); static int InitVirtVhdHeader(); static int InitVirtualVmdkFile(); static int InitVirtImageInfoFile(); static int InitCacheFile(); static int LoadLibs(); static int FindInputLib(pts_InputImage); static int FindMorphingLib(); static void InitResources(); static void FreeResources(); 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 implementing FUSE functions static int FuseGetAttr(const char*, struct stat*); static int FuseMkDir(const char*, mode_t); static int FuseMkNod(const char*, mode_t, dev_t); static int FuseReadDir(const char*, void*, fuse_fill_dir_t, off_t, struct fuse_file_info*); static int FuseOpen(const char*, struct fuse_file_info*); static int FuseRead(const char*, char*, size_t, off_t, struct fuse_file_info*); static int FuseRename(const char*, const char*); static int FuseRmDir(const char*); static int FuseUnlink(const char*); //static int FuseStatFs(const char*, struct statvfs*); static int FuseWrite(const char *p_path, const char*, size_t, off_t, struct fuse_file_info*); /******************************************************************************* * 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] \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 : Specify fuse mount options. Will also disable " "automatic addition of FUSE's allow_other option!\n"); printf("\n"); printf(" xopts:\n"); printf(" --cache : Enable virtual write support.\n"); printf(" specifies the cache file to use.\n"); printf(" --in : Input image format and source file(s). " "May be specified multiple times.\n"); printf(" can be "); // List supported input formats first=1; for(uint32_t i=0;ip_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(" specifies the source file. If your image is split into " "multiple files, you have to specify them all!\n"); printf(" --inopts : Specify input library specific options.\n"); printf(" 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 : Morphing function to apply to input image(s). " "If not specified, defaults to \"combine\".\n"); printf(" can be "); // List supported morphing functions first=1; for(uint32_t i=0;ip_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 : Specify morphing library specific " "options.\n"); printf(" specifies a comma separated list of key=value options. " "See below for details.\n"); printf(" --offset : Move the output image data start bytes " "into the input image(s).\n"); printf(" --out : Output image format. If not specified, " "defaults to "); #ifdef __APPLE__ printf("\"dmg\".\n"); #else printf("\"raw\".\n"); #endif printf(" can be "); // List supported output formats printf("\"raw\", \"dmg\", \"vdi\", \"vhd\", \"vmdk\", \"vmdks\".\n"); printf(" --owcache : Same as --cache but overwrites " "existing cache file.\n"); printf(" --sizelimit : The data end of input image(s) is set to no " "more than 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 and morphing lib options for(uint32_t i=0;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 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"); } } //! 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 */ #define SUPPORT_DEPRECATED_IN 1 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; #ifdef SUPPORT_DEPRECATED_IN int use_old_in_syntax=FALSE; #endif // 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(i1 && *(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)p_type); free(p_input_image); return FALSE; } #endif if((i+2)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) { #ifndef SUPPORT_DEPRECATED_IN 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; #else use_old_in_syntax=TRUE; i++; continue; #endif } // 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)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;iip_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("\n"); exit(0); } else { LOG_ERROR("Unknown command line option \"%s\"\n",pp_argv[i]); return FALSE; } } i++; } #ifdef SUPPORT_DEPRECATED_IN if(use_old_in_syntax==TRUE && i<(argc-1)) { LOG_WARNING("You are using a deprecated --in option syntax which will be " "removed in the next release. Please see the man page " "on how to use the new syntax.\n"); while(i<(argc-1)) { 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++; } // 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 if(use_old_in_syntax==TRUE) { free(p_input_image->p_type); free(p_input_image); LOG_ERROR("You must specify an input image type, source file(s) and a " "mount point!\n"); return FALSE; } #endif // 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="); XMOUNT_STRAPP(glob_xmount.pp_fuse_argv[glob_xmount.fuse_argc-1], glob_xmount.input.pp_images[0]->pp_files[0]); } 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 virtual 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 ExtractVirtFileNames(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,"/"); if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { XMOUNT_STRSET(glob_xmount.output.vmdk.p_virtual_vmdk_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); if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { XMOUNT_STRAPP(glob_xmount.output.vmdk.p_virtual_vmdk_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)); if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { XMOUNT_STRNAPP(glob_xmount.output.vmdk.p_virtual_vmdk_path,p_orig_name, strlen(p_orig_name)-strlen(tmp)); } XMOUNT_STRAPP(glob_xmount.output.p_info_path,".info"); } // Add virtual file extensions switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,".dd"); break; case VirtImageType_DMG: XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,".dmg"); break; case VirtImageType_VDI: XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,".vdi"); break; case VirtImageType_VHD: XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,".vhd"); break; case VirtImageType_VMDK: case VirtImageType_VMDKS: XMOUNT_STRAPP(glob_xmount.output.p_virtual_image_path,".dd"); XMOUNT_STRAPP(glob_xmount.output.vmdk.p_virtual_vmdk_path,".vmdk"); break; default: LOG_ERROR("Unknown virtual image type!\n") return FALSE; } 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); if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { LOG_DEBUG("Set virtual vmdk name to \"%s\"\n", glob_xmount.output.vmdk.p_virtual_vmdk_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 virtual image /*! * \param p_size Pointer to an uint64_t to which the size will be written to * \return TRUE on success, FALSE on error */ static int GetVirtImageSize(uint64_t *p_size) { if(glob_xmount.output.image_size!=0) { *p_size=glob_xmount.output.image_size; return TRUE; } switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: case VirtImageType_DMG: case VirtImageType_VMDK: case VirtImageType_VMDKS: // Virtual image is a DD, DMG or VMDK file. Just return the size of the // original image if(!GetMorphedImageSize(p_size)) { LOG_ERROR("Couldn't get size of input image!\n") return FALSE; } break; case VirtImageType_VDI: // Virtual image is a VDI file. Get size of original image and add size // of VDI header etc. if(!GetMorphedImageSize(p_size)) { LOG_ERROR("Couldn't get size of input image!\n") return FALSE; } (*p_size)+=(sizeof(ts_VdiFileHeader)+ glob_xmount.output.vdi.vdi_block_map_size); break; case VirtImageType_VHD: // Virtual image is a VHD file. Get size of original image and add size // of VHD footer. if(!GetMorphedImageSize(p_size)) { LOG_ERROR("Couldn't get size of input image!\n") return FALSE; } (*p_size)+=sizeof(ts_VhdFileHeader); break; default: LOG_ERROR("Unsupported image type!\n") return FALSE; } glob_xmount.output.image_size=*p_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 GetInputImageData(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 GetMorphedImageData(char *p_buf, off_t offset, size_t size, size_t *p_read) { int ret; size_t to_read=0; size_t read; uint64_t image_size=0; // 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 0; } 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 %zu.\n", size, to_read); } else to_read=size; // Read data from morphed image ret=glob_xmount.morphing.p_functions->Read(glob_xmount.morphing.p_handle, p_buf, offset, to_read, &read); if(ret!=0) { LOG_ERROR("Couldn't read %zu bytes at offset %zu from morphed image: %s!\n", to_read, offset, glob_xmount.morphing.p_functions->GetErrorMessage(ret)); return -EIO; } *p_read=to_read; return TRUE; } //! Read data from virtual 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 */ static int GetVirtImageData(char *p_buf, off_t offset, size_t size) { uint32_t cur_block=0; uint64_t morphed_image_size, virt_image_size; size_t read, to_read=0, cur_to_read=0; off_t file_off=offset, block_off=0; size_t to_read_later=0; int ret; // Get virtual image size if(GetVirtImageSize(&virt_image_size)!=TRUE) { LOG_ERROR("Couldn't get size of virtual image!\n") return -EIO; } if(offset>=virt_image_size) { LOG_DEBUG("Offset %zu is at / beyond size of virtual image.\n",offset); return 0; } if(offset+size>virt_image_size) { LOG_DEBUG("Attempt to read data past EOF of virtual image. Corrected size " "from %zu to %zu.\n", size, virt_image_size-offset); size=virt_image_size-offset; } to_read=size; // Get morphed image size if(GetMorphedImageSize(&morphed_image_size)!=TRUE) { LOG_ERROR("Couldn't get morphed image size!") return -EIO; } // Read virtual image type specific data preceeding morphed image data switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: case VirtImageType_DMG: case VirtImageType_VMDK: case VirtImageType_VMDKS: break; case VirtImageType_VDI: if(file_offglob_xmount.output.vdi.vdi_header_size) { cur_to_read=glob_xmount.output.vdi.vdi_header_size-file_off; } else { cur_to_read=to_read; } if(glob_xmount.output.writable==TRUE && glob_xmount.cache.p_cache_header->VdiFileHeaderCached==TRUE) { // VDI header was already cached if(fseeko(glob_xmount.cache.h_cache_file, glob_xmount.cache.p_cache_header->pVdiFileHeader+file_off, SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to cached VDI header at offset %" PRIu64 "\n", glob_xmount.cache.p_cache_header->pVdiFileHeader+file_off) return -EIO; } if(fread(p_buf,cur_to_read,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't read %zu bytes from cache file at offset %" PRIu64 "\n", cur_to_read, glob_xmount.cache.p_cache_header->pVdiFileHeader+file_off) return -EIO; } LOG_DEBUG("Read %zd bytes from cached VDI header at offset %" PRIu64 " at cache file offset %" PRIu64 "\n", cur_to_read, file_off, glob_xmount.cache.p_cache_header->pVdiFileHeader+file_off) } else { // VDI header isn't cached memcpy(p_buf, ((char*)glob_xmount.output.vdi.p_vdi_header)+file_off, cur_to_read); LOG_DEBUG("Read %zd bytes at offset %" PRIu64 " from virtual VDI header\n",cur_to_read, file_off) } if(to_read==cur_to_read) return to_read; else { // Adjust values to read from morphed image to_read-=cur_to_read; p_buf+=cur_to_read; file_off=0; } } else file_off-=glob_xmount.output.vdi.vdi_header_size; break; case VirtImageType_VHD: // When emulating VHD, make sure the while loop below only reads data // available in the morphed image. Any VHD footer data must be read // afterwards. if(file_off>=morphed_image_size) { to_read_later=to_read; to_read=0; } else if((file_off+to_read)>morphed_image_size) { to_read_later=(file_off+to_read)-morphed_image_size; to_read-=to_read_later; } break; } // Calculate block to read data from cur_block=file_off/CACHE_BLOCK_SIZE; block_off=file_off%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; if(glob_xmount.output.writable==TRUE && glob_xmount.cache.p_cache_blkidx[cur_block].Assigned==TRUE) { // Write support enabled and need to read altered data from cachefile if(fseeko(glob_xmount.cache.h_cache_file, glob_xmount.cache.p_cache_blkidx[cur_block].off_data+block_off, SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to offset %" PRIu64 " in cache file\n") return -EIO; } if(fread(p_buf,cur_to_read,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't read data from cache file!\n") return -EIO; } LOG_DEBUG("Read %zd bytes at offset %" PRIu64 " from cache file\n",cur_to_read,file_off) } else { // No write support or data not cached ret=GetMorphedImageData(p_buf,file_off,cur_to_read,&read); if(ret!=TRUE || read!=cur_to_read) { LOG_ERROR("Couldn't read data from virtual image!\n") return -EIO; } LOG_DEBUG("Read %zu bytes at offset %zu from virtual image file\n", cur_to_read, file_off); } cur_block++; block_off=0; p_buf+=cur_to_read; to_read-=cur_to_read; file_off+=cur_to_read; } if(to_read_later!=0) { // Read virtual image type specific data following morphed image data switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: case VirtImageType_DMG: case VirtImageType_VMDK: case VirtImageType_VMDKS: case VirtImageType_VDI: break; case VirtImageType_VHD: // Micro$oft has choosen to use a footer rather then a header. if(glob_xmount.output.writable==TRUE && glob_xmount.cache.p_cache_header->VhdFileHeaderCached==TRUE) { // VHD footer was already cached if(fseeko(glob_xmount.cache.h_cache_file, glob_xmount.cache.p_cache_header->pVhdFileHeader+ (file_off-morphed_image_size), SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to cached VHD footer at offset %" PRIu64 "\n", glob_xmount.cache.p_cache_header->pVhdFileHeader+ (file_off-morphed_image_size)) return -EIO; } if(fread(p_buf,to_read_later,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't read %zu bytes from cache file at offset %" PRIu64 "\n", to_read_later, glob_xmount.cache.p_cache_header->pVhdFileHeader+ (file_off-morphed_image_size)) return -EIO; } LOG_DEBUG("Read %zd bytes from cached VHD footer at offset %" PRIu64 " at cache file offset %" PRIu64 "\n", to_read_later, (file_off-morphed_image_size), glob_xmount.cache.p_cache_header->pVhdFileHeader+ (file_off-morphed_image_size)) } else { // VHD header isn't cached memcpy(p_buf, ((char*)glob_xmount.output.vhd.p_vhd_header)+ (file_off-morphed_image_size), to_read_later); LOG_DEBUG("Read %zd bytes at offset %" PRIu64 " from virtual VHD header\n", to_read_later, (file_off-morphed_image_size)) } break; } } return size; } //! Write data to virtual VDI file header /*! * \param p_buf Buffer containing data to write * \param offset Offset of changes * \param size Amount of bytes to write * \return Number of written bytes on success or "-1" on error */ static int SetVdiFileHeaderData(char *p_buf,off_t offset,size_t size) { if(offset+size>glob_xmount.output.vdi.vdi_header_size) { size=glob_xmount.output.vdi.vdi_header_size-offset; } LOG_DEBUG("Need to cache %zu bytes at offset %" PRIu64 " from VDI header\n", size, offset); if(glob_xmount.cache.p_cache_header->VdiFileHeaderCached==1) { // Header was already cached if(fseeko(glob_xmount.cache.h_cache_file, glob_xmount.cache.p_cache_header->pVdiFileHeader+offset, SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to cached VDI header at address %" PRIu64 "\n", glob_xmount.cache.p_cache_header->pVdiFileHeader+offset) return -1; } if(fwrite(p_buf,size,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write %zu bytes to cache file at offset %" PRIu64 "\n", size, glob_xmount.cache.p_cache_header->pVdiFileHeader+offset) return -1; } LOG_DEBUG("Wrote %zd bytes at offset %" PRIu64 " to cache file\n", size, glob_xmount.cache.p_cache_header->pVdiFileHeader+offset) } else { // Header wasn't already cached. if(fseeko(glob_xmount.cache.h_cache_file, 0, SEEK_END)!=0) { LOG_ERROR("Couldn't seek to end of cache file!") return -1; } glob_xmount.cache.p_cache_header->pVdiFileHeader= ftello(glob_xmount.cache.h_cache_file); LOG_DEBUG("Caching whole VDI header\n") if(offset>0) { // Changes do not begin at offset 0, need to prepend with data from // VDI header if(fwrite((char*)glob_xmount.output.vdi.p_vdi_header, offset, 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Error while writing %" PRIu64 " bytes " "to cache file at offset %" PRIu64 "!\n", offset, glob_xmount.cache.p_cache_header->pVdiFileHeader); return -1; } LOG_DEBUG("Prepended changed data with %" PRIu64 " bytes at cache file offset %" PRIu64 "\n", offset, glob_xmount.cache.p_cache_header->pVdiFileHeader) } // Cache changed data if(fwrite(p_buf,size,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write %zu bytes to cache file at offset %" PRIu64 "\n",size, glob_xmount.cache.p_cache_header->pVdiFileHeader+offset) return -1; } LOG_DEBUG("Wrote %zu bytes of changed data to cache file offset %" PRIu64 "\n", size, glob_xmount.cache.p_cache_header->pVdiFileHeader+offset) if(offset+size!=glob_xmount.output.vdi.vdi_header_size) { // Need to append data from VDI header to cache whole data struct if(fwrite(((char*)glob_xmount.output.vdi.p_vdi_header)+offset+size, glob_xmount.output.vdi.vdi_header_size-(offset+size), 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write %zu bytes to cache file at offset %" PRIu64 "\n", glob_xmount.output.vdi.vdi_header_size-(offset+size), (uint64_t)(glob_xmount.cache.p_cache_header->pVdiFileHeader+ offset+size)); return -1; } LOG_DEBUG("Appended %" PRIu32 " bytes to changed data at cache file offset %" PRIu64 "\n", glob_xmount.output.vdi.vdi_header_size-(offset+size), glob_xmount.cache.p_cache_header->pVdiFileHeader+offset+size) } // Mark header as cached and update header in cache file glob_xmount.cache.p_cache_header->VdiFileHeaderCached=1; if(fseeko(glob_xmount.cache.h_cache_file,0,SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to offset 0 of cache file!\n") return -1; } if(fwrite((char*)glob_xmount.cache.p_cache_header, sizeof(ts_CacheFileHeader), 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write changed cache file header!\n") return -1; } } // All important data has been written, now flush all buffers to make // sure data is written to cache file fflush(glob_xmount.cache.h_cache_file); #ifndef __APPLE__ ioctl(fileno(glob_xmount.cache.h_cache_file),BLKFLSBUF,0); #endif return size; } //! Write data to virtual VHD file footer /*! * \param p_buf Buffer containing data to write * \param offset Offset of changes * \param size Amount of bytes to write * \return Number of written bytes on success or "-1" on error */ static int SetVhdFileHeaderData(char *p_buf,off_t offset,size_t size) { LOG_DEBUG("Need to cache %zu bytes at offset %" PRIu64 " from VHD footer\n",size,offset) if(glob_xmount.cache.p_cache_header->VhdFileHeaderCached==1) { // Header has already been cached if(fseeko(glob_xmount.cache.h_cache_file, glob_xmount.cache.p_cache_header->pVhdFileHeader+offset, SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to cached VHD header at address %" PRIu64 "\n", glob_xmount.cache.p_cache_header->pVhdFileHeader+offset); return -1; } if(fwrite(p_buf,size,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write %zu bytes to cache file at offset %" PRIu64 "\n", size, glob_xmount.cache.p_cache_header->pVhdFileHeader+offset); return -1; } LOG_DEBUG("Wrote %zd bytes at offset %" PRIu64 " to cache file\n", size, glob_xmount.cache.p_cache_header->pVhdFileHeader+offset); } else { // Header hasn't been cached yet. if(fseeko(glob_xmount.cache.h_cache_file, 0, SEEK_END)!=0) { LOG_ERROR("Couldn't seek to end of cache file!") return -1; } glob_xmount.cache.p_cache_header->pVhdFileHeader= ftello(glob_xmount.cache.h_cache_file); LOG_DEBUG("Caching whole VHD header\n") if(offset>0) { // Changes do not begin at offset 0, need to prepend with data from // VHD header if(fwrite((char*)glob_xmount.output.vhd.p_vhd_header, offset, 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Error while writing %" PRIu64 " bytes " "to cache file at offset %" PRIu64 "!\n", offset, glob_xmount.cache.p_cache_header->pVhdFileHeader); return -1; } LOG_DEBUG("Prepended changed data with %" PRIu64 " bytes at cache file offset %" PRIu64 "\n", offset, glob_xmount.cache.p_cache_header->pVhdFileHeader); } // Cache changed data if(fwrite(p_buf,size,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write %zu bytes to cache file at offset %" PRIu64 "\n", size, glob_xmount.cache.p_cache_header->pVhdFileHeader+offset); return -1; } LOG_DEBUG("Wrote %zu bytes of changed data to cache file offset %" PRIu64 "\n", size, glob_xmount.cache.p_cache_header->pVhdFileHeader+offset); if(offset+size!=sizeof(ts_VhdFileHeader)) { // Need to append data from VHD header to cache whole data struct if(fwrite(((char*)glob_xmount.output.vhd.p_vhd_header)+offset+size, sizeof(ts_VhdFileHeader)-(offset+size), 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write %zu bytes to cache file at offset %" PRIu64 "\n", sizeof(ts_VhdFileHeader)-(offset+size), (uint64_t)(glob_xmount.cache.p_cache_header->pVhdFileHeader+ offset+size)) return -1; } LOG_DEBUG("Appended %" PRIu32 " bytes to changed data at cache file offset %" PRIu64 "\n", sizeof(ts_VhdFileHeader)-(offset+size), glob_xmount.cache.p_cache_header->pVhdFileHeader+offset+size); } // Mark header as cached and update header in cache file glob_xmount.cache.p_cache_header->VhdFileHeaderCached=1; if(fseeko(glob_xmount.cache.h_cache_file,0,SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to offset 0 of cache file!\n") return -1; } if(fwrite((char*)glob_xmount.cache.p_cache_header, sizeof(ts_CacheFileHeader), 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write changed cache file header!\n") return -1; } } // All important data has been written, now flush all buffers to make // sure data is written to cache file fflush(glob_xmount.cache.h_cache_file); #ifndef __APPLE__ ioctl(fileno(glob_xmount.cache.h_cache_file),BLKFLSBUF,0); #endif return size; } //! Write data to virtual image /*! * \param p_buf Buffer containing data to write * \param offset Offset to start writing at * \param size Size of data to be written * \return Number of written bytes on success or "-1" on error */ static int SetVirtImageData(const char *p_buf, off_t offset, size_t size) { uint64_t cur_block=0; uint64_t virt_image_size; uint64_t orig_image_size; size_t to_write=0; size_t to_write_later=0; size_t to_write_now=0; off_t file_offset=offset; off_t block_offset=0; char *p_write_buf=(char*)p_buf; char *p_buf2; int ret; size_t read; // Get virtual image size if(!GetVirtImageSize(&virt_image_size)) { LOG_ERROR("Couldn't get virtual image size!\n") return -1; } if(offset>=virt_image_size) { LOG_ERROR("Attempt to write beyond EOF of virtual image file!\n") return -1; } if(offset+size>virt_image_size) { LOG_DEBUG("Attempt to write past EOF of virtual image file\n") size=virt_image_size-offset; } to_write=size; // Get original image size if(!GetMorphedImageSize(&orig_image_size)) { LOG_ERROR("Couldn't get morphed image size!\n") return -1; } // Cache virtual image type specific data preceeding original image data switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: case VirtImageType_DMG: case VirtImageType_VMDK: case VirtImageType_VMDKS: break; case VirtImageType_VDI: if(file_offset=orig_image_size) { to_write_later=to_write; to_write=0; } else if((file_offset+to_write)>orig_image_size) { to_write_later=(file_offset+to_write)-orig_image_size; to_write-=to_write_later; } break; } // Calculate block to write data to cur_block=file_offset/CACHE_BLOCK_SIZE; block_offset=file_offset%CACHE_BLOCK_SIZE; while(to_write!=0) { // Calculate how many bytes we have to write to this block if(block_offset+to_write>CACHE_BLOCK_SIZE) { to_write_now=CACHE_BLOCK_SIZE-block_offset; } else to_write_now=to_write; if(glob_xmount.cache.p_cache_blkidx[cur_block].Assigned==1) { // Block was already cached // Seek to data offset in cache file if(fseeko(glob_xmount.cache.h_cache_file, glob_xmount.cache.p_cache_blkidx[cur_block].off_data+block_offset, SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to cached block at address %" PRIu64 "\n", glob_xmount.cache.p_cache_blkidx[cur_block].off_data+ block_offset); return -1; } if(fwrite(p_write_buf,to_write_now,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Error while writing %zu bytes " "to cache file at offset %" PRIu64 "!\n", to_write_now, glob_xmount.cache.p_cache_blkidx[cur_block].off_data+ block_offset); return -1; } LOG_DEBUG("Wrote %zd bytes at offset %" PRIu64 " to cache file\n",to_write_now, glob_xmount.cache.p_cache_blkidx[cur_block].off_data+ block_offset); } else { // Uncached block. Need to cache entire new block // Seek to end of cache file to append new cache block fseeko(glob_xmount.cache.h_cache_file,0,SEEK_END); glob_xmount.cache.p_cache_blkidx[cur_block].off_data= ftello(glob_xmount.cache.h_cache_file); if(block_offset!=0) { // Changed data does not begin at block boundry. Need to prepend // with data from virtual image file XMOUNT_MALLOC(p_buf2,char*,block_offset*sizeof(char)); ret=GetMorphedImageData(p_buf2, file_offset-block_offset, block_offset, &read); if(ret!=TRUE || read!=block_offset) { LOG_ERROR("Couldn't read data from morphed image!\n") return -1; } if(fwrite(p_buf2,block_offset,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't writing %" PRIu64 " bytes " "to cache file at offset %" PRIu64 "!\n", block_offset, glob_xmount.cache.p_cache_blkidx[cur_block].off_data); return -1; } LOG_DEBUG("Prepended changed data with %" PRIu64 " bytes from virtual image file at offset %" PRIu64 "\n",block_offset,file_offset-block_offset) free(p_buf2); } if(fwrite(p_write_buf,to_write_now,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Error while writing %zd bytes " "to cache file at offset %" PRIu64 "!\n", to_write_now, glob_xmount.cache.p_cache_blkidx[cur_block].off_data+ block_offset); return -1; } if(block_offset+to_write_now!=CACHE_BLOCK_SIZE) { // Changed data does not end at block boundry. Need to append // with data from virtual image file XMOUNT_MALLOC(p_buf2,char*,(CACHE_BLOCK_SIZE- (block_offset+to_write_now))*sizeof(char)) memset(p_buf2,0,CACHE_BLOCK_SIZE-(block_offset+to_write_now)); if((file_offset-block_offset)+CACHE_BLOCK_SIZE>orig_image_size) { // Original image is smaller than full cache block ret=GetMorphedImageData(p_buf2, file_offset+to_write_now, orig_image_size-(file_offset+to_write_now), &read); if(ret!=TRUE || read!=orig_image_size-(file_offset+to_write_now)) { LOG_ERROR("Couldn't read data from virtual image file!\n") return -1; } } else { ret=GetMorphedImageData(p_buf2, file_offset+to_write_now, CACHE_BLOCK_SIZE-(block_offset+to_write_now), &read); if(ret!=TRUE || read!=CACHE_BLOCK_SIZE-(block_offset+to_write_now)) { LOG_ERROR("Couldn't read data from virtual image file!\n") return -1; } } if(fwrite(p_buf2, CACHE_BLOCK_SIZE-(block_offset+to_write_now), 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Error while writing %zd bytes " "to cache file at offset %" PRIu64 "!\n", CACHE_BLOCK_SIZE-(block_offset+to_write_now), glob_xmount.cache.p_cache_blkidx[cur_block].off_data+ block_offset+to_write_now); return -1; } free(p_buf2); } // All important data for this cache block has been written, // flush all buffers and mark cache block as assigned fflush(glob_xmount.cache.h_cache_file); #ifndef __APPLE__ ioctl(fileno(glob_xmount.cache.h_cache_file),BLKFLSBUF,0); #endif glob_xmount.cache.p_cache_blkidx[cur_block].Assigned=1; // Update cache block index entry in cache file fseeko(glob_xmount.cache.h_cache_file, sizeof(ts_CacheFileHeader)+ (cur_block*sizeof(ts_CacheFileBlockIndex)), SEEK_SET); if(fwrite(&(glob_xmount.cache.p_cache_blkidx[cur_block]), sizeof(ts_CacheFileBlockIndex), 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't update cache file block index!\n"); return -1; } LOG_DEBUG("Updated cache file block index: Number=%" PRIu64 ", Data offset=%" PRIu64 "\n", cur_block, glob_xmount.cache.p_cache_blkidx[cur_block].off_data); } // Flush buffers fflush(glob_xmount.cache.h_cache_file); #ifndef __APPLE__ ioctl(fileno(glob_xmount.cache.h_cache_file),BLKFLSBUF,0); #endif block_offset=0; cur_block++; p_write_buf+=to_write_now; to_write-=to_write_now; file_offset+=to_write_now; } if(to_write_later!=0) { // Cache virtual image type specific data preceeding original image data switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: case VirtImageType_DMG: case VirtImageType_VMDK: case VirtImageType_VMDKS: case VirtImageType_VDI: break; case VirtImageType_VHD: // Micro$oft has choosen to use a footer rather then a header. ret=SetVhdFileHeaderData(p_write_buf, file_offset-orig_image_size, to_write_later); if(ret==-1) { LOG_ERROR("Couldn't write data to virtual VHD file footer!\n") return -1; } break; } } 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=GetMorphedImageData(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; } //! Build and init virtual VDI file header /*! * \return TRUE on success, FALSE on error */ static int InitVirtVdiHeader() { // See http://forums.virtualbox.org/viewtopic.php?t=8046 for a // "description" of the various header fields uint64_t image_size; off_t offset; uint32_t i,block_entries; // Get input image size if(!GetMorphedImageSize(&image_size)) { LOG_ERROR("Couldn't get morphed image size!\n") return FALSE; } // Calculate how many VDI blocks we need block_entries=image_size/VDI_IMAGE_BLOCK_SIZE; if((image_size%VDI_IMAGE_BLOCK_SIZE)!=0) block_entries++; glob_xmount.output.vdi.vdi_block_map_size=block_entries*sizeof(uint32_t); LOG_DEBUG("BlockMap: %d (%08X) entries, %d (%08X) bytes!\n", block_entries, block_entries, glob_xmount.output.vdi.vdi_block_map_size, glob_xmount.output.vdi.vdi_block_map_size) // Allocate memory for vdi header and block map glob_xmount.output.vdi.vdi_header_size= sizeof(ts_VdiFileHeader)+glob_xmount.output.vdi.vdi_block_map_size; XMOUNT_MALLOC(glob_xmount.output.vdi.p_vdi_header, pts_VdiFileHeader, glob_xmount.output.vdi.vdi_header_size); memset(glob_xmount.output.vdi.p_vdi_header, 0, glob_xmount.output.vdi.vdi_header_size); glob_xmount.output.vdi.p_vdi_block_map= ((void*)glob_xmount.output.vdi.p_vdi_header)+sizeof(ts_VdiFileHeader); // Init header values strncpy(glob_xmount.output.vdi.p_vdi_header->szFileInfo, VDI_FILE_COMMENT, strlen(VDI_FILE_COMMENT)+1); glob_xmount.output.vdi.p_vdi_header->u32Signature=VDI_IMAGE_SIGNATURE; glob_xmount.output.vdi.p_vdi_header->u32Version=VDI_IMAGE_VERSION; // No idea what the following value is for! Testimage had same value glob_xmount.output.vdi.p_vdi_header->cbHeader=0x00000180; glob_xmount.output.vdi.p_vdi_header->u32Type=VDI_IMAGE_TYPE_FIXED; glob_xmount.output.vdi.p_vdi_header->fFlags=VDI_IMAGE_FLAGS; strncpy(glob_xmount.output.vdi.p_vdi_header->szComment, VDI_HEADER_COMMENT, strlen(VDI_HEADER_COMMENT)+1); glob_xmount.output.vdi.p_vdi_header->offData= glob_xmount.output.vdi.vdi_header_size; glob_xmount.output.vdi.p_vdi_header->offBlocks=sizeof(ts_VdiFileHeader); glob_xmount.output.vdi.p_vdi_header->cCylinders=0; // Legacy info glob_xmount.output.vdi.p_vdi_header->cHeads=0; // Legacy info glob_xmount.output.vdi.p_vdi_header->cSectors=0; // Legacy info glob_xmount.output.vdi.p_vdi_header->cbSector=512; // Legacy info glob_xmount.output.vdi.p_vdi_header->u32Dummy=0; glob_xmount.output.vdi.p_vdi_header->cbDisk=image_size; // Seems as VBox is always using a 1MB blocksize glob_xmount.output.vdi.p_vdi_header->cbBlock=VDI_IMAGE_BLOCK_SIZE; glob_xmount.output.vdi.p_vdi_header->cbBlockExtra=0; glob_xmount.output.vdi.p_vdi_header->cBlocks=block_entries; glob_xmount.output.vdi.p_vdi_header->cBlocksAllocated=block_entries; // Use partial MD5 input file hash as creation UUID and generate a random // modification UUID. VBox won't accept immages where create and modify UUIDS // aren't set. glob_xmount.output.vdi.p_vdi_header->uuidCreate_l= glob_xmount.input.image_hash_lo; glob_xmount.output.vdi.p_vdi_header->uuidCreate_h= glob_xmount.input.image_hash_hi; #define rand64(var) { \ *((uint32_t*)&(var))=rand(); \ *(((uint32_t*)&(var))+1)=rand(); \ } rand64(glob_xmount.output.vdi.p_vdi_header->uuidModify_l); rand64(glob_xmount.output.vdi.p_vdi_header->uuidModify_h); #undef rand64 // Generate block map i=0; for(offset=0;offsetcookie=VHD_IMAGE_HVAL_COOKIE; glob_xmount.output.vhd.p_vhd_header->features=VHD_IMAGE_HVAL_FEATURES; glob_xmount.output.vhd.p_vhd_header->file_format_version= VHD_IMAGE_HVAL_FILE_FORMAT_VERSION; glob_xmount.output.vhd.p_vhd_header->data_offset=VHD_IMAGE_HVAL_DATA_OFFSET; glob_xmount.output.vhd.p_vhd_header->creation_time= htobe32(time(NULL)-VHD_IMAGE_TIME_CONVERSION_OFFSET); glob_xmount.output.vhd.p_vhd_header->creator_app= VHD_IMAGE_HVAL_CREATOR_APPLICATION; glob_xmount.output.vhd.p_vhd_header->creator_ver= VHD_IMAGE_HVAL_CREATOR_VERSION; glob_xmount.output.vhd.p_vhd_header->creator_os= VHD_IMAGE_HVAL_CREATOR_HOST_OS; glob_xmount.output.vhd.p_vhd_header->size_original=htobe64(orig_image_size); glob_xmount.output.vhd.p_vhd_header->size_current= glob_xmount.output.vhd.p_vhd_header->size_original; // Convert size to sectors if(orig_image_size>136899993600) { // image is larger then CHS values can address. // Set sectors to max (C65535*H16*S255). geom_tot_s=267382800; } else { // Calculate actual sectors geom_tot_s=orig_image_size/512; if((orig_image_size%512)!=0) geom_tot_s++; } // Calculate CHS values. This is done according to the VHD specs if(geom_tot_s>=66059280) { // C65535 * H16 * S63 geom_s=255; geom_h=16; geom_c_x_h=geom_tot_s/geom_s; } else { geom_s=17; geom_c_x_h=geom_tot_s/geom_s; geom_h=(geom_c_x_h+1023)/1024; if(geom_h<4) geom_h=4; if(geom_c_x_h>=(geom_h*1024) || geom_h>16) { geom_s=31; geom_h=16; geom_c_x_h=geom_tot_s/geom_s; } if(geom_c_x_h>=(geom_h*1024)) { geom_s=63; geom_h=16; geom_c_x_h=geom_tot_s/geom_s; } } geom_c=geom_c_x_h/geom_h; glob_xmount.output.vhd.p_vhd_header->disk_geometry_c=htobe16(geom_c); glob_xmount.output.vhd.p_vhd_header->disk_geometry_h=geom_h; glob_xmount.output.vhd.p_vhd_header->disk_geometry_s=geom_s; glob_xmount.output.vhd.p_vhd_header->disk_type=VHD_IMAGE_HVAL_DISK_TYPE; glob_xmount.output.vhd.p_vhd_header->uuid_l=glob_xmount.input.image_hash_lo; glob_xmount.output.vhd.p_vhd_header->uuid_h=glob_xmount.input.image_hash_hi; glob_xmount.output.vhd.p_vhd_header->saved_state=0x00; // Calculate footer checksum for(i=0;ichecksum=htobe32(~checksum); LOG_DEBUG("VHD header size = %u\n",sizeof(ts_VhdFileHeader)); return TRUE; } //! Init the virtual VMDK file /*! * \return TRUE on success, FALSE on error */ static int InitVirtualVmdkFile() { uint64_t image_size=0; uint64_t image_blocks=0; char buf[500]; // Get original image size if(!GetMorphedImageSize(&image_size)) { LOG_ERROR("Couldn't get morphed image size!\n") return FALSE; } image_blocks=image_size/512; if(image_size%512!=0) image_blocks++; #define VMDK_DESC_FILE "# Disk DescriptorFile\n" \ "version=1\n" \ "CID=fffffffe\n" \ "parentCID=ffffffff\n" \ "createType=\"monolithicFlat\"\n\n" \ "# Extent description\n" \ "RW %" PRIu64 " FLAT \"%s\" 0\n\n" \ "# The Disk Data Base\n" \ "#DDB\n" \ "ddb.virtualHWVersion = \"3\"\n" \ "ddb.adapterType = \"%s\"\n" \ "ddb.geometry.cylinders = \"0\"\n" \ "ddb.geometry.heads = \"0\"\n" \ "ddb.geometry.sectors = \"0\"\n" if(glob_xmount.output.VirtImageType==VirtImageType_VMDK) { // VMDK with IDE bus sprintf(buf, VMDK_DESC_FILE, image_blocks, (glob_xmount.output.p_virtual_image_path)+1, "ide"); } else if(glob_xmount.output.VirtImageType==VirtImageType_VMDKS){ // VMDK with SCSI bus sprintf(buf, VMDK_DESC_FILE, image_blocks, (glob_xmount.output.p_virtual_image_path)+1, "scsi"); } else { LOG_ERROR("Unknown virtual VMDK file format!\n") return FALSE; } #undef VMDK_DESC_FILE // Do not use XMOUNT_STRSET here to avoid adding '\0' to the buffer! XMOUNT_MALLOC(glob_xmount.output.vmdk.p_vmdk_file,char*,strlen(buf)) strncpy(glob_xmount.output.vmdk.p_vmdk_file,buf,strlen(buf)); glob_xmount.output.vmdk.vmdk_file_size=strlen(buf); return TRUE; } //! Create virtual image info file /*! * \return TRUE on success, FALSE on error */ static int InitVirtImageInfoFile() { 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;ip_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 image_size=0; uint64_t blockindex_size=0; uint64_t cachefile_header_size=0; uint64_t cachefile_size=0; uint32_t needed_blocks=0; uint64_t buf; if(!glob_xmount.cache.overwrite_cache) { // Try to open an existing cache file or create a new one glob_xmount.cache.h_cache_file=(FILE*)FOPEN(glob_xmount.cache.p_cache_file, "rb+"); if(glob_xmount.cache.h_cache_file==NULL) { // As the c lib seems to have no possibility to open a file rw wether it // exists or not (w+ does not work because it truncates an existing file), // when r+ returns NULL the file could simply not exist LOG_DEBUG("Cache file does not exist. Creating new one\n") glob_xmount.cache.h_cache_file= (FILE*)FOPEN(glob_xmount.cache.p_cache_file,"wb+"); if(glob_xmount.cache.h_cache_file==NULL) { // There is really a problem opening the file LOG_ERROR("Couldn't open cache file \"%s\"!\n", glob_xmount.cache.p_cache_file) return FALSE; } } } else { // Overwrite existing cache file or create a new one glob_xmount.cache.h_cache_file=(FILE*)FOPEN(glob_xmount.cache.p_cache_file, "wb+"); if(glob_xmount.cache.h_cache_file==NULL) { LOG_ERROR("Couldn't open cache file \"%s\"!\n", glob_xmount.cache.p_cache_file) return FALSE; } } // Get input image size if(!GetMorphedImageSize(&image_size)) { LOG_ERROR("Couldn't get morphed image size!\n") return FALSE; } // Calculate how many blocks are needed and how big the buffers must be // for the actual cache file version needed_blocks=image_size/CACHE_BLOCK_SIZE; if((image_size%CACHE_BLOCK_SIZE)!=0) needed_blocks++; blockindex_size=needed_blocks*sizeof(ts_CacheFileBlockIndex); cachefile_header_size=sizeof(ts_CacheFileHeader)+blockindex_size; LOG_DEBUG("Cache blocks: %u (%04X) entries, %zd (%08zX) bytes\n", needed_blocks, needed_blocks, blockindex_size, blockindex_size) // Get cache file size // fseeko64 had massive problems! if(fseeko(glob_xmount.cache.h_cache_file,0,SEEK_END)!=0) { LOG_ERROR("Couldn't seek to end of cache file!\n") return FALSE; } // Same here, ftello64 didn't work at all and returned 0 all the times cachefile_size=ftello(glob_xmount.cache.h_cache_file); LOG_DEBUG("Cache file has %zd bytes\n",cachefile_size) if(cachefile_size>0) { // Cache file isn't empty, parse block header LOG_DEBUG("Cache file not empty. Parsing block header\n") if(fseeko(glob_xmount.cache.h_cache_file,0,SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to beginning of cache file!\n") return FALSE; } // Read and check file signature if(fread(&buf,8,1,glob_xmount.cache.h_cache_file)!=1 || buf!=CACHE_FILE_SIGNATURE) { LOG_ERROR("Not an xmount cache file or cache file corrupt!\n") return FALSE; } // Now get cache file version (Has only 32bit!) if(fread(&buf,4,1,glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Not an xmount cache file or cache file corrupt!\n") return FALSE; } switch((uint32_t)buf) { case 0x00000001: // Old v1 cache file. LOG_ERROR("Unsupported cache file version!\n") LOG_ERROR("Please use xmount-tool to upgrade your cache file.\n") return FALSE; case CUR_CACHE_FILE_VERSION: // Current version if(fseeko(glob_xmount.cache.h_cache_file,0,SEEK_SET)!=0) { LOG_ERROR("Couldn't seek to beginning of cache file!\n") return FALSE; } // Alloc memory for header and block index XMOUNT_MALLOC(glob_xmount.cache.p_cache_header, pts_CacheFileHeader, cachefile_header_size); memset(glob_xmount.cache.p_cache_header,0,cachefile_header_size); // Read header and block index from file if(fread(glob_xmount.cache.p_cache_header, cachefile_header_size, 1, glob_xmount.cache.h_cache_file)!=1) { // Cache file isn't big enough LOG_ERROR("Cache file corrupt!\n") return FALSE; } break; default: LOG_ERROR("Unknown cache file version!\n") return FALSE; } // Check if cache file has same block size as we do if(glob_xmount.cache.p_cache_header->BlockSize!=CACHE_BLOCK_SIZE) { LOG_ERROR("Cache file does not use default cache block size!\n") return FALSE; } // Set pointer to block index glob_xmount.cache.p_cache_blkidx= (pts_CacheFileBlockIndex)((void*)glob_xmount.cache.p_cache_header+ glob_xmount.cache.p_cache_header->pBlockIndex); } else { // New cache file, generate a new block header LOG_DEBUG("Cache file is empty. Generating new block header\n"); // Alloc memory for header and block index XMOUNT_MALLOC(glob_xmount.cache.p_cache_header, pts_CacheFileHeader, cachefile_header_size) memset(glob_xmount.cache.p_cache_header,0,cachefile_header_size); glob_xmount.cache.p_cache_header->FileSignature=CACHE_FILE_SIGNATURE; glob_xmount.cache.p_cache_header->CacheFileVersion=CUR_CACHE_FILE_VERSION; glob_xmount.cache.p_cache_header->BlockSize=CACHE_BLOCK_SIZE; glob_xmount.cache.p_cache_header->BlockCount=needed_blocks; //glob_xmount.cache.p_cache_header->UsedBlocks=0; // The following pointer is only usuable when reading data from cache file glob_xmount.cache.p_cache_header->pBlockIndex=sizeof(ts_CacheFileHeader); glob_xmount.cache.p_cache_blkidx= (pts_CacheFileBlockIndex)((void*)glob_xmount.cache.p_cache_header+ sizeof(ts_CacheFileHeader)); glob_xmount.cache.p_cache_header->VdiFileHeaderCached=FALSE; glob_xmount.cache.p_cache_header->pVdiFileHeader=0; glob_xmount.cache.p_cache_header->VmdkFileCached=FALSE; glob_xmount.cache.p_cache_header->VmdkFileSize=0; glob_xmount.cache.p_cache_header->pVmdkFile=0; glob_xmount.cache.p_cache_header->VhdFileHeaderCached=FALSE; glob_xmount.cache.p_cache_header->pVhdFileHeader=0; // Write header to file if(fwrite(glob_xmount.cache.p_cache_header, cachefile_header_size, 1, glob_xmount.cache.h_cache_file)!=1) { LOG_ERROR("Couldn't write cache file header to file!\n"); 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; 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; 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); } 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 and %u morphing libs were loaded.\n", glob_xmount.input.libs_count, glob_xmount.morphing.libs_count); free(p_library_path); closedir(p_dir); return ((glob_xmount.input.libs_count>0 && glob_xmount.morphing.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;ip_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;ip_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 input type 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.p_cache_file=NULL; glob_xmount.cache.h_cache_file=NULL; glob_xmount.cache.p_cache_header=NULL; glob_xmount.cache.p_cache_blkidx=NULL; glob_xmount.cache.overwrite_cache=FALSE; // Output #ifndef __APPLE__ glob_xmount.output.VirtImageType=VirtImageType_DD; #else glob_xmount.output.VirtImageType=VirtImageType_DMG; #endif 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; glob_xmount.output.vdi.vdi_header_size=0; glob_xmount.output.vdi.p_vdi_header=NULL; glob_xmount.output.vdi.vdi_block_map_size=0; glob_xmount.output.vdi.p_vdi_block_map=NULL; glob_xmount.output.vhd.p_vhd_header=NULL; glob_xmount.output.vmdk.p_virtual_vmdk_path=NULL; glob_xmount.output.vmdk.p_vmdk_file=NULL; glob_xmount.output.vmdk.vmdk_file_size=0; glob_xmount.output.vmdk.p_vmdk_lockdir1=NULL; glob_xmount.output.vmdk.p_vmdk_lockdir2=NULL; glob_xmount.output.vmdk.p_vmdk_lockfile_data=NULL; glob_xmount.output.vmdk.vmdk_lockfile_size=0; glob_xmount.output.vmdk.p_vmdk_lockfile_name=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; LOG_DEBUG("Freeing all resources\n"); // Misc if(glob_xmount.pp_fuse_argv!=NULL) { for(int i=0;i 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;ip_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;ip_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;iifiles_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;ip_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;ip_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 GetInputImageData(glob_xmount.input.pp_images[image], p_buf, offset, count, p_read); } /******************************************************************************* * FUSE function implementation ******************************************************************************/ //! FUSE access implementation /*! * \param p_path Path of file to get attributes from * \param perm Requested permissisons * \return 0 on success, negated error code on error */ /* static int FuseAccess(const char *path, int perm) { // TODO: Implement propper file permission handling // http://www.cs.cf.ac.uk/Dave/C/node20.html // Values for the second argument to access. // These may be OR'd together. //#define R_OK 4 // Test for read permission. //#define W_OK 2 // Test for write permission. //#define X_OK 1 // Test for execute permission. //#define F_OK 0 // Test for existence. return 0; } */ //! FUSE getattr implementation /*! * \param p_path Path of file to get attributes from * \param p_stat Pointer to stat structure to save attributes to * \return 0 on success, negated error code on error */ static int FuseGetAttr(const char *p_path, struct stat *p_stat) { memset(p_stat,0,sizeof(struct stat)); if(strcmp(p_path,"/")==0) { // Attributes of mountpoint p_stat->st_mode=S_IFDIR | 0777; p_stat->st_nlink=2; } else if(strcmp(p_path,glob_xmount.output.p_virtual_image_path)==0) { // Attributes of virtual image if(!glob_xmount.output.writable) p_stat->st_mode=S_IFREG | 0444; else p_stat->st_mode=S_IFREG | 0666; p_stat->st_nlink=1; // Get virtual image file size if(!GetVirtImageSize((uint64_t*)&(p_stat->st_size))) { LOG_ERROR("Couldn't get image size!\n"); return -ENOENT; } if(glob_xmount.output.VirtImageType==VirtImageType_VHD) { // Make sure virtual image seems to be fully allocated (not sparse file). // Without this, Windows won't attach the vhd file! p_stat->st_blocks=p_stat->st_size/512; if(p_stat->st_size%512!=0) p_stat->st_blocks++; } } else if(strcmp(p_path,glob_xmount.output.p_info_path)==0) { // Attributes of virtual image info file p_stat->st_mode=S_IFREG | 0444; p_stat->st_nlink=1; // Get virtual image info file size if(glob_xmount.output.p_info_file!=NULL) { p_stat->st_size=strlen(glob_xmount.output.p_info_file); } else p_stat->st_size=0; } else if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { // Some special files only present when emulating VMDK files if(strcmp(p_path,glob_xmount.output.vmdk.p_virtual_vmdk_path)==0) { // Attributes of virtual vmdk file if(!glob_xmount.output.writable) p_stat->st_mode=S_IFREG | 0444; else p_stat->st_mode=S_IFREG | 0666; p_stat->st_nlink=1; // Get virtual image info file size if(glob_xmount.output.vmdk.p_vmdk_file!=NULL) { p_stat->st_size=glob_xmount.output.vmdk.vmdk_file_size; } else p_stat->st_size=0; } else if(glob_xmount.output.vmdk.p_vmdk_lockdir1!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockdir1)==0) { p_stat->st_mode=S_IFDIR | 0777; p_stat->st_nlink=2; } else if(glob_xmount.output.vmdk.p_vmdk_lockdir2!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockdir2)==0) { p_stat->st_mode=S_IFDIR | 0777; p_stat->st_nlink=2; } else if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockfile_name)==0) { p_stat->st_mode=S_IFREG | 0666; if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL) { p_stat->st_size=strlen(glob_xmount.output.vmdk.p_vmdk_lockfile_name); } else p_stat->st_size=0; } else return -ENOENT; } else return -ENOENT; // Set uid and gid of all files to uid and gid of current process p_stat->st_uid=getuid(); p_stat->st_gid=getgid(); return 0; } //! FUSE mkdir implementation /*! * \param p_path Directory path * \param mode Directory permissions * \return 0 on success, negated error code on error */ static int FuseMkDir(const char *p_path, mode_t mode) { // Only allow creation of VMWare's lock directories if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { if(glob_xmount.output.vmdk.p_vmdk_lockdir1==NULL) { char aVmdkLockDir[strlen(glob_xmount.output.vmdk.p_virtual_vmdk_path)+5]; sprintf(aVmdkLockDir, "%s.lck", glob_xmount.output.vmdk.p_virtual_vmdk_path); if(strcmp(p_path,aVmdkLockDir)==0) { LOG_DEBUG("Creating virtual directory \"%s\"\n",aVmdkLockDir) XMOUNT_STRSET(glob_xmount.output.vmdk.p_vmdk_lockdir1,aVmdkLockDir) return 0; } else { LOG_ERROR("Attempt to create illegal directory \"%s\"!\n",p_path) LOG_DEBUG("Supposed: %s\n",aVmdkLockDir) return -1; } } else if(glob_xmount.output.vmdk.p_vmdk_lockdir2==NULL && strncmp(p_path, glob_xmount.output.vmdk.p_vmdk_lockdir1, strlen(glob_xmount.output.vmdk.p_vmdk_lockdir1))==0) { LOG_DEBUG("Creating virtual directory \"%s\"\n",p_path) XMOUNT_STRSET(glob_xmount.output.vmdk.p_vmdk_lockdir2,p_path) return 0; } else { LOG_ERROR("Attempt to create illegal directory \"%s\"!\n",p_path) LOG_DEBUG("Compared to first %u chars of \"%s\"\n", strlen(glob_xmount.output.vmdk.p_vmdk_lockdir1), glob_xmount.output.vmdk.p_vmdk_lockdir1) return -1; } } LOG_ERROR("Attempt to create directory \"%s\" " "on read-only filesystem!\n",p_path) return -1; } //! FUSE create implementation. /*! * Currently only allows to create VMWare's lock file * * \param p_path File to create * \param mode File mode * \param dev ??? but not used * \return 0 on success, negated error code on error */ static int FuseMkNod(const char *p_path, mode_t mode, dev_t dev) { if((glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) && glob_xmount.output.vmdk.p_vmdk_lockdir1!=NULL && glob_xmount.output.vmdk.p_vmdk_lockfile_name==NULL) { LOG_DEBUG("Creating virtual file \"%s\"\n",p_path) XMOUNT_STRSET(glob_xmount.output.vmdk.p_vmdk_lockfile_name,p_path); return 0; } else { LOG_ERROR("Attempt to create illegal file \"%s\"\n",p_path) return -1; } } //! FUSE readdir implementation /*! * \param p_path Path from where files should be listed * \param p_buf Buffer to write file entrys to * \param filler Function to write dir entrys to buffer * \param offset ??? but not used * \param p_fi File info struct * \return 0 on success, negated error code on error */ static int FuseReadDir(const char *p_path, void *p_buf, fuse_fill_dir_t filler, off_t offset, struct fuse_file_info *p_fi) { // Ignore some params (void)offset; (void)p_fi; if(strcmp(p_path,"/")==0) { // Add std . and .. entrys filler(p_buf,".",NULL,0); filler(p_buf,"..",NULL,0); // Add our virtual files (p+1 to ignore starting "/") filler(p_buf,glob_xmount.output.p_virtual_image_path+1,NULL,0); filler(p_buf,glob_xmount.output.p_info_path+1,NULL,0); if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { // For VMDK's, we use an additional descriptor file filler(p_buf,glob_xmount.output.vmdk.p_virtual_vmdk_path+1,NULL,0); // And there could also be a lock directory if(glob_xmount.output.vmdk.p_vmdk_lockdir1!=NULL) { filler(p_buf,glob_xmount.output.vmdk.p_vmdk_lockdir1+1,NULL,0); } } } else if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { // For VMDK emulation, there could be a lock directory if(glob_xmount.output.vmdk.p_vmdk_lockdir1!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockdir1)==0) { filler(p_buf,".",NULL,0); filler(p_buf,"..",NULL,0); if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL) { filler(p_buf, glob_xmount.output.vmdk.p_vmdk_lockfile_name+ strlen(glob_xmount.output.vmdk.p_vmdk_lockdir1)+1, NULL, 0); } } else if(glob_xmount.output.vmdk.p_vmdk_lockdir2!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockdir2)==0) { filler(p_buf,".",NULL,0); filler(p_buf,"..",NULL,0); } else return -ENOENT; } else return -ENOENT; return 0; } //! FUSE open implementation /*! * \param p_path Path to file to open * \param p_fi File info struct * \return 0 on success, negated error code on error */ static int FuseOpen(const char *p_path, struct fuse_file_info *p_fi) { #define CHECK_OPEN_PERMS() { \ if(!glob_xmount.output.writable && (p_fi->flags & 3)!=O_RDONLY) { \ LOG_DEBUG("Attempt to open the read-only file \"%s\" for writing.\n", \ p_path) \ return -EACCES; \ } \ return 0; \ } if(strcmp(p_path,glob_xmount.output.p_virtual_image_path)==0 || strcmp(p_path,glob_xmount.output.p_info_path)==0) { CHECK_OPEN_PERMS(); } else if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { if(strcmp(p_path,glob_xmount.output.vmdk.p_virtual_vmdk_path)==0 || (glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockfile_name)==0)) { CHECK_OPEN_PERMS(); } } #undef CHECK_OPEN_PERMS LOG_DEBUG("Attempt to open inexistant file \"%s\".\n",p_path); return -ENOENT; } //! FUSE read implementation /*! * \param p_path Path (relative to mount folder) of file to read data from * \param p_buf Pre-allocated buffer where read data should be written to * \param size Number of bytes to read * \param offset Offset to start reading at * \param p_fi: File info struct * \return Read bytes on success, negated error code on error */ static int FuseRead(const char *p_path, char *p_buf, size_t size, off_t offset, struct fuse_file_info *p_fi) { (void)p_fi; int ret; uint64_t len; #define READ_MEM_FILE(filebuf,filesize,filetypestr,mutex) { \ len=filesize; \ if(offsetlen) { \ LOG_DEBUG("Attempt to read past EOF of virtual " filetypestr " file\n"); \ LOG_DEBUG("Adjusting read size from %u to %u\n",size,len-offset); \ size=len-offset; \ } \ pthread_mutex_lock(&mutex); \ memcpy(p_buf,filebuf+offset,size); \ pthread_mutex_unlock(&mutex); \ LOG_DEBUG("Read %" PRIu64 " bytes at offset %" PRIu64 \ " from virtual " filetypestr " file\n",size,offset); \ ret=size; \ } else { \ LOG_DEBUG("Attempt to read behind EOF of virtual " filetypestr " file\n"); \ ret=0; \ } \ } if(strcmp(p_path,glob_xmount.output.p_virtual_image_path)==0) { // Read data from virtual output file // Wait for other threads to end reading/writing data pthread_mutex_lock(&(glob_xmount.mutex_image_rw)); // Get requested data if((ret=GetVirtImageData(p_buf,offset,size))<0) { LOG_ERROR("Couldn't read data from virtual image file!\n") } // Allow other threads to read/write data again pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); } else if(strcmp(p_path,glob_xmount.output.p_info_path)==0) { // Read data from virtual info file READ_MEM_FILE(glob_xmount.output.p_info_file, strlen(glob_xmount.output.p_info_file), "info", glob_xmount.mutex_info_read); } else if(strcmp(p_path,glob_xmount.output.vmdk.p_virtual_vmdk_path)==0) { // Read data from virtual vmdk file READ_MEM_FILE(glob_xmount.output.vmdk.p_vmdk_file, glob_xmount.output.vmdk.vmdk_file_size, "vmdk", glob_xmount.mutex_image_rw); } else if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockfile_name)==0) { // Read data from virtual lock file READ_MEM_FILE(glob_xmount.output.vmdk.p_vmdk_lockfile_data, glob_xmount.output.vmdk.vmdk_lockfile_size, "vmdk lock", glob_xmount.mutex_image_rw); } else { // Attempt to read non existant file LOG_DEBUG("Attempt to read from non existant file \"%s\"\n",p_path) ret=-ENOENT; } #undef READ_MEM_FILE // TODO: Return size of read data!!!!! return ret; } //! FUSE rename implementation /*! * \param p_path File to rename * \param p_npath New filename * \return 0 on error, negated error code on error */ static int FuseRename(const char *p_path, const char *p_npath) { if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockfile_name)==0) { LOG_DEBUG("Renaming virtual lock file from \"%s\" to \"%s\"\n", glob_xmount.output.vmdk.p_vmdk_lockfile_name, p_npath) XMOUNT_REALLOC(glob_xmount.output.vmdk.p_vmdk_lockfile_name,char*, (strlen(p_npath)+1)*sizeof(char)); strcpy(glob_xmount.output.vmdk.p_vmdk_lockfile_name,p_npath); return 0; } } return -ENOENT; } //! FUSE rmdir implementation /*! * \param p_path Directory to delete * \return 0 on success, negated error code on error */ static int FuseRmDir(const char *p_path) { // Only VMWare's lock directories can be deleted if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { if(glob_xmount.output.vmdk.p_vmdk_lockdir1!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockdir1)==0) { LOG_DEBUG("Deleting virtual lock dir \"%s\"\n", glob_xmount.output.vmdk.p_vmdk_lockdir1) free(glob_xmount.output.vmdk.p_vmdk_lockdir1); glob_xmount.output.vmdk.p_vmdk_lockdir1=NULL; return 0; } else if(glob_xmount.output.vmdk.p_vmdk_lockdir2!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockdir2)==0) { LOG_DEBUG("Deleting virtual lock dir \"%s\"\n", glob_xmount.output.vmdk.p_vmdk_lockdir1) free(glob_xmount.output.vmdk.p_vmdk_lockdir2); glob_xmount.output.vmdk.p_vmdk_lockdir2=NULL; return 0; } } return -1; } //! FUSE unlink implementation /*! * \param p_path File to delete * \return 0 on success, negated error code on error */ static int FuseUnlink(const char *p_path) { // Only VMWare's lock file can be deleted if(glob_xmount.output.VirtImageType==VirtImageType_VMDK || glob_xmount.output.VirtImageType==VirtImageType_VMDKS) { if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockfile_name)==0) { LOG_DEBUG("Deleting virtual file \"%s\"\n", glob_xmount.output.vmdk.p_vmdk_lockfile_name) free(glob_xmount.output.vmdk.p_vmdk_lockfile_name); free(glob_xmount.output.vmdk.p_vmdk_lockfile_data); glob_xmount.output.vmdk.p_vmdk_lockfile_name=NULL; glob_xmount.output.vmdk.p_vmdk_lockfile_data=NULL; glob_xmount.output.vmdk.vmdk_lockfile_size=0; return 0; } } return -1; } //! FUSE statfs implementation /*! * \param p_path Get stats for fs that the specified file resides in * \param stats Stats * \return 0 on success, negated error code on error */ /* static int FuseStatFs(const char *p_path, struct statvfs *stats) { struct statvfs CacheFileFsStats; int ret; if(glob_xmount.writable==TRUE) { // If write support is enabled, return stats of fs upon which cache file // resides in if((ret=statvfs(glob_xmount.p_cache_file,&CacheFileFsStats))==0) { memcpy(stats,&CacheFileFsStats,sizeof(struct statvfs)); return 0; } else { LOG_ERROR("Couldn't get stats for fs upon which resides \"%s\"\n", glob_xmount.p_cache_file) return ret; } } else { // TODO: Return read only return 0; } } */ // FUSE write implementation /*! * \param p_buf Buffer containing data to write * \param size Number of bytes to write * \param offset Offset to start writing at * \param p_fi: File info struct * * Returns: * Written bytes on success, negated error code on error */ static int FuseWrite(const char *p_path, const char *p_buf, size_t size, off_t offset, struct fuse_file_info *p_fi) { (void)p_fi; uint64_t len; if(strcmp(p_path,glob_xmount.output.p_virtual_image_path)==0) { // Wait for other threads to end reading/writing data pthread_mutex_lock(&(glob_xmount.mutex_image_rw)); // Get virtual image file size if(!GetVirtImageSize(&len)) { LOG_ERROR("Couldn't get virtual image size!\n") pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); return 0; } if(offsetlen) size=len-offset; if(SetVirtImageData(p_buf,offset,size)!=size) { LOG_ERROR("Couldn't write data to virtual image file!\n") pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); return 0; } } else { LOG_DEBUG("Attempt to write past EOF of virtual image file\n") pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); return 0; } // Allow other threads to read/write data again pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); } else if(strcmp(p_path,glob_xmount.output.vmdk.p_virtual_vmdk_path)==0) { pthread_mutex_lock(&(glob_xmount.mutex_image_rw)); len=glob_xmount.output.vmdk.vmdk_file_size; if((offset+size)>len) { // Enlarge or create buffer if needed if(len==0) { len=offset+size; XMOUNT_MALLOC(glob_xmount.output.vmdk.p_vmdk_file, char*, len*sizeof(char)); } else { len=offset+size; XMOUNT_REALLOC(glob_xmount.output.vmdk.p_vmdk_file, char*, len*sizeof(char)); } glob_xmount.output.vmdk.vmdk_file_size=offset+size; } // Copy data to buffer memcpy(glob_xmount.output.vmdk.p_vmdk_file+offset,p_buf,size); pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); } else if(glob_xmount.output.vmdk.p_vmdk_lockfile_name!=NULL && strcmp(p_path,glob_xmount.output.vmdk.p_vmdk_lockfile_name)==0) { pthread_mutex_lock(&(glob_xmount.mutex_image_rw)); if((offset+size)>glob_xmount.output.vmdk.vmdk_lockfile_size) { // Enlarge or create buffer if needed if(glob_xmount.output.vmdk.vmdk_lockfile_size==0) { glob_xmount.output.vmdk.vmdk_lockfile_size=offset+size; XMOUNT_MALLOC(glob_xmount.output.vmdk.p_vmdk_lockfile_data,char*, glob_xmount.output.vmdk.vmdk_lockfile_size*sizeof(char)) } else { glob_xmount.output.vmdk.vmdk_lockfile_size=offset+size; XMOUNT_REALLOC(glob_xmount.output.vmdk.p_vmdk_lockfile_data,char*, glob_xmount.output.vmdk.vmdk_lockfile_size*sizeof(char)) } } // Copy data to buffer memcpy(glob_xmount.output.vmdk.p_vmdk_lockfile_data+offset,p_buf,size); pthread_mutex_unlock(&(glob_xmount.mutex_image_rw)); } else if(strcmp(p_path,glob_xmount.output.p_info_path)==0) { // Attempt to write data to read only image info file LOG_DEBUG("Attempt to write data to virtual info file\n"); return -ENOENT; } else { // Attempt to write to non existant file LOG_DEBUG("Attempt to write to the non existant file \"%s\"\n",p_path) return -ENOENT; } return size; } /******************************************************************************* * 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=FuseGetAttr, .mkdir=FuseMkDir, .mknod=FuseMkNod, .open=FuseOpen, .readdir=FuseReadDir, .read=FuseRead, .rename=FuseRename, .rmdir=FuseRmDir, //.statfs=FuseStatFs, .unlink=FuseUnlink, .write=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;ifiles_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(!ExtractVirtFileNames(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") // Gather infos for info file if(!InitVirtImageInfoFile()) { LOG_ERROR("Couldn't gather infos for virtual image info file!\n") FreeResources(); return 1; } LOG_DEBUG("Virtual image info file build successfully\n") // Do some virtual image type specific initialisations switch(glob_xmount.output.VirtImageType) { case VirtImageType_DD: case VirtImageType_DMG: break; case VirtImageType_VDI: // When mounting as VDI, we need to construct a vdi header if(!InitVirtVdiHeader()) { LOG_ERROR("Couldn't initialize virtual VDI file header!\n") FreeResources(); return 1; } LOG_DEBUG("Virtual VDI file header build successfully\n") break; case VirtImageType_VHD: // When mounting as VHD, we need to construct a vhd footer if(!InitVirtVhdHeader()) { LOG_ERROR("Couldn't initialize virtual VHD file footer!\n") FreeResources(); return 1; } LOG_DEBUG("Virtual VHD file footer build successfully\n") break; case VirtImageType_VMDK: case VirtImageType_VMDKS: // When mounting as VMDK, we need to construct the VMDK descriptor file if(!InitVirtualVmdkFile()) { LOG_ERROR("Couldn't initialize virtual VMDK file!\n") FreeResources(); return 1; } break; } 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; } /* ----- Change log ----- 20090131: v0.1.0 released * Some minor things have still to be done. * Mounting ewf as dd: Seems to work. Diff didn't complain about changes between original dd and emulated dd. * Mounting ewf as vdi: Seems to work too. VBox accepts the emulated vdi as valid vdi file and I was able to mount the containing fs under Debian. INFO: Debian freezed when not using mount -r !! 20090203: v0.1.1 released * Multiple code improvements. For ex. cleaner vdi header allocation. * Fixed severe bug in image block calculation. Didn't check for odd input in conversion from bytes to megabytes. * Added more debug output 20090210: v0.1.2 released * Fixed compilation problem (Typo in image_init_info() function). * Fixed some problems with the debian scripts to be able to build packages. * Added random generator initialisation (Makes it possible to use more than one image in VBox at a time). 20090215: * Added function init_cache_blocks which creates / loads a cache file used to implement virtual write capability. 20090217: * Implemented the fuse write function. Did already some basic tests with dd and it seems to work. But there are certainly still some bugs left as there are also still some TODO's left. 20090226: * Changed program name from mountewf to xmount. * Began with massive code cleanups to ease full implementation of virtual write support and to be able to support multiple input image formats (DD, EWF and AFF are planned for now). * Added defines for supported input formats so it should be possible to compile xmount without supporting all input formats. (DD input images are always supported as these do not require any additional libs). Input formats should later be en/disabled by the configure script in function to which libs it detects. * GetOrigImageSize function added to get the size of the original image whatever type it is in. * GetOrigImageData function added to retrieve data from original image file whatever type it is in. * GetVirtImageSize function added to get the size of the virtual image file. * Cleaned function mountewf_getattr and renamed it to GetVirtFileAttr * Cleaned function mountewf_readdir and renamed it to GetVirtFiles * Cleaned function mountewf_open and renamed it to OpenVirtFile 20090227: * Cleaned function init_info_file and renamed it to InitVirtImageInfoFile 20090228: * Cleaned function init_cache_blocks and renamed it to InitCacheFile * Added LogMessage function to ease error and debug logging (See also LOG_ERROR and LOG_DEBUG macros in xmount.h) * Cleaned function init_vdi_header and renamed it to InitVirtVdiHeader * Added PrintUsage function to print out xmount usage informations * Cleaned function parse_cmdline and renamed it to ParseCmdLine * Cleaned function main * Added ExtractVirtFileNames function to extract virtual file names from input image name * Added function GetVirtImageData to retrieve data from the virtual image file. This includes reading data from cache file if virtual write support is enabled. * Added function ReadVirtFile to replace mountewf_read 20090229: * Fixed a typo in virtual file name creation * Added function SetVirtImageData to write data to virtual image file. This includes writing data to cache file and caching entire new blocks * Added function WriteVirtFile to replace mountewf_write 20090305: * Solved a problem that made it impossible to access offsets >32bit 20090308: * Added SetVdiFileHeaderData function to handle virtual image type specific data to be cached. This makes cache files independent from virtual image type 20090316: v0.2.0 released 20090327: v0.2.1 released * Fixed a bug in virtual write support. Checking whether data is cached didn't use semaphores. This could corrupt cache files when running multi-threaded. * Added IsVdiFileHeaderCached function to check whether VDI file header was already cached * Added IsBlockCached function to check whether a block was already cached 20090331: v0.2.2 released (Internal release) * Further changes to semaphores to fix write support bug. 20090410: v0.2.3 released * Reverted most of the fixes from v0.2.1 and v0.2.2 as those did not solve the write support bug. * Removed all semaphores * Added two pthread mutexes to protect virtual image and virtual info file. 20090508: * Configure script will now exit when needed libraries aren't found * Added support for newest libewf beta version 20090506 as it seems to reduce memory usage when working with EWF files by about 1/2. * Added LIBEWF_BETA define to adept source to new libewf API. * Added function InitVirtualVmdkFile to build a VmWare virtual disk descriptor file. 20090519: * Added function CreateVirtDir implementing FUSE's mkdir to allow VMWare to create his .vmdk.lck lock folder. Function does not allow to create other folders! * Changed cache file handling as VMDK caching will need new cache file structure incompatible to the old one. 20090522: v0.3.0 released * Added function DeleteVirtFile and DeleteVirtDir so VMWare can remove his lock directories and files. * Added function RenameVirtFile because VMWare needs to rename his lock files. * VMDK support should work now but descriptor file won't get cached as I didn't implement it yet. 20090604: * Added --cache commandline parameter doing the same as --rw. * Added --owcache commandline parameter doing the same as --rw but overwrites any existing cache data. This can be handy for debugging and testing purposes. * Added "vmdks" output type. Same as "vmdk" but generates a disk connected to the SCSI bus rather than the IDE bus. 20090710: v0.3.1 released 20090721: * Added function CheckFuseAllowOther to check wether FUSE supports the "-o allow_other" option. It is supported when "user_allow_other" is set in /etc/fuse.conf or when running xmount as root. * Automatic addition of FUSE's "-o allow_other" option if it is supported. * Added special "-o no_allow_other" command line parameter to disable automatic addition of the above option. * Reorganisation of FUSE's and xmount's command line options processing. * Added LogWarnMessage function to output a warning message. 20090722: * Added function CalculateInputImageHash to calculate an MD5 hash of the first input image's HASH_AMOUNT bytes of data. This hash is used as VDI creation UUID and will later be used to match cache files to input images. 20090724: v0.3.2 released 20090725: v0.4.0 released * Added AFF input image support. * Due to various problems with libewf and libaff packages (Mainly in Debian and Ubuntu), I decided to include them into xmount's source tree and link them in statically. This has the advantage that I can use whatever version I want. 20090727: v0.4.1 released * Added again the ability to compile xmount with shared libs as the Debian folks don't like the static ones :) 20090812: * Added TXMountConfData.OrigImageSize and TXMountConfData.VirtImageSize to save the size of the input and output image in order to avoid regetting it always from disk. 20090814: * Replaced all malloc and realloc occurences with the two macros XMOUNT_MALLOC and XMOUNT_REALLOC. 20090816: * Replaced where applicable all occurences of str(n)cpy or alike with their corresponding macros XMOUNT_STRSET, XMOUNT_STRCPY and XMOUNT_STRNCPY pendants. 20090907: v0.4.2 released * Fixed a bug in VMDK lock file access. glob_vmdk_lockfile_size wasn't reset to 0 when the file was deleted. * Fixed a bug in VMDK descriptor file access. Had to add glob_vmdk_file_size to track the size of this file as strlen was a bad idea :). 20100324: v0.4.3 released * Changed all header structs to prevent different sizes on i386 and amd64. See xmount.h for more details. 20100810: v0.4.4 released * Found a bug in InitVirtVdiHeader(). The 64bit values were addressed incorrectly while filled with rand(). This leads to an error message when trying to add a VDI file to VirtualBox 3.2.8. 20110210: * Adding subtype and fsname FUSE options in order to display mounted source in mount command output. 20110211: v0.4.5 released 20111011: * Changes to deal with libewf v2 API (thx to Joachim Metz) 20111109: v0.4.6 released * Added support for DMG output type (actually a DD with .dmg file extension). This type is used as default output type when using xmount under Mac OS X. 20120130: v0.4.7 released * Made InitVirtImageInfoFile less picky about missing EWF infos. 20120507: * Added support for VHD output image as requested by various people. * Statically linked libs updated to 20120504 (libewf) and 3.7.0 (afflib). 20120510: v0.5.0 released * Added stbuf->st_blocks calculation for VHD images in function GetVirtFileAttr. This makes Windows not think the emulated file would be a sparse file. Sparse vhd files are not attachable in Windows. 20130726: v0.6.0 released * Added libaaff to replace libaff (thx to Guy Voncken). * Added libdd to replace raw dd input file handling and finally support split dd files (thx to Guy Voncken). 20140311: * Added libaewf (thx to Guy Voncken). 20140726: * Added support for dynamically loading of input libs. This should ease adding support for new input image formats in the future. * Moved input image functions to their corresponding dynamically loadable libs. * Prepended "glob_" to all global vars for better identification. 20140731: * Added --offset option as requested by HPM. * Began massive code cleanup. 20140803: * Added correct return code handling when calling input lib functions including getting error messages using GetErrorMessage. * Added input lib specific option parsing. * Re-implemented InitVirtImageInfoFile() using input lib's GetInfofileContent() function. * Further code cleanups. 20140807: * Further code cleanups. * Renamed GetVirtFileAttr() to FuseGetAttr() * Renamed CreateVirtDir() to FuseMkDir() * Renamed CreateVirtDir() to FuseMkNod() * Renamed OpenVirtFile() to FuseOpen() * Renamed GetVirtFiles() to FuseReadDir() * Renamed ReadVirtFile() to FuseRead() * Renamed RenameVirtFile() to FuseRename() * Renamed DeleteVirtDir() to FuseRmDir() * Renamed DeleteVirtFile() to FuseUnlink() * Renamed WriteVirtFile() to FuseWrite() * Fixed bug in CalculateInputImageHash() where always HASH_AMOUNT bytes were hased even if input image is smaller. * Fixed a newly introduced bug in FuseRead() and GetVirtImageData() returning -EIO when trying to read behind EOF. The correct return value is 0. 20140811: * Renamed CheckFuseAllowOther() to CheckFuseSettings() and added a check to see if user is part of the fuse group. 20140814: * Replaced cuserid() with getpwuid(geteuid()) in CheckFuseSettings() as it is deprecated on Linux and not available on OSx. * Only build fuse group checks from CheckFuseSettings() on Linux. 20140820: * Added libxmount_morphing handling. * Renamed LoadInputLibs() to LoadLibs() and added code to find and load morphing libs. * Added FindMorphingLib() function. 20140821: * Changed command line syntax to support specification of multiple input images of different types. * Added code in main() to initialize and open multiple input images. * Added GetMorphedImageSize() and GetMorphedImageData(). 20140822: * Removed GetInputImageSize() as it isn't needed anymore. * Quick test showed morphing using libxmount_morphing_combine seems to work. 20140825: * Added InitResources() and FreeResources() functions. 20140926: v0.7.0 released 20140930: v0.7.1 released * Fixed bug with --sizelimit command line option. 20141001: v0.7.2 released * Fixed bug in FreeResources(). Do not free vdi.p_vdi_block_map as it is part of vdi.p_vdi_header. 20141113: v0.7.3 released * Fixed minor bug in ParseCmdLine() which was not checking upper argc bounds correctly. */ diff --git a/trunk/src/xmount.h b/trunk/src/xmount.h index 20fb68b..19fd882 100755 --- a/trunk/src/xmount.h +++ b/trunk/src/xmount.h @@ -1,490 +1,490 @@ /******************************************************************************* -* xmount Copyright (c) 2008-2014 by Gillen Daniel * +* xmount Copyright (c) 2008-2015 by Gillen Daniel * * * * 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 . * *******************************************************************************/ #include "../libxmount_input/libxmount_input.h" #include "../libxmount_morphing/libxmount_morphing.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" /******************************************************************************* * Structures of output images ******************************************************************************/ #define VDI_FILE_COMMENT "<<< This is a virtual VDI image >>>" #define VDI_HEADER_COMMENT "This VDI was emulated using xmount v" XMOUNT_VERSION #define VDI_IMAGE_SIGNATURE 0xBEDA107F // 1:1 copy from hp #define VDI_IMAGE_VERSION 0x00010001 // Vers 1.1 #define VDI_IMAGE_TYPE_FIXED 0x00000002 // Type 2 (fixed size) #define VDI_IMAGE_FLAGS 0 #define VDI_IMAGE_BLOCK_SIZE (1024*1024) // 1 Megabyte //! VDI Binary File Header structure typedef struct s_VdiFileHeader { // ----- VDIPREHEADER ------ //! Just text info about image type, for eyes only char szFileInfo[64]; //! The image signature (VDI_IMAGE_SIGNATURE) uint32_t u32Signature; //! The image version (VDI_IMAGE_VERSION) uint32_t u32Version; // ----- VDIHEADER1PLUS ----- //! Size of header structure in bytes. uint32_t cbHeader; //! The image type (VDI_IMAGE_TYPE_*) uint32_t u32Type; //! Image flags (VDI_IMAGE_FLAGS_*) uint32_t fFlags; //! Image comment (UTF-8) char szComment[256]; //! Offset of Blocks array from the begining of image file. // Should be sector-aligned for HDD access optimization. uint32_t offBlocks; //! Offset of image data from the begining of image file. // Should be sector-aligned for HDD access optimization. uint32_t offData; //! Legacy image geometry - Cylinders uint32_t cCylinders; //! Legacy image geometry - Heads uint32_t cHeads; //! Legacy image geometry - Sectors per track uint32_t cSectors; //! Legacy image geometry - Sector size (bytes per sector) uint32_t cbSector; //! Was BIOS HDD translation mode, now unused uint32_t u32Dummy; //! Size of disk (in bytes) uint64_t cbDisk; //! Block size (for instance VDI_IMAGE_BLOCK_SIZE). Must be a power of 2! uint32_t cbBlock; //! Size of additional service information of every data block. // Prepended before block data. May be 0. // Should be a power of 2 and sector-aligned for optimization reasons. uint32_t cbBlockExtra; //! Number of blocks uint32_t cBlocks; //! Number of allocated blocks uint32_t cBlocksAllocated; //! UUID of image (lower 64 bit) uint64_t uuidCreate_l; //! UUID of image (higher 64 bit) uint64_t uuidCreate_h; //! UUID of image's last modification (lower 64 bit) uint64_t uuidModify_l; //! UUID of image's last modification (higher 64 bit) uint64_t uuidModify_h; //! Only for secondary images - UUID of previous image (lower 64 bit) uint64_t uuidLinkage_l; //! Only for secondary images - UUID of previous image (higher 64 bit) uint64_t uuidLinkage_h; //! Only for secondary images - UUID of prev image's last mod (lower 64 bit) uint64_t uuidParentModify_l; //! Only for secondary images - UUID of prev image's last mod (higher 64 bit) uint64_t uuidParentModify_h; //! Padding to get 512 byte alignment char padding[56]; } __attribute__ ((packed)) ts_VdiFileHeader, *pts_VdiFileHeader; // /** The way the UUID is declared by the DCE specification. */ // struct // { // uint32_t u32TimeLow; // uint16_t u16TimeMid; // uint16_t u16TimeHiAndVersion; // uint8_t u8ClockSeqHiAndReserved; // uint8_t u8ClockSeqLow; // uint8_t au8Node[6]; // } Gen; /* * VHD Binary File footer structure * * At the time of writing, the specs could be found here: * http://www.microsoft.com/downloads/details.aspx? * FamilyID=C2D03242-2FFB-48EF-A211-F0C44741109E * * Warning: All values are big-endian! */ // #ifdef __LP64__ #define VHD_IMAGE_HVAL_COOKIE 0x78697463656E6F63 // "conectix" #else #define VHD_IMAGE_HVAL_COOKIE 0x78697463656E6F63LL #endif #define VHD_IMAGE_HVAL_FEATURES 0x02000000 #define VHD_IMAGE_HVAL_FILE_FORMAT_VERSION 0x00000100 #ifdef __LP64__ #define VHD_IMAGE_HVAL_DATA_OFFSET 0xFFFFFFFFFFFFFFFF #else #define VHD_IMAGE_HVAL_DATA_OFFSET 0xFFFFFFFFFFFFFFFFLL #endif #define VHD_IMAGE_HVAL_CREATOR_APPLICATION 0x746E6D78 // "xmnt" #define VHD_IMAGE_HVAL_CREATOR_VERSION 0x00000500 // This one is funny! According to VHD specs, I can only choose between Windows // and Macintosh. I'm going to choose the most common one. #define VHD_IMAGE_HVAL_CREATOR_HOST_OS 0x6B326957 // "Win2k" #define VHD_IMAGE_HVAL_DISK_TYPE 0x02000000 // Seconds from January 1st, 1970 to January 1st, 2000 #define VHD_IMAGE_TIME_CONVERSION_OFFSET 0x386D97E0 typedef struct s_VhdFileHeader { uint64_t cookie; uint32_t features; uint32_t file_format_version; uint64_t data_offset; uint32_t creation_time; uint32_t creator_app; uint32_t creator_ver; uint32_t creator_os; uint64_t size_original; uint64_t size_current; uint16_t disk_geometry_c; uint8_t disk_geometry_h; uint8_t disk_geometry_s; uint32_t disk_type; uint32_t checksum; uint64_t uuid_l; uint64_t uuid_h; uint8_t saved_state; char reserved[427]; } __attribute__ ((packed)) ts_VhdFileHeader, *pts_VhdFileHeader; /******************************************************************************* * Xmount specific structures ******************************************************************************/ #ifdef __LP64__ #define CACHE_BLOCK_FREE 0xFFFFFFFFFFFFFFFF #else #define CACHE_BLOCK_FREE 0xFFFFFFFFFFFFFFFFLL #endif //! Cache file block index array element 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; //! Virtual image types typedef enum e_VirtImageType { //! Virtual image is a DD file VirtImageType_DD, //! Virtual image is a DMG file VirtImageType_DMG, //! Virtual image is a VDI file VirtImageType_VDI, //! Virtual image is a VMDK file (IDE bus) VirtImageType_VMDK, //! Virtual image is a VMDK file (SCSI bus) VirtImageType_VMDKS, //! Virtual image is a VHD file VirtImageType_VHD } te_VirtImageType; //! 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 typedef struct s_CacheData { //! Cache file to save changes to char *p_cache_file; //! Handle to cache file FILE *h_cache_file; //! Overwrite existing cache uint8_t overwrite_cache; //! Cache header pts_CacheFileHeader p_cache_header; //! Cache block index pts_CacheFileBlockIndex p_cache_blkidx; } ts_CacheData; //! Structures and vars needed for VDI support typedef struct s_OutputImageVdiData { //! VDI header size uint32_t vdi_header_size; //! VDI header pts_VdiFileHeader p_vdi_header; //! VDI blockmap size uint32_t vdi_block_map_size; //! VDI block map char *p_vdi_block_map; } ts_OutputImageVdiData; //! Structures and vars needed for VHD support typedef struct s_OutputImageVhdData { //! VHD header ts_VhdFileHeader *p_vhd_header; } ts_OutputImageVhdData; //! Structures and vars needed for VMDK support typedef struct s_OutputImageVmdkData { //! Path of virtual VMDK file char *p_virtual_vmdk_path; //! VMDK content char *p_vmdk_file; //! VMDK content size int vmdk_file_size; //! Path of 1st lockdir char *p_vmdk_lockdir1; //! Path of 2nd lockdir char *p_vmdk_lockdir2; //! Lockfile content char *p_vmdk_lockfile_data; //! Lockfile content size int vmdk_lockfile_size; //! Path of lockfile char *p_vmdk_lockfile_name; } ts_OutputImageVmdkData; //! Structure containing infos about output image typedef struct s_OutputData { //! Virtual image type te_VirtImageType VirtImageType; //! 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; //! VDI related data ts_OutputImageVdiData vdi; //! VHD related data ts_OutputImageVhdData vhd; //! VMDK related data ts_OutputImageVmdkData vmdk; } 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; /* ----- Change log ----- 20090226: * Added change history information to this file. * Added TVirtImageType enum to identify virtual image type. * Added TOrigImageType enum to identify input image type. * Added TMountimgConfData struct to hold various mountimg runtime options. * Renamed VDIFILEHEADER to ts_VdiFileHeader. 20090228: * Added LOG_ERROR and LOG_DEBUG macros * Added defines for various static VDI header values * Added defines for TRUE and FALSE 20090307: * Added defines for various static cache file header values * Added VdiFileHeaderCached and pVdiFileHeader values to be able to cache the VDI file header separatly. 20090519: * Added new cache file header structure and moved old one to ts_CacheFileHeader_v1. * New cache file structure includes VmdkFileCached and pVmdkFile to cache virtual VMDK file and makes room for further additions so current cache file version 2 cache files can be easily converted to newer ones. 20090814: * Added XMOUNT_MALLOC and XMOUNT_REALLOC macros. 20090816: * Added XMOUNT_STRSET, XMOUNT_STRNSET, XMOUNT_STRAPP and XMOUNT_STRNAPP macros. 20100324: * Added "__attribute__ ((packed))" to all header structs to prevent different sizes on i386 and amd64. 20111109: * Added TVirtImageType_DMG type. 20120130: * Added LOG_WARNING macro. 20120507: * Added ts_VhdFileHeader structure. 20120511: * Added endianness conversation macros 20140809: * Moved endianness macros to separate file 20140810: * Moved convenience macros to separate file 20140726: * Added ts_InputLib 20140821: * Added ts_InputImage and ts_InputData * Moved data from various places to the above structs. 20140825: * Added ts_MorphingLib, ts_CacheData, ts_OutputImageVdiData, ts_OutputImageVhdData, ts_OutputImageVmdkData and ts_OutputData. * Moved data from various places to the above structs. */