"

 COPYRIGHT (c) 1992 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' }"

NonPositionableExternalStream subclass:#Socket

	classVariableNames:''

	poolDictionaries:''

	category:'Streams-External'

!

!Socket primitiveDefinitions!

%{

/* #define DGRAM_DEBUG /* */

#include "stxOSDefs.h"

#ifdef WIN32

# undef __BEGIN_INTERRUPTABLE__

# undef __END_INTERRUPTABLE__

# define __BEGIN_INTERRUPTABLE__ ..

# define __END_INTERRUPTABLE__ ..

# define WRAP_STDIO

# define DO_WRAP_CALLS

# ifndef DO_WRAP_CALLS

#  define STX_C_CALL0(__nm__, __f__)                                                    __f__()

#  define STX_C_CALL1(__nm__, __f__, __a1__)                                            __f__((__a1__))

#  define STX_C_CALL2(__nm__, __f__, __a1__, __a2__)                                    __f__((__a1__), (__a2__))

#  define STX_C_CALL3(__nm__, __f__, __a1__, __a2__, __a3__)                            __f__((__a1__), (__a2__), (__a3__))

#  define STX_API_CALL0(__nm__, __f__)                                                  __f__((__a1__))

#  define STX_API_CALL1(__nm__, __f__, __a1__)                                          __f__((__a1__))

#  define STX_API_CALL2(__nm__, __f__, __a1__, __a2__)                                  __f__((__a1__), (__a2__))

#  define STX_API_CALL3(__nm__, __f__, __a1__, __a2__, __a3__)                          __f__((__a1__), (__a2__), (__a3__))

#  define STX_API_CALL4(__nm__, __f__, __a1__, __a2__, __a3__, __a4__)                  __f__((__a1__), (__a2__), (__a3__), (__a4__))

#  define STX_API_CALL5(__nm__, __f__, __a1__, __a2__, __a3__, __a4__, __a5__)          __f__((__a1__), (__a2__), (__a3__), (__a4__), (__a5__))

#  define STX_API_CALL6(__nm__, __f__, __a1__, __a2__, __a3__, __a4__, __a5__, __a6__)  __f__((__a1__), (__a2__), (__a3__), (__a4__), (__a5__), (__a6__))

# else

#  define STX_C_CALL0(__nm__, __f__)                                                    __STX_C_CALL0(__nm__,(void*)__f__)

#  define STX_C_CALL1(__nm__, __f__, __a1__)                                            __STX_C_CALL1(__nm__,(void*)__f__, (void*)(__a1__))

#  define STX_C_CALL2(__nm__, __f__, __a1__, __a2__)                                    __STX_C_CALL2(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__))

#  define STX_C_CALL3(__nm__, __f__, __a1__, __a2__, __a3__)                            __STX_C_CALL3(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__), (void*)(__a3__))

#  define STX_API_CALL0(__nm__, __f__)                                                  __STX_API_CALL0(__nm__,(void*)__f__)

#  define STX_API_CALL1(__nm__, __f__, __a1__)                                          __STX_API_CALL1(__nm__,(void*)__f__, (void*)(__a1__))

#  define STX_API_CALL2(__nm__, __f__, __a1__, __a2__)                                  __STX_API_CALL2(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__))

#  define STX_API_CALL3(__nm__, __f__, __a1__, __a2__, __a3__)                          __STX_API_CALL3(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__), (void*)(__a3__))

#  define STX_API_CALL4(__nm__, __f__, __a1__, __a2__, __a3__, __a4__)                  __STX_API_CALL4(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__), (void*)(__a3__), (void*)(__a4__))

#  define STX_API_CALL5(__nm__, __f__, __a1__, __a2__, __a3__, __a4__, __a5__)          __STX_API_CALL5(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__), (void*)(__a3__), (void*)(__a4__), (void*)(__a5__))

#  define STX_API_CALL6(__nm__, __f__, __a1__, __a2__, __a3__, __a4__, __a5__, __a6__)  __STX_API_CALL6(__nm__,(void*)__f__, (void*)(__a1__), (void*)(__a2__), (void*)(__a3__), (void*)(__a4__), (void*)(__a5__), (void*)(__a6__))

# endif

# ifdef DO_WRAP_CALLS

#  define GETHOSTBYNAME(hp, name)                    \

      {                                              \

	char __hostNm[512];                          \

						     \

	strncpy(__hostNm, name, sizeof(__hostNm)-1); \

	__hostNm[sizeof(__hostNm)-1] = '\0';         \

	do {                                         \

	  __threadErrno = 0;                         \

	  hp = STX_API_CALL1("gethostbyname", gethostbyname, __hostNm); \

	} while ((hp < 0) && (__threadErrno == EINTR));      \

      }

#  define GETHOSTBYADDR(hp, addr, alen, af) \

      do {                                 \

	__threadErrno = 0;                 \

	hp = STX_API_CALL3("gethostbyaddr", gethostbyaddr, addr, alen, af);    \

      } while ((hp < 0) && (__threadErrno == EINTR));

# else

#  define GETHOSTBYNAME(hp, name) \

	__BEGIN_INTERRUPTABLE__  \

	hp = gethostbyname((char *) name); \

	__END_INTERRUPTABLE__

#  define GETHOSTBYADDR(hp, addr, alen, af) \

	__BEGIN_INTERRUPTABLE__ \

	hp = gethostbyaddr(addr, alen, af); \

	__END_INTERRUPTABLE__

# endif /* DO_WRAP_CALLS */

#else /* not WIN32 */

# define STX_C_CALL0(__nm__, __f__)                                                    __f__()

# define STX_C_CALL1(__nm__, __f__, __a1__)                                            __f__((__a1__))

# define STX_C_CALL2(__nm__, __f__, __a1__, __a2__)                                    __f__((__a1__), (__a2__))

# define STX_C_CALL3(__nm__, __f__, __a1__, __a2__, __a3__)                            __f__((__a1__), (__a2__), (__a3__))

# define STX_API_CALL0(__nm__, __f__)                                                  __f__((__a1__))

# define STX_API_CALL1(__nm__, __f__, __a1__)                                          __f__((__a1__))

# define STX_API_CALL2(__nm__, __f__, __a1__, __a2__)                                  __f__((__a1__), (__a2__))

# define STX_API_CALL3(__nm__, __f__, __a1__, __a2__, __a3__)                          __f__((__a1__), (__a2__), (__a3__))

# define STX_API_CALL4(__nm__, __f__, __a1__, __a2__, __a3__, __a4__)                  __f__((__a1__), (__a2__), (__a3__), (__a4__))

# define STX_API_CALL5(__nm__, __f__, __a1__, __a2__, __a3__, __a4__, __a5__)          __f__((__a1__), (__a2__), (__a3__), (__a4__), (__a5__))

# define STX_API_CALL6(__nm__, __f__, __a1__, __a2__, __a3__, __a4__, __a5__, __a6__)  __f__((__a1__), (__a2__), (__a3__), (__a4__), (__a5__), (__a6__))

#endif /* WIN32 */

#include <stdio.h>

#include <errno.h>

#ifdef WANT__TCP_DOT_H

# include <netinet/tcp.h>

#endif

#ifdef WANT__NETDB_DOT_H

# include <netdb.h>

#endif

#if defined(TRY_AGAIN) || defined(HOST_NOT_FOUND)

# define USE_H_ERRNO

#endif

/*

 * on some systems errno is a macro ... check for it here

 */

#ifndef errno

 extern errno;

#endif

#ifdef USE_H_ERRNO

# ifndef h_errno

 extern h_errno;

# endif

#endif

#ifdef DEBUG

# define DBGPRINTF(x)    { if (__debugging__) printf x; }

# define DBGFPRINTF(x)   { if (__debugging__) fprintf x; }

#else

# define DBGPRINTF(x)    /* as nothing */

# define DBGFPRINTF(x)   /* as nothing */

#endif

#ifdef SRWDEBUG

# define SRWPRINTF(x)    { if (__debugging__) printf x; }

#else

# define SRWPRINTF(x)    /* as nothing */

#endif

%}

! !

!Socket primitiveVariables!

%{

static int __debugging__ = 0;

%}

! !

!Socket primitiveFunctions!

%{

static

setupBufferParameters(aDataBuffer, startIndex, p_extPtr, p_offs, p_objSize)

    OBJ aDataBuffer, startIndex;

    char **p_extPtr;

    int *p_offs;

    int *p_objSize;

{

	char *extPtr = 0;

	int sIdx = 0, objSize = 0, offs = 0;

	sIdx = 0;

	if (__isSmallInteger(startIndex)) {

	    sIdx = __intVal(startIndex) - 1;

	}

	if (__isExternalBytesLike(aDataBuffer)) {

	    OBJ sz;

	    extPtr = (char *)(__externalBytesAddress(aDataBuffer));

	    sz = __externalBytesSize(aDataBuffer);

	    if (__isSmallInteger(sz)) {

		objSize = __intVal(sz);

	    } else {

		objSize = 0; /* unknown */

	    }

	    offs = sIdx;

	} else {

	    OBJ oClass;

	    int nInstVars, nInstBytes;

	    oClass = __Class(aDataBuffer);

	    switch (__intVal(__ClassInstPtr(oClass)->c_flags) & ARRAYMASK) {

		case BYTEARRAY:

		    offs = sIdx;

		    break;

		case WORDARRAY:

		case SWORDARRAY:

		    offs = sIdx * 2;

		    break;

		case LONGARRAY:

		case SLONGARRAY:

		    offs = sIdx * 4;

		    break;

		case LONGLONGARRAY:

		case SLONGLONGARRAY:

		    offs = sIdx * 8;

# ifdef __NEED_LONGLONG_ALIGN

		    offs += 4;

# endif

		    break;

		case FLOATARRAY:

		    offs = sIdx * sizeof(float);

		    break;

		case DOUBLEARRAY:

		    offs = sIdx * sizeof(double);

# ifdef __NEED_DOUBLE_ALIGN

		    offs += 4;

# endif

		    break;

		default:

		    objSize = -1;

		    break;

	    }

	    extPtr = (char *)0;

	    nInstVars = __intVal(__ClassInstPtr(oClass)->c_ninstvars);

	    nInstBytes = OHDR_SIZE + nInstVars * sizeof(OBJ);

	    offs = offs + nInstBytes;

	    objSize = _Size(aDataBuffer) - offs;

	}

	*p_extPtr = extPtr;

	*p_objSize = objSize;

	*p_offs = offs;

}

%}

! !

!Socket class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 1992 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

"

    This class provides access to sockets for interprocess communication.

    The message protocol is preliminary, until someone tells me how

    other smalltalk's socket interfaces look like.

    Currently, only IP and UNIX domain sockets are supported.

    Code for appletalk is included, but was never tested ...

    More may be added in the future.

    (the code is prepared for things like SNA or decNet;

     however, right now, this code is empty and needs a little work.

     Implementing those is pretty straight forward, once the address

     data structures are known.)

    Due to historic reasons (I started this class, before I got hold of some

    code using ST-80 Sockets i.e. RemoteInvocation), there is some old interface

    still provided.

    This will vanish; use the #family:type: or #newTCPxxx and #newUDPxxx interfaces,

    together with the bind/listen and accept calls,

    which are meant to be compatible to ST-80's UnixSocketAccessor interface.

    ST/X does not yet use IPSocketAddress, UDSocketAddress etc; all addressing

    is done by passing appropriate string- or byteArray objects containing

    the addresses. This may change, too.

    TODO: cleanup historic leftovers, implement other than inet domain stuff.

	  (mhmh - how can I test those ?)

	  change to raise more signals on errors.

	  Help - could someone who has a linux machine connected to an appletalk

		 net please test this (and send bug fixes ...)

    [author:]

	Claus Gittinger

"

!

examples

"

    example (get help info from an nntp server):

									[exBegin]

	|sock host|

	host := OperatingSystem getEnvironment:'NNTPSERVER'.

	sock := Socket newTCPclientToHost:host port:'nntp'.

	Transcript showCR:sock nextLine.

	sock buffered:false.

	sock nextPutAll:'HELP'; cr.

	[:exit |

	    |line|

	    line := sock nextLine.

	    line = '.' ifTrue:[exit value:nil].

	    Transcript showCR:line.

	] loopWithExit.

	sock close

									[exEnd]

    example (connect to finger daemon, get users entry):

									[exBegin]

	|sock host entry|

	host := OperatingSystem getHostName.

	sock := Socket newTCPclientToHost:host port:'finger'.

	sock isNil ifTrue:[

	    self warn:'no finger daemon is running'.

	    ^ self

	].

	sock useCRLF:true.

	sock buffered:false.

	sock isNil ifTrue:[

	    Transcript showCR:'cannot connect to local finger daemon'

	] ifFalse:[

	    sock nextPutAll:(OperatingSystem getLoginName).

	    sock cr.

	    entry := sock nextLine.

	    Transcript showCR:entry.

	    sock close

	]

									[exEnd]

    example (connect to an ftp server):

									[exBegin]

	|sock host|

	host := OperatingSystem getHostName.

	sock := Socket newTCPclientToHost:host port:'ftp'.

	sock buffered:false.

	Transcript showCR:sock nextLine.

	sock nextPutAll:('USER ' , 'anonymous'); cr.

	Transcript showCR:sock nextLine.

	sock nextPutAll:('PASS ' , 'fooBar'); cr.

	Transcript showCR:sock nextLine.

	sock nextPutAll:'LIST'; cr.

	Transcript showCR:sock nextLine.

	sock close.

	'dont know enough of the ftp protocol to continue here ...'

									[exEnd]

    example (connect to an snmp server [UDP]):

									[exBegin]

	|sock port|

	sock := Socket newUDP.

	port := Socket portOfService:'snmp'.

	sock connectTo:(OperatingSystem getHostName) port:port.

	sock buffered:false.

	Transcript showCR:'got it'.

	sock close.

									[exEnd]

    example (await connection from a client and read some data):

									[exBegin]

	|connectSock sock|

	connectSock := Socket newTCPserverAtPort:9998.

	connectSock isNil ifTrue:[

	    Transcript showCR:'socket setup failed.'.

	] ifFalse:[

	    Transcript showCR:'listen ..'.

	    (connectSock listenFor:5) ifFalse:[

		Transcript showCR:'listen failed.'.

	    ] ifTrue:[

		Transcript showCR:'wait'.

		connectSock readWait.

		Transcript showCR:'accept'.

		sock := connectSock accept.

		sock isNil ifTrue:[

		    Transcript showCR:'accept failed.'.

		] ifFalse:[

		    sock buffered:false.

		    Transcript showCR:'server: got it'.

		    'can now do transfer via sock'.

		    Transcript showCR:'read'.

		    Transcript showCR:('got: ' , sock nextLine).

		    Transcript showCR:'close'.

		    sock close

		].

		connectSock close.

	    ]

	]

									[exEnd]

    example (connect to above server and send some data):

									[exBegin]

	|sock|

	sock := Socket newTCPclientToHost:(OperatingSystem getHostName) port:9998.

	sock isNil ifTrue:[

	    Transcript showCR:'nope'

	] ifFalse:[

	    sock buffered:false.

	    Transcript showCR:'client: got it'.

	    'can now do transfer via sock'.

	    Transcript showCR:'sending <hello>'.

	    sock nextPutLine:'hello'.

	    sock close

	]

									[exEnd]

    example: UNIX domain socket (await connection from a client and read some data):

	|connectSock sock|

	'/tmp/ud_socket' asFilename remove.

	connectSock := Socket newUNIXserverAt:'/tmp/ud_socket'.

	connectSock isNil ifTrue:[

	    Transcript showCR:'socket setup failed.'.

	] ifFalse:[

	    Transcript showCR:'listen ..'.

	    (connectSock listenFor:5) ifFalse:[

		Transcript showCR:'listen failed.'.

	    ] ifTrue:[

		Transcript showCR:'wait'.

		connectSock buffered:false.

		connectSock readWait.

		Transcript showCR:'accept'.

		sock := connectSock accept.

		sock isNil ifTrue:[

		    Transcript showCR:'accept failed.'.

		] ifFalse:[

		    sock buffered:false.

		    Transcript showCR:'server: got it'.

		    'can now do transfer via sock'.

		    Transcript showCR:'read'.

		    Transcript showCR:('got: ' , sock nextLine).

		    Transcript showCR:'close'.

		    sock close

		].

		connectSock close.

	    ]

	]

    example (connect to above server and send some data;

	     Notice, this fails, if above server code is executed in the same ST/X image

		     (at least on LINUX), since the OS does not correctly handle

		     a connect from within an interrupted accept system call

		     On SGI's SVR4, this works ok

									[exBegin]

	|sock|

	sock := Socket newUNIXclientTo:'/tmp/ud_socket'.

	sock isNil ifTrue:[

	    Transcript showCR:'nope'

	] ifFalse:[

	    sock buffered:false.

	    Transcript showCR:'client: got it'.

	    'can now do transfer via sock'.

	    Transcript showCR:'sending <hello>'.

	    sock nextPutLine:'hello'.

	    sock close

	]

									[exEnd]

    example (UDP await packet from a client and read some data):

									[exBegin]

	|udpSock sock addr n dataBuffer|

	udpSock := Socket newUDPserverAtPort:9999.

	udpSock isNil ifTrue:[