292 lines
7.4 KiB
C
292 lines
7.4 KiB
C
/*
|
||
* tclIOSock.c --
|
||
*
|
||
* Common routines used by all socket based channel types.
|
||
*
|
||
* Copyright (c) 1995-1997 Sun Microsystems, Inc.
|
||
*
|
||
* See the file "license.terms" for information on usage and redistribution of
|
||
* this file, and for a DISCLAIMER OF ALL WARRANTIES.
|
||
*/
|
||
|
||
#include "tclInt.h"
|
||
|
||
#if defined(_WIN32)
|
||
/* On Windows, we need to do proper Unicode->UTF-8 conversion. */
|
||
|
||
typedef struct ThreadSpecificData {
|
||
int initialized;
|
||
Tcl_DString errorMsg; /* UTF-8 encoded error-message */
|
||
} ThreadSpecificData;
|
||
static Tcl_ThreadDataKey dataKey;
|
||
|
||
#undef gai_strerror
|
||
static const char *gai_strerror(int code) {
|
||
ThreadSpecificData *tsdPtr = TCL_TSD_INIT(&dataKey);
|
||
|
||
if (tsdPtr->initialized) {
|
||
Tcl_DStringFree(&tsdPtr->errorMsg);
|
||
} else {
|
||
tsdPtr->initialized = 1;
|
||
}
|
||
Tcl_WinTCharToUtf((TCHAR *)gai_strerrorW(code), -1, &tsdPtr->errorMsg);
|
||
return Tcl_DStringValue(&tsdPtr->errorMsg);
|
||
}
|
||
#endif
|
||
|
||
/*
|
||
*---------------------------------------------------------------------------
|
||
*
|
||
* TclSockGetPort --
|
||
*
|
||
* Maps from a string, which could be a service name, to a port. Used by
|
||
* socket creation code to get port numbers and resolve registered
|
||
* service names to port numbers.
|
||
*
|
||
* Results:
|
||
* A standard Tcl result. On success, the port number is returned in
|
||
* portPtr. On failure, an error message is left in the interp's result.
|
||
*
|
||
* Side effects:
|
||
* None.
|
||
*
|
||
*---------------------------------------------------------------------------
|
||
*/
|
||
|
||
int
|
||
TclSockGetPort(
|
||
Tcl_Interp *interp,
|
||
const char *string, /* Integer or service name */
|
||
const char *proto, /* "tcp" or "udp", typically */
|
||
int *portPtr) /* Return port number */
|
||
{
|
||
struct servent *sp; /* Protocol info for named services */
|
||
Tcl_DString ds;
|
||
const char *native;
|
||
|
||
if (Tcl_GetInt(NULL, string, portPtr) != TCL_OK) {
|
||
/*
|
||
* Don't bother translating 'proto' to native.
|
||
*/
|
||
|
||
native = Tcl_UtfToExternalDString(NULL, string, -1, &ds);
|
||
sp = getservbyname(native, proto); /* INTL: Native. */
|
||
Tcl_DStringFree(&ds);
|
||
if (sp != NULL) {
|
||
*portPtr = ntohs((unsigned short) sp->s_port);
|
||
return TCL_OK;
|
||
}
|
||
}
|
||
if (Tcl_GetInt(interp, string, portPtr) != TCL_OK) {
|
||
return TCL_ERROR;
|
||
}
|
||
if (*portPtr > 0xFFFF) {
|
||
Tcl_SetObjResult(interp, Tcl_NewStringObj(
|
||
"couldn't open socket: port number too high", -1));
|
||
return TCL_ERROR;
|
||
}
|
||
return TCL_OK;
|
||
}
|
||
|
||
/*
|
||
*----------------------------------------------------------------------
|
||
*
|
||
* TclSockMinimumBuffers --
|
||
*
|
||
* Ensure minimum buffer sizes (non zero).
|
||
*
|
||
* Results:
|
||
* A standard Tcl result.
|
||
*
|
||
* Side effects:
|
||
* Sets SO_SNDBUF and SO_RCVBUF sizes.
|
||
*
|
||
*----------------------------------------------------------------------
|
||
*/
|
||
|
||
#if !defined(_WIN32) && !defined(__CYGWIN__)
|
||
# define SOCKET int
|
||
#endif
|
||
|
||
int
|
||
TclSockMinimumBuffers(
|
||
void *sock, /* Socket file descriptor */
|
||
int size) /* Minimum buffer size */
|
||
{
|
||
int current;
|
||
socklen_t len;
|
||
|
||
len = sizeof(int);
|
||
getsockopt((SOCKET)(size_t) sock, SOL_SOCKET, SO_SNDBUF,
|
||
(char *) ¤t, &len);
|
||
if (current < size) {
|
||
len = sizeof(int);
|
||
setsockopt((SOCKET)(size_t) sock, SOL_SOCKET, SO_SNDBUF,
|
||
(char *) &size, len);
|
||
}
|
||
len = sizeof(int);
|
||
getsockopt((SOCKET)(size_t) sock, SOL_SOCKET, SO_RCVBUF,
|
||
(char *) ¤t, &len);
|
||
if (current < size) {
|
||
len = sizeof(int);
|
||
setsockopt((SOCKET)(size_t) sock, SOL_SOCKET, SO_RCVBUF,
|
||
(char *) &size, len);
|
||
}
|
||
return TCL_OK;
|
||
}
|
||
|
||
/*
|
||
*----------------------------------------------------------------------
|
||
*
|
||
* TclCreateSocketAddress --
|
||
*
|
||
* This function initializes a sockaddr structure for a host and port.
|
||
*
|
||
* Results:
|
||
* 1 if the host was valid, 0 if the host could not be converted to an IP
|
||
* address.
|
||
*
|
||
* Side effects:
|
||
* Fills in the *sockaddrPtr structure.
|
||
*
|
||
*----------------------------------------------------------------------
|
||
*/
|
||
|
||
int
|
||
TclCreateSocketAddress(
|
||
Tcl_Interp *interp, /* Interpreter for querying
|
||
* the desired socket family */
|
||
struct addrinfo **addrlist, /* Socket address list */
|
||
const char *host, /* Host. NULL implies INADDR_ANY */
|
||
int port, /* Port number */
|
||
int willBind, /* Is this an address to bind() to or
|
||
* to connect() to? */
|
||
const char **errorMsgPtr) /* Place to store the error message
|
||
* detail, if available. */
|
||
{
|
||
struct addrinfo hints;
|
||
struct addrinfo *p;
|
||
struct addrinfo *v4head = NULL, *v4ptr = NULL;
|
||
struct addrinfo *v6head = NULL, *v6ptr = NULL;
|
||
char *native = NULL, portbuf[TCL_INTEGER_SPACE], *portstring;
|
||
const char *family = NULL;
|
||
Tcl_DString ds;
|
||
int result;
|
||
|
||
if (host != NULL) {
|
||
native = Tcl_UtfToExternalDString(NULL, host, -1, &ds);
|
||
}
|
||
|
||
/*
|
||
* Workaround for OSX's apparent inability to resolve "localhost", "0"
|
||
* when the loopback device is the only available network interface.
|
||
*/
|
||
if (host != NULL && port == 0) {
|
||
portstring = NULL;
|
||
} else {
|
||
TclFormatInt(portbuf, port);
|
||
portstring = portbuf;
|
||
}
|
||
|
||
(void) memset(&hints, 0, sizeof(hints));
|
||
hints.ai_family = AF_UNSPEC;
|
||
|
||
/*
|
||
* Magic variable to enforce a certain address family - to be superseded
|
||
* by a TIP that adds explicit switches to [socket]
|
||
*/
|
||
|
||
if (interp != NULL) {
|
||
family = Tcl_GetVar(interp, "::tcl::unsupported::socketAF", 0);
|
||
if (family != NULL) {
|
||
if (strcmp(family, "inet") == 0) {
|
||
hints.ai_family = AF_INET;
|
||
} else if (strcmp(family, "inet6") == 0) {
|
||
hints.ai_family = AF_INET6;
|
||
}
|
||
}
|
||
}
|
||
|
||
hints.ai_socktype = SOCK_STREAM;
|
||
|
||
#if 0
|
||
/*
|
||
* We found some problems when using AI_ADDRCONFIG, e.g. on systems that
|
||
* have no networking besides the loopback interface and want to resolve
|
||
* localhost. See [Bugs 3385024, 3382419, 3382431]. As the advantage of
|
||
* using AI_ADDRCONFIG in situations where it works, is probably low,
|
||
* we'll leave it out for now. After all, it is just an optimisation.
|
||
*
|
||
* Missing on: OpenBSD, NetBSD.
|
||
* Causes failure when used on AIX 5.1 and HP-UX
|
||
*/
|
||
|
||
#if defined(AI_ADDRCONFIG) && !defined(_AIX) && !defined(__hpux)
|
||
hints.ai_flags |= AI_ADDRCONFIG;
|
||
#endif /* AI_ADDRCONFIG && !_AIX && !__hpux */
|
||
#endif /* 0 */
|
||
|
||
if (willBind) {
|
||
hints.ai_flags |= AI_PASSIVE;
|
||
}
|
||
|
||
result = getaddrinfo(native, portstring, &hints, addrlist);
|
||
|
||
if (host != NULL) {
|
||
Tcl_DStringFree(&ds);
|
||
}
|
||
|
||
if (result != 0) {
|
||
*errorMsgPtr =
|
||
#ifdef EAI_SYSTEM /* Doesn't exist on Windows */
|
||
(result == EAI_SYSTEM) ? Tcl_PosixError(interp) :
|
||
#endif /* EAI_SYSTEM */
|
||
gai_strerror(result);
|
||
return 0;
|
||
}
|
||
|
||
/*
|
||
* Put IPv4 addresses before IPv6 addresses to maximize backwards
|
||
* compatibility of [fconfigure -sockname] output.
|
||
*
|
||
* There might be more elegant/efficient ways to do this.
|
||
*/
|
||
if (willBind) {
|
||
for (p = *addrlist; p != NULL; p = p->ai_next) {
|
||
if (p->ai_family == AF_INET) {
|
||
if (v4head == NULL) {
|
||
v4head = p;
|
||
} else {
|
||
v4ptr->ai_next = p;
|
||
}
|
||
v4ptr = p;
|
||
} else {
|
||
if (v6head == NULL) {
|
||
v6head = p;
|
||
} else {
|
||
v6ptr->ai_next = p;
|
||
}
|
||
v6ptr = p;
|
||
}
|
||
}
|
||
*addrlist = NULL;
|
||
if (v6head != NULL) {
|
||
*addrlist = v6head;
|
||
v6ptr->ai_next = NULL;
|
||
}
|
||
if (v4head != NULL) {
|
||
v4ptr->ai_next = *addrlist;
|
||
*addrlist = v4head;
|
||
}
|
||
}
|
||
return 1;
|
||
}
|
||
|
||
/*
|
||
* Local Variables:
|
||
* mode: c
|
||
* c-basic-offset: 4
|
||
* fill-column: 78
|
||
* End:
|
||
*/
|