prog: fix restoration of default arguments

Currently fails on:
 - pointers
 - VMAs
 - structs
 - fixed-size structs

1 files changed, 16 insertions, 0 deletions

diff --git a/prog/prog_test.go b/prog/prog_test.go
index 2d05058ea..369b8dc78 100644
--- a/prog/prog_test.go
+++ b/prog/prog_test.go
@@ -5,6 +5,7 @@ package prog
 
 import (
 	"bytes"
+	"fmt"
 	"math/rand"
 	"testing"
 	"time"
@@ -44,6 +45,21 @@ func TestDefault(t *testing.T) {
 	}
 }
 
+func TestDefaultCallArgs(t *testing.T) {
+	initTest(t)
+	for _, meta := range sys.CallMap {
+		// Ensure that we can restore all arguments of all calls.
+		prog := fmt.Sprintf("%v()", meta.Name)
+		p, err := Deserialize([]byte(prog))
+		if err != nil {
+			t.Fatalf("failed to restore default args in prog %q: %v", prog, err)
+		}
+		if len(p.Calls) != 1 || p.Calls[0].Meta.Name != meta.Name {
+			t.Fatalf("restored bad program from prog %q: %q", prog, p.Serialize())
+		}
+	}
+}
+
 func TestSerialize(t *testing.T) {
 	rs, iters := initTest(t)
 	for i := 0; i < iters; i++ {