"{ Encoding: utf8 }"

"

 COPYRIGHT (c) 1995 by Claus Gittinger

	      All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

"{ Package: 'stx:libbasic2' }"

"{ NameSpace: Smalltalk }"

SocketAddress variableByteSubclass:#IPSocketAddress

	instanceVariableNames:''

	classVariableNames:'CacheInvalidationTimeInterval AddressCacheSize NameCacheSize'

	poolDictionaries:''

	category:'OS-Sockets'

!

IPSocketAddress class instanceVariableNames:'addrCache nameCache'

"

 No other class instance variables are inherited by this class.

"

!

!IPSocketAddress class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 1995 by Claus Gittinger

	      All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

!

documentation

"

    Instances of IPSocketAddress represent tcp/ip-domain socket addresses.

    These consist of an ip address (4 bytes) and a port number.

    Notice that for performance, name and address translations are cached here

    for some time, in order to speed up heavy lookup such as when operating a webserver or similar

    application which does many address-to-hostname translations.

    For systems, when the name translation is fast (unix), you may want to disable it,

    as it prevents new addresses to be detected for some time (for example, if the own address

    changes due to a new dhcp address being acquired).

    You can also change the caching-interval time 

    (see caching protocol, flush*Cache, cachingIntervalTime, *CacheSize).

    [author:]

        Claus Gittinger

"

! !

!IPSocketAddress class methodsFor:'instance creation'!

addressString:aString

    "convert an address given in a dot notation like '123.456.78.9'.

     Handles IPv6SocketAddresses too (if the string contains colons or brackets)."

    (self == IPSocketAddress and:[aString includesAny:'[:']) ifTrue:[

        ^ IPv6SocketAddress hostAddress:(IPv6SocketAddress hostAddressFromString:aString).

    ].

    ^ self hostAddress:(self hostAddressFromString:aString).

    "

     IPSocketAddress addressString:'1.2.3.4'

     IPSocketAddress addressString:'2001:4dd0:ffa3::1'

     IPSocketAddress addressString:'[2001:4dd0:ffa3::1]'

    "

    "Modified (comment): / 13-06-2018 / 10:59:41 / Claus Gittinger"

!

allForHostName:name serviceName:portNrOrName type:socketTypeSymbol

    "get a collection of new instances, given a hostname, port or service and type.

     Multi-homed hosts return more than one entry.

     Redefined to cache the result of the name-lookup.

     Take care, this resolves to IPv4-addresses only!!

     Use SocketAddress to resolve to both IPv4 and IPv6-addresses."

    |addressList|

    portNrOrName isString ifTrue:[

        ^ super allForHostName:name serviceName:portNrOrName type:socketTypeSymbol

    ].

    addressList := self addressCacheAt:name.

    addressList notNil ifTrue:[

        addressList := addressList collect:[:eachSocketAddress|

                                eachSocketAddress copy

                                    port:(portNrOrName ? 0);

                                    yourself

                            ].

    ] ifFalse:[

        addressList := super allForHostName:name serviceName:portNrOrName type:socketTypeSymbol.

        addressList notNil ifTrue:[

            self addressCacheAt:name put:(addressList deepCopy).

        ].

    ].

    ^ addressList

    "

     IPSocketAddress allForHostName:nil serviceName:10 type:#stream

     SocketAddress allForHostName:'localhost' serviceName:10 type:#stream

     IPSocketAddress allForHostName:'localhost' serviceName:'echo' type:#datagram

     IPSocketAddress allForHostName:'www.google.com' serviceName:80 type:nil

     IPSocketAddress allForHostName:'localhost' serviceName:'echo' type:#datagram

     IPv6SocketAddress allForHostName:'localhost' serviceName:'echo' type:#datagram

    "

    "Modified: / 17-06-2009 / 15:21:00 / sr"

    "Modified (comment): / 21-02-2017 / 20:59:52 / stefan"

    "Modified (comment): / 13-06-2018 / 11:00:04 / Claus Gittinger"

!

hostAddress:aByteArray

    aByteArray size = 16 ifTrue:[

        ^ IPv6SocketAddress new hostAddress:aByteArray.

    ].

    ^ self new hostAddress:aByteArray.

    "Created: / 02-03-2020 / 19:03:44 / Stefan Vogel"

!

localHost

    "get a new instance representing the local-host address as seen internally"

    ^ self hostAddress:self local

    "

     self localHost

     self localHost hostName

    "

    "Modified: / 04-06-2007 / 21:34:22 / cg"

!

localHostAddress

    "NO NO NEVER!!

     Do not use this, because:

     1. It does not work on hosts that get their IP address via dhcp (at least on unix/linux)

     2. Usually OperatingSystem>>#getHostName does not return a fully qualified domain name

     3. This does not work well on multi-homed hosts"

    <resource: #obsolete>

    self obsoleteMethodWarning:'Do not use this method'.

    ^ self hostName:(OperatingSystem getHostName).

    "

      self localHostAddress

    "

    "Modified: / 04-06-2007 / 21:17:25 / cg"

! !

!IPSocketAddress class methodsFor:'addressing'!

anyAddress

    "return the anonymous address's bytes"

    ^ #[0 0 0 0]

    "Modified (comment): / 13-06-2018 / 11:00:59 / Claus Gittinger"

!

anyPort

    "return the anon port number"

    ^ 0

!

broadcastAddress

    "return the broadcast address"

    ^ #[255 255 255 255]

!

firstUnreservedPort

    "return the first unreserved port number"

    ^ 1024

!

local

    "return IN_ADDR_ANY, the address matching any local address"

    ^ #[127 0 0 1]

!

maxPort

    "return the maximum port number"

    ^ 16rffff

!

thisHost

    <resource: #obsolete>

    "return the bytes of IN_ADDR_ANY, the address matching any local address"

    self obsoleteMethodWarning.

    ^ self anyAddress

! !

!IPSocketAddress class methodsFor:'caching'!

addressCacheAt:aHostName

    "retrieve a cached address for aHostName,

     or nil if not in the cache"

    |addrAndTime addressList time|

    addrCache notNil ifTrue:[

        addrAndTime := addrCache at:aHostName ifAbsent:nil.

        addrAndTime notNil ifTrue:[

            addressList := addrAndTime key.

            time := addrAndTime value.

            ((Timestamp now) secondDeltaFrom:time) > self cacheInvalidationTimeInterval ifFalse:[

                ^ addressList.

            ]

        ].

    ].

    ^ nil

    "Modified: / 26-04-2007 / 11:21:37 / cg"

    "Modified (comment): / 19-09-2018 / 12:28:21 / Claus Gittinger"

!

addressCacheAt:aName put:anAddressList

    "put a aHostName -> list-of-addresses into the cache"

    (addrCache isNil or:[addrCache size > self addressCacheSize]) ifTrue:[

        addrCache := Dictionary new.

    ].

    addrCache at:aName put:(anAddressList -> Timestamp now).

    "Modified: / 26-04-2007 / 11:23:26 / cg"

    "Modified (comment): / 13-06-2018 / 11:04:00 / Claus Gittinger"

!

addressCacheSize

    "get the number of cached name entries.

     If never set explicitly, the default is 50 names"

    ^ AddressCacheSize ? 50

    "Created: / 26-04-2007 / 11:22:40 / cg"

    "Modified (comment): / 13-06-2018 / 11:02:49 / Claus Gittinger"

!

addressCacheSize:aNumber

    "set the number of cached name entries.

     If never set explicitly, the default is 50 names"

    AddressCacheSize := aNumber

    "Created: / 26-04-2007 / 11:24:05 / cg"

    "Modified (comment): / 13-06-2018 / 11:02:57 / Claus Gittinger"

!

cacheInvalidationTimeInterval

    "return the cache invalidation interval.

     If never set explicitly, the default is 30 seconds"

    ^ CacheInvalidationTimeInterval ? 30

    "Created: / 26-04-2007 / 11:21:31 / cg"

    "Modified (comment): / 13-06-2018 / 11:01:55 / Claus Gittinger"

!

cacheInvalidationTimeInterval:seconds

    "set the cache invalidation interval.

     If never set explicitly, the default is 30 seconds"

    CacheInvalidationTimeInterval := seconds

    "Created: / 26-04-2007 / 11:29:28 / cg"

    "Modified (comment): / 13-06-2018 / 11:02:07 / Claus Gittinger"

!

flushAddressCache

    addrCache := nil

    "

     self flushAddressCache

    "

!

flushNameCache

    nameCache := Dictionary new.

    "

     self flushNameCache

    "

    "Modified: / 26-04-2007 / 11:25:32 / cg"

!

nameCacheAt:aHostAddress

    "return a cached hostname, given a host address.

     The cache is flushed after cacheInvalidationTime seconds"

    |nameAndTime name time|

    nameCache notNil ifTrue:[

        nameAndTime := nameCache at:aHostAddress ifAbsent:nil.

        nameAndTime notNil ifTrue:[

            name := nameAndTime key.

            time := nameAndTime value.

            (Timestamp now secondDeltaFrom:time) > self cacheInvalidationTimeInterval ifFalse:[

                ^ name.

            ]

        ].

    ].

    ^ nil

    "Modified: / 26-04-2007 / 11:25:11 / cg"

    "Modified (comment): / 19-09-2018 / 12:29:30 / Claus Gittinger"

!

nameCacheAt:aHostAddress put:aName

    (nameCache isNil or:[nameCache size > self nameCacheSize]) ifTrue:[

        nameCache := Dictionary new.

    ].

    nameCache at:aHostAddress put:(aName -> Timestamp now).

    "Modified: / 26-04-2007 / 11:23:35 / cg"

!

nameCacheSize

    ^ NameCacheSize ? 50

    "Created: / 26-04-2007 / 11:22:44 / cg"

!

nameCacheSize:aNumber

    NameCacheSize := aNumber

    "Created: / 26-04-2007 / 11:23:49 / cg"

! !

!IPSocketAddress class methodsFor:'conversion'!

hostAddressFromString:aString

    "convert an address given in a dot notation like 123.456.78.9"

    |components bytes|

    components := aString asCollectionOfSubstringsSeparatedBy:$..

    components size ~~ 4 ifTrue:[

        ^ NameLookupError raiseRequestWith:aString errorString:' - bad address string'.

    ].

    bytes := components collect:[:eachComponent| 

        Integer readFromString:eachComponent onError:[^ NameLookupError raiseRequestWith:aString errorString:' - bad address string'].

    ].

    ^ bytes asByteArray.

    "

     IPSocketAddress hostAddressFromString:'1.2.3.4'

     IPSocketAddress hostAddressFromString:'255.255.255.255'

    "

    "Modified (comment): / 19-09-2018 / 12:29:58 / Claus Gittinger"

! !

!IPSocketAddress class methodsFor:'queries'!

domain

    ^ #'AF_INET'

!

hostAddressLen

    "answer the number of bytes of the host address"

    ^ 4

!

obsoleteDomainSymbol

    ^ #inet

!

vwDomainSymbol

    ^ #afInet

! !

!IPSocketAddress methodsFor:'accessing'!

hostAddress

    ^ (ByteArray new:4) replaceFrom:1 to:4 with:self startingAt:5

!

hostAddress:aByteArray

    ^ self replaceFrom:5 to:8 with:aByteArray startingAt:1

!

port

    ^ self unsignedInt16At:3 MSB:true

!

port:aPortNr

    self unsignedInt16At:3 put:aPortNr MSB:true

! !

!IPSocketAddress methodsFor:'comparing'!

sameHostAddress:aSocketAddress

    "answer true, if myself and aSocketAddress have the same host address

     (but possibly different ports)."

    ^ aSocketAddress class == self class 

        and:[self sameContentsFrom:5 to:9 as:aSocketAddress startingAt:5]

! !

!IPSocketAddress methodsFor:'converting'!

asIPv4SocketAddress

    ^ self

!

asIPv6SocketAddress

    "convert a IPv4 address to a mapped IPv6SocketAddress"

    |ipv6|

    ipv6 := IPv6SocketAddress new.

    ipv6

        port:self port;

        at:19 put:16rff;

        at:20 put:16rff.

    ipv6 replaceFrom:21 to:24 with:self startingAt:5.

    ^ ipv6.

    "

        (self localHost port:80) asIPv6SocketAddress

        (self localHost port:80) asIPv6SocketAddress asIPv4SocketAddress

    "

! !

!IPSocketAddress methodsFor:'obsolete'!

address

    <resource: #obsolete>

    ^ self hostAddress

! !

!IPSocketAddress methodsFor:'printing & storing'!

printAddressOn:aStream

    |i1 i2|

    i1 := self adrBytesStart.

    i2 := i1 + self numAdrBytes - 1.

    i1 to:i2 do:[:i | 

        i ~~ i1 ifTrue:[

            aStream nextPut:$.

        ].

        (self at:i) printOn:aStream

    ].

    "

     String streamContents:[:s | (IPSocketAddress hostName:'localhost') printAddressOn:s]