GroupKey Management Invite

injae · Nov 12, 2017 · 7a981db · 7a981db
1 parent 5149944
commit 7a981db
Show file tree

Hide file tree

Showing 3 changed files with 92 additions and 56 deletions.
diff --git a/Cryptonite/src/Server/Server_Delete_Group.java b/Cryptonite/src/Server/Server_Delete_Group.java
@@ -74,6 +74,7 @@ public void running(int count) throws IOException
 					db.Update("update test set mygrouplist = '" + save + "' where uscode = " + uscode + ";");
 				}
 				db.Update("delete from grouplist where gpcode = "+Server_Code_Manager.codeCutter(gpCode)+";");
+				db.Update("delete from groupkey where gpcode = "+Server_Code_Manager.codeCutter(gpCode)+";");
 
 				File forDelete = new File("Server_Folder/Backup/" + gpCode);
 				String[] arrayTemp = forDelete.list();

diff --git a/Cryptonite/src/Server/Server_Group_Invite.java b/Cryptonite/src/Server/Server_Group_Invite.java
@@ -3,126 +3,163 @@
 import java.io.IOException;
 import java.security.KeyFactory;
 import java.security.NoSuchAlgorithmException;
+import java.security.PrivateKey;
 import java.security.PublicKey;
+import java.security.spec.EncodedKeySpec;
 import java.security.spec.InvalidKeySpecException;
+import java.security.spec.PKCS8EncodedKeySpec;
 import java.security.spec.X509EncodedKeySpec;
 import java.sql.ResultSet;
 import java.sql.SQLException;
+import java.util.ArrayList;
 import java.util.Base64;
 import java.util.StringTokenizer;
 
 import javax.crypto.BadPaddingException;
 import javax.crypto.Cipher;
 import javax.crypto.IllegalBlockSizeException;
+import javax.crypto.spec.SecretKeySpec;
 
 import Crypto.Crypto_Factory;
 
-public class Server_Group_Invite extends Server_Funtion 
-{
+public class Server_Group_Invite extends Server_Funtion {
 	private String _id;
 	private String _gpCode;
 	private boolean _passCheck;
-
-	public Server_Group_Invite(Server_Client_Activity activity) 
-	{
+
+	public Server_Group_Invite(Server_Client_Activity activity) {
 		super(activity);
 	}
 
-	private void setInformation(byte[] packet)
-	{
+	private void setInformation(byte[] packet) {
 		int end = 0;
 		byte[] idTemp = new byte[packet[1]];
-		for(int i = 0; i < idTemp.length; i++)
-		{
+		for (int i = 0; i < idTemp.length; i++) {
 			idTemp[i] = packet[i + 3];
 			end = i + 3;
 		}
 		_id = new String(idTemp).trim();
-		
+
 		byte[] gpCodeTemp = new byte[packet[2]];
-		for(int i = 0; i < gpCodeTemp.length; i++)
-		{
+		for (int i = 0; i < gpCodeTemp.length; i++) {
 			gpCodeTemp[i] = packet[i + end + 1];
 		}
 		_gpCode = new String(gpCodeTemp).trim();
 	}
-	
+
 	@Override
-	public void Checker(byte[] packet) 
-	{
+	public void Checker(byte[] packet) {
 		setInformation(packet);
 		_packetMaxCount = 1;
 		_cutSize = 1;
 	}
 
 	@Override
-	public void running(int count) throws IOException 
-	{
-		if(count == 1) 
-		{ 
-			Checker(_activity.getReceiveEvent()); 
-			try 
-			{
+	public void running(int count) throws IOException {
+		if (count == 1) {
+			Checker(_activity.getReceiveEvent());
+			try {
 				_passCheck = true;
 				Server_DataBase db = Server_DataBase.getInstance();
 				ResultSet rs = db.Query("select * from test where id = '" + _id + "';");
 				rs.next();
 				String uscode = "@" + rs.getString(6);
-
-				ResultSet rs2 = db.Query("select * from grouplist where gpcode = " + Server_Code_Manager.codeCutter(_gpCode) + ";");
+
+				ResultSet rs2 = db.Query(
+						"select * from grouplist where gpcode = " + Server_Code_Manager.codeCutter(_gpCode) + ";");
 				rs2.next();
 				String gplist = rs2.getString(2);
 				StringTokenizer st = new StringTokenizer(gplist, ":");
-				while(st.hasMoreTokens())
-				{
-					if(st.nextToken().equals(uscode))
-					{
+				while (st.hasMoreTokens()) {
+					if (st.nextToken().equals(uscode)) {
 						_passCheck = false;
 					}
 				}
-				if(_passCheck)
-				{
+				if (_passCheck) {
 					gplist += ":" + uscode;
-					db.Update("update grouplist set gplist = '" + gplist + "' where gpcode = " + Server_Code_Manager.codeCutter(_gpCode) + ";");
-
+					db.Update("update grouplist set gplist = '" + gplist + "' where gpcode = "
+							+ Server_Code_Manager.codeCutter(_gpCode) + ";");
+
 					ResultSet rs3 = db.Query("select * from test where id = '" + _id + "';");
 					rs3.next();
 					String mygrouplist = rs3.getString(10);
-					if(mygrouplist.equals("NULL"))
-					{
+					String pk = rs3.getString(14);
+					byte[] Pk = Base64.getDecoder().decode(pk);
+
+					if (mygrouplist.equals("NULL")) {
 						mygrouplist = _gpCode;
-					}
-					else
-					{
+					} else {
 						mygrouplist += ":" + _gpCode;
 					}
-					db.Update("update test set mygrouplist = '" + mygrouplist + "' where uscode = " + Server_Code_Manager.codeCutter(uscode) + ";");
-
+
+					db.Update("update test set mygrouplist = '" + mygrouplist + "' where uscode = "
+							+ Server_Code_Manager.codeCutter(uscode) + ";");
+
 					ResultSet rs4 = db.Query("select * from test where id = '" + _id + "';");
 					rs4.next();
 					String pubkey = rs4.getString(14);
 					byte[] pubKey = Base64.getDecoder().decode(pubkey);
 
-					PublicKey pk = KeyFactory.getInstance("RSA").generatePublic(new X509EncodedKeySpec(pubKey));
-
-					ResultSet rs5 = db.Query("select * from grouplist where gpcode = " + Server_Code_Manager.codeCutter(_gpCode) + ";");
+					ResultSet rs5 = db.Query("Select secretkey from test where uscode = '" + Server_Code_Manager.codeCutter(_activity.getClientCode()) + "';");
 					rs5.next();
+					String sk = rs5.getString(1);
+					byte[] Sk = Base64.getDecoder().decode(sk);
+
+					EncodedKeySpec privateKeySpec = new PKCS8EncodedKeySpec(Sk);
+			        KeyFactory generator = KeyFactory.getInstance("RSA");
+
+			        PublicKey PK = KeyFactory.getInstance("RSA").generatePublic(new X509EncodedKeySpec(pubKey));	//초대받는사람 공개키
+			        PrivateKey privateKey = generator.generatePrivate(privateKeySpec);								//초대하는사람 비밀키
+
+					ResultSet rs6 = db.Query("Select groupkeynum, groupkey from groupkey where uscode = '"
+							+ Server_Code_Manager.codeCutter(_activity.getClientCode()) + "' and gpcode = '"
+							+ Server_Code_Manager.codeCutter(_gpCode) + "';");
 
-					byte[] groupkey = Crypto_Factory.create("RSA1024", Cipher.ENCRYPT_MODE, pk).doFinal(Base64.getDecoder().decode(rs5.getString(4)));
+					ArrayList<Integer> gknumList = new ArrayList<>();
+					ArrayList<String> GroupkeyList = new ArrayList<>(); 
 
-					//System.out.println();
-					db.Update("INSERT INTO groupkey values('"+ Server_Code_Manager.codeCutter(_gpCode) +"'," + rs5.getInt(11) + ",'" + Server_Code_Manager.codeCutter(uscode) + "','" + Base64.getEncoder().encodeToString(groupkey) + "')");
+					while(rs6.next())
+					{
+						int gknum = rs6.getInt(1);
+						String gk = rs6.getString(2);
+						byte[] Gk = Base64.getDecoder().decode(gk);
+
+						byte[] groupkey = Crypto_Factory.create("RSA1024", Cipher.DECRYPT_MODE, privateKey).doFinal(Gk);
+						groupkey = Crypto_Factory.create("RSA1024", Cipher.ENCRYPT_MODE, PK).doFinal(groupkey);
+						String Groupkey = Base64.getEncoder().encodeToString(groupkey);
+
+						gknumList.add(gknum);
+						GroupkeyList.add(Groupkey);
+
+					}
 
+					for (int i=0; i<gknumList.size();i++)
+					{
+						db.Update("Insert into groupkey values ('" + Server_Code_Manager.codeCutter(_gpCode) + "','" + gknumList.get(i) + "','" + Server_Code_Manager.codeCutter(uscode) + "','" + GroupkeyList.get(i) + "');");
+					}
 
+
+					/*
+					 * PublicKey pk = KeyFactory.getInstance("RSA").generatePublic(new
+					 * X509EncodedKeySpec(pubKey));
+					 * 
+					 * ResultSet rs5 = db.Query("select * from grouplist where gpcode = " +
+					 * Server_Code_Manager.codeCutter(_gpCode) + ";"); rs5.next();
+					 * 
+					 * byte[] groupkey = Crypto_Factory.create("RSA1024", Cipher.ENCRYPT_MODE,
+					 * pk).doFinal(Base64.getDecoder().decode(rs5.getString(4)));
+					 * 
+					 * //System.out.println(); db.Update("INSERT INTO groupkey values('"+
+					 * Server_Code_Manager.codeCutter(_gpCode) +"'," + rs5.getInt(11) + ",'" +
+					 * Server_Code_Manager.codeCutter(uscode) + "','" +
+					 * Base64.getEncoder().encodeToString(groupkey) + "')");
+					 */
+
 					_activity.send.setPacket("TRUE".getBytes(), 100).write();
-				}
-				else
-				{
+				} else {
 					_activity.send.setPacket("FALSE".getBytes(), 100).write();
 				}
-			} 
-			catch (SQLException e) 
-			{
+			} catch (SQLException e) {
 				e.printStackTrace();
 			} catch (InvalidKeySpecException e) {
 				// TODO Auto-generated catch block
@@ -137,10 +174,8 @@ public void running(int count) throws IOException
 				// TODO Auto-generated catch block
 				e.printStackTrace();
 			}
-		}
-		else
-		{
-
+		} else {
+
 		}
 	}
 }
diff --git a/Cryptonite/src/Server/Server_Group_Withdrawal.java b/Cryptonite/src/Server/Server_Group_Withdrawal.java
@@ -57,7 +57,7 @@ public void running(int count) throws IOException {
 			if(save.length() == 0)
 			{
 				db.Update("delete from grouplist where gpcode = "+Server_Code_Manager.codeCutter(_gpcode)+";"); 
-				db.Update("Delete from groupkey where gpcode = "+_gpcode+";"); 
+				db.Update("Delete from groupkey where gpcode = "+Server_Code_Manager.codeCutter(_gpcode)+";"); 
 			}
 			else 
 			{