James Moger
2012-01-05 cb285cbfddfc0b633d6b8cdb4dc0d2bd2b8b51ef
src/com/gitblit/utils/JsonUtils.java
@@ -22,29 +22,31 @@
import java.io.OutputStream;
import java.lang.reflect.Type;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.security.SecureRandom;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Date;
import java.util.Locale;
import java.util.Map;
import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSession;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import org.eclipse.jgit.util.Base64;
import java.util.TimeZone;
import com.gitblit.GitBlitException.ForbiddenException;
import com.gitblit.GitBlitException.NotAllowedException;
import com.gitblit.GitBlitException.UnauthorizedException;
import com.gitblit.GitBlitException.UnknownRequestException;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonPrimitive;
import com.google.gson.JsonSerializationContext;
import com.google.gson.JsonSerializer;
import com.google.gson.JsonSyntaxException;
import com.google.gson.reflect.TypeToken;
/**
@@ -55,30 +57,11 @@
 */
public class JsonUtils {
   public static final String CHARSET;
   public static final Type REPOSITORIES_TYPE = new TypeToken<Map<String, RepositoryModel>>() {
   }.getType();
   public static final Type USERS_TYPE = new TypeToken<Collection<UserModel>>() {
   }.getType();
   private static final SSLContext SSL_CONTEXT;
   private static final DummyHostnameVerifier HOSTNAME_VERIFIER;
   static {
      SSLContext context = null;
      try {
         context = SSLContext.getInstance("SSL");
         context.init(null, new TrustManager[] { new DummyTrustManager() }, new SecureRandom());
      } catch (Throwable t) {
         t.printStackTrace();
      }
      SSL_CONTEXT = context;
      HOSTNAME_VERIFIER = new DummyHostnameVerifier();
      CHARSET = "UTF-8";
   }
   /**
    * Creates JSON from the specified object.
@@ -87,8 +70,7 @@
    * @return json
    */
   public static String toJsonString(Object o) {
      Gson gson = new GsonBuilder().setPrettyPrinting().create();
      String json = gson.toJson(o);
      String json = gson().toJson(o);
      return json;
   }
@@ -100,8 +82,7 @@
    * @return an object
    */
   public static <X> X fromJsonString(String json, Class<X> clazz) {
      Gson gson = new Gson();
      return gson.fromJson(json, clazz);
      return gson().fromJson(json, clazz);
   }
   /**
@@ -112,8 +93,7 @@
    * @return an object
    */
   public static <X> X fromJsonString(String json, Type type) {
      Gson gson = new Gson();
      return gson.fromJson(json, type);
      return gson().fromJson(json, type);
   }
   /**
@@ -127,6 +107,19 @@
   public static <X> X retrieveJson(String url, Type type) throws IOException,
         UnauthorizedException {
      return retrieveJson(url, type, null, null);
   }
   /**
    * Reads a gson object from the specified url.
    *
    * @param url
    * @param type
    * @return the deserialized object
    * @throws {@link IOException}
    */
   public static <X> X retrieveJson(String url, Class<? extends X> clazz) throws IOException,
         UnauthorizedException {
      return retrieveJson(url, clazz, null, null);
   }
   /**
@@ -145,8 +138,26 @@
      if (StringUtils.isEmpty(json)) {
         return null;
      }
      Gson gson = new Gson();
      return gson.fromJson(json, type);
      return gson().fromJson(json, type);
   }
   /**
    * Reads a gson object from the specified url.
    *
    * @param url
    * @param clazz
    * @param username
    * @param password
    * @return the deserialized object
    * @throws {@link IOException}
    */
   public static <X> X retrieveJson(String url, Class<X> clazz, String username, char[] password)
         throws IOException {
      String json = retrieveJsonString(url, username, password);
      if (StringUtils.isEmpty(json)) {
         return null;
      }
      return gson().fromJson(json, clazz);
   }
   /**
@@ -158,20 +169,10 @@
    */
   public static String retrieveJsonString(String url, String username, char[] password)
         throws IOException {
      try {
         URL urlObject = new URL(url);
         URLConnection conn = urlObject.openConnection();
         conn.setRequestProperty("Accept-Charset", CHARSET);
         setAuthorization(conn, username, password);
         conn.setUseCaches(false);
         conn.setDoInput(true);
         if (conn instanceof HttpsURLConnection) {
            HttpsURLConnection secureConn = (HttpsURLConnection) conn;
            secureConn.setSSLSocketFactory(SSL_CONTEXT.getSocketFactory());
            secureConn.setHostnameVerifier(HOSTNAME_VERIFIER);
         }
      try {
         URLConnection conn = ConnectionUtils.openReadConnection(url, username, password);
         InputStream is = conn.getInputStream();
         BufferedReader reader = new BufferedReader(new InputStreamReader(is, CHARSET));
         BufferedReader reader = new BufferedReader(new InputStreamReader(is, ConnectionUtils.CHARSET));
         StringBuilder json = new StringBuilder();
         char[] buffer = new char[4096];
         int len = 0;
@@ -187,6 +188,12 @@
         } else if (e.getMessage().indexOf("403") > -1) {
            // requested url is forbidden by the requesting user
            throw new ForbiddenException(url);
         } else if (e.getMessage().indexOf("405") > -1) {
            // requested url is not allowed by the server
            throw new NotAllowedException(url);
         } else if (e.getMessage().indexOf("501") > -1) {
            // requested url is not recognized by the server
            throw new UnknownRequestException(url);
         }
         throw e;
      }
@@ -221,19 +228,10 @@
   public static int sendJsonString(String url, String json, String username, char[] password)
         throws IOException {
      try {
         byte[] jsonBytes = json.getBytes(CHARSET);
         URL urlObject = new URL(url);
         URLConnection conn = urlObject.openConnection();
         conn.setRequestProperty("Content-Type", "text/plain;charset=" + CHARSET);
         byte[] jsonBytes = json.getBytes(ConnectionUtils.CHARSET);
         URLConnection conn = ConnectionUtils.openConnection(url, username, password);
         conn.setRequestProperty("Content-Type", "text/plain;charset=" + ConnectionUtils.CHARSET);
         conn.setRequestProperty("Content-Length", "" + jsonBytes.length);
         setAuthorization(conn, username, password);
         conn.setUseCaches(false);
         conn.setDoOutput(true);
         if (conn instanceof HttpsURLConnection) {
            HttpsURLConnection secureConn = (HttpsURLConnection) conn;
            secureConn.setSSLSocketFactory(SSL_CONTEXT.getSocketFactory());
            secureConn.setHostnameVerifier(HOSTNAME_VERIFIER);
         }
         // write json body
         OutputStream os = conn.getOutputStream();
@@ -249,48 +247,53 @@
         } else if (e.getMessage().indexOf("403") > -1) {
            // requested url is forbidden by the requesting user
            throw new ForbiddenException(url);
         } else if (e.getMessage().indexOf("405") > -1) {
            // requested url is not allowed by the server
            throw new NotAllowedException(url);
         } else if (e.getMessage().indexOf("501") > -1) {
            // requested url is not recognized by the server
            throw new UnknownRequestException(url);
         }
         throw e;
      }
   }
   private static void setAuthorization(URLConnection conn, String username, char[] password) {
      if (!StringUtils.isEmpty(username) && (password != null && password.length > 0)) {
         conn.setRequestProperty(
               "Authorization",
               "Basic "
                     + Base64.encodeBytes((username + ":" + new String(password)).getBytes()));
      }
   // build custom gson instance with GMT date serializer/deserializer
   // http://code.google.com/p/google-gson/issues/detail?id=281
   private static Gson gson() {
      GsonBuilder builder = new GsonBuilder();
      builder.registerTypeAdapter(Date.class, new GmtDateTypeAdapter());
      builder.setPrettyPrinting();
      return builder.create();
   }
   /**
    * DummyTrustManager trusts all certificates.
    */
   private static class DummyTrustManager implements X509TrustManager {
   private static class GmtDateTypeAdapter implements JsonSerializer<Date>, JsonDeserializer<Date> {
      private final DateFormat dateFormat;
      @Override
      public void checkClientTrusted(X509Certificate[] certs, String authType)
            throws CertificateException {
      private GmtDateTypeAdapter() {
         dateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'", Locale.US);
         dateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
      }
      @Override
      public void checkServerTrusted(X509Certificate[] certs, String authType)
            throws CertificateException {
      public synchronized JsonElement serialize(Date date, Type type,
            JsonSerializationContext jsonSerializationContext) {
         synchronized (dateFormat) {
            String dateFormatAsString = dateFormat.format(date);
            return new JsonPrimitive(dateFormatAsString);
         }
      }
      @Override
      public X509Certificate[] getAcceptedIssuers() {
         return null;
      }
   }
   /**
    * Trusts all hostnames from a certificate, including self-signed certs.
    */
   private static class DummyHostnameVerifier implements HostnameVerifier {
      @Override
      public boolean verify(String hostname, SSLSession session) {
         return true;
      public synchronized Date deserialize(JsonElement jsonElement, Type type,
            JsonDeserializationContext jsonDeserializationContext) {
         try {
            synchronized (dateFormat) {
               return dateFormat.parse(jsonElement.getAsString());
            }
         } catch (ParseException e) {
            throw new JsonSyntaxException(jsonElement.getAsString(), e);
         }
      }
   }
}