Perrrfect

Author: Seggan

src/main/java/io/github/seggan/javaclasslib/attributes/Attribute.java

@@ -7,9 +7,8 @@
 public abstract class Attribute {
 
     private final UTF8Entry nameIndex;
-    private final int length;
 
-    public Attribute(UTF8Entry nameIndex, String requiredName, int length) {
+    public Attribute(UTF8Entry nameIndex, String requiredName) {
         if (!new String(nameIndex.getData(), StandardCharsets.UTF_8).equals(requiredName)) {
             throw new IllegalArgumentException(String.format(
                 "Name of \"%s\" is not \"%s\"",
@@ -19,16 +18,11 @@ public Attribute(UTF8Entry nameIndex, String requiredName, int length) {
         }
 
         this.nameIndex = nameIndex;
-        this.length = length;
     }
 
     public abstract byte[] getBytes();
 
     public UTF8Entry getNameIndex() {
         return nameIndex;
     }
-
-    public int getLength() {
-        return length;
-    }
 }

src/main/java/io/github/seggan/javaclasslib/attributes/code/CodeAttribute.java

@@ -14,8 +14,8 @@ public final class CodeAttribute extends Attribute {
     private final int maxLocalVarSize;
     private final List<JvmInstructionImpl> code;
 
-    public CodeAttribute(UTF8Entry nameIndex, int length, int maxStackSize, int maxLocalVarSize, List<JvmInstructionImpl> code) {
-        super(nameIndex, "Code", length);
+    public CodeAttribute(UTF8Entry nameIndex, int maxStackSize, int maxLocalVarSize, List<JvmInstructionImpl> code) {
+        super(nameIndex, "Code");
         this.maxStackSize = maxStackSize;
         this.maxLocalVarSize = maxLocalVarSize;
         this.code = code;

src/main/java/io/github/seggan/javaclasslib/methods/Method.java

@@ -52,7 +52,7 @@ public MethodAccessFlags[] getAccessFlags() {
         return accessFlags;
     }
 
-    public void setAccessFlags(MethodAccessFlags[] accessFlags) {
+    public void setAccessFlags(MethodAccessFlags... accessFlags) {
         this.accessFlags = accessFlags;
     }
 }

src/test/java/io/github/seggan/javaclasslib/tests/Test.java

@@ -2,16 +2,36 @@
 
 import io.github.seggan.javaclasslib.ClassAccessFlags;
 import io.github.seggan.javaclasslib.JavaClass;
+import io.github.seggan.javaclasslib.attributes.code.CodeAttribute;
+import io.github.seggan.javaclasslib.attributes.code.instructions.JvmInstructions;
+import io.github.seggan.javaclasslib.constantpool.UTF8Entry;
+import io.github.seggan.javaclasslib.methods.Method;
+import io.github.seggan.javaclasslib.methods.MethodAccessFlags;
 
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.util.Arrays;
 
 public class Test {
 
     public static void main(String[] args) {
 
         JavaClass javaClass = new JavaClass("Testy", "java.lang.Object", 16);
         javaClass.setClassAccessFlags(ClassAccessFlags.PUBLIC, ClassAccessFlags.FINAL);
+
+        Method main = new Method(new UTF8Entry(javaClass.getConstantPool(), "main"),
+            new UTF8Entry(javaClass.getConstantPool(), "([Ljava/jang/String;)V"));
+        main.setAccessFlags(MethodAccessFlags.PUBLIC, MethodAccessFlags.STATIC);
+
+        CodeAttribute code = new CodeAttribute(new UTF8Entry(javaClass.getConstantPool(), "Code"),
+            2, 2, Arrays.asList(
+            JvmInstructions.ALOAD.create(0)
+        ));
+
+        main.getAttributes().add(code);
+
+        javaClass.getMethods().add(main);
+
         try (FileOutputStream outputStream = new FileOutputStream("Testy.class")) {
             javaClass.write(outputStream);
         } catch (IOException e) {