public static void close(Object o) {
try {
if (o == null) return;
if (o instanceof InputStream) {
((InputStream) o).close();
} else if (o instanceof OutputStream) {
((OutputStream) o).flush();
((OutputStream) o).close();
}
} catch (Exception e) {
e.printStackTrace();
}
}
public void run() {
try {
URL url = new URL(protocol + "://localhost:" + port + "/test1/" + f);
HttpURLConnection urlc = (HttpURLConnection) url.openConnection();
if (urlc instanceof HttpsURLConnection) {
HttpsURLConnection urlcs = (HttpsURLConnection) urlc;
urlcs.setHostnameVerifier(
new HostnameVerifier() {
public boolean verify(String s, SSLSession s1) {
return true;
}
});
urlcs.setSSLSocketFactory(ctx.getSocketFactory());
}
byte[] buf = new byte[4096];
if (fixedLen) {
urlc.setRequestProperty("XFixed", "yes");
}
InputStream is = urlc.getInputStream();
File temp = File.createTempFile("Test1", null);
temp.deleteOnExit();
OutputStream fout = new BufferedOutputStream(new FileOutputStream(temp));
int c, count = 0;
while ((c = is.read(buf)) != -1) {
count += c;
fout.write(buf, 0, c);
}
is.close();
fout.close();
if (count != size) {
throw new RuntimeException("wrong amount of data returned");
}
String orig = root + "/" + f;
compare(new File(orig), temp);
temp.delete();
} catch (Exception e) {
e.printStackTrace();
fail = true;
}
}
SOAPMessage post(SOAPMessage message, URL endPoint) throws SOAPException {
boolean isFailure = false;
URL url = null;
HttpURLConnection httpConnection = null;
int responseCode = 0;
try {
if (endPoint.getProtocol().equals("https"))
// if(!setHttps)
initHttps();
// Process the URL
JaxmURI uri = new JaxmURI(endPoint.toString());
String userInfo = uri.getUserinfo();
url = endPoint;
if (dL > 0) d("uri: " + userInfo + " " + url + " " + uri);
// TBD
// Will deal with https later.
if (!url.getProtocol().equalsIgnoreCase("http")
&& !url.getProtocol().equalsIgnoreCase("https")) {
log.severe("SAAJ0052.p2p.protocol.mustbe.http.or.https");
throw new IllegalArgumentException(
"Protocol " + url.getProtocol() + " not supported in URL " + url);
}
httpConnection = (HttpURLConnection) createConnection(url);
httpConnection.setRequestMethod("POST");
httpConnection.setDoOutput(true);
httpConnection.setDoInput(true);
httpConnection.setUseCaches(false);
httpConnection.setInstanceFollowRedirects(true);
if (message.saveRequired()) message.saveChanges();
MimeHeaders headers = message.getMimeHeaders();
Iterator it = headers.getAllHeaders();
boolean hasAuth = false; // true if we find explicit Auth header
while (it.hasNext()) {
MimeHeader header = (MimeHeader) it.next();
String[] values = headers.getHeader(header.getName());
if (values.length == 1)
httpConnection.setRequestProperty(header.getName(), header.getValue());
else {
StringBuffer concat = new StringBuffer();
int i = 0;
while (i < values.length) {
if (i != 0) concat.append(',');
concat.append(values[i]);
i++;
}
httpConnection.setRequestProperty(header.getName(), concat.toString());
}
if ("Authorization".equals(header.getName())) {
hasAuth = true;
log.fine("SAAJ0091.p2p.https.auth.in.POST.true");
}
}
if (!hasAuth && userInfo != null) {
initAuthUserInfo(httpConnection, userInfo);
}
OutputStream out = httpConnection.getOutputStream();
message.writeTo(out);
out.flush();
out.close();
httpConnection.connect();
try {
responseCode = httpConnection.getResponseCode();
// let HTTP_INTERNAL_ERROR (500) through because it is used for SOAP faults
if (responseCode == HttpURLConnection.HTTP_INTERNAL_ERROR) {
isFailure = true;
}
// else if (responseCode != HttpURLConnection.HTTP_OK)
// else if (!(responseCode >= HttpURLConnection.HTTP_OK && responseCode < 207))
else if ((responseCode / 100) != 2) {
log.log(
Level.SEVERE,
"SAAJ0008.p2p.bad.response",
new String[] {httpConnection.getResponseMessage()});
throw new SOAPExceptionImpl(
"Bad response: (" + responseCode + httpConnection.getResponseMessage());
}
} catch (IOException e) {
// on JDK1.3.1_01, we end up here, but then getResponseCode() succeeds!
responseCode = httpConnection.getResponseCode();
if (responseCode == HttpURLConnection.HTTP_INTERNAL_ERROR) {
isFailure = true;
} else {
throw e;
}
}
} catch (SOAPException ex) {
throw ex;
} catch (Exception ex) {
log.severe("SAAJ0009.p2p.msg.send.failed");
throw new SOAPExceptionImpl("Message send failed", ex);
}
SOAPMessage response = null;
if (responseCode == HttpURLConnection.HTTP_OK || isFailure) {
try {
MimeHeaders headers = new MimeHeaders();
String key, value;
// Header field 0 is the status line so we skip it.
int i = 1;
while (true) {
key = httpConnection.getHeaderFieldKey(i);
value = httpConnection.getHeaderField(i);
if (key == null && value == null) break;
if (key != null) {
StringTokenizer values = new StringTokenizer(value, ",");
while (values.hasMoreTokens()) headers.addHeader(key, values.nextToken().trim());
}
i++;
}
InputStream httpIn =
(isFailure ? httpConnection.getErrorStream() : httpConnection.getInputStream());
byte[] bytes = readFully(httpIn);
int length =
httpConnection.getContentLength() == -1
? bytes.length
: httpConnection.getContentLength();
// If no reply message is returned,
// content-Length header field value is expected to be zero.
if (length == 0) {
response = null;
log.warning("SAAJ0014.p2p.content.zero");
} else {
ByteInputStream in = new ByteInputStream(bytes, length);
response = messageFactory.createMessage(headers, in);
}
httpIn.close();
httpConnection.disconnect();
} catch (SOAPException ex) {
throw ex;
} catch (Exception ex) {
log.log(Level.SEVERE, "SAAJ0010.p2p.cannot.read.resp", ex);
throw new SOAPExceptionImpl("Unable to read response: " + ex.getMessage());
}
}
return response;
}
public static void main(String[] args) {
try {
if (args[0].equals("-genkey")) {
KeyPairGenerator pairgen = KeyPairGenerator.getInstance("RSA");
SecureRandom random = new SecureRandom();
pairgen.initialize(KEYSIZE, random);
KeyPair keyPair = pairgen.generateKeyPair();
ObjectOutputStream out = new ObjectOutputStream(new FileOutputStream(args[1]));
out.writeObject(keyPair.getPublic());
out.close();
out = new ObjectOutputStream(new FileOutputStream(args[2]));
out.writeObject(keyPair.getPrivate());
out.close();
} else if (args[0].equals("-encrypt")) {
KeyGenerator keygen = KeyGenerator.getInstance("AES");
SecureRandom random = new SecureRandom();
keygen.init(random);
SecretKey key = keygen.generateKey();
// wrap with RSA public key
ObjectInputStream keyIn = new ObjectInputStream(new FileInputStream(args[3]));
Key publicKey = (Key) keyIn.readObject();
keyIn.close();
Cipher cipher = Cipher.getInstance("RSA");
cipher.init(Cipher.WRAP_MODE, publicKey);
byte[] wrappedKey = cipher.wrap(key);
DataOutputStream out = new DataOutputStream(new FileOutputStream(args[2]));
out.writeInt(wrappedKey.length);
out.write(wrappedKey);
InputStream in = new FileInputStream(args[1]);
cipher = Cipher.getInstance("AES");
cipher.init(Cipher.ENCRYPT_MODE, key);
crypt(in, out, cipher);
in.close();
out.close();
} else {
DataInputStream in = new DataInputStream(new FileInputStream(args[1]));
int length = in.readInt();
byte[] wrappedKey = new byte[length];
in.read(wrappedKey, 0, length);
// unwrap with RSA private key
ObjectInputStream keyIn = new ObjectInputStream(new FileInputStream(args[3]));
Key privateKey = (Key) keyIn.readObject();
keyIn.close();
Cipher cipher = Cipher.getInstance("RSA");
cipher.init(Cipher.UNWRAP_MODE, privateKey);
Key key = cipher.unwrap(wrappedKey, "AES", Cipher.SECRET_KEY);
OutputStream out = new FileOutputStream(args[2]);
cipher = Cipher.getInstance("AES");
cipher.init(Cipher.DECRYPT_MODE, key);
crypt(in, out, cipher);
in.close();
out.close();
}
} catch (IOException e) {
e.printStackTrace();
} catch (GeneralSecurityException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
/** Unpacks a resource to a temp. file */
public static File unpackInstaller(String resourceName) {
// Array to hold all results (this code is slightly more
// generally that it needs to be)
File[] results = new File[1];
URL[] urls = new URL[1];
// Determine size of download
ClassLoader cl = Main.class.getClassLoader();
urls[0] = cl.getResource(Config.getInstallerResource());
if (urls[0] == null) {
Config.trace("Could not find resource: " + Config.getInstallerResource());
return null;
}
int totalSize = 0;
int totalRead = 0;
for (int i = 0; i < urls.length; i++) {
if (urls[i] != null) {
try {
URLConnection connection = urls[i].openConnection();
totalSize += connection.getContentLength();
} catch (IOException ioe) {
Config.trace("Got exception: " + ioe);
return null;
}
}
}
// Unpack each file
for (int i = 0; i < urls.length; i++) {
if (urls[i] != null) {
// Create temp. file to store unpacked file in
InputStream in = null;
OutputStream out = null;
try {
// Use extension from URL (important for dll files)
String extension = new File(urls[i].getFile()).getName();
int lastdotidx = (extension != null) ? extension.lastIndexOf('.') : -1;
if (lastdotidx == -1) {
extension = ".dat";
} else {
extension = extension.substring(lastdotidx);
}
// Create output stream
results[i] = File.createTempFile("jre", extension);
results[i].deleteOnExit();
out = new FileOutputStream(results[i]);
// Create inputstream
URLConnection connection = urls[i].openConnection();
in = connection.getInputStream();
int read = 0;
byte[] buf = new byte[BUFFER_SIZE];
while ((read = in.read(buf)) != -1) {
out.write(buf, 0, read);
// Notify delegate
totalRead += read;
if (totalRead > totalSize && totalSize != 0) totalSize = totalRead;
// Update UI
if (totalSize != 0) {
int percent = (100 * totalRead) / totalSize;
setStepText(STEP_UNPACK, Config.getWindowStepProgress(STEP_UNPACK, percent));
}
}
} catch (IOException ie) {
Config.trace("Got exception while downloading resource: " + ie);
for (int j = 0; j < results.length; j++) {
if (results[j] != null) results[j].delete();
}
return null;
} finally {
try {
if (in != null) in.close();
if (out != null) out.close();
} catch (IOException io) {
/* ignore */
}
}
}
}
setStepText(STEP_UNPACK, Config.getWindowStep(STEP_UNPACK));
return results[0];
}
