nub31/nub-lang-archive-2
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Found issue with stack allocated structs and possibly arrays. Must be investigated further

Browse Source
This commit is contained in:
nub31
2025-01-31 23:48:30 +01:00
parent 16a031823e
commit 8c21483f74
6 changed files with 172 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
Nub.Lang/Nub.Lang/Backend/Custom/Generator.cs
View File

@@ -354,6 +354,9 @@ public class Generator
case StructInitializerNode structInitializer:
GenerateStructInitializer(structInitializer, func);
break;
case StructMemberAccessorNode structMemberAccessor:
GenerateStructMemberAccessor(structMemberAccessor, func);
break;
case SyscallExpressionNode syscallExpression:
GenerateSyscall(syscallExpression.Syscall, func);
break;
@@ -362,6 +365,45 @@ public class Generator
}
}
private void GenerateStructMemberAccessor(StructMemberAccessorNode structMemberAccessor, LocalFunc func)
{
var variable = func.ResolveLocalVariable(structMemberAccessor.Members[0]);
if (variable.Type is not StructType structType)
{
throw new Exception($"Cannot access struct member on {variable} since it is not a struct type");
}
_builder.AppendLine($" mov rax, [rbp - {variable.Offset}]");
_builder.AppendLine(" mov rax, [rax]");
Type prevMemberType = structType;
for (var i = 1; i < structMemberAccessor.Members.Count; i++)
{
if (prevMemberType is not StructType prevMemberStructType)
{
throw new Exception($"Cannot access {structMemberAccessor.Members[i]} on type {prevMemberType} because it is not a struct type");
}
var structDefinition = _definitions.OfType<StructDefinitionNode>().FirstOrDefault(sd => sd.Name == prevMemberStructType.Name);
if (structDefinition == null)
{
throw new Exception($"Struct {prevMemberStructType} is not defined");
}
var member = structDefinition.Members.FirstOrDefault(m => m.Name == structMemberAccessor.Members[i]);
if (member == null)
{
throw new Exception($"Struct {prevMemberStructType} has no member with name {structMemberAccessor.Members[i]}");
}
var offset = structDefinition.Members.IndexOf(member);
_builder.AppendLine($" mov rax, [rax + {offset * 8}]");
prevMemberType = member.Type;
}
}
private void GenerateArrayIndexAccess(ArrayIndexAccessNode arrayIndexAccess, LocalFunc func)
{
GenerateArrayIndexPointerAccess(arrayIndexAccess.Identifier, arrayIndexAccess.Index, func);
@@ -601,7 +643,7 @@ public class Generator
throw new Exception($"Struct {structInitializer.StructType} is not defined");
}
_builder.AppendLine($" add rsp, {structDefinition.Members.Count * 8}");
_builder.AppendLine($" sub rsp, {structDefinition.Members.Count * 8}");
foreach (var initializer in structInitializer.Initializers)
{

2
Nub.Lang/Nub.Lang/Backend/Custom/SymbolTable.cs
View File

@@ -137,6 +137,8 @@ public abstract class Variable(string name, Type type)
{
public string Name { get; } = name;
public Type Type { get; } = type;
public override string ToString() => $"{Name}: {Type}";
}
public class LocalVariable(string name, Type type, int offset) : Variable(name, type)

63
Nub.Lang/Nub.Lang/Frontend/Parsing/Parser.cs
View File

@@ -423,28 +423,61 @@ public class Parser
private ExpressionNode ParseExpressionIdentifier(IdentifierToken identifier)
{
if (TryExpectSymbol(Symbol.OpenBracket))
var token = Peek();
if (!token.HasValue)
{
var index = ParseExpression();
ExpectSymbol(Symbol.CloseBracket);
return new ArrayIndexAccessNode(new IdentifierNode(identifier.Value), index);
return new IdentifierNode(identifier.Value);
}
if (TryExpectSymbol(Symbol.OpenParen))
switch (token.Value)
{
List<ExpressionNode> parameters = [];
while (!TryExpectSymbol(Symbol.CloseParen))
case SymbolToken symbolToken:
{
parameters.Add(ParseExpression());
TryExpectSymbol(Symbol.Comma);
}
switch (symbolToken.Symbol)
{
case Symbol.Period:
{
Next();
List<string> members =
[
identifier.Value,
ExpectIdentifier().Value
];
if (identifier.Value == "syscall")
{
return new SyscallExpressionNode(new Syscall(parameters));
}
while (TryExpectSymbol(Symbol.Period))
{
members.Add(ExpectIdentifier().Value);
}
return new FuncCallExpressionNode(new FuncCall(identifier.Value, parameters));
return new StructMemberAccessorNode(members);
}
case Symbol.OpenBracket:
{
Next();
var index = ParseExpression();
ExpectSymbol(Symbol.CloseBracket);
return new ArrayIndexAccessNode(new IdentifierNode(identifier.Value), index);
}
case Symbol.OpenParen:
{
Next();
List<ExpressionNode> parameters = [];
while (!TryExpectSymbol(Symbol.CloseParen))
{
parameters.Add(ParseExpression());
TryExpectSymbol(Symbol.Comma);
}
if (identifier.Value == "syscall")
{
return new SyscallExpressionNode(new Syscall(parameters));
}
return new FuncCallExpressionNode(new FuncCall(identifier.Value, parameters));
}
}
break;
}
}
return new IdentifierNode(identifier.Value);

6
Nub.Lang/Nub.Lang/Frontend/Parsing/StructMemberAccessorNode.cs Normal file
View File

@@ -0,0 +1,6 @@
namespace Nub.Lang.Frontend.Parsing;
public class StructMemberAccessorNode(List<string> members) : ExpressionNode
{
public List<string> Members { get; } = members;
}

58
Nub.Lang/Nub.Lang/Frontend/Typing/ExpressionTyper.cs
View File

@@ -15,7 +15,7 @@ public class ExpressionTyper
{
private readonly List<Func> _functions;
private readonly List<GlobalVariableDefinitionNode> _variableDefinitions;
private readonly List<StructDefinitionNode> _classes;
private readonly List<StructDefinitionNode> _structDefinitions;
private readonly Stack<Variable> _variables;
public ExpressionTyper(List<DefinitionNode> definitions)
@@ -24,7 +24,7 @@ public class ExpressionTyper
_functions = [];
_variableDefinitions = [];
_classes = definitions.OfType<StructDefinitionNode>().ToList();
_structDefinitions = definitions.OfType<StructDefinitionNode>().ToList();
var functions = definitions
.OfType<LocalFuncDefinitionNode>()
@@ -45,7 +45,7 @@ public class ExpressionTyper
{
_variables.Clear();
foreach (var @class in _classes)
foreach (var @class in _structDefinitions)
{
foreach (var variable in @class.Members)
{
@@ -216,6 +216,9 @@ public class ExpressionTyper
case StructInitializerNode structInitializer:
PopulateStructInitializer(structInitializer);
break;
case StructMemberAccessorNode structMemberAccessor:
GenerateStructMemberAccessorNode(structMemberAccessor);
break;
case SyscallExpressionNode syscall:
PopulateSyscallExpression(syscall);
break;
@@ -323,6 +326,55 @@ public class ExpressionTyper
structInitializer.Type = structInitializer.StructType;
}
// TODO: Fix this ugly ass code
private void GenerateStructMemberAccessorNode(StructMemberAccessorNode structMemberAccessor)
{
var variable = _variables.FirstOrDefault(v => v.Name == structMemberAccessor.Members[0]);
if (variable == null)
{
throw new Exception($"Variable {structMemberAccessor.Members[0]} is not defined");
}
if (variable.Type is not StructType variableType)
{
throw new Exception("Variable " + structMemberAccessor.Members[0] + " is not a struct");
}
var definition = _structDefinitions.FirstOrDefault(sd => sd.Name == variableType.Name);
if (definition == null)
{
throw new Exception($"Struct {structMemberAccessor.Members[0]} is not defined");
}
for (var i = 1; i < structMemberAccessor.Members.Count - 1; i++)
{
var member = definition.Members.FirstOrDefault(m => m.Name == structMemberAccessor.Members[i]);
if (member == null)
{
throw new Exception($"Member {structMemberAccessor.Members[i]} does not exist on struct {definition.Name}");
}
if (member.Type is not StructType memberType)
{
throw new Exception($"Member {structMemberAccessor.Members[i]} on struct {definition.Name} is not a struct");
}
definition = _structDefinitions.FirstOrDefault(sd => sd.Name == memberType.Name);
if (definition == null)
{
throw new Exception($"Struct {structMemberAccessor.Members[i]} is not defined");
}
}
var tmp = definition.Members.FirstOrDefault(m => m.Name == structMemberAccessor.Members.Last());
if (tmp == null)
{
throw new Exception($"Member {structMemberAccessor.Members.Last()} does not exist on struct {definition.Name}");
}
structMemberAccessor.Type = tmp.Type;
}
private void PopulateSyscallExpression(SyscallExpressionNode syscall)
{
foreach (var parameter in syscall.Syscall.Parameters)

15
input/program.nub
View File

@@ -3,12 +3,25 @@ import "core";
func main() {
let x = new Test
{
some_string = "test2",
some_struct = new Test
{
some_struct = 0,
some_int = 420,
some_string = "blaze it"
},
some_int = 69,
some_string = "nice"
};
println(x.some_int);
println(x.some_string);
println(x.some_struct.some_int);
println(x.some_struct.some_string);
}
struct Test {
let some_struct: Test;
let some_string: String;
let some_int: int64;
}