storage/mmap.go

   1 //go:build !wasm
   2 // +build !wasm
   3
   4 package storage
   5
   6 import (
   7         "errors"
   8         "fmt"
   9         "io"
  10         "os"
  11         "path/filepath"
  12
  13         "github.com/anacrolix/missinggo/v2"
  14         "github.com/edsrzf/mmap-go"
  15
  16         "github.com/anacrolix/torrent/metainfo"
  17         "github.com/anacrolix/torrent/mmap_span"
  18 )
  19
  20 type mmapClientImpl struct {
  21         baseDir string
  22         pc      PieceCompletion
  23 }
  24
  25 // TODO: Support all the same native filepath configuration that NewFileOpts provides.
  26 func NewMMap(baseDir string) ClientImplCloser {
  27         return NewMMapWithCompletion(baseDir, pieceCompletionForDir(baseDir))
  28 }
  29
  30 func NewMMapWithCompletion(baseDir string, completion PieceCompletion) *mmapClientImpl {
  31         return &mmapClientImpl{
  32                 baseDir: baseDir,
  33                 pc:      completion,
  34         }
  35 }
  36
  37 func (s *mmapClientImpl) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (_ TorrentImpl, err error) {
  38         span, err := mMapTorrent(info, s.baseDir)
  39         t := &mmapTorrentStorage{
  40                 infoHash: infoHash,
  41                 span:     span,
  42                 pc:       s.pc,
  43         }
  44         return TorrentImpl{Piece: t.Piece, Close: t.Close}, err
  45 }
  46
  47 func (s *mmapClientImpl) Close() error {
  48         return s.pc.Close()
  49 }
  50
  51 type mmapTorrentStorage struct {
  52         infoHash metainfo.Hash
  53         span     *mmap_span.MMapSpan
  54         pc       PieceCompletionGetSetter
  55 }
  56
  57 func (ts *mmapTorrentStorage) Piece(p metainfo.Piece) PieceImpl {
  58         return mmapStoragePiece{
  59                 pc:       ts.pc,
  60                 p:        p,
  61                 ih:       ts.infoHash,
  62                 ReaderAt: io.NewSectionReader(ts.span, p.Offset(), p.Length()),
  63                 WriterAt: missinggo.NewSectionWriter(ts.span, p.Offset(), p.Length()),
  64         }
  65 }
  66
  67 func (ts *mmapTorrentStorage) Close() error {
  68         errs := ts.span.Close()
  69         if len(errs) > 0 {
  70                 return errs[0]
  71         }
  72         return nil
  73 }
  74
  75 type mmapStoragePiece struct {
  76         pc PieceCompletionGetSetter
  77         p  metainfo.Piece
  78         ih metainfo.Hash
  79         io.ReaderAt
  80         io.WriterAt
  81 }
  82
  83 func (me mmapStoragePiece) pieceKey() metainfo.PieceKey {
  84         return metainfo.PieceKey{me.ih, me.p.Index()}
  85 }
  86
  87 func (sp mmapStoragePiece) Completion() Completion {
  88         c, err := sp.pc.Get(sp.pieceKey())
  89         if err != nil {
  90                 panic(err)
  91         }
  92         return c
  93 }
  94
  95 func (sp mmapStoragePiece) MarkComplete() error {
  96         sp.pc.Set(sp.pieceKey(), true)
  97         return nil
  98 }
  99
 100 func (sp mmapStoragePiece) MarkNotComplete() error {
 101         sp.pc.Set(sp.pieceKey(), false)
 102         return nil
 103 }
 104
 105 func mMapTorrent(md *metainfo.Info, location string) (mms *mmap_span.MMapSpan, err error) {
 106         mms = &mmap_span.MMapSpan{}
 107         defer func() {
 108                 if err != nil {
 109                         mms.Close()
 110                 }
 111         }()
 112         for _, miFile := range md.UpvertedFiles() {
 113                 var safeName string
 114                 safeName, err = ToSafeFilePath(append([]string{md.Name}, miFile.Path...)...)
 115                 if err != nil {
 116                         return
 117                 }
 118                 fileName := filepath.Join(location, safeName)
 119                 var mm mmap.MMap
 120                 mm, err = mmapFile(fileName, miFile.Length)
 121                 if err != nil {
 122                         err = fmt.Errorf("file %q: %s", miFile.DisplayPath(md), err)
 123                         return
 124                 }
 125                 if mm != nil {
 126                         mms.Append(mm)
 127                 }
 128         }
 129         mms.InitIndex()
 130         return
 131 }
 132
 133 func mmapFile(name string, size int64) (ret mmap.MMap, err error) {
 134         dir := filepath.Dir(name)
 135         err = os.MkdirAll(dir, 0750)
 136         if err != nil {
 137                 err = fmt.Errorf("making directory %q: %s", dir, err)
 138                 return
 139         }
 140         var file *os.File
 141         file, err = os.OpenFile(name, os.O_CREATE|os.O_RDWR, 0666)
 142         if err != nil {
 143                 return
 144         }
 145         defer file.Close()
 146         var fi os.FileInfo
 147         fi, err = file.Stat()
 148         if err != nil {
 149                 return
 150         }
 151         if fi.Size() < size {
 152                 // I think this is necessary on HFS+. Maybe Linux will SIGBUS too if
 153                 // you overmap a file but I'm not sure.
 154                 err = file.Truncate(size)
 155                 if err != nil {
 156                         return
 157                 }
 158         }
 159         if size == 0 {
 160                 // Can't mmap() regions with length 0.
 161                 return
 162         }
 163         intLen := int(size)
 164         if int64(intLen) != size {
 165                 err = errors.New("size too large for system")
 166                 return
 167         }
 168         ret, err = mmap.MapRegion(file, intLen, mmap.RDWR, 0, 0)
 169         if err != nil {
 170                 err = fmt.Errorf("error mapping region: %s", err)
 171                 return
 172         }
 173         if int64(len(ret)) != size {
 174                 panic(len(ret))
 175         }
 176         return
 177 }