private Map<String, String> parseHeaders(HttpURLConnection conn) {
Map<String, String> headers = new HashMap<String, String>();
for (String key : conn.getHeaderFields().keySet()) {
headers.put(key, conn.getHeaderFields().get(key).get(0));
}
return headers;
}
/**
* Loads the drawing. By convention this method is invoked on a worker thread.
*
* @param progress A ProgressIndicator to inform the user about the progress of the operation.
* @return The Drawing that was loaded.
*/
protected Drawing loadDrawing(ProgressIndicator progress) throws IOException {
Drawing drawing = createDrawing();
if (getParameter("datafile") != null) {
URL url = new URL(getDocumentBase(), getParameter("datafile"));
URLConnection uc = url.openConnection();
// Disable caching. This ensures that we always request the
// newest version of the drawing from the server.
// (Note: The server still needs to set the proper HTTP caching
// properties to prevent proxies from caching the drawing).
if (uc instanceof HttpURLConnection) {
((HttpURLConnection) uc).setUseCaches(false);
}
// Read the data into a buffer
int contentLength = uc.getContentLength();
InputStream in = uc.getInputStream();
try {
if (contentLength != -1) {
in = new BoundedRangeInputStream(in);
((BoundedRangeInputStream) in).setMaximum(contentLength + 1);
progress.setProgressModel((BoundedRangeModel) in);
progress.setIndeterminate(false);
}
BufferedInputStream bin = new BufferedInputStream(in);
bin.mark(512);
// Read the data using all supported input formats
// until we succeed
IOException formatException = null;
for (InputFormat format : drawing.getInputFormats()) {
try {
bin.reset();
} catch (IOException e) {
uc = url.openConnection();
in = uc.getInputStream();
in = new BoundedRangeInputStream(in);
((BoundedRangeInputStream) in).setMaximum(contentLength + 1);
progress.setProgressModel((BoundedRangeModel) in);
bin = new BufferedInputStream(in);
bin.mark(512);
}
try {
bin.reset();
format.read(bin, drawing, true);
formatException = null;
break;
} catch (IOException e) {
formatException = e;
}
}
if (formatException != null) {
throw formatException;
}
} finally {
in.close();
}
}
return drawing;
}
/**
* 运行时，添加JVM参数“-Dsun.net.http.retryPost=false”，可阻止自动重连。
*
* @see 'http://www.coderanch.com/t/490463/sockets/java/Timeout-retry-URLHTTPRequest'
*/
@Test
public void testConnectionResetByHttpURLConnection() throws IOException {
testConnectionResetCount = 0;
String resp = null;
try {
HttpURLConnection conn =
(HttpURLConnection) new URL("http://localhost:65532/soso").openConnection();
conn.setDoOutput(true);
conn.setRequestMethod("POST");
conn.getOutputStream().write("username".getBytes());
resp = conn.getResponseCode() + "";
} catch (IOException e) {
Throwable ee = ExceptionUtils.getRootCause(e);
if (ee == null) {
ee = e;
}
Logger.error(this, "", ee);
Assert.assertNotSame(NoHttpResponseException.class, ee.getClass());
Assert.assertSame(SocketException.class, ee.getClass());
Assert.assertTrue(
"Connection reset".equals(ee.getMessage())
|| "Socket closed".equals(ee.getMessage())
|| "Unexpected end of file from server".equals(ee.getMessage()));
} finally {
Logger.info(
this,
"resp[HttpURLConnection]-["
+ testConnectionResetCount
+ "]=========["
+ resp
+ "]=========");
}
Assert.assertEquals(2, testConnectionResetCount);
}
// прочитать весь json в строку
private static String readAll() throws IOException {
StringBuilder data = new StringBuilder();
try {
HttpURLConnection con = (HttpURLConnection) ((new URL(PRODUCT_URL).openConnection()));
con.setRequestMethod("GET");
con.setDoInput(true);
String s;
try (BufferedReader in = new BufferedReader(new InputStreamReader(con.getInputStream()))) {
while ((s = in.readLine()) != null) {
data.append(s);
}
}
} catch (MalformedURLException e) {
e.printStackTrace();
throw new MalformedURLException("Url is not valid");
} catch (ProtocolException e) {
e.printStackTrace();
throw new ProtocolException("No such protocol, it must be POST,GET,PATCH,DELETE etc.");
} catch (IOException e) {
e.printStackTrace();
throw new IOException("cannot read from server");
}
return data.toString();
}
public SetIfModifiedSince() throws Exception {
serverSock = new ServerSocket(0);
int port = serverSock.getLocalPort();
Thread thr = new Thread(this);
thr.start();
Date date = new Date(new Date().getTime() - 1440000); // this time yesterday
URL url;
HttpURLConnection con;
// url = new URL(args[0]);
url = new URL("http://localhost:" + String.valueOf(port) + "/anything");
con = (HttpURLConnection) url.openConnection();
con.setIfModifiedSince(date.getTime());
con.connect();
int ret = con.getResponseCode();
if (ret == 304) {
System.out.println("Success!");
} else {
throw new RuntimeException(
"Test failed! Http return code using setIfModified method is:"
+ ret
+ "\nNOTE:some web servers are not implemented according to RFC, thus a failed test does not necessarily indicate a bug in setIfModifiedSince method");
}
}
public boolean shutdown(int port, boolean ssl) {
try {
String protocol = "http" + (ssl ? "s" : "");
URL url = new URL(protocol, "127.0.0.1", port, "shutdown");
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
conn.setRequestMethod("GET");
conn.setRequestProperty("servicemanager", "shutdown");
conn.connect();
StringBuffer sb = new StringBuffer();
BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream(), "UTF-8"));
int n;
char[] cbuf = new char[1024];
while ((n = br.read(cbuf, 0, cbuf.length)) != -1) sb.append(cbuf, 0, n);
br.close();
String message = sb.toString().replace("<br>", "\n");
if (message.contains("Goodbye")) {
cp.appendln("Shutting down the server:");
String[] lines = message.split("\n");
for (String line : lines) {
cp.append("...");
cp.appendln(line);
}
return true;
}
} catch (Exception ex) {
}
cp.appendln("Unable to shutdown CTP");
return false;
}
Response(HttpURLConnection connection) throws IOException {
try {
connection.connect();
code = connection.getResponseCode();
headers = parseHeaders(connection);
stream = isSuccessful() ? connection.getInputStream() : connection.getErrorStream();
} catch (UnknownHostException e) {
throw new OAuthException("The IP address of a host could not be determined.", e);
}
}
void addBody(HttpURLConnection conn, byte[] content) throws IOException {
conn.setRequestProperty(CONTENT_LENGTH, String.valueOf(content.length));
// Set default content type if none is set.
if (conn.getRequestProperty(CONTENT_TYPE) == null) {
conn.setRequestProperty(CONTENT_TYPE, DEFAULT_CONTENT_TYPE);
}
conn.setDoOutput(true);
conn.getOutputStream().write(content);
}
private T requestInfo(URI baseUrl, RenderingContext context)
throws IOException, URISyntaxException, ParserConfigurationException, SAXException {
URL url = loader.createURL(baseUrl, context);
GetMethod method = null;
try {
final InputStream stream;
if ((url.getProtocol().equals("http") || url.getProtocol().equals("https"))
&& context.getConfig().localHostForwardIsFrom(url.getHost())) {
String scheme = url.getProtocol();
final String host = url.getHost();
if (url.getProtocol().equals("https")
&& context.getConfig().localHostForwardIsHttps2http()) {
scheme = "http";
}
URL localUrl = new URL(scheme, "localhost", url.getPort(), url.getFile());
HttpURLConnection connexion = (HttpURLConnection) localUrl.openConnection();
connexion.setRequestProperty("Host", host);
for (Map.Entry<String, String> entry : context.getHeaders().entrySet()) {
connexion.setRequestProperty(entry.getKey(), entry.getValue());
}
stream = connexion.getInputStream();
} else {
method = new GetMethod(url.toString());
for (Map.Entry<String, String> entry : context.getHeaders().entrySet()) {
method.setRequestHeader(entry.getKey(), entry.getValue());
}
context.getConfig().getHttpClient(baseUrl).executeMethod(method);
int code = method.getStatusCode();
if (code < 200 || code >= 300) {
throw new IOException(
"Error "
+ code
+ " while reading the Capabilities from "
+ url
+ ": "
+ method.getStatusText());
}
stream = method.getResponseBodyAsStream();
}
final T result;
try {
result = loader.parseInfo(stream);
} finally {
stream.close();
}
return result;
} finally {
if (method != null) {
method.releaseConnection();
}
}
}
public ErrorInfo listPolicy(String polname, String token) throws IOException, RestException {
String realm = "/";
String data = null;
ErrorInfo ei = null;
InputStreamReader iss = null;
BufferedReader br = null;
HttpURLConnection urlc = null;
InputStream inputStream = null;
try {
data =
"policynames="
+ URLEncoder.encode(polname, "UTF-8")
+ "&realm="
+ URLEncoder.encode(realm, "UTF-8")
+ "&submit="
+ URLEncoder.encode("Submit", "UTF-8");
} catch (UnsupportedEncodingException e) {
System.out.println("OpenssoHelper: " + e.getMessage());
e.printStackTrace();
}
if (data != null) {
try {
r.Connect(new URL(url + ssoadm_list));
} catch (MalformedURLException e) {
System.out.println("OpenssoHelper: " + e.getMessage());
e.printStackTrace();
}
urlc = (HttpURLConnection) r.c;
urlc.addRequestProperty("Cookie", "iPlanetDirectoryPro=\"" + token + "\"");
r.Send(urlc, data);
String answer = null;
int status = 0;
inputStream = urlc.getInputStream();
iss = new InputStreamReader(inputStream);
br = new BufferedReader(iss);
answer = BrToString(br);
status = urlc.getResponseCode();
if (answer != null) {
ei = new ErrorInfo(answer, status);
}
br.close();
iss.close();
inputStream.close();
urlc.disconnect();
}
return ei;
}
@Test(
description = "Verify the downloads links return 200 rather than 404",
groups = {"functional"})
public void DownloadsTab_02() throws HarnessException {
// Determine which links should be present
List<String> locators = new ArrayList<String>();
if (ZimbraSeleniumProperties.zimbraGetVersionString().contains("NETWORK")) {
locators.addAll(Arrays.asList(NetworkOnlyLocators));
locators.addAll(Arrays.asList(CommonLocators));
} else if (ZimbraSeleniumProperties.zimbraGetVersionString().contains("FOSS")) {
locators.addAll(Arrays.asList(FossOnlyLocators));
locators.addAll(Arrays.asList(CommonLocators));
} else {
throw new HarnessException(
"Unable to find NETWORK or FOSS in version string: "
+ ZimbraSeleniumProperties.zimbraGetVersionString());
}
for (String locator : locators) {
String href = app.zPageDownloads.sGetAttribute("xpath=" + locator + "@href");
String page = ZimbraSeleniumProperties.getBaseURL() + href;
HttpURLConnection connection = null;
try {
URL url = new URL(page);
connection = (HttpURLConnection) url.openConnection();
connection.setRequestMethod("HEAD");
int code = connection.getResponseCode();
// TODO: why is 400 returned for the PDF links?
// 200 and 400 are acceptable
ZAssert.assertStringContains(
"200 400", "" + code, "Verify the download URL is valid: " + url.toString());
} catch (MalformedURLException e) {
throw new HarnessException(e);
} catch (IOException e) {
throw new HarnessException(e);
} finally {
if (connection != null) {
connection.disconnect();
connection = null;
}
}
}
}
private byte[] downloadByteArray(URL url) throws IOException {
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
conn.setRequestProperty("Cookie", cookies);
if (conn.getResponseCode() == HttpURLConnection.HTTP_NOT_FOUND) return (null);
InputStream in = conn.getInputStream();
byte[] buf = new byte[conn.getContentLength()];
int read, offset = 0;
while ((read = in.read(buf, offset, buf.length - offset)) != -1) {
offset += read;
}
return (buf);
}
public ErrorInfo doLogin() throws IOException, RestException {
String data = null;
ErrorInfo ei = null;
InputStreamReader iss = null;
BufferedReader br = null;
HttpURLConnection urlc = null;
InputStream inputStream = null;
try {
data =
"username="******"UTF-8")
+ "&password="******"UTF-8");
} catch (UnsupportedEncodingException e) {
System.out.println("OpenssoHelper: " + e.getMessage());
e.printStackTrace();
}
if (data != null) {
try {
r.Connect(new URL(url + authenticate));
} catch (MalformedURLException e) {
System.out.println("OpenssoHelper: " + e.getMessage());
e.printStackTrace();
}
urlc = (HttpURLConnection) r.c;
r.Send(urlc, data);
String answer = null;
int status = 0;
inputStream = urlc.getInputStream();
iss = new InputStreamReader(inputStream);
br = new BufferedReader(iss);
answer = BrToString(br);
status = urlc.getResponseCode();
if (answer != null) {
ei = new ErrorInfo(answer, status);
}
br.close();
iss.close();
inputStream.close();
urlc.disconnect();
}
return ei;
}
private String getResponse(String link) {
try {
URL urlObject = new URL(link);
HttpURLConnection httpUrlConnection = (HttpURLConnection) urlObject.openConnection();
httpUrlConnection.setRequestMethod("GET");
httpUrlConnection.setDoOutput(true);
httpUrlConnection.setRequestProperty("Cookie", cookieValue);
return getResponse(httpUrlConnection.getInputStream());
} catch (Exception e) {
e.printStackTrace();
}
return "";
}
/**
* 以阻塞的方式立即下载一个文件，该方法会覆盖已经存在的文件
*
* @param task
* @return "ok" if download success (else return errmessage);
*/
static String download(DownloadTask task) {
if (!openedStatus && show) openStatus();
URL url;
HttpURLConnection conn;
try {
url = new URL(task.getOrigin());
conn = (HttpURLConnection) url.openConnection();
conn.setConnectTimeout(30000);
conn.setReadTimeout(30000);
conn.setRequestProperty(
"User-Agent",
"Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/43.0.2357.134 Safari/"
+ Math.random());
if ("www.imgjav.com".equals(url.getHost())) { // 该网站需要带cookie请求
conn.setRequestProperty(
"User-Agent",
"Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/43.0.2357.134 Safari/537.36");
// conn.setRequestProperty("Cookie", "__cfduid=d219ea333c7a9b5743b572697b631925a1446093229;
// cf_clearance=6ae62d843f5d09acf393f9e4eb130d9366840c82-1446093303-28800");
conn.setRequestProperty(
"Cookie",
"__cfduid=d6ee846b378bb7d5d173a05541f8a2b6a1446090548; cf_clearance=ea10e8db31f8b6ee51570b118dd89b7e616d7b62-1446099714-28800");
conn.setRequestProperty("Host", "www.imgjav.com");
}
Path directory = Paths.get(task.getDest()).getParent();
if (!Files.exists(directory)) Files.createDirectories(directory);
} catch (Exception e) {
e.printStackTrace();
return e.getMessage();
}
try (InputStream is = conn.getInputStream();
BufferedInputStream in = new BufferedInputStream(is);
FileOutputStream fos = new FileOutputStream(task.getDest());
OutputStream out = new BufferedOutputStream(fos); ) {
int length = conn.getContentLength();
if (length < 1) throw new IOException("length<1");
byte[] binary = new byte[length];
byte[] buff = new byte[65536];
int len;
int index = 0;
while ((len = in.read(buff)) != -1) {
System.arraycopy(buff, 0, binary, index, len);
index += len;
allLen += len; // allLen有线程安全的问题 ，可能会不正确。无需精确数据，所以不同步了
task.setReceivePercent(String.format("%.2f", ((float) index / length) * 100) + "%");
}
out.write(binary);
} catch (IOException e) {
e.printStackTrace();
return e.getMessage();
}
return "ok";
}
private Map executeHTTP(Map data_to_send, boolean v6) throws Exception {
if (v6 && !enable_v6) {
throw (new Exception("IPv6 is disabled"));
}
String host = getHost(v6, HTTP_SERVER_ADDRESS_V6, HTTP_SERVER_ADDRESS_V4);
if (Logger.isEnabled())
Logger.log(
new LogEvent(
LOGID,
"VersionCheckClient retrieving "
+ "version information from "
+ host
+ ":"
+ HTTP_SERVER_PORT
+ " via HTTP"));
String url_str =
"http://"
+ (v6 ? UrlUtils.convertIPV6Host(host) : host)
+ (HTTP_SERVER_PORT == 80 ? "" : (":" + HTTP_SERVER_PORT))
+ "/version?";
url_str +=
URLEncoder.encode(new String(BEncoder.encode(data_to_send), "ISO-8859-1"), "ISO-8859-1");
URL url = new URL(url_str);
HttpURLConnection url_connection = (HttpURLConnection) url.openConnection();
url_connection.connect();
try {
InputStream is = url_connection.getInputStream();
Map reply = BDecoder.decode(new BufferedInputStream(is));
preProcessReply(reply, v6);
return (reply);
} finally {
url_connection.disconnect();
}
}
// добавить продукт в базу json server
public static void add(ProductREST product) throws NotValidProductException, IOException {
try {
check(product); // рповеряем продукт
// connection к серверу
HttpURLConnection con = (HttpURLConnection) ((new URL(PRODUCT_URL).openConnection()));
con.setRequestMethod("POST"); // метод post для добавления
// генерируем запрос
StringBuilder urlParameters = new StringBuilder();
urlParameters
.append("name=")
.append(URLEncoder.encode(product.getName(), "UTF8"))
.append("&");
urlParameters.append("price=").append(product.getPrice()).append("&");
urlParameters.append("weight=").append(product.getWeight()).append("&");
urlParameters
.append("manufacturer=")
.append(product.getManufacturer().getCountry())
.append("&");
urlParameters.append("category=").append(URLEncoder.encode(product.getCategory(), "UTF8"));
con.setDoOutput(true); // разрешаем отправку данных
// отправляем
try (DataOutputStream out = new DataOutputStream(con.getOutputStream())) {
out.writeBytes(urlParameters.toString());
}
// код ответа
int responseCode = con.getResponseCode();
System.out.println("\nSending 'POST' request to URL : " + PRODUCT_URL);
System.out.println("Response Code : " + responseCode);
} catch (NotValidProductException e) {
e.printStackTrace();
throw e;
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
throw new UnsupportedEncodingException("cannot recognize encoding");
} catch (ProtocolException e) {
e.printStackTrace();
throw new ProtocolException("No such protocol, protocol must be POST,DELETE,PATCH,GET etc.");
} catch (MalformedURLException e) {
e.printStackTrace();
throw new MalformedURLException("Url is not valid");
} catch (IOException e) {
e.printStackTrace();
throw new IOException("cannot write information to server");
}
}
public static void main(String args[]) {
for (int i = 0; i < args.length; i++) {
try {
URL u = new URL(args[i]);
HttpURLConnection http = (HttpURLConnection) u.openConnection();
http.setRequestMethod("HEAD");
System.out.println(u + "was last modified at " + new Date(http.getLastModified()));
} // end try
catch (MalformedURLException ex) {
System.err.println(args[i] + " is not a URL I understand");
} catch (IOException ex) {
System.err.println(ex);
}
System.out.println();
} // end for
} // end main
Response doSend() throws IOException {
connection.setRequestMethod(this.verb.name());
if (connectTimeout != null) {
connection.setConnectTimeout(connectTimeout.intValue());
}
if (readTimeout != null) {
connection.setReadTimeout(readTimeout.intValue());
}
if (boundary != null) {
connection.setRequestProperty(CONTENT_TYPE, "multipart/form-data; boundary=" + boundary);
}
addHeaders(connection);
if (verb.equals(Verb.PUT) || verb.equals(Verb.POST)) {
addBody(connection, getByteBodyContents());
}
return new Response(connection);
}
@Override
public void run() {
try {
URL url = buildPingUrl();
if (logger.isDebugEnabled()) {
logger.debug("Sending UDC information to {}...", url);
}
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
conn.setConnectTimeout((int) HTTP_TIMEOUT.millis());
conn.setReadTimeout((int) HTTP_TIMEOUT.millis());
if (conn.getResponseCode() >= 300) {
throw new Exception(
String.format("%s Responded with Code %d", url.getHost(), conn.getResponseCode()));
}
if (logger.isDebugEnabled()) {
BufferedReader reader = new BufferedReader(new InputStreamReader(conn.getInputStream()));
String line = reader.readLine();
while (line != null) {
logger.debug(line);
line = reader.readLine();
}
reader.close();
} else {
conn.getInputStream().close();
}
successCounter.incrementAndGet();
} catch (Exception e) {
if (logger.isDebugEnabled()) {
logger.debug("Error sending UDC information", e);
}
failCounter.incrementAndGet();
}
}
public void run() {
try {
Thread.sleep(10);
byte[] buf = getBuf();
URL url = new URL("http://127.0.0.1:" + port + "/test");
HttpURLConnection con = (HttpURLConnection) url.openConnection();
con.setDoOutput(true);
con.setDoInput(true);
con.setRequestMethod("POST");
con.setRequestProperty(
"Content-Type",
"Multipart/Related; type=\"application/xop+xml\"; boundary=\"----=_Part_0_6251267.1128549570165\"; start-info=\"text/xml\"");
OutputStream out = con.getOutputStream();
out.write(buf);
out.close();
InputStream in = con.getInputStream();
byte[] newBuf = readFully(in);
in.close();
if (buf.length != newBuf.length) {
System.out.println("Doesn't match");
error = true;
}
synchronized (lock) {
++received;
if ((received % 1000) == 0) {
System.out.println("Received=" + received);
}
}
} catch (Exception e) {
// e.printStackTrace();
System.out.print(".");
error = true;
}
}
private byte[] sendHttpMessage(byte[] body) throws IOException {
HttpMessageContext context = HttpMessageContext.getInstance();
context.Debug("GNHttpConnection [sendHttpMessage] starts");
if (context.getLogheader()) {
context.Info(logMessageSetting());
}
connection.setDoInput(true);
if (body != null) {
connection.setDoOutput(true);
OutputStream os = TimedURLConnection.getOutputStream(connection, timeout * 1000);
context.Debug("GNHttpConnection [sendHttpMessage] sending message ...");
os.write(body);
context.Debug("GNHttpConnection [sendHttpMessage] message sent");
}
context.Debug(
"GNHttpConnection [sendHttpMessage] TimedURLConnection.getInputStream timeout["
+ timeout
+ " S]");
InputStream is = TimedURLConnection.getInputStream(connection, timeout * 1000);
responseCode = connection.getResponseCode();
context.Debug("GNHttpConnection [sendHttpMessage] responseCode[" + responseCode + "]");
responseMessage = HttpMessageContext.getMessage(is);
responseheaders = new Hashtable<String, String>();
int no = 0;
while (true) {
String headerName = connection.getHeaderFieldKey(no);
String headerValue = connection.getHeaderField(no);
if (headerName != null && headerName.length() > 0) {
responseheaders.put(headerName, headerValue);
} else {
if (headerValue == null || headerValue.length() <= 0) break;
}
no++;
}
if (context.getLogheader()) {
GTConfigFile head = new GTConfigFile(responseheaders);
context.Debug("GNHttpConnection [sendHttpMessage] responseHeader\r\n" + head);
context.Debug(
"GNHttpConnection [sendHttpMessage] responseMessage\r\n"
+ new String(getResponseMessage()));
context.Info("GNHttpConnection [sendHttpMessage] success for " + url);
} else context.Info("GNHttpConnection [sendHttpMessage] success for " + url);
return responseMessage;
}
Response doSend(RequestTuner tuner) throws IOException
{
connection.setRequestMethod(this.verb.name());
if (connectTimeout != null)
{
connection.setConnectTimeout(connectTimeout.intValue());
}
if (readTimeout != null)
{
connection.setReadTimeout(readTimeout.intValue());
}
tuner.tune(this);
addHeaders(connection);
if (verb.equals(Verb.PUT) || verb.equals(Verb.POST))
{
addBody(connection, getByteBodyContents());
}
return new Response(connection);
}
private void createConnection() throws IOException
{
String completeUrl = getCompleteUrl();
if (connection == null)
{
System.setProperty("http.keepAlive", connectionKeepAlive ? "true" : "false");
connection = (HttpURLConnection) new URL(completeUrl).openConnection();
connection.setInstanceFollowRedirects(followRedirects);
}
}
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;
}
}
private boolean checkServer(int port, boolean ssl) {
try {
URL url = new URL("http://127.0.0.1:" + port);
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
conn.setRequestMethod("GET");
conn.connect();
int length = conn.getContentLength();
StringBuffer text = new StringBuffer();
InputStream is = conn.getInputStream();
InputStreamReader isr = new InputStreamReader(is);
int size = 256;
char[] buf = new char[size];
int len;
while ((len = isr.read(buf, 0, size)) != -1) text.append(buf, 0, len);
isr.close();
if (programName.equals("ISN")) return !shutdown(port, ssl);
return true;
} catch (Exception ex) {
return false;
}
}
public static String visitWeb(String urlStr) {
URL url = null;
HttpURLConnection httpConn = null;
InputStream in = null;
try {
url = new URL(urlStr);
httpConn = (HttpURLConnection) url.openConnection();
HttpURLConnection.setFollowRedirects(true);
httpConn.setRequestMethod("GET");
httpConn.setRequestProperty("User-Agent", "Mozilla/4.0(compatible;MSIE 6.0;Windows 2000)");
in = httpConn.getInputStream();
return convertStreamToString(in);
} catch (MalformedURLException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
in.close();
httpConn.disconnect();
} catch (Exception ex) {
ex.printStackTrace();
}
}
return null;
}
private boolean handshake() throws Exception {
URL homePage = new URL("http://mangaonweb.com/viewer.do?ctsn=" + ctsn);
HttpURLConnection urlConn = (HttpURLConnection) homePage.openConnection();
urlConn.connect();
if (urlConn.getResponseCode() == HttpURLConnection.HTTP_NOT_FOUND) return (false);
// save the cookie
String headerName = null;
for (int i = 1; (headerName = urlConn.getHeaderFieldKey(i)) != null; i++) {
if (headerName.equals("Set-Cookie")) {
cookies = urlConn.getHeaderField(i);
}
}
// save cdn and crcod
String page = "", line;
BufferedReader stream =
new BufferedReader(new InputStreamReader(urlConn.getInputStream(), "UTF-8"));
while ((line = stream.readLine()) != null) page += line;
cdn = param(page, "cdn");
crcod = param(page, "crcod");
return (true);
}
@Test(groups = {"pulse"}) // test method
public void testUserTx() throws Exception {
try {
String testurl =
"http://" + host + ":" + port + "/" + strContextRoot + "/MyServlet?testcase=usertx";
URL url = new URL(testurl);
echo("Connecting to: " + url.toString());
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
conn.connect();
int responseCode = conn.getResponseCode();
InputStream is = conn.getInputStream();
BufferedReader input = new BufferedReader(new InputStreamReader(is));
String line = null;
boolean result = false;
String testLine = null;
String EXPECTED_RESPONSE = "user-tx-commit:true";
String EXPECTED_RESPONSE2 = "user-tx-rollback:true";
while ((line = input.readLine()) != null) {
// echo(line);
if (line.indexOf(EXPECTED_RESPONSE) != -1 && line.indexOf(EXPECTED_RESPONSE2) != -1) {
testLine = line;
echo(testLine);
result = true;
break;
}
}
Assert.assertEquals(result, true, "Unexpected Results");
} catch (Exception e) {
e.printStackTrace();
throw new Exception(e);
}
}
private boolean test(String c) throws Exception {
String EXPECTED_RESPONSE = c + ":pass";
boolean result = false;
String url = "http://" + host + ":" + port + strContextRoot + "/jpa?testcase=" + c;
// System.out.println("url="+url);
HttpURLConnection conn = (HttpURLConnection) (new URL(url)).openConnection();
int code = conn.getResponseCode();
if (code != 200) {
System.err.println("Unexpected return code: " + code);
} else {
InputStream is = conn.getInputStream();
BufferedReader input = new BufferedReader(new InputStreamReader(is));
String line = null;
while ((line = input.readLine()) != null) {
if (line.contains(EXPECTED_RESPONSE)) {
result = true;
break;
}
}
}
return result;
}
