public static void main(String[] args) {
boolean sender = false;
File fi = null;
try {
if (args.length == 0) {
int ret =
JOptionPane.showConfirmDialog(
null,
"Are you the sender? (no = reciever)",
"Send/Recieve",
JOptionPane.YES_NO_OPTION);
JFileChooser chooser = new JFileChooser();
if (ret == JOptionPane.YES_OPTION) {
chooser.showOpenDialog(null);
sender = true;
} else {
chooser.showSaveDialog(null);
sender = false;
}
fi = chooser.getSelectedFile();
} else {
if (args[0].equalsIgnoreCase("-s")) {
sender = true;
} else if (args[0].equalsIgnoreCase("-r")) {
sender = false;
}
fi = new File(args[1]);
}
} catch (Exception e) {
e.printStackTrace();
exit();
}
if (sender && !fi.exists()) {
System.err.println("Cannot send, file doesn't exist");
exit();
}
try {
if (sender) {
ServerSocket ss = new ServerSocket(DEFAULT_PORT);
Socket sck = ss.accept();
byte[] hsh = SecureUtils.hash(fi);
System.out.println(SecureUtils.hexify(hsh));
sendFile(sck, fi, hsh);
sck.close();
} else {
Socket sck = new Socket("localhost", DEFAULT_PORT);
byte[] hsh = recvFile(sck, fi);
System.out.println(SecureUtils.hexify(hsh));
sck.close();
}
} catch (NoSuchAlgorithmException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* Transform an URI accessible from the server into a URI accessible from the client. The mapping
* expires with the session.
*
* @param propertyContext context to obtain session
* @param uri server URI to transform
* @param filename file name
* @param contentType type of the content referred to by the URI, or null if unknown
* @param lastModified last modification timestamp
* @return client URI
*/
public static String proxyURI(
PropertyContext propertyContext,
String uri,
String filename,
String contentType,
long lastModified) {
// Create a digest, so that for a given URI we always get the same key
final String digest = SecureUtils.digestString(uri, "MD5", "hex");
// Get session
final ExternalContext externalContext =
(ExternalContext) propertyContext.getAttribute(PipelineContext.EXTERNAL_CONTEXT);
final ExternalContext.Session session =
externalContext.getSession(
true); // NOTE: We force session creation here. Should we? What's the alternative?
if (session != null) {
// Store mapping into session
session
.getAttributesMap(ExternalContext.Session.APPLICATION_SCOPE)
.put(
DYNAMIC_RESOURCES_SESSION_KEY + digest,
new DynamicResource(uri, filename, contentType, -1, lastModified));
}
// Rewrite new URI to absolute path without the context
return DYNAMIC_RESOURCES_PATH + digest;
}
/**
* This digester is based on some existing public document (not sure which). There are some
* changes though. It is not clear anymore why we used that document as a base, as this is purely
* internal.
*
* <p>The bottom line is that the digest should change whenever the infoset of the source XML
* document changes.
*/
public static class DigestContentHandler implements XMLReceiver {
private static final int ELEMENT_CODE = Node.ELEMENT_NODE;
private static final int ATTRIBUTE_CODE = Node.ATTRIBUTE_NODE;
private static final int TEXT_CODE = Node.TEXT_NODE;
private static final int PROCESSING_INSTRUCTION_CODE = Node.PROCESSING_INSTRUCTION_NODE;
private static final int NAMESPACE_CODE = 0XAA01; // some code that is none of the above
private static final int COMMENT_CODE = 0XAA02; // some code that is none of the above
/**
* 4/6/2005 d : Previously we were using String.getBytes( "UnicodeBigUnmarked" ). ( Believe the
* code was copied from RFC 2803 ). This first tries to get a java.nio.Charset with the name if
* this fails it uses a sun.io.CharToByteConverter. Now in the case of "UnicodeBigUnmarked"
* there is no such Charset so a CharToByteConverter, utf-16be, is used. Unfortunately this
* negative lookup is expensive. ( Costing us a full second in the 50thread/512MB test. ) The
* solution, of course, is just to use get the appropriate Charset and hold on to it.
*/
private static final Charset utf16BECharset = Charset.forName("UTF-16BE");
/** Encoder has state and therefore cannot be shared across threads. */
private final CharsetEncoder charEncoder = utf16BECharset.newEncoder();
private java.nio.CharBuffer charBuff = java.nio.CharBuffer.allocate(64);
private java.nio.ByteBuffer byteBuff = java.nio.ByteBuffer.allocate(128);
private final MessageDigest digest = SecureUtils.defaultMessageDigest();
private void ensureCharBuffRemaining(final int size) {
if (charBuff.remaining() < size) {
final int cpcty = (charBuff.capacity() + size) * 2;
final java.nio.CharBuffer newChBuf = java.nio.CharBuffer.allocate(cpcty);
newChBuf.put(charBuff);
charBuff = newChBuf;
}
}
private void updateWithCharBuf() {
final int reqSize = (int) charEncoder.maxBytesPerChar() * charBuff.position();
if (byteBuff.capacity() < reqSize) {
byteBuff = java.nio.ByteBuffer.allocate(2 * reqSize);
}
// Make ready for read
charBuff.flip();
final CoderResult cr = charEncoder.encode(charBuff, byteBuff, true);
try {
if (cr.isError()) cr.throwException();
// Make ready for read
byteBuff.flip();
final byte[] byts = byteBuff.array();
final int len = byteBuff.remaining();
final int strt = byteBuff.arrayOffset();
digest.update(byts, strt, len);
} catch (final CharacterCodingException e) {
throw new OXFException(e);
} catch (java.nio.BufferOverflowException e) {
throw new OXFException(e);
} catch (java.nio.BufferUnderflowException e) {
throw new OXFException(e);
} finally {
// Make ready for write
charBuff.clear();
byteBuff.clear();
}
}
private void updateWith(final String s) {
addToCharBuff(s);
updateWithCharBuf();
}
private void updateWith(final char[] chArr, final int ofst, final int len) {
ensureCharBuffRemaining(len);
charBuff.put(chArr, ofst, len);
updateWithCharBuf();
}
private void addToCharBuff(final char c) {
ensureCharBuffRemaining(1);
charBuff.put(c);
}
private void addToCharBuff(final String s) {
final int size = s.length();
ensureCharBuffRemaining(size);
charBuff.put(s);
}
public byte[] getResult() {
return digest.digest();
}
public void setDocumentLocator(Locator locator) {}
public void startDocument() throws SAXException {
charBuff.clear();
byteBuff.clear();
charEncoder.reset();
}
public void endDocument() throws SAXException {}
public void startPrefixMapping(String prefix, String uri) throws SAXException {
digest.update((byte) ((NAMESPACE_CODE >> 24) & 0xff));
digest.update((byte) ((NAMESPACE_CODE >> 16) & 0xff));
digest.update((byte) ((NAMESPACE_CODE >> 8) & 0xff));
digest.update((byte) (NAMESPACE_CODE & 0xff));
updateWith(prefix);
digest.update((byte) 0);
digest.update((byte) 0);
updateWith(uri);
digest.update((byte) 0);
digest.update((byte) 0);
}
public void endPrefixMapping(String prefix) throws SAXException {}
public void startElement(String namespaceURI, String localName, String qName, Attributes atts)
throws SAXException {
digest.update((byte) ((ELEMENT_CODE >> 24) & 0xff));
digest.update((byte) ((ELEMENT_CODE >> 16) & 0xff));
digest.update((byte) ((ELEMENT_CODE >> 8) & 0xff));
digest.update((byte) (ELEMENT_CODE & 0xff));
addToCharBuff('{');
addToCharBuff(namespaceURI);
addToCharBuff('}');
addToCharBuff(localName);
updateWithCharBuf();
digest.update((byte) 0);
digest.update((byte) 0);
int attCount = atts.getLength();
digest.update((byte) ((attCount >> 24) & 0xff));
digest.update((byte) ((attCount >> 16) & 0xff));
digest.update((byte) ((attCount >> 8) & 0xff));
digest.update((byte) (attCount & 0xff));
for (int i = 0; i < attCount; i++) {
digest.update((byte) ((ATTRIBUTE_CODE >> 24) & 0xff));
digest.update((byte) ((ATTRIBUTE_CODE >> 16) & 0xff));
digest.update((byte) ((ATTRIBUTE_CODE >> 8) & 0xff));
digest.update((byte) (ATTRIBUTE_CODE & 0xff));
final String attURI = atts.getURI(i);
final String attNam = atts.getLocalName(i);
addToCharBuff('{');
addToCharBuff(attURI);
addToCharBuff('}');
addToCharBuff(attNam);
updateWithCharBuf();
digest.update((byte) 0);
digest.update((byte) 0);
final String val = atts.getValue(i);
updateWith(val);
}
}
public void endElement(String namespaceURI, String localName, String qName)
throws SAXException {}
public void characters(char ch[], int start, int length) throws SAXException {
digest.update((byte) ((TEXT_CODE >> 24) & 0xff));
digest.update((byte) ((TEXT_CODE >> 16) & 0xff));
digest.update((byte) ((TEXT_CODE >> 8) & 0xff));
digest.update((byte) (TEXT_CODE & 0xff));
updateWith(ch, start, length);
digest.update((byte) 0);
digest.update((byte) 0);
}
public void ignorableWhitespace(char ch[], int start, int length) throws SAXException {}
public void processingInstruction(String target, String data) throws SAXException {
digest.update((byte) ((PROCESSING_INSTRUCTION_CODE >> 24) & 0xff));
digest.update((byte) ((PROCESSING_INSTRUCTION_CODE >> 16) & 0xff));
digest.update((byte) ((PROCESSING_INSTRUCTION_CODE >> 8) & 0xff));
digest.update((byte) (PROCESSING_INSTRUCTION_CODE & 0xff));
updateWith(target);
digest.update((byte) 0);
digest.update((byte) 0);
updateWith(data);
digest.update((byte) 0);
digest.update((byte) 0);
}
public void skippedEntity(String name) throws SAXException {}
public void startDTD(String name, String publicId, String systemId) throws SAXException {}
public void endDTD() throws SAXException {}
public void startEntity(String name) throws SAXException {}
public void endEntity(String name) throws SAXException {}
public void startCDATA() throws SAXException {}
public void endCDATA() throws SAXException {}
public void comment(char[] ch, int start, int length) throws SAXException {
// We do consider comments significant for the purpose of digesting. But should this be an
// option?
digest.update((byte) ((COMMENT_CODE >> 24) & 0xff));
digest.update((byte) ((COMMENT_CODE >> 16) & 0xff));
digest.update((byte) ((COMMENT_CODE >> 8) & 0xff));
digest.update((byte) (COMMENT_CODE & 0xff));
updateWith(ch, start, length);
digest.update((byte) 0);
digest.update((byte) 0);
}
}
