from hashtree import HashTree

import collections

import socket

import logging as log

import config as conf

from networkers import NetworkReader,NetworkWriter

class Connection:

	def __init__(self):

		self.socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

		self.socket.connect((conf.hosts[0], conf.port))

		fr=self.socket.makefile(mode="rb")

		fw=self.socket.makefile(mode="wb")

import hashlib

import os

class HashTree:

	HASH_LEN=16 # bytes

	BLOCK_SIZE=4096 # bytes

	## Prepares a tree containing leafCount leaves.

	def __init__(self,leafCount):

		self.store=[b""]*(leafCount*2-1)

		self.leafStart=leafCount-1

		self.index=self.leafStart

		self.leafCount=leafCount

	@classmethod

	def fromFile(cls,filename):

		with open(filename,"rb") as f:

			stat=os.fstat(f.fileno())

			size=stat.st_size # !! symlinks

			leafCount=(size-1)//HashTree.BLOCK_SIZE+1 # number of leaf blocks

			res=cls(leafCount)

			for i in range(leafCount):

				data=f.read(HashTree.BLOCK_SIZE)

				res.insertLeaf(hashlib.sha256(data).digest()[HashTree.HASH_LEN:])

		res.buildTree()

		return res

	@classmethod

	def load(cls,filename):

		with open(filename,"rb") as f:

			stat=os.fstat(f.fileno())

			size=stat.st_size

			nodeCount=size//HashTree.HASH_LEN

			res=cls((nodeCount+1)//2)

		self.store[index]=h

		self.updateNode((index-1)//2)

	## Updates the node at index and all its ancestors.

	def updateNode(self,index):

		while index>=0:

			self.store[index]=hashlib.sha256(self.store[index*2+1]+self.store[index*2+2]).digest()[HashTree.HASH_LEN:]

			index=(index-1)//2

	## Fast construction of the tree over the leaves. O(n).

	def buildTree(self):

		for i in range(self.leafStart-1,-1,-1):

			self.store[i]=hashlib.sha256(self.store[i*2+1]+self.store[i*2+2]).digest()[HashTree.HASH_LEN:]

if __name__=="__main__":

	f1=HashTree.fromFile(open("serverFile.txt",mode='rb'))

	f2=HashTree.fromFile(open("clientFile.txt",mode='rb'))

	for i,(h1,h2) in enumerate(zip(f1.store,f2.store)):

		print("{0:2}".format(i),h1.hex(),h2.hex(),h1==h2)

	import shutil

	origFilename=args.datafile

	filename=origFilename+"_"

	shutil.copyfile(origFilename,filename)

	s=Server(filename,args.tree)

	s.serve()

parser=ArgumentParser()

subparsers=parser.add_subparsers()

pUpdate=subparsers.add_parser("update")

pUpdate.add_argument("-p","--port",type=int)

pUpdate.add_argument("--host",default="127.0.0.1")

pUpdate.add_argument("-t","--tree",help="stored hash tree location")

pUpdate.add_argument("datafile")

pUpdate.set_defaults(func=update)

pServe=subparsers.add_parser("serve")

pServe.add_argument("-p","--port",type=int)

pServe.add_argument("--host",default="")

pServe.add_argument("-t","--tree",help="stored hash tree location")