// 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])
		}
	}
}