/tools/Ruby/lib/ruby/1.8/webrick/utils.rb

  1#
  2# utils.rb -- Miscellaneous utilities
  3#
  4# Author: IPR -- Internet Programming with Ruby -- writers
  5# Copyright (c) 2001 TAKAHASHI Masayoshi, GOTOU Yuuzou
  6# Copyright (c) 2002 Internet Programming with Ruby writers. All rights
  7# reserved.
  8#
  9# $IPR: utils.rb,v 1.10 2003/02/16 22:22:54 gotoyuzo Exp $
 10
 11require 'socket'
 12require 'fcntl'
 13begin
 14  require 'etc'
 15rescue LoadError
 16  nil
 17end
 18
 19module WEBrick
 20  module Utils
 21    def set_non_blocking(io)
 22      flag = File::NONBLOCK
 23      if defined?(Fcntl::F_GETFL)
 24        flag |= io.fcntl(Fcntl::F_GETFL)
 25      end
 26      io.fcntl(Fcntl::F_SETFL, flag)
 27    end
 28    module_function :set_non_blocking
 29
 30    def set_close_on_exec(io)
 31      if defined?(Fcntl::FD_CLOEXEC)
 32        io.fcntl(Fcntl::FD_CLOEXEC, 1)
 33      end
 34    end
 35    module_function :set_close_on_exec
 36
 37    def su(user)
 38      if defined?(Etc)
 39        pw = Etc.getpwnam(user)
 40        Process::initgroups(user, pw.gid)
 41        Process::Sys::setgid(pw.gid)
 42        Process::Sys::setuid(pw.uid)
 43      else
 44        warn("WEBrick::Utils::su doesn't work on this platform")
 45      end
 46    end
 47    module_function :su
 48
 49    def getservername
 50      host = Socket::gethostname
 51      begin
 52        Socket::gethostbyname(host)[0]
 53      rescue
 54        host
 55      end
 56    end
 57    module_function :getservername
 58
 59    def create_listeners(address, port, logger=nil)
 60      unless port
 61        raise ArgumentError, "must specify port"
 62      end
 63      res = Socket::getaddrinfo(address, port,
 64                                Socket::AF_UNSPEC,   # address family
 65                                Socket::SOCK_STREAM, # socket type
 66                                0,                   # protocol
 67                                Socket::AI_PASSIVE)  # flag
 68      last_error = nil
 69      sockets = []
 70      res.each{|ai|
 71        begin
 72          logger.debug("TCPServer.new(#{ai[3]}, #{port})") if logger
 73          sock = TCPServer.new(ai[3], port)
 74          port = sock.addr[1] if port == 0
 75          Utils::set_close_on_exec(sock)
 76          sockets << sock
 77        rescue => ex
 78          logger.warn("TCPServer Error: #{ex}") if logger
 79          last_error  = ex
 80        end
 81      }
 82      raise last_error if sockets.empty?
 83      return sockets
 84    end
 85    module_function :create_listeners
 86
 87    RAND_CHARS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" +
 88                 "0123456789" +
 89                 "abcdefghijklmnopqrstuvwxyz" 
 90
 91    def random_string(len)
 92      rand_max = RAND_CHARS.size
 93      ret = "" 
 94      len.times{ ret << RAND_CHARS[rand(rand_max)] }
 95      ret 
 96    end
 97    module_function :random_string
 98
 99  end
100end