BitTorrent-Simulator/Message.java at master · bharathaiys/BitTorrent-Simulator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import java.io.Serializable;

public class Message implements Serializable {
    protected int messageLength;
    protected byte messageType;
    protected int pieceIndex;
    protected boolean[] bitField;
    protected byte[] piece;
    public TCPConnectionInfo messageOrigin;

    Message(byte messageType, int messageLength) {
        this.messageType = messageType;
        this.messageLength = messageLength;
    }

    /***
     * Constructor for "request" and "have" messages
     */
    Message(byte messageType, int messageLength, int pieceIndex) {
        this.messageType = messageType;
        this.messageLength = messageLength;
        this.pieceIndex = pieceIndex;
    }

    /***
     * Constructor for "bitfield" message
     */
    Message(byte messageType, int messageLength, boolean[] bitField) {
        this.messageType = messageType;
        this.messageLength = messageLength;
        this.bitField = bitField;
    }

    /**
     * Constructor for "piece" message.
     */
    Message(byte messageType, int messageLength, int pieceIndex, byte[] piece) {
        this.messageType = messageType;
        this.messageLength = messageLength;
        this.pieceIndex = pieceIndex;
        this.piece = piece;
    }

    /***
     * Default constructor for custom messages. Message type '100' represents custom message'
     * This is used to notify the peerProcess that a new Handshake message is received.
     * */
    Message() {
        this.messageType = 100;
    }

}