package torrent import ( "container/heap" "expvar" "fmt" "io" "log" "net" "sort" "sync" "time" "github.com/anacrolix/missinggo" "github.com/anacrolix/missinggo/bitmap" "github.com/anacrolix/missinggo/perf" "github.com/anacrolix/missinggo/pubsub" "github.com/bradfitz/iter" "github.com/anacrolix/torrent/bencode" "github.com/anacrolix/torrent/metainfo" pp "github.com/anacrolix/torrent/peer_protocol" "github.com/anacrolix/torrent/tracker" ) func (t *torrent) chunkIndexSpec(chunkIndex, piece int) chunkSpec { return chunkIndexSpec(chunkIndex, t.pieceLength(piece), t.chunkSize) } func (t *torrent) pieceNumPendingBytes(index int) (count pp.Integer) { if t.pieceComplete(index) { return } piece := &t.Pieces[index] count = t.pieceLength(index) if !piece.EverHashed { return } for i, dirty := range piece.DirtyChunks { if dirty { count -= t.chunkIndexSpec(i, index).Length } } return } type peersKey struct { IPBytes string Port int } // Is not aware of Client. Maintains state of torrent for with-in a Client. type torrent struct { stateMu sync.Mutex closing chan struct{} // Closed when no more network activity is desired. This includes // announcing, and communicating with peers. ceasingNetworking chan struct{} InfoHash InfoHash Pieces []piece // Values are the piece indices that changed. pieceStateChanges *pubsub.PubSub chunkSize pp.Integer // Total length of the torrent in bytes. Stored because it's not O(1) to // get this from the info dict. length int64 data Data // The info dict. Nil if we don't have it (yet). Info *metainfo.Info // Active peer connections, running message stream loops. Conns []*connection // Set of addrs to which we're attempting to connect. Connections are // half-open until all handshakes are completed. HalfOpen map[string]struct{} // Reserve of peers to connect to. A peer can be both here and in the // active connections if were told about the peer after connecting with // them. That encourages us to reconnect to peers that are well known. Peers map[peersKey]Peer wantPeers sync.Cond // BEP 12 Multitracker Metadata Extension. The tracker.Client instances // mirror their respective URLs from the announce-list metainfo key. Trackers [][]tracker.Client // Name used if the info name isn't available. displayName string // The bencoded bytes of the info dict. MetaData []byte // Each element corresponds to the 16KiB metadata pieces. If true, we have // received that piece. metadataHave []bool // Closed when .Info is set. gotMetainfo chan struct{} readers map[*Reader]struct{} pendingPieces *bitmap.Bitmap } var ( piecePrioritiesReused = expvar.NewInt("piecePrioritiesReused") piecePrioritiesNew = expvar.NewInt("piecePrioritiesNew") ) func (t *torrent) setDisplayName(dn string) { t.displayName = dn } func (t *torrent) pieceComplete(piece int) bool { // TODO: This is called when setting metadata, and before storage is // assigned, which doesn't seem right. return t.data != nil && t.data.PieceComplete(piece) } func (t *torrent) numConnsUnchoked() (num int) { for _, c := range t.Conns { if !c.PeerChoked { num++ } } return } // There's a connection to that address already. func (t *torrent) addrActive(addr string) bool { if _, ok := t.HalfOpen[addr]; ok { return true } for _, c := range t.Conns { if c.remoteAddr().String() == addr { return true } } return false } func (t *torrent) worstConns(cl *Client) (wcs *worstConns) { wcs = &worstConns{ c: make([]*connection, 0, len(t.Conns)), t: t, cl: cl, } for _, c := range t.Conns { select { case <-c.closing: default: wcs.c = append(wcs.c, c) } } return } func (t *torrent) ceaseNetworking() { t.stateMu.Lock() defer t.stateMu.Unlock() select { case <-t.ceasingNetworking: return default: } close(t.ceasingNetworking) for _, c := range t.Conns { c.Close() } } func (t *torrent) addPeer(p Peer, cl *Client) { cl.openNewConns(t) if len(t.Peers) >= torrentPeersHighWater { return } key := peersKey{string(p.IP), p.Port} if _, ok := t.Peers[key]; ok { return } t.Peers[key] = p peersAddedBySource.Add(string(p.Source), 1) cl.openNewConns(t) } func (t *torrent) invalidateMetadata() { t.MetaData = nil t.metadataHave = nil t.Info = nil } func (t *torrent) saveMetadataPiece(index int, data []byte) { if t.haveInfo() { return } if index >= len(t.metadataHave) { log.Printf("%s: ignoring metadata piece %d", t, index) return } copy(t.MetaData[(1<<14)*index:], data) t.metadataHave[index] = true } func (t *torrent) metadataPieceCount() int { return (len(t.MetaData) + (1 << 14) - 1) / (1 << 14) } func (t *torrent) haveMetadataPiece(piece int) bool { if t.haveInfo() { return (1<<14)*piece < len(t.MetaData) } else { return piece < len(t.metadataHave) && t.metadataHave[piece] } } func (t *torrent) metadataSizeKnown() bool { return t.MetaData != nil } func (t *torrent) metadataSize() int { return len(t.MetaData) } func infoPieceHashes(info *metainfo.Info) (ret []string) { for i := 0; i < len(info.Pieces); i += 20 { ret = append(ret, string(info.Pieces[i:i+20])) } return } // Called when metadata for a torrent becomes available. func (t *torrent) setMetadata(md *metainfo.Info, infoBytes []byte) (err error) { err = validateInfo(md) if err != nil { err = fmt.Errorf("bad info: %s", err) return } t.Info = md t.length = 0 for _, f := range t.Info.UpvertedFiles() { t.length += f.Length } t.MetaData = infoBytes t.metadataHave = nil hashes := infoPieceHashes(md) t.Pieces = make([]piece, len(hashes)) for i, hash := range hashes { piece := &t.Pieces[i] piece.noPendingWrites.L = &piece.pendingWritesMutex missinggo.CopyExact(piece.Hash[:], hash) } for _, conn := range t.Conns { if err := conn.setNumPieces(t.numPieces()); err != nil { log.Printf("closing connection: %s", err) conn.Close() } } return } func (t *torrent) setStorage(td Data) (err error) { if t.data != nil { t.data.Close() } t.data = td return } func (t *torrent) haveAllMetadataPieces() bool { if t.haveInfo() { return true } if t.metadataHave == nil { return false } for _, have := range t.metadataHave { if !have { return false } } return true } func (t *torrent) setMetadataSize(bytes int64, cl *Client) { if t.haveInfo() { // We already know the correct metadata size. return } if bytes <= 0 || bytes > 10000000 { // 10MB, pulled from my ass. log.Printf("received bad metadata size: %d", bytes) return } if t.MetaData != nil && len(t.MetaData) == int(bytes) { return } t.MetaData = make([]byte, bytes) t.metadataHave = make([]bool, (bytes+(1<<14)-1)/(1<<14)) for _, c := range t.Conns { cl.requestPendingMetadata(t, c) } } // The current working name for the torrent. Either the name in the info dict, // or a display name given such as by the dn value in a magnet link, or "". func (t *torrent) Name() string { if t.haveInfo() { return t.Info.Name } return t.displayName } func (t *torrent) pieceState(index int) (ret PieceState) { p := &t.Pieces[index] ret.Priority = t.piecePriority(index) if t.pieceComplete(index) { ret.Complete = true } if p.QueuedForHash || p.Hashing { ret.Checking = true } if !ret.Complete && t.piecePartiallyDownloaded(index) { ret.Partial = true } return } func (t *torrent) metadataPieceSize(piece int) int { return metadataPieceSize(len(t.MetaData), piece) } func (t *torrent) newMetadataExtensionMessage(c *connection, msgType int, piece int, data []byte) pp.Message { d := map[string]int{ "msg_type": msgType, "piece": piece, } if data != nil { d["total_size"] = len(t.MetaData) } p, err := bencode.Marshal(d) if err != nil { panic(err) } return pp.Message{ Type: pp.Extended, ExtendedID: byte(c.PeerExtensionIDs["ut_metadata"]), ExtendedPayload: append(p, data...), } } func (t *torrent) pieceStateRuns() (ret []PieceStateRun) { rle := missinggo.NewRunLengthEncoder(func(el interface{}, count uint64) { ret = append(ret, PieceStateRun{ PieceState: el.(PieceState), Length: int(count), }) }) for index := range t.Pieces { rle.Append(t.pieceState(index), 1) } rle.Flush() return } // Produces a small string representing a PieceStateRun. func pieceStateRunStatusChars(psr PieceStateRun) (ret string) { ret = fmt.Sprintf("%d", psr.Length) ret += func() string { switch psr.Priority { case PiecePriorityNext: return "N" case PiecePriorityNormal: return "." case PiecePriorityReadahead: return "R" case PiecePriorityNow: return "!" default: return "" } }() if psr.Checking { ret += "H" } if psr.Partial { ret += "P" } if psr.Complete { ret += "C" } return } func (t *torrent) writeStatus(w io.Writer, cl *Client) { fmt.Fprintf(w, "Infohash: %x\n", t.InfoHash) fmt.Fprintf(w, "Metadata length: %d\n", t.metadataSize()) if !t.haveInfo() { fmt.Fprintf(w, "Metadata have: ") for _, h := range t.metadataHave { fmt.Fprintf(w, "%c", func() rune { if h { return 'H' } else { return '.' } }()) } fmt.Fprintln(w) } fmt.Fprintf(w, "Piece length: %s\n", func() string { if t.haveInfo() { return fmt.Sprint(t.usualPieceSize()) } else { return "?" } }()) if t.haveInfo() { fmt.Fprintf(w, "Num Pieces: %d\n", t.numPieces()) fmt.Fprint(w, "Piece States:") for _, psr := range t.pieceStateRuns() { w.Write([]byte(" ")) w.Write([]byte(pieceStateRunStatusChars(psr))) } fmt.Fprintln(w) } fmt.Fprintf(w, "Reader Pieces:") t.forReaderOffsetPieces(func(begin, end int) (again bool) { fmt.Fprintf(w, " %d:%d", begin, end) return true }) fmt.Fprintln(w) fmt.Fprintf(w, "Trackers: ") for _, tier := range t.Trackers { for _, tr := range tier { fmt.Fprintf(w, "%q ", tr.String()) } } fmt.Fprintf(w, "\n") fmt.Fprintf(w, "Pending peers: %d\n", len(t.Peers)) fmt.Fprintf(w, "Half open: %d\n", len(t.HalfOpen)) fmt.Fprintf(w, "Active peers: %d\n", len(t.Conns)) sort.Sort(&worstConns{ c: t.Conns, t: t, cl: cl, }) for i, c := range t.Conns { fmt.Fprintf(w, "%2d. ", i+1) c.WriteStatus(w, t) } } func (t *torrent) String() string { s := t.Name() if s == "" { s = fmt.Sprintf("%x", t.InfoHash) } return s } func (t *torrent) haveInfo() bool { return t.Info != nil } // TODO: Include URIs that weren't converted to tracker clients. func (t *torrent) announceList() (al [][]string) { for _, tier := range t.Trackers { var l []string for _, tr := range tier { l = append(l, tr.URL()) } al = append(al, l) } return } // Returns a run-time generated MetaInfo that includes the info bytes and // announce-list as currently known to the client. func (t *torrent) MetaInfo() *metainfo.MetaInfo { if t.MetaData == nil { panic("info bytes not set") } return &metainfo.MetaInfo{ Info: metainfo.InfoEx{ Info: *t.Info, Bytes: t.MetaData, }, CreationDate: time.Now().Unix(), Comment: "dynamic metainfo from client", CreatedBy: "go.torrent", AnnounceList: t.announceList(), } } func (t *torrent) bytesLeft() (left int64) { if !t.haveInfo() { return -1 } for i := 0; i < t.numPieces(); i++ { left += int64(t.pieceNumPendingBytes(i)) } return } func (t *torrent) piecePartiallyDownloaded(index int) bool { pendingBytes := t.pieceNumPendingBytes(index) return pendingBytes != 0 && pendingBytes != t.pieceLength(index) } func numChunksForPiece(chunkSize int, pieceSize int) int { return (pieceSize + chunkSize - 1) / chunkSize } func (t *torrent) usualPieceSize() int { return int(t.Info.PieceLength) } func (t *torrent) lastPieceSize() int { return int(t.pieceLength(t.numPieces() - 1)) } func (t *torrent) numPieces() int { return t.Info.NumPieces() } func (t *torrent) numPiecesCompleted() (num int) { for i := range iter.N(t.Info.NumPieces()) { if t.pieceComplete(i) { num++ } } return } func (t *torrent) isClosed() bool { select { case <-t.closing: return true default: return false } } func (t *torrent) close() (err error) { if t.isClosed() { return } t.ceaseNetworking() close(t.closing) if c, ok := t.data.(io.Closer); ok { c.Close() } for _, conn := range t.Conns { conn.Close() } t.pieceStateChanges.Close() return } func (t *torrent) requestOffset(r request) int64 { return torrentRequestOffset(t.length, int64(t.usualPieceSize()), r) } // Return the request that would include the given offset into the torrent // data. Returns !ok if there is no such request. func (t *torrent) offsetRequest(off int64) (req request, ok bool) { return torrentOffsetRequest(t.length, t.Info.PieceLength, int64(t.chunkSize), off) } func (t *torrent) writeChunk(piece int, begin int64, data []byte) (err error) { tr := perf.NewTimer() n, err := t.data.WriteAt(data, int64(piece)*t.Info.PieceLength+begin) if err == nil && n != len(data) { err = io.ErrShortWrite } if err == nil { tr.Stop("write chunk") } return } func (t *torrent) bitfield() (bf []bool) { for i := range t.Pieces { bf = append(bf, t.havePiece(i)) } return } func (t *torrent) validOutgoingRequest(r request) bool { if r.Index >= pp.Integer(t.Info.NumPieces()) { return false } if r.Begin%t.chunkSize != 0 { return false } if r.Length > t.chunkSize { return false } pieceLength := t.pieceLength(int(r.Index)) if r.Begin+r.Length > pieceLength { return false } return r.Length == t.chunkSize || r.Begin+r.Length == pieceLength } func (t *torrent) pieceChunks(piece int) (css []chunkSpec) { css = make([]chunkSpec, 0, (t.pieceLength(piece)+t.chunkSize-1)/t.chunkSize) var cs chunkSpec for left := t.pieceLength(piece); left != 0; left -= cs.Length { cs.Length = left if cs.Length > t.chunkSize { cs.Length = t.chunkSize } css = append(css, cs) cs.Begin += cs.Length } return } func (t *torrent) pieceNumChunks(piece int) int { return int((t.pieceLength(piece) + t.chunkSize - 1) / t.chunkSize) } func (t *torrent) pendAllChunkSpecs(pieceIndex int) { t.Pieces[pieceIndex].DirtyChunks = nil } type Peer struct { Id [20]byte IP net.IP Port int Source peerSource // Peer is known to support encryption. SupportsEncryption bool } func (t *torrent) pieceLength(piece int) (len_ pp.Integer) { if piece < 0 || piece > t.Info.NumPieces() { return } if int(piece) == t.numPieces()-1 { len_ = pp.Integer(t.length % t.Info.PieceLength) } if len_ == 0 { len_ = pp.Integer(t.Info.PieceLength) } return } func (t *torrent) hashPiece(piece int) (ps pieceSum) { hash := pieceHash.New() p := &t.Pieces[piece] p.waitNoPendingWrites() pl := t.Info.Piece(int(piece)).Length() n, err := t.data.WriteSectionTo(hash, int64(piece)*t.Info.PieceLength, pl) if err != nil { if err != io.ErrUnexpectedEOF { log.Printf("error hashing piece with %T: %s", t.data, err) } return } if n != pl { panic(fmt.Sprintf("%T: %d != %d", t.data, n, pl)) } missinggo.CopyExact(ps[:], hash.Sum(nil)) return } func (t *torrent) haveAllPieces() bool { if !t.haveInfo() { return false } for i := range t.Pieces { if !t.pieceComplete(i) { return false } } return true } func (me *torrent) haveAnyPieces() bool { for i := range me.Pieces { if me.pieceComplete(i) { return true } } return false } func (t *torrent) havePiece(index int) bool { return t.haveInfo() && t.pieceComplete(index) } func (t *torrent) haveChunk(r request) (ret bool) { // defer func() { // log.Println("have chunk", r, ret) // }() if !t.haveInfo() { return false } if t.pieceComplete(int(r.Index)) { return true } p := &t.Pieces[r.Index] return !p.pendingChunk(r.chunkSpec, t.chunkSize) } func chunkIndex(cs chunkSpec, chunkSize pp.Integer) int { return int(cs.Begin / chunkSize) } // TODO: This should probably be called wantPiece. func (t *torrent) wantChunk(r request) bool { if !t.wantPiece(int(r.Index)) { return false } if t.Pieces[r.Index].pendingChunk(r.chunkSpec, t.chunkSize) { return true } // TODO: What about pieces that were wanted, but aren't now, and aren't // completed either? That used to be done here. return false } // TODO: This should be called wantPieceIndex. func (t *torrent) wantPiece(index int) bool { if !t.haveInfo() { return false } p := &t.Pieces[index] if p.QueuedForHash { return false } if p.Hashing { return false } if t.pieceComplete(index) { return false } if t.pendingPieces.Contains(index) { return true } return !t.forReaderOffsetPieces(func(begin, end int) bool { return index < begin || index >= end }) } func (t *torrent) forNeededPieces(f func(piece int) (more bool)) (all bool) { return t.forReaderOffsetPieces(func(begin, end int) (more bool) { for i := begin; begin < end; i++ { if !f(i) { return false } } return true }) } func (t *torrent) connHasWantedPieces(c *connection) bool { for it := t.pendingPieces.Iter(); it.Next(); { if c.PeerHasPiece(it.Value()) { return true } } return !t.forReaderOffsetPieces(func(begin, end int) (again bool) { for i := begin; i < end; i++ { if c.PeerHasPiece(i) { return false } } return true }) } func (t *torrent) extentPieces(off, _len int64) (pieces []int) { for i := off / int64(t.usualPieceSize()); i*int64(t.usualPieceSize()) < off+_len; i++ { pieces = append(pieces, int(i)) } return } func (t *torrent) worstBadConn(cl *Client) *connection { wcs := t.worstConns(cl) heap.Init(wcs) for wcs.Len() != 0 { c := heap.Pop(wcs).(*connection) if c.UnwantedChunksReceived >= 6 && c.UnwantedChunksReceived > c.UsefulChunksReceived { return c } if wcs.Len() >= (socketsPerTorrent+1)/2 { // Give connections 1 minute to prove themselves. if time.Since(c.completedHandshake) > time.Minute { return c } } } return nil } func (t *torrent) publishPieceChange(piece int) { cur := t.pieceState(piece) p := &t.Pieces[piece] if cur != p.PublicPieceState { t.pieceStateChanges.Publish(piece) } p.PublicPieceState = cur } func (t *torrent) pieceNumPendingChunks(piece int) int { if t.pieceComplete(piece) { return 0 } return t.pieceNumChunks(piece) - t.Pieces[piece].numDirtyChunks() } func (t *torrent) pieceAllDirty(piece int) bool { p := &t.Pieces[piece] if len(p.DirtyChunks) != t.pieceNumChunks(piece) { return false } for _, dirty := range p.DirtyChunks { if !dirty { return false } } return true } func (t *torrent) forUrgentPieces(f func(piece int) (again bool)) (all bool) { return t.forReaderOffsetPieces(func(begin, end int) (again bool) { if begin < end { if !f(begin) { return false } } return true }) } func (t *torrent) readersChanged(cl *Client) { // Accept new connections. cl.event.Broadcast() for _, c := range t.Conns { c.updateRequests() } cl.openNewConns(t) } func (t *torrent) byteRegionPieces(off, size int64) (begin, end int) { if off >= t.length { return } if off < 0 { size += off off = 0 } if size <= 0 { return } begin = int(off / t.Info.PieceLength) end = int((off + size + t.Info.PieceLength - 1) / t.Info.PieceLength) if end > t.Info.NumPieces() { end = t.Info.NumPieces() } return } // Returns true if all iterations complete without breaking. func (t *torrent) forReaderOffsetPieces(f func(begin, end int) (more bool)) (all bool) { for r := range t.readers { r.mu.Lock() pos, readahead := r.pos, r.readahead r.mu.Unlock() if readahead < 1 { readahead = 1 } begin, end := t.byteRegionPieces(pos, readahead) if begin >= end { continue } if !f(begin, end) { return false } } return true } func (t *torrent) piecePriority(piece int) (ret piecePriority) { ret = PiecePriorityNone if t.pieceComplete(piece) { return } if t.pendingPieces.Contains(piece) { ret = PiecePriorityNormal } raiseRet := func(prio piecePriority) { if prio > ret { ret = prio } } t.forReaderOffsetPieces(func(begin, end int) (again bool) { if piece == begin { raiseRet(PiecePriorityNow) } if begin <= piece && piece < end { raiseRet(PiecePriorityReadahead) } return true }) return } func (t *torrent) pendPiece(piece int, cl *Client) { if t.pendingPieces == nil { t.pendingPieces = bitmap.New() } if t.pendingPieces.Contains(piece) { return } if t.havePiece(piece) { return } t.pendingPieces.Add(piece) for _, c := range t.Conns { if !c.PeerHasPiece(piece) { continue } c.updateRequests() } cl.openNewConns(t) cl.pieceChanged(t, piece) } func (t *torrent) connRequestPiecePendingChunks(c *connection, piece int) (more bool) { if !c.PeerHasPiece(piece) { return true } for _, cs := range t.Pieces[piece].shuffledPendingChunkSpecs(t, piece) { req := request{pp.Integer(piece), cs} if !c.Request(req) { return false } } return true } func (t *torrent) pendRequest(req request) { ci := chunkIndex(req.chunkSpec, t.chunkSize) t.Pieces[req.Index].pendChunkIndex(ci) }