ext/Socket/Socket.xs

   1 #define PERL_NO_GET_CONTEXT
   2 #include "EXTERN.h"
   3 #include "perl.h"
   4 #include "XSUB.h"
   5
   6 #include <stddef.h>
   7
   8 #ifndef VMS
   9 # ifdef I_SYS_TYPES
  10 #  include <sys/types.h>
  11 # endif
  12 # include <sys/socket.h>
  13 # if defined(USE_SOCKS) && defined(I_SOCKS)
  14 #   include <socks.h>
  15 # endif
  16 # ifdef MPE
  17 #  define PF_INET AF_INET
  18 #  define PF_UNIX AF_UNIX
  19 #  define SOCK_RAW 3
  20 # endif
  21 # ifdef I_SYS_UN
  22 #  include <sys/un.h>
  23 # endif
  24 /* XXX Configure test for <netinet/in_systm.h needed XXX */
  25 # if defined(NeXT) || defined(__NeXT__)
  26 #  include <netinet/in_systm.h>
  27 # endif
  28 # if defined(__sgi) && !defined(AF_LINK) && defined(PF_LINK) && PF_LINK == AF_LNK
  29 #  undef PF_LINK
  30 # endif
  31 # ifdef I_NETINET_IN
  32 #  include <netinet/in.h>
  33 # endif
  34 # ifdef I_NETDB
  35 #  include <netdb.h>
  36 # endif
  37 # ifdef I_ARPA_INET
  38 #  include <arpa/inet.h>
  39 # endif
  40 # ifdef I_NETINET_TCP
  41 #  include <netinet/tcp.h>
  42 # endif
  43 #else
  44 # include "sockadapt.h"
  45 #endif
  46
  47 #ifdef NETWARE
  48 NETDB_DEFINE_CONTEXT
  49 NETINET_DEFINE_CONTEXT
  50 #endif
  51
  52 #ifdef I_SYSUIO
  53 # include <sys/uio.h>
  54 #endif
  55
  56 #ifndef AF_NBS
  57 # undef PF_NBS
  58 #endif
  59
  60 #ifndef AF_X25
  61 # undef PF_X25
  62 #endif
  63
  64 #ifndef INADDR_NONE
  65 # define INADDR_NONE    0xffffffff
  66 #endif /* INADDR_NONE */
  67 #ifndef INADDR_BROADCAST
  68 # define INADDR_BROADCAST       0xffffffff
  69 #endif /* INADDR_BROADCAST */
  70 #ifndef INADDR_LOOPBACK
  71 # define INADDR_LOOPBACK         0x7F000001
  72 #endif /* INADDR_LOOPBACK */
  73
  74 #ifndef HAS_INET_ATON
  75
  76 /*
  77  * Check whether "cp" is a valid ascii representation
  78  * of an Internet address and convert to a binary address.
  79  * Returns 1 if the address is valid, 0 if not.
  80  * This replaces inet_addr, the return value from which
  81  * cannot distinguish between failure and a local broadcast address.
  82  */
  83 static int
  84 my_inet_aton(register const char *cp, struct in_addr *addr)
  85 {
  86         dTHX;
  87         register U32 val;
  88         register int base;
  89         register char c;
  90         int nparts;
  91         const char *s;
  92         unsigned int parts[4];
  93         register unsigned int *pp = parts;
  94
  95        if (!cp || !*cp)
  96                 return 0;
  97         for (;;) {
  98                 /*
  99                  * Collect number up to ``.''.
 100                  * Values are specified as for C:
 101                  * 0x=hex, 0=octal, other=decimal.
 102                  */
 103                 val = 0; base = 10;
 104                 if (*cp == '0') {
 105                         if (*++cp == 'x' || *cp == 'X')
 106                                 base = 16, cp++;
 107                         else
 108                                 base = 8;
 109                 }
 110                 while ((c = *cp) != '\0') {
 111                         if (isDIGIT(c)) {
 112                                 val = (val * base) + (c - '0');
 113                                 cp++;
 114                                 continue;
 115                         }
 116                         if (base == 16 && (s=strchr(PL_hexdigit,c))) {
 117                                 val = (val << 4) +
 118                                         ((s - PL_hexdigit) & 15);
 119                                 cp++;
 120                                 continue;
 121                         }
 122                         break;
 123                 }
 124                 if (*cp == '.') {
 125                         /*
 126                          * Internet format:
 127                          *      a.b.c.d
 128                          *      a.b.c   (with c treated as 16-bits)
 129                          *      a.b     (with b treated as 24 bits)
 130                          */
 131                         if (pp >= parts + 3 || val > 0xff)
 132                                 return 0;
 133                         *pp++ = val, cp++;
 134                 } else
 135                         break;
 136         }
 137         /*
 138          * Check for trailing characters.
 139          */
 140         if (*cp && !isSPACE(*cp))
 141                 return 0;
 142         /*
 143          * Concoct the address according to
 144          * the number of parts specified.
 145          */
 146         nparts = pp - parts + 1;        /* force to an int for switch() */
 147         switch (nparts) {
 148
 149         case 1:                         /* a -- 32 bits */
 150                 break;
 151
 152         case 2:                         /* a.b -- 8.24 bits */
 153                 if (val > 0xffffff)
 154                         return 0;
 155                 val |= parts[0] << 24;
 156                 break;
 157
 158         case 3:                         /* a.b.c -- 8.8.16 bits */
 159                 if (val > 0xffff)
 160                         return 0;
 161                 val |= (parts[0] << 24) | (parts[1] << 16);
 162                 break;
 163
 164         case 4:                         /* a.b.c.d -- 8.8.8.8 bits */
 165                 if (val > 0xff)
 166                         return 0;
 167                 val |= (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8);
 168                 break;
 169         }
 170         addr->s_addr = htonl(val);
 171         return 1;
 172 }
 173
 174 #undef inet_aton
 175 #define inet_aton my_inet_aton
 176
 177 #endif /* ! HAS_INET_ATON */
 178
 179
 180 static int
 181 not_here(char *s)
 182 {
 183     croak("Socket::%s not implemented on this architecture", s);
 184     return -1;
 185 }
 186
 187 #define PERL_IN_ADDR_S_ADDR_SIZE 4
 188
 189 /*
 190 * Bad assumptions possible here.
 191 *
 192 * Bad Assumption 1: struct in_addr has no other fields
 193 * than the s_addr (which is the field we care about
 194 * in here, really). However, we can be fed either 4-byte
 195 * addresses (from pack("N", ...), or va.b.c.d, or ...),
 196 * or full struct in_addrs (from e.g. pack_sockaddr_in()),
 197 * which may or may not be 4 bytes in size.
 198 *
 199 * Bad Assumption 2: the s_addr field is a simple type
 200 * (such as an int, u_int32_t).  It can be a bit field,
 201 * in which case using & (address-of) on it or taking sizeof()
 202 * wouldn't go over too well.  (Those are not attempted
 203 * now but in case someone thinks to change the below code
 204 * to use addr.s_addr instead of addr, you have been warned.)
 205 *
 206 * Bad Assumption 3: the s_addr is the first field in
 207 * an in_addr, or that its bytes are the first bytes in
 208 * an in_addr.
 209 *
 210 * These bad assumptions are wrong in UNICOS which has
 211 * struct in_addr { struct { u_long  st_addr:32; } s_da };
 212 * #define s_addr s_da.st_addr
 213 * and u_long is 64 bits.
 214 *
 215 * --jhi */
 216
 217 #include "const-c.inc"
 218
 219 MODULE = Socket         PACKAGE = Socket
 220
 221 INCLUDE: const-xs.inc
 222
 223 void
 224 inet_aton(host)
 225         char *  host
 226         CODE:
 227         {
 228         struct in_addr ip_address;
 229         struct hostent * phe;
 230         int ok =
 231                 (host != NULL) &&
 232                 (*host != '\0') &&
 233                 inet_aton(host, &ip_address);
 234
 235         if (!ok && (phe = gethostbyname(host))) {
 236                 Copy( phe->h_addr, &ip_address, phe->h_length, char );
 237                 ok = 1;
 238         }
 239
 240         ST(0) = sv_newmortal();
 241         if (ok)
 242                 sv_setpvn( ST(0), (char *)&ip_address, sizeof ip_address );
 243         }
 244
 245 void
 246 inet_ntoa(ip_address_sv)
 247         SV *    ip_address_sv
 248         CODE:
 249         {
 250         STRLEN addrlen;
 251         struct in_addr addr;
 252         char * addr_str;
 253         char * ip_address;
 254         if (DO_UTF8(ip_address_sv) && !sv_utf8_downgrade(ip_address_sv, 1))
 255              croak("Wide character in Socket::inet_ntoa");
 256         ip_address = SvPVbyte(ip_address_sv, addrlen);
 257         if (addrlen == sizeof(addr) || addrlen == 4)
 258                 addr.s_addr =
 259                     (ip_address[0] & 0xFF) << 24 |
 260                     (ip_address[1] & 0xFF) << 16 |
 261                     (ip_address[2] & 0xFF) <<  8 |
 262                     (ip_address[3] & 0xFF);
 263         else
 264                 croak("Bad arg length for %s, length is %d, should be %d",
 265                       "Socket::inet_ntoa",
 266                       addrlen, sizeof(addr));
 267         /* We could use inet_ntoa() but that is broken
 268          * in HP-UX + GCC + 64bitint (returns "0.0.0.0"),
 269          * so let's use this sprintf() workaround everywhere.
 270          * This is also more threadsafe than using inet_ntoa(). */
 271         New(1138, addr_str, 4 * 3 + 3 + 1, char); /* IPv6? */
 272         sprintf(addr_str, "%d.%d.%d.%d",
 273                 ((addr.s_addr >> 24) & 0xFF),
 274                 ((addr.s_addr >> 16) & 0xFF),
 275                 ((addr.s_addr >>  8) & 0xFF),
 276                 ( addr.s_addr        & 0xFF));
 277         ST(0) = sv_2mortal(newSVpvn(addr_str, strlen(addr_str)));
 278         Safefree(addr_str);
 279         }
 280
 281 void
 282 sockaddr_family(sockaddr)
 283         SV *    sockaddr
 284         PREINIT:
 285         STRLEN sockaddr_len;
 286         char *sockaddr_pv = SvPVbyte(sockaddr, sockaddr_len);
 287         CODE:
 288         if (sockaddr_len < offsetof(struct sockaddr, sa_data)) {
 289             croak("Bad arg length for %s, length is %d, should be at least %d",
 290                   "Socket::sockaddr_family", sockaddr_len,
 291                   offsetof(struct sockaddr, sa_data));
 292         }
 293         ST(0) = sv_2mortal(newSViv(((struct sockaddr*)sockaddr_pv)->sa_family));
 294
 295 void
 296 pack_sockaddr_un(pathname)
 297         char *  pathname
 298         CODE:
 299         {
 300 #ifdef I_SYS_UN
 301         struct sockaddr_un sun_ad; /* fear using sun */
 302         STRLEN len;
 303
 304         Zero( &sun_ad, sizeof sun_ad, char );
 305         sun_ad.sun_family = AF_UNIX;
 306         len = strlen(pathname);
 307         if (len > sizeof(sun_ad.sun_path))
 308             len = sizeof(sun_ad.sun_path);
 309 #  ifdef OS2    /* Name should start with \socket\ and contain backslashes! */
 310         {
 311             int off;
 312             char *s, *e;
 313
 314             if (pathname[0] != '/' && pathname[0] != '\\')
 315                 croak("Relative UNIX domain socket name '%s' unsupported", pathname);
 316             else if (len < 8
 317                      || pathname[7] != '/' && pathname[7] != '\\'
 318                      || !strnicmp(pathname + 1, "socket", 6))
 319                 off = 7;
 320             else
 321                 off = 0;                /* Preserve names starting with \socket\ */
 322             Copy( "\\socket", sun_ad.sun_path, off, char);
 323             Copy( pathname, sun_ad.sun_path + off, len, char );
 324
 325             s = sun_ad.sun_path + off - 1;
 326             e = s + len + 1;
 327             while (++s < e)
 328                 if (*s = '/')
 329                     *s = '\\';
 330         }
 331 #  else /* !( defined OS2 ) */
 332         Copy( pathname, sun_ad.sun_path, len, char );
 333 #  endif
 334         if (0) not_here("dummy");
 335         ST(0) = sv_2mortal(newSVpvn((char *)&sun_ad, sizeof sun_ad));
 336 #else
 337         ST(0) = (SV *) not_here("pack_sockaddr_un");
 338 #endif
 339
 340         }
 341
 342 void
 343 unpack_sockaddr_un(sun_sv)
 344         SV *    sun_sv
 345         CODE:
 346         {
 347 #ifdef I_SYS_UN
 348         struct sockaddr_un addr;
 349         STRLEN sockaddrlen;
 350         char * sun_ad = SvPVbyte(sun_sv,sockaddrlen);
 351         char * e;
 352 #   ifndef __linux__
 353         /* On Linux sockaddrlen on sockets returned by accept, recvfrom,
 354            getpeername and getsockname is not equal to sizeof(addr). */
 355         if (sockaddrlen != sizeof(addr)) {
 356             croak("Bad arg length for %s, length is %d, should be %d",
 357                         "Socket::unpack_sockaddr_un",
 358                         sockaddrlen, sizeof(addr));
 359         }
 360 #   endif
 361
 362         Copy( sun_ad, &addr, sizeof addr, char );
 363
 364         if ( addr.sun_family != AF_UNIX ) {
 365             croak("Bad address family for %s, got %d, should be %d",
 366                         "Socket::unpack_sockaddr_un",
 367                         addr.sun_family,
 368                         AF_UNIX);
 369         }
 370         e = addr.sun_path;
 371         while (*e && e < addr.sun_path + sizeof addr.sun_path)
 372             ++e;
 373         ST(0) = sv_2mortal(newSVpvn(addr.sun_path, e - addr.sun_path));
 374 #else
 375         ST(0) = (SV *) not_here("unpack_sockaddr_un");
 376 #endif
 377         }
 378
 379 void
 380 pack_sockaddr_in(port, ip_address_sv)
 381         unsigned short  port
 382         SV *    ip_address_sv
 383         CODE:
 384         {
 385         struct sockaddr_in sin;
 386         struct in_addr addr;
 387         STRLEN addrlen;
 388         char * ip_address;
 389         if (DO_UTF8(ip_address_sv) && !sv_utf8_downgrade(ip_address_sv, 1))
 390              croak("Wide character in Socket::pack_sockaddr_in");
 391         ip_address = SvPVbyte(ip_address_sv, addrlen);
 392         if (addrlen == sizeof(addr) || addrlen == 4)
 393                 addr.s_addr =
 394                     (ip_address[0] & 0xFF) << 24 |
 395                     (ip_address[1] & 0xFF) << 16 |
 396                     (ip_address[2] & 0xFF) <<  8 |
 397                     (ip_address[3] & 0xFF);
 398         else
 399                 croak("Bad arg length for %s, length is %d, should be %d",
 400                       "Socket::pack_sockaddr_in",
 401                       addrlen, sizeof(addr));
 402         Zero( &sin, sizeof sin, char );
 403         sin.sin_family = AF_INET;
 404         sin.sin_port = htons(port);
 405         sin.sin_addr.s_addr = htonl(addr.s_addr);
 406         ST(0) = sv_2mortal(newSVpvn((char *)&sin, sizeof sin));
 407         }
 408
 409 void
 410 unpack_sockaddr_in(sin_sv)
 411         SV *    sin_sv
 412         PPCODE:
 413         {
 414         STRLEN sockaddrlen;
 415         struct sockaddr_in addr;
 416         unsigned short  port;
 417         struct in_addr  ip_address;
 418         char *  sin = SvPVbyte(sin_sv,sockaddrlen);
 419         if (sockaddrlen != sizeof(addr)) {
 420             croak("Bad arg length for %s, length is %d, should be %d",
 421                         "Socket::unpack_sockaddr_in",
 422                         sockaddrlen, sizeof(addr));
 423         }
 424         Copy( sin, &addr,sizeof addr, char );
 425         if ( addr.sin_family != AF_INET ) {
 426             croak("Bad address family for %s, got %d, should be %d",
 427                         "Socket::unpack_sockaddr_in",
 428                         addr.sin_family,
 429                         AF_INET);
 430         }
 431         port = ntohs(addr.sin_port);
 432         ip_address = addr.sin_addr;
 433
 434         EXTEND(SP, 2);
 435         PUSHs(sv_2mortal(newSViv((IV) port)));
 436         PUSHs(sv_2mortal(newSVpvn((char *)&ip_address, sizeof ip_address)));
 437         }