from luks import LuksFile
import sys, StringIO

if len(sys.argv) < 2:
	print "Usage: createio.py <file> <password>"
	sys.exit(1)

sbuf = StringIO.StringIO()

luks = LuksFile()
luks.create(sbuf, "aes", "cbc-essiv:sha256", "sha1", 16, 4000)
luks.set_key(0, sys.argv[2], 10000)

data = "this is really cool:"
data += "5#" * ((512-len(data))/2)

luks.encrypt_data(0, data)
luks.encrypt_data(512, data)
luks.encrypt_data(1024, data)
luks.truncate(512*3)

f = open(sys.argv[1], "wb+")
sbuf.seek(0)
f.write(sbuf.read())
f.close()

luks.close()