@libp2p/utils 4.0.6 → 4.0.7-0b4a2ee79

package/src/abstract-stream.ts

@@ -0,0 +1,509 @@
+import { CodeError } from '@libp2p/interface/errors'
+import { type Pushable, pushable } from 'it-pushable'
+import defer, { type DeferredPromise } from 'p-defer'
+import { raceSignal } from 'race-signal'
+import { Uint8ArrayList } from 'uint8arraylist'
+import { closeSource } from './close-source.js'
+import type { AbortOptions } from '@libp2p/interface'
+import type { Direction, ReadStatus, Stream, StreamStatus, StreamTimeline, WriteStatus } from '@libp2p/interface/connection'
+import type { Logger } from '@libp2p/logger'
+import type { Source } from 'it-stream-types'
+
+const ERR_STREAM_RESET = 'ERR_STREAM_RESET'
+const ERR_SINK_INVALID_STATE = 'ERR_SINK_INVALID_STATE'
+const DEFAULT_SEND_CLOSE_WRITE_TIMEOUT = 5000
+
+export interface AbstractStreamInit {
+  /**
+   * A unique identifier for this stream
+   */
+  id: string
+
+  /**
+   * The stream direction
+   */
+  direction: Direction
+
+  /**
+   * A Logger implementation used to log stream-specific information
+   */
+  log: Logger
+
+  /**
+   * User specific stream metadata
+   */
+  metadata?: Record<string, unknown>
+
+  /**
+   * Invoked when the stream ends
+   */
+  onEnd?(err?: Error | undefined): void
+
+  /**
+   * Invoked when the readable end of the stream is closed
+   */
+  onCloseRead?(): void
+
+  /**
+   * Invoked when the writable end of the stream is closed
+   */
+  onCloseWrite?(): void
+
+  /**
+   * Invoked when the the stream has been reset by the remote
+   */
+  onReset?(): void
+
+  /**
+   * Invoked when the the stream has errored
+   */
+  onAbort?(err: Error): void
+
+  /**
+   * How long to wait in ms for stream data to be written to the underlying
+   * connection when closing the writable end of the stream. (default: 500)
+   */
+  closeTimeout?: number
+
+  /**
+   * After the stream sink has closed, a limit on how long it takes to send
+   * a close-write message to the remote peer.
+   */
+  sendCloseWriteTimeout?: number
+}
+
+function isPromise <T = unknown> (thing: any): thing is Promise<T> {
+  if (thing == null) {
+    return false
+  }
+
+  return typeof thing.then === 'function' &&
+    typeof thing.catch === 'function' &&
+    typeof thing.finally === 'function'
+}
+
+export abstract class AbstractStream implements Stream {
+  public id: string
+  public direction: Direction
+  public timeline: StreamTimeline
+  public protocol?: string
+  public metadata: Record<string, unknown>
+  public source: AsyncGenerator<Uint8ArrayList, void, unknown>
+  public status: StreamStatus
+  public readStatus: ReadStatus
+  public writeStatus: WriteStatus
+  public readonly log: Logger
+
+  private readonly sinkController: AbortController
+  private readonly sinkEnd: DeferredPromise<void>
+  private readonly closed: DeferredPromise<void>
+  private endErr: Error | undefined
+  private readonly streamSource: Pushable<Uint8ArrayList>
+  private readonly onEnd?: (err?: Error | undefined) => void
+  private readonly onCloseRead?: () => void
+  private readonly onCloseWrite?: () => void
+  private readonly onReset?: () => void
+  private readonly onAbort?: (err: Error) => void
+  private readonly sendCloseWriteTimeout: number
+
+  constructor (init: AbstractStreamInit) {
+    this.sinkController = new AbortController()
+    this.sinkEnd = defer()
+    this.closed = defer()
+    this.log = init.log
+
+    // stream status
+    this.status = 'open'
+    this.readStatus = 'ready'
+    this.writeStatus = 'ready'
+
+    this.id = init.id
+    this.metadata = init.metadata ?? {}
+    this.direction = init.direction
+    this.timeline = {
+      open: Date.now()
+    }
+    this.sendCloseWriteTimeout = init.sendCloseWriteTimeout ?? DEFAULT_SEND_CLOSE_WRITE_TIMEOUT
+
+    this.onEnd = init.onEnd
+    this.onCloseRead = init?.onCloseRead
+    this.onCloseWrite = init?.onCloseWrite
+    this.onReset = init?.onReset
+    this.onAbort = init?.onAbort
+
+    this.source = this.streamSource = pushable<Uint8ArrayList>({
+      onEnd: (err) => {
+        if (err != null) {
+          this.log.trace('source ended with error', err)
+        } else {
+          this.log.trace('source ended')
+        }
+
+        this.onSourceEnd(err)
+      }
+    })
+
+    // necessary because the libp2p upgrader wraps the sink function
+    this.sink = this.sink.bind(this)
+  }
+
+  async sink (source: Source<Uint8ArrayList | Uint8Array>): Promise<void> {
+    if (this.writeStatus !== 'ready') {
+      throw new CodeError(`writable end state is "${this.writeStatus}" not "ready"`, ERR_SINK_INVALID_STATE)
+    }
+
+    try {
+      this.writeStatus = 'writing'
+
+      const options: AbortOptions = {
+        signal: this.sinkController.signal
+      }
+
+      if (this.direction === 'outbound') { // If initiator, open a new stream
+        const res = this.sendNewStream(options)
+
+        if (isPromise(res)) {
+          await res
+        }
+      }
+
+      const abortListener = (): void => {
+        closeSource(source, this.log)
+      }
+
+      try {
+        this.sinkController.signal.addEventListener('abort', abortListener)
+
+        this.log.trace('sink reading from source')
+
+        for await (let data of source) {
+          data = data instanceof Uint8Array ? new Uint8ArrayList(data) : data
+
+          const res = this.sendData(data, options)
+
+          if (isPromise(res)) { // eslint-disable-line max-depth
+            await res
+          }
+        }
+      } finally {
+        this.sinkController.signal.removeEventListener('abort', abortListener)
+      }
+
+      this.log.trace('sink finished reading from source, write status is "%s"', this.writeStatus)
+
+      if (this.writeStatus === 'writing') {
+        this.writeStatus = 'closing'
+
+        this.log.trace('send close write to remote')
+        await this.sendCloseWrite({
+          signal: AbortSignal.timeout(this.sendCloseWriteTimeout)
+        })
+
+        this.writeStatus = 'closed'
+      }
+
+      this.onSinkEnd()
+    } catch (err: any) {
+      this.log.trace('sink ended with error, calling abort with error', err)
+      this.abort(err)
+
+      throw err
+    } finally {
+      this.log.trace('resolve sink end')
+      this.sinkEnd.resolve()
+    }
+  }
+
+  protected onSourceEnd (err?: Error): void {
+    if (this.timeline.closeRead != null) {
+      return
+    }
+
+    this.timeline.closeRead = Date.now()
+    this.readStatus = 'closed'
+
+    if (err != null && this.endErr == null) {
+      this.endErr = err
+    }
+
+    this.onCloseRead?.()
+
+    if (this.timeline.closeWrite != null) {
+      this.log.trace('source and sink ended')
+      this.timeline.close = Date.now()
+
+      if (this.status !== 'aborted' && this.status !== 'reset') {
+        this.status = 'closed'
+      }
+
+      if (this.onEnd != null) {
+        this.onEnd(this.endErr)
+      }
+
+      this.closed.resolve()
+    } else {
+      this.log.trace('source ended, waiting for sink to end')
+    }
+  }
+
+  protected onSinkEnd (err?: Error): void {
+    if (this.timeline.closeWrite != null) {
+      return
+    }
+
+    this.timeline.closeWrite = Date.now()
+    this.writeStatus = 'closed'
+
+    if (err != null && this.endErr == null) {
+      this.endErr = err
+    }
+
+    this.onCloseWrite?.()
+
+    if (this.timeline.closeRead != null) {
+      this.log.trace('sink and source ended')
+      this.timeline.close = Date.now()
+
+      if (this.status !== 'aborted' && this.status !== 'reset') {
+        this.status = 'closed'
+      }
+
+      if (this.onEnd != null) {
+        this.onEnd(this.endErr)
+      }
+
+      this.closed.resolve()
+    } else {
+      this.log.trace('sink ended, waiting for source to end')
+    }
+  }
+
+  // Close for both Reading and Writing
+  async close (options?: AbortOptions): Promise<void> {
+    this.log.trace('closing gracefully')
+
+    this.status = 'closing'
+
+    await Promise.all([
+      this.closeRead(options),
+      this.closeWrite(options)
+    ])
+
+    // wait for read and write ends to close
+    await raceSignal(this.closed.promise, options?.signal)
+
+    this.status = 'closed'
+
+    this.log.trace('closed gracefully')
+  }
+
+  async closeRead (options: AbortOptions = {}): Promise<void> {
+    if (this.readStatus === 'closing' || this.readStatus === 'closed') {
+      return
+    }
+
+    this.log.trace('closing readable end of stream with starting read status "%s"', this.readStatus)
+
+    const readStatus = this.readStatus
+    this.readStatus = 'closing'
+
+    if (this.status !== 'reset' && this.status !== 'aborted' && this.timeline.closeRead == null) {
+      this.log.trace('send close read to remote')
+      await this.sendCloseRead(options)
+    }
+
+    if (readStatus === 'ready') {
+      this.log.trace('ending internal source queue with %d queued bytes', this.streamSource.readableLength)
+      this.streamSource.end()
+    }
+
+    this.log.trace('closed readable end of stream')
+  }
+
+  async closeWrite (options: AbortOptions = {}): Promise<void> {
+    if (this.writeStatus === 'closing' || this.writeStatus === 'closed') {
+      return
+    }
+
+    this.log.trace('closing writable end of stream with starting write status "%s"', this.writeStatus)
+
+    if (this.writeStatus === 'ready') {
+      this.log.trace('sink was never sunk, sink an empty array')
+
+      await raceSignal(this.sink([]), options.signal)
+    }
+
+    if (this.writeStatus === 'writing') {
+      // stop reading from the source passed to `.sink` in the microtask queue
+      // - this lets any data queued by the user in the current tick get read
+      // before we exit
+      await new Promise((resolve, reject) => {
+        queueMicrotask(() => {
+          this.log.trace('aborting source passed to .sink')
+          this.sinkController.abort()
+          raceSignal(this.sinkEnd.promise, options.signal)
+            .then(resolve, reject)
+        })
+      })
+    }
+
+    this.writeStatus = 'closed'
+
+    this.log.trace('closed writable end of stream')
+  }
+
+  /**
+   * Close immediately for reading and writing and send a reset message (local
+   * error)
+   */
+  abort (err: Error): void {
+    if (this.status === 'closed' || this.status === 'aborted' || this.status === 'reset') {
+      return
+    }
+
+    this.log('abort with error', err)
+
+    // try to send a reset message
+    this.log('try to send reset to remote')
+    const res = this.sendReset()
+
+    if (isPromise(res)) {
+      res.catch((err) => {
+        this.log.error('error sending reset message', err)
+      })
+    }
+
+    this.status = 'aborted'
+    this.timeline.abort = Date.now()
+    this._closeSinkAndSource(err)
+    this.onAbort?.(err)
+  }
+
+  /**
+   * Receive a reset message - close immediately for reading and writing (remote
+   * error)
+   */
+  reset (): void {
+    if (this.status === 'closed' || this.status === 'aborted' || this.status === 'reset') {
+      return
+    }
+
+    const err = new CodeError('stream reset', ERR_STREAM_RESET)
+
+    this.status = 'reset'
+    this.timeline.reset = Date.now()
+    this._closeSinkAndSource(err)
+    this.onReset?.()
+  }
+
+  _closeSinkAndSource (err?: Error): void {
+    this._closeSink(err)
+    this._closeSource(err)
+  }
+
+  _closeSink (err?: Error): void {
+    // if the sink function is running, cause it to end
+    if (this.writeStatus === 'writing') {
+      this.log.trace('end sink source')
+      this.sinkController.abort()
+    }
+
+    this.onSinkEnd(err)
+  }
+
+  _closeSource (err?: Error): void {
+    // if the source is not ending, end it
+    if (this.readStatus !== 'closing' && this.readStatus !== 'closed') {
+      this.log.trace('ending source with %d bytes to be read by consumer', this.streamSource.readableLength)
+      this.readStatus = 'closing'
+      this.streamSource.end(err)
+    }
+  }
+
+  /**
+   * The remote closed for writing so we should expect to receive no more
+   * messages
+   */
+  remoteCloseWrite (): void {
+    if (this.readStatus === 'closing' || this.readStatus === 'closed') {
+      this.log('received remote close write but local source is already closed')
+      return
+    }
+
+    this.log.trace('remote close write')
+    this._closeSource()
+  }
+
+  /**
+   * The remote closed for reading so we should not send any more
+   * messages
+   */
+  remoteCloseRead (): void {
+    if (this.writeStatus === 'closing' || this.writeStatus === 'closed') {
+      this.log('received remote close read but local sink is already closed')
+      return
+    }
+
+    this.log.trace('remote close read')
+    this._closeSink()
+  }
+
+  /**
+   * The underlying muxer has closed, no more messages can be sent or will
+   * be received, close immediately to free up resources
+   */
+  destroy (): void {
+    if (this.status === 'closed' || this.status === 'aborted' || this.status === 'reset') {
+      this.log('received destroy but we are already closed')
+      return
+    }
+
+    this.log.trace('stream destroyed')
+
+    this._closeSinkAndSource()
+  }
+
+  /**
+   * When an extending class reads data from it's implementation-specific source,
+   * call this method to allow the stream consumer to read the data.
+   */
+  sourcePush (data: Uint8ArrayList): void {
+    this.streamSource.push(data)
+  }
+
+  /**
+   * Returns the amount of unread data - can be used to prevent large amounts of
+   * data building up when the stream consumer is too slow.
+   */
+  sourceReadableLength (): number {
+    return this.streamSource.readableLength
+  }
+
+  /**
+   * Send a message to the remote muxer informing them a new stream is being
+   * opened
+   */
+  abstract sendNewStream (options?: AbortOptions): void | Promise<void>
+
+  /**
+   * Send a data message to the remote muxer
+   */
+  abstract sendData (buf: Uint8ArrayList, options?: AbortOptions): void | Promise<void>
+
+  /**
+   * Send a reset message to the remote muxer
+   */
+  abstract sendReset (options?: AbortOptions): void | Promise<void>
+
+  /**
+   * Send a message to the remote muxer, informing them no more data messages
+   * will be sent by this end of the stream
+   */
+  abstract sendCloseWrite (options?: AbortOptions): void | Promise<void>
+
+  /**
+   * Send a message to the remote muxer, informing them no more data messages
+   * will be read by this end of the stream
+   */
+  abstract sendCloseRead (options?: AbortOptions): void | Promise<void>
+}

package/src/close-source.ts

@@ -0,0 +1,14 @@
+import { getIterator } from 'get-iterator'
+import { isPromise } from './is-promise.js'
+import type { Logger } from '@libp2p/logger'
+import type { Source } from 'it-stream-types'
+
+export function closeSource (source: Source<unknown>, log: Logger): void {
+  const res = getIterator(source).return?.()
+
+  if (isPromise(res)) {
+    res.catch(err => {
+      log.error('could not cause iterator to return', err)
+    })
+  }
+}

package/src/ip-port-to-multiaddr.ts

@@ -1,10 +1,7 @@
 import { isIPv4, isIPv6 } from '@chainsafe/is-ip'
 import { CodeError } from '@libp2p/interface/errors'
-import { logger } from '@libp2p/logger'
 import { type Multiaddr, multiaddr } from '@multiformats/multiaddr'
 
-const log = logger('libp2p:ip-port-to-multiaddr')
-
 export const Errors = {
   ERR_INVALID_IP_PARAMETER: 'ERR_INVALID_IP_PARAMETER',
   ERR_INVALID_PORT_PARAMETER: 'ERR_INVALID_PORT_PARAMETER',
@@ -35,7 +32,5 @@ export function ipPortToMultiaddr (ip: string, port: number | string): Multiaddr
     return multiaddr(`/ip6/${ip}/tcp/${port}`)
   }
 
-  const errMsg = `invalid ip:port for creating a multiaddr: ${ip}:${port}`
-  log.error(errMsg)
-  throw new CodeError(errMsg, Errors.ERR_INVALID_IP)
+  throw new CodeError(`invalid ip:port for creating a multiaddr: ${ip}:${port}`, Errors.ERR_INVALID_IP)
 }

package/src/is-promise.ts

@@ -0,0 +1,9 @@
+export function isPromise <T = unknown> (thing: any): thing is Promise<T> {
+  if (thing == null) {
+    return false
+  }
+
+  return typeof thing.then === 'function' &&
+    typeof thing.catch === 'function' &&
+    typeof thing.finally === 'function'
+}

package/src/peer-job-queue.ts

@@ -0,0 +1,118 @@
+/* eslint-disable @typescript-eslint/no-non-null-assertion */
+
+import { CodeError, ERR_INVALID_PARAMETERS } from '@libp2p/interface/errors'
+import PQueue from 'p-queue'
+import type { PeerId } from '@libp2p/interface/peer-id'
+import type { QueueAddOptions, Options, Queue } from 'p-queue'
+
+// Port of lower_bound from https://en.cppreference.com/w/cpp/algorithm/lower_bound
+// Used to compute insertion index to keep queue sorted after insertion
+function lowerBound<T> (array: readonly T[], value: T, comparator: (a: T, b: T) => number): number {
+  let first = 0
+  let count = array.length
+
+  while (count > 0) {
+    const step = Math.trunc(count / 2)
+    let it = first + step
+
+    if (comparator(array[it]!, value) <= 0) {
+      first = ++it
+      count -= step + 1
+    } else {
+      count = step
+    }
+  }
+
+  return first
+}
+
+interface RunFunction { (): Promise<unknown> }
+
+export interface PeerPriorityQueueOptions extends QueueAddOptions {
+  peerId: PeerId
+}
+
+interface PeerJob {
+  priority: number
+  peerId: PeerId
+  run: RunFunction
+}
+
+/**
+ * Port of https://github.com/sindresorhus/p-queue/blob/main/source/priority-queue.ts
+ * that adds support for filtering jobs by peer id
+ */
+class PeerPriorityQueue implements Queue<RunFunction, PeerPriorityQueueOptions> {
+  readonly #queue: PeerJob[] = []
+
+  enqueue (run: RunFunction, options?: Partial<PeerPriorityQueueOptions>): void {
+    const peerId = options?.peerId
+    const priority = options?.priority ?? 0
+
+    if (peerId == null) {
+      throw new CodeError('missing peer id', ERR_INVALID_PARAMETERS)
+    }
+
+    const element: PeerJob = {
+      priority,
+      peerId,
+      run
+    }
+
+    if (this.size > 0 && this.#queue[this.size - 1]!.priority >= priority) {
+      this.#queue.push(element)
+      return
+    }
+
+    const index = lowerBound(
+      this.#queue, element,
+      (a: Readonly<PeerPriorityQueueOptions>, b: Readonly<PeerPriorityQueueOptions>) => b.priority! - a.priority!
+    )
+    this.#queue.splice(index, 0, element)
+  }
+
+  dequeue (): RunFunction | undefined {
+    const item = this.#queue.shift()
+    return item?.run
+  }
+
+  filter (options: Readonly<Partial<PeerPriorityQueueOptions>>): RunFunction[] {
+    if (options.peerId != null) {
+      const peerId = options.peerId
+
+      return this.#queue.filter(
+        (element: Readonly<PeerPriorityQueueOptions>) => peerId.equals(element.peerId)
+      ).map((element: Readonly<{ run: RunFunction }>) => element.run)
+    }
+
+    return this.#queue.filter(
+      (element: Readonly<PeerPriorityQueueOptions>) => element.priority === options.priority
+    ).map((element: Readonly<{ run: RunFunction }>) => element.run)
+  }
+
+  get size (): number {
+    return this.#queue.length
+  }
+}
+
+/**
+ * Extends PQueue to add support for querying queued jobs by peer id
+ */
+export class PeerJobQueue extends PQueue<PeerPriorityQueue, PeerPriorityQueueOptions> {
+  constructor (options: Options<PeerPriorityQueue, PeerPriorityQueueOptions> = {}) {
+    super({
+      ...options,
+      queueClass: PeerPriorityQueue
+    })
+  }
+
+  /**
+   * Returns true if this queue has a job for the passed peer id that has not yet
+   * started to run
+   */
+  hasJob (peerId: PeerId): boolean {
+    return this.sizeBy({
+      peerId
+    }) > 0
+  }
+}