def test_poa(self):
self.poa_a.verify()
self.poa_b.verify()
# Test poa expiretime
expire_time = time.mktime(time.gmtime()) + 60 # Expire in one minute
self.poa_a = ClosedSwarm.create_poa(
self.torrent_id, self.torrent_keypair, self.node_a_pub_permid, expire_time=expire_time
)
try:
self.poa_a.verify()
except ClosedSwarm.POAExpiredException:
self.fail("POA verify means expired, but it is not")
expire_time = time.mktime(time.gmtime()) - 1 # Expire one second ago
self.poa_a = ClosedSwarm.create_poa(
self.torrent_id, self.torrent_keypair, self.node_a_pub_permid, expire_time=expire_time
)
try:
self.poa_a.verify()
self.fail("POA verify does not honor expire time")
except ClosedSwarm.POAExpiredException:
pass
def setUp(self):
self.keyfiles = [".node_a_keypair", ".node_b_keypair", ".torrent_keypair"]
for filename in self.keyfiles:
if not os.path.exists(filename):
keypair = permid.generate_keypair()
permid.save_keypair(keypair, filename)
self.node_a_keypair = permid.read_keypair(".node_a_keypair")
self.node_b_keypair = permid.read_keypair(".node_b_keypair")
self.torrent_keypair = permid.read_keypair(".torrent_keypair")
self.torrent_id = "1234"
# Shortcuts
self.node_a_pub_permid = str(self.node_a_keypair.pub().get_der())
self.node_b_pub_permid = str(self.node_b_keypair.pub().get_der())
self.torrent_pubkeys = [encodestring(str(self.torrent_keypair.pub().get_der())).replace("\n", "")]
# Create the certificate for this torrent ("proof of access")
self.poa_a = ClosedSwarm.create_poa(self.torrent_id, self.torrent_keypair, self.node_a_pub_permid)
self.poa_b = ClosedSwarm.create_poa(self.torrent_id, self.torrent_keypair, self.node_b_pub_permid)
self.cs_a = ClosedSwarm.ClosedSwarm(self.node_a_keypair, self.torrent_id, self.torrent_pubkeys, self.poa_a)
self.cs_b = ClosedSwarm.ClosedSwarm(self.node_b_keypair, self.torrent_id, self.torrent_pubkeys, self.poa_b)
def test_poa(self):
self.poa_a.verify()
self.poa_b.verify()
# Test poa expiretime
expire_time = time.mktime(time.gmtime()) + 60 # Expire in one minute
self.poa_a = ClosedSwarm.create_poa(self.torrent_id,
self.torrent_keypair,
self.node_a_pub_permid,
expire_time=expire_time)
try:
self.poa_a.verify()
except ClosedSwarm.POAExpiredException:
self.fail("POA verify means expired, but it is not")
expire_time = time.mktime(time.gmtime()) - 1 # Expire one second ago
self.poa_a = ClosedSwarm.create_poa(self.torrent_id,
self.torrent_keypair,
self.node_a_pub_permid,
expire_time=expire_time)
try:
self.poa_a.verify()
self.fail("POA verify does not honor expire time")
except ClosedSwarm.POAExpiredException:
pass
def create_poa(torrent, torrent_keypair, node_id, target_file):
# Sanity - check that this key matches the torrent
pubkey = torrent_keypair.pub()
good_key = False
for key in torrent.get_cs_keys():
if pubkey.get_der() == key.get_der():
good_key = True
break
if not good_key:
raise Exception("Bad key given for this torrent")
# if the node_id is base64 encoded, decode it
try:
actual_node_id = decodestring(node_id)
print "Node ID was base64 encoded"
except:
actual_node_id = node_id
# Got the right key, now create POA
poa = ClosedSwarm.create_poa(t.infohash, torrent_keypair, actual_node_id)
# Now we save it
if target_file:
ClosedSwarm.write_poa_to_file(target_file)
tf = target_file
else:
tf = ClosedSwarm.trivial_save_poa("./", decodestring(node_id), t.infohash, poa)
print "Proof of access written to file '%s'"%tf
def setUp(self):
self.keyfiles = [
".node_a_keypair", ".node_b_keypair", ".torrent_keypair"
]
for filename in self.keyfiles:
if not os.path.exists(filename):
print "Generating", filename
keypair = permid.generate_keypair()
permid.save_keypair(keypair, filename)
self.node_a_keypair = permid.read_keypair(".node_a_keypair")
self.node_b_keypair = permid.read_keypair(".node_b_keypair")
self.torrent_keypair = permid.read_keypair(".torrent_keypair")
self.torrent_id = "1234"
# Shortcuts
self.node_a_pub_permid = str(self.node_a_keypair.pub().get_der())
self.node_b_pub_permid = str(self.node_b_keypair.pub().get_der())
self.torrent_pubkeys = [
encodestring(str(self.torrent_keypair.pub().get_der())).replace(
"\n", "")
]
# Create the certificate for this torrent ("proof of access")
self.poa_a = ClosedSwarm.create_poa(self.torrent_id,
self.torrent_keypair,
self.node_a_pub_permid)
self.poa_b = ClosedSwarm.create_poa(self.torrent_id,
self.torrent_keypair,
self.node_b_pub_permid)
self.cs_a = ClosedSwarm.ClosedSwarm(self.node_a_keypair,
self.torrent_id,
self.torrent_pubkeys, self.poa_a)
self.cs_b = ClosedSwarm.ClosedSwarm(self.node_b_keypair,
self.torrent_id,
self.torrent_pubkeys, self.poa_b)
def create_poa(torrent, torrent_keypair, node_id, target_file):
# Sanity - check that this key matches the torrent
pubkey = torrent_keypair.pub()
good_key = False
for key in torrent.get_cs_keys():
if pubkey.get_der() == key.get_der():
good_key = True
break
if not good_key:
raise Exception("Bad key given for this torrent")
# Got the right key, now create POA
poa = ClosedSwarm.create_poa(t.infohash, torrent_keypair, node_id)
f = open(target_file, "wb")
f.write(poa.serialize())
print "Proof of access written to file '%s'"%target_file
status.get_status_element("poas_failed").inc()
raise Exception("Randomly denied...")
key_file = os.path.join(KEY_PATH, swarm_id + ".tkey")
if not os.path.exists(key_file):
raise Exception("Missing key file")
# Load keys
try:
torrent_keypair = ClosedSwarm.read_cs_keypair(key_file)
except Exception,e:
raise Exception("Bad torrent key file")
# TODO? Sanity - check that this key matches the torrent
poa = ClosedSwarm.create_poa(swarm_id, torrent_keypair, perm_id)
status.create_and_add_event("allowed", [swarm_id, perm_id])
status.get_status_element("poas_generated").inc()
return poa.serialize()
class WebServer(threading.Thread):
def __init__(self, port):
threading.Thread.__init__(self)
print "Starting WebServer on port %s"%port
self.server = MyWebServer(('', int(port)), WebHandler)
self.port = port
self.running = False
