root / GetchunkTask.java @ 1
History | View | Annotate | Download (3 KB)
| 1 | 1 | up20150476 | import java.io.File; |
|---|---|---|---|
| 2 | import java.io.FileInputStream; |
||
| 3 | import java.io.IOException; |
||
| 4 | import java.net.DatagramPacket; |
||
| 5 | import java.net.DatagramSocket; |
||
| 6 | import java.rmi.NotBoundException; |
||
| 7 | import java.rmi.registry.LocateRegistry; |
||
| 8 | import java.rmi.registry.Registry; |
||
| 9 | import java.util.Random; |
||
| 10 | |||
| 11 | /**
|
||
| 12 | * GetchunkTask
|
||
| 13 | */
|
||
| 14 | public class GetchunkTask implements Runnable { |
||
| 15 | private static final String CRLF = Integer.toHexString(0xD) + Integer.toHexString(0xA); |
||
| 16 | |||
| 17 | private Peer peer;
|
||
| 18 | private String fileId; |
||
| 19 | private int chunkNo; |
||
| 20 | private MCListener mc;
|
||
| 21 | |||
| 22 | private String version; |
||
| 23 | |||
| 24 | private String senderId; |
||
| 25 | |||
| 26 | // GETCHUNK <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF>
|
||
| 27 | GetchunkTask(Peer peer, String[] args, MCListener mc) { |
||
| 28 | this.peer = peer;
|
||
| 29 | this.fileId = args[3]; |
||
| 30 | this.version = args[1]; |
||
| 31 | this.senderId = args[2]; |
||
| 32 | this.chunkNo = Integer.parseInt(args[4]); |
||
| 33 | this.mc = mc;
|
||
| 34 | |||
| 35 | this.mc.print("starting GETCHUNK protocol: [fileId] " + fileId + "[chunkNo] " + chunkNo); |
||
| 36 | } |
||
| 37 | |||
| 38 | @Override
|
||
| 39 | public void run() { |
||
| 40 | File file = new File("Peer-" + this.peer.getSenderId() + File.separator + "Chunks" + File.separator + fileId |
||
| 41 | + File.separator + chunkNo);
|
||
| 42 | try {
|
||
| 43 | FileInputStream stream = new FileInputStream(file); |
||
| 44 | byte[] chunk = new byte[(int) file.length()]; |
||
| 45 | stream.read(chunk); |
||
| 46 | stream.close(); |
||
| 47 | Random random = new Random(); |
||
| 48 | Thread.sleep(random.nextInt(401)); // [0-400] ms |
||
| 49 | |||
| 50 | if (this.peer.chunksMap.get(fileId).getChunksMap().get(chunkNo).getRecentlyRequested()) { |
||
| 51 | this.mc.print("Chunk was sent by another peer already"); |
||
| 52 | return;
|
||
| 53 | } |
||
| 54 | |||
| 55 | //Protocol enhancement -
|
||
| 56 | if (this.version.equals("2.0")) { |
||
| 57 | Registry registry = LocateRegistry.getRegistry(); |
||
| 58 | TestAppRemote stub = (TestAppRemote) registry.lookup(this.senderId);
|
||
| 59 | stub.chunk(this.fileId, chunk);
|
||
| 60 | } else {
|
||
| 61 | // CHUNK <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF><Body>
|
||
| 62 | String message = "CHUNK " + this.version + " " + this.peer.getSenderId() + " " + fileId + " " |
||
| 63 | + chunkNo + " " + CRLF + CRLF;
|
||
| 64 | |||
| 65 | byte[] packetData = new byte[message.getBytes().length + chunk.length]; |
||
| 66 | System.arraycopy(message.getBytes(), 0, packetData, 0, message.getBytes().length); |
||
| 67 | System.arraycopy(chunk, 0, packetData, message.getBytes().length, chunk.length); |
||
| 68 | |||
| 69 | DatagramSocket socket = new DatagramSocket(); |
||
| 70 | DatagramPacket packet = new DatagramPacket(packetData, packetData.length, this.peer.getMDRaddress(), |
||
| 71 | this.peer.getMDRport());
|
||
| 72 | socket.send(packet); |
||
| 73 | socket.close(); |
||
| 74 | } |
||
| 75 | |||
| 76 | } catch (IOException | InterruptedException | NotBoundException e) { |
||
| 77 | e.printStackTrace(); |
||
| 78 | } |
||
| 79 | this.mc.print("GETCHUNK protocol finished"); |
||
| 80 | } |
||
| 81 | } |