15 "github.com/anacrolix/fuse"
16 fusefs "github.com/anacrolix/fuse/fs"
17 _ "github.com/anacrolix/envpprof"
18 "github.com/anacrolix/missinggo/v2"
19 "github.com/stretchr/testify/assert"
20 "github.com/stretchr/testify/require"
22 "github.com/anacrolix/torrent"
23 "github.com/anacrolix/torrent/internal/testutil"
24 "github.com/anacrolix/torrent/metainfo"
25 "github.com/anacrolix/torrent/storage"
29 log.SetFlags(log.Flags() | log.Lshortfile)
32 func TestTCPAddrString(t *testing.T) {
33 l, err := net.Listen("tcp4", "localhost:0")
38 c, err := net.Dial("tcp", l.Addr().String())
43 ras := c.RemoteAddr().String()
45 IP: net.IPv4(127, 0, 0, 1),
46 Port: missinggo.AddrPort(l.Addr()),
54 type testLayout struct {
58 Metainfo *metainfo.MetaInfo
61 func (tl *testLayout) Destroy() error {
62 return os.RemoveAll(tl.BaseDir)
65 func newGreetingLayout() (tl testLayout, err error) {
66 tl.BaseDir, err = ioutil.TempDir("", "torrentfs")
70 tl.Completed = filepath.Join(tl.BaseDir, "completed")
71 os.Mkdir(tl.Completed, 0o777)
72 tl.MountDir = filepath.Join(tl.BaseDir, "mnt")
73 os.Mkdir(tl.MountDir, 0o777)
74 testutil.CreateDummyTorrentData(tl.Completed)
75 tl.Metainfo = testutil.GreetingMetaInfo()
79 // Unmount without first killing the FUSE connection while there are FUSE
80 // operations blocked inside the filesystem code.
81 func TestUnmountWedged(t *testing.T) {
82 layout, err := newGreetingLayout()
83 require.NoError(t, err)
85 err := layout.Destroy()
90 cfg := torrent.NewDefaultClientConfig()
91 cfg.DataDir = filepath.Join(layout.BaseDir, "incomplete")
92 cfg.DisableTrackers = true
96 client, err := torrent.NewClient(cfg)
97 require.NoError(t, err)
99 tt, err := client.AddTorrent(layout.Metainfo)
100 require.NoError(t, err)
102 fuseConn, err := fuse.Mount(layout.MountDir)
105 case "cannot locate OSXFUSE":
107 case "fusermount: exit status 1":
113 server := fusefs.New(fuseConn, &fusefs.Config{
114 Debug: func(msg interface{}) {
121 if err := fuseConn.MountError; err != nil {
122 t.Fatalf("mount error: %s", err)
124 ctx, cancel := context.WithCancel(context.Background())
125 // Read the greeting file, though it will never be available. This should
126 // "wedge" FUSE, requiring the fs object to be forcibly destroyed. The
127 // read call will return with a FS error.
136 _, err := ioutil.ReadFile(filepath.Join(layout.MountDir, tt.Info().Name))
137 require.Error(t, err)
140 // Wait until the read has blocked inside the filesystem code.
142 for fs.blockedReads != 1 && ctx.Err() == nil {
150 err = fuse.Unmount(layout.MountDir)
152 t.Logf("error unmounting: %s", err)
153 time.Sleep(time.Millisecond)
159 err = fuseConn.Close()
160 assert.NoError(t, err)
163 func TestDownloadOnDemand(t *testing.T) {
164 layout, err := newGreetingLayout()
165 require.NoError(t, err)
166 defer layout.Destroy()
167 cfg := torrent.NewDefaultClientConfig()
168 cfg.DataDir = layout.Completed
169 cfg.DisableTrackers = true
173 cfg.ListenHost = torrent.LoopbackListenHost
174 seeder, err := torrent.NewClient(cfg)
175 require.NoError(t, err)
177 defer testutil.ExportStatusWriter(seeder, "s", t)()
178 // Just to mix things up, the seeder starts with the data, but the leecher
179 // starts with the metainfo.
180 seederTorrent, err := seeder.AddMagnet(fmt.Sprintf("magnet:?xt=urn:btih:%s", layout.Metainfo.HashInfoBytes().HexString()))
181 require.NoError(t, err)
183 // Wait until we get the metainfo, then check for the data.
184 <-seederTorrent.GotInfo()
185 seederTorrent.VerifyData()
187 cfg = torrent.NewDefaultClientConfig()
188 cfg.DisableTrackers = true
190 cfg.DisableTCP = true
191 cfg.DefaultStorage = storage.NewMMap(filepath.Join(layout.BaseDir, "download"))
192 cfg.ListenHost = torrent.LoopbackListenHost
194 leecher, err := torrent.NewClient(cfg)
195 require.NoError(t, err)
196 testutil.ExportStatusWriter(leecher, "l", t)()
197 defer leecher.Close()
198 leecherTorrent, err := leecher.AddTorrent(layout.Metainfo)
199 require.NoError(t, err)
200 leecherTorrent.AddClientPeer(seeder)
204 node, _ := root.(fusefs.NodeStringLookuper).Lookup(context.Background(), "greeting")
206 node.Attr(context.Background(), &attr)
208 data := make([]byte, size)
209 h, err := node.(fusefs.NodeOpener).Open(context.TODO(), nil, nil)
210 require.NoError(t, err)
212 // torrent.Reader.Read no longer tries to fill the entire read buffer, so this is a ReadFull for
216 resp := fuse.ReadResponse{Data: data[n:]}
217 err := h.(fusefs.HandleReader).Read(context.Background(), &fuse.ReadRequest{
221 assert.NoError(t, err)
225 assert.EqualValues(t, testutil.GreetingFileContents, data)
228 func TestIsSubPath(t *testing.T) {
229 for _, case_ := range []struct {
236 {"a/b", "a/bc", false},
237 {"a/b", "a/b", false},
238 {"a/b", "a/b/c", true},
239 {"a/b", "a//b", false},
241 assert.Equal(t, case_.is, isSubPath(case_.parent, case_.child))