Module: Socket::Foreign

Extended by:
Library
Defined in:
lib/rubysl/socket.rb

Defined Under Namespace

Classes: Addrinfo, Linger

Class Method Summary collapse

Class Method Details

.bind(descriptor, sockaddr) ⇒ Object



347
348
349
350
351
352
353
# File 'lib/rubysl/socket.rb', line 347

def self.bind(descriptor, sockaddr)
  FFI::MemoryPointer.new :char, sockaddr.bytesize do |sockaddr_p|
    sockaddr_p.write_string sockaddr, sockaddr.bytesize

    _bind descriptor, sockaddr_p, sockaddr.bytesize
  end
end

.connect(descriptor, sockaddr) ⇒ Object



355
356
357
358
359
360
361
362
363
364
# File 'lib/rubysl/socket.rb', line 355

def self.connect(descriptor, sockaddr)
  err = 0
  FFI::MemoryPointer.new :char, sockaddr.bytesize do |sockaddr_p|
    sockaddr_p.write_string sockaddr, sockaddr.bytesize

    err = _connect descriptor, sockaddr_p, sockaddr.bytesize
  end

  err
end

.getaddress(host) ⇒ Object



438
439
440
441
# File 'lib/rubysl/socket.rb', line 438

def self.getaddress(host)
  addrinfos = getaddrinfo(host)
  unpack_sockaddr_in(addrinfos.first[4], false).first
end

.getaddrinfo(host, service = nil, family = nil, socktype = nil, protocol = nil, flags = nil) ⇒ Object



380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
# File 'lib/rubysl/socket.rb', line 380

def self.getaddrinfo(host, service = nil, family = nil, socktype = nil,  protocol = nil, flags = nil)
  hints = Addrinfo.new
  hints[:ai_family] = family || 0
  hints[:ai_socktype] = socktype || 0
  hints[:ai_protocol] = protocol || 0
  hints[:ai_flags] = flags || 0

  if host && (host.empty? || host == '<any>')
    host = "0.0.0.0"
  elsif host == '<broadcast>'
    host = '255.255.255.255'
  end

  res_p = FFI::MemoryPointer.new :pointer

  err = _getaddrinfo host, service, hints.pointer, res_p

  raise SocketError, gai_strerror(err) unless err == 0

  ptr = res_p.read_pointer

  return [] unless ptr

  res = Addrinfo.new ptr

  addrinfos = []

  while true
    addrinfo = []
    addrinfo << res[:ai_flags]
    addrinfo << res[:ai_family]
    addrinfo << res[:ai_socktype]
    addrinfo << res[:ai_protocol]
    addrinfo << res[:ai_addr].read_string(res[:ai_addrlen])
    addrinfo << res[:ai_canonname]

    addrinfos << addrinfo

    break unless res[:ai_next]

    res = Addrinfo.new res[:ai_next]
  end

  return addrinfos
ensure
  hints.free if hints

  if res_p
    ptr = res_p.read_pointer

    # Be sure to feed a legit pointer to freeaddrinfo
    if ptr and !ptr.null?
      freeaddrinfo ptr
    end
    res_p.free
  end
end

.getnameinfo(sockaddr, flags = Socket::Constants::NI_NUMERICHOST | Socket::Constants::NI_NUMERICSERV, reverse_lookup = !BasicSocket.do_not_reverse_lookup)) ⇒ Object



443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
# File 'lib/rubysl/socket.rb', line 443

def self.getnameinfo(sockaddr, flags = Socket::Constants::NI_NUMERICHOST | Socket::Constants::NI_NUMERICSERV,
                     reverse_lookup = !BasicSocket.do_not_reverse_lookup)
  name_info = []
  value = nil

  FFI::MemoryPointer.new :char, sockaddr.bytesize do |sockaddr_p|
    FFI::MemoryPointer.new :char, Socket::Constants::NI_MAXHOST do |node|
      FFI::MemoryPointer.new :char, Socket::Constants::NI_MAXSERV do |service|
        sockaddr_p.write_string sockaddr, sockaddr.bytesize

        if reverse_lookup then
          err = _getnameinfo(sockaddr_p, sockaddr.bytesize,
                             node, Socket::Constants::NI_MAXHOST, nil, 0, 0)

          name_info[2] = node.read_string if err == 0
        end

        err = _getnameinfo(sockaddr_p, sockaddr.bytesize,
                           node, Socket::Constants::NI_MAXHOST,
                           service, Socket::Constants::NI_MAXSERV,
                           flags)

        unless err == 0 then
          raise SocketError, gai_strerror(err)
        end

        sa_family = SockAddr_In.new(sockaddr)[:sin_family]

        name_info[0] = Socket::Constants::AF_TO_FAMILY[sa_family]
        name_info[1] = service.read_string
        name_info[3] = node.read_string
      end
    end
  end

  name_info[2] = name_info[3] if name_info[2].nil?
  name_info
end

.getpeername(descriptor) ⇒ Object



482
483
484
485
486
487
488
489
490
491
492
493
494
# File 'lib/rubysl/socket.rb', line 482

def self.getpeername(descriptor)
  FFI::MemoryPointer.new :char, 128 do |sockaddr_storage_p|
    FFI::MemoryPointer.new :socklen_t do |len_p|
      len_p.write_int 128

      err = _getpeername descriptor, sockaddr_storage_p, len_p

      Errno.handle 'getpeername(2)' unless err == 0

      sockaddr_storage_p.read_string len_p.read_int
    end
  end
end

.getsockname(descriptor) ⇒ Object



496
497
498
499
500
501
502
503
504
505
506
507
508
# File 'lib/rubysl/socket.rb', line 496

def self.getsockname(descriptor)
  FFI::MemoryPointer.new :char, 128 do |sockaddr_storage_p|
    FFI::MemoryPointer.new :socklen_t do |len_p|
      len_p.write_int 128

      err = _getsockname descriptor, sockaddr_storage_p, len_p

      Errno.handle 'getsockname(2)' unless err == 0

      sockaddr_storage_p.read_string len_p.read_int
    end
  end
end

.getsockopt(descriptor, level, optname) ⇒ Object



366
367
368
369
370
371
372
373
374
375
376
377
378
# File 'lib/rubysl/socket.rb', line 366

def self.getsockopt(descriptor, level, optname)
  FFI::MemoryPointer.new 256 do |val| # HACK magic number
    FFI::MemoryPointer.new :socklen_t do |length|
      length.write_int 256 # HACK magic number

      err = _getsockopt descriptor, level, optname, val, length

      Errno.handle "Unable to get socket option" unless err == 0

      return val.read_string(length.read_int)
    end
  end
end

.pack_sockaddr_in(host, port, family, type, flags) ⇒ Object



510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
# File 'lib/rubysl/socket.rb', line 510

def self.pack_sockaddr_in(host, port, family, type, flags)
  hints = Addrinfo.new
  hints[:ai_family] = family
  hints[:ai_socktype] = type
  hints[:ai_flags] = flags

  if host && host.empty?
    host = "0.0.0.0"
  end

  res_p = FFI::MemoryPointer.new :pointer

  err = _getaddrinfo host, port.to_s, hints.pointer, res_p

  raise SocketError, gai_strerror(err) unless err == 0

  return [] if res_p.read_pointer.null?

  res = Addrinfo.new res_p.read_pointer

  return res[:ai_addr].read_string(res[:ai_addrlen])

ensure
  hints.free if hints

  if res_p then
    ptr = res_p.read_pointer

    freeaddrinfo ptr if ptr and not ptr.null?

    res_p.free
  end
end

.unpack_sockaddr_in(sockaddr, reverse_lookup) ⇒ Object

Raises:

  • (ArgumentError)


544
545
546
547
548
549
550
# File 'lib/rubysl/socket.rb', line 544

def self.unpack_sockaddr_in(sockaddr, reverse_lookup)
  family, port, host, ip = getnameinfo sockaddr, Socket::Constants::NI_NUMERICHOST | Socket::Constants::NI_NUMERICSERV, reverse_lookup
  # On some systems this doesn't fail for families other than AF_INET(6)
  # so we raise manually here.
  raise ArgumentError, 'not an AF_INET/AF_INET6 sockaddr' unless family =~ /AF_INET/
  return host, ip, port.to_i
end