Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
P
Practical-Cryptdb
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Zhaozhen
Practical-Cryptdb
Commits
a1ae02da
Commit
a1ae02da
authored
Jan 26, 2018
by
yiwenshao
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
able to use pipeline
parent
ec3617d5
Expand all
Show whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
490 additions
and
98 deletions
+490
-98
final_load.cc
debug/final_load.cc
+69
-6
final_load.cc.back
debug/final_load.cc.back
+8
-9
final_load_low_memory.cc
debug/final_load_low_memory.cc
+59
-70
final_load_low_memory.cc.back
debug/final_load_low_memory.cc.back
+312
-0
final_store.cc.back
debug/final_store.cc.back
+20
-6
verify_timer.cc
test_util/verify_timer.cc
+14
-0
timer.hh
util/timer.hh
+8
-7
No files found.
debug/final_load.cc
View file @
a1ae02da
...
@@ -202,7 +202,7 @@ static ResType load_files(std::string db, std::string table){
...
@@ -202,7 +202,7 @@ static ResType load_files(std::string db, std::string table){
unsigned
long
gcount
=
0
;
unsigned
long
gcount
=
0
;
static
static
void
local_wrapper
(
const
Item
&
i
,
const
FieldMeta
&
fm
,
Analysis
&
a
,
void
local_wrapper
(
const
Item
&
i
,
const
FieldMeta
&
fm
,
Analysis
&
a
,
List
<
Item
>
*
const
append_list
)
{
List
<
Item
>
*
append_list
)
{
//append_list->push_back(&(const_cast<Item&>(i)));
//append_list->push_back(&(const_cast<Item&>(i)));
//do not use the plain strategy
//do not use the plain strategy
std
::
vector
<
Item
*>
l
;
std
::
vector
<
Item
*>
l
;
...
@@ -252,6 +252,21 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
...
@@ -252,6 +252,21 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
}
}
}
}
static
List
<
Item
>
*
processRow
(
const
std
::
vector
<
Item
*>
&
row
,
const
std
::
vector
<
std
::
string
>
&
names
,
Analysis
&
analysis
,
std
::
string
db
,
std
::
string
table
)
{
List
<
Item
>
*
const
newList0
=
new
List
<
Item
>
();
for
(
auto
i
=
0u
;
i
<
names
.
size
();
i
++
){
std
::
string
field_name
=
names
[
i
];
FieldMeta
&
fm
=
analysis
.
getFieldMeta
(
db
,
table
,
field_name
);
local_wrapper
(
*
row
[
i
],
fm
,
analysis
,
newList0
);
}
return
newList0
;
}
int
int
main
(
int
argc
,
char
*
argv
[]){
main
(
int
argc
,
char
*
argv
[]){
...
@@ -277,21 +292,69 @@ main(int argc, char* argv[]){
...
@@ -277,21 +292,69 @@ main(int argc, char* argv[]){
const
std
::
string
head
=
std
::
string
(
"INSERT INTO `"
)
+
db
+
"`.`"
+
annoTableName
+
"` "
;
const
std
::
string
head
=
std
::
string
(
"INSERT INTO `"
)
+
db
+
"`.`"
+
annoTableName
+
"` "
;
/*reencryption to get the encrypted insert!!!*/
/*reencryption to get the encrypted insert!!!*/
for
(
auto
&
row
:
res
.
rows
)
{
/* int currentCount=0;
List<List_item> newList;
List<List_item> newList;
List
<
Item
>
*
const
newList0
=
new
List
<
Item
>
();
for(auto &row:res.rows) {
for
(
auto
i
=
0u
;
i
<
res
.
names
.
size
();
i
++
){
List<Item> * newList0 = new List<Item>();
for(auto i=0u;i<res.names.size();i++) {
std::string field_name = res.names[i];
std::string field_name = res.names[i];
FieldMeta & fm = analysis.getFieldMeta(db,table,field_name);
FieldMeta & fm = analysis.getFieldMeta(db,table,field_name);
local_wrapper(*row[i],fm,analysis,newList0);
local_wrapper(*row[i],fm,analysis,newList0);
}
}
newList.push_back(newList0);
newList.push_back(newList0);
currentCount++;
if(currentCount==constGlobalConstants.pipelineCount){
currentCount=0;
std::ostringstream o;
std::ostringstream o;
insertManyValues(o,newList);
insertManyValues(o,newList);
std::cout<<(head+o.str())<<std::endl;
std::cout<<(head+o.str())<<std::endl;
//newList = List<List_item>();
}
}
} */
/* if(currentCount!=0){
std::ostringstream o;
insertManyValues(o,newList);
std::cout<<(head+o.str())<<std::endl;
}*/
std
::
cout
<<
"gcount<<"
<<
gcount
<<
std
::
endl
;
/* for(auto &row:res.rows) {
List<List_item> newList;
List<Item> * newList0 = processRow(row,res.names,analysis,db,table);
newList.push_back(newList0);
std::ostringstream o;
insertManyValues(o,newList);
std::cout<<(head+o.str())<<std::endl;
}
*/
unsigned
int
i
=
0u
;
while
(
true
){
List
<
List_item
>
newList
;
int
localCount
=
0
;
for
(;
i
<
res
.
rows
.
size
();
i
++
){
List
<
Item
>
*
newList0
=
processRow
(
res
.
rows
[
i
],
res
.
names
,
analysis
,
db
,
table
);
newList
.
push_back
(
newList0
);
localCount
++
;
if
(
localCount
==
constGlobalConstants
.
pipelineCount
){
std
::
ostringstream
o
;
insertManyValues
(
o
,
newList
);
std
::
cout
<<
(
head
+
o
.
str
())
<<
std
::
endl
;
i
++
;
break
;
}
}
if
(
i
>=
res
.
rows
.
size
()){
if
(
localCount
!=
constGlobalConstants
.
pipelineCount
)
{
std
::
ostringstream
o
;
insertManyValues
(
o
,
newList
);
std
::
cout
<<
(
head
+
o
.
str
())
<<
std
::
endl
;
}
break
;
}
}
// std::cout<<"gcount<<"<<gcount<<std::endl;
return
0
;
return
0
;
}
}
debug/final_load.cc.back
View file @
a1ae02da
...
@@ -131,16 +131,18 @@ void initGfb(std::vector<FieldMetaTrans> &res,std::string db,std::string table){
...
@@ -131,16 +131,18 @@ void initGfb(std::vector<FieldMetaTrans> &res,std::string db,std::string table){
gfb.field_names = field_names;
gfb.field_names = field_names;
gfb.field_types = field_types;
gfb.field_types = field_types;
gfb.field_lengths = field_lengths;
gfb.field_lengths = field_lengths;
//then we should read the vector
//then we should read the vector
std::string prefix = std::string("data/")+db+"/"+table+"/";
std::string prefix = std::string("data/")+db+"/"+table+"/";
for(unsigned int i=0u; i<gfb.field_names.size(); i++) {
for(unsigned int i=0u; i<gfb.field_names.size(); i++) {
std::string filename = prefix + gfb.field_names[i];
std::string filename = prefix + gfb.field_names[i];
std::vector<std::string> column;
std::vector<std::string> column;
if(IS_NUM(gfb.field_types[i])){
if(IS_NUM(gfb.field_types[i])){
load_num_file_c
ount(filename,column,constGlobalConstants.loadCount);
loadFileNoEscapeLimitC
ount(filename,column,constGlobalConstants.loadCount);
}else{
}else{
load_string_file_c
ount(filename,column,gfb.field_lengths[i],constGlobalConstants.loadCount);
loadFileEscapeLimitC
ount(filename,column,gfb.field_lengths[i],constGlobalConstants.loadCount);
}
}
std::reverse(column.begin(),column.end());
gfb.annoOnionNameToFileVector[gfb.field_names[i]] = std::move(column);
gfb.annoOnionNameToFileVector[gfb.field_names[i]] = std::move(column);
}
}
//init another map
//init another map
...
@@ -171,8 +173,6 @@ static ResType load_files(std::string db, std::string table){
...
@@ -171,8 +173,6 @@ static ResType load_files(std::string db, std::string table){
//why do we need this??
//why do we need this??
create_embedded_thd(0);
create_embedded_thd(0);
rawMySQLReturnValue resraw;
rawMySQLReturnValue resraw;
// vector<vector<string>> resss_field = loadTableFieldsForDecryption(db,
// table,field_names, field_types, field_lengths);
vector<vector<string>> res_field;
vector<vector<string>> res_field;
for(auto item:field_names){
for(auto item:field_names){
res_field.push_back(gfb.annoOnionNameToFileVector[item]);
res_field.push_back(gfb.annoOnionNameToFileVector[item]);
...
@@ -199,6 +199,7 @@ static ResType load_files(std::string db, std::string table){
...
@@ -199,6 +199,7 @@ static ResType load_files(std::string db, std::string table){
return finalresults;
return finalresults;
}
}
unsigned long gcount=0;
static
static
void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
List<Item> *const append_list) {
List<Item> *const append_list) {
...
@@ -216,7 +217,6 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
...
@@ -216,7 +217,6 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
std::vector<std::string> &tempFileVector = gfb.annoOnionNameToFileVector[annoOnionName];
std::vector<std::string> &tempFileVector = gfb.annoOnionNameToFileVector[annoOnionName];
std::string in = tempFileVector.back();
std::string in = tempFileVector.back();
if(IS_NUM(type)){
if(IS_NUM(type)){
//std::string in("11");
unsigned int len = annoOnionName.size();
unsigned int len = annoOnionName.size();
if(len>4u&&annoOnionName.substr(len-4)=="ASHE"){
if(len>4u&&annoOnionName.substr(len-4)=="ASHE"){
l.push_back(MySQLFieldTypeToItem(type,in));
l.push_back(MySQLFieldTypeToItem(type,in));
...
@@ -225,13 +225,12 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
...
@@ -225,13 +225,12 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
Item_int(static_cast<ulonglong>(valFromStr(in))) );
Item_int(static_cast<ulonglong>(valFromStr(in))) );
}
}
}else{
}else{
//std::string in("hehe");
l.push_back(MySQLFieldTypeToItem(type,in));
l.push_back(MySQLFieldTypeToItem(type,in));
}
}
tempFileVector.pop_back();
tempFileVector.pop_back();
//l.push_back(&(const_cast<Item&>(i)));
}else{
}else{
l.push_back(my_encrypt_item_layers(i, o, *om, a, IV));
l.push_back(my_encrypt_item_layers(i, o, *om, a, IV));
gcount++;
}
}
}
}
std::string saltName = fm.getSaltName();
std::string saltName = fm.getSaltName();
...
@@ -239,8 +238,6 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
...
@@ -239,8 +238,6 @@ void local_wrapper(const Item &i, const FieldMeta &fm, Analysis &a,
if(gfb.annoOnionNameToFileVector.find(saltName)!=gfb.annoOnionNameToFileVector.end()){
if(gfb.annoOnionNameToFileVector.find(saltName)!=gfb.annoOnionNameToFileVector.end()){
std::vector<std::string> &tempFileVector = gfb.annoOnionNameToFileVector[saltName];
std::vector<std::string> &tempFileVector = gfb.annoOnionNameToFileVector[saltName];
std::string in = tempFileVector.back();
std::string in = tempFileVector.back();
// enum_field_types type = static_cast<enum_field_types>(gfb.annoOnionNameToType[saltName]);
// l.push_back(MySQLFieldTypeToItem(type,in));
l.push_back( new (current_thd->mem_root)
l.push_back( new (current_thd->mem_root)
Item_int(static_cast<ulonglong>(valFromStr(in)))
Item_int(static_cast<ulonglong>(valFromStr(in)))
);
);
...
@@ -293,6 +290,8 @@ main(int argc, char* argv[]){
...
@@ -293,6 +290,8 @@ main(int argc, char* argv[]){
insertManyValues(o,newList);
insertManyValues(o,newList);
std::cout<<(head+o.str())<<std::endl;
std::cout<<(head+o.str())<<std::endl;
}
}
std::cout<<"gcount<<"<<gcount<<std::endl;
return 0;
return 0;
}
}
debug/final_load_low_memory.cc
View file @
a1ae02da
...
@@ -131,58 +131,31 @@ void initGfb(std::vector<FieldMetaTrans> &res,std::string db,std::string table){
...
@@ -131,58 +131,31 @@ void initGfb(std::vector<FieldMetaTrans> &res,std::string db,std::string table){
gfb
.
field_names
=
field_names
;
gfb
.
field_names
=
field_names
;
gfb
.
field_types
=
field_types
;
gfb
.
field_types
=
field_types
;
gfb
.
field_lengths
=
field_lengths
;
gfb
.
field_lengths
=
field_lengths
;
//then we should read the vector
//then we should read the vector
std
::
string
prefix
=
std
::
string
(
"data/"
)
+
db
+
"/"
+
table
+
"/"
;
std
::
string
prefix
=
std
::
string
(
"data/"
)
+
db
+
"/"
+
table
+
"/"
;
unsigned
long
tupleNum
=
0
;
for
(
unsigned
int
i
=
0u
;
i
<
gfb
.
field_names
.
size
();
i
++
)
{
for
(
unsigned
int
i
=
0u
;
i
<
gfb
.
field_names
.
size
();
i
++
)
{
std
::
string
filename
=
prefix
+
gfb
.
field_names
[
i
];
std
::
string
filename
=
prefix
+
gfb
.
field_names
[
i
];
std
::
vector
<
std
::
string
>
column
;
std
::
vector
<
std
::
string
>
column
;
if
(
IS_NUM
(
gfb
.
field_types
[
i
])){
if
(
IS_NUM
(
gfb
.
field_types
[
i
])){
load_num_file_c
ount
(
filename
,
column
,
constGlobalConstants
.
loadCount
);
loadFileNoEscapeLimitC
ount
(
filename
,
column
,
constGlobalConstants
.
loadCount
);
}
else
{
}
else
{
load_string_file_c
ount
(
filename
,
column
,
gfb
.
field_lengths
[
i
],
constGlobalConstants
.
loadCount
);
loadFileEscapeLimitC
ount
(
filename
,
column
,
gfb
.
field_lengths
[
i
],
constGlobalConstants
.
loadCount
);
}
}
tupleNum
=
column
.
size
(
);
std
::
reverse
(
column
.
begin
(),
column
.
end
()
);
gfb
.
annoOnionNameToFileVector
[
gfb
.
field_names
[
i
]]
=
std
::
move
(
column
);
gfb
.
annoOnionNameToFileVector
[
gfb
.
field_names
[
i
]]
=
std
::
move
(
column
);
}
//get memory 31%
}
//init another map
//init another map
for
(
unsigned
int
i
=
0
;
i
<
gfb
.
field_names
.
size
();
i
++
){
for
(
unsigned
int
i
=
0
;
i
<
gfb
.
field_names
.
size
();
i
++
){
gfb
.
annoOnionNameToType
[
gfb
.
field_names
[
i
]]
=
gfb
.
field_types
[
i
];
gfb
.
annoOnionNameToType
[
gfb
.
field_names
[
i
]]
=
gfb
.
field_types
[
i
];
}
}
//extra transformation. transform rows to item*
for
(
unsigned
int
i
=
0
;
i
<
gfb
.
field_names
.
size
();
i
++
){
gfb
.
annoOnionNameToItemVector
[
gfb
.
field_names
[
i
]]
=
std
::
move
(
itemNullVector
(
tupleNum
));
auto
&
dest
=
gfb
.
annoOnionNameToItemVector
[
gfb
.
field_names
[
i
]];
auto
&
src
=
gfb
.
annoOnionNameToFileVector
[
gfb
.
field_names
[
i
]];
enum_field_types
ct
=
static_cast
<
enum_field_types
>
(
field_types
[
i
]);
for
(
unsigned
int
j
=
0
;
j
<
tupleNum
;
j
++
){
if
(
IS_NUM
(
ct
)){
unsigned
int
len
=
gfb
.
field_names
[
i
].
size
();
if
(
len
>
4u
&&
gfb
.
field_names
[
i
].
substr
(
len
-
4
)
==
"ASHE"
){
dest
[
j
]
=
MySQLFieldTypeToItem
(
static_cast
<
enum_field_types
>
(
gfb
.
field_types
[
i
]),
src
[
j
]);
}
else
{
//other fields should be unsigned
dest
[
j
]
=
new
(
current_thd
->
mem_root
)
Item_int
(
static_cast
<
ulonglong
>
(
valFromStr
(
src
[
j
])));
}
}
else
{
dest
[
j
]
=
MySQLFieldTypeToItem
(
static_cast
<
enum_field_types
>
(
gfb
.
field_types
[
i
]),
src
[
j
]);
}
}
gfb
.
annoOnionNameToFileVector
.
erase
(
gfb
.
field_names
[
i
]);
}
//here we get memory 100% and segment fault
}
}
/*load file, decrypt, and then return data plain fields in the type ResType*/
static
static
ResType
ResType
tempfunction
(
std
::
vector
<
std
::
string
>
names
,
load_files
(
std
::
string
db
,
std
::
vector
<
enum_field_types
>
types
,
std
::
vector
<
std
::
vector
<
Item
*>>
&
rows
){
std
::
string
table
)
{
return
ResType
(
true
,
0
,
0
,
std
::
move
(
names
),
std
::
move
(
types
),
std
::
move
(
rows
));
}
/*load file, decrypt, and then return data plain fields in the type ResType*/
static
ResType
load_files_low_memory
(
std
::
string
db
,
std
::
string
table
){
std
::
unique_ptr
<
SchemaInfo
>
schema
=
myLoadSchemaInfo
(
embeddedDir
);
std
::
unique_ptr
<
SchemaInfo
>
schema
=
myLoadSchemaInfo
(
embeddedDir
);
//get all the fields in the tables.
//get all the fields in the tables.
std
::
vector
<
FieldMeta
*>
fms
=
getFieldMeta
(
*
schema
,
db
,
table
);
std
::
vector
<
FieldMeta
*>
fms
=
getFieldMeta
(
*
schema
,
db
,
table
);
...
@@ -191,7 +164,6 @@ static ResType load_files_low_memory(std::string db, std::string table){
...
@@ -191,7 +164,6 @@ static ResType load_files_low_memory(std::string db, std::string table){
for
(
unsigned
int
i
=
0
;
i
<
fms
.
size
();
i
++
){
for
(
unsigned
int
i
=
0
;
i
<
fms
.
size
();
i
++
){
res
[
i
].
trans
(
fms
[
i
]);
res
[
i
].
trans
(
fms
[
i
]);
}
}
create_embedded_thd
(
0
);
//then we should load all the fields available
//then we should load all the fields available
initGfb
(
res
,
db
,
table
);
initGfb
(
res
,
db
,
table
);
...
@@ -201,39 +173,42 @@ static ResType load_files_low_memory(std::string db, std::string table){
...
@@ -201,39 +173,42 @@ static ResType load_files_low_memory(std::string db, std::string table){
vector
<
int
>
field_types
=
ggbt
.
field_types
;
vector
<
int
>
field_types
=
ggbt
.
field_types
;
vector
<
int
>
field_lengths
=
ggbt
.
field_lengths
;
vector
<
int
>
field_lengths
=
ggbt
.
field_lengths
;
//why do we need this??
the error comes from itemNullVector
//why do we need this??
//
create_embedded_thd(0);
create_embedded_thd
(
0
);
rawMySQLReturnValue
resraw
;
vector
<
vector
<
Item
*>>
res_field_item
;
vector
<
vector
<
string
>>
res_field
;
for
(
auto
item
:
field_names
){
for
(
auto
item
:
field_names
){
res_field
_item
.
push_back
(
gfb
.
annoOnionNameToItem
Vector
[
item
]);
res_field
.
push_back
(
gfb
.
annoOnionNameToFile
Vector
[
item
]);
}
}
//then transform it to ress_fields
//then transform it to ress_fields
unsigned
int
length
=
res_field_item
[
0
].
size
();
unsigned
int
length
=
res_field
[
0
].
size
();
vector
<
vector
<
Item
*>>
ress_field_item
;
vector
<
vector
<
string
>>
ress_field
;
for
(
unsigned
int
i
=
0u
;
i
<
length
;
i
++
){
for
(
unsigned
int
i
=
0u
;
i
<
length
;
i
++
){
vector
<
Item
*>
row
=
itemNullVector
(
res_field_item
.
size
())
;
vector
<
string
>
row
;
for
(
unsigned
int
j
=
0u
;
j
<
res_field
_item
.
size
();
j
++
){
for
(
unsigned
int
j
=
0u
;
j
<
res_field
.
size
();
j
++
){
row
[
j
]
=
res_field_item
[
j
][
i
]
;
row
.
push_back
(
res_field
[
j
][
i
])
;
}
}
ress_field
_item
.
push_back
(
row
);
ress_field
.
push_back
(
row
);
}
}
std
::
vector
<
enum_field_types
>
fieldTypes
;
resraw
.
rowValues
=
ress_field
;
resraw
.
fieldNames
=
field_names
;
for
(
unsigned
int
i
=
0
;
i
<
field_types
.
size
();
++
i
){
for
(
unsigned
int
i
=
0
;
i
<
field_types
.
size
();
++
i
){
fieldTypes
.
push_back
(
static_cast
<
enum_field_types
>
(
field_types
[
i
]));
resraw
.
fieldTypes
.
push_back
(
static_cast
<
enum_field_types
>
(
field_types
[
i
]));
}
}
ResType
rawtorestype
=
tempfunction
(
field_names
,
fieldTypes
,
ress_field_item
);
ResType
rawtorestype
=
rawMySQLReturnValue_to_ResType
(
false
,
&
resraw
);
auto
finalresults
=
decryptResults
(
rawtorestype
,
*
rm
);
auto
finalresults
=
decryptResults
(
rawtorestype
,
*
rm
);
return
std
::
move
(
finalresults
)
;
return
finalresults
;
}
}
unsigned
long
gcount
=
0
;
static
static
void
local_wrapper_low_memory_item
(
const
Item
&
i
,
const
FieldMeta
&
fm
,
Analysis
&
a
,
void
local_wrapper
(
const
Item
&
i
,
const
FieldMeta
&
fm
,
Analysis
&
a
,
List
<
Item
>
*
const
append_list
){
List
<
Item
>
*
const
append_list
)
{
//append_list->push_back(&(const_cast<Item&>(i)));
//do not use the plain strategy
std
::
vector
<
Item
*>
l
;
std
::
vector
<
Item
*>
l
;
const
uint64_t
salt
=
fm
.
getHasSalt
()
?
randomValue
()
:
0
;
const
uint64_t
salt
=
fm
.
getHasSalt
()
?
randomValue
()
:
0
;
uint64_t
IV
=
salt
;
uint64_t
IV
=
salt
;
...
@@ -242,21 +217,35 @@ void local_wrapper_low_memory_item(const Item &i, const FieldMeta &fm, Analysis
...
@@ -242,21 +217,35 @@ void local_wrapper_low_memory_item(const Item &i, const FieldMeta &fm, Analysis
OnionMeta
*
const
om
=
it
.
second
;
OnionMeta
*
const
om
=
it
.
second
;
std
::
string
annoOnionName
=
om
->
getAnonOnionName
();
std
::
string
annoOnionName
=
om
->
getAnonOnionName
();
if
(
gfb
.
annoOnionNameToFileVector
.
find
(
annoOnionName
)
!=
gfb
.
annoOnionNameToFileVector
.
end
()){
if
(
gfb
.
annoOnionNameToFileVector
.
find
(
annoOnionName
)
!=
gfb
.
annoOnionNameToFileVector
.
end
()){
std
::
vector
<
Item
*>
&
tempItemVector
=
gfb
.
annoOnionNameToItemVector
[
annoOnionName
];
enum_field_types
type
=
static_cast
<
enum_field_types
>
(
gfb
.
annoOnionNameToType
[
annoOnionName
]);
Item
*
in
=
tempItemVector
.
back
();
std
::
vector
<
std
::
string
>
&
tempFileVector
=
gfb
.
annoOnionNameToFileVector
[
annoOnionName
];
l
.
push_back
(
in
);
std
::
string
in
=
tempFileVector
.
back
();
tempItemVector
.
pop_back
();
if
(
IS_NUM
(
type
)){
unsigned
int
len
=
annoOnionName
.
size
();
if
(
len
>
4u
&&
annoOnionName
.
substr
(
len
-
4
)
==
"ASHE"
){
l
.
push_back
(
MySQLFieldTypeToItem
(
type
,
in
));
}
else
{
l
.
push_back
(
new
(
current_thd
->
mem_root
)
Item_int
(
static_cast
<
ulonglong
>
(
valFromStr
(
in
)))
);
}
}
else
{
l
.
push_back
(
MySQLFieldTypeToItem
(
type
,
in
));
}
tempFileVector
.
pop_back
();
}
else
{
}
else
{
l
.
push_back
(
my_encrypt_item_layers
(
i
,
o
,
*
om
,
a
,
IV
));
l
.
push_back
(
my_encrypt_item_layers
(
i
,
o
,
*
om
,
a
,
IV
));
gcount
++
;
}
}
}
}
std
::
string
saltName
=
fm
.
getSaltName
();
std
::
string
saltName
=
fm
.
getSaltName
();
if
(
fm
.
getHasSalt
())
{
if
(
fm
.
getHasSalt
())
{
if
(
gfb
.
annoOnionNameToFileVector
.
find
(
saltName
)
!=
gfb
.
annoOnionNameToFileVector
.
end
()){
if
(
gfb
.
annoOnionNameToFileVector
.
find
(
saltName
)
!=
gfb
.
annoOnionNameToFileVector
.
end
()){
std
::
vector
<
Item
*>
&
tempItemVector
=
gfb
.
annoOnionNameToItemVector
[
saltName
];
std
::
vector
<
std
::
string
>
&
tempFileVector
=
gfb
.
annoOnionNameToFileVector
[
saltName
];
Item
*
in
=
tempItemVector
.
back
();
std
::
string
in
=
tempFileVector
.
back
();
l
.
push_back
(
in
);
l
.
push_back
(
new
(
current_thd
->
mem_root
)
tempItemVector
.
pop_back
();
Item_int
(
static_cast
<
ulonglong
>
(
valFromStr
(
in
)))
);
tempFileVector
.
pop_back
();
}
else
{
}
else
{
l
.
push_back
(
new
Item_int
(
static_cast
<
ulonglong
>
(
salt
)));
l
.
push_back
(
new
Item_int
(
static_cast
<
ulonglong
>
(
salt
)));
}
}
...
@@ -268,14 +257,12 @@ void local_wrapper_low_memory_item(const Item &i, const FieldMeta &fm, Analysis
...
@@ -268,14 +257,12 @@ void local_wrapper_low_memory_item(const Item &i, const FieldMeta &fm, Analysis
}
}
int
int
main
(
int
argc
,
char
*
argv
[]){
main
(
int
argc
,
char
*
argv
[]){
init
();
init
();
create_embedded_thd
(
0
);
create_embedded_thd
(
0
);
std
::
string
ip
=
"localhost"
;
std
::
string
db
=
"tdb"
,
table
=
"student"
;
std
::
string
db
=
"tdb"
,
table
=
"student"
;
std
::
string
ip
=
"localhost"
;
if
(
argc
==
4
){
if
(
argc
==
4
){
ip
=
std
::
string
(
argv
[
1
]);
ip
=
std
::
string
(
argv
[
1
]);
db
=
std
::
string
(
argv
[
2
]);
db
=
std
::
string
(
argv
[
2
]);
...
@@ -288,25 +275,27 @@ main(int argc, char* argv[]){
...
@@ -288,25 +275,27 @@ main(int argc, char* argv[]){
Analysis
analysis
(
db
,
*
schema
,
TK
,
SECURITY_RATING
::
SENSITIVE
);
Analysis
analysis
(
db
,
*
schema
,
TK
,
SECURITY_RATING
::
SENSITIVE
);
/*choose decryption onion, load and decrypt to plain text*/
/*choose decryption onion, load and decrypt to plain text*/
ResType
res
=
load_files
_low_memory
(
db
,
table
);
ResType
res
=
load_files
(
db
,
table
);
std
::
string
annoTableName
=
analysis
.
getTableMeta
(
db
,
table
).
getAnonTableName
();
std
::
string
annoTableName
=
analysis
.
getTableMeta
(
db
,
table
).
getAnonTableName
();
const
std
::
string
head
=
std
::
string
(
"INSERT INTO `"
)
+
db
+
"`.`"
+
annoTableName
+
"` "
;
const
std
::
string
head
=
std
::
string
(
"INSERT INTO `"
)
+
db
+
"`.`"
+
annoTableName
+
"` "
;
/*reencryption to get the encrypted insert!!!*/
/*reencryption to get the encrypted insert!!!*/
for
(
auto
&
row
:
res
.
rows
){
for
(
auto
&
row
:
res
.
rows
)
{
List
<
List_item
>
newList
;
List
<
List_item
>
newList
;
List
<
Item
>
*
const
newList0
=
new
List
<
Item
>
();
List
<
Item
>
*
const
newList0
=
new
List
<
Item
>
();
for
(
auto
i
=
0u
;
i
<
res
.
names
.
size
();
i
++
){
for
(
auto
i
=
0u
;
i
<
res
.
names
.
size
();
i
++
){
std
::
string
field_name
=
res
.
names
[
i
];
std
::
string
field_name
=
res
.
names
[
i
];
FieldMeta
&
fm
=
analysis
.
getFieldMeta
(
db
,
table
,
field_name
);
FieldMeta
&
fm
=
analysis
.
getFieldMeta
(
db
,
table
,
field_name
);
local_wrapper
_low_memory_item
(
*
row
[
i
],
fm
,
analysis
,
newList0
);
local_wrapper
(
*
row
[
i
],
fm
,
analysis
,
newList0
);
}
}
newList
.
push_back
(
newList0
);
newList
.
push_back
(
newList0
);
std
::
ostringstream
o
;
std
::
ostringstream
o
;
insertManyValues
(
o
,
newList
);
insertManyValues
(
o
,
newList
);
std
::
cout
<<
(
head
+
o
.
str
())
<<
std
::
endl
;
std
::
cout
<<
(
head
+
o
.
str
())
<<
std
::
endl
;
}
}
std
::
cout
<<
"gcount<<"
<<
gcount
<<
std
::
endl
;
return
0
;
return
0
;
}
}
debug/final_load_low_memory.cc.back
0 → 100644
View file @
a1ae02da
This diff is collapsed.
Click to expand it.
debug/final_store.cc.back
View file @
a1ae02da
...
@@ -57,18 +57,32 @@ std::string getTestQuery(SchemaInfo &schema, std::vector<FieldMetaTrans> &tfds,
...
@@ -57,18 +57,32 @@ std::string getTestQuery(SchemaInfo &schema, std::vector<FieldMetaTrans> &tfds,
return res;
return res;
}
}
static void write_meta(
rawMySQLReturnValue& resraw,
std::vector<FieldMetaTrans> &res,string db,string table){
static void write_meta(std::vector<FieldMetaTrans> &res,string db,string table){
TableMetaTrans mf(db,table,res);
TableMetaTrans mf(db,table,res);
mf.set_db_table(db,table);
mf.set_db_table(db,table);
mf.serialize();
mf.serialize();
}
}
static
static
void write_raw_data_to_files(
rawMySQLReturnValue
& resraw,std::vector<FieldMetaTrans> &res ,string db,string table){
void write_raw_data_to_files(
MySQLColumnData
& resraw,std::vector<FieldMetaTrans> &res ,string db,string table){
//write metafiles
//write metafiles
write_meta(res
raw,res
,db,table);
write_meta(res,db,table);
//write datafiles
//write datafiles
write_row_data(resraw,db,table);
std::string prefix = std::string("data/") +db+"/"+table+"/";
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])){
writeColumndataNum(resraw.columnData[i],filenames[i]);
}else{
writeColumndataEscapeString(resraw.columnData[i],filenames[i],resraw.maxLengths[i]);
}
}
}
}
static void store(std::string db, std::string table){
static void store(std::string db, std::string table){
...
@@ -90,11 +104,11 @@ static void store(std::string db, std::string table){
...
@@ -90,11 +104,11 @@ static void store(std::string db, std::string table){
//generate the backup query and then fetch the tuples
//generate the backup query and then fetch the tuples
std::string backup_query = getTestQuery(*schema,res,db,table);
std::string backup_query = getTestQuery(*schema,res,db,table);
rawMySQLReturnValue resraw = executeAndGetResultRemote
(globalConn,backup_query);
MySQLColumnData resraw = executeAndGetColumnData
(globalConn,backup_query);
//then we should set the type and length of FieldMetaTrans
//then we should set the type and length of FieldMetaTrans
auto types = resraw.fieldTypes;
auto types = resraw.fieldTypes;
auto lengths = resraw.
l
engths;
auto lengths = resraw.
maxL
engths;
int base_types = 0;
int base_types = 0;
int base_lengths = 0;
int base_lengths = 0;
for(auto &item:res){
for(auto &item:res){
...
...
test_util/verify_timer.cc
0 → 100644
View file @
a1ae02da
#include "util/timer.hh"
#include <iostream>
#include <string>
#include <unistd.h>
using
std
::
cout
;
using
std
::
endl
;
int
main
(){
timer
t
;
sleep
(
1
);
cout
<<
t
.
lap
()
<<
endl
;
return
0
;
}
util/timer.hh
View file @
a1ae02da
#pragma once
#pragma once
#include <sys/time.h>
#include <sys/time.h>
#include <time.h>
class
timer
{
class
timer
{
private
:
private
:
...
@@ -9,19 +10,19 @@ class timer {
...
@@ -9,19 +10,19 @@ class timer {
public
:
public
:
timer
()
{
lap
();
}
timer
()
{
lap
();
}
u
int64_t
lap
()
{
/* returns microseconds */
u
nsigned
long
lap
()
{
/* returns microseconds */
u
int64_t
t0
=
start
;
u
nsigned
long
t0
=
start
;
u
int64_t
t1
=
cur_usec
();
u
nsigned
long
t1
=
cur_usec
();
start
=
t1
;
start
=
t1
;
return
t1
-
t0
;
return
t1
-
t0
;
}
}
private
:
private
:
//uint64_t ?? not a type
static
u
int64_t
cur_usec
()
{
static
u
nsigned
long
cur_usec
()
{
struct
timeval
tv
;
struct
timeval
tv
;
gettimeofday
(
&
tv
,
0
);
gettimeofday
(
&
tv
,
0
);
return
((
u
int64_t
)
tv
.
tv_sec
)
*
1000000
+
tv
.
tv_usec
;
return
((
u
nsigned
long
)
tv
.
tv_sec
)
*
1000000
+
tv
.
tv_usec
;
}
}
u
int64_t
start
;
u
nsigned
long
start
;
};
};
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment