HellPot/internal/extra/banner.go

101 lines
2.9 KiB
Go

package extra
import (
crip "crypto/rand"
"encoding/binary"
"fmt"
"os"
"runtime"
"strings"
"time"
"git.tcp.direct/kayos/common/squish"
"github.com/yunginnanet/HellPot/internal/config"
)
const hellpot = "H4sIAAAAAAACA8VXvW7bQAze9QpZOGQUZNXntBD6Ahm7Gx1cx0jdRnKRKAUCdPDgQavOgB/QTxLZ1P3oRJ5Obo0CtnE5feSR30fylOhmfjv9PEtzwIXIj4dds/xw2jsequNB2gizXd3Mxad2O81PX7AAe+UNGneuR8aUOuTsqQUDXAMv1cJE5Tfbn6GaKz45kpid+lQc3zoNY5zmEUEt+jCGNZUjeYr0StZYmbwtwNavuCaUFWA8MjxVIImjNas6TPQT9Tnq4MnYJF0zkhVU4rLvqflscU/ox0Lg45qKTjoSmiLQPA+ZuTT7BbrckpfWKMkUquTErIPEYbPoKjamy6SjR0feGssPUMYTCDWEnrR8c0m7hJ2B4jekK2KUsBfa7bpTD0ftnmKPE9nN2IzcLc99vxhIUbszlwqrJoklpQWlI6AeQh9nDHXj2ldOvyat/vZdDxVfzZdbSuspRUe/+IKZtxq2GWlbZzS6jnrnDEXGCkXUGnahuTgAA+DY9HU8FUoYH3ji/q84HetDWmT/Y3ml6oX21/eCNzB46+6UuVTSQHXgGmzUTJT/zeNQ3zCvysEBuH3hER9CbhNa6FoLHSBfT2gmK/rFKCj/K1nTfcBduKHVwgjo+Y+HilXBEAqhKg1X6lQzMaIF6ZK6ipVILR0Awh16SWy9KsxvZXWbL34oGpNmMcPNdYFmiE40+qV9cg4Logjm2uXjukzK5a/kYf28WpaTn4u3zcvkfvX09GVTnuFfEYzBNujvr9+S5SafvL0Wj+uiWBSrsov/I6axmMXiLhYf40zE2TTOZnF2F2fNn2n0DpcvBxhQEAAA"
func rc(s []string) string {
return strings.TrimSpace(s[ru()%uint32(len(s))])
}
func process(in string) (s string) {
var v = strings.Split(config.Version, "")
var maj, min, smin = "", "", ""
if len(config.Version) > 0 {
maj = v[0]
}
if len(config.Version) > 2 {
min = v[2]
}
if len(config.Version) > 4 {
smin = v[4]
}
defl8, _ := squish.UnpackStr(in)
sp := strings.Split(defl8, "|")
s = sp[0]
if smin == "" || len(config.Version) == 7 || config.Version == "dev" {
s = strings.ReplaceAll(s, "$1;40m.", "$1;40m")
if len(config.Version) == 7 || config.Version == "dev" {
s = strings.ReplaceAll(s, "$3;40m.", "$3;40m")
}
}
c := strings.Split(sp[1], ",")
cproc := func(in, num string) (inr string) {
inr = in
tor := fmt.Sprintf("$%s", num)
for n := strings.Count(inr, tor); n > 0; n-- {
inr = strings.Replace(inr, tor, rc(c), 1)
}
return
}
for n := 1; n < 5; n++ {
s = cproc(s, fmt.Sprintf("%d", n))
}
if len(config.Version) == 7 || config.Version == "dev" {
maj = "[" + config.Version + "]"
min = ""
smin = ""
}
s = strings.ReplaceAll(s, "$maj", maj)
s = strings.ReplaceAll(s, "$min", min)
s = strings.ReplaceAll(s, "$smin", smin)
return
}
func ru() uint32 {
b := make([]byte, 8192)
if _, err := crip.Read(b); err != nil {
bannerFail(err)
}
return binary.LittleEndian.Uint32(b)
}
// printBanner prints our entropic banner
func printBanner() {
time.Sleep(5 * time.Millisecond)
println("\n" + process(hellpot) + "\n\n")
time.Sleep(5 * time.Millisecond)
}
func bannerFail(errs ...error) {
println("failed printing banner, consider using --nocolor")
for _, err := range errs {
if err != nil {
println(err.Error())
}
}
os.Exit(1)
}
// Banner prints out our banner (using spooky magic)
func Banner() {
//goland:noinspection GoBoolExpressions
if runtime.GOOS == "windows" || config.NoColor {
println(config.Title + " " + config.Version)
return
}
printBanner()
}