Java tutorial
/** * personium.io * Copyright 2014 FUJITSU LIMITED * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.fujitsu.dc.test.utils; import java.io.BufferedOutputStream; import java.io.BufferedReader; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.OutputStream; import java.io.StringWriter; import java.io.UnsupportedEncodingException; import java.net.Socket; import java.net.URL; import java.security.KeyManagementException; import java.security.KeyStoreException; import java.security.NoSuchAlgorithmException; import java.security.SecureRandom; import java.security.cert.CertificateException; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import javax.net.SocketFactory; import javax.net.ssl.KeyManager; import javax.net.ssl.SSLContext; import javax.net.ssl.SSLSocket; import javax.net.ssl.TrustManager; import org.apache.commons.codec.CharEncoding; import org.apache.http.HttpHeaders; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import com.fujitsu.dc.test.DcCoreTestConfig; import com.fujitsu.dc.test.unit.core.UrlUtils; import com.sun.jersey.core.header.OutBoundHeaders; /** * Http?. * <h2>??</h2> * <pre> * HttpTest.request("{}") * .with("", "") * .with("", "") * .returns() * .statusCode(201); * </pre> * ????HTTP??????? * <pre> * GET / HTTP/1.1 * Host: ? * Connection: close * Content-Length: 0 * </pre> */ public class Http { /** * . */ private static Logger log = LoggerFactory.getLogger(Http.class); String method; String path; InputStream is; Map<String, String> params; byte[] paraBody; OutBoundHeaders headers = new OutBoundHeaders(); URL url; Socket socket; InputStream sIn; OutputStream sOut; BufferedReader sReader; BufferedOutputStream sWriter; private Http() { this.params = new HashMap<String, String>(); } /** * ????. * @return TResponse */ public TResponse returns() { BufferedReader br = null; try { // ??? InputStreamReader isr = new InputStreamReader(is, CharEncoding.UTF_8); br = new BufferedReader(isr); String firstLine = br.readLine(); firstLine = this.processParams(firstLine); String[] l1 = firstLine.split(" "); this.method = l1[0]; this.path = l1[1]; String protoVersion = l1[2]; // Open this.url = new URL(baseUrl + this.path); try { this.socket = createSocket(this.url); } catch (KeyManagementException e) { throw new RuntimeException(e); } catch (KeyStoreException e) { throw new RuntimeException(e); } catch (NoSuchAlgorithmException e) { throw new RuntimeException(e); } catch (CertificateException e) { throw new RuntimeException(e); } this.sIn = this.socket.getInputStream(); this.sOut = this.socket.getOutputStream(); this.sReader = new BufferedReader(new InputStreamReader(this.sIn, CharEncoding.UTF_8)); this.sWriter = new BufferedOutputStream(this.sOut); // ? StringBuilder sb = new StringBuilder(); sb.append(this.method); sb.append(" "); sb.append(this.url.getPath()); if (this.url.getQuery() != null) { sb.append("?"); sb.append(this.url.getQuery()); } sb.append(" "); sb.append(protoVersion); this.sWriter.write(sb.toString().getBytes(CharEncoding.UTF_8)); this.sWriter.write(CRLF.getBytes(CharEncoding.UTF_8)); log.debug("Req Start -------"); log.debug(sb.toString()); // Header String line = null; String lastLine = null; int contentLengthLineNum = -1; List<String> lines = new ArrayList<String>(); int i = 0; while ((line = br.readLine()) != null) { if (line.length() == 0) { break; } line = this.processParams(line); if (line.toLowerCase().startsWith(HttpHeaders.CONTENT_LENGTH.toLowerCase())) { // Content-Length??????????????????? contentLengthLineNum = i; } else if (line.toLowerCase().startsWith(HttpHeaders.HOST.toLowerCase())) { line = line.replaceAll("\\?", this.url.getAuthority()); } lines.add(line + CRLF); lastLine = line; i++; } // Version? lines.add("X-Dc-Version: " + DcCoreTestConfig.getCoreVersion() + CRLF); String body = null; // ????Break??????Body????? if (line != null) { log.debug("Req Body-------"); i = 1; StringWriter sw = new StringWriter(); int chr; while ((chr = br.read()) != -1) { sw.write((char) chr); } body = sw.toString(); body = this.processParams(body); // Content-Length? if (contentLengthLineNum != -1) { String contentLength = lines.get(contentLengthLineNum).replaceAll("\\?", String.valueOf(body.getBytes().length)); lines.set(contentLengthLineNum, contentLength); } } else { if (this.paraBody != null) { log.debug("Req Body-------"); // ?Body?? String contentLength = lines.get(contentLengthLineNum).replaceAll("\\?", String.valueOf(this.paraBody.length)); lines.set(contentLengthLineNum, contentLength); } else { // ??????. if (lastLine.length() > 0) { log.debug("one more CRLF"); this.sWriter.write(CRLF.getBytes(CharEncoding.UTF_8)); } } } // Header?? for (String l : lines) { this.sWriter.write(l.getBytes(CharEncoding.UTF_8)); if (log.isDebugEnabled()) { l.replaceAll(CRLF, ""); log.debug(l); } } // this.sWriter.write(CRLF.getBytes(CharEncoding.UTF_8)); // Body?? if (body != null) { this.sWriter.write(body.getBytes(CharEncoding.UTF_8)); log.debug(body); } // ?Body?? if (this.paraBody != null) { this.sWriter.write(this.paraBody); this.sWriter.write(CRLF.getBytes(CharEncoding.UTF_8)); } this.sWriter.flush(); // ?? TResponse ret = new TResponse(this.sReader); return ret; } catch (UnsupportedEncodingException e) { throw new RuntimeException(e); } catch (IOException e) { throw new RuntimeException(e); } catch (Exception e) { throw new RuntimeException(e); } finally { try { if (br != null) { br.close(); } if (this.sWriter != null) { this.sWriter.close(); } if (this.sReader != null) { this.sReader.close(); } } catch (IOException e) { throw new RuntimeException(e); } } } // ??? String processParams(String in) { String ret = in; for (String k : this.params.keySet()) { ret = ret.replaceAll("\\$\\{" + k + "\\}", this.params.get(k)); } return ret; } static final String CRLF = "\r\n"; /** * ??URL?? ??????URL? */ private static String baseUrl = System.getProperty(UrlUtils.PROP_TARGET_URL, UrlUtils.DEFAULT_TARGET_URL); /** * @param baseUrltoSet baseUrl */ public static void setBaseUrl(String baseUrltoSet) { baseUrl = baseUrltoSet; } /** * @return BaseURL */ public static String getBaseUrl() { return baseUrl; } /** * ??HttpTest?. * @param resPath ? * @return */ public static Http request(String resPath) { Http ret = new Http(); ret.is = ClassLoader.getSystemResourceAsStream("request/" + resPath); return ret; } /** * ????. * @param key key * @param value value * @return HttpTest */ public Http with(final String key, final String value) { this.params.put(key, value); return this; } /** * ????. * @param value value * @return HttpTest */ public Http setBodyBinary(final byte[] value) { this.paraBody = value; return this; } static final int PORT_HTTP = 80; static final int PORT_HTTPS = 443; static Socket createSocket(URL url) throws IOException, KeyStoreException, NoSuchAlgorithmException, CertificateException, KeyManagementException { String host = url.getHost(); int port = url.getPort(); String proto = url.getProtocol(); if (port < 0) { if ("https".equals(proto)) { port = PORT_HTTPS; } if ("http".equals(proto)) { port = PORT_HTTP; } } log.debug("sock: " + host + ":" + port); log.debug("proto: " + proto); // HTTPS?????????????SSLSocket??? if ("https".equals(proto)) { KeyManager[] km = null; TrustManager[] tm = { new javax.net.ssl.X509TrustManager() { public void checkClientTrusted(java.security.cert.X509Certificate[] arg0, String arg1) throws java.security.cert.CertificateException { log.debug("Insecure SSLSocket Impl for Testing: NOP at X509TrustManager#checkClientTrusted"); } public void checkServerTrusted(java.security.cert.X509Certificate[] arg0, String arg1) throws java.security.cert.CertificateException { log.debug("Insecure SSLSocket Impl for Testing: NOP at X509TrustManager#checkServerTrusted"); } public java.security.cert.X509Certificate[] getAcceptedIssuers() { return null; } } }; SSLContext sslContext = SSLContext.getInstance("TLS"); sslContext.init(km, tm, new SecureRandom()); SocketFactory sf = sslContext.getSocketFactory(); return (SSLSocket) sf.createSocket(host, port); } // HTTPS???????? return new Socket(host, port); } }