work

Author: asiegel-jt

src/discof/rpcserver/fd_rpc_history.c

@@ -208,6 +208,12 @@ fd_rpc_history_save(fd_rpc_history_t * hist, fd_replay_notif_msg_t * info) {
   } FD_SPAD_FRAME_END;
 }
 
+void
+fd_rpc_history_save_shred(fd_rpc_history_t * hist, fd_shred_t * shred) {
+  (void)hist;
+  (void)shred;
+}
+
 ulong
 fd_rpc_history_first_slot(fd_rpc_history_t * hist) {
   return hist->first_slot;

src/discof/rpcserver/fd_rpc_history.h

@@ -2,6 +2,7 @@
 #define HEADER_fd_src_discof_rpcserver_fd_rpc_history_h
 
 #include "fd_rpc_service.h"
+#include "../../ballet/shred/fd_shred.h"
 
 struct fd_rpc_history;
 typedef struct fd_rpc_history fd_rpc_history_t;
@@ -15,6 +16,8 @@ fd_rpc_history_t * fd_rpc_history_create(fd_rpcserver_args_t * args);
 
 void fd_rpc_history_save(fd_rpc_history_t * hist, fd_replay_notif_msg_t * msg);
 
+void fd_rpc_history_save_shred(fd_rpc_history_t * hist, fd_shred_t * shred);
+
 ulong fd_rpc_history_first_slot(fd_rpc_history_t * hist);
 
 ulong fd_rpc_history_latest_slot(fd_rpc_history_t * hist);

src/discof/rpcserver/fd_rpc_service.c

@@ -6,6 +6,7 @@
 #include "../../flamenco/runtime/fd_acc_mgr.h"
 #include "../../ballet/base58/fd_base58.h"
 #include "../../ballet/base64/fd_base64.h"
+#include "../../ballet/shred/fd_shred.h"
 #include "fd_rpc_history.h"
 #include "fd_block_to_json.h"
 #include "keywords.h"
@@ -2434,3 +2435,19 @@ fd_rpc_stake_after_frag(fd_rpc_ctx_t * ctx, fd_multi_epoch_leaders_t * state) {
   (void)ctx;
   fd_multi_epoch_leaders_stake_msg_fini( state );
 }
+
+void
+fd_rpc_shred_repair_during_frag(fd_rpc_ctx_t * ctx, uchar * shred, void const * msg, int sz) {
+  (void)ctx;
+  FD_TEST( sz <= (int)FD_SHRED_MAX_SZ );
+  fd_memcpy(shred, msg, (ulong)sz);
+}
+
+void
+fd_rpc_shred_repair_after_frag(fd_rpc_ctx_t * ctx, uchar * shred) {
+  fd_rpc_global_ctx_t * subs = ctx->global;
+  fd_shred_t * shred_p = (fd_shred_t *)shred;
+  if( fd_shred_is_data( shred_p->variant ) ) {
+     fd_rpc_history_save_shred( subs->history, shred_p );
+ }
+}

src/discof/rpcserver/fd_rpc_service.h

@@ -43,4 +43,8 @@ void fd_rpc_stake_during_frag(fd_rpc_ctx_t * ctx, fd_multi_epoch_leaders_t * sta
 
 void fd_rpc_stake_after_frag(fd_rpc_ctx_t * ctx, fd_multi_epoch_leaders_t * state);
 
+void fd_rpc_shred_repair_during_frag(fd_rpc_ctx_t * ctx, uchar * shred, void const * msg, int sz);
+
+void fd_rpc_shred_repair_after_frag(fd_rpc_ctx_t * ctx, uchar * shred);
+
 #endif /* HEADER_fd_src_discof_rpcserver_fd_rpc_service_h */

src/discof/rpcserver/fd_rpcserv_tile.c

@@ -1,16 +1,15 @@
 /* Repair tile runs the repair protocol for a Firedancer node. */
 #define _GNU_SOURCE
 
-#include "../../disco/topo/fd_topo.h"
 #include <sys/socket.h>
 
+#include "../../disco/topo/fd_topo.h"
 #include "generated/fd_rpcserv_tile_seccomp.h"
-
 #include "../rpcserver/fd_rpc_service.h"
-
 #include "../../flamenco/leaders/fd_multi_epoch_leaders.h"
 #include "../../disco/keyguard/fd_keyload.h"
 #include "../../disco/keyguard/fd_keyswitch.h"
+#include "../../ballet/shred/fd_shred.h"
 
 #include <fcntl.h>
 #include <unistd.h>
@@ -39,6 +38,7 @@ struct fd_rpcserv_tile_ctx {
   fd_keyswitch_t * keyswitch;
 
   fd_replay_notif_msg_t replay_notif_in_state;
+  uchar                 shred_in_state[ FD_SHRED_MAX_SZ ];
 
   uchar               in_kind[ MAX_IN_LINKS ];
   fd_rpcserv_in_ctx_t in_links[ MAX_IN_LINKS ];
@@ -109,6 +109,8 @@ during_frag( fd_rpcserv_tile_ctx_t * ctx,
     fd_rpc_replay_during_frag( ctx->ctx, &ctx->replay_notif_in_state, fd_chunk_to_laddr_const( ctx->in_links[in_idx].mem, chunk ), (int)sz );
   } else if( FD_UNLIKELY( kind == IN_KIND_STAKE_OUT ) ) {
     fd_rpc_stake_during_frag( ctx->ctx, ctx->args.leaders, fd_chunk_to_laddr_const( ctx->in_links[in_idx].mem, chunk ), (int)sz );
+  } else if( FD_UNLIKELY( kind == IN_KIND_SHRED_REPAIR ) ) {
+    fd_rpc_shred_repair_during_frag( ctx->ctx, ctx->shred_in_state, fd_chunk_to_laddr_const( ctx->in_links[in_idx].mem, chunk ), (int)sz );
   } else {
     FD_LOG_ERR(("Unknown in_idx %lu for rpc", in_idx));
   }
@@ -135,6 +137,8 @@ after_frag( fd_rpcserv_tile_ctx_t * ctx,
     fd_rpc_replay_after_frag( ctx->ctx, &ctx->replay_notif_in_state );
   } else if( FD_UNLIKELY( kind == IN_KIND_STAKE_OUT ) ) {
     fd_rpc_stake_after_frag( ctx->ctx, ctx->args.leaders );
+  } else if( FD_UNLIKELY( kind == IN_KIND_SHRED_REPAIR ) ) {
+    fd_rpc_shred_repair_after_frag( ctx->ctx, ctx->shred_in_state );
   } else {
     FD_LOG_ERR(("Unknown in_idx %lu for rpc", in_idx));
   }