scyther/test/protocollist.py

#!/usr/bin/python
#
#	protocol list
#
#
def list_ppfix(list, prefix, postfix):
	newlist = []
	for i in list:
		newlist.append(prefix + i + postfix)
	return newlist

def from_good_literature():
	list = [ \
	"ccitt509-1c.spdl",
	"ccitt509-1.spdl",
	"ccitt509-3.spdl",
	"ccitt509-ban3.spdl",
	"denning-sacco-lowe.spdl",
	"denning-sacco.spdl",
	"kaochow.spdl",
	"kaochow-v2.spdl",
	"kaochow-v3.spdl",
	"ksl-lowe.spdl",
	"ksl.spdl",
	"needham-schroeder-lowe.spdl",
	"needham-schroeder-sk-amend.spdl",
	"needham-schroeder-sk.spdl",
	"neumannstub-hwang.spdl",
	"neumannstub.spdl",
	"otwayrees.spdl",
	"smartright.spdl",
	"splice-as-cj.spdl",
	"splice-as-hc.spdl",
	"splice-as.spdl",
	"woo-lam-pi-f.spdl",
	"woo-lam-pi.spdl",
	"woo-lam.spdl",
	"yahalom-lowe.spdl",
	"yahalom-paulson.spdl" ]

	return list_ppfix(list, "../spdl/SPORE/","")

def from_bad_literature():
	list = [ \
	"andrew-ban-concrete.spdl",
	"andrew-ban.spdl",
	"andrew-lowe-ban.spdl",
	"andrew.spdl",
	"needham-schroeder.spdl",
	"tmn.spdl",
	"wmf-lowe.spdl",
	"wmf.spdl",
	"woo-lam-pi-1.spdl",
	"woo-lam-pi-2.spdl",
	"woo-lam-pi-3.spdl",
	"yahalom-ban.spdl",
	"yahalom.spdl" ]

	return list_ppfix(list, "../spdl/SPORE/","")

def from_literature():
	return from_good_literature() + from_bad_literature()

def from_others():
	list = [ \
			]

	return list_ppfix(list, "../spdl/","")

def from_all():
	return from_literature() + from_others()

def select(type):
	n = int(type)
	if n == 0:
		# 0 means all protocols
		return from_all()
	elif n == 1:
		# 1 means from literature
		return from_literature()
	elif n == 2:
		# 2 means from literature, no known attacks
		return from_good_literature()
	else:
		# Otherwise empty list
		return []




def main():
	for l in [from_literature(), from_others()]:
		for p in l:
			print p
		print

if __name__ == '__main__':
	main()