"

 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

	instanceVariableNames:'domain socketType protocol port serviceName peer peerName

		listening'

	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 SET_LINGER_WHEN_CREATING_SOCKET

/*

 * mhmh - WIN32-fclose  has a problem in trying to close() the socket

 * instead of closesocket()'ing it

 */

# ifdef WIN32

#  define CLOSESOCKET_AFTER_FCLOSE

#  define xxCLOSESOCKET_BEFORE_FCLOSE

# endif

# define SOCKET_FROM_FD(fd)               (_get_osfhandle(fd))

#else /* not WIN32 */

# define SOCKET_FROM_FD(fd)               (fd)

#endif /* WIN32 */

#define SOCKET_FROM_FILE(f)             (SOCKET_FROM_FD(fileno(f)))

#define SOCKET_FROM_FILE_OBJECT(f)      (SOCKET_FROM_FILE(__FILEVal(f)))

#ifndef WIN32

# define closesocket(sock)    close(sock)

# define SOCKET               int

#endif

#include <stdio.h>

#include <errno.h>

#ifdef WIN32

#else

# include <netinet/ip.h>

#endif

#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__) console_printf x; }

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

#else

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

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

#endif

#ifndef TRUE

# define TRUE   1

#endif

#ifndef FALSE

# define FALSE  0

#endif

#ifndef WIN32

typedef int BOOL;

#endif

#ifdef WIN32

# undef stdout

# undef stderr

# define stdout __win32_stdout()

# define stderr __win32_stderr()

#endif

%}

! !

!Socket primitiveVariables!

%{

static int __debugging__ = 0;

%}

! !

!Socket primitiveFunctions!

%{

static int

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;

	    extPtr = (char *)0;

	    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:

		    *p_objSize = -1;

		    return 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;

	return 1;

}

%}

! !

!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 really fully

    tested and 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.

    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'.

	sock isNil ifTrue:[

	    self warn:'no nntp daemon is running'.

	    ^ self

	].

	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:[

	    Transcript showCR:'socket setup failed.'.

	] ifFalse:[

	    Transcript showCR:'wait'.

	    udpSock readWait.

	    addr := IPSocketAddress new.

	    dataBuffer := ByteArray new:1000.

	    n := udpSock receiveFrom:addr buffer:dataBuffer start:1 for:dataBuffer size.

	    n > 0 ifTrue:[

		Transcript showCR:('got: ' , n printString , 'bytes  from ' , addr printString).

		Transcript showCR:('data: ' , (dataBuffer copyTo:n) printString).

	    ] ifFalse:[

		Transcript showCR:'read failed'.

	    ].

	    Transcript showCR:'close'.

	    udpSock close

	]

									[exEnd]

    example (connect to above UDP server and send some data;

									[exBegin]

	|sock|

	sock := Socket newUDP.

	sock isNil ifTrue:[