Request.java :  » UnTagged » growlforandroid » com » growlforandroid » gntp » Android Open Source

package com.growlforandroid.gntp;

import java.io.*;
import java.net.*;
import java.nio.channels.SocketChannel;
import java.util.*;

import android.util.Log;

import com.growlforandroid.common.*;

public class Request
  extends GntpMessage {
  
  /* private enum RequestState {
    Connected,
    ReadingRequestHeaders,
    ReadingNotificationHeaders,
    EndOfRequest,
    ResponseSent
  }
  
  private RequestState _currentState = RequestState.Connected; */
  
  private RequestType _requestType;
  private EncryptionType _encryptionType;
  private HashAlgorithm _hashAlgorithm;
  private String _initVector = null;
  private String _hash;
  private String _salt;
  
  /*private int _notificationsCount = 0;
  private int _notificationIndex = 0;
  private Map<Integer, Map<String, String>> _notificationsHeaders = new HashMap<Integer, Map<String, String>>(); */
  
  public Request(RequestType type, EncryptionType encryption, HashAlgorithm algorithm, String password) throws GntpException {
    _requestType = type;
    _encryptionType = encryption;
    _hashAlgorithm = algorithm;
    
    if (_encryptionType != EncryptionType.None)
      throw new GntpException(GntpError.NotAuthorized, "Encryption is not yet supported");
    
    // Generate the hash and salt
    _salt = UUID.randomUUID().toString().replace("-", "").toUpperCase();
    byte[] salt = Utility.hexStringToByteArray(_salt);
    byte[] hash = _hashAlgorithm.calculateHash(password, salt);
    _hash = Utility.getHexStringFromByteArray(hash);
  }

  public void sendTo(long connectionID, String hostAndPort) throws GntpException, UnknownHostException, IOException {
    String host = hostAndPort;
    int port = 23053;
    
    int colon = hostAndPort.indexOf(':');
    if (colon >= 0) {
      port = Integer.parseInt(hostAndPort.substring(colon));
      host = hostAndPort.substring(0, colon);
    }
    
    sendTo(connectionID, host, port);
  }
  
  public void sendTo(long connectionID, String host, int port) throws GntpException, UnknownHostException, IOException {
    SocketChannel channel = null;
    try {
      Log.i("Request.sendTo[" + connectionID + "]", "Connecting to " + host + " on port " + port + "...");
      channel = SocketChannel.open();
      Socket socket = channel.socket();
      socket.setSoTimeout(READ_TIMEOUT_MS); // Maximum time to block while waiting for data
      socket.connect(new InetSocketAddress(host, port), CONNECTION_TIMEOUT_MS);
      
      ChannelWriter writer = new ChannelWriter(channel, Constants.CHARSET);
      EncryptedChannelReader reader = new EncryptedChannelReader(channel);
      
      // Write the request line and headers
      Log.i("Request.sendTo[" + connectionID + "]", "Sending " + _requestType.toString() + " request...");
      try {
        write(writer);
      } catch (SocketException se) {
        // We capture this and ignore it, as the socket may have prematurely closed,
        // but we may still have received response (eg: bad password)
        se.printStackTrace();
      }
      
      // Wait for the response
      Log.i("Request.sendTo[" + connectionID + "]", "Waiting for response...");
      Response response = Response.read(reader);
      GntpException error = response.getError();
      if (error != null) {
        Log.i("Request.sendTo[" + connectionID + "]", "Failed: " + error.getMessage());
        throw error;
      }
      Log.i("Request.sendTo[" + connectionID + "]", "Succeeded");
      
    } finally {
      if (channel != null) {
        channel.close();
        channel = null;
      }
      Log.i("Request.sendTo[" + connectionID + "]", "Done");
    }
  }

  protected String getLeaderLine() throws IOException {
    String leaderLine = Constants.GNTP_PROTOCOL_VERSION + " " +
      _requestType.toString() + " " + _encryptionType.toString();
    if (_initVector != null) {
      leaderLine += ":" + _initVector;
    }
    if (_hashAlgorithm != HashAlgorithm.NONE) {
      leaderLine += " " + _hashAlgorithm.toString() + ":" + _hash + "." + _salt;
    }
    return leaderLine;
  }
}