Update Fuzz_DSL.py

Fuzz_DSL.py

@@ -33,14 +33,20 @@ def get_sk(target, my_logger, session, node, edge, *args, **kwargs):
 	print my_key+"!!!!!!!!!!!!!!!!!!!!"
 	node.names['my_key']._value = my_key
 
-    
 
+def mytest(target, my_logger, session, node, edge, *args, **kwargs):
+	try:
+		banner = target.recv(10000)
+		print banner
+	except:
+		print "Failed!!!!!!!!!!"
     
 def main():
 	tar=Target(connection=TCPSocketConnection("192.168.1.1", 80))
 	session = Session(
 		target=tar,
 		receive_data_after_each_request=True,
+		# post_test_case_callbacks=mytest,
 		)
 	
 	s_initialize(name="Login")
@@ -113,7 +119,9 @@ def main():
 
 
 	s_initialize(name="Final")
-	s_static("GET /cgi-bin/New_GUI/Diagnostics.asp HTTP/1.1")
+	s_static("GET")
+	s_string(" ", max_len=10)
+	s_static("/cgi-bin/New_GUI/Diagnostics.asp HTTP/1.1")
 	s_static("\r\n")
 
 	s_static("Host: 192.168.1.1")
@@ -140,6 +148,8 @@ def main():
 
 	s_static("Upgrade-Insecure-Requests: 1")
 	s_static("\r\n")
+	s_static("\r\n")
+	s_static("\r\n")
 
 	
 
@@ -202,7 +212,7 @@ def main():
 	s_static("X-Requested-With: XMLHttpRequest")
 	s_static("\r\n")
 
-	s_static("Content-Length: 56")
+	s_static("Content-Length: 156")
 	# s_size('data', output_format='ascii', fuzzable=True)
 	s_static("\r\n")
 
@@ -223,7 +233,7 @@ def main():
 		s_static("my_key", name="my_key")
 		s_static("&Addr=")
 		s_static("XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX")
-		s_string("hello", max_len=10)
+		
 	session.connect(s_get("Login"))
 	session.connect(s_get("Login"), s_get("Get_key"), callback=change_port)
 	my_edge=session.connect(s_get("Get_key"), s_get("Post"), callback=get_sk)