P2PTesting.java

/**
 * Copyright (C) 2011 Adriano Monteiro Marques
 *
 * Author:  Zubair Nabi <zn.zubairnabi@gmail.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

package org.umit.icm.mobile.p2p;

import java.io.IOException;

import org.umit.icm.mobile.process.Globals;
import org.umit.icm.mobile.proto.MessageProtos.AgentData;
import org.umit.icm.mobile.proto.MessageProtos.AuthenticatePeer;
import org.umit.icm.mobile.proto.MessageProtos.AuthenticatePeerResponse;
import org.umit.icm.mobile.proto.MessageProtos.P2PGetPeerList;
import org.umit.icm.mobile.proto.MessageProtos.P2PGetPeerListResponse;
import org.umit.icm.mobile.proto.MessageProtos.P2PGetSuperPeerList;
import org.umit.icm.mobile.proto.MessageProtos.P2PGetSuperPeerListResponse;
import org.umit.icm.mobile.proto.MessageProtos.RSAKey;

import android.util.Log;

public class P2PTesting {
	public static void testRequestResponse() throws IOException {
		String ip = "202.206.64.11";
		int port = 3128;
		Globals.tcpClient.openConnection(ip, port);		
		Globals.tcpClient.writeLine(MessageBuilder.generateMessage(
				getTestMessage().getMessageID(), getTestMessage().getMessage()));
		/*read total length*/
		byte [] response = Globals.tcpClient.readBytes(4);
		Log.w("###Len ResponseSize: ", Integer.toString(response.length));	
		if(response.length!=0) {
			byte [] sizeBytes = MessageBuilder.getSubArray(response, 0, 3);
			Log.w("###Len sizeBytes: ", Integer.toString(sizeBytes.length));									
			int size = MessageBuilder.byteArrayToInt(sizeBytes);						
			Log.w("###size: ", Integer.toString(size));				
			
			/*read actual message now*/
			response = Globals.tcpClient.readBytes(size);
			Log.w("###Len ResponseMessage: ", Integer.toString(response.length));	
			if(response.length!=0) {
				byte[] idBytes = MessageBuilder.getSubArray(response, 0, 3);
				Log.w("###Len idBytes: ", Integer.toString(idBytes.length));																
				int id = MessageBuilder.byteArrayToInt(idBytes);				
				Log.w("###id ", Integer.toString(id));
				byte [] msg = MessageBuilder.getSubArray(response, 4, size - 1);
				
				AuthenticatePeerResponse authenticatePeerResponse
				= AuthenticatePeerResponse.parseFrom(msg);
				Log.w("###msg ", authenticatePeerResponse.getCipheredPublicKey().getExp());
				
				} else {
					Log.w("### ", "Blank response");
				}					
	        			
			} else {
				Log.w("### ", "Blank response");
		}
		Globals.tcpClient.closeConnection();
	}
	
	private static QueueObject getTestMessage() {	
		RSAKey rsaKey = RSAKey.newBuilder()
		.setExp("exp")
		.setMod("mod")
		.build();
		AuthenticatePeer authenticatePeer = AuthenticatePeer.newBuilder()
		.setAgentID(Integer.toString(10))
		.setAgentType(3)
		.setAgentPort(8000)		
		.setCipheredPublicKey(rsaKey)
		.build();
		
		AgentData agentData = AgentData.newBuilder()
		.setAgentID(Integer.toString(10))
		.setAgentIP("")
		.setAgentPort(20)
		.setPeerStatus("On")
		.setPublicKey(rsaKey)
		.setToken("token")
		.build();
		
		return new QueueObject(agentData, authenticatePeer.toByteArray(), MessageID.AuthenticatePeer);		
	}
	
	public static void testRequestResponse2() throws IOException {
		String ip = "202.206.64.11";
		int port = 3128;
		Globals.tcpClient.openConnection(ip, port);		
		Globals.tcpClient.writeLine(MessageBuilder.generateMessage(
				getTestMessage2().getMessageID(), getTestMessage2().getMessage()));
		/*read total length*/
		byte [] response = Globals.tcpClient.readBytes(4);
		Log.w("###Len ResponseSize: ", Integer.toString(response.length));	
		if(response.length!=0) {
			byte [] sizeBytes = MessageBuilder.getSubArray(response, 0, 3);
			Log.w("###Len sizeBytes: ", Integer.toString(sizeBytes.length));									
			int size = MessageBuilder.byteArrayToInt(sizeBytes);						
			Log.w("###size: ", Integer.toString(size));				
			
			/*read actual message now*/
			response = Globals.tcpClient.readBytes(size);
			Log.w("###Len ResponseMessage: ", Integer.toString(response.length));	
			if(response.length!=0) {
				byte[] idBytes = MessageBuilder.getSubArray(response, 0, 3);
				Log.w("###Len idBytes: ", Integer.toString(idBytes.length));																
				int id = MessageBuilder.byteArrayToInt(idBytes);				
				Log.w("###id ", Integer.toString(id));
				byte [] msg = MessageBuilder.getSubArray(response, 4, size - 1);
				
				P2PGetPeerListResponse p2pGetPeerListResponse
				= P2PGetPeerListResponse.parseFrom(msg);
				Log.w("###msg ", Integer.toString(p2pGetPeerListResponse.getPeersCount()));
				
				} else {
					Log.w("### ", "Blank response");
				}					
	        			
			} else {
				Log.w("### ", "Blank response");
		}
		Globals.tcpClient.closeConnection();
	}
	
	private static QueueObject getTestMessage2() {		
		P2PGetPeerList p2pGetPeerList = P2PGetPeerList.newBuilder()
		.setCount(10)
		.build();
		
		RSAKey rsaKey = RSAKey.newBuilder()
		.setExp("exp")
		.setMod("mod")
		.build();
		
		AgentData agentData = AgentData.newBuilder()
		.setAgentID(Integer.toString(10))
		.setAgentIP("")
		.setAgentPort(20)
		.setPeerStatus("On")
		.setPublicKey(rsaKey)
		.setToken("token")
		.build();
		
		return new QueueObject(agentData, p2pGetPeerList.toByteArray(), MessageID.P2PGetPeerList);		
	}
	
	public static void testRequestResponse3() throws IOException {
		String ip = "202.206.64.11";
		int port = 3128;
		Globals.tcpClient.openConnection(ip, port);		
		Globals.tcpClient.writeLine(MessageBuilder.generateMessage(
				getTestMessage3().getMessageID(), getTestMessage3().getMessage()));
		/*read total length*/
		byte [] response = Globals.tcpClient.readBytes(4);
		Log.w("###Len ResponseSize: ", Integer.toString(response.length));	
		if(response.length!=0) {
			byte [] sizeBytes = MessageBuilder.getSubArray(response, 0, 3);
			Log.w("###Len sizeBytes: ", Integer.toString(sizeBytes.length));									
			int size = MessageBuilder.byteArrayToInt(sizeBytes);						
			Log.w("###size: ", Integer.toString(size));				
			
			/*read actual message now*/
			response = Globals.tcpClient.readBytes(size);
			Log.w("###Len ResponseMessage: ", Integer.toString(response.length));	
			if(response.length!=0) {
				byte[] idBytes = MessageBuilder.getSubArray(response, 0, 3);
				Log.w("###Len idBytes: ", Integer.toString(idBytes.length));																
				int id = MessageBuilder.byteArrayToInt(idBytes);				
				Log.w("###id ", Integer.toString(id));
				byte [] msg = MessageBuilder.getSubArray(response, 4, size - 1);
				
				P2PGetSuperPeerListResponse p2pGetPeerListResponse
				= P2PGetSuperPeerListResponse.parseFrom(msg);
				Log.w("###msg ", Integer.toString(p2pGetPeerListResponse.getPeersCount()));
				
				} else {
					Log.w("### ", "Blank response");
				}					
	        			
			} else {
				Log.w("### ", "Blank response");
		}
		Globals.tcpClient.closeConnection();
	}
	
	private static QueueObject getTestMessage3() {		
		P2PGetSuperPeerList p2pGetPeerList = P2PGetSuperPeerList.newBuilder()
		.setCount(10)
		.build();
		
		RSAKey rsaKey = RSAKey.newBuilder()
		.setExp("exp")
		.setMod("mod")
		.build();
		
		AgentData agentData = AgentData.newBuilder()
		.setAgentID(Integer.toString(10))
		.setAgentIP("")
		.setAgentPort(20)
		.setPeerStatus("On")
		.setPublicKey(rsaKey)
		.setToken("token")
		.build();
		
		return new QueueObject(agentData, p2pGetPeerList.toByteArray(), MessageID.P2PGetSuperPeerList);		
	}
}