16 fusefs "bazil.org/fuse/fs"
17 _ "github.com/anacrolix/envpprof"
18 "github.com/anacrolix/missinggo"
19 "github.com/stretchr/testify/assert"
20 "github.com/stretchr/testify/require"
21 netContext "golang.org/x/net/context"
23 "github.com/anacrolix/torrent"
24 "github.com/anacrolix/torrent/internal/testutil"
25 "github.com/anacrolix/torrent/metainfo"
26 "github.com/anacrolix/torrent/storage"
30 log.SetFlags(log.Flags() | log.Lshortfile)
33 func TestTCPAddrString(t *testing.T) {
34 l, err := net.Listen("tcp4", "localhost:0")
39 c, err := net.Dial("tcp", l.Addr().String())
44 ras := c.RemoteAddr().String()
46 IP: net.IPv4(127, 0, 0, 1),
47 Port: missinggo.AddrPort(l.Addr()),
55 type testLayout struct {
59 Metainfo *metainfo.MetaInfo
62 func (tl *testLayout) Destroy() error {
63 return os.RemoveAll(tl.BaseDir)
66 func newGreetingLayout() (tl testLayout, err error) {
67 tl.BaseDir, err = ioutil.TempDir("", "torrentfs")
71 tl.Completed = filepath.Join(tl.BaseDir, "completed")
72 os.Mkdir(tl.Completed, 0777)
73 tl.MountDir = filepath.Join(tl.BaseDir, "mnt")
74 os.Mkdir(tl.MountDir, 0777)
75 testutil.CreateDummyTorrentData(tl.Completed)
76 tl.Metainfo = testutil.GreetingMetaInfo()
80 // Unmount without first killing the FUSE connection while there are FUSE
81 // operations blocked inside the filesystem code.
82 func TestUnmountWedged(t *testing.T) {
83 layout, err := newGreetingLayout()
84 require.NoError(t, err)
86 err := layout.Destroy()
91 client, err := torrent.NewClient(&torrent.Config{
92 DataDir: filepath.Join(layout.BaseDir, "incomplete"),
93 DisableTrackers: true,
98 require.NoError(t, err)
100 tt, err := client.AddTorrent(layout.Metainfo)
101 require.NoError(t, err)
103 fuseConn, err := fuse.Mount(layout.MountDir)
106 case "cannot locate OSXFUSE",
107 "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))
138 t.Fatal("expected error reading greeting")
142 // Wait until the read has blocked inside the filesystem code.
144 for fs.blockedReads != 1 && ctx.Err() == nil {
152 err = fuse.Unmount(layout.MountDir)
154 t.Logf("error unmounting: %s", err)
155 time.Sleep(time.Millisecond)
161 err = fuseConn.Close()
163 t.Fatalf("error closing fuse conn: %s", err)
167 func TestDownloadOnDemand(t *testing.T) {
168 layout, err := newGreetingLayout()
169 require.NoError(t, err)
170 defer layout.Destroy()
171 seeder, err := torrent.NewClient(&torrent.Config{
172 DataDir: layout.Completed,
173 DisableTrackers: true,
175 ListenAddr: "localhost:0",
178 require.NoError(t, err)
180 testutil.ExportStatusWriter(seeder, "s")
181 // Just to mix things up, the seeder starts with the data, but the leecher
182 // starts with the metainfo.
183 seederTorrent, err := seeder.AddMagnet(fmt.Sprintf("magnet:?xt=urn:btih:%s", layout.Metainfo.HashInfoBytes().HexString()))
184 require.NoError(t, err)
186 // Wait until we get the metainfo, then check for the data.
187 <-seederTorrent.GotInfo()
188 seederTorrent.VerifyData()
190 leecher, err := torrent.NewClient(&torrent.Config{
191 DisableTrackers: true,
193 ListenAddr: "localhost:0",
195 DefaultStorage: storage.NewMMap(filepath.Join(layout.BaseDir, "download")),
196 // This can be used to check if clients can connect to other clients
198 // PeerID: seeder.PeerID(),
200 require.NoError(t, err)
201 testutil.ExportStatusWriter(leecher, "l")
202 defer leecher.Close()
203 leecherTorrent, _ := leecher.AddTorrent(layout.Metainfo)
204 leecherTorrent.AddPeers([]torrent.Peer{
206 IP: missinggo.AddrIP(seeder.ListenAddr()),
207 Port: missinggo.AddrPort(seeder.ListenAddr()),
213 node, _ := root.(fusefs.NodeStringLookuper).Lookup(netContext.Background(), "greeting")
215 node.Attr(netContext.Background(), &attr)
217 resp := &fuse.ReadResponse{
218 Data: make([]byte, size),
220 h, err := node.(fusefs.NodeOpener).Open(nil, nil, nil)
221 require.NoError(t, err)
222 h.(fusefs.HandleReader).Read(netContext.Background(), &fuse.ReadRequest{
225 assert.EqualValues(t, testutil.GreetingFileContents, resp.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))