moonlight-stream/moonlight-android
1
0
Fork 0
mirror of https://github.com/moonlight-stream/moonlight-android.git synced 2025-07-18 18:42:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Created the NvmDNS class to issue the mDNS query.

Browse Source
This commit is contained in:
Andrew Hennessy 2013-09-26 01:12:42 -04:00
parent adf6b1a01b
commit 0d9e63f700
1 changed files with 221 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

231
src/com/limelight/nvstream/NvMDNS.java
View File

@ -1,27 +1,236 @@
package com.limelight.nvstream; package com.limelight.nvstream;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.InetAddress; import java.net.InetAddress;
import java.net.MulticastSocket;
import java.net.UnknownHostException; import java.net.UnknownHostException;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
public class NvMDNS { import org.xbill.DNS.DClass;
import org.xbill.DNS.Header;
import org.xbill.DNS.Message;
import org.xbill.DNS.Name;
import org.xbill.DNS.Record;
import org.xbill.DNS.Section;
import org.xbill.DNS.Type;
/**
* NvmDNS implements a clone of the NVidia Shield mDNS service for use on Limelight
* @author yetanothername
*
*/
public class NvmDNS implements Runnable {
/**
* The mDNS query that the Shield issues to look for computers
*/
public static String NVSTREAM_MDNS_QUERY = "_nvstream._tcp.local."; public static String NVSTREAM_MDNS_QUERY = "_nvstream._tcp.local.";
public static final short MDNS_PORT = 5353; /**
public static final int REPLY_TIMEOUT = 3000; * The multicast group NVSTREAM_MDNS_QUERY gets sent to
public static final int REPLY_TRIES = 10; */
public static InetAddress MDNS_ADDRESS; public static String NVSTREAM_MDNS_MULTICAST_GROUP = "224.0.0.251";
/**
* The IP Address that NVSTREAM_MDNS_MULTICAST_GROUP points to, in Java's IP object format
*/
public static InetAddress NVSTREAM_MDNS_MULTICAST_ADDRESS;
/**
* The port that the NVSTREAM_MDNS_QUERY gets sent to at NVSTREAM_MULTICAST_GROUP
*/
public static final short NVSTREAM_MDNS_PORT = 5353;
/**
* A collection of responses that we have received from our query
*/
private LinkedList<NvmDNSResponse> nvstream_mdns_responses;
/**
* The send and receive socket for our queries
*/
private MulticastSocket nvstream_query_socket;
/**
* We need to convert the IP Address into an IP Object
*/
static { static {
try { try {
// 224.0.0.251 is the mDNS multicast address NVSTREAM_MDNS_MULTICAST_ADDRESS = InetAddress.getByName(NvmDNS.NVSTREAM_MDNS_MULTICAST_GROUP);
MDNS_ADDRESS = InetAddress.getByName("224.0.0.251");
} catch (UnknownHostException e) { } catch (UnknownHostException e) {
MDNS_ADDRESS = null; NVSTREAM_MDNS_MULTICAST_ADDRESS = null;
} }
} }
// TODO: Implement this shit /**
* This sets up the query sockets and the list, as well as sends out the query and listens for resposnes
* @throws IOException When shit breaks
*/
public NvmDNS() throws IOException {
this.nvstream_mdns_responses = new LinkedList<NvmDNSResponse>();
this.nvstream_query_socket = new MulticastSocket(NvmDNS.NVSTREAM_MDNS_PORT);
this.nvstream_query_socket.setLoopbackMode(false);
this.nvstream_query_socket.joinGroup(NvmDNS.NVSTREAM_MDNS_MULTICAST_ADDRESS);
this.sendNVQuery();
// Implement the socket reads in a different thread
//this.recieveNVResponse();
Thread t = new Thread(this);
t.start();
} }
/**
* Sends out the NVSTREAM_MDNS_QUERY to the NVSTREAM_MDNS_MULTICAST_ADDRESS
* @return True if our query was sent, false otherwise
*/
private boolean sendNVQuery() {
try {
Header queryHeader = new Header();
queryHeader.setFlag(org.xbill.DNS.Flags.RA);
queryHeader.setID(0);
Record question = Record.newRecord(new Name(NvmDNS.NVSTREAM_MDNS_QUERY), Type.PTR, DClass.IN);
Message query = new Message();
query.setHeader(queryHeader);
query.addRecord(question, Section.QUESTION);
byte[] wireQuery = query.toWire();
DatagramPacket packet = new DatagramPacket(wireQuery, wireQuery.length);
packet.setAddress(NvmDNS.NVSTREAM_MDNS_MULTICAST_ADDRESS);
packet.setPort(NvmDNS.NVSTREAM_MDNS_PORT);
this.nvstream_query_socket.send(packet);
return true;
} catch (IOException e) {
e.printStackTrace();
return false;
}
}
@Override
public void run() {
try {
this.recieveNVResponse();
} catch (IOException e) {
e.printStackTrace();
}
}
private void recieveNVResponse() throws IOException {
byte[] data = new byte[1500];
DatagramPacket packet = new DatagramPacket(data, data.length);
Message message = null;
while (true) {
this.nvstream_query_socket.receive(packet);
message = new Message(packet.getData());
Record[] responses = message.getSectionArray(Section.ADDITIONAL);
if (responses.length != 0 && message.getSectionArray(Section.ANSWER)[0].getName().toString().equals(NvmDNS.NVSTREAM_MDNS_QUERY)) {
System.out.println("\nGot a packet from " + packet.getAddress().getCanonicalHostName());
System.out.println("\tQuestion: " + message.getSectionArray(Section.ANSWER)[0].getName());
System.out.println("\tResponse: " + responses[0].getName());
String[] txtRecords = responses[0].rdataToString().split("\" \"");
// No, but really, there has to be a better way of doing this...
txtRecords[0] = txtRecords[0].substring(1);
txtRecords[txtRecords.length - 1] = txtRecords[txtRecords.length - 1].split("\"")[0];
int state = -1;
int numOfApps = -1;
String gpuType = "Intel(R) Extreme Graphics 2";
String mac = "DE:AD:BE:EF:CA:FE";
String uniqueID = "4";
for (int i = 0; i < txtRecords.length; i++) {
System.out.println("\t\t" + txtRecords[i]);
if (i == 0) {
state = Integer.parseInt(txtRecords[i].split("=")[1]);
} else if (i == 1) {
numOfApps = Integer.parseInt(txtRecords[i].split("=")[1]);
} else if (i == 2) {
gpuType = txtRecords[i].split("=")[1];
} else if (i == 3) {
mac = txtRecords[i].split("=")[1];
} else if (i == 4) {
uniqueID = txtRecords[i].split("=")[1];
}
}
this.nvstream_mdns_responses.add(new NvmDNSResponse(packet.getAddress(), state, numOfApps, gpuType, mac, uniqueID));
}
}
}
public List<NvmDNSResponse> getmDNSResponses() {
return Collections.unmodifiableList(this.nvstream_mdns_responses);
}
public class NvmDNSResponse {
private InetAddress ipAddress;
private int state;
private int numOfApps;
private String gpuType;
private String mac;
private String uniqueID;
public NvmDNSResponse(InetAddress ipAddress, int state, int numOfApps, String gpuType, String mac, String uniqueID) {
this.ipAddress = ipAddress;
this.state = state;
this.numOfApps = numOfApps;
this.gpuType = gpuType;
this.mac = mac;
this.uniqueID = uniqueID;
}
public InetAddress getIpAddress() {
return ipAddress;
}
public int getState() {
return state;
}
public int getNumOfApps() {
return numOfApps;
}
public String getGpuType() {
return gpuType;
}
public String getMac() {
return mac;
}
public String getUniqueID() {
return uniqueID;
}
public String toString() {
StringBuilder returnStringBuilder = new StringBuilder();
returnStringBuilder.append("`IP Address: ");
returnStringBuilder.append(this.ipAddress.toString());
returnStringBuilder.append(" State: ");
returnStringBuilder.append(this.state);
returnStringBuilder.append(" Apps: ");
returnStringBuilder.append(this.numOfApps);
returnStringBuilder.append(" GPU: \'");
returnStringBuilder.append(this.gpuType);
returnStringBuilder.append("\' MAC: \'");
returnStringBuilder.append(this.mac);
returnStringBuilder.append("\' UniqueID: \'");
returnStringBuilder.append(this.uniqueID);
returnStringBuilder.append("\'`\n");
return returnStringBuilder.toString();
}
}
}