mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-11 09:30:59 +01:00
134 lines
2.9 KiB
Go
134 lines
2.9 KiB
Go
|
package ebpf
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/cilium/ebpf/asm"
|
||
|
"github.com/cilium/ebpf/internal/btf"
|
||
|
)
|
||
|
|
||
|
// link resolves bpf-to-bpf calls.
|
||
|
//
|
||
|
// Each library may contain multiple functions / labels, and is only linked
|
||
|
// if prog references one of these functions.
|
||
|
//
|
||
|
// Libraries also linked.
|
||
|
func link(prog *ProgramSpec, libs []*ProgramSpec) error {
|
||
|
var (
|
||
|
linked = make(map[*ProgramSpec]bool)
|
||
|
pending = []asm.Instructions{prog.Instructions}
|
||
|
insns asm.Instructions
|
||
|
)
|
||
|
for len(pending) > 0 {
|
||
|
insns, pending = pending[0], pending[1:]
|
||
|
for _, lib := range libs {
|
||
|
if linked[lib] {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
needed, err := needSection(insns, lib.Instructions)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("linking %s: %w", lib.Name, err)
|
||
|
}
|
||
|
|
||
|
if !needed {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
linked[lib] = true
|
||
|
prog.Instructions = append(prog.Instructions, lib.Instructions...)
|
||
|
pending = append(pending, lib.Instructions)
|
||
|
|
||
|
if prog.BTF != nil && lib.BTF != nil {
|
||
|
if err := btf.ProgramAppend(prog.BTF, lib.BTF); err != nil {
|
||
|
return fmt.Errorf("linking BTF of %s: %w", lib.Name, err)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func needSection(insns, section asm.Instructions) (bool, error) {
|
||
|
// A map of symbols to the libraries which contain them.
|
||
|
symbols, err := section.SymbolOffsets()
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
|
||
|
for _, ins := range insns {
|
||
|
if ins.Reference == "" {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if ins.OpCode.JumpOp() != asm.Call || ins.Src != asm.PseudoCall {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if ins.Constant != -1 {
|
||
|
// This is already a valid call, no need to link again.
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if _, ok := symbols[ins.Reference]; !ok {
|
||
|
// Symbol isn't available in this section
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
// At this point we know that at least one function in the
|
||
|
// library is called from insns, so we have to link it.
|
||
|
return true, nil
|
||
|
}
|
||
|
|
||
|
// None of the functions in the section are called.
|
||
|
return false, nil
|
||
|
}
|
||
|
|
||
|
func fixupJumpsAndCalls(insns asm.Instructions) error {
|
||
|
symbolOffsets := make(map[string]asm.RawInstructionOffset)
|
||
|
iter := insns.Iterate()
|
||
|
for iter.Next() {
|
||
|
ins := iter.Ins
|
||
|
|
||
|
if ins.Symbol == "" {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if _, ok := symbolOffsets[ins.Symbol]; ok {
|
||
|
return fmt.Errorf("duplicate symbol %s", ins.Symbol)
|
||
|
}
|
||
|
|
||
|
symbolOffsets[ins.Symbol] = iter.Offset
|
||
|
}
|
||
|
|
||
|
iter = insns.Iterate()
|
||
|
for iter.Next() {
|
||
|
i := iter.Index
|
||
|
offset := iter.Offset
|
||
|
ins := iter.Ins
|
||
|
|
||
|
switch {
|
||
|
case ins.IsFunctionCall() && ins.Constant == -1:
|
||
|
// Rewrite bpf to bpf call
|
||
|
callOffset, ok := symbolOffsets[ins.Reference]
|
||
|
if !ok {
|
||
|
return fmt.Errorf("instruction %d: reference to missing symbol %q", i, ins.Reference)
|
||
|
}
|
||
|
|
||
|
ins.Constant = int64(callOffset - offset - 1)
|
||
|
|
||
|
case ins.OpCode.Class() == asm.JumpClass && ins.Offset == -1:
|
||
|
// Rewrite jump to label
|
||
|
jumpOffset, ok := symbolOffsets[ins.Reference]
|
||
|
if !ok {
|
||
|
return fmt.Errorf("instruction %d: reference to missing symbol %q", i, ins.Reference)
|
||
|
}
|
||
|
|
||
|
ins.Offset = int16(jumpOffset - offset - 1)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|