OptionsHelper.nim

import Structs
import AuxFunctions

proc PrintBanner*():void = 
    var banner = """
 
                                                                                             _..._     
                                                                                          .-'_..._''.  
   _..._   .--. __  __   ___         __.....__                          __.....__       .' .'      '.\ 
 .'     '. |__||  |/  `.'   `.   .-''         '.                    .-''         '.    / .'            
.   .-.   ..--.|   .-.  .-.   ' /     .-''"'-.  `.                 /     .-''"'-.  `. . '              
|  '   '  ||  ||  |  |  |  |  |/     /________\   \ ____     _____/     /________\   \| |              
|  |   |  ||  ||  |  |  |  |  ||                  |`.   \  .'    /|                  || |              
|  |   |  ||  ||  |  |  |  |  |\    .-------------'  `.  `'    .' \    .-------------'. '              
|  |   |  ||  ||  |  |  |  |  | \    '-.____...---.    '.    .'    \    '-.____...---. \ '.          . 
|  |   |  ||__||__|  |__|  |__|  `.             .'     .'     `.    `.             .'   '. `._____.-'/ 
|  |   |  |                        `''-...... -'     .'  .'`.   `.    `''-...... -'       `-.______ /  
|  |   |  |                                        .'   /    `.   `.                               `   
'--'   '--'                                       '----'       '----'                                  

                                            @R0h1rr1m          

"""
    echo banner


proc PrintHelp*():void =
    var optionsString = """

    -v | --verbose                          Enable more verbose output.
    -u | --username <Username>              Username for NTLM Authentication.* 
    -h | --hash <NTLM Hash>                 NTLM password hash for NTLM Authentication.**
    -p | --password <Password>              Plaintext password.**
    -t | --target <Target>                  Lateral movement target.*
    -c | --command <Command>                Command to execute.*
    -d | --domain <Domain>                  Domain name for NTLM Authentication.
    -s | --service <Service Name>           Name of the service instead of a random one.
    --help                                  Show the help message.
    """
    echo(optionsString)


proc ParseArgs*(argc:int, argv:seq[string], optionsStruct:ptr OPTIONS):bool =
    optionsStruct.Service = ""
    optionsStruct.Username = ""
    optionsStruct.Hash = ""
    optionsStruct.Domain = ""
    optionsStruct.Target = ""
    optionsStruct.Command = ""
    optionsStruct.IsVerbose = false
    optionsStruct.OutputUsername = ""
    var i:int = 0
    while(i<argc):
        if(argv[i] == "-v" or argv[i] == "--verbose"):
            optionsStruct.IsVerbose = true
        elif(argv[i] == "--help"):
            PrintHelp()
            quit(0)
        elif(argv[i] == "-u" or argv[i] == "--username"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Username = argv[i]
        elif(argv[i] == "-p" or argv[i] == "--password"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Hash = NtlmCalculator(argv[i])
        elif(argv[i] == "-h" or argv[i] == "--hash"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Hash = argv[i]
        elif(argv[i] == "-d" or argv[i] == "--domain"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Domain = argv[i]
        elif(argv[i] == "-t" or argv[i] == "--target"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Target = argv[i]
        elif(argv[i] == "-s" or argv[i] == "--service"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Service = argv[i]
        elif(argv[i] == "-c" or argv[i] == "--command"):
            i+=1
            if(i>=argc):
                return false
            optionsStruct.Command = argv[i]
        elif(i != 0):
            echo "[!] Unknown argument!"
            return false
        i+=1
    if(optionsStruct.Username == "" or optionsStruct.Hash == "" or optionsStruct.Target == "" or optionsStruct.Command == ""):
        echo "[!] Missing one or more arguments!"
        return false
    return true