17 "github.com/anacrolix/torrent/data"
19 "golang.org/x/net/context"
21 "github.com/anacrolix/torrent"
22 "github.com/anacrolix/torrent/data/mmap"
23 "github.com/anacrolix/torrent/internal/testutil"
24 "github.com/anacrolix/torrent/util"
26 "github.com/anacrolix/libtorgo/metainfo"
29 fusefs "bazil.org/fuse/fs"
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: util.AddrPort(l.Addr()),
54 type testLayout struct {
58 Metainfo *metainfo.MetaInfo
61 func (me *testLayout) Destroy() error {
62 return os.RemoveAll(me.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, 0777)
72 tl.MountDir = filepath.Join(tl.BaseDir, "mnt")
73 os.Mkdir(tl.MountDir, 0777)
74 name := testutil.CreateDummyTorrentData(tl.Completed)
75 metaInfoBuf := &bytes.Buffer{}
76 testutil.CreateMetaInfo(name, metaInfoBuf)
77 tl.Metainfo, err = metainfo.Load(metaInfoBuf)
81 // Unmount without first killing the FUSE connection while there are FUSE
82 // operations blocked inside the filesystem code.
83 func TestUnmountWedged(t *testing.T) {
84 layout, err := newGreetingLayout()
89 err := layout.Destroy()
94 client, err := torrent.NewClient(&torrent.Config{
95 DataDir: filepath.Join(layout.BaseDir, "incomplete"),
96 DisableTrackers: true,
99 NoDefaultBlocklist: true,
102 client.AddTorrent(layout.Metainfo)
104 fuseConn, err := fuse.Mount(layout.MountDir)
106 if strings.Contains(err.Error(), "fuse") {
112 server := fusefs.Server{
114 Debug: func(msg interface{}) {
118 server.Serve(fuseConn)
121 if err := fuseConn.MountError; err != nil {
124 // Read the greeting file, though it will never be available. This should
125 // "wedge" FUSE, requiring the fs object to be forcibly destroyed. The
126 // read call will return with a FS error.
128 _, err := ioutil.ReadFile(filepath.Join(layout.MountDir, layout.Metainfo.Info.Name))
130 t.Fatal("expected error reading greeting")
134 // Wait until the read has blocked inside the filesystem code.
136 for fs.blockedReads != 1 {
144 err = fuse.Unmount(layout.MountDir)
146 t.Logf("error unmounting: %s", err)
147 time.Sleep(time.Millisecond)
153 err = fuseConn.Close()
155 t.Fatalf("error closing fuse conn: %s", err)
159 func TestDownloadOnDemand(t *testing.T) {
160 layout, err := newGreetingLayout()
164 defer layout.Destroy()
165 seeder, err := torrent.NewClient(&torrent.Config{
166 DataDir: layout.Completed,
167 DisableTrackers: true,
171 NoDefaultBlocklist: true,
172 // Ensure that the metainfo is obtained over the wire, since we added
173 // the torrent to the seeder by magnet.
174 DisableMetainfoCache: true,
177 t.Fatalf("error creating seeder client: %s", err)
179 seeder.SetIPBlockList(nil)
181 http.HandleFunc("/seeder", func(w http.ResponseWriter, req *http.Request) {
182 seeder.WriteStatus(w)
184 _, err = seeder.AddMagnet(fmt.Sprintf("magnet:?xt=urn:btih:%x", layout.Metainfo.Info.Hash))
188 leecher, err := torrent.NewClient(&torrent.Config{
189 DisableTrackers: true,
194 NoDefaultBlocklist: true,
196 TorrentDataOpener: func(info *metainfo.Info) data.Data {
197 ret, _ := mmap.TorrentData(info, filepath.Join(layout.BaseDir, "download"))
201 // This can be used to check if clients can connect to other clients
204 // PeerID: seeder.PeerID(),
206 leecher.SetIPBlockList(nil)
207 http.HandleFunc("/leecher", func(w http.ResponseWriter, req *http.Request) {
208 leecher.WriteStatus(w)
210 defer leecher.Close()
211 leecherTorrent, _ := leecher.AddTorrent(layout.Metainfo)
212 leecherTorrent.AddPeers([]torrent.Peer{func() torrent.Peer {
213 _, port, err := net.SplitHostPort(seeder.ListenAddr().String())
217 portInt64, err := strconv.ParseInt(port, 0, 0)
223 ret, _ := net.ResolveIPAddr("ip", "localhost")
226 Port: int(portInt64),
232 node, _ := root.(fusefs.NodeStringLookuper).Lookup(context.Background(), "greeting")
233 size := int(node.Attr().Size)
234 resp := &fuse.ReadResponse{
235 Data: make([]byte, size),
237 node.(fusefs.HandleReader).Read(context.Background(), &fuse.ReadRequest{
241 if string(content) != testutil.GreetingFileContents {