s3-selftest: Verify GK and GF flag behaviour

At least this ensures that the helper has not crashed, it will require
a little more to ensure that the values are correct.

Andrew Bartlett

Autobuild-User: Andrew Bartlett <abartlet@samba.org>
Autobuild-Date: Fri Feb 24 03:53:38 CET 2012 on sn-devel-104

1 files changed, 26 insertions, 0 deletions

diff --git a/source3/torture/test_ntlm_auth.py b/source3/torture/test_ntlm_auth.py
index b6e06a9bd5..be725485a0 100755
--- a/source3/torture/test_ntlm_auth.py
+++ b/source3/torture/test_ntlm_auth.py
@@ -209,6 +209,32 @@ def main():
 	if buf.count("AF ", 0, 3) != 1:
 		sys.exit(4)
 
+	if opts.client_helper == "ntlmssp-client-1":
+		writeLine(client_out, "GK")
+		buf = readLine(client_in)
+
+		if buf.count("GK ", 0, 3) != 1:
+			sys.exit(4)
+
+		writeLine(client_out, "GF")
+		buf = readLine(client_in)
+
+		if buf.count("GF ", 0, 3) != 1:
+			sys.exit(4)
+
+	if opts.server_helper == "squid-2.5-ntlmssp":
+		writeLine(server_out, "GK")
+		buf = readLine(server_in)
+
+		if buf.count("GK ", 0, 3) != 1:
+			sys.exit(4)
+
+		writeLine(server_out, "GF")
+		buf = readLine(server_in)
+
+		if buf.count("GF ", 0, 3) != 1:
+			sys.exit(4)
+
 	os.close(server_in)
 	os.close(server_out)
 	os.close(client_in)