-
Notifications
You must be signed in to change notification settings - Fork 45
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
bd8331f
commit 6c6ce51
Showing
35 changed files
with
1,538 additions
and
40 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,52 @@ | ||
| # -*- coding: utf-8 -*- | ||
| """The FAT format analyzer helper implementation.""" | ||
|
|
||
| from dfvfs.analyzer import analyzer | ||
| from dfvfs.analyzer import analyzer_helper | ||
| from dfvfs.analyzer import specification | ||
| from dfvfs.lib import definitions | ||
|
|
||
|
|
||
| class FATAnalyzerHelper(analyzer_helper.AnalyzerHelper): | ||
| """FAT analyzer helper.""" | ||
|
|
||
| FORMAT_CATEGORIES = frozenset([ | ||
| definitions.FORMAT_CATEGORY_FILE_SYSTEM]) | ||
|
|
||
| TYPE_INDICATOR = definitions.TYPE_INDICATOR_FAT | ||
|
|
||
| def GetFormatSpecification(self): | ||
| """Retrieves the format specification. | ||
| Returns: | ||
| FormatSpecification: format specification or None if the format cannot | ||
| be defined by a specification object. | ||
| """ | ||
| format_specification = specification.FormatSpecification( | ||
| self.type_indicator) | ||
|
|
||
| # Boot sector signature. | ||
| format_specification.AddNewSignature(b'\x55\xaa', offset=510) | ||
|
|
||
| # FAT-12 and FAT-16 file system hint. | ||
| format_specification.AddNewSignature(b'FAT12 ', offset=54) | ||
| format_specification.AddNewSignature(b'FAT16 ', offset=54) | ||
|
|
||
| # FAT-32 file system hint. | ||
| format_specification.AddNewSignature(b'FAT32 ', offset=82) | ||
|
|
||
| # exFAT file system signature. | ||
| format_specification.AddNewSignature(b'EXFAT ', offset=3) | ||
|
|
||
| return format_specification | ||
|
|
||
| def IsEnabled(self): | ||
| """Determines if the analyzer helper is enabled. | ||
| Returns: | ||
| bool: True if the analyzer helper is enabled. | ||
| """ | ||
| return definitions.PREFERRED_FAT_BACK_END == self.TYPE_INDICATOR | ||
|
|
||
|
|
||
| analyzer.Analyzer.RegisterHelper(FATAnalyzerHelper()) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,149 @@ | ||
| # -*- coding: utf-8 -*- | ||
| """The File Allocation Table (FAT) file-like object implementation.""" | ||
|
|
||
| import os | ||
|
|
||
| from dfvfs.file_io import file_io | ||
| from dfvfs.resolver import resolver | ||
|
|
||
|
|
||
| class FATFile(file_io.FileIO): | ||
| """File input/output (IO) object using pyfsfat.file_entry""" | ||
|
|
||
| def __init__(self, resolver_context, path_spec): | ||
| """Initializes a file input/output (IO) object. | ||
| Args: | ||
| resolver_context (Context): resolver context. | ||
| path_spec (PathSpec): a path specification. | ||
| """ | ||
| super(FATFile, self).__init__(resolver_context, path_spec) | ||
| self._file_system = None | ||
| self._fsfat_data_stream = None | ||
| self._fsfat_file_entry = None | ||
|
|
||
| def _Close(self): | ||
| """Closes the file-like object.""" | ||
| self._fsfat_data_stream = None | ||
| self._fsfat_file_entry = None | ||
|
|
||
| self._file_system = None | ||
|
|
||
| def _Open(self, mode='rb'): | ||
| """Opens the file-like object defined by path specification. | ||
| Args: | ||
| mode (Optional[str]): file access mode. | ||
| Raises: | ||
| AccessError: if the access to open the file was denied. | ||
| IOError: if the file-like object could not be opened. | ||
| NotSupported: if a data stream, like the resource or named fork, is | ||
| requested to be opened. | ||
| OSError: if the file-like object could not be opened. | ||
| PathSpecError: if the path specification is incorrect. | ||
| """ | ||
| data_stream_name = getattr(self._path_spec, 'data_stream', None) | ||
|
|
||
| self._file_system = resolver.Resolver.OpenFileSystem( | ||
| self._path_spec, resolver_context=self._resolver_context) | ||
|
|
||
| file_entry = self._file_system.GetFileEntryByPathSpec(self._path_spec) | ||
| if not file_entry: | ||
| raise IOError('Unable to open file entry.') | ||
|
|
||
| fsfat_data_stream = None | ||
| fsfat_file_entry = file_entry.GetFATFileEntry() | ||
| if not fsfat_file_entry: | ||
| raise IOError('Unable to open FAT file entry.') | ||
|
|
||
| if data_stream_name == 'rsrc': | ||
| fsfat_data_stream = fsfat_file_entry.get_resource_fork() | ||
| elif data_stream_name: | ||
| raise IOError('Unable to open data stream: {0:s}.'.format( | ||
| data_stream_name)) | ||
|
|
||
| self._fsfat_data_stream = fsfat_data_stream | ||
| self._fsfat_file_entry = fsfat_file_entry | ||
|
|
||
| # Note: that the following functions do not follow the style guide | ||
| # because they are part of the file-like object interface. | ||
| # pylint: disable=invalid-name | ||
|
|
||
| def read(self, size=None): | ||
| """Reads a byte string from the file-like object at the current offset. | ||
| The function will read a byte string of the specified size or | ||
| all of the remaining data if no size was specified. | ||
| Args: | ||
| size (Optional[int]): number of bytes to read, where None is all | ||
| remaining data. | ||
| Returns: | ||
| bytes: data read. | ||
| Raises: | ||
| IOError: if the read failed. | ||
| OSError: if the read failed. | ||
| """ | ||
| if not self._is_open: | ||
| raise IOError('Not opened.') | ||
|
|
||
| if self._fsfat_data_stream: | ||
| return self._fsfat_data_stream.read(size=size) | ||
| return self._fsfat_file_entry.read(size=size) | ||
|
|
||
| def seek(self, offset, whence=os.SEEK_SET): | ||
| """Seeks to an offset within the file-like object. | ||
| Args: | ||
| offset (int): offset to seek to. | ||
| whence (Optional(int)): value that indicates whether offset is an absolute | ||
| or relative position within the file. | ||
| Raises: | ||
| IOError: if the seek failed. | ||
| OSError: if the seek failed. | ||
| """ | ||
| if not self._is_open: | ||
| raise IOError('Not opened.') | ||
|
|
||
| if self._fsfat_data_stream: | ||
| self._fsfat_data_stream.seek(offset, whence) | ||
| else: | ||
| self._fsfat_file_entry.seek(offset, whence) | ||
|
|
||
| def get_offset(self): | ||
| """Retrieves the current offset into the file-like object. | ||
| Return: | ||
| int: current offset into the file-like object. | ||
| Raises: | ||
| IOError: if the file-like object has not been opened. | ||
| OSError: if the file-like object has not been opened. | ||
| """ | ||
| if not self._is_open: | ||
| raise IOError('Not opened.') | ||
|
|
||
| if self._fsfat_data_stream: | ||
| return self._fsfat_data_stream.get_offset() | ||
| return self._fsfat_file_entry.get_offset() | ||
|
|
||
| def get_size(self): | ||
| """Retrieves the size of the file-like object. | ||
| Returns: | ||
| int: size of the file-like object data. | ||
| Raises: | ||
| IOError: if the file-like object has not been opened. | ||
| OSError: if the file-like object has not been opened. | ||
| """ | ||
| if not self._is_open: | ||
| raise IOError('Not opened.') | ||
|
|
||
| if self._fsfat_data_stream: | ||
| return self._fsfat_data_stream.get_size() | ||
| return self._fsfat_file_entry.get_size() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.