BackEnd_BSFFSS_V 1.0.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from Crypto.PublicKey import RSA\n",
    "from Crypto.Hash import SHA\n",
    "from Crypto.Signature import PKCS1_v1_5\n",
    "from datetime import datetime as DT\n",
    "import hmac\n",
    "from Crypto.Cipher import AES\n",
    "from merklelib import MerkleTree\n",
    "import warnings\n",
    "import random\n",
    "warnings.filterwarnings(\"ignore\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "invalidTxns=[]\n",
    "flag=0\n",
    "miner=[]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "class device:\n",
    "    deviceCount = 0\n",
    "    \n",
    "    def __init__(self):\n",
    "        self.id = device.deviceCount\n",
    "        self.sk,self.pk= self.getKeys()\n",
    "        self.balance=100\n",
    "        \n",
    "        self.idProof = {}\n",
    "        self.idProof['public_KEY'] = self.pk.hex()\n",
    "        self.idProof['block_no'] = None\n",
    "        self.idProof['block_hash'] = None\n",
    "        self.idProof['difficulty'] = None\n",
    "        self.idProof['txn_Index'] = None\n",
    "        self.idProof['Node_ID'] = None\n",
    "        self.idProof['okToSend'] = None\n",
    "        self.idProof['okToRecieve'] = None\n",
    "        self.confi=0.5\n",
    "        self.resi=100\n",
    "        self.txncount=0\n",
    "        self.D=0.01\n",
    "        \n",
    "        data=str(self.id)+self.pk.hex()\n",
    "        self.address=SHA.new(data.encode()).hexdigest()\n",
    "        device.deviceCount += 1\n",
    "    \n",
    "    def getKeys(self):\n",
    "        self.keypair=RSA.generate(1024)\n",
    "        sk = self.keypair.exportKey('DER')\n",
    "        pk = self.keypair.publickey().exportKey('DER')\n",
    "        return sk,pk\n",
    "    \n",
    "    def getDeviceDetails(self):\n",
    "        print(\"deviceID: \"+str(self.id),\"publicKEY:\",self.pk.hex(),\"deviceADDRESS:\",self.address,sep=\"\\n\",end=\"\\n\\n\")\n",
    "        \n",
    "    @classmethod\n",
    "    def transact(self, S,R,amt):\n",
    "        \n",
    "        if amt < dList[S.id].balance:\n",
    "            dList[S.id].balance-=amt\n",
    "            dList[R.id].balance+=amt\n",
    "            Txn= S.address+R.address+str(amt)\n",
    "            sign = device.signTxn(Txn,S.sk)\n",
    "            verified = device.verifyTxn(Txn, sign, S.pk)\n",
    "            if(verified):\n",
    "                print(\"Transaction is verified\")\n",
    "                if dList[S.id].resi>0:\n",
    "                    dList[S.id].resi-=(random.randint(1,2)/100)\n",
    "                    dList[S.id].txncount+=1\n",
    "                    return [{\"Sender\":S,\"Receiver\":R,\"Amount\":amt}]\n",
    "                else:\n",
    "                    dList[S.id].resi=0\n",
    "                return []\n",
    "            else:\n",
    "                invalidTxns.append({\"Sender\":dList[S],\"Receiver\":dList[R],\"Amount\":amt,\"status\":\"TxnNotVerified\"})\n",
    "                print(\"Fraudulent transaction\")\n",
    "                return []\n",
    "        else:\n",
    "            invalidTxns.append({\"Sender\":S.id,\"Receiver\":R.id,\"Amount\":amt,\"status\":\"TxnNotVerified\"})\n",
    "            print(\"Sybil node\")\n",
    "            return []\n",
    "            \n",
    "    @classmethod\n",
    "    def dataTransfer(self,S,R,msg,dtKEY):\n",
    "        sF,rF=device.checkID(S,R)\n",
    "        if(not sF):\n",
    "            print(\"TODO sF triggered\")\n",
    "        if(not rF):\n",
    "            print(\"TODO rF triggered\")\n",
    "        else:\n",
    "            dList[S.id].resi-=(random.randint(1,2)/100)\n",
    "            pad=16-len(msg)%16\n",
    "            msg=msg.zfill(pad+len(msg))\n",
    "            secret=AES.new(dtKEY)\n",
    "            cipherTxt=secret.encrypt(msg)\n",
    "            signature=device.signTxn(cipherTxt,S.sk)\n",
    "            dList[S.id].txncount+=1\n",
    "            return [S,R,cipherTxt,signature]\n",
    "        return None\n",
    "        \n",
    "    @classmethod\n",
    "    def checkID(self, S, R):\n",
    "        SID=S.idProof\n",
    "        RID=R.idProof\n",
    "        sF,rF=True,True    #F-> flag\n",
    "        for i in SID.values():\n",
    "            if(i is None):\n",
    "                sF=False\n",
    "                invalidTxns.append({\"Sender\":S,\"Receiver\":R,\"Amount\":amt,\"status\":\"SenderIDNotVerified\"})\n",
    "                print(\"Sender ID cannot be verified\")\n",
    "                break\n",
    "        for i in RID.values():\n",
    "            if(i is None):\n",
    "                rF=False\n",
    "                invalidTxns.append({\"Sender\":S,\"Receiver\":R,\"Amount\":amt,\"status\":\"ReceiverNotVerified\"})\n",
    "                print(\"Receiver ID cannot be verified\")\n",
    "                break\n",
    "        return (sF,rF)\n",
    "        \n",
    "    \n",
    "    @classmethod\n",
    "    def signTxn(self,txn, sk):\n",
    "        txn=str(txn)\n",
    "        RSA_key=RSA.importKey(sk)\n",
    "        author=PKCS1_v1_5.new(RSA_key)\n",
    "        msg=SHA.new(txn.encode())\n",
    "        signature = author.sign(msg)\n",
    "        return signature\n",
    "    \n",
    "    @classmethod\n",
    "    def verifyTxn(self, txn, sign, pk):\n",
    "        txn=str(txn)\n",
    "        msg=SHA.new(txn.encode())\n",
    "        RSA_key=RSA.importKey(pk)\n",
    "        verifier = PKCS1_v1_5.new(RSA_key)\n",
    "        verified=verifier.verify(msg, sign)\n",
    "        return verified\n",
    "    \n",
    "    def __del__(self):\n",
    "        print(\"Device:\",self.id,\"deleted\")\n",
    "        device.deviceCount-=1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "class block:\n",
    "    blockCount=0\n",
    "    def __init__(self,pH,txnList,PoW,mR):\n",
    "        block.blockCount+=1\n",
    "        self.blockNo=block.blockCount\n",
    "        self.prevHash=pH\n",
    "        self.transactions=txnList\n",
    "        self.timestamp=DT.timestamp(DT.now())\n",
    "        self.hash=PoW['Hash']\n",
    "        self.nonce=PoW['nonce']\n",
    "        self.mRoot=mR\n",
    "    \n",
    "    def getBlockDetails(self):\n",
    "        print(\"Block#:\",self.blockNo)\n",
    "        print(\"previous HASH:\",self.prevHash)\n",
    "        print(\"current HASH:\",self.hash)\n",
    "        print()\n",
    "        \n",
    "        print(\"Transactions:\",len(self.transactions),\"     Time:\",DT.fromtimestamp(self.timestamp))\n",
    "        for T in self.transactions:\n",
    "            print(\"Sender:\",T['Sender'].id, end=\",  \")\n",
    "            print(\"Receiver:\",T['Receiver'].id, end=\",  \")\n",
    "            print(\"Amount:\",T['Amount'])\n",
    "        \n",
    "        print(\"Merkley Root:\",self.mRoot)\n",
    "        print(\"\\nNONCE:\",self.nonce)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "def HF(data):\n",
    "    data=data.hex()\n",
    "    return SHA.new(data.encode()).hexdigest()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "class blockChain:\n",
    "    chain=[]\n",
    "    allTxns={}\n",
    "    Tcount=1\n",
    "    def __init__(self,H):\n",
    "        pH='0'\n",
    "        PoW={}\n",
    "        txn=[{\"Sender\":H,\"Receiver\":H,\"Amount\":0}]\n",
    "        PoW['Hash'],PoW['nonce']=self.PoW(pH,txn)\n",
    "        mR=SHA.new(str(txn).encode()).hexdigest()\n",
    "        txID = str(1)\n",
    "        diff=3\n",
    "        H.idProof['block_no'] = len(blockChain.chain)\n",
    "        H.idProof['block_hash'] = PoW['Hash']\n",
    "        H.idProof['difficulty'] = diff\n",
    "        H.idProof['txn_Index'] = txID\n",
    "        pk=H.pk\n",
    "        key=hmac.HMAC(pk)\n",
    "        H.idProof['Node_ID'] = hmac.HMAC(key.digest(),txID.encode()).hexdigest()\n",
    "        H.idProof['okToSend'] = '0'\n",
    "        H.idProof['okToRecieve'] = '0'\n",
    "        \n",
    "        BLOC=block(pH,txn,PoW,mR)\n",
    "        blockChain.chain.append(BLOC)\n",
    "        \n",
    "        \n",
    "    def newBlock(self,txn):\n",
    "        if(txn==[]):\n",
    "            return\n",
    "        \n",
    "        Tno=\"Txn-\"+str(blockChain.Tcount)\n",
    "        blockChain.allTxns[Tno]=txn[0]\n",
    "        blockChain.Tcount+=1\n",
    "        pH=blockChain.chain[-1].hash\n",
    "        PoW={}\n",
    "        PoW['Hash'],PoW['nonce']=self.PoW(pH,txn)\n",
    "        mT=MerkleTree(txn, HF)\n",
    "        mR=mT.merkle_root\n",
    "        #ID PROOF GENERATION\n",
    "        S = txn[0]['Sender']\n",
    "        txID = str(1)\n",
    "        diff=3\n",
    "        S.idProof['block_no'] = len(blockChain.chain)\n",
    "        S.idProof['block_hash'] = PoW['Hash']\n",
    "        S.idProof['difficulty'] = diff\n",
    "        S.idProof['txn_Index'] = txID\n",
    "        pk=S.pk\n",
    "        key=hmac.HMAC(pk)\n",
    "        S.idProof['Node_ID'] = hmac.HMAC(key.digest(),txID.encode()).hexdigest()\n",
    "        S.idProof['okToSend'] = '0'\n",
    "        S.idProof['okToRecieve'] = '0'\n",
    "        \n",
    "        BLOC=block(pH,txn,PoW,mR)\n",
    "        miner.append(self.PoC())\n",
    "        blockChain.chain.append(BLOC)\n",
    "        \n",
    "    def addNewBlock(self,txnList):\n",
    "        if(txnList==[]):\n",
    "            return\n",
    "        \n",
    "        for T in txnList:\n",
    "            Tno=\"Txn-\"+str(blockChain.Tcount)\n",
    "            blockChain.allTxns[Tno]=T\n",
    "            blockChain.Tcount+=1\n",
    "            \n",
    "        pH=blockChain.chain[-1].hash\n",
    "        PoW={}\n",
    "        PoW['Hash'],PoW['nonce']=self.PoW(pH,txnList)        \n",
    "        mT = MerkleTree(txnList, HF)\n",
    "        mR=mT.merkle_root\n",
    "        BLOC=block(pH,txnList,PoW,mR)\n",
    "        miner.append(self.PoC())\n",
    "        blockChain.chain.append(BLOC)\n",
    "    \n",
    "    def PoC(self):\n",
    "        maxr=0\n",
    "        maxt=0\n",
    "        for i in range(N):\n",
    "            if dList[i].resi>=dList[maxr].resi:\n",
    "                maxr=i\n",
    "            if dList[i].txncount>=dList[maxt].txncount:\n",
    "                maxt=i\n",
    "        print(maxr)\n",
    "        print(maxt)\n",
    "        for i in range(N):\n",
    "            dList[i].confi= (dList[i].confi*(1-dList[i].D))+ ((dList[i].txncount/dList[maxt].txncount)*(dList[i].resi/dList[maxr].resi))\n",
    "            if i>0:\n",
    "                if dList[i].confi > dList[i-1].confi:\n",
    "                    maxco=dList[i].confi\n",
    "        for i in range(N):\n",
    "            dList[i].confi=dList[i].confi/maxco\n",
    "            if(dList[i].confi==1):\n",
    "                t=i\n",
    "        for i in range(N):\n",
    "            dList[i].txncount=0\n",
    "        return t\n",
    "        \n",
    "    \n",
    "    def PoW(self,pH,txn,diff=2):\n",
    "        nonce=0\n",
    "        data=pH+str(txn)+str(nonce)\n",
    "        nH=SHA.new(data.encode()).hexdigest()\n",
    "        while nH[:diff] != '0'*diff:\n",
    "            nonce += 1\n",
    "            data=pH+str(txn)+str(nonce)\n",
    "            nH=SHA.new(data.encode()).hexdigest()\n",
    "        return nH,nonce\n",
    "    \n",
    "    def getChainDetails(self):\n",
    "        for D in blockChain.chain:\n",
    "            D.getBlockDetails()\n",
    "            print(\"-------------------------------------------\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Enter no.of devices:3\n",
      "press 'Y' to show device details, any other key to skip:y\n",
      "deviceID: 0\n",
      "publicKEY:\n",
      "30819f300d06092a864886f70d010101050003818d0030818902818100b9ed705df8b0c535379e954413d2cce084fe3b7f9f38f67d7481bd28b79104bda36c8ffd124735971c286da8665255f08d1ee8c803915f430721b30cd156477e6809ce853b79e54cd8846d41fd1e17253d9e4aae0a2965dd76466cd60b15299a46129b6bfa520c2df5a031a30edc182665627569c7054273053415bf404057650203010001\n",
      "deviceADDRESS:\n",
      "5e0e23183b5575cb20fa13fec43d8e660d45777c\n",
      "\n",
      "deviceID: 1\n",
      "publicKEY:\n",
      "30819f300d06092a864886f70d010101050003818d0030818902818100cf8fe68e29cf9b6735824a9779998fea3a578e2055a4d4e901adb3fab45febfeabec53f74708928039169dc834fab8d011c0fcc40b669611ef6249efb599ec543feb8ee0a73b8be1a4b6fa8d68de9b986b931f94149b7200e1b2f135cbd2d41799d4edb029b2cefdd3aa0b75155d75e4201ecaeda21c7eab0dda5d44af47fe150203010001\n",
      "deviceADDRESS:\n",
      "06d247ceb957a30e7bc014fc2b9b5d13e6193d16\n",
      "\n",
      "deviceID: 2\n",
      "publicKEY:\n",
      "30819f300d06092a864886f70d010101050003818d0030818902818100f6589e90258ba871032e3654320b0bce49c190a69ba7d477a70b772044eb0441e9aa224d110297a487bf04698a45f16e2fd6b3a05ca5331a5206660ccafdfe0e0dff9a18fd0f3ce5a7809b555a1a5b298752b96dc7e6dda0bcadd7882711bde2ca6223486c828e851b467efa21b2f4684c180d3a4e539006a84147fdc4afbbe50203010001\n",
      "deviceADDRESS:\n",
      "8ff34cc3e9c102d807333459a7f49731dd26a94c\n",
      "\n",
      "Sender#:   Receiver#:   Amount:1 0 0\n",
      "Transaction is verified\n",
      "2\n",
      "1\n",
      "Sender#:   Receiver#:   Amount:2 0 0\n",
      "Transaction is verified\n",
      "0\n",
      "2\n"
     ]
    }
   ],
   "source": [
    "dtKEY=\"sixteen--SIXTEEN\"\n",
    "N=int(input(\"Enter no.of devices:\"))\n",
    "dList=[]\n",
    "\n",
    "\n",
    "for i in range(N):\n",
    "    dList.append(device())\n",
    "\n",
    "show=input(\"press \\'Y' to show device details, any other key to skip:\")\n",
    "if(show in ['Y','y']):\n",
    "    for i in dList:\n",
    "        i.getDeviceDetails()\n",
    "\n",
    "myChain=blockChain(dList[0])\n",
    "for i in range(1,N):\n",
    "    Txn = input(\"Sender#:   Receiver#:   Amount:\").strip().split()\n",
    "    S,R,amt = [int(x) for x in Txn]\n",
    "    if(S>=len(dList)):\n",
    "        invalidTxns.append({\"Sender\":dList[S],\"Receiver\":dList[R],\"Amount\":amt,\"status\":\"SenderNotFound\"})\n",
    "        print(\"Sender not found\")\n",
    "    elif(S==0):\n",
    "        invalidTxns.append({\"Sender\":dList[S],\"Receiver\":dList[R],\"Amount\":amt,\"status\":\"inValidSender\"})\n",
    "        print(\"Invalid Sender\")\n",
    "    elif(R != 0):\n",
    "        invalidTxns.append({\"Sender\":dList[S],\"Receiver\":dList[R],\"Amount\":amt,\"status\":\"inValidReceiver\"})\n",
    "        print(\"Invalid Receiver\")\n",
    "    else:\n",
    "        oneTxnList=device.transact(dList[S],dList[R],amt)\n",
    "        myChain.newBlock(oneTxnList)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[]"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "invalidTxns"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Displaying each block"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Block#: 1\n",
      "previous HASH: 0\n",
      "current HASH: 00712cd6cf4ebc14c9ba7fbb4963d675d224a856\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:08:07.859978\n",
      "Sender: 0,  Receiver: 0,  Amount: 0\n",
      "Merkley Root: ddbaaf2a58ea4e923d156dc70b18614095399e41\n",
      "\n",
      "NONCE: 256\n",
      "-------------------------------------------\n",
      "Block#: 2\n",
      "previous HASH: 00712cd6cf4ebc14c9ba7fbb4963d675d224a856\n",
      "current HASH: 00e35824f78df09583e13c1ddcf6e47175c445de\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:09:02.427059\n",
      "Sender: 1,  Receiver: 0,  Amount: 0\n",
      "Merkley Root: 68f958e693017513ceb7f6af04d089d14174517c\n",
      "\n",
      "NONCE: 385\n",
      "-------------------------------------------\n",
      "Block#: 3\n",
      "previous HASH: 00e35824f78df09583e13c1ddcf6e47175c445de\n",
      "current HASH: 0073c95fda71ba7e2d7ed1f39745b6b3cbc12315\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:09:14.280881\n",
      "Sender: 2,  Receiver: 0,  Amount: 0\n",
      "Merkley Root: 11608d6f51cf2ec61d2bfaeccba50314f4bbea27\n",
      "\n",
      "NONCE: 530\n",
      "-------------------------------------------\n"
     ]
    }
   ],
   "source": [
    "myChain.getChainDetails()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Txn-1 : Sender: 06d247ceb957a30e7bc014fc2b9b5d13e6193d16    Receiver: 5e0e23183b5575cb20fa13fec43d8e660d45777c    Amount: 0\n",
      "Txn-2 : Sender: 8ff34cc3e9c102d807333459a7f49731dd26a94c    Receiver: 5e0e23183b5575cb20fa13fec43d8e660d45777c    Amount: 0\n"
     ]
    }
   ],
   "source": [
    "for k,v in myChain.allTxns.items():\n",
    "    print(k,\": Sender:\",v[\"Sender\"].address, \"   Receiver:\",v[\"Receiver\"].address,   \"   Amount:\",v['Amount'])        "
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Displaying Identity Set"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Device: 0\n",
      "public_KEY : 30819f300d06092a864886f70d010101050003818d0030818902818100b9ed705df8b0c535379e954413d2cce084fe3b7f9f38f67d7481bd28b79104bda36c8ffd124735971c286da8665255f08d1ee8c803915f430721b30cd156477e6809ce853b79e54cd8846d41fd1e17253d9e4aae0a2965dd76466cd60b15299a46129b6bfa520c2df5a031a30edc182665627569c7054273053415bf404057650203010001\n",
      "block_no : 0\n",
      "block_hash : 00712cd6cf4ebc14c9ba7fbb4963d675d224a856\n",
      "difficulty : 3\n",
      "txn_Index : 1\n",
      "Node_ID : 6a025c75ce39aef27f8ecae43cf1d9bd\n",
      "okToSend : 0\n",
      "okToRecieve : 0\n",
      "\n",
      "Device: 1\n",
      "public_KEY : 30819f300d06092a864886f70d010101050003818d0030818902818100cf8fe68e29cf9b6735824a9779998fea3a578e2055a4d4e901adb3fab45febfeabec53f74708928039169dc834fab8d011c0fcc40b669611ef6249efb599ec543feb8ee0a73b8be1a4b6fa8d68de9b986b931f94149b7200e1b2f135cbd2d41799d4edb029b2cefdd3aa0b75155d75e4201ecaeda21c7eab0dda5d44af47fe150203010001\n",
      "block_no : 1\n",
      "block_hash : 00e35824f78df09583e13c1ddcf6e47175c445de\n",
      "difficulty : 3\n",
      "txn_Index : 1\n",
      "Node_ID : bd0f328f88845959af07e0cc8298ef26\n",
      "okToSend : 0\n",
      "okToRecieve : 0\n",
      "\n",
      "Device: 2\n",
      "public_KEY : 30819f300d06092a864886f70d010101050003818d0030818902818100f6589e90258ba871032e3654320b0bce49c190a69ba7d477a70b772044eb0441e9aa224d110297a487bf04698a45f16e2fd6b3a05ca5331a5206660ccafdfe0e0dff9a18fd0f3ce5a7809b555a1a5b298752b96dc7e6dda0bcadd7882711bde2ca6223486c828e851b467efa21b2f4684c180d3a4e539006a84147fdc4afbbe50203010001\n",
      "block_no : 2\n",
      "block_hash : 0073c95fda71ba7e2d7ed1f39745b6b3cbc12315\n",
      "difficulty : 3\n",
      "txn_Index : 1\n",
      "Node_ID : c51cd183db906d2d891b49203c3b53b0\n",
      "okToSend : 0\n",
      "okToRecieve : 0\n",
      "\n"
     ]
    }
   ],
   "source": [
    "for D in dList:\n",
    "    print(\"Device:\",D.id)\n",
    "    for k,v in D.idProof.items():\n",
    "        print(k,\":\",v)\n",
    "    print()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Data Transfer"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Press N to send new data, any other key to stop sending:n\n",
      "Enter Sender's id:1\n",
      "Enter Receiver's id:2\n",
      "Enter data to be sent:m1\n",
      "SUCCESSFUL\n",
      "\n",
      "Press N to send new data, any other key to stop sending:n\n",
      "Enter Sender's id:2\n",
      "Enter Receiver's id:1\n",
      "Enter data to be sent:m2\n",
      "SUCCESSFUL\n",
      "\n",
      "Press N to send new data, any other key to stop sending:1\n",
      "0\n",
      "2\n"
     ]
    }
   ],
   "source": [
    "txnList=[]\n",
    "while(1):\n",
    "    cmd=input(\"Press N to send new data, any other key to stop sending:\")\n",
    "    if(cmd in ['N','n']):\n",
    "\n",
    "        S=int(input(\"Enter Sender's id:\"))\n",
    "        if(S>=len(dList) or S<0):\n",
    "            invalidTxns.append({\"Sender\":dList[S],\"Receiver\":dList[R],\"Amount\":amt,\"status\":\"inValidSender\"})\n",
    "            print(\"invalid sender\")\n",
    "            continue\n",
    "        R=int(input(\"Enter Receiver's id:\"))\n",
    "        if(R>=len(dList) or R<0):\n",
    "            invalidTxns.append({\"Sender\":dList[S],\"Receiver\":dList[R],\"Amount\":amt,\"status\":\"inValidReceiver\"})\n",
    "            print(\"invalid Receiver\")\n",
    "            continue\n",
    "        \n",
    "        msg=input(\"Enter data to be sent:\")\n",
    "        \n",
    "        data=device.dataTransfer(dList[S],dList[R],msg,dtKEY)\n",
    "        if(not bool(data)):\n",
    "            continue\n",
    "        verified=device.verifyTxn(data[2],data[3],data[0].pk)\n",
    "        if(verified):\n",
    "            txnList.append({\"Sender\":data[0],\"Receiver\":data[1],\"Amount\":data[2].hex()})\n",
    "#             print(\"Sender Address:  \",data[0].address)\n",
    "#             print(\"Receiver Address:\",data[1].address)\n",
    "#             print(\"Encrypted msg:   \",data[2].hex())\n",
    "            print(\"SUCCESSFUL\")\n",
    "            print()\n",
    "        if(len(txnList)==3):\n",
    "            myChain.addNewBlock(txnList)\n",
    "            txnList=[]\n",
    "    else:\n",
    "        if(bool(txnList)):\n",
    "            myChain.addNewBlock(txnList)\n",
    "            txnList=[]\n",
    "        break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Block#: 1\n",
      "previous HASH: 0\n",
      "current HASH: 00712cd6cf4ebc14c9ba7fbb4963d675d224a856\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:08:07.859978\n",
      "Sender: 0,  Receiver: 0,  Amount: 0\n",
      "Merkley Root: ddbaaf2a58ea4e923d156dc70b18614095399e41\n",
      "\n",
      "NONCE: 256\n",
      "-------------------------------------------\n",
      "Block#: 2\n",
      "previous HASH: 00712cd6cf4ebc14c9ba7fbb4963d675d224a856\n",
      "current HASH: 00e35824f78df09583e13c1ddcf6e47175c445de\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:09:02.427059\n",
      "Sender: 1,  Receiver: 0,  Amount: 0\n",
      "Merkley Root: 68f958e693017513ceb7f6af04d089d14174517c\n",
      "\n",
      "NONCE: 385\n",
      "-------------------------------------------\n",
      "Block#: 3\n",
      "previous HASH: 00e35824f78df09583e13c1ddcf6e47175c445de\n",
      "current HASH: 0073c95fda71ba7e2d7ed1f39745b6b3cbc12315\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:09:14.280881\n",
      "Sender: 2,  Receiver: 0,  Amount: 0\n",
      "Merkley Root: 11608d6f51cf2ec61d2bfaeccba50314f4bbea27\n",
      "\n",
      "NONCE: 530\n",
      "-------------------------------------------\n",
      "Block#: 4\n",
      "previous HASH: 0073c95fda71ba7e2d7ed1f39745b6b3cbc12315\n",
      "current HASH: 008c5218b0893231a356e57e7722daeee3dbd4f4\n",
      "\n",
      "Transactions: 1      Time: 2020-02-04 14:14:17.192642\n",
      "Sender: 1,  Receiver: 2,  Amount: 7872d61dcfe7c4db4cfa311c921805d1\n",
      "Merkley Root: e49763d717428e1955d9de3b6123f64f1cf17b76\n",
      "\n",
      "NONCE: 314\n",
      "-------------------------------------------\n",
      "Block#: 5\n",
      "previous HASH: 008c5218b0893231a356e57e7722daeee3dbd4f4\n",
      "current HASH: 0002987b8e055ffb566c9211ab4f2ddb4dbb1419\n",
      "\n",
      "Transactions: 2      Time: 2020-02-04 14:15:02.084633\n",
      "Sender: 1,  Receiver: 2,  Amount: 0a41fdca2020db77a109978e31a2d344\n",
      "Sender: 2,  Receiver: 1,  Amount: 463a7152bbf1e53f5bc9b12ff41ee829\n",
      "Merkley Root: ccf174d3a8085ab9df3832cca795fa94e072bc72\n",
      "\n",
      "NONCE: 405\n",
      "-------------------------------------------\n"
     ]
    }
   ],
   "source": [
    "myChain.getChainDetails()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}