/
become.go
88 lines (77 loc) · 1.42 KB
/
become.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
package main
import (
"fmt"
"io"
"io/ioutil"
"net"
"os"
"plugin"
"sync"
)
const (
LISTEN = "localhost:3333"
)
var (
wait sync.WaitGroup
)
type Become interface {
Become()
}
func main() {
l, err := net.Listen("tcp", LISTEN)
if err != nil {
fmt.Println("Error listening: ", err.Error())
os.Exit(1)
}
defer l.Close()
conn, err := l.Accept()
if err != nil {
fmt.Println("Error accepting: ", err.Error())
} else {
wait.Add(1)
go handleRequest(conn)
}
wait.Wait()
}
func handleRequest(conn net.Conn) {
defer wait.Done()
defer conn.Close()
file, err := ioutil.TempFile("", "become.*")
if err != nil {
fmt.Println("Error creating temporary file: ", err.Error())
return
}
_, err = io.Copy(file, conn)
if err != nil {
fmt.Println("Error copying: ", err.Error())
return
}
err = file.Close()
if err != nil {
fmt.Println("Error closing temporary file: ", err.Error())
return
}
wait.Add(1)
go becomePlugin(file.Name())
}
func becomePlugin(filename string) {
defer wait.Done()
defer os.Remove(filename)
plugin, err := plugin.Open(filename)
if err != nil {
fmt.Println("Error opening becoming: ", err.Error())
return
}
symBecoming, err := plugin.Lookup("Become")
if err != nil {
fmt.Println("Error looking up becoming: ", err.Error())
return
}
var becoming Become
becoming, ok := symBecoming.(Become)
if !ok {
fmt.Println("Becoming not of correct type")
return
}
becoming.Become()
}