1 files changed, 161 insertions, 0 deletions
diff --git a/src/cmd/link/prog_test.go b/src/cmd/link/prog_test.go
new file mode 100644
index 0000000000..8229b5b91f
--- /dev/null
+++ b/src/cmd/link/prog_test.go
@@ -0,0 +1,161 @@
+// Copyright 2014 The Go Authors.  All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"testing"
+)
+
+// shiftProg adjusts the addresses in p.
+// It adds vdelta to all virtual addresses and fdelta to all file offsets.
+func shiftProg(p *Prog, vdelta Addr, fdelta Addr) {
+	p.Entry += vdelta
+	for _, seg := range p.Segments {
+		seg.FileOffset += fdelta
+		seg.VirtAddr += vdelta
+		for _, sect := range seg.Sections {
+			sect.VirtAddr += vdelta
+			for _, sym := range sect.Syms {
+				sym.Addr += vdelta
+			}
+		}
+	}
+}
+
+// diffProg returns a list of differences between p and q,
+// assuming p is being checked and q is the correct answer.
+func diffProg(p, q *Prog) []string {
+	var errors []string
+	if p.UnmappedSize != q.UnmappedSize {
+		errors = append(errors, fmt.Sprintf("p.UnmappedSize = %#x, want %#x", p.UnmappedSize, q.UnmappedSize))
+	}
+	if p.HeaderSize != q.HeaderSize {
+		errors = append(errors, fmt.Sprintf("p.HeaderSize = %#x, want %#x", p.HeaderSize, q.HeaderSize))
+	}
+	if p.Entry != q.Entry {
+		errors = append(errors, fmt.Sprintf("p.Entry = %#x, want %#x", p.Entry, q.Entry))
+	}
+	for i := 0; i < len(p.Segments) || i < len(q.Segments); i++ {
+		if i >= len(p.Segments) {
+			errors = append(errors, fmt.Sprintf("p missing segment %q", q.Segments[i].Name))
+			continue
+		}
+		if i >= len(q.Segments) {
+			errors = append(errors, fmt.Sprintf("p has extra segment %q", p.Segments[i].Name))
+			continue
+		}
+		pseg := p.Segments[i]
+		qseg := q.Segments[i]
+		if pseg.Name != qseg.Name {
+			errors = append(errors, fmt.Sprintf("segment %d Name = %q, want %q", i, pseg.Name, qseg.Name))
+			continue // probably out of sync
+		}
+		if pseg.VirtAddr != qseg.VirtAddr {
+			errors = append(errors, fmt.Sprintf("segment %q VirtAddr = %#x, want %#x", pseg.Name, pseg.VirtAddr, qseg.VirtAddr))
+		}
+		if pseg.VirtSize != qseg.VirtSize {
+			errors = append(errors, fmt.Sprintf("segment %q VirtSize = %#x, want %#x", pseg.Name, pseg.VirtSize, qseg.VirtSize))
+		}
+		if pseg.FileOffset != qseg.FileOffset {
+			errors = append(errors, fmt.Sprintf("segment %q FileOffset = %#x, want %#x", pseg.Name, pseg.FileOffset, qseg.FileOffset))
+		}
+		if pseg.FileSize != qseg.FileSize {
+			errors = append(errors, fmt.Sprintf("segment %q FileSize = %#x, want %#x", pseg.Name, pseg.FileSize, qseg.FileSize))
+		}
+		if len(pseg.Data) != len(qseg.Data) {
+			errors = append(errors, fmt.Sprintf("segment %q len(Data) = %d, want %d", pseg.Name, len(pseg.Data), len(qseg.Data)))
+		} else if !bytes.Equal(pseg.Data, qseg.Data) {
+			errors = append(errors, fmt.Sprintf("segment %q Data mismatch:\n\thave %x\n\twant %x", pseg.Name, pseg.Data, qseg.Data))
+		}
+
+		for j := 0; j < len(pseg.Sections) || j < len(qseg.Sections); j++ {
+			if j >= len(pseg.Sections) {
+				errors = append(errors, fmt.Sprintf("segment %q missing section %q", qseg.Sections[i].Name))
+				continue
+			}
+			if j >= len(qseg.Sections) {
+				errors = append(errors, fmt.Sprintf("segment %q has extra section %q", pseg.Sections[i].Name))
+				continue
+			}
+			psect := pseg.Sections[j]
+			qsect := qseg.Sections[j]
+			if psect.Name != qsect.Name {
+				errors = append(errors, fmt.Sprintf("segment %q, section %d Name = %q, want %q", pseg.Name, j, psect.Name, qsect.Name))
+				continue // probably out of sync
+			}
+
+			if psect.VirtAddr != qsect.VirtAddr {
+				errors = append(errors, fmt.Sprintf("segment %q section %q VirtAddr = %#x, want %#x", pseg.Name, psect.Name, psect.VirtAddr, qsect.VirtAddr))
+			}
+			if psect.Size != qsect.Size {
+				errors = append(errors, fmt.Sprintf("segment %q section %q Size = %#x, want %#x", pseg.Name, psect.Name, psect.Size, qsect.Size))
+			}
+			if psect.Align != qsect.Align {
+				errors = append(errors, fmt.Sprintf("segment %q section %q Align = %#x, want %#x", pseg.Name, psect.Name, psect.Align, qsect.Align))
+			}
+		}
+	}
+
+	return errors
+}
+
+// cloneProg returns a deep copy of p.
+func cloneProg(p *Prog) *Prog {
+	q := new(Prog)
+	*q = *p
+	q.Segments = make([]*Segment, len(p.Segments))
+	for i, seg := range p.Segments {
+		q.Segments[i] = cloneSegment(seg)
+	}
+	return p
+}
+
+// cloneSegment returns a deep copy of seg.
+func cloneSegment(seg *Segment) *Segment {
+	t := new(Segment)
+	*t = *seg
+	t.Sections = make([]*Section, len(seg.Sections))
+	for i, sect := range seg.Sections {
+		t.Sections[i] = cloneSection(sect)
+	}
+	t.Data = make([]byte, len(seg.Data))
+	copy(t.Data, seg.Data)
+	return t
+}
+
+// cloneSection returns a deep copy of section.
+func cloneSection(sect *Section) *Section {
+	// At the moment, there's nothing we need to make a deep copy of.
+	t := new(Section)
+	*t = *sect
+	return t
+}
+
+// checkGolden checks that data matches the named file.
+// If not, it reports the error to the test.
+func checkGolden(t *testing.T, data []byte, name string) {
+	golden, err := ioutil.ReadFile(name)
+	if err != nil {
+		t.Errorf("%s: %v", name, err)
+		return
+	}
+	if !bytes.Equal(data, golden) {
+		// TODO(rsc): A better diff would be nice, as needed.
+		i := 0
+		for i < len(data) && i < len(golden) && data[i] == golden[i] {
+			i++
+		}
+		if i >= len(data) {
+			t.Errorf("%s: output file shorter than expected: have %d bytes, want %d", name, len(data), len(golden))
+		} else if i >= len(golden) {
+			t.Errorf("%s: output file larger than expected: have %d bytes, want %d", name, len(data), len(golden))
+		} else {
+			t.Errorf("%s: output file differs at byte %d: have %#02x, want %#02x", name, i, data[i], golden[i])
+		}
+	}
+}