@libp2p/perf 1.1.15 → 2.0.0-fb8a6f188

package/src/perf-service.ts

@@ -0,0 +1,207 @@
+import { logger } from '@libp2p/logger'
+import { pushable } from 'it-pushable'
+import { MAX_INBOUND_STREAMS, MAX_OUTBOUND_STREAMS, PROTOCOL_NAME, RUN_ON_TRANSIENT_CONNECTION, WRITE_BLOCK_SIZE } from './constants.js'
+import type { PerfOptions, PerfOutput, PerfServiceComponents, PerfServiceInit, PerfService as PerfServiceInterface } from './index.js'
+import type { Startable } from '@libp2p/interface/startable'
+import type { IncomingStreamData } from '@libp2p/interface-internal/registrar'
+import type { Multiaddr } from '@multiformats/multiaddr'
+
+const log = logger('libp2p:perf')
+
+export class PerfService implements Startable, PerfServiceInterface {
+  public readonly protocol: string
+  private readonly components: PerfServiceComponents
+  private started: boolean
+  private readonly databuf: ArrayBuffer
+  private readonly writeBlockSize: number
+  private readonly maxInboundStreams: number
+  private readonly maxOutboundStreams: number
+  private readonly runOnTransientConnection: boolean
+
+  constructor (components: PerfServiceComponents, init: PerfServiceInit = {}) {
+    this.components = components
+    this.started = false
+    this.protocol = init.protocolName ?? PROTOCOL_NAME
+    this.writeBlockSize = init.writeBlockSize ?? WRITE_BLOCK_SIZE
+    this.databuf = new ArrayBuffer(this.writeBlockSize)
+    this.maxInboundStreams = init.maxInboundStreams ?? MAX_INBOUND_STREAMS
+    this.maxOutboundStreams = init.maxOutboundStreams ?? MAX_OUTBOUND_STREAMS
+    this.runOnTransientConnection = init.runOnTransientConnection ?? RUN_ON_TRANSIENT_CONNECTION
+  }
+
+  async start (): Promise<void> {
+    await this.components.registrar.handle(this.protocol, (data: IncomingStreamData) => {
+      void this.handleMessage(data).catch((err) => {
+        log.error('error handling perf protocol message', err)
+      })
+    }, {
+      maxInboundStreams: this.maxInboundStreams,
+      maxOutboundStreams: this.maxOutboundStreams,
+      runOnTransientConnection: this.runOnTransientConnection
+    })
+    this.started = true
+  }
+
+  async stop (): Promise<void> {
+    await this.components.registrar.unhandle(this.protocol)
+    this.started = false
+  }
+
+  isStarted (): boolean {
+    return this.started
+  }
+
+  async handleMessage (data: IncomingStreamData): Promise<void> {
+    const { stream } = data
+
+    try {
+      const writeBlockSize = this.writeBlockSize
+
+      let bytesToSendBack: number | undefined
+
+      for await (const buf of stream.source) {
+        if (bytesToSendBack == null) {
+          // downcast 64 to 52 bits to avoid bigint arithmetic performance penalty
+          bytesToSendBack = Number(buf.getBigUint64(0, false))
+        }
+        // Ingest all the bufs and wait for the read side to close
+      }
+
+      if (bytesToSendBack == null) {
+        throw new Error('bytesToSendBack was not set')
+      }
+
+      const uint8Buf = new Uint8Array(this.databuf)
+
+      await stream.sink(async function * () {
+        while (bytesToSendBack > 0) {
+          let toSend: number = writeBlockSize
+          if (toSend > bytesToSendBack) {
+            toSend = bytesToSendBack
+          }
+
+          bytesToSendBack = bytesToSendBack - toSend
+          yield uint8Buf.subarray(0, toSend)
+        }
+      }())
+    } catch (err: any) {
+      stream.abort(err)
+    }
+  }
+
+  async * measurePerformance (ma: Multiaddr, sendBytes: number, receiveBytes: number, options: PerfOptions = {}): AsyncGenerator<PerfOutput> {
+    log('opening stream on protocol %s to %a', this.protocol, ma)
+
+    const uint8Buf = new Uint8Array(this.databuf)
+    const writeBlockSize = this.writeBlockSize
+
+    // start time should include connection establishment
+    const initialStartTime = Date.now()
+    const connection = await this.components.connectionManager.openConnection(ma, {
+      ...options,
+      force: options.reuseExistingConnection !== true
+    })
+    const stream = await connection.newStream(this.protocol, options)
+
+    let lastAmountOfBytesSent = 0
+    let lastReportedTime = Date.now()
+    let totalBytesSent = 0
+
+    // tell the remote how many bytes we will send. Up cast to 64 bit number
+    // as if we send as ui32 we limit total transfer size to 4GB
+    const view = new DataView(this.databuf)
+    view.setBigUint64(0, BigInt(receiveBytes), false)
+
+    log('sending %i bytes to %p', sendBytes, connection.remotePeer)
+
+    try {
+      const sendOutput = pushable<PerfOutput>({
+        objectMode: true
+      })
+
+      void stream.sink(async function * () {
+        // Send the number of bytes to receive
+        yield uint8Buf.subarray(0, 8)
+
+        while (sendBytes > 0) {
+          options.signal?.throwIfAborted()
+
+          let toSend: number = writeBlockSize
+          if (toSend > sendBytes) {
+            toSend = sendBytes
+          }
+          sendBytes = sendBytes - toSend
+          yield uint8Buf.subarray(0, toSend)
+
+          if (Date.now() - lastReportedTime > 1000) {
+            sendOutput.push({
+              type: 'intermediary',
+              timeSeconds: (Date.now() - lastReportedTime) / 1000,
+              uploadBytes: lastAmountOfBytesSent,
+              downloadBytes: 0
+            })
+
+            // record last reported time after `console.log` because it can
+            // affect benchmark timings
+            lastReportedTime = Date.now()
+            lastAmountOfBytesSent = 0
+          }
+
+          lastAmountOfBytesSent += toSend
+          totalBytesSent += toSend
+        }
+
+        sendOutput.end()
+      }())
+        .catch(err => {
+          sendOutput.end(err)
+        })
+
+      yield * sendOutput
+
+      // Read the received bytes
+      let lastAmountOfBytesReceived = 0
+      lastReportedTime = Date.now()
+      let totalBytesReceived = 0
+
+      for await (const buf of stream.source) {
+        options.signal?.throwIfAborted()
+
+        if (Date.now() - lastReportedTime > 1000) {
+          yield {
+            type: 'intermediary',
+            timeSeconds: (Date.now() - lastReportedTime) / 1000,
+            uploadBytes: 0,
+            downloadBytes: lastAmountOfBytesReceived
+          }
+
+          // record last reported time after `console.log` because it can
+          // affect benchmark timings
+          lastReportedTime = Date.now()
+          lastAmountOfBytesReceived = 0
+        }
+
+        lastAmountOfBytesReceived += buf.byteLength
+        totalBytesReceived += buf.byteLength
+      }
+
+      if (totalBytesReceived !== receiveBytes) {
+        throw new Error(`Expected to receive ${receiveBytes} bytes, but received ${totalBytesReceived}`)
+      }
+
+      yield {
+        type: 'final',
+        timeSeconds: (Date.now() - initialStartTime) / 1000,
+        uploadBytes: totalBytesSent,
+        downloadBytes: totalBytesReceived
+      }
+
+      log('performed %s to %p', this.protocol, connection.remotePeer)
+      await stream.close()
+    } catch (err: any) {
+      log('error sending %s bytes to %p: %s', totalBytesSent, connection.remotePeer, err)
+      stream.abort(err)
+      throw err
+    }
+  }
+}

package/dist/src/main.d.ts

@@ -1,2 +0,0 @@
-export declare function main(runServer: boolean, serverIpAddress: string, transport: string, uploadBytes: number, downloadBytes: number): Promise<void>;
-//# sourceMappingURL=main.d.ts.map

package/dist/src/main.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"main.d.ts","sourceRoot":"","sources":["../../src/main.ts"],"names":[],"mappings":"AAgDA,wBAAsB,IAAI,CAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAiDrJ"}

package/dist/src/main.js

@@ -1,107 +0,0 @@
-import { noise } from '@chainsafe/libp2p-noise';
-import { yamux } from '@chainsafe/libp2p-yamux';
-import { unmarshalPrivateKey } from '@libp2p/crypto/keys';
-import { createFromPrivKey } from '@libp2p/peer-id-factory';
-import { tcp } from '@libp2p/tcp';
-import { multiaddr } from '@multiformats/multiaddr';
-import { createLibp2p } from 'libp2p';
-import { fromString as uint8ArrayFromString } from 'uint8arrays/from-string';
-import yargs from 'yargs';
-import { hideBin } from 'yargs/helpers';
-import { defaultInit, perfService } from '../src/index.js';
-const argv = yargs(hideBin(process.argv))
-    .options({
-    'run-server': {
-        type: 'boolean',
-        demandOption: true,
-        default: false,
-        description: 'Whether to run as a server'
-    },
-    'server-address': {
-        type: 'string',
-        demandOption: false,
-        description: 'Server IP address',
-        default: ''
-    },
-    transport: {
-        type: 'string',
-        demandOption: false,
-        description: 'Transport to use',
-        default: 'tcp'
-    },
-    'upload-bytes': {
-        type: 'number',
-        demandOption: false,
-        description: 'Number of bytes to upload',
-        default: 0
-    },
-    'download-bytes': {
-        type: 'number',
-        demandOption: false,
-        description: 'Number of bytes to download',
-        default: 0
-    }
-})
-    .command('help', 'Print usage information', yargs.help)
-    .parseSync();
-export async function main(runServer, serverIpAddress, transport, uploadBytes, downloadBytes) {
-    const listenAddrs = [];
-    const { host, port } = splitHostPort(serverIpAddress);
-    // #TODO: right now we only support tcp
-    const tcpMultiaddr = multiaddr(`/ip4/${host}/tcp/${port}`);
-    const config = {
-        transports: [tcp()],
-        streamMuxers: [yamux()],
-        connectionEncryption: [
-            noise()
-        ],
-        services: {
-            perf: perfService(defaultInit)
-        }
-    };
-    const testPrivKey = 'CAASpgkwggSiAgEAAoIBAQC2SKo/HMFZeBml1AF3XijzrxrfQXdJzjePBZAbdxqKR1Mc6juRHXij6HXYPjlAk01BhF1S3Ll4Lwi0cAHhggf457sMg55UWyeGKeUv0ucgvCpBwlR5cQ020i0MgzjPWOLWq1rtvSbNcAi2ZEVn6+Q2EcHo3wUvWRtLeKz+DZSZfw2PEDC+DGPJPl7f8g7zl56YymmmzH9liZLNrzg/qidokUv5u1pdGrcpLuPNeTODk0cqKB+OUbuKj9GShYECCEjaybJDl9276oalL9ghBtSeEv20kugatTvYy590wFlJkkvyl+nPxIH0EEYMKK9XRWlu9XYnoSfboiwcv8M3SlsjAgMBAAECggEAZtju/bcKvKFPz0mkHiaJcpycy9STKphorpCT83srBVQi59CdFU6Mj+aL/xt0kCPMVigJw8P3/YCEJ9J+rS8BsoWE+xWUEsJvtXoT7vzPHaAtM3ci1HZd302Mz1+GgS8Epdx+7F5p80XAFLDUnELzOzKftvWGZmWfSeDnslwVONkL/1VAzwKy7Ce6hk4SxRE7l2NE2OklSHOzCGU1f78ZzVYKSnS5Ag9YrGjOAmTOXDbKNKN/qIorAQ1bovzGoCwx3iGIatQKFOxyVCyO1PsJYT7JO+kZbhBWRRE+L7l+ppPER9bdLFxs1t5CrKc078h+wuUr05S1P1JjXk68pk3+kQKBgQDeK8AR11373Mzib6uzpjGzgNRMzdYNuExWjxyxAzz53NAR7zrPHvXvfIqjDScLJ4NcRO2TddhXAfZoOPVH5k4PJHKLBPKuXZpWlookCAyENY7+Pd55S8r+a+MusrMagYNljb5WbVTgN8cgdpim9lbbIFlpN6SZaVjLQL3J8TWH6wKBgQDSChzItkqWX11CNstJ9zJyUE20I7LrpyBJNgG1gtvz3ZMUQCn3PxxHtQzN9n1P0mSSYs+jBKPuoSyYLt1wwe10/lpgL4rkKWU3/m1Myt0tveJ9WcqHh6tzcAbb/fXpUFT/o4SWDimWkPkuCb+8j//2yiXk0a/T2f36zKMuZvujqQKBgC6B7BAQDG2H2B/ijofp12ejJU36nL98gAZyqOfpLJ+FeMz4TlBDQ+phIMhnHXA5UkdDapQ+zA3SrFk+6yGk9Vw4Hf46B+82SvOrSbmnMa+PYqKYIvUzR4gg34rL/7AhwnbEyD5hXq4dHwMNsIDq+l2elPjwm/U9V0gdAl2+r50HAoGALtsKqMvhv8HucAMBPrLikhXP/8um8mMKFMrzfqZ+otxfHzlhI0L08Bo3jQrb0Z7ByNY6M8epOmbCKADsbWcVre/AAY0ZkuSZK/CaOXNX/AhMKmKJh8qAOPRY02LIJRBCpfS4czEdnfUhYV/TYiFNnKRj57PPYZdTzUsxa/yVTmECgYBr7slQEjb5Onn5mZnGDh+72BxLNdgwBkhO0OCdpdISqk0F0Pxby22DFOKXZEpiyI9XYP1C8wPiJsShGm2yEwBPWXnrrZNWczaVuCbXHrZkWQogBDG3HGXNdU4MAWCyiYlyinIBpPpoAJZSzpGLmWbMWh28+RJS6AQX6KHrK1o2uw==';
-    const encoded = uint8ArrayFromString(testPrivKey, 'base64pad');
-    const privateKey = await unmarshalPrivateKey(encoded);
-    const peerId = await createFromPrivKey(privateKey);
-    const tcpMultiaddrAddress = `${tcpMultiaddr.toString()}/p2p/${peerId.toString()}`;
-    if (runServer) {
-        listenAddrs.push(tcpMultiaddrAddress);
-        Object.assign(config, {
-            peerId,
-            addresses: {
-                listen: listenAddrs
-            }
-        });
-    }
-    const node = await createLibp2p(config);
-    await node.start();
-    const startTime = Date.now();
-    if (!runServer) {
-        const connection = await node.dial(multiaddr(tcpMultiaddrAddress));
-        const duration = await node.services.perf.measurePerformance(startTime, connection, BigInt(uploadBytes), BigInt(downloadBytes));
-        // Output latency to stdout in seconds
-        // eslint-disable-next-line no-console
-        console.log(JSON.stringify({ latency: duration / 1000 }));
-        await node.stop();
-    }
-}
-function splitHostPort(address) {
-    try {
-        const parts = address.split(':');
-        const host = parts[0];
-        const port = parts[1];
-        return {
-            host,
-            port
-        };
-    }
-    catch (error) {
-        throw Error('Invalid server address');
-    }
-}
-main(argv['run-server'], argv['server-address'], argv.transport, argv['upload-bytes'], argv['download-bytes']).catch((err) => {
-    // eslint-disable-next-line no-console
-    console.error(err);
-    process.exit(1);
-});
-//# sourceMappingURL=main.js.map

package/dist/src/main.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"main.js","sourceRoot":"","sources":["../../src/main.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAA;AAC/C,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAA;AAC/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAA;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAA;AAC3D,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AACjC,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,EAAE,UAAU,IAAI,oBAAoB,EAAE,MAAM,yBAAyB,CAAA;AAC5E,OAAO,KAAK,MAAM,OAAO,CAAA;AACzB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAA;AACvC,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAA;AAE1D,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;KACtC,OAAO,CAAC;IACP,YAAY,EAAE;QACZ,IAAI,EAAE,SAAS;QACf,YAAY,EAAE,IAAI;QAClB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,4BAA4B;KAC1C;IACD,gBAAgB,EAAE;QAChB,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,KAAK;QACnB,WAAW,EAAE,mBAAmB;QAChC,OAAO,EAAE,EAAE;KACZ;IACD,SAAS,EAAE;QACT,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,KAAK;QACnB,WAAW,EAAE,kBAAkB;QAC/B,OAAO,EAAE,KAAK;KACf;IACD,cAAc,EAAE;QACd,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,KAAK;QACnB,WAAW,EAAE,2BAA2B;QACxC,OAAO,EAAE,CAAC;KACX;IACD,gBAAgB,EAAE;QAChB,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,KAAK;QACnB,WAAW,EAAE,6BAA6B;QAC1C,OAAO,EAAE,CAAC;KACX;CACF,CAAC;KACD,OAAO,CAAC,MAAM,EAAE,yBAAyB,EAAE,KAAK,CAAC,IAAI,CAAC;KACtD,SAAS,EAAE,CAAA;AAEd,MAAM,CAAC,KAAK,UAAU,IAAI,CAAE,SAAkB,EAAE,eAAuB,EAAE,SAAiB,EAAE,WAAmB,EAAE,aAAqB;IACpI,MAAM,WAAW,GAAa,EAAE,CAAA;IAEhC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,aAAa,CAAC,eAAe,CAAC,CAAA;IACrD,uCAAuC;IACvC,MAAM,YAAY,GAAG,SAAS,CAAC,QAAQ,IAAI,QAAQ,IAAI,EAAE,CAAC,CAAA;IAE1D,MAAM,MAAM,GAAG;QACb,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC;QACnB,YAAY,EAAE,CAAC,KAAK,EAAE,CAAC;QACvB,oBAAoB,EAAE;YACpB,KAAK,EAAE;SACR;QACD,QAAQ,EAAE;YACR,IAAI,EAAE,WAAW,CAAC,WAAW,CAAC;SAC/B;KACF,CAAA;IAED,MAAM,WAAW,GAAG,8jDAA8jD,CAAA;IACllD,MAAM,OAAO,GAAG,oBAAoB,CAAC,WAAW,EAAE,WAAW,CAAC,CAAA;IAC9D,MAAM,UAAU,GAAG,MAAM,mBAAmB,CAAC,OAAO,CAAC,CAAA;IACrD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,UAAU,CAAC,CAAA;IAClD,MAAM,mBAAmB,GAAG,GAAG,YAAY,CAAC,QAAQ,EAAE,QAAQ,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAA;IAEjF,IAAI,SAAS,EAAE;QACb,WAAW,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAErC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;YACpB,MAAM;YACN,SAAS,EAAE;gBACT,MAAM,EAAE,WAAW;aACpB;SACF,CAAC,CAAA;KACH;IAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,CAAA;IAEvC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAA;IAElB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;IAE5B,IAAI,CAAC,SAAS,EAAE;QACd,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC,CAAA;QAClE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC,CAAA;QAC/H,sCAAsC;QACtC,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,QAAQ,GAAG,IAAI,EAAE,CAAC,CAAC,CAAA;QACzD,MAAM,IAAI,CAAC,IAAI,EAAE,CAAA;KAClB;AACH,CAAC;AAED,SAAS,aAAa,CAAE,OAAe;IACrC,IAAI;QACF,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;QAChC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAA;QACrB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAA;QACrB,OAAO;YACL,IAAI;YACJ,IAAI;SACL,CAAA;KACF;IAAC,OAAO,KAAK,EAAE;QACd,MAAM,KAAK,CAAC,wBAAwB,CAAC,CAAA;KACtC;AACH,CAAC;AAED,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAC3H,sCAAsC;IACtC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAClB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;AACjB,CAAC,CAAC,CAAA"}

package/dist/typedoc-urls.json

@@ -1,12 +0,0 @@
-{
-  "PerfService": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_perf.PerfService.html",
-  ".:PerfService": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_perf.PerfService.html",
-  "PerfServiceComponents": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_perf.PerfServiceComponents.html",
-  ".:PerfServiceComponents": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_perf.PerfServiceComponents.html",
-  "PerfServiceInit": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_perf.PerfServiceInit.html",
-  ".:PerfServiceInit": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_perf.PerfServiceInit.html",
-  "defaultInit": "https://libp2p.github.io/js-libp2p/variables/_libp2p_perf.defaultInit.html",
-  ".:defaultInit": "https://libp2p.github.io/js-libp2p/variables/_libp2p_perf.defaultInit.html",
-  "perfService": "https://libp2p.github.io/js-libp2p/functions/_libp2p_perf.perfService-1.html",
-  ".:perfService": "https://libp2p.github.io/js-libp2p/functions/_libp2p_perf.perfService-1.html"
-}

package/src/main.ts

@@ -1,118 +0,0 @@
-import { noise } from '@chainsafe/libp2p-noise'
-import { yamux } from '@chainsafe/libp2p-yamux'
-import { unmarshalPrivateKey } from '@libp2p/crypto/keys'
-import { createFromPrivKey } from '@libp2p/peer-id-factory'
-import { tcp } from '@libp2p/tcp'
-import { multiaddr } from '@multiformats/multiaddr'
-import { createLibp2p } from 'libp2p'
-import { fromString as uint8ArrayFromString } from 'uint8arrays/from-string'
-import yargs from 'yargs'
-import { hideBin } from 'yargs/helpers'
-import { defaultInit, perfService } from '../src/index.js'
-
-const argv = yargs(hideBin(process.argv))
-  .options({
-    'run-server': {
-      type: 'boolean',
-      demandOption: true,
-      default: false,
-      description: 'Whether to run as a server'
-    },
-    'server-address': {
-      type: 'string',
-      demandOption: false,
-      description: 'Server IP address',
-      default: ''
-    },
-    transport: {
-      type: 'string',
-      demandOption: false,
-      description: 'Transport to use',
-      default: 'tcp'
-    },
-    'upload-bytes': {
-      type: 'number',
-      demandOption: false,
-      description: 'Number of bytes to upload',
-      default: 0
-    },
-    'download-bytes': {
-      type: 'number',
-      demandOption: false,
-      description: 'Number of bytes to download',
-      default: 0
-    }
-  })
-  .command('help', 'Print usage information', yargs.help)
-  .parseSync()
-
-export async function main (runServer: boolean, serverIpAddress: string, transport: string, uploadBytes: number, downloadBytes: number): Promise<void> {
-  const listenAddrs: string[] = []
-
-  const { host, port } = splitHostPort(serverIpAddress)
-  // #TODO: right now we only support tcp
-  const tcpMultiaddr = multiaddr(`/ip4/${host}/tcp/${port}`)
-
-  const config = {
-    transports: [tcp()],
-    streamMuxers: [yamux()],
-    connectionEncryption: [
-      noise()
-    ],
-    services: {
-      perf: perfService(defaultInit)
-    }
-  }
-
-  const testPrivKey = 'CAASpgkwggSiAgEAAoIBAQC2SKo/HMFZeBml1AF3XijzrxrfQXdJzjePBZAbdxqKR1Mc6juRHXij6HXYPjlAk01BhF1S3Ll4Lwi0cAHhggf457sMg55UWyeGKeUv0ucgvCpBwlR5cQ020i0MgzjPWOLWq1rtvSbNcAi2ZEVn6+Q2EcHo3wUvWRtLeKz+DZSZfw2PEDC+DGPJPl7f8g7zl56YymmmzH9liZLNrzg/qidokUv5u1pdGrcpLuPNeTODk0cqKB+OUbuKj9GShYECCEjaybJDl9276oalL9ghBtSeEv20kugatTvYy590wFlJkkvyl+nPxIH0EEYMKK9XRWlu9XYnoSfboiwcv8M3SlsjAgMBAAECggEAZtju/bcKvKFPz0mkHiaJcpycy9STKphorpCT83srBVQi59CdFU6Mj+aL/xt0kCPMVigJw8P3/YCEJ9J+rS8BsoWE+xWUEsJvtXoT7vzPHaAtM3ci1HZd302Mz1+GgS8Epdx+7F5p80XAFLDUnELzOzKftvWGZmWfSeDnslwVONkL/1VAzwKy7Ce6hk4SxRE7l2NE2OklSHOzCGU1f78ZzVYKSnS5Ag9YrGjOAmTOXDbKNKN/qIorAQ1bovzGoCwx3iGIatQKFOxyVCyO1PsJYT7JO+kZbhBWRRE+L7l+ppPER9bdLFxs1t5CrKc078h+wuUr05S1P1JjXk68pk3+kQKBgQDeK8AR11373Mzib6uzpjGzgNRMzdYNuExWjxyxAzz53NAR7zrPHvXvfIqjDScLJ4NcRO2TddhXAfZoOPVH5k4PJHKLBPKuXZpWlookCAyENY7+Pd55S8r+a+MusrMagYNljb5WbVTgN8cgdpim9lbbIFlpN6SZaVjLQL3J8TWH6wKBgQDSChzItkqWX11CNstJ9zJyUE20I7LrpyBJNgG1gtvz3ZMUQCn3PxxHtQzN9n1P0mSSYs+jBKPuoSyYLt1wwe10/lpgL4rkKWU3/m1Myt0tveJ9WcqHh6tzcAbb/fXpUFT/o4SWDimWkPkuCb+8j//2yiXk0a/T2f36zKMuZvujqQKBgC6B7BAQDG2H2B/ijofp12ejJU36nL98gAZyqOfpLJ+FeMz4TlBDQ+phIMhnHXA5UkdDapQ+zA3SrFk+6yGk9Vw4Hf46B+82SvOrSbmnMa+PYqKYIvUzR4gg34rL/7AhwnbEyD5hXq4dHwMNsIDq+l2elPjwm/U9V0gdAl2+r50HAoGALtsKqMvhv8HucAMBPrLikhXP/8um8mMKFMrzfqZ+otxfHzlhI0L08Bo3jQrb0Z7ByNY6M8epOmbCKADsbWcVre/AAY0ZkuSZK/CaOXNX/AhMKmKJh8qAOPRY02LIJRBCpfS4czEdnfUhYV/TYiFNnKRj57PPYZdTzUsxa/yVTmECgYBr7slQEjb5Onn5mZnGDh+72BxLNdgwBkhO0OCdpdISqk0F0Pxby22DFOKXZEpiyI9XYP1C8wPiJsShGm2yEwBPWXnrrZNWczaVuCbXHrZkWQogBDG3HGXNdU4MAWCyiYlyinIBpPpoAJZSzpGLmWbMWh28+RJS6AQX6KHrK1o2uw=='
-  const encoded = uint8ArrayFromString(testPrivKey, 'base64pad')
-  const privateKey = await unmarshalPrivateKey(encoded)
-  const peerId = await createFromPrivKey(privateKey)
-  const tcpMultiaddrAddress = `${tcpMultiaddr.toString()}/p2p/${peerId.toString()}`
-
-  if (runServer) {
-    listenAddrs.push(tcpMultiaddrAddress)
-
-    Object.assign(config, {
-      peerId,
-      addresses: {
-        listen: listenAddrs
-      }
-    })
-  }
-
-  const node = await createLibp2p(config)
-
-  await node.start()
-
-  const startTime = Date.now()
-
-  if (!runServer) {
-    const connection = await node.dial(multiaddr(tcpMultiaddrAddress))
-    const duration = await node.services.perf.measurePerformance(startTime, connection, BigInt(uploadBytes), BigInt(downloadBytes))
-    // Output latency to stdout in seconds
-    // eslint-disable-next-line no-console
-    console.log(JSON.stringify({ latency: duration / 1000 }))
-    await node.stop()
-  }
-}
-
-function splitHostPort (address: string): { host: string, port?: string } {
-  try {
-    const parts = address.split(':')
-    const host = parts[0]
-    const port = parts[1]
-    return {
-      host,
-      port
-    }
-  } catch (error) {
-    throw Error('Invalid server address')
-  }
-}
-
-main(argv['run-server'], argv['server-address'], argv.transport, argv['upload-bytes'], argv['download-bytes']).catch((err) => {
-  // eslint-disable-next-line no-console
-  console.error(err)
-  process.exit(1)
-})