Homec4science
Diffusion Genome Privacy - SHCS App (master) Browse

genopri-shcs/project/PPPCertifiedInstitution/src/io/ReaderCF.java
5bc09aefc5dcmaster

Last Change
Raw File

ReaderCF.java
View Options

package io;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.math.BigInteger;
import java.sql.SQLException;
import java.util.Date;
import java.util.HashMap;
import javax.swing.JTextPane;
import javax.swing.text.StyledDocument;
import org.bouncycastle.asn1.nist.NISTNamedCurves;
import org.bouncycastle.asn1.x9.X9ECParameters;
import org.bouncycastle.math.ec.ECPoint;
import utils.CertifiedInstitutionProperties;
import utils.ConfigProperties;
import utils.Validator;
import crypto.elgamal.ECElGamal;
import crypto.elgamal.fragment.FragmentEncrypted;
import database.DBconnectorClient;
import database.DBconnectorServer;
public class ReaderCF implements Runnable {
private String path;
private DBconnectorClient conn_mu;
private DBconnectorServer conn_s;
private StyledDocument doc;
private HashMap<String, Integer> attributes;
private String test;
private X9ECParameters ecParams;
public ReaderCF(CertifiedInstitutionProperties props, String csv_filepath, JTextPane txp, String test){
path = csv_filepath;
conn_mu = new DBconnectorClient(props.getConfigMU());
conn_s = new DBconnectorServer(props.getConfigSPU());
doc = txp.getStyledDocument();
attributes = new HashMap<>();
this.test = test;
this.ecParams = NISTNamedCurves.getByName(props.getProperty(ConfigProperties.KEY_CURVE));
}
@Override
public void run() {
try{
//check extension file
if(!Validator.checkExtension(path, "csv")) {
throw new IOException("Wrong Format for CSV File!");
}
StyledDocumentWriter.appendMessage(doc, "Analysing CSV File, please wait..." + "\n\n");
// open connection to SPU and MU
conn_s.openConnection();
conn_mu.openConnection();
// open input stream
BufferedReader br = new BufferedReader(new FileReader(new File(path)));
long start = System.currentTimeMillis();
switch(test){
case "Coronary Artery Disease":
int i = 0;
String line;
while((line = br.readLine()) != null ){
i++;
String[] tokens = line.split(";");
if(i == 1){
for(int j = 0; j<tokens.length;j++){
attributes.put(tokens[j],j);
}
}else{
int patient = Integer.parseInt(tokens[0]);
StyledDocumentWriter.appendMessage(doc, "Encrypting clinical factors for patient " + patient + "\n");
ECPoint pPubKey = conn_s.getPublicKey(patient, ecParams);
ECElGamal ecceg = new ECElGamal(ecParams, pPubKey);
// Current Smoking
processCurrentSmoking(patient, tokens, ecceg);
// Age
processAge(patient, tokens, ecceg);
// Family history of CAD
processCAD(patient, tokens, ecceg);
//Past Smoking
processPastSmoking(patient, tokens, ecceg);
//Hyper-cholesterolemia
processHyperChol(patient, tokens, ecceg);
//Hypertension
processHypertension(patient, tokens, ecceg);
}
}
break;
}
// close all connections
br.close();
conn_s.closeConnection();
conn_mu.closeConnection();
long end = System.currentTimeMillis();
StyledDocumentWriter.appendMessage(doc, "Parsing CSV File and encryption, completed!" + "\n");
StyledDocumentWriter.appendMessage(doc, (end-start) + " ms\n");
}catch(IOException | NullPointerException e ){
ErrorHandler.handle(this, e, doc);
} catch (SQLException e) {
ErrorHandler.handle(this, e, doc, "No Connection with the Server");
}
}
private void processCurrentSmoking(int patient, String[] tokens, ECElGamal ecceg) throws SQLException {
BigInteger plain = new BigInteger(tokens[attributes.get("SMOKE")]);
FragmentEncrypted cipher = ecceg.encrypt(plain);
conn_s.insertEncryptedCF(patient, conn_mu.getCF("Current smoking"), cipher.getFragmentEncoded(), null);
}
private void processAge(int patient, String[] tokens, ECElGamal ecceg) throws SQLException {
String birthdate = tokens[attributes.get("BIRTHDATE")];
String[] date = birthdate.split("/");
Date today = new Date();
@SuppressWarnings("deprecation")
BigInteger plain = BigInteger.valueOf(1900 + today.getYear() - Integer.parseInt(date[2]));
FragmentEncrypted cipher = ecceg.encrypt(plain);
conn_s.insertEncryptedCF(patient, conn_mu.getCF("Age"), cipher.getFragmentEncoded(), null);
}
private void processCAD(int patient, String[] tokens, ECElGamal ecceg) throws SQLException {
String value = tokens[attributes.get("CV_POS_FAMILY")];
BigInteger plain;
if(Integer.parseInt(value) == 0)
plain = BigInteger.ZERO;
else
plain = BigInteger.ONE;
FragmentEncrypted cipher = ecceg.encrypt(plain);
conn_s.insertEncryptedCF(patient, conn_mu.getCF("Family history of CAD"), cipher.getFragmentEncoded(), null);
}
private void processPastSmoking(int patient, String[] tokens,
ECElGamal ecceg) throws SQLException {
BigInteger plain;
if(tokens[attributes.get("PAST_SMOKE")].equals("1"))
plain = BigInteger.ONE;
else
plain = BigInteger.ZERO;
FragmentEncrypted cipher = ecceg.encrypt(plain);
conn_s.insertEncryptedCF(patient, conn_mu.getCF("Past smoking"), cipher.getFragmentEncoded(), null);
}
private void processHyperChol(int patient, String[] tokens, ECElGamal ecceg) throws SQLException {
double chol = Double.parseDouble(tokens[attributes.get("CHOL")]);
BigInteger plain;
if(chol > 6.2d)
plain = BigInteger.ONE;
else
plain = BigInteger.ZERO;
FragmentEncrypted cipher = ecceg.encrypt(plain);
conn_s.insertEncryptedCF(patient, conn_mu.getCF("Hyper-cholesterolemia"), cipher.getFragmentEncoded(), null);
}
private void processHypertension(int patient, String[] tokens,
ECElGamal ecceg) throws SQLException {
int diastolic = Integer.parseInt(tokens[attributes.get("DIASTOLIC")]);
int systolic = Integer.parseInt(tokens[attributes.get("SYSTOLIC")]);
BigInteger plain;
if(diastolic >= 90 || systolic >= 140 || !tokens[attributes.get("CURRENT_ART")].equals(""))
plain = BigInteger.ONE;
else
plain = BigInteger.ZERO;
FragmentEncrypted cipher = ecceg.encrypt(plain);
conn_s.insertEncryptedCF(patient, conn_mu.getCF("Hypertension"), cipher.getFragmentEncoded(), null);
}
}
c4science · Help