Validate that operator account has positive balance

Signed-off-by: Luis Mastrangelo <[email protected]>

Author: acuarica

packages/relay/src/lib/relay.ts

@@ -329,4 +329,14 @@ export class Relay {
   mirrorClient(): MirrorNodeClient {
     return this.mirrorNodeClient;
   }
+
+  async ensureOperatorHasBalance() {
+    if (!ConfigService.get('READ_ONLY')) {
+      const operator = this.clientMain.operatorAccountId!.toString();
+      const balance = await this.ethImpl.getBalance(operator, 'latest', {} as RequestDetails);
+      if (BigInt(balance) === BigInt(0)) {
+        throw new Error(`Operator account \`${operator}\` has no balance`);
+      }
+    }
+  }
 }

packages/server/src/index.ts

@@ -1,10 +1,18 @@
 // SPDX-License-Identifier: Apache-2.0
 
-import app from './server';
 import { ConfigService } from '@hashgraph/json-rpc-config-service/dist/services';
+
 import { setServerTimeout } from './koaJsonRpc/lib/utils'; // Import the 'setServerTimeout' function from the correct location
+import app, { logger, relay } from './server';
 
 async function main() {
+  try {
+    await relay.ensureOperatorHasBalance();
+  } catch (error) {
+    logger.fatal(error);
+    process.exit(1);
+  }
+
   const server = app.listen({ port: ConfigService.get('SERVER_PORT'), host: ConfigService.get('SERVER_HOST') });
 
   // set request timeout to ensure sockets are closed after specified time of inactivity

packages/server/src/server.ts

@@ -26,9 +26,9 @@ const mainLogger = pino({
   },
 });
 
-const logger = mainLogger.child({ name: 'rpc-server' });
+export const logger = mainLogger.child({ name: 'rpc-server' });
 const register = new Registry();
-const relay: Relay = new Relay(logger.child({ name: 'relay' }), register);
+export const relay: Relay = new Relay(logger.child({ name: 'relay' }), register);
 const app = new KoaJsonRpc(logger.child({ name: 'koa-rpc' }), register, relay, {
   limit: ConfigService.get('INPUT_SIZE_LIMIT') + 'mb',
 });

packages/ws-server/src/index.ts

@@ -1,15 +1,21 @@
 // SPDX-License-Identifier: Apache-2.0
 
-import { app, httpApp } from './webSocketServer';
-import constants from '@hashgraph/json-rpc-relay/dist/lib/constants';
 import { ConfigService } from '@hashgraph/json-rpc-config-service/dist/services';
+import constants from '@hashgraph/json-rpc-relay/dist/lib/constants';
+
+import { app, httpApp, logger, relay } from './webSocketServer';
 
 async function main() {
+  try {
+    await relay.ensureOperatorHasBalance();
+  } catch (error) {
+    logger.fatal(error);
+    process.exit(1);
+  }
+
   const host = ConfigService.get('SERVER_HOST');
   app.listen({ port: constants.WEB_SOCKET_PORT, host });
   httpApp.listen({ port: constants.WEB_SOCKET_HTTP_PORT, host });
 }
 
-(async () => {
-  await main();
-})();
+main();

packages/ws-server/src/webSocketServer.ts

@@ -237,4 +237,4 @@ process.on('uncaughtException', (err) => {
   logger.error(err, 'Uncaught Exception!');
 });
 
-export { app, httpApp };
+export { app, httpApp, relay, logger };