-
Notifications
You must be signed in to change notification settings - Fork 117
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implement file storage abstraction to allow using any fs.File as a st…
…orage Separate interfaces for Read and Write storage operations Introduce diskfs.OpenStorage(fs.File,...) to newly implemented features.
- Loading branch information
Showing
53 changed files
with
838 additions
and
467 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,33 @@ | ||
| package backend | ||
|
|
||
| import ( | ||
| "errors" | ||
| "io" | ||
| "io/fs" | ||
| "os" | ||
| ) | ||
|
|
||
| var ( | ||
| ErrIncorrectOpenMode = errors.New("disk file or device not open for write") | ||
| ErrNotSuitable = errors.New("backing file is not suitable") | ||
| ) | ||
|
|
||
| type File interface { | ||
| fs.File | ||
| io.ReaderAt | ||
| io.Seeker | ||
| io.Closer | ||
| } | ||
|
|
||
| type WritableFile interface { | ||
| File | ||
| io.WriterAt | ||
| } | ||
|
|
||
| type Storage interface { | ||
| File | ||
| // OS-stecific file for ioctl calls via fd | ||
| Sys() (*os.File, error) | ||
| // file for read-write operations | ||
| Writable() (WritableFile, error) | ||
| } |
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,127 @@ | ||
| package raw | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "io" | ||
| "io/fs" | ||
| "os" | ||
|
|
||
| "github.com/diskfs/go-diskfs/backend" | ||
| ) | ||
|
|
||
| type rawBackend struct { | ||
| storage fs.File | ||
| readOnly bool | ||
| } | ||
|
|
||
| // Create a backend.Storage from provided fs.File | ||
| func New(f fs.File, readOnly bool) backend.Storage { | ||
| return rawBackend{ | ||
| storage: f, | ||
| readOnly: readOnly, | ||
| } | ||
| } | ||
|
|
||
| // Create a backend.Storage from a path to a device | ||
| // Should pass a path to a block device e.g. /dev/sda or a path to a file /tmp/foo.img | ||
| // The provided device/file must exist at the time you call OpenFromPath() | ||
| func OpenFromPath(pathName string, readOnly bool) (backend.Storage, error) { | ||
| if pathName == "" { | ||
| return nil, errors.New("must pass device of file name") | ||
| } | ||
|
|
||
| if _, err := os.Stat(pathName); os.IsNotExist(err) { | ||
| return nil, fmt.Errorf("provided device/file %s does not exist", pathName) | ||
| } | ||
|
|
||
| openMode := os.O_RDONLY | ||
|
|
||
| if !readOnly { | ||
| openMode |= os.O_RDWR | os.O_EXCL | ||
| } | ||
|
|
||
| f, err := os.OpenFile(pathName, openMode, 0o600) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("could not open device %s with mode %v: %w", pathName, openMode, err) | ||
| } | ||
|
|
||
| return rawBackend{ | ||
| storage: f, | ||
| readOnly: readOnly, | ||
| }, nil | ||
| } | ||
|
|
||
| // Create a backend.Storage from a path to an image file. | ||
| // Should pass a path to a file /tmp/foo.img | ||
| // The provided file must not exist at the time you call CreateFromPath() | ||
| func CreateFromPath(pathName string, size int64) (backend.Storage, error) { | ||
| if pathName == "" { | ||
| return nil, errors.New("must pass device name") | ||
| } | ||
| if size <= 0 { | ||
| return nil, errors.New("must pass valid device size to create") | ||
| } | ||
| f, err := os.OpenFile(pathName, os.O_RDWR|os.O_EXCL|os.O_CREATE, 0o666) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("could not create device %s: %w", pathName, err) | ||
| } | ||
| err = os.Truncate(pathName, size) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("could not expand device %s to size %d: %w", pathName, size, err) | ||
| } | ||
|
|
||
| return rawBackend{ | ||
| storage: f, | ||
| readOnly: false, | ||
| }, nil | ||
| } | ||
|
|
||
| // backend.Storage interface guard | ||
| var _ backend.Storage = (*rawBackend)(nil) | ||
|
|
||
| // OS-stecific file for ioctl calls via fd | ||
| func (f rawBackend) Sys() (*os.File, error) { | ||
| if osFile, ok := f.storage.(*os.File); ok { | ||
| return osFile, nil | ||
| } | ||
| return nil, backend.ErrNotSuitable | ||
| } | ||
|
|
||
| // file for read-write operations | ||
| func (f rawBackend) Writable() (backend.WritableFile, error) { | ||
| if rwFile, ok := f.storage.(backend.WritableFile); ok { | ||
| if !f.readOnly { | ||
| return rwFile, nil | ||
| } | ||
|
|
||
| return nil, backend.ErrIncorrectOpenMode | ||
| } | ||
| return nil, backend.ErrNotSuitable | ||
| } | ||
|
|
||
| func (f rawBackend) Stat() (fs.FileInfo, error) { | ||
| return f.storage.Stat() | ||
| } | ||
|
|
||
| func (f rawBackend) Read(b []byte) (int, error) { | ||
| return f.storage.Read(b) | ||
| } | ||
|
|
||
| func (f rawBackend) Close() error { | ||
| return f.storage.Close() | ||
| } | ||
|
|
||
| func (f rawBackend) ReadAt(p []byte, off int64) (n int, err error) { | ||
| if readerAt, ok := f.storage.(io.ReaderAt); ok { | ||
| return readerAt.ReadAt(p, off) | ||
| } | ||
| return -1, backend.ErrNotSuitable | ||
| } | ||
|
|
||
| func (f rawBackend) Seek(offset int64, whence int) (int64, error) { | ||
| if seeker, ok := f.storage.(io.Seeker); ok { | ||
| return seeker.Seek(offset, whence) | ||
| } | ||
| return -1, backend.ErrNotSuitable | ||
| } |
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.