/
rawcmd.go
44 lines (40 loc) · 1.09 KB
/
rawcmd.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package cmd
import (
"encoding/binary"
"fmt"
"github.com/lunixbochs/usercorn/go"
"github.com/lunixbochs/usercorn/go/loader"
"github.com/lunixbochs/usercorn/go/models"
)
func NewUsercornRawCmd() *UsercornCmd {
c := NewUsercornCmd()
var entry *uint64
var arch, osStr, endian *string
c.MakeUsercorn = func(exe string) (models.Usercorn, error) {
var byteOrder binary.ByteOrder
switch *endian {
case "little":
byteOrder = binary.LittleEndian
case "big":
byteOrder = binary.BigEndian
default:
return nil, fmt.Errorf("%s is not a valid byte order ('little' or 'big')", endian)
}
var err error
l := loader.NewNullLoader(*arch, *osStr, byteOrder, *entry)
u, err := usercorn.NewUsercornRaw(l, c.Config)
if err != nil {
return nil, err
}
u.SetEntry(*entry)
return u, nil
}
c.SetupFlags = func() error {
entry = c.Flags.Uint64("entry", 0, "entry point")
arch = c.Flags.String("arch", "x86", "target architecture")
osStr = c.Flags.String("os", "linux", "target OS")
endian = c.Flags.String("endian", "little", "'big' or 'little' endian")
return nil
}
return c
}