Module: Socket::Foreign

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

Defined Under Namespace

Classes: Addrinfo

Class Method Summary collapse

Class Method Details

.bind(descriptor, sockaddr) ⇒ Object



234
235
236
237
238
239
240
# File 'lib/rubysl/socket/socket.rb', line 234

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

    _bind descriptor, sockaddr_p, sockaddr.length
  end
end

.connect(descriptor, sockaddr) ⇒ Object



242
243
244
245
246
247
248
249
250
251
# File 'lib/rubysl/socket/socket.rb', line 242

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

    err = _connect descriptor, sockaddr_p, sockaddr.length
  end

  err
end

.getaddress(host) ⇒ Object



325
326
327
328
# File 'lib/rubysl/socket/socket.rb', line 325

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



267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
# File 'lib/rubysl/socket/socket.rb', line 267

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



330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
# File 'lib/rubysl/socket/socket.rb', line 330

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

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

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

        err = _getnameinfo(sockaddr_p, sockaddr.length,
                           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



369
370
371
372
373
374
375
376
377
378
379
380
381
# File 'lib/rubysl/socket/socket.rb', line 369

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



383
384
385
386
387
388
389
390
391
392
393
394
395
# File 'lib/rubysl/socket/socket.rb', line 383

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



253
254
255
256
257
258
259
260
261
262
263
264
265
# File 'lib/rubysl/socket/socket.rb', line 253

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



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
# File 'lib/rubysl/socket/socket.rb', line 397

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)


431
432
433
434
435
436
437
# File 'lib/rubysl/socket/socket.rb', line 431

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