mirror of
https://github.com/nsacyber/HIRS.git
synced 2024-12-21 05:53:27 +00:00
250 lines
9.0 KiB
C#
250 lines
9.0 KiB
C#
using System.Security.Cryptography;
|
|
using System.Security.Cryptography.X509Certificates;
|
|
using System.Security.Cryptography.Xml;
|
|
using System.Xml;
|
|
using System.CommandLine;
|
|
using System.Text;
|
|
|
|
/**
|
|
* This command line program has three commands:
|
|
* 1. sign - append a signature calculated from a user-provided private key
|
|
* 2. validate - validate a signature with a user-provided certificate
|
|
* 3. debug - print out important components of a signed XML document
|
|
*
|
|
* The validate functioin strictly checks the cryptographic integrity of the signature,
|
|
* it does not verify the integrity of the certificate chain.
|
|
*/
|
|
class XmlDsigTool
|
|
{
|
|
|
|
static async Task<int> Main(String[] args)
|
|
{
|
|
var fileOption = new Option<string>(
|
|
name: "--file",
|
|
description: "The filename input for the command.",
|
|
parseArgument: result =>
|
|
{
|
|
string? filePath = result.Tokens.Single().Value;
|
|
if (!File.Exists(filePath))
|
|
{
|
|
result.ErrorMessage = "File " + filePath + " does not exist.";
|
|
return null;
|
|
}
|
|
else
|
|
{
|
|
return filePath;
|
|
}
|
|
});
|
|
var privateKeyOption = new Option<string>(
|
|
name: "--private-key",
|
|
description: "The private key with which to sign."
|
|
);
|
|
var certificateOption = new Option<string>(
|
|
name: "--certificate",
|
|
description: "The certificate with which to validate the signature."
|
|
);
|
|
|
|
var rootCommand = new RootCommand("A tool for signing, validating, and debugging base RIMs.");
|
|
var signCommand = new Command("sign", "Sign the given file with the given key.")
|
|
{
|
|
fileOption,
|
|
privateKeyOption
|
|
};
|
|
var validateCommand = new Command("validate", "Validate the signature in the given base RIM.")
|
|
{
|
|
fileOption,
|
|
certificateOption
|
|
};
|
|
var debugCommand = new Command("debug", "Print out the significant portions of a base RIM and the expected signature value.")
|
|
{
|
|
fileOption,
|
|
privateKeyOption
|
|
};
|
|
|
|
signCommand.SetHandler(async (file, privateKey) =>
|
|
{
|
|
await SignXml(file, privateKey);
|
|
}, fileOption, privateKeyOption);
|
|
validateCommand.SetHandler(async (file, certificate) =>
|
|
{
|
|
await ValidateXml(file, certificate);
|
|
}, fileOption, certificateOption);
|
|
debugCommand.SetHandler(async (file, privateKey) =>
|
|
{
|
|
await DebugRim(file, privateKey);
|
|
}, fileOption, privateKeyOption);
|
|
|
|
rootCommand.AddCommand(signCommand);
|
|
rootCommand.AddCommand(validateCommand);
|
|
rootCommand.AddCommand(debugCommand);
|
|
|
|
return rootCommand.InvokeAsync(args).Result;
|
|
}
|
|
internal static async Task SignXml(string xmlFilename, string keyFilename)
|
|
{
|
|
if (String.IsNullOrWhiteSpace(xmlFilename))
|
|
throw new ArgumentException(nameof(xmlFilename));
|
|
if (String.IsNullOrWhiteSpace(keyFilename))
|
|
throw new ArgumentException(nameof(keyFilename));
|
|
|
|
Console.Write("Signing xml...");
|
|
|
|
// Load an XML file into a SignedXML object.
|
|
XmlDocument unsignedDoc = new XmlDocument();
|
|
unsignedDoc.Load(xmlFilename);
|
|
SignedXml signedXml = new SignedXml(unsignedDoc);
|
|
|
|
//Load private key from file
|
|
string privateKeyText = File.ReadAllText(keyFilename);
|
|
var privateKey = RSA.Create();
|
|
privateKey.ImportFromPem(privateKeyText);
|
|
|
|
// Add the key to the SignedXml document.
|
|
signedXml.SigningKey = privateKey;
|
|
|
|
// Create a reference to be signed.
|
|
Reference reference = new Reference();
|
|
reference.Uri = "";
|
|
|
|
// Add an enveloped transformation to the reference.
|
|
XmlDsigEnvelopedSignatureTransform env = new XmlDsigEnvelopedSignatureTransform();
|
|
reference.AddTransform(env);
|
|
|
|
// Add the reference to the SignedXml object.
|
|
signedXml.AddReference(reference);
|
|
|
|
// Add keyinfo block
|
|
KeyInfo keyInfo = new KeyInfo();
|
|
keyInfo.AddClause(new RSAKeyValue((RSA)privateKey));
|
|
signedXml.KeyInfo = keyInfo;
|
|
|
|
// Compute the signature.
|
|
signedXml.ComputeSignature();
|
|
|
|
// Get the XML representation of the signature and save
|
|
// it to an XmlElement object.
|
|
XmlElement xmlDigitalSignature = signedXml.GetXml();
|
|
|
|
// Append the element to the XML document.
|
|
unsignedDoc.DocumentElement.AppendChild(unsignedDoc.ImportNode(xmlDigitalSignature, true));
|
|
string signedFilename = "signed_" + xmlFilename;
|
|
unsignedDoc.Save(signedFilename);
|
|
Console.WriteLine("Xml signed and written to " + signedFilename);
|
|
}
|
|
|
|
// Verify the signature of an XML file against an asymmetric
|
|
// algorithm and return the result.
|
|
internal static async Task ValidateXml(string signedFilename, string certFilename)
|
|
{
|
|
// Check arguments.
|
|
if (String.IsNullOrWhiteSpace(signedFilename))
|
|
throw new ArgumentException(nameof(signedFilename));
|
|
if (certFilename == null)
|
|
throw new ArgumentException(nameof(certFilename));
|
|
|
|
Console.Write("Verifying signature...");
|
|
// Create a new SignedXml object and pass it
|
|
// the XML document class.
|
|
XmlDocument signedDoc = new XmlDocument();
|
|
signedDoc.Load(signedFilename);
|
|
SignedXml signedXml = new SignedXml(signedDoc);
|
|
|
|
//Load public cert from file
|
|
X509Certificate2 signingCert = new X509Certificate2(certFilename);
|
|
RSA publicKey = signingCert.GetRSAPublicKey();
|
|
|
|
// Find the "Signature" node and create a new
|
|
// XmlNodeList object.
|
|
XmlNodeList nodeList = signedDoc.GetElementsByTagName("Signature");
|
|
|
|
// Throw an exception if no signature was found.
|
|
if (nodeList.Count <= 0)
|
|
{
|
|
throw new CryptographicException("Verification failed: No Signature was found in the document.");
|
|
}
|
|
|
|
// This example only supports one signature for
|
|
// the entire XML document. Throw an exception
|
|
// if more than one signature was found.
|
|
if (nodeList.Count >= 2)
|
|
{
|
|
throw new CryptographicException("Verification failed: More than one signature was found for the document.");
|
|
}
|
|
|
|
// Load the first <signature> node.
|
|
signedXml.LoadXml((XmlElement)nodeList[0]);
|
|
Boolean isValid = false;
|
|
try
|
|
{
|
|
isValid = signedXml.CheckSignature(publicKey);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Console.WriteLine(e.Message);
|
|
}
|
|
|
|
// Check the signature and return the result.
|
|
if (isValid)
|
|
{
|
|
Console.WriteLine("Signature is valid!");
|
|
}
|
|
else
|
|
{
|
|
Console.WriteLine("Signature is not valid.");
|
|
}
|
|
}
|
|
|
|
internal static async Task DebugRim(string filename, string keyFilename)
|
|
{
|
|
if (String.IsNullOrWhiteSpace(filename))
|
|
{
|
|
throw new ArgumentException(nameof(filename));
|
|
} else if (String.IsNullOrWhiteSpace(keyFilename))
|
|
{
|
|
throw new ArgumentException(nameof(keyFilename));
|
|
}
|
|
|
|
XmlDocument xmlToBeSigned = new XmlDocument();
|
|
XmlDocument xmlDoc = new XmlDocument();
|
|
xmlDoc.Load(filename);
|
|
XmlNodeList nodes = xmlDoc.GetElementsByTagName("SoftwareIdentity");
|
|
XmlNodeList signatureNodes = xmlDoc.GetElementsByTagName("Signature");
|
|
|
|
//Assumes there is only one signature; may change in the future for multiple signatures
|
|
if (signatureNodes.Count > 0)
|
|
{
|
|
nodes[0].RemoveChild(signatureNodes[0]);
|
|
}
|
|
xmlToBeSigned.AppendChild(xmlToBeSigned.ImportNode(nodes[0], true));
|
|
string outFileName = "ToBeSigned_" + filename;
|
|
xmlToBeSigned.Save(outFileName);
|
|
Console.WriteLine("Xml data to be signed parsed to " + outFileName);
|
|
|
|
//Load private key from file
|
|
string privateKeyText = File.ReadAllText(keyFilename);
|
|
var privateKey = RSA.Create();
|
|
privateKey.ImportFromPem(privateKeyText);
|
|
|
|
// Add the key to the SignedXml document.
|
|
SignedXml signedXml = new SignedXml(xmlToBeSigned);
|
|
signedXml.SigningKey = privateKey;
|
|
|
|
// Create a reference to be signed.
|
|
Reference reference = new Reference();
|
|
reference.Uri = "";
|
|
|
|
// Add an enveloped transformation to the reference.
|
|
XmlDsigEnvelopedSignatureTransform env = new XmlDsigEnvelopedSignatureTransform();
|
|
reference.AddTransform(env);
|
|
|
|
// Add the reference to the SignedXml object.
|
|
signedXml.AddReference(reference);
|
|
|
|
signedXml.ComputeSignature();
|
|
Signature signature = signedXml.Signature;
|
|
Console.WriteLine("For the data to be signed the expected signature value is "
|
|
+ Encoding.Default.GetString(signature.SignatureValue));
|
|
}
|
|
|
|
}
|