From 4fc78877072c4098bdcf50fd257d65aad511f467 Mon Sep 17 00:00:00 2001 From: Jesse de Wit Date: Thu, 15 Jun 2023 12:48:28 +0200 Subject: [PATCH] notifications: server implementation --- notifications/genproto.sh | 4 + notifications/notifications.pb.go | 218 +++++++++++++++++++++++++ notifications/notifications.proto | 18 ++ notifications/notifications_grpc.pb.go | 105 ++++++++++++ notifications/server.go | 58 +++++++ notifications/store.go | 10 ++ 6 files changed, 413 insertions(+) create mode 100755 notifications/genproto.sh create mode 100644 notifications/notifications.pb.go create mode 100644 notifications/notifications.proto create mode 100644 notifications/notifications_grpc.pb.go create mode 100644 notifications/server.go create mode 100644 notifications/store.go diff --git a/notifications/genproto.sh b/notifications/genproto.sh new file mode 100755 index 0000000..aff50da --- /dev/null +++ b/notifications/genproto.sh @@ -0,0 +1,4 @@ +#!/bin/bash +SCRIPTDIR=$(dirname $0) + +protoc --go_out=$SCRIPTDIR --go_opt=paths=source_relative --go-grpc_out=$SCRIPTDIR --go-grpc_opt=paths=source_relative -I=$SCRIPTDIR $SCRIPTDIR/*.proto diff --git a/notifications/notifications.pb.go b/notifications/notifications.pb.go new file mode 100644 index 0000000..36db039 --- /dev/null +++ b/notifications/notifications.pb.go @@ -0,0 +1,218 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// versions: +// protoc-gen-go v1.28.1 +// protoc v3.21.12 +// source: notifications.proto + +package notifications + +import ( + protoreflect "google.golang.org/protobuf/reflect/protoreflect" + protoimpl "google.golang.org/protobuf/runtime/protoimpl" + reflect "reflect" + sync "sync" +) + +const ( + // Verify that this generated code is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) + // Verify that runtime/protoimpl is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) +) + +type SubscribeNotificationsRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Url string `protobuf:"bytes,1,opt,name=url,proto3" json:"url,omitempty"` + Signature []byte `protobuf:"bytes,2,opt,name=signature,proto3" json:"signature,omitempty"` +} + +func (x *SubscribeNotificationsRequest) Reset() { + *x = SubscribeNotificationsRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_notifications_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *SubscribeNotificationsRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*SubscribeNotificationsRequest) ProtoMessage() {} + +func (x *SubscribeNotificationsRequest) ProtoReflect() protoreflect.Message { + mi := &file_notifications_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use SubscribeNotificationsRequest.ProtoReflect.Descriptor instead. +func (*SubscribeNotificationsRequest) Descriptor() ([]byte, []int) { + return file_notifications_proto_rawDescGZIP(), []int{0} +} + +func (x *SubscribeNotificationsRequest) GetUrl() string { + if x != nil { + return x.Url + } + return "" +} + +func (x *SubscribeNotificationsRequest) GetSignature() []byte { + if x != nil { + return x.Signature + } + return nil +} + +type SubscribeNotificationsReply struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields +} + +func (x *SubscribeNotificationsReply) Reset() { + *x = SubscribeNotificationsReply{} + if protoimpl.UnsafeEnabled { + mi := &file_notifications_proto_msgTypes[1] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *SubscribeNotificationsReply) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*SubscribeNotificationsReply) ProtoMessage() {} + +func (x *SubscribeNotificationsReply) ProtoReflect() protoreflect.Message { + mi := &file_notifications_proto_msgTypes[1] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use SubscribeNotificationsReply.ProtoReflect.Descriptor instead. +func (*SubscribeNotificationsReply) Descriptor() ([]byte, []int) { + return file_notifications_proto_rawDescGZIP(), []int{1} +} + +var File_notifications_proto protoreflect.FileDescriptor + +var file_notifications_proto_rawDesc = []byte{ + 0x0a, 0x13, 0x6e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, + 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x0d, 0x6e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, + 0x69, 0x6f, 0x6e, 0x73, 0x22, 0x4f, 0x0a, 0x1d, 0x53, 0x75, 0x62, 0x73, 0x63, 0x72, 0x69, 0x62, + 0x65, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x52, 0x65, + 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x10, 0x0a, 0x03, 0x75, 0x72, 0x6c, 0x18, 0x01, 0x20, 0x01, + 0x28, 0x09, 0x52, 0x03, 0x75, 0x72, 0x6c, 0x12, 0x1c, 0x0a, 0x09, 0x73, 0x69, 0x67, 0x6e, 0x61, + 0x74, 0x75, 0x72, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x09, 0x73, 0x69, 0x67, 0x6e, + 0x61, 0x74, 0x75, 0x72, 0x65, 0x22, 0x1d, 0x0a, 0x1b, 0x53, 0x75, 0x62, 0x73, 0x63, 0x72, 0x69, + 0x62, 0x65, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x52, + 0x65, 0x70, 0x6c, 0x79, 0x32, 0x85, 0x01, 0x0a, 0x0d, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x12, 0x74, 0x0a, 0x16, 0x53, 0x75, 0x62, 0x73, 0x63, 0x72, + 0x69, 0x62, 0x65, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, + 0x12, 0x2c, 0x2e, 0x6e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, + 0x2e, 0x53, 0x75, 0x62, 0x73, 0x63, 0x72, 0x69, 0x62, 0x65, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x69, + 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x2a, + 0x2e, 0x6e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x53, + 0x75, 0x62, 0x73, 0x63, 0x72, 0x69, 0x62, 0x65, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, + 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x22, 0x00, 0x42, 0x25, 0x5a, 0x23, + 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x62, 0x72, 0x65, 0x65, 0x7a, + 0x2f, 0x6c, 0x73, 0x70, 0x64, 0x2f, 0x6e, 0x6f, 0x74, 0x69, 0x66, 0x69, 0x63, 0x61, 0x74, 0x69, + 0x6f, 0x6e, 0x73, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, +} + +var ( + file_notifications_proto_rawDescOnce sync.Once + file_notifications_proto_rawDescData = file_notifications_proto_rawDesc +) + +func file_notifications_proto_rawDescGZIP() []byte { + file_notifications_proto_rawDescOnce.Do(func() { + file_notifications_proto_rawDescData = protoimpl.X.CompressGZIP(file_notifications_proto_rawDescData) + }) + return file_notifications_proto_rawDescData +} + +var file_notifications_proto_msgTypes = make([]protoimpl.MessageInfo, 2) +var file_notifications_proto_goTypes = []interface{}{ + (*SubscribeNotificationsRequest)(nil), // 0: notifications.SubscribeNotificationsRequest + (*SubscribeNotificationsReply)(nil), // 1: notifications.SubscribeNotificationsReply +} +var file_notifications_proto_depIdxs = []int32{ + 0, // 0: notifications.Notifications.SubscribeNotifications:input_type -> notifications.SubscribeNotificationsRequest + 1, // 1: notifications.Notifications.SubscribeNotifications:output_type -> notifications.SubscribeNotificationsReply + 1, // [1:2] is the sub-list for method output_type + 0, // [0:1] is the sub-list for method input_type + 0, // [0:0] is the sub-list for extension type_name + 0, // [0:0] is the sub-list for extension extendee + 0, // [0:0] is the sub-list for field type_name +} + +func init() { file_notifications_proto_init() } +func file_notifications_proto_init() { + if File_notifications_proto != nil { + return + } + if !protoimpl.UnsafeEnabled { + file_notifications_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*SubscribeNotificationsRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_notifications_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*SubscribeNotificationsReply); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } + type x struct{} + out := protoimpl.TypeBuilder{ + File: protoimpl.DescBuilder{ + GoPackagePath: reflect.TypeOf(x{}).PkgPath(), + RawDescriptor: file_notifications_proto_rawDesc, + NumEnums: 0, + NumMessages: 2, + NumExtensions: 0, + NumServices: 1, + }, + GoTypes: file_notifications_proto_goTypes, + DependencyIndexes: file_notifications_proto_depIdxs, + MessageInfos: file_notifications_proto_msgTypes, + }.Build() + File_notifications_proto = out.File + file_notifications_proto_rawDesc = nil + file_notifications_proto_goTypes = nil + file_notifications_proto_depIdxs = nil +} diff --git a/notifications/notifications.proto b/notifications/notifications.proto new file mode 100644 index 0000000..2590dff --- /dev/null +++ b/notifications/notifications.proto @@ -0,0 +1,18 @@ +syntax = "proto3"; + +option go_package = "github.com/breez/lspd/notifications"; + +package notifications; + +service Notifications { + rpc SubscribeNotifications(SubscribeNotificationsRequest) + returns (SubscribeNotificationsReply) {} +} + +message SubscribeNotificationsRequest { + string url = 1; + bytes signature = 2; +} + +message SubscribeNotificationsReply { +} \ No newline at end of file diff --git a/notifications/notifications_grpc.pb.go b/notifications/notifications_grpc.pb.go new file mode 100644 index 0000000..13a2f77 --- /dev/null +++ b/notifications/notifications_grpc.pb.go @@ -0,0 +1,105 @@ +// Code generated by protoc-gen-go-grpc. DO NOT EDIT. +// versions: +// - protoc-gen-go-grpc v1.2.0 +// - protoc v3.21.12 +// source: notifications.proto + +package notifications + +import ( + context "context" + grpc "google.golang.org/grpc" + codes "google.golang.org/grpc/codes" + status "google.golang.org/grpc/status" +) + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the grpc package it is being compiled against. +// Requires gRPC-Go v1.32.0 or later. +const _ = grpc.SupportPackageIsVersion7 + +// NotificationsClient is the client API for Notifications service. +// +// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream. +type NotificationsClient interface { + SubscribeNotifications(ctx context.Context, in *SubscribeNotificationsRequest, opts ...grpc.CallOption) (*SubscribeNotificationsReply, error) +} + +type notificationsClient struct { + cc grpc.ClientConnInterface +} + +func NewNotificationsClient(cc grpc.ClientConnInterface) NotificationsClient { + return ¬ificationsClient{cc} +} + +func (c *notificationsClient) SubscribeNotifications(ctx context.Context, in *SubscribeNotificationsRequest, opts ...grpc.CallOption) (*SubscribeNotificationsReply, error) { + out := new(SubscribeNotificationsReply) + err := c.cc.Invoke(ctx, "/notifications.Notifications/SubscribeNotifications", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + +// NotificationsServer is the server API for Notifications service. +// All implementations must embed UnimplementedNotificationsServer +// for forward compatibility +type NotificationsServer interface { + SubscribeNotifications(context.Context, *SubscribeNotificationsRequest) (*SubscribeNotificationsReply, error) + mustEmbedUnimplementedNotificationsServer() +} + +// UnimplementedNotificationsServer must be embedded to have forward compatible implementations. +type UnimplementedNotificationsServer struct { +} + +func (UnimplementedNotificationsServer) SubscribeNotifications(context.Context, *SubscribeNotificationsRequest) (*SubscribeNotificationsReply, error) { + return nil, status.Errorf(codes.Unimplemented, "method SubscribeNotifications not implemented") +} +func (UnimplementedNotificationsServer) mustEmbedUnimplementedNotificationsServer() {} + +// UnsafeNotificationsServer may be embedded to opt out of forward compatibility for this service. +// Use of this interface is not recommended, as added methods to NotificationsServer will +// result in compilation errors. +type UnsafeNotificationsServer interface { + mustEmbedUnimplementedNotificationsServer() +} + +func RegisterNotificationsServer(s grpc.ServiceRegistrar, srv NotificationsServer) { + s.RegisterService(&Notifications_ServiceDesc, srv) +} + +func _Notifications_SubscribeNotifications_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(SubscribeNotificationsRequest) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(NotificationsServer).SubscribeNotifications(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/notifications.Notifications/SubscribeNotifications", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(NotificationsServer).SubscribeNotifications(ctx, req.(*SubscribeNotificationsRequest)) + } + return interceptor(ctx, in, info, handler) +} + +// Notifications_ServiceDesc is the grpc.ServiceDesc for Notifications service. +// It's only intended for direct use with grpc.RegisterService, +// and not to be introspected or modified (even as a copy) +var Notifications_ServiceDesc = grpc.ServiceDesc{ + ServiceName: "notifications.Notifications", + HandlerType: (*NotificationsServer)(nil), + Methods: []grpc.MethodDesc{ + { + MethodName: "SubscribeNotifications", + Handler: _Notifications_SubscribeNotifications_Handler, + }, + }, + Streams: []grpc.StreamDesc{}, + Metadata: "notifications.proto", +} diff --git a/notifications/server.go b/notifications/server.go new file mode 100644 index 0000000..ab73a18 --- /dev/null +++ b/notifications/server.go @@ -0,0 +1,58 @@ +package notifications + +import ( + context "context" + "crypto/sha256" + "encoding/hex" + "fmt" + "log" + + "github.com/btcsuite/btcd/btcec/v2/ecdsa" +) + +var ErrInvalidSignature = fmt.Errorf("invalid signature") +var ErrInternal = fmt.Errorf("internal error") + +type server struct { + store Store + NotificationsServer +} + +func NewNotificationsServer(store Store) NotificationsServer { + return &server{ + store: store, + } +} + +func (s *server) SubscribeNotifications( + ctx context.Context, + request *SubscribeNotificationsRequest, +) (*SubscribeNotificationsReply, error) { + first := sha256.Sum256([]byte(request.Url)) + second := sha256.Sum256(first[:]) + pubkey, wasCompressed, err := ecdsa.RecoverCompact( + request.Signature, + second[:], + ) + if err != nil { + return nil, ErrInvalidSignature + } + + if !wasCompressed { + return nil, ErrInvalidSignature + } + + err = s.store.Register(ctx, hex.EncodeToString(pubkey.SerializeCompressed()), request.Url) + if err != nil { + log.Printf( + "failed to register %x for notifications on url %s: %v", + pubkey.SerializeCompressed(), + request.Url, + err, + ) + + return nil, ErrInternal + } + + return &SubscribeNotificationsReply{}, nil +} diff --git a/notifications/store.go b/notifications/store.go new file mode 100644 index 0000000..fb9a352 --- /dev/null +++ b/notifications/store.go @@ -0,0 +1,10 @@ +package notifications + +import ( + "context" +) + +type Store interface { + Register(ctx context.Context, pubkey string, url string) error + GetRegistrations(ctx context.Context, pubkey string) ([]string, error) +}