9 // Accesses torrent data via a client.
17 var _ io.ReadCloser = &Reader{}
19 // Don't wait for pieces to complete and be verified. Read calls return as
20 // soon as they can when the underlying chunks become available.
21 func (r *Reader) SetResponsive() {
25 // Configure the number of bytes ahead of a read that should also be
26 // prioritized in preparation for further reads.
27 func (r *Reader) SetReadahead(readahead int64) {
28 r.readahead = readahead
31 func (r *Reader) raisePriorities(off int64, n int) {
33 r.t.cl.addUrgentRequests(r.t.torrent, off, n)
35 if !r.responsive || r.readahead != 0 {
36 r.t.cl.readRaisePiecePriorities(r.t.torrent, off, int64(n)+r.readahead)
40 func (r *Reader) readable(off int64) (ret bool) {
41 // log.Println("readable", off)
43 // log.Println("readable", ret)
48 req, ok := r.t.offsetRequest(off)
53 return r.t.haveChunk(req)
55 return r.t.pieceComplete(int(req.Index))
58 // How many bytes are available to read. Max is the most we could require.
59 func (r *Reader) available(off, max int64) (ret int64) {
61 req, ok := r.t.offsetRequest(off)
65 if !r.t.haveChunk(req) {
68 len1 := int64(req.Length) - (off - r.t.requestOffset(req))
73 // Ensure that ret hasn't exceeded our original max.
80 func (r *Reader) waitReadable(off int64) {
84 func (r *Reader) ReadAt(b []byte, off int64) (n int, err error) {
87 n1, err = r.readAt(b, off)
91 if err != nil || len(b) == 0 || n1 == 0 {
97 func (r *Reader) Read(b []byte) (n int, err error) {
98 n, err = r.readAt(b, r.pos)
103 // Must only return EOF at the end of the torrent.
104 func (r *Reader) readAt(b []byte, pos int64) (n int, err error) {
106 // log.Println(pos, n, err)
108 maxLen := r.t.torrent.Info.TotalLength() - pos
113 if int64(len(b)) > maxLen {
118 r.raisePriorities(pos, len(b))
119 for !r.readable(pos) {
120 r.raisePriorities(pos, len(b))
123 avail := r.available(pos, int64(len(b)))
124 // log.Println("available", avail)
127 pi := int(pos / r.t.Info().PieceLength)
128 tp := &r.t.torrent.Pieces[pi]
129 ip := r.t.Info().Piece(pi)
130 po := pos % ip.Length()
131 if int64(len(b1)) > ip.Length()-po {
132 b1 = b1[:ip.Length()-po]
134 tp.pendingWritesMutex.Lock()
135 for tp.pendingWrites != 0 {
136 tp.noPendingWrites.Wait()
138 tp.pendingWritesMutex.Unlock()
139 n, err = dataReadAt(r.t.data, b1, pos)
146 err = errors.New("torrent closed")
150 if err == io.ErrUnexpectedEOF {
156 func (r *Reader) Close() error {
161 func (r *Reader) Seek(off int64, whence int) (ret int64, err error) {
168 r.pos = r.t.torrent.Info.TotalLength() + off
170 err = errors.New("bad whence")