diff options
Diffstat (limited to 'daemons')
-rw-r--r-- | daemons/ipa-otpd/test.py | 36 |
1 files changed, 21 insertions, 15 deletions
diff --git a/daemons/ipa-otpd/test.py b/daemons/ipa-otpd/test.py index 0ccd8b0e4..ea55f9b2a 100644 --- a/daemons/ipa-otpd/test.py +++ b/daemons/ipa-otpd/test.py @@ -41,22 +41,28 @@ ATTRIBUTE User-Password 2 string ATTRIBUTE NAS-Identifier 32 string """ -dct = Dictionary(StringIO(DICTIONARY)) -proc = subprocess.Popen(["./ipa-otpd", sys.argv[1]], - stdin=subprocess.PIPE, stdout=subprocess.PIPE) +def main(): + dct = Dictionary(StringIO(DICTIONARY)) -pkt = packet.AuthPacket(secret="", dict=dct) -pkt["User-Name"] = sys.argv[2] -pkt["User-Password"] = pkt.PwCrypt(sys.argv[3]) -pkt["NAS-Identifier"] = "localhost" -proc.stdin.write(pkt.RequestPacket()) + proc = subprocess.Popen(["./ipa-otpd", sys.argv[1]], + stdin=subprocess.PIPE, + stdout=subprocess.PIPE) -rsp = packet.Packet(secret="", dict=dict) -buf = proc.stdout.read(4) -buf += proc.stdout.read(struct.unpack("!BBH", buf)[2] - 4) -rsp.DecodePacket(buf) -pkt.VerifyReply(rsp) + pkt = packet.AuthPacket(secret="", dict=dct) + pkt["User-Name"] = sys.argv[2] + pkt["User-Password"] = pkt.PwCrypt(sys.argv[3]) + pkt["NAS-Identifier"] = "localhost" + proc.stdin.write(pkt.RequestPacket()) -proc.terminate() #pylint: disable=E1101 -proc.wait() + rsp = packet.Packet(secret="", dict=dict) + buf = proc.stdout.read(4) + buf += proc.stdout.read(struct.unpack("!BBH", buf)[2] - 4) + rsp.DecodePacket(buf) + pkt.VerifyReply(rsp) + + proc.terminate() # pylint: disable=E1101 + proc.wait() + +if __name__ == '__main__': + main() |