fmt

AztecProtocol · Jan 19, 2025 · cc0ca50 · cc0ca50
1 parent b755ab5
commit cc0ca50
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 6 deletions.
diff --git a/yarn-project/p2p/src/services/libp2p/libp2p_service.ts b/yarn-project/p2p/src/services/libp2p/libp2p_service.ts
@@ -132,6 +132,12 @@ export class LibP2PService<T extends P2PClientType> extends WithTracer implement
       this.reqresp,
     );
 
+    // Update gossipsub score params
+    this.node.services.pubsub.score.params.appSpecificScore = (peerId: string) => {
+      return this.peerManager.getPeerScore(peerId);
+    };
+    this.node.services.pubsub.score.params.appSpecificWeight = 10;
+
     this.attestationValidator = new AttestationValidator(epochCache);
     this.blockProposalValidator = new BlockProposalValidator(epochCache);
     this.epochProofQuoteValidator = new EpochProofQuoteValidator(epochCache);

diff --git a/yarn-project/p2p/src/services/peer-manager/peer_scoring.ts b/yarn-project/p2p/src/services/peer-manager/peer_scoring.ts
@@ -88,7 +88,7 @@ export class PeerScoring {
   }
 
   public getScoreState(peerId: string): PeerScoreState {
-    // TODO: permanently store banned peers?
+    // TODO(#11329): permanently store banned peers?
     const score = this.getScore(peerId);
     if (score < MIN_SCORE_BEFORE_BAN) {
       return PeerScoreState.Banned;

diff --git a/yarn-project/p2p/src/services/reqresp/protocols/goodbye.ts b/yarn-project/p2p/src/services/reqresp/protocols/goodbye.ts
@@ -54,7 +54,7 @@ export function prettyGoodbyeReason(reason: GoodByeReason): string {
       return 'low_score';
     case GoodByeReason.BANNED:
       return 'banned';
-    // TODO(https://github.com/AztecProtocol/aztec-packages/issues/11328): implement
+    // TODO(#11328): implement
     case GoodByeReason.WRONG_NETWORK:
       return 'wrong_network';
     case GoodByeReason.UNKNOWN:
@@ -81,8 +81,12 @@ export class GoodbyeProtocolHandler {
 }
 
 /**
- * Handles the goodbye request.
- * @param _msg - The goodbye request message.
+ * Handles the goodbye request. In request response, the goodbye request is handled by the peer manager.
+ *
+ * @dev Implemented as returning a function as the function is bound in the libp2p service, however
+ *      its implementation is here to keep functionality together.
+ *
+ * @param peerManager - The peer manager.
  * @returns A resolved promise with the goodbye response.
  */
 export function reqGoodbyeHandler(peerManager: PeerManager): ReqRespSubProtocolHandler {

diff --git a/...rc/services/reqresp/rate_limiter/index.ts → ...rc/services/reqresp/rate-limiter/index.ts b/...rc/services/reqresp/rate_limiter/index.ts → ...rc/services/reqresp/rate-limiter/index.ts
diff --git a/...reqresp/rate_limiter/rate_limiter.test.ts → ...reqresp/rate-limiter/rate_limiter.test.ts b/...reqresp/rate_limiter/rate_limiter.test.ts → ...reqresp/rate-limiter/rate_limiter.test.ts
diff --git a/...ices/reqresp/rate_limiter/rate_limiter.ts → ...ices/reqresp/rate-limiter/rate_limiter.ts b/...ices/reqresp/rate_limiter/rate_limiter.ts → ...ices/reqresp/rate-limiter/rate_limiter.ts
diff --git a/...vices/reqresp/rate_limiter/rate_limits.ts → ...vices/reqresp/rate-limiter/rate_limits.ts b/...vices/reqresp/rate_limiter/rate_limits.ts → ...vices/reqresp/rate-limiter/rate_limits.ts
@@ -32,7 +32,6 @@ export const DEFAULT_RATE_LIMITS: ReqRespSubProtocolRateLimits = {
       quotaCount: 10,
     },
   },
-  // TODO(md): feels like goodbye is an exception to the rule here
   [ReqRespSubProtocol.GOODBYE]: {
     peerLimit: {
       quotaTimeMs: 1000,

diff --git a/yarn-project/p2p/src/services/reqresp/reqresp.test.ts b/yarn-project/p2p/src/services/reqresp/reqresp.test.ts
@@ -15,7 +15,7 @@ import {
   startNodes,
   stopNodes,
 } from '../../mocks/index.js';
-import { PeerManager } from '../peer-manager/peer_manager.js';
+import { type PeerManager } from '../peer-manager/peer_manager.js';
 import { type PeerScoring } from '../peer-manager/peer_scoring.js';
 import { ReqRespSubProtocol, RequestableBuffer } from './interface.js';
 import { GoodByeReason, reqGoodbyeHandler } from './protocols/goodbye.js';