com.twitter.finagle.util.InetAddressUtil.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finagle-core_2.9.2 Show documentation

finagle-core

The newest version!

package com.twitter.finagle.core.util

import com.twitter.util.{NetUtil => UtilNetUtil}
import java.net.{InetAddress, Inet4Address}


object InetAddressUtil {

  val Loopback = InetAddress.getByAddress(Array[Byte](127,0,0,1))

  /** Check if string is an IPv4 private address. */
  def isPrivateAddress(ip: InetAddress): Boolean =
    ip match {
      case ip: Inet4Address =>
        val addr = ip.getAddress
        if (addr(0) == 10.toByte) // 10/8
          true
        else if (addr(0) == 172.toByte && (addr(1) & 0xf0) == 16.toByte) // 172/12
          true
        else if (addr(0) == 192.toByte && addr(1) == 168.toByte) // 192.168/16
          true
        else
            false
      case _ =>
        false
    }

  /**
   * Faster InetAddress.getByName that uses NetUtil to parse the address instead
   * of regular expressions.
   */
  def getByName(host: String): InetAddress = {
    UtilNetUtil.ipToOptionInt(host) match {
      case Some(i) =>
        val bytes = Array[Byte](
          ((i & 0xff000000) >> 24).toByte,
          ((i & 0x00ff0000) >> 16).toByte,
          ((i & 0x0000ff00) >>  8).toByte,
          ((i & 0x000000ff)      ).toByte)
        InetAddress.getByAddress(host, bytes)
      case None =>
        InetAddress.getByName(host)
    }
  }
}