From 466133e5aa53f3bc70ea1645e4caa0b60c55b3f7 Mon Sep 17 00:00:00 2001
From: ardevd <ardevd@protonmail.com>
Date: Mon, 15 Jan 2024 22:30:50 +0100
Subject: [PATCH] feat: implemented main rpc message verification

added support for the main RPC message verification API endpoint.
---
 lightning_client.go      | 26 ++++++++++++++++++++++++++
 macaroon_recipes_test.go |  2 +-
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/lightning_client.go b/lightning_client.go
index f326c05..48ccf7c 100644
--- a/lightning_client.go
+++ b/lightning_client.go
@@ -278,6 +278,14 @@ type LightningClient interface {
 	// are needed for verification.
 	SignMessage(ctx context.Context, data []byte) (string,
 		error)
+
+	// VerifyMessage verifies a signature over a msg. The signature must
+	// be zbase32 encoded and signed by an active node in the resident
+	// node's channel database. In addition to returning the validity of
+	// the signature, VerifyMessage also returns the recovered pubkey
+	// from the signature.
+	VerifyMessage(ctx context.Context, data []byte, signature string) (bool,
+		string, error)
 }
 
 // Info contains info about the connected lnd node.
@@ -4156,6 +4164,24 @@ func (s *lightningClient) SignMessage(ctx context.Context,
 	return rpcRes.Signature, nil
 }
 
+// VerifyMessage verifies a signature over a msg. The signature must
+// be zbase32 encoded and signed by an active node in the resident
+// node's channel database. In addition to returning the validity of
+// the signature, VerifyMessage also returns the recovered pubkey
+// from the signature.
+func (s *lightningClient) VerifyMessage(ctx context.Context,
+	data []byte, signature string) (bool, string, error) {
+
+	rpcCtx := s.adminMac.WithMacaroonAuth(ctx)
+	rpcReq := &lnrpc.VerifyMessageRequest{Msg: data, Signature: signature}
+	rpcRes, err := s.client.VerifyMessage(rpcCtx, rpcReq)
+	if err != nil {
+		return false, "", err
+	}
+
+	return rpcRes.Valid, rpcRes.Pubkey, nil
+}
+
 // SubscribeCustomMessages subscribes to a stream of custom messages, optionally
 // filtering by peer and message type. The channels returned will be closed
 // when the subscription exits.
diff --git a/macaroon_recipes_test.go b/macaroon_recipes_test.go
index dcb5baa..2bf927c 100644
--- a/macaroon_recipes_test.go
+++ b/macaroon_recipes_test.go
@@ -12,7 +12,7 @@ import (
 
 var (
 	expectedPermissions = map[string]int{
-		"lnrpc":       12,
+		"lnrpc":       13,
 		"chainrpc":    1,
 		"invoicesrpc": 2,
 		"routerrpc":   2,