macho/macho-go/pkg/ios/macho/lc_segment.go

package macho

import (
	"bytes"
	"encoding/binary"
	"fmt"
)

type Section interface {
	// SegName() []byte
	SectName() []byte
	Addr() uint64
	Size() uint64
	Offset() uint32
	Type() uint8
	Attribute() uint32
}

type Segment interface {
	SegName() []byte
	Vmaddr() uint64
	Vmsize() uint64
	Fileoff() uint64
	Filesize() uint64
	Flags() uint32
	Sections() []Section
}

type Segment32 struct {
	c        LoadCmd
	segname  []byte
	vmaddr   uint32
	vmsize   uint32
	fileoff  uint32
	filesize uint32
	maxprot  uint32
	initprot uint32
	nsects   uint32
	flags    uint32
	sections []*Section32
}

func (lcmd *Segment32) Cmd() uint32 {
	return lcmd.c.Cmd()
}

func (lcmd *Segment32) Cmdsize() uint32 {
	segment_size := uint32(56)
	section_size := uint32(4*9 + 16*2)
	return segment_size + section_size*lcmd.nsects
}

func (lcmd *Segment32) Cmdname() string {
	var s string
	s += fmt.Sprintf(
		"%s %s vm[vmaddr:0x%x vmsize:0x%x] [fileoff:0x%x filesize:0x%x]",
		lcmd.c.Cmdname(),
		string(lcmd.segname),
		lcmd.vmaddr,
		lcmd.vmsize,
		lcmd.fileoff,
		lcmd.filesize,
	)
	for i, section := range lcmd.sections {
		s += fmt.Sprintf("\n  %d. %s [addr:0x%x size:0x%x offset:0x%x]",
			i, string(section.sectname),
			section.addr, section.size, section.offset,
		)
	}
	return s
}

func (lcmd *Segment32) SegName() []byte {
	return lcmd.segname
}
func (lcmd *Segment32) Vmaddr() uint64 {
	return uint64(lcmd.vmaddr)
}
func (lcmd *Segment32) Vmsize() uint64 {
	return uint64(lcmd.vmaddr)
}
func (lcmd *Segment32) Fileoff() uint64 {
	return uint64(lcmd.fileoff)
}
func (lcmd *Segment32) Filesize() uint64 {
	return uint64(lcmd.filesize)
}
func (lcmd *Segment32) Flags() uint32 {
	return lcmd.flags
}
func (lcmd *Segment32) Sections() []Section {
	var sections []Section
	for _, section := range lcmd.sections {
		sections = append(sections, section)
	}
	return sections
}

func (lcmd *Segment32) Serialize(mc *MachoContext) []byte {
	buf := new(bytes.Buffer)
	binary.Write(buf, mc.byteorder, lcmd.Cmd())
	binary.Write(buf, mc.byteorder, lcmd.Cmdsize())
	buf.Write(lcmd.segname)
	binary.Write(buf, mc.byteorder, lcmd.vmaddr)
	binary.Write(buf, mc.byteorder, lcmd.vmsize)
	binary.Write(buf, mc.byteorder, lcmd.fileoff)
	binary.Write(buf, mc.byteorder, lcmd.filesize)
	binary.Write(buf, mc.byteorder, lcmd.maxprot)
	binary.Write(buf, mc.byteorder, lcmd.initprot)
	binary.Write(buf, mc.byteorder, lcmd.nsects)
	binary.Write(buf, mc.byteorder, lcmd.flags)
	for _, section := range lcmd.sections {
		binary.Write(buf, mc.byteorder, section.Serialize(mc))
	}
	return buf.Bytes()
}

func (lcmd *Segment32) Deserialize(mc *MachoContext, buf []byte) {
	r := bytes.NewBuffer(buf)
	binary.Read(r, mc.byteorder, &lcmd.c.cmd)
	binary.Read(r, mc.byteorder, &lcmd.c.cmdsize)
	lcmd.segname = r.Next(16)
	binary.Read(r, mc.byteorder, &lcmd.vmaddr)
	binary.Read(r, mc.byteorder, &lcmd.vmsize)
	binary.Read(r, mc.byteorder, &lcmd.fileoff)
	binary.Read(r, mc.byteorder, &lcmd.filesize)
	binary.Read(r, mc.byteorder, &lcmd.maxprot)
	binary.Read(r, mc.byteorder, &lcmd.initprot)
	binary.Read(r, mc.byteorder, &lcmd.nsects)
	binary.Read(r, mc.byteorder, &lcmd.flags)

	for i := uint32(0); i < lcmd.nsects; i++ {
		section_buf := make([]byte, 68)
		r.Read(section_buf)

		var section Section32
		section.Deserialize(mc, section_buf)
		lcmd.sections = append(lcmd.sections, &section)
	}
}

type Segment64 struct {
	c        LoadCmd
	segname  []byte
	vmaddr   uint64
	vmsize   uint64
	fileoff  uint64
	filesize uint64
	maxprot  uint32
	initprot uint32
	nsects   uint32
	flags    uint32
	sections []*Section64
}

func (lcmd *Segment64) Cmd() uint32 {
	return lcmd.c.Cmd()
}

func (lcmd *Segment64) Cmdsize() uint32 {
	segment_size := uint32(72)
	section_size := uint32(8*2 + 4*8 + 16*2)
	return segment_size + section_size*lcmd.nsects
}

func (lcmd *Segment64) Cmdname() string {
	var s string
	s += fmt.Sprintf(
		"%s %s vm[vmaddr:0x%x vmsize:0x%x] [fileoff:0x%x filesize:0x%x]",
		lcmd.c.Cmdname(),
		string(lcmd.segname),
		lcmd.vmaddr,
		lcmd.vmsize,
		lcmd.fileoff,
		lcmd.filesize,
	)
	for i, section := range lcmd.sections {
		s += fmt.Sprintf("\n  %d. %s [addr:0x%x size:0x%x offset:0x%x]",
			i, string(section.sectname),
			section.addr, section.size, section.offset,
		)
	}
	return s
}

func (lcmd *Segment64) SegName() []byte {
	return lcmd.segname
}
func (lcmd *Segment64) Vmaddr() uint64 {
	return lcmd.vmaddr
}
func (lcmd *Segment64) Vmsize() uint64 {
	return lcmd.vmsize
}
func (lcmd *Segment64) Fileoff() uint64 {
	return lcmd.fileoff
}
func (lcmd *Segment64) Filesize() uint64 {
	return lcmd.filesize
}
func (lcmd *Segment64) Flags() uint32 {
	return lcmd.flags
}
func (lcmd *Segment64) Sections() []Section {
	var sections []Section
	for _, section := range lcmd.sections {
		sections = append(sections, section)
	}
	return sections
}

func (lcmd *Segment64) Serialize(mc *MachoContext) []byte {
	buf := new(bytes.Buffer)
	binary.Write(buf, mc.byteorder, lcmd.Cmd())
	binary.Write(buf, mc.byteorder, lcmd.Cmdsize())
	buf.Write(lcmd.segname)
	binary.Write(buf, mc.byteorder, lcmd.vmaddr)
	binary.Write(buf, mc.byteorder, lcmd.vmsize)
	binary.Write(buf, mc.byteorder, lcmd.fileoff)
	binary.Write(buf, mc.byteorder, lcmd.filesize)
	binary.Write(buf, mc.byteorder, lcmd.maxprot)
	binary.Write(buf, mc.byteorder, lcmd.initprot)
	binary.Write(buf, mc.byteorder, lcmd.nsects)
	binary.Write(buf, mc.byteorder, lcmd.flags)
	for _, section := range lcmd.sections {
		binary.Write(buf, mc.byteorder, section.Serialize(mc))
	}
	return buf.Bytes()
}

func (lcmd *Segment64) Deserialize(mc *MachoContext, buf []byte) {
	r := bytes.NewBuffer(buf)
	binary.Read(r, mc.byteorder, &lcmd.c.cmd)
	binary.Read(r, mc.byteorder, &lcmd.c.cmdsize)
	lcmd.segname = r.Next(16)
	binary.Read(r, mc.byteorder, &lcmd.vmaddr)
	binary.Read(r, mc.byteorder, &lcmd.vmsize)
	binary.Read(r, mc.byteorder, &lcmd.fileoff)
	binary.Read(r, mc.byteorder, &lcmd.filesize)
	binary.Read(r, mc.byteorder, &lcmd.maxprot)
	binary.Read(r, mc.byteorder, &lcmd.initprot)
	binary.Read(r, mc.byteorder, &lcmd.nsects)
	binary.Read(r, mc.byteorder, &lcmd.flags)

	for i := uint32(0); i < lcmd.nsects; i++ {
		section_buf := make([]byte, 80)
		r.Read(section_buf)

		var section Section64
		section.Deserialize(mc, section_buf)
		lcmd.sections = append(lcmd.sections, &section)
	}
}

type Section32 struct {
	sectname  []byte
	segname   []byte
	addr      uint32
	size      uint32
	offset    uint32
	align     uint32
	reloff    uint32
	nreloc    uint32
	flags     uint32
	reserved1 uint32
	reserved2 uint32
}

func (sec *Section32) SectName() []byte {
	return sec.sectname
}
func (sec *Section32) Addr() uint64 {
	return uint64(sec.addr)
}
func (sec *Section32) Size() uint64 {
	return uint64(sec.size)
}
func (sec *Section32) Offset() uint32 {
	return sec.offset
}
func (sec *Section32) Type() uint8 {
	return uint8(sec.flags & 0xff)
}
func (sec *Section32) Attribute() uint32 {
	return sec.flags & 0xffffff00
}

func (section *Section32) Serialize(mc *MachoContext) []byte {
	buf := new(bytes.Buffer)
	buf.Write(section.sectname)
	buf.Write(section.segname)
	binary.Write(buf, mc.byteorder, section.addr)
	binary.Write(buf, mc.byteorder, section.size)
	binary.Write(buf, mc.byteorder, section.offset)
	binary.Write(buf, mc.byteorder, section.align)
	binary.Write(buf, mc.byteorder, section.reloff)
	binary.Write(buf, mc.byteorder, section.nreloc)
	binary.Write(buf, mc.byteorder, section.flags)
	binary.Write(buf, mc.byteorder, section.reserved1)
	binary.Write(buf, mc.byteorder, section.reserved2)
	return buf.Bytes()
}

func (section *Section32) Deserialize(mc *MachoContext, buf []byte) {
	r := bytes.NewBuffer(buf)
	section.sectname = r.Next(16)
	section.segname = r.Next(16)
	binary.Read(r, mc.byteorder, &section.addr)
	binary.Read(r, mc.byteorder, &section.size)
	binary.Read(r, mc.byteorder, &section.offset)
	binary.Read(r, mc.byteorder, &section.align)
	binary.Read(r, mc.byteorder, &section.reloff)
	binary.Read(r, mc.byteorder, &section.nreloc)
	binary.Read(r, mc.byteorder, &section.flags)
	binary.Read(r, mc.byteorder, &section.reserved1)
	binary.Read(r, mc.byteorder, &section.reserved2)
}

type Section64 struct {
	sectname  []byte
	segname   []byte
	addr      uint64
	size      uint64
	offset    uint32
	align     uint32
	reloff    uint32
	nreloc    uint32
	flags     uint32
	reserved1 uint32
	reserved2 uint32
	reserved3 uint32
}

func (sec *Section64) SectName() []byte {
	return sec.sectname
}
func (sec *Section64) Addr() uint64 {
	return sec.addr
}
func (sec *Section64) Size() uint64 {
	return sec.size
}
func (sec *Section64) Offset() uint32 {
	return sec.offset
}
func (sec *Section64) Type() uint8 {
	return uint8(sec.flags & 0xff)
}
func (sec *Section64) Attribute() uint32 {
	return sec.flags & 0xffffff00
}

func (section *Section64) Serialize(mc *MachoContext) []byte {
	buf := new(bytes.Buffer)
	buf.Write(section.sectname)
	buf.Write(section.segname)
	binary.Write(buf, mc.byteorder, section.addr)
	binary.Write(buf, mc.byteorder, section.size)
	binary.Write(buf, mc.byteorder, section.offset)
	binary.Write(buf, mc.byteorder, section.align)
	binary.Write(buf, mc.byteorder, section.reloff)
	binary.Write(buf, mc.byteorder, section.nreloc)
	binary.Write(buf, mc.byteorder, section.flags)
	binary.Write(buf, mc.byteorder, section.reserved1)
	binary.Write(buf, mc.byteorder, section.reserved2)
	binary.Write(buf, mc.byteorder, section.reserved3)
	return buf.Bytes()
}

func (section *Section64) Deserialize(mc *MachoContext, buf []byte) {
	r := bytes.NewBuffer(buf)
	section.sectname = r.Next(16)
	section.segname = r.Next(16)
	binary.Read(r, mc.byteorder, &section.addr)
	binary.Read(r, mc.byteorder, &section.size)
	binary.Read(r, mc.byteorder, &section.offset)
	binary.Read(r, mc.byteorder, &section.align)
	binary.Read(r, mc.byteorder, &section.reloff)
	binary.Read(r, mc.byteorder, &section.nreloc)
	binary.Read(r, mc.byteorder, &section.flags)
	binary.Read(r, mc.byteorder, &section.reserved1)
	binary.Read(r, mc.byteorder, &section.reserved2)
	binary.Read(r, mc.byteorder, &section.reserved3)
}