start-os/sdk/base/lib/util/ip.ts

export class IpAddress {
  private renderedOctets: number[]
  protected constructor(
    public octets: number[],
    private renderedAddress: string,
  ) {
    this.renderedOctets = [...octets]
  }
  static parse(address: string): IpAddress {
    let octets
    if (address.includes(':')) {
      octets = new Array(16).fill(0)
      const segs = address.split(':')
      let idx = 0
      let octIdx = 0
      while (segs[idx]) {
        const num = parseInt(segs[idx], 16)
        octets[octIdx++] = num >> 8
        octets[octIdx++] = num & 255
        idx += 1
      }
      const lastSegIdx = segs.length - 1
      if (idx < lastSegIdx) {
        idx = lastSegIdx
        octIdx = 15
        while (segs[idx]) {
          const num = parseInt(segs[idx], 16)
          octets[octIdx--] = num & 255
          octets[octIdx--] = num >> 8
          idx -= 1
        }
      }
    } else {
      octets = address.split('.').map(Number)
      if (octets.length !== 4) throw new Error('invalid ipv4 address')
    }
    if (octets.some((o) => isNaN(o) || o > 255)) {
      throw new Error('invalid ip address')
    }
    return new IpAddress(octets, address)
  }
  static fromOctets(octets: number[]) {
    if (octets.length == 4) {
      if (octets.some((o) => o > 255)) {
        throw new Error('invalid ip address')
      }
      return new IpAddress(octets, octets.join('.'))
    } else if (octets.length == 16) {
      if (octets.some((o) => o > 255)) {
        throw new Error('invalid ip address')
      }
      let pre = octets.slice(0, 8)
      while (pre[pre.length - 1] == 0) {
        pre.pop()
      }
      let post = octets.slice(8)
      while (post[0] == 0) {
        post.unshift()
      }
      if (pre.length + post.length == 16) {
        return new IpAddress(octets, octets.join(':'))
      } else {
        return new IpAddress(octets, pre.join(':') + '::' + post.join(':'))
      }
    } else {
      throw new Error('invalid ip address')
    }
  }
  isIpv4(): boolean {
    return this.octets.length === 4
  }
  isIpv6(): boolean {
    return this.octets.length === 16
  }
  isPublic(): boolean {
    return this.isIpv4() && !PRIVATE_IPV4_RANGES.some((r) => r.contains(this))
  }
  add(n: number): IpAddress {
    let octets = [...this.octets]
    n = Math.floor(n)
    for (let i = octets.length - 1; i >= 0; i--) {
      octets[i] += n
      if (octets[i] > 255) {
        n = octets[i] >> 8
        octets[i] &= 255
      } else {
        break
      }
    }
    if (octets[0] > 255) {
      throw new Error('overflow incrementing ip')
    }
    return IpAddress.fromOctets(octets)
  }
  sub(n: number): IpAddress {
    let octets = [...this.octets]
    n = Math.floor(n)
    for (let i = octets.length - 1; i >= 0; i--) {
      octets[i] -= n
      if (octets[i] < 0) {
        n = Math.ceil(Math.abs(octets[i]) / 256)
        octets[i] = ((octets[i] % 256) + 256) % 256
      } else {
        break
      }
    }
    if (octets[0] < 0) {
      throw new Error('underflow decrementing ip')
    }
    return IpAddress.fromOctets(octets)
  }
  cmp(other: string | IpAddress): -1 | 0 | 1 {
    if (typeof other === 'string') other = IpAddress.parse(other)
    const len = Math.max(this.octets.length, other.octets.length)
    for (let i = 0; i < len; i++) {
      const left = this.octets[i] || 0
      const right = other.octets[i] || 0
      if (left > right) {
        return 1
      } else if (left < right) {
        return -1
      }
    }
    return 0
  }
  get address(): string {
    if (
      this.renderedOctets.length === this.octets.length &&
      this.renderedOctets.every((o, idx) => o === this.octets[idx])
    ) {
      // already rendered
    } else if (this.octets.length === 4) {
      this.renderedAddress = this.octets.join('.')
      this.renderedOctets = [...this.octets]
    } else if (this.octets.length === 16) {
      const contigZeros = this.octets.reduce(
        (acc, x, idx) => {
          if (x === 0) {
            acc.current++
          } else {
            acc.current = 0
          }
          if (acc.current > acc.end - acc.start) {
            acc.end = idx + 1
            acc.start = acc.end - acc.current
          }
          return acc
        },
        { start: 0, end: 0, current: 0 },
      )
      if (contigZeros.end - contigZeros.start >= 2) {
        return `${this.octets.slice(0, contigZeros.start).join(':')}::${this.octets.slice(contigZeros.end).join(':')}`
      }
      this.renderedAddress = this.octets.join(':')
      this.renderedOctets = [...this.octets]
    } else {
      console.warn('invalid octet length for IpAddress', this.octets)
    }
    return this.renderedAddress
  }
}

export class IpNet extends IpAddress {
  private constructor(
    octets: number[],
    public prefix: number,
    address: string,
  ) {
    super(octets, address)
  }
  static fromIpPrefix(ip: IpAddress, prefix: number): IpNet {
    if (prefix > ip.octets.length * 8) {
      throw new Error('invalid prefix')
    }
    return new IpNet(ip.octets, prefix, ip.address)
  }
  static parse(ipnet: string): IpNet {
    const [address, prefixStr] = ipnet.split('/', 2)
    const ip = IpAddress.parse(address)
    const prefix = Number(prefixStr)
    return IpNet.fromIpPrefix(ip, prefix)
  }
  contains(address: string | IpAddress | IpNet): boolean {
    if (typeof address === 'string') address = IpAddress.parse(address)
    if (address instanceof IpNet && address.prefix < this.prefix) return false
    if (this.octets.length !== address.octets.length) return false
    let prefix = this.prefix
    let idx = 0
    while (idx < this.octets.length && prefix >= 8) {
      if (this.octets[idx] !== address.octets[idx]) {
        return false
      }
      idx += 1
      prefix -= 8
    }
    if (prefix === 0 || idx >= this.octets.length) return true
    const mask = 255 ^ (255 >> prefix)
    return (this.octets[idx] & mask) === (address.octets[idx] & mask)
  }
  zero(): IpAddress {
    let octets: number[] = []
    let prefix = this.prefix
    for (let idx = 0; idx < this.octets.length; idx++) {
      if (prefix >= 8) {
        octets[idx] = this.octets[idx]
        prefix -= 8
      } else {
        const mask = 255 ^ (255 >> prefix)
        octets[idx] = this.octets[idx] & mask
        prefix = 0
      }
    }

    return IpAddress.fromOctets(octets)
  }
  broadcast(): IpAddress {
    let octets: number[] = []
    let prefix = this.prefix
    for (let idx = 0; idx < this.octets.length; idx++) {
      if (prefix >= 8) {
        octets[idx] = this.octets[idx]
        prefix -= 8
      } else {
        const mask = 255 >> prefix
        octets[idx] = this.octets[idx] | mask
        prefix = 0
      }
    }

    return IpAddress.fromOctets(octets)
  }
  get ipnet() {
    return `${this.address}/${this.prefix}`
  }
}

export const PRIVATE_IPV4_RANGES = [
  IpNet.parse('127.0.0.0/8'),
  IpNet.parse('10.0.0.0/8'),
  IpNet.parse('172.16.0.0/12'),
  IpNet.parse('192.168.0.0/16'),
]

export const IPV4_LOOPBACK = IpNet.parse('127.0.0.0/8')
export const IPV6_LOOPBACK = IpNet.parse('::1/128')
export const IPV6_LINK_LOCAL = IpNet.parse('fe80::/10')

export const CGNAT = IpNet.parse('100.64.0.0/10')