internal override void Generate()
{
WriteAutoGeneratedSourceHeader();
Write("import { TpmMarshaller, TpmBuffer } from \"./TpmMarshaller.js\";\r\n" +
"import { TpmStructure, ReqStructure, RespStructure, SessEncInfo } from \"./TpmStructure.js\";\r\n" +
"\r\n" +
"import { Crypto } from \"./Crypt.js\";\r\n" +
"\r\n");
// First generate enums
foreach (var e in TpmTypes.Get <TpmEnum>())
{
GenEnum(e);
}
foreach (var b in TpmTypes.Get <TpmBitfield>())
{
GenBitfield(b);
}
// Then generate unions and structures
GenUnions();
foreach (var s in TpmTypes.Get <TpmStruct>())
{
GenStruct(s);
}
File.WriteAllText(RootDir + "TpmTypes.ts", GeneratedCode.ToString());
GeneratedCode.Clear();
// Now generate the TPM methods
GenCommands();
File.WriteAllText(RootDir + "Tpm.ts", GeneratedCode.ToString());
GeneratedCode.Clear();
}
internal override void Generate()
{
WriteAutoGeneratedSourceHeader();
Write("from .TpmStructure import *");
Write("from .TpmEnum import *");
Write("");
// First generate enums
foreach (var e in TpmTypes.Get <TpmEnum>())
{
GenEnum(e);
}
foreach (var b in TpmTypes.Get <TpmBitfield>())
{
GenBitfield(b);
}
Write("from .Crypt import *" + "\r\n");
// Then generate unions and structures
GenUnions();
foreach (var s in TpmTypes.Get <TpmStruct>())
{
GenStruct(s);
}
File.WriteAllText(RootDir + "TpmTypes.py", GeneratedCode.ToString());
GeneratedCode.Clear();
// Now generate the TPM methods
GenCommands();
File.WriteAllText(RootDir + "Tpm.py", GeneratedCode.ToString());
GeneratedCode.Clear();
}
/// <summary> Writes GeneratedCode to the destination using appropriate template </summary>
/// <remarks> Replaces everything between //>>> and //<<< in template file </remarks>
/// <param name="fileName"> Destination file </param>
/// <param name="toRemove"> Lines containing toRemove entries are deleted </param>
/// <param name="tpmCommands"> Selects the appropriate template </param>
void WriteDef(string fileName, string[] toRemove, bool tpmCommands)
{
string template = tpmCommands ? tpmObjectTemplate : typesTemplate;
StringBuilder b = new StringBuilder();
string terminatorLine;
using (StringReader sr = new StringReader(template))
{
while (true)
{
string line = sr.ReadLine();
if (line == null)
{
break;
}
if (toRemove != null && toRemove.Any(frag => line.Contains(frag)))
{
continue;
}
b.AppendLine(line);
if (line.Contains("//>>>"))
{
// skip everything until the terminator
while (true)
{
var l = sr.ReadLine();
if (l == null)
{
throw new Exception("Terminator not found");
}
if (l.Contains("//<<<"))
{
terminatorLine = l;
break;
}
}
// and insert the defs
b.AppendLine(GeneratedCode.ToString());
GeneratedCode.Clear();
b.AppendLine(terminatorLine);
}
}
}
string temp = b.ToString();
b.Clear();
temp = temp.Replace(ExtraImportsTag, needsUtil ? "import java.util.*;" : "");
needsUtil = false;
File.WriteAllText(fileName, temp);
}
