public static ICSequence TranslateStatement(Statement stat, Program program)
{
ICSequence re = new ICSequence();
if (stat is SSequence)
{
SSequence ss = (SSequence)stat;
foreach (var entry in ss.GetStatementsList())
{
re.AddRange(TranslateStatement(/*condition, */ entry, program));
}
}
else if (stat is SAssignment)
{
SAssignment sa = (SAssignment)stat;
System.Diagnostics.Debug.Assert(sa.result is EVariable);
EVariable result = (EVariable)sa.result;
//if(condition != null)
//{
// Expression
// notExp = new EBinaryOperation(new ENumericLiteral("1", 0), condition, OperationType.Substraction),
// term1Exp = new EBinaryOperation(condition, sa.value, OperationType.Multiplication),
// term2Exp = new EBinaryOperation(notExp, result, OperationType.Multiplication),
// newExp = new EBinaryOperation(term1Exp, term2Exp, OperationType.Addition);
// re.AddRange(TranslateExpression(newExp, result));
//}
//else
//{
if (sa.value is ETerminal)
{
re.Add(new ICAssignment(OperationType.None, result, (ETerminal)sa.value, null));
}
else
{
re.AddRange(TranslateExpression(sa.value, result, program));
}
//}
}
else if (stat is SIfElse)
{
SIfElse sie = (SIfElse)stat;
ETerminal cond = null;
ICIfElse icie = new ICIfElse();
icie.prob = sie.prob;
icie.revealCond = sie.revealCond;
if (sie.condition is ETerminal)
{
cond = (ETerminal)sie.condition;
}
else
{
cond = new EVariable(program);
icie.conditionCodes.AddRange(TranslateExpression(sie.condition, (EVariable)cond, program));
}
icie.condition = cond;
if (!ReferenceEquals(sie.statIf, null))
{
icie.codesIf.AddRange(TranslateStatement(sie.statIf, program));
}
if (!ReferenceEquals(sie.statElse, null))
{
icie.codesElse.AddRange(TranslateStatement(sie.statElse, program));
}
re.Add(icie);
}
else if (stat is SReturn)
{
SReturn sr = (SReturn)stat;
foreach (var exp in sr.exps)
{
if (exp is EVariable)
{
program.vReturn.Add(((EVariable)exp).name);
}
else
{
var place = new EVariable(program);
re.AddRange(TranslateExpression(exp, place, program));
program.vReturn.Add(place.name);
}
}
}
else if (stat is SWhile)
{
SWhile sw = (SWhile)stat;
ICWhile icw = new ICWhile();
ETerminal cond = null;
if (sw.condition is ETerminal)
{
cond = (ETerminal)sw.condition;
}
else
{
cond = new EVariable(program);
icw.conditionCodes.AddRange(TranslateExpression(sw.condition, (EVariable)cond, program));
}
icw.condition = cond;
icw.codes.AddRange(TranslateStatement(/*cond, */ sw.stat, program));
re.Add(icw);
}
return(re);
}
// tranform intermediate code at runtime
public static ICSequence TransformIntermediateCode(ETerminal condition, IntermediateCode code, Program program)
{
ICSequence re = new ICSequence();
if (code is ICAssignment)
{
ICAssignment ica = (ICAssignment)code;
if (ReferenceEquals(condition, null))
{
re.Add(code);
}
else
{
EVariable
result = ica.result,
temp1 = new EVariable(program, "$ICA_1_" + code.index),
temp2 = new EVariable(program, "$ICA_2_" + code.index),
temp3 = new EVariable(program, "$ICA_3_" + code.index),
temp4 = new EVariable(program, "$ICA_4_" + code.index);
ICAssignment
// condition * newResult + !condition * oldResult
// Command index is the same as the old command
// such that EVH and KH can receive the correct message.
// Commands generated at runtime cannot be run in parallel.
newCode = new ICAssignment(ica.op, temp1, ica.operand1, ica.operand2, code.index),
notCondition = new ICAssignment(OperationType.NOT, temp2, condition, null, code.index),
term1Exp = new ICAssignment(OperationType.Multiplication, temp3, condition, temp1, code.index),
term2Exp = new ICAssignment(OperationType.Multiplication, temp4, temp2, result, code.index),
term3Exp = new ICAssignment(OperationType.Addition, result, temp3, temp4, code.index);
re.AddRange(new ICAssignment[] { newCode, notCondition, term1Exp, term2Exp, term3Exp });
}
}
else if (code is ICIfElse)
{
// for if-else statement, simply set the outerCondition
ICIfElse icie = (ICIfElse)code;
icie.outerCondition = condition;
re.Add(icie);
}
else if (code is ICWhile)
{
ICWhile icw = (ICWhile)code;
// outerCondition && while condition. This command will be executed very time condition codes are evaluated
if (!ReferenceEquals(condition, null))
{
EVariable temp1 = new EVariable(program, "&ICW_1_" + code.index);
icw.conditionCodes.Add(new ICAssignment(OperationType.AND, temp1, icw.condition, condition, icw.conditionCodes[icw.conditionCodes.Count - 1].index));
icw.condition = temp1;
}
ICSequence icwCodes = new ICSequence();
foreach (var entry in icw.codes.GetCodes())
{
// if statement is Assignment and result is a temporarily variable, no need to transform
if (entry is ICAssignment && ((ICAssignment)entry).result.isTemporary)
{
icwCodes.Add(entry);
continue;
}
icwCodes.AddRange(TransformIntermediateCode(icw.condition, entry, program));
}
icw.codes = icwCodes;
re.Add(icw);
}
return(re);
}
private ICSequence[] EncBlock(ICSequence codes)
{
var toKH = new ICSequence();
var toEVH = new ICSequence();
foreach (var ic in codes.GetCodes())
{
if (ic is ICAssignment)
{
ICAssignment ica = (ICAssignment)ic;
Numeric key1 = null, key2 = null;
toKH.Add(new ICAssignment(ica, ref key1, ref key2, PartyType.KH));
toEVH.Add(new ICAssignment(ica, ref key1, ref key2, PartyType.EVH));
}
else if (ic is ICWhile)
{
ICWhile icw = (ICWhile)ic, icwEVH = new ICWhile(), icwKH = new ICWhile();
icwEVH.index = icw.index;
icwKH.index = icw.index;
if (icw.condition is ENumericLiteral)
{
var cond = (ENumericLiteral)icw.condition;
if (!cond.needEnc)
{
var val = new Numeric(cond.GetValue());
val.SetEncType(EncryptionType.None);
icwEVH.condition = new ENumericLiteral(val);
((ENumericLiteral)icwEVH.condition).needEnc = false;
icwKH.condition = new ENumericLiteral(val);
((ENumericLiteral)icwKH.condition).needEnc = false;
}
else
{
var keyTemp = Utility.NextUnsignedNumeric(0, 1);
icwEVH.condition = new ENumericLiteral(GetValue(icw.condition) ^ keyTemp);
icwKH.condition = new ENumericLiteral(keyTemp);
}
}
else
{
icwEVH.condition = icw.condition;
icwKH.condition = icw.condition;
}
var encCondCodes = EncBlock(icw.conditionCodes);
icwEVH.conditionCodes.AddRange(encCondCodes[0]);
icwKH.conditionCodes.AddRange(encCondCodes[1]);
var encCodes = EncBlock(icw.codes);
icwEVH.codes.AddRange(encCodes[0]);
icwKH.codes.AddRange(encCodes[1]);
toEVH.Add(icwEVH);
toKH.Add(icwKH);
}
else
{
ICIfElse icie = (ICIfElse)ic, icieEVH = new ICIfElse(), icieKH = new ICIfElse();
icieEVH.revealCond = icie.revealCond;
icieKH.revealCond = icie.revealCond;
icieEVH.prob = icie.prob;
icieKH.prob = icie.prob;
icieEVH.index = icie.index;
icieKH.index = icie.index;
if (icie.condition is ENumericLiteral)
{
var cond = (ENumericLiteral)icie.condition;
if (!cond.needEnc)
{
var val = new Numeric(cond.GetValue());
val.SetEncType(EncryptionType.None);
icieEVH.condition = new ENumericLiteral(val);
((ENumericLiteral)icieEVH.condition).needEnc = false;
icieKH.condition = new ENumericLiteral(val);
((ENumericLiteral)icieKH.condition).needEnc = false;
}
else
{
var keyTemp = Utility.NextUnsignedNumeric(0, 1);
icieEVH.condition = new ENumericLiteral(GetValue(icie.condition) ^ keyTemp);
icieKH.condition = new ENumericLiteral(keyTemp);
}
}
else
{
icieEVH.condition = icie.condition;
icieKH.condition = icie.condition;
}
var encCondCodes = EncBlock(icie.conditionCodes);
icieEVH.conditionCodes.AddRange(encCondCodes[0]);
icieKH.conditionCodes.AddRange(encCondCodes[1]);
var encIfCodes = EncBlock(icie.codesIf);
icieEVH.codesIf = encIfCodes[0];
icieKH.codesIf = encIfCodes[1];
var encElseCodes = EncBlock(icie.codesElse);
icieEVH.codesElse = encElseCodes[0];
icieKH.codesElse = encElseCodes[1];
toEVH.Add(icieEVH);
toKH.Add(icieKH);
}
}
return(new ICSequence[] { toEVH, toKH });
}