njelib.py 文件源码

def test():
    """Test program for njelib.

    Usage: python njelib.py [-d] ... [host [port]] [RHOST OHOST]

    Default host is localhost; default port is 175.

    """
    debuglevel = 0

    while sys.argv[1:] and sys.argv[1] == '-d':
        debuglevel = debuglevel+1
        del sys.argv[1]

    host = 'localhost'
    if sys.argv[1:]:
        host = sys.argv[1]

    port = 175
    if sys.argv[2:]:
        portstr = sys.argv[2]
        try:
            port = int(portstr)
        except ValueError:
            port = socket.getservbyname(portstr, 'tcp')

    rhost = ohost = 'FAKE'
    if sys.argv[3:]:
        rhost = sys.argv[3]
        ohost = sys.argv[4]

    password = ''
    if sys.argv[5:]:
        password = sys.argv[5]

    nje = NJE(ohost,rhost)
    nje.set_debuglevel(debuglevel)
    t = nje.signon(host=host,port=port, timeout=2, password=password)

    if t:
        print "[+] Connection Successful"
    else:
        print "[!] Connection Failed"