15 _ "github.com/anacrolix/envpprof"
16 "github.com/anacrolix/fuse"
17 fusefs "github.com/anacrolix/fuse/fs"
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(t *testing.T) (tl testLayout, err error) {
66 tl.BaseDir = t.TempDir()
67 tl.Completed = filepath.Join(tl.BaseDir, "completed")
68 os.Mkdir(tl.Completed, 0o777)
69 tl.MountDir = filepath.Join(tl.BaseDir, "mnt")
70 os.Mkdir(tl.MountDir, 0o777)
71 testutil.CreateDummyTorrentData(tl.Completed)
72 tl.Metainfo = testutil.GreetingMetaInfo()
76 // Unmount without first killing the FUSE connection while there are FUSE
77 // operations blocked inside the filesystem code.
78 func TestUnmountWedged(t *testing.T) {
79 layout, err := newGreetingLayout(t)
80 require.NoError(t, err)
82 err := layout.Destroy()
87 cfg := torrent.NewDefaultClientConfig()
88 cfg.DataDir = filepath.Join(layout.BaseDir, "incomplete")
89 cfg.DisableTrackers = true
93 client, err := torrent.NewClient(cfg)
94 require.NoError(t, err)
96 tt, err := client.AddTorrent(layout.Metainfo)
97 require.NoError(t, err)
99 fuseConn, err := fuse.Mount(layout.MountDir)
102 case "cannot locate OSXFUSE":
104 case "fusermount: exit status 1":
110 server := fusefs.New(fuseConn, &fusefs.Config{
111 Debug: func(msg interface{}) {
118 if err := fuseConn.MountError; err != nil {
119 t.Fatalf("mount error: %s", err)
121 ctx, cancel := context.WithCancel(context.Background())
122 // Read the greeting file, though it will never be available. This should
123 // "wedge" FUSE, requiring the fs object to be forcibly destroyed. The
124 // read call will return with a FS error.
133 _, err := ioutil.ReadFile(filepath.Join(layout.MountDir, tt.Info().Name))
134 require.Error(t, err)
137 // Wait until the read has blocked inside the filesystem code.
139 for fs.blockedReads != 1 && ctx.Err() == nil {
147 err = fuse.Unmount(layout.MountDir)
149 t.Logf("error unmounting: %s", err)
150 time.Sleep(time.Millisecond)
156 err = fuseConn.Close()
157 assert.NoError(t, err)
160 func TestDownloadOnDemand(t *testing.T) {
161 layout, err := newGreetingLayout(t)
162 require.NoError(t, err)
163 defer layout.Destroy()
164 cfg := torrent.NewDefaultClientConfig()
165 cfg.DataDir = layout.Completed
166 cfg.DisableTrackers = true
170 cfg.ListenHost = torrent.LoopbackListenHost
171 seeder, err := torrent.NewClient(cfg)
172 require.NoError(t, err)
174 defer testutil.ExportStatusWriter(seeder, "s", t)()
175 // Just to mix things up, the seeder starts with the data, but the leecher
176 // starts with the metainfo.
177 seederTorrent, err := seeder.AddMagnet(fmt.Sprintf("magnet:?xt=urn:btih:%s", layout.Metainfo.HashInfoBytes().HexString()))
178 require.NoError(t, err)
180 // Wait until we get the metainfo, then check for the data.
181 <-seederTorrent.GotInfo()
182 seederTorrent.VerifyData()
184 cfg = torrent.NewDefaultClientConfig()
185 cfg.DisableTrackers = true
187 cfg.DisableTCP = true
188 cfg.DefaultStorage = storage.NewMMap(filepath.Join(layout.BaseDir, "download"))
189 cfg.ListenHost = torrent.LoopbackListenHost
191 leecher, err := torrent.NewClient(cfg)
192 require.NoError(t, err)
193 testutil.ExportStatusWriter(leecher, "l", t)()
194 defer leecher.Close()
195 leecherTorrent, err := leecher.AddTorrent(layout.Metainfo)
196 require.NoError(t, err)
197 leecherTorrent.AddClientPeer(seeder)
201 node, _ := root.(fusefs.NodeStringLookuper).Lookup(context.Background(), "greeting")
203 node.Attr(context.Background(), &attr)
205 data := make([]byte, size)
206 h, err := node.(fusefs.NodeOpener).Open(context.TODO(), nil, nil)
207 require.NoError(t, err)
209 // torrent.Reader.Read no longer tries to fill the entire read buffer, so this is a ReadFull for
213 resp := fuse.ReadResponse{Data: data[n:]}
214 err := h.(fusefs.HandleReader).Read(context.Background(), &fuse.ReadRequest{
218 assert.NoError(t, err)
222 assert.EqualValues(t, testutil.GreetingFileContents, data)
225 func TestIsSubPath(t *testing.T) {
226 for _, case_ := range []struct {
233 {"a/b", "a/bc", false},
234 {"a/b", "a/b", false},
235 {"a/b", "a/b/c", true},
236 {"a/b", "a//b", false},
238 assert.Equal(t, case_.is, isSubPath(case_.parent, case_.child))