Validate input
This commit is contained in:
@@ -68,10 +68,8 @@ public class Generator
|
|||||||
_builder.AppendLine($" mov [{symbol.Identifier}], rax");
|
_builder.AppendLine($" mov [{symbol.Identifier}], rax");
|
||||||
}
|
}
|
||||||
|
|
||||||
_builder.AppendLine();
|
|
||||||
_builder.AppendLine($" call {main.StartLabel}");
|
_builder.AppendLine($" call {main.StartLabel}");
|
||||||
|
|
||||||
_builder.AppendLine();
|
|
||||||
_builder.AppendLine(main.ReturnType.HasValue
|
_builder.AppendLine(main.ReturnType.HasValue
|
||||||
? " mov rdi, rax"
|
? " mov rdi, rax"
|
||||||
: " mov rdi, 0");
|
: " mov rdi, 0");
|
||||||
|
|||||||
@@ -10,10 +10,38 @@ internal static class Program
|
|||||||
private static readonly Lexer Lexer = new();
|
private static readonly Lexer Lexer = new();
|
||||||
private static readonly Parser Parser = new();
|
private static readonly Parser Parser = new();
|
||||||
|
|
||||||
public static void Main(string[] args)
|
public static int Main(string[] args)
|
||||||
{
|
{
|
||||||
var modules = RunFrontend(args[0]);
|
if (args.Length != 2)
|
||||||
|
{
|
||||||
|
Console.WriteLine("Usage: nub <input-dir> <output-file>");
|
||||||
|
Console.WriteLine("Example: nub src out.asm");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
var input = args[0];
|
||||||
|
var output = args[1];
|
||||||
|
|
||||||
|
if (!Directory.Exists(input))
|
||||||
|
{
|
||||||
|
Console.WriteLine($"Error: Input directory '{input}' does not exist.");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
var outputDir = Path.GetDirectoryName(output);
|
||||||
|
if (outputDir == null || !Directory.Exists(outputDir))
|
||||||
|
{
|
||||||
|
Console.WriteLine($"Error: Output directory '{outputDir}' does not exist.");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (string.IsNullOrWhiteSpace(Path.GetFileName(output)))
|
||||||
|
{
|
||||||
|
Console.WriteLine("Error: Output path must specify a file, not a directory.");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
var modules = RunFrontend(input);
|
||||||
var definitions = modules.SelectMany(f => f.Definitions).ToArray();
|
var definitions = modules.SelectMany(f => f.Definitions).ToArray();
|
||||||
|
|
||||||
var typer = new ExpressionTyper(definitions);
|
var typer = new ExpressionTyper(definitions);
|
||||||
@@ -22,9 +50,8 @@ internal static class Program
|
|||||||
var generator = new Generator(definitions);
|
var generator = new Generator(definitions);
|
||||||
var asm = generator.Generate();
|
var asm = generator.Generate();
|
||||||
|
|
||||||
Console.WriteLine(asm);
|
File.WriteAllText(output, asm);
|
||||||
|
return 0;
|
||||||
File.WriteAllText(args[1], asm);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private static IEnumerable<ModuleNode> RunFrontend(string path)
|
private static IEnumerable<ModuleNode> RunFrontend(string path)
|
||||||
|
|||||||
Reference in New Issue
Block a user