ext/Socket/Socket.xs

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