package net.pterodactylus.fcp.highlevel;
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.FileWriter;
import java.io.IOException;
import java.net.InetAddress;
+import java.net.URL;
import java.net.UnknownHostException;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
+import java.util.Map.Entry;
+import java.util.logging.Logger;
+import net.pterodactylus.fcp.AddPeer;
import net.pterodactylus.fcp.AllData;
import net.pterodactylus.fcp.ClientHello;
import net.pterodactylus.fcp.CloseConnectionDuplicateClientName;
import net.pterodactylus.fcp.FcpConnection;
import net.pterodactylus.fcp.FcpListener;
import net.pterodactylus.fcp.FcpMessage;
+import net.pterodactylus.fcp.FcpUtils;
import net.pterodactylus.fcp.FinishedCompression;
import net.pterodactylus.fcp.GenerateSSK;
import net.pterodactylus.fcp.GetFailed;
import net.pterodactylus.fcp.ListPeers;
import net.pterodactylus.fcp.NodeData;
import net.pterodactylus.fcp.NodeHello;
+import net.pterodactylus.fcp.NodeRef;
import net.pterodactylus.fcp.Peer;
import net.pterodactylus.fcp.PeerNote;
import net.pterodactylus.fcp.PeerRemoved;
import net.pterodactylus.fcp.SubscribedUSKUpdate;
import net.pterodactylus.fcp.TestDDAComplete;
import net.pterodactylus.fcp.TestDDAReply;
+import net.pterodactylus.fcp.TestDDARequest;
+import net.pterodactylus.fcp.TestDDAResponse;
import net.pterodactylus.fcp.URIGenerated;
import net.pterodactylus.fcp.UnknownNodeIdentifier;
import net.pterodactylus.fcp.UnknownPeerNoteType;
*/
public class HighLevelClient {
+ /** Logger. */
+ private static final Logger logger = Logger.getLogger(HighLevelClient.class.getName());
+
/** Object for internal synchronization. */
private final Object syncObject = new Object();
/** Mapping from request identifier to peer list callbacks. */
private Map<String, HighLevelCallback<PeerListResult>> peerListCallbacks = Collections.synchronizedMap(new HashMap<String, HighLevelCallback<PeerListResult>>());
+ /** Mapping from request identifier to peer callbacks. */
+ private Map<String, HighLevelCallback<PeerResult>> peerCallbacks = Collections.synchronizedMap(new HashMap<String, HighLevelCallback<PeerResult>>());
+
+ /** Mapping from directories to DDA callbacks. */
+ private Map<String, HighLevelCallback<DirectDiskAccessResult>> directDiskAccessCallbacks = Collections.synchronizedMap(new HashMap<String, HighLevelCallback<DirectDiskAccessResult>>());
+
/**
* Creates a new high-level client that connects to a node on
* <code>localhost</code>.
}
/**
+ * Adds the peer whose noderef is stored in the given file.
+ *
+ * @param nodeRefFile
+ * The name of the file the peer’s noderef is stored in
+ * @return A peer callback
+ * @throws IOException
+ * if an I/O error occurs communicating with the node
+ */
+ public HighLevelCallback<PeerResult> addPeer(String nodeRefFile) throws IOException {
+ String identifier = generateIdentifier("addPeer");
+ AddPeer addPeer = new AddPeer(nodeRefFile);
+ HighLevelCallback<PeerResult> peerCallback = new HighLevelCallback<PeerResult>(new PeerResult());
+ peerCallbacks.put(identifier, peerCallback);
+ fcpConnection.sendMessage(addPeer);
+ return peerCallback;
+ }
+
+ /**
+ * Adds the peer whose noderef is stored in the given file.
+ *
+ * @param nodeRefURL
+ * The URL where the peer’s noderef is stored
+ * @return A peer callback
+ * @throws IOException
+ * if an I/O error occurs communicating with the node
+ */
+ public HighLevelCallback<PeerResult> addPeer(URL nodeRefURL) throws IOException {
+ String identifier = generateIdentifier("addPeer");
+ AddPeer addPeer = new AddPeer(nodeRefURL);
+ HighLevelCallback<PeerResult> peerCallback = new HighLevelCallback<PeerResult>(new PeerResult());
+ peerCallbacks.put(identifier, peerCallback);
+ fcpConnection.sendMessage(addPeer);
+ return peerCallback;
+ }
+
+ /**
+ * Adds the peer whose noderef is stored in the given file.
+ *
+ * @param nodeRef
+ * The peer’s noderef
+ * @return A peer callback
+ * @throws IOException
+ * if an I/O error occurs communicating with the node
+ */
+ public HighLevelCallback<PeerResult> addPeer(NodeRef nodeRef) throws IOException {
+ String identifier = generateIdentifier("addPeer");
+ AddPeer addPeer = new AddPeer(nodeRef);
+ HighLevelCallback<PeerResult> peerCallback = new HighLevelCallback<PeerResult>(new PeerResult());
+ peerCallbacks.put(identifier, peerCallback);
+ fcpConnection.sendMessage(addPeer);
+ return peerCallback;
+ }
+
+ /**
+ * Checks whether direct disk access for the given directory is possible.
+ * You have to perform this check before you can upload or download anything
+ * from or the disk directly!
+ *
+ * @param directory
+ * The directory to check
+ * @param wantRead
+ * Whether you want to read the given directory
+ * @param wantWrite
+ * Whether you want to write to the given directory
+ * @return A direct disk access callback
+ * @throws IOException
+ */
+ public HighLevelCallback<DirectDiskAccessResult> checkDirectDiskAccess(String directory, boolean wantRead, boolean wantWrite) throws IOException {
+ TestDDARequest testDDARequest = new TestDDARequest(directory, wantRead, wantWrite);
+ HighLevelCallback<DirectDiskAccessResult> directDiskAccessCallback = new HighLevelCallback<DirectDiskAccessResult>(new DirectDiskAccessResult());
+ directDiskAccessCallbacks.put(directory, directDiskAccessCallback);
+ fcpConnection.sendMessage(testDDARequest);
+ return directDiskAccessCallback;
+ }
+
+ //
+ // PRIVATE METHODS
+ //
+
+ /**
* Generates an identifier for the given function.
*
* @param function
*/
private class HighLevelClientFcpListener implements FcpListener {
+ /** Mapping from directory to written file (for cleanup). */
+ private final Map<DirectDiskAccessResult, String> writtenFiles = new HashMap<DirectDiskAccessResult, String>();
+
/**
* Creates a new FCP listener for {@link HighLevelClient}.
*/
HighLevelClientFcpListener() {
}
+ //
+ // PRIVATE METHODS
+ //
+
+ /**
+ * Searches all callback collections for a callback with the given
+ * identifier and cancels it.
+ *
+ * @param identifier
+ * The identifier to search for, or <code>null</code> to
+ * cancel all pending requests
+ */
+ @SuppressWarnings("synthetic-access")
+ private void cancelIdentifier(String identifier) {
+ synchronized (syncObject) {
+ if (connectCallback != null) {
+ connectCallback.getIntermediaryResult().setFailed(true);
+ connectCallback.setDone();
+ connectCallback = null;
+ }
+ }
+ if (identifier == null) {
+ /* key generation callbacks */
+ for (Entry<String, HighLevelCallback<KeyGenerationResult>> keyGenerationEntry: keyGenerationCallbacks.entrySet()) {
+ keyGenerationEntry.getValue().getIntermediaryResult().setFailed(true);
+ keyGenerationEntry.getValue().setDone();
+ }
+ keyGenerationCallbacks.clear();
+ /* peer list callbacks. */
+ for (Entry<String, HighLevelCallback<PeerListResult>> peerListEntry: peerListCallbacks.entrySet()) {
+ peerListEntry.getValue().getIntermediaryResult().setFailed(true);
+ peerListEntry.getValue().setDone();
+ }
+ peerListCallbacks.clear();
+ /* peer callbacks. */
+ for (Entry<String, HighLevelCallback<PeerResult>> peerEntry: peerCallbacks.entrySet()) {
+ peerEntry.getValue().getIntermediaryResult().setFailed(true);
+ peerEntry.getValue().setDone();
+ }
+ peerCallbacks.clear();
+ /* direct disk access callbacks. */
+ for (Entry<String, HighLevelCallback<DirectDiskAccessResult>> directDiskAccessEntry: directDiskAccessCallbacks.entrySet()) {
+ directDiskAccessEntry.getValue().getIntermediaryResult().setFailed(true);
+ directDiskAccessEntry.getValue().setDone();
+ }
+ directDiskAccessCallbacks.clear();
+ } else {
+ HighLevelCallback<KeyGenerationResult> keyGenerationCallback = keyGenerationCallbacks.remove(identifier);
+ if (keyGenerationCallback != null) {
+ keyGenerationCallback.getIntermediaryResult().setFailed(true);
+ keyGenerationCallback.setDone();
+ return;
+ }
+ HighLevelCallback<PeerListResult> peerListCallback = peerListCallbacks.remove(identifier);
+ if (peerListCallback != null) {
+ peerListCallback.getIntermediaryResult().setFailed(true);
+ peerListCallback.setDone();
+ return;
+ }
+ HighLevelCallback<PeerResult> peerCallback = peerCallbacks.remove(identifier);
+ if (peerCallback != null) {
+ peerCallback.getIntermediaryResult().setFailed(true);
+ peerCallback.setDone();
+ return;
+ }
+ HighLevelCallback<DirectDiskAccessResult> directDiskAccessCallback = directDiskAccessCallbacks.remove(identifier);
+ if (directDiskAccessCallback != null) {
+ directDiskAccessCallback.getIntermediaryResult().setFailed(true);
+ directDiskAccessCallback.setDone();
+ return;
+ }
+ }
+ }
+
+ /**
+ * Reads the given file and returns the first line of the file.
+ *
+ * @param readFilename
+ * The name of the file to read
+ * @return The content of the file
+ */
+ private String readContent(String readFilename) {
+ FileReader fileReader = null;
+ BufferedReader bufferedFileReader = null;
+ try {
+ fileReader = new FileReader(readFilename);
+ bufferedFileReader = new BufferedReader(fileReader);
+ String content = bufferedFileReader.readLine();
+ return content;
+ } catch (IOException ioe1) {
+ /* swallow. */
+ } finally {
+ FcpUtils.close(bufferedFileReader);
+ FcpUtils.close(fileReader);
+ }
+ return null;
+ }
+
+ /**
+ * Writes the given content to the given file.
+ *
+ * @param directDiskAccessResult
+ * The DDA result
+ * @param writeFilename
+ * The name of the file to write to
+ * @param writeContent
+ * The content to write to the file
+ */
+ private void writeContent(DirectDiskAccessResult directDiskAccessResult, String writeFilename, String writeContent) {
+ if ((writeFilename == null) || (writeContent == null)) {
+ return;
+ }
+ writtenFiles.put(directDiskAccessResult, writeFilename);
+ FileWriter fileWriter = null;
+ try {
+ fileWriter = new FileWriter(writeFilename);
+ fileWriter.write(writeContent);
+ } catch (IOException ioe1) {
+ /* swallow. */
+ } finally {
+ FcpUtils.close(fileWriter);
+ }
+ }
+
+ /**
+ * Cleans up any files that written for the given result.
+ *
+ * @param directDiskAccessResult
+ * The direct disk access result
+ */
+ @SuppressWarnings("synthetic-access")
+ private void cleanFiles(DirectDiskAccessResult directDiskAccessResult) {
+ String writeFilename = writtenFiles.remove(directDiskAccessResult);
+ if (writeFilename != null) {
+ if (!new File(writeFilename).delete()) {
+ logger.warning("could not delete " + writeFilename);
+ }
+ }
+ }
+
+ //
+ // INTERFACE FcpListener
+ //
+
/**
* @see net.pterodactylus.fcp.FcpListener#connectionClosed(net.pterodactylus.fcp.FcpConnection)
*/
+ @SuppressWarnings("synthetic-access")
public void connectionClosed(FcpConnection fcpConnection) {
+ if (fcpConnection != HighLevelClient.this.fcpConnection) {
+ return;
+ }
+ cancelIdentifier(null);
}
/**
return;
}
String identifier = peer.getIdentifier();
+ if (identifier == null) {
+ return;
+ }
HighLevelCallback<PeerListResult> peerListCallback = peerListCallbacks.get(identifier);
- if (peerListCallback == null) {
+ if (peerListCallback != null) {
+ peerListCallback.getIntermediaryResult().addPeer(peer);
return;
}
- peerListCallback.getIntermediaryResult().addPeer(peer);
+ HighLevelCallback<PeerResult> peerResult = peerCallbacks.remove(identifier);
+ if (peerResult != null) {
+ peerResult.getIntermediaryResult().setPeer(peer);
+ peerResult.setDone();
+ }
}
/**
if (identifier == null) {
return;
}
- /* now check all callbacks. */
- synchronized (syncObject) {
- if (connectCallback != null) {
- connectCallback.getIntermediaryResult().setFailed(true);
- connectCallback.setDone();
- connectCallback = null;
- }
- }
- HighLevelCallback<KeyGenerationResult> keyGenerationCallback = keyGenerationCallbacks.remove(identifier);
- if (keyGenerationCallback != null) {
- keyGenerationCallback.getIntermediaryResult().setFailed(true);
- keyGenerationCallback.setDone();
- }
+ cancelIdentifier(identifier);
}
/**
* @see net.pterodactylus.fcp.FcpListener#receivedTestDDAComplete(net.pterodactylus.fcp.FcpConnection,
* net.pterodactylus.fcp.TestDDAComplete)
*/
+ @SuppressWarnings("synthetic-access")
public void receivedTestDDAComplete(FcpConnection fcpConnection, TestDDAComplete testDDAComplete) {
+ if (fcpConnection != HighLevelClient.this.fcpConnection) {
+ return;
+ }
+ String directory = testDDAComplete.getDirectory();
+ if (directory == null) {
+ return;
+ }
+ HighLevelCallback<DirectDiskAccessResult> directDiskAccessCallback = directDiskAccessCallbacks.remove(directory);
+ DirectDiskAccessResult directDiskAccessResult = directDiskAccessCallback.getIntermediaryResult();
+ cleanFiles(directDiskAccessResult);
+ directDiskAccessResult.setReadAllowed(testDDAComplete.isReadDirectoryAllowed());
+ directDiskAccessResult.setWriteAllowed(testDDAComplete.isWriteDirectoryAllowed());
+ directDiskAccessCallback.setDone();
}
/**
* @see net.pterodactylus.fcp.FcpListener#receivedTestDDAReply(net.pterodactylus.fcp.FcpConnection,
* net.pterodactylus.fcp.TestDDAReply)
*/
+ @SuppressWarnings("synthetic-access")
public void receivedTestDDAReply(FcpConnection fcpConnection, TestDDAReply testDDAReply) {
+ if (fcpConnection != HighLevelClient.this.fcpConnection) {
+ return;
+ }
+ String directory = testDDAReply.getDirectory();
+ if (directory == null) {
+ return;
+ }
+ DirectDiskAccessResult directDiskAccessResult = directDiskAccessCallbacks.get(directory).getIntermediaryResult();
+ String readFilename = testDDAReply.getReadFilename();
+ String readContent = readContent(readFilename);
+ String writeFilename = testDDAReply.getWriteFilename();
+ String writeContent = testDDAReply.getContentToWrite();
+ writeContent(directDiskAccessResult, writeFilename, writeContent);
+ TestDDAResponse testDDAResponse = new TestDDAResponse(directory, readContent);
+ try {
+ fcpConnection.sendMessage(testDDAResponse);
+ } catch (IOException e) {
+ /* swallow. I’m verry unhappy about this. */
+ }
}
/**