add tools/type_store.cc

tools/type_store.cc

+#include "wrapper/common.hh"
+#include "wrapper/reuse.hh"
+#include "util/util.hh"
+#include "util/constants.hh"
+static std::string embeddedDir="/t/cryt/shadow";
+//global map, for each client, we have one WrapperState which contains ProxyState.
+static std::map<std::string, WrapperState*> clients;
+//This connection mimics the behaviour of MySQL-Proxy
+Connect  *globalConn;
+
+static void init(std::string ip,int port){
+    std::string client="192.168.1.1:1234";
+    //one Wrapper per user.
+    clients[client] = new WrapperState();    
+    //Connect phase
+    ConnectionInfo ci("localhost", "root", "letmein",port);
+    const std::string master_key = "113341234";
+    char *buffer;
+    if((buffer = getcwd(NULL, 0)) == NULL){  
+        perror("getcwd error");  
+    }
+    embeddedDir = std::string(buffer)+"/shadow";
+    SharedProxyState *shared_ps = 
+			new SharedProxyState(ci, embeddedDir , master_key, 
+                                            determineSecurityRating());
+    assert(0 == mysql_thread_init());
+    //we init embedded database here.
+    clients[client]->ps = std::unique_ptr<ProxyState>(new ProxyState(*shared_ps));
+    clients[client]->ps->safeCreateEmbeddedTHD();
+    //Connect end!!
+    globalConn = new Connect(ip, ci.user, ci.passwd, port);
+}
+
+
+
+static void 
+write_meta_new(std::vector<FieldMetaTrans> &res,string db,string table){
+    std::string filename = std::string("data/") +db+"/"+table+"/metadata.data";
+    TableMetaTrans mf(db,table,res);
+    mf.set_db_table(db,table);
+    mf.serializeNew(filename);
+}
+
+static
+std::string
+getSelectField(SchemaInfo &schema, FieldMetaTrans &tf,std::string db,std::string table){
+    std::string res = "SELECT ";
+    const std::unique_ptr<IdentityMetaKey> dbmeta_key(new IdentityMetaKey(db));
+    //get databaseMeta, search in the map
+    DatabaseMeta * dbm = schema.getChild(*dbmeta_key);
+    const TableMeta & tbm = *((*dbm).getChild(IdentityMetaKey(table)));
+    std::string annotablename = tbm.getAnonTableName();
+
+    //then a list of onion names
+    for(auto item : tf.getChoosenOnionName()){
+        res += item;
+        res += " , ";
+    }
+    //actually the salt should be selected if RND is used,this should be changed later.
+    if(tf.getHasSalt()){
+        res += tf.getSaltName() + " , ";
+    }
+    res = res.substr(0,res.size()-2);
+    res = res + "FROM `"+db+std::string("`.`")+annotablename+"`";
+    return res;
+}
+
+static
+void
+write_field_data_to_files(MySQLColumnData& resraw, FieldMetaTrans &res, string db, string table,string field) {
+    std::string prefix = std::string("data/") +db+"/"+table+"/"+field+"/";
+    g_make_path(prefix);
+    std::vector<std::string> filenames;
+    for(auto item:resraw.fieldNames){
+        item=prefix+item;
+        filenames.push_back(item);
+    }
+    int len = resraw.fieldNames.size();
+    for(int i=0;i<len;i++){
+        if(IS_NUM(resraw.fieldTypes[i])){
+            filenames[i]+="NUMF/";
+            writeColumndataNum(resraw.columnData[i],filenames[i]);
+        }else{
+            filenames[i]+="STRF/";
+            writeColumndataEscapeString(resraw.columnData[i],filenames[i],resraw.maxLengths[i]);
+        }
+    }
+}
+
+static void store(std::string db, std::string table){
+    std::unique_ptr<SchemaInfo> schema =  myLoadSchemaInfo(embeddedDir);
+    //get all the fields in the tables
+    std::vector<FieldMeta*> fms = getFieldMeta(*schema,db,table);
+    //transform the field so that selected onions can be used
+    std::vector<FieldMetaTrans> res;
+    for(auto i=0u;i<fms.size();i++){
+        FieldMetaTrans ft;
+        res.push_back(ft);
+        res.back().trans(fms[i]);
+    }
+    /*this is our strategy, each field should be able to choose the selected onion*/
+    storeStrategyNew(res);
+    for(auto &tf:res){
+        std::string back_field_query = getSelectField(*schema,tf,db,table);
+        std::cout<<back_field_query<<std::endl;
+        MySQLColumnData resraw =  executeAndGetColumnData(globalConn,back_field_query);
+        std::vector<int> types;
+        for(unsigned int i=0u;i<resraw.fieldTypes.size();i++){
+            types.push_back((int)(resraw.fieldTypes[i]));
+        }
+        auto lengths = resraw.maxLengths;
+        auto stype = types.back();
+        auto slength = lengths.back();
+        types.pop_back();
+        lengths.pop_back();
+        tf.setChoosenFieldTypes(types);
+        tf.setChoosenFieldLengths(lengths);
+        tf.setSaltType(stype);
+        tf.setSaltLength(slength);
+        write_field_data_to_files(resraw,tf,db,table,tf.getOriginalFieldMeta()->getFieldName());
+    }
+    write_meta_new(res,db,table);
+    //generate the backup query and then fetch the tuples
+}
+
+int
+main(int argc, char* argv[]){    
+    std::string db="tdb",table="student";
+    std::string ip="127.0.0.1";
+    int port=3306;
+    if(argc==4){
+        ip = std::string(argv[1]);
+        db = std::string(argv[2]);
+        table = std::string(argv[3]);
+    }
+    init(ip,port);
+
+    store(db,table);
+    return 0;
+}