Wallet check payment hash (#649)

* check preimage

* make format

* adjust logs

---------

Co-authored-by: callebtc <[email protected]>

cashu/wallet/cli/cli.py

@@ -11,6 +11,7 @@
 from os.path import isdir, join
 from typing import Optional, Union
 
+import bolt11
 import click
 from click import Context
 from loguru import logger
@@ -49,6 +50,7 @@
     verify_mint,
 )
 from ..helpers import (
+    check_payment_preimage,
     deserialize_token_from_string,
     init_wallet,
     list_mints,
@@ -209,6 +211,7 @@ async def pay(
     wallet: Wallet = ctx.obj["WALLET"]
     await wallet.load_mint()
     await print_balance(ctx)
+    payment_hash = bolt11.decode(invoice).payment_hash
     quote = await wallet.melt_quote(invoice, amount)
     logger.debug(f"Quote: {quote}")
     total_amount = quote.amount + quote.fee_reserve
@@ -252,9 +255,11 @@ async def pay(
             melt_response.payment_preimage
             and melt_response.payment_preimage != "0" * 64
         ):
+            if not check_payment_preimage(payment_hash, melt_response.payment_preimage):
+                print(" Error: Invalid preimage!", end="", flush=True)
             print(f" (Preimage: {melt_response.payment_preimage}).")
         else:
-            print(".")
+            print(" Mint did not provide a preimage.")
     elif MintQuoteState(melt_response.state) == MintQuoteState.pending:
         print(" Invoice pending.")
     elif MintQuoteState(melt_response.state) == MintQuoteState.unpaid:

cashu/wallet/helpers.py

@@ -1,3 +1,4 @@
+import hashlib
 import os
 from typing import Optional
 
@@ -169,3 +170,11 @@ async def send(
 
     await wallet.set_reserved(send_proofs, reserved=True)
     return wallet.available_balance, token
+
+def check_payment_preimage(
+    payment_hash: str,
+    preimage: str,
+) -> bool:
+    return bytes.fromhex(payment_hash) == hashlib.sha256(
+        bytes.fromhex(preimage)
+    ).digest()