sidekick-client 0.2.1 → 0.2.2

data/lib/ext/eventmachine-0.12.10/lib/em/protocols/postgres3.rb

@@ -1,247 +0,0 @@
-#--
-#
-# Author:: Francis Cianfrocca (gmail: blackhedd)
-# Homepage::  http://rubyeventmachine.com
-# Date:: 15 November 2006
-# 
-# See EventMachine and EventMachine::Connection for documentation and
-# usage examples.
-#
-#----------------------------------------------------------------------------
-#
-# Copyright (C) 2006-08 by Francis Cianfrocca. All Rights Reserved.
-# Gmail: blackhedd
-# 
-# This program is free software; you can redistribute it and/or modify
-# it under the terms of either: 1) the GNU General Public License
-# as published by the Free Software Foundation; either version 2 of the
-# License, or (at your option) any later version; or 2) Ruby's License.
-# 
-# See the file COPYING for complete licensing information.
-#
-#---------------------------------------------------------------------------
-#
-# 
-# 
-
-require 'readbytes'
-require 'postgres-pr/message'
-require 'postgres-pr/connection'
-require 'stringio'
-
-class StringIO # :nodoc:
-  # Reads exactly +n+ bytes.
-  #
-  # If the data read is nil an EOFError is raised.
-  #
-  # If the data read is too short a TruncatedDataError is raised and the read
-  # data is obtainable via its #data method.
-  def readbytes(n)
-    str = read(n)
-    if str == nil
-      raise EOFError, "End of file reached"
-    end
-    if str.size < n
-      raise TruncatedDataError.new("data truncated", str) 
-    end
-    str
-  end
-  alias read_exactly_n_bytes readbytes
-end
-
-
-module EventMachine
-  module Protocols
-    # PROVISIONAL IMPLEMENTATION of an evented Postgres client.
-    # This implements version 3 of the Postgres wire protocol, which will work
-    # with any Postgres version from roughly 7.4 onward.
-    #
-    # Objective: we want to access Postgres databases without requiring threads.
-    # Until now this has been a problem because the Postgres client implementations
-    # have all made use of blocking I/O calls, which is incompatible with a
-    # thread-free evented model.
-    #
-    # But rather than re-implement the Postgres Wire3 protocol, we're taking advantage
-    # of the existing postgres-pr library, which was originally written by Michael
-    # Neumann but (at this writing) appears to be no longer maintained. Still, it's
-    # in basically a production-ready state, and the wire protocol isn't that complicated
-    # anyway.
-    #
-    # We need to monkeypatch StringIO because it lacks the #readbytes method needed
-    # by postgres-pr.
-    #
-    # We're tucking in a bunch of require statements that may not be present in garden-variety
-    # EM installations. Until we find a good way to only require these if a program
-    # requires postgres, this file will need to be required explicitly.
-    #
-    # The StringIO monkeypatch is lifted verbatim from the standard library readbytes.rb,
-    # which adds method #readbytes directly to class IO. But StringIO is not a subclass of IO.
-    #
-    # We cloned the handling of postgres messages from lib/postgres-pr/connection.rb
-    # in the postgres-pr library, and modified it for event-handling.
-    #
-    # TODO: The password handling in dispatch_conn_message is totally incomplete.
-    #
-    #
-    # We return Deferrables from the user-level operations surfaced by this interface.
-    # Experimentally, we're using the pattern of always returning a boolean value as the
-    # first argument of a deferrable callback to indicate success or failure. This is
-    # instead of the traditional pattern of calling Deferrable#succeed or #fail, and
-    # requiring the user to define both a callback and an errback function.
-    #
-    # === Usage
-    #  EM.run {
-    #    db = EM.connect_unix_domain( "/tmp/.s.PGSQL.5432", EM::P::Postgres3 )
-    #    db.connect( dbname, username, psw ).callback do |status|
-    #      if status
-    #        db.query( "select * from some_table" ).callback do |status, result, errors|
-    #          if status
-    #            result.rows.each do |row|
-    #              p row
-    #            end
-    #          end
-    #        end
-    #      end
-    #    end
-    #  }
-    class Postgres3 < EventMachine::Connection
-      include PostgresPR
-
-      def initialize
-        @data = ""
-        @params = {}
-      end
-
-      def connect db, user, psw=nil
-        d = EM::DefaultDeferrable.new
-        d.timeout 15
-
-        if @pending_query || @pending_conn
-          d.succeed false, "Operation already in progress"
-        else
-          @pending_conn = d
-          prms = {"user"=>user, "database"=>db}
-          @user = user
-          if psw
-            @password = psw
-            #prms["password"] = psw
-          end
-          send_data PostgresPR::StartupMessage.new( 3 << 16, prms ).dump
-        end
-
-        d
-      end
-
-      def query sql
-        d = EM::DefaultDeferrable.new
-        d.timeout 15
-
-        if @pending_query || @pending_conn
-          d.succeed false, "Operation already in progress"
-        else
-          @r = PostgresPR::Connection::Result.new
-          @e = []
-          @pending_query = d
-          send_data PostgresPR::Query.dump(sql)
-        end
-
-        d
-      end
-
-
-      def receive_data data
-        @data << data
-        while @data.length >= 5
-          pktlen = @data[1...5].unpack("N").first
-          if @data.length >= (1 + pktlen)
-            pkt = @data.slice!(0...(1+pktlen))
-            m = StringIO.open( pkt, "r" ) {|io| PostgresPR::Message.read( io ) }
-            if @pending_conn
-              dispatch_conn_message m
-            elsif @pending_query
-              dispatch_query_message m
-            else
-              raise "Unexpected message from database"
-            end
-          else
-            break # very important, break out of the while
-          end
-        end
-      end
-
-
-      def unbind
-        if o = (@pending_query || @pending_conn)
-          o.succeed false, "lost connection"
-        end
-      end
-
-      # Cloned and modified from the postgres-pr.
-      def dispatch_conn_message msg
-        case msg
-        when AuthentificationClearTextPassword
-          raise ArgumentError, "no password specified" if @password.nil?
-          send_data PasswordMessage.new(@password).dump
-
-        when AuthentificationCryptPassword
-          raise ArgumentError, "no password specified" if @password.nil?
-          send_data PasswordMessage.new(@password.crypt(msg.salt)).dump
-
-        when AuthentificationMD5Password
-          raise ArgumentError, "no password specified" if @password.nil?
-          require 'digest/md5'
-
-          m = Digest::MD5.hexdigest(@password + @user)
-          m = Digest::MD5.hexdigest(m + msg.salt)
-          m = 'md5' + m
-          send_data PasswordMessage.new(m).dump
-
-        when AuthentificationKerberosV4, AuthentificationKerberosV5, AuthentificationSCMCredential
-          raise "unsupported authentification"
-
-        when AuthentificationOk
-        when ErrorResponse
-          raise msg.field_values.join("\t")
-        when NoticeResponse
-          @notice_processor.call(msg) if @notice_processor
-        when ParameterStatus
-          @params[msg.key] = msg.value
-        when BackendKeyData
-          # TODO
-          #p msg
-        when ReadyForQuery
-          # TODO: use transaction status
-          pc,@pending_conn = @pending_conn,nil
-          pc.succeed true
-        else
-          raise "unhandled message type"
-        end
-      end
-
-      # Cloned and modified from the postgres-pr.
-      def dispatch_query_message msg
-        case msg
-        when DataRow
-          @r.rows << msg.columns
-        when CommandComplete
-          @r.cmd_tag = msg.cmd_tag
-        when ReadyForQuery
-          pq,@pending_query = @pending_query,nil
-          pq.succeed true, @r, @e
-        when RowDescription
-          @r.fields = msg.fields
-        when CopyInResponse
-        when CopyOutResponse
-        when EmptyQueryResponse
-        when ErrorResponse
-          # TODO
-          @e << msg
-        when NoticeResponse
-          @notice_processor.call(msg) if @notice_processor
-        else
-          # TODO
-        end
-      end
-    end
-  end
-end

data/lib/ext/eventmachine-0.12.10/lib/em/protocols/saslauth.rb

@@ -1,175 +0,0 @@
-#--
-#
-# Author:: Francis Cianfrocca (gmail: blackhedd)
-# Homepage::  http://rubyeventmachine.com
-# Date:: 15 November 2006
-# 
-# See EventMachine and EventMachine::Connection for documentation and
-# usage examples.
-#
-#----------------------------------------------------------------------------
-#
-# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
-# Gmail: blackhedd
-# 
-# This program is free software; you can redistribute it and/or modify
-# it under the terms of either: 1) the GNU General Public License
-# as published by the Free Software Foundation; either version 2 of the
-# License, or (at your option) any later version; or 2) Ruby's License.
-# 
-# See the file COPYING for complete licensing information.
-#
-#---------------------------------------------------------------------------
-#
-# 
-# 
-
-module EventMachine
-  module Protocols
-
-    # Implements SASL authd.
-    # This is a very, very simple protocol that mimics the one used
-    # by saslauthd and pwcheck, two outboard daemons included in the
-    # standard SASL library distro.
-    # The only thing this is really suitable for is SASL PLAIN
-    # (user+password) authentication, but the SASL libs that are
-    # linked into standard servers (like imapd and sendmail) implement
-    # the other ones.
-    #
-    # SASL-auth is intended for reasonably fast operation inside a
-    # single machine, so it has no transport-security (although there
-    # have been multi-machine extensions incorporating transport-layer
-    # encryption).
-    #
-    # The standard saslauthd module generally runs privileged and does
-    # its work by referring to the system-account files.
-    #
-    # This feature was added to EventMachine to enable the development
-    # of custom authentication/authorization engines for standard servers.
-    #
-    # To use SASLauth, include it in a class that subclasses EM::Connection,
-    # and reimplement the validate method.
-    #
-    # The typical way to incorporate this module into an authentication
-    # daemon would be to set it as the handler for a UNIX-domain socket.
-    # The code might look like this:
-    #
-    #  EM.start_unix_domain_server( "/var/run/saslauthd/mux", MyHandler )
-    #  File.chmod( 0777, "/var/run/saslauthd/mux")
-    #
-    # The chmod is probably needed to ensure that unprivileged clients can
-    # access the UNIX-domain socket.
-    #
-    # It's also a very good idea to drop superuser privileges (if any), after
-    # the UNIX-domain socket has been opened.
-    #--
-    # Implementation details: assume the client can send us pipelined requests,
-    # and that the client will close the connection.
-    #
-    # The client sends us four values, each encoded as a two-byte length field in
-    # network order followed by the specified number of octets.
-    # The fields specify the username, password, service name (such as imap),
-    # and the "realm" name. We send back the barest minimum reply, a single
-    # field also encoded as a two-octet length in network order, followed by
-    # either "NO" or "OK" - simplicity itself.
-    #
-    # We enforce a maximum field size just as a sanity check.
-    # We do NOT automatically time out the connection.
-    #
-    # The code we use to parse out the values is ugly and probably slow.
-    # Improvements welcome.
-    #
-    module SASLauth
-
-      MaxFieldSize = 128*1024
-      def post_init
-        super
-        @sasl_data = ""
-        @sasl_values = []
-      end
-
-      def receive_data data
-        @sasl_data << data
-        while @sasl_data.length >= 2
-          len = (@sasl_data[0,2].unpack("n")).first
-          raise "SASL Max Field Length exceeded" if len > MaxFieldSize
-          if @sasl_data.length >= (len + 2)
-            @sasl_values << @sasl_data[2,len]
-            @sasl_data.slice!(0...(2+len))
-            if @sasl_values.length == 4
-              send_data( validate(*@sasl_values) ? "\0\002OK" : "\0\002NO" )
-              @sasl_values.clear
-            end
-          else
-            break
-          end
-        end
-      end
-
-      def validate username, psw, sysname, realm
-        p username
-        p psw
-        p sysname
-        p realm
-        true
-      end
-    end
-
-    # Implements the SASL authd client protocol.
-    # This is a very, very simple protocol that mimics the one used
-    # by saslauthd and pwcheck, two outboard daemons included in the
-    # standard SASL library distro.
-    # The only thing this is really suitable for is SASL PLAIN
-    # (user+password) authentication, but the SASL libs that are
-    # linked into standard servers (like imapd and sendmail) implement
-    # the other ones.
-    #
-    # You can use this module directly as a handler for EM Connections,
-    # or include it in a module or handler class of your own.
-    #
-    # First connect to a SASL server (it's probably a TCP server, or more
-    # likely a Unix-domain socket). Then call the #validate? method,
-    # passing at least a username and a password. #validate? returns
-    # a Deferrable which will either succeed or fail, depending
-    # on the status of the authentication operation.
-    #
-    module SASLauthclient
-      MaxFieldSize = 128*1024
-
-      def validate? username, psw, sysname=nil, realm=nil
-
-        str = [username, psw, sysname, realm].map {|m|
-          [(m || "").length, (m || "")]
-        }.flatten.pack( "nA*" * 4 )
-        send_data str
-
-        d = EM::DefaultDeferrable.new
-        @queries.unshift d
-        d
-      end
-
-      def post_init
-        @sasl_data = ""
-        @queries = []
-      end
-
-      def receive_data data
-        @sasl_data << data
-
-        while @sasl_data.length > 2
-          len = (@sasl_data[0,2].unpack("n")).first
-          raise "SASL Max Field Length exceeded" if len > MaxFieldSize
-          if @sasl_data.length >= (len + 2)
-            val = @sasl_data[2,len]
-            @sasl_data.slice!(0...(2+len))
-            q = @queries.pop
-            (val == "NO") ? q.fail : q.succeed
-          else
-            break
-          end
-        end
-      end
-    end
-
-  end
-end