-
Notifications
You must be signed in to change notification settings - Fork 33
/
module.go
165 lines (130 loc) · 3.14 KB
/
module.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
package main
import (
"bufio"
"fmt"
"io/ioutil"
"os"
"os/exec"
)
type IModule interface {
compile() error
setSource(string)
getName() string
getLanguage() string
getSourceFromDisk() string
getShellcode() []byte
writeToDisk()
}
type Module struct {
Name string
Source string
Language string
extension string
}
func newModule(name string, source string, language string) *Module {
var extension string
if language == "Go" {
extension = ".go"
} else if language == "C#" {
extension = ".cs"
}
return &Module{name, source, language, extension}
}
func (mod Module) getShellcode() []byte {
fileName, err := mod.compile(false)
shellcodeFileName := "modules/" + mod.Name + ".bin"
if err != nil {
info("Failed to compile module: " + err.Error())
return []byte{}
}
output := ""
cmdHandle := exec.Command("/bin/sh", "-c", "./includes/donut -x 1 -c TestModule2 -m Main "+fileName+" -o "+shellcodeFileName)
stdout, err := cmdHandle.StdoutPipe()
stderr, err := cmdHandle.StderrPipe()
if err = cmdHandle.Start(); err == nil {
scanner := bufio.NewScanner(stdout)
for scanner.Scan() {
output += scanner.Text()
output += "\n"
}
scanner = bufio.NewScanner(stderr)
for scanner.Scan() {
output += scanner.Text()
output += "\n"
}
}
info(output)
data, err := ioutil.ReadFile(shellcodeFileName)
//os.Remove(fileName)
//os.Remove(shellcodeFileName)
if err != nil {
return []byte{}
} else {
return data
}
}
func (mod Module) writeToDisk() {
ioutil.WriteFile("modules/"+mod.Name+mod.extension, []byte(mod.Source), 0644)
}
func (mod Module) setSource(source string) {
mod.Source = source
}
func (mod Module) getSourceFromDisk() string {
source, err := ioutil.ReadFile("modules/" + mod.Name + mod.extension)
if err != nil {
return ""
} else {
mod.Source = string(source)
return mod.Source
}
}
func (mod Module) getLanguage() string {
return mod.Language
}
func (mod Module) getName() string {
return mod.Name
}
func (mod Module) compile(delete bool) (string, error) {
var cmdHandle *exec.Cmd
filename := "/tmp/" + genRandID()
outfile := "modules/" + mod.Name + ".exe"
err := ioutil.WriteFile(filename, []byte(mod.Source), 0644)
fmt.Println(mod.Language)
if mod.Language == "C#" {
cmdHandle = exec.Command("/bin/sh", "-c", "mcs -out:"+outfile+" "+filename)
} else if mod.Language == "Go" {
cmdHandle = exec.Command("/bin/sh", "-c", "env CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -ldflags '-s -w' -o "+outfile+" modules/"+mod.Name+".go")
}
stderr, err := cmdHandle.StderrPipe()
stdout, err := cmdHandle.StdoutPipe()
output := ""
if err = cmdHandle.Start(); err == nil {
scanner := bufio.NewScanner(stderr)
if err != nil {
output += scanner.Text()
output += "\n"
}
for scanner.Scan() {
output += scanner.Text()
output += "\n"
}
scanner = bufio.NewScanner(stdout)
if err != nil {
output += scanner.Text()
output += "\n"
}
for scanner.Scan() {
output += scanner.Text()
output += "\n"
}
}
os.Remove(filename)
if delete {
os.Remove(outfile)
}
if len(output) == 0 {
return outfile, nil
} else {
return outfile, fmt.Errorf("%s", output)
}
}