package storage
import (
+ "fmt"
"io"
"os"
"path/filepath"
- "github.com/anacrolix/missinggo"
+ "github.com/anacrolix/missinggo/v2"
+ "github.com/anacrolix/torrent/common"
"github.com/anacrolix/torrent/metainfo"
+ "github.com/anacrolix/torrent/segments"
)
-// File-based storage for torrents, that isn't yet bound to a particular
-// torrent.
-type fileStorage struct {
- baseDir string
- completed map[[20]byte]bool
+// File-based storage for torrents, that isn't yet bound to a particular torrent.
+type fileClientImpl struct {
+ opts NewFileClientOpts
}
-func NewFile(baseDir string) Client {
- return &fileStorage{
- baseDir: baseDir,
+// All Torrent data stored in this baseDir. The info names of each torrent are used as directories.
+func NewFile(baseDir string) ClientImplCloser {
+ return NewFileWithCompletion(baseDir, pieceCompletionForDir(baseDir))
+}
+
+type NewFileClientOpts struct {
+ // The base directory for all downloads.
+ ClientBaseDir string
+ FilePathMaker FilePathMaker
+ TorrentDirMaker TorrentDirFilePathMaker
+ PieceCompletion PieceCompletion
+}
+
+// NewFileOpts creates a new ClientImplCloser that stores files using the OS native filesystem.
+func NewFileOpts(opts NewFileClientOpts) ClientImplCloser {
+ if opts.TorrentDirMaker == nil {
+ opts.TorrentDirMaker = defaultPathMaker
+ }
+ if opts.FilePathMaker == nil {
+ opts.FilePathMaker = func(opts FilePathMakerOpts) string {
+ var parts []string
+ if opts.Info.Name != metainfo.NoName {
+ parts = append(parts, opts.Info.Name)
+ }
+ return filepath.Join(append(parts, opts.File.Path...)...)
+ }
+ }
+ if opts.PieceCompletion == nil {
+ opts.PieceCompletion = pieceCompletionForDir(opts.ClientBaseDir)
+ }
+ return fileClientImpl{opts}
+}
+
+func (me fileClientImpl) Close() error {
+ return me.opts.PieceCompletion.Close()
+}
+
+func (fs fileClientImpl) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (_ TorrentImpl, err error) {
+ dir := fs.opts.TorrentDirMaker(fs.opts.ClientBaseDir, info, infoHash)
+ upvertedFiles := info.UpvertedFiles()
+ files := make([]file, 0, len(upvertedFiles))
+ for i, fileInfo := range upvertedFiles {
+ filePath := filepath.Join(dir, fs.opts.FilePathMaker(FilePathMakerOpts{
+ Info: info,
+ File: &fileInfo,
+ }))
+ if !isSubFilepath(dir, filePath) {
+ err = fmt.Errorf("file %v: path %q is not sub path of %q", i, filePath, dir)
+ return
+ }
+ f := file{
+ path: filePath,
+ length: fileInfo.Length,
+ }
+ if f.length == 0 {
+ err = CreateNativeZeroLengthFile(f.path)
+ if err != nil {
+ err = fmt.Errorf("creating zero length file: %w", err)
+ return
+ }
+ }
+ files = append(files, f)
+ }
+ t := &fileTorrentImpl{
+ files,
+ segments.NewIndex(common.LengthIterFromUpvertedFiles(upvertedFiles)),
+ infoHash,
+ fs.opts.PieceCompletion,
}
+ return TorrentImpl{
+ Piece: t.Piece,
+ Close: t.Close,
+ }, nil
}
-func (fs *fileStorage) OpenTorrent(info *metainfo.InfoEx) (Torrent, error) {
- return fileTorrentStorage{fs}, nil
+type file struct {
+ // The safe, OS-local file path.
+ path string
+ length int64
}
-// File-based torrent storage, not yet bound to a Torrent.
-type fileTorrentStorage struct {
- *fileStorage
+type fileTorrentImpl struct {
+ files []file
+ segmentLocater segments.Index
+ infoHash metainfo.Hash
+ completion PieceCompletion
}
-func (fs *fileStorage) Piece(p metainfo.Piece) Piece {
+func (fts *fileTorrentImpl) Piece(p metainfo.Piece) PieceImpl {
// Create a view onto the file-based torrent storage.
- _io := &fileStorageTorrent{
- p.Info,
- fs.baseDir,
- }
+ _io := fileTorrentImplIO{fts}
// Return the appropriate segments of this.
- return &fileStoragePiece{
- fs,
+ return &filePieceImpl{
+ fts,
p,
missinggo.NewSectionWriter(_io, p.Offset(), p.Length()),
io.NewSectionReader(_io, p.Offset(), p.Length()),
}
}
-func (fs *fileStorage) Close() error {
+func (fs *fileTorrentImpl) Close() error {
return nil
}
-type fileStoragePiece struct {
- *fileStorage
- p metainfo.Piece
- io.WriterAt
- io.ReaderAt
-}
-
-func (fs *fileStoragePiece) GetIsComplete() bool {
- return fs.completed[fs.p.Hash()]
-}
-
-func (fs *fileStoragePiece) MarkComplete() error {
- if fs.completed == nil {
- fs.completed = make(map[[20]byte]bool)
+// A helper to create zero-length files which won't appear for file-orientated storage since no
+// writes will ever occur to them (no torrent data is associated with a zero-length file). The
+// caller should make sure the file name provided is safe/sanitized.
+func CreateNativeZeroLengthFile(name string) error {
+ os.MkdirAll(filepath.Dir(name), 0o777)
+ var f io.Closer
+ f, err := os.Create(name)
+ if err != nil {
+ return err
}
- fs.completed[fs.p.Hash()] = true
- return nil
+ return f.Close()
}
// Exposes file-based storage of a torrent, as one big ReadWriterAt.
-type fileStorageTorrent struct {
- info *metainfo.InfoEx
- baseDir string
+type fileTorrentImplIO struct {
+ fts *fileTorrentImpl
}
// Returns EOF on short or missing file.
-func (fst *fileStorageTorrent) readFileAt(fi metainfo.FileInfo, b []byte, off int64) (n int, err error) {
- f, err := os.Open(fst.fileInfoName(fi))
+func (fst *fileTorrentImplIO) readFileAt(file file, b []byte, off int64) (n int, err error) {
+ f, err := os.Open(file.path)
if os.IsNotExist(err) {
// File missing is treated the same as a short file.
err = io.EOF
}
defer f.Close()
// Limit the read to within the expected bounds of this file.
- if int64(len(b)) > fi.Length-off {
- b = b[:fi.Length-off]
+ if int64(len(b)) > file.length-off {
+ b = b[:file.length-off]
}
- for off < fi.Length && len(b) != 0 {
+ for off < file.length && len(b) != 0 {
n1, err1 := f.ReadAt(b, off)
b = b[n1:]
n += n1
}
// Only returns EOF at the end of the torrent. Premature EOF is ErrUnexpectedEOF.
-func (fst *fileStorageTorrent) ReadAt(b []byte, off int64) (n int, err error) {
- for _, fi := range fst.info.UpvertedFiles() {
- for off < fi.Length {
- n1, err1 := fst.readFileAt(fi, b, off)
- n += n1
- off += int64(n1)
- b = b[n1:]
- if len(b) == 0 {
- // Got what we need.
- return
- }
- if n1 != 0 {
- // Made progress.
- continue
- }
- err = err1
- if err == io.EOF {
- // Lies.
- err = io.ErrUnexpectedEOF
- }
- return
- }
- off -= fi.Length
+func (fst fileTorrentImplIO) ReadAt(b []byte, off int64) (n int, err error) {
+ fst.fts.segmentLocater.Locate(segments.Extent{off, int64(len(b))}, func(i int, e segments.Extent) bool {
+ n1, err1 := fst.readFileAt(fst.fts.files[i], b[:e.Length], e.Start)
+ n += n1
+ b = b[n1:]
+ err = err1
+ return err == nil // && int64(n1) == e.Length
+ })
+ if len(b) != 0 && err == nil {
+ err = io.EOF
}
- err = io.EOF
return
}
-func (fst *fileStorageTorrent) WriteAt(p []byte, off int64) (n int, err error) {
- for _, fi := range fst.info.UpvertedFiles() {
- if off >= fi.Length {
- off -= fi.Length
- continue
- }
- n1 := len(p)
- if int64(n1) > fi.Length-off {
- n1 = int(fi.Length - off)
- }
- name := fst.fileInfoName(fi)
- os.MkdirAll(filepath.Dir(name), 0770)
+func (fst fileTorrentImplIO) WriteAt(p []byte, off int64) (n int, err error) {
+ // log.Printf("write at %v: %v bytes", off, len(p))
+ fst.fts.segmentLocater.Locate(segments.Extent{off, int64(len(p))}, func(i int, e segments.Extent) bool {
+ name := fst.fts.files[i].path
+ os.MkdirAll(filepath.Dir(name), 0o777)
var f *os.File
- f, err = os.OpenFile(name, os.O_WRONLY|os.O_CREATE, 0660)
- if err != nil {
- return
- }
- n1, err = f.WriteAt(p[:n1], off)
- f.Close()
+ f, err = os.OpenFile(name, os.O_WRONLY|os.O_CREATE, 0o666)
if err != nil {
- return
+ return false
}
+ var n1 int
+ n1, err = f.WriteAt(p[:e.Length], e.Start)
+ // log.Printf("%v %v wrote %v: %v", i, e, n1, err)
+ closeErr := f.Close()
n += n1
- off = 0
p = p[n1:]
- if len(p) == 0 {
- break
+ if err == nil {
+ err = closeErr
}
- }
+ if err == nil && int64(n1) != e.Length {
+ err = io.ErrShortWrite
+ }
+ return err == nil
+ })
return
}
-
-func (fst *fileStorageTorrent) fileInfoName(fi metainfo.FileInfo) string {
- return filepath.Join(append([]string{fst.baseDir, fst.info.Name}, fi.Path...)...)
-}