mallcop 0.0.1

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2010 Carl Lerche, Aaron Patterson
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,48 @@
+= mall_cop
+
+* FIX (url)
+
+== DESCRIPTION:
+
+FIX (describe your package)
+
+== FEATURES/PROBLEMS:
+
+* FIX (list of features or problems)
+
+== SYNOPSIS:
+
+  FIX (code sample of usage)
+
+== REQUIREMENTS:
+
+* FIX (list of requirements)
+
+== INSTALL:
+
+* FIX (sudo gem install, anything else)
+
+== LICENSE:
+
+(The MIT License)
+
+Copyright (c) 2010 FIX
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+'Software'), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/ext/mallcop/channel.c

@@ -0,0 +1,222 @@
+#include <mallcop.h>
+
+VALUE rb_cMallCopChannel;
+
+static void dealloc(MallCopChannel *m_channel)
+{
+  BLOCK(libssh2_channel_free( m_channel->libssh2_channel ));
+  mallcop_session_release( m_channel->m_session );
+  free( m_channel );
+}
+
+static VALUE allocate(VALUE klass)
+{
+  MallCopChannel *m_channel = calloc( 1, sizeof(MallCopChannel) );
+  return Data_Wrap_Struct(klass, 0, dealloc, m_channel);
+}
+
+static VALUE initialize(VALUE self)
+{
+  MallCopChannel *m_channel;
+  MallCopSession *m_session;
+  VALUE session;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  if ( m_channel->libssh2_channel ) {
+    rb_raise(rb_eRuntimeError, "the libssh2 channel is already initialized");
+    return Qnil;
+  }
+
+  session = rb_iv_get(self, "@session");
+
+  if ( NIL_P(session) ) {
+    rb_raise(rb_eRuntimeError, "@session is nil");
+    return Qfalse;
+  }
+
+  Data_Get_Struct(session, MallCopSession, m_session);
+
+  m_channel->libssh2_channel = libssh2_channel_open_session(m_session->libssh2_session);
+
+  if ( !m_channel->libssh2_channel ) {
+    if ( libssh2_session_last_errno(m_session->libssh2_session) == LIBSSH2_ERROR_EAGAIN ) {
+      return INT2FIX(LIBSSH2_ERROR_EAGAIN);
+    }
+
+    mallcop_raise_last_error(m_session, rb_eRuntimeError);
+    return Qnil;
+  }
+
+  // Initializing the libssh2 channel is successful, so we can associate
+  // the session object with the channel object.
+  m_channel->m_session = m_session;
+  mallcop_session_retain(m_session);
+
+  return Qtrue;
+}
+
+static VALUE request_pty(VALUE self, VALUE term)
+{
+  MallCopChannel *m_channel;
+  int ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_request_pty(m_channel->libssh2_channel, StringValuePtr(term));
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE shell(VALUE self)
+{
+  MallCopChannel *m_channel;
+  int ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_shell(m_channel->libssh2_channel);
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE channel_exec(VALUE self, VALUE command)
+{
+  MallCopChannel *m_channel;
+  int ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_exec(m_channel->libssh2_channel, StringValuePtr(command));
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE channel_read(VALUE self)
+{
+  MallCopChannel *m_channel;
+  char buffer[0x4000];
+  int count;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  count = libssh2_channel_read(m_channel->libssh2_channel, buffer, sizeof(buffer));
+
+  if ( count >= 0 ) {
+    return rb_str_new((const char *) &buffer, count);
+  } else if ( count == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(LIBSSH2_ERROR_EAGAIN);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE channel_write(VALUE self, VALUE string)
+{
+  MallCopChannel *m_channel;
+  ssize_t ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_write(m_channel->libssh2_channel, StringValuePtr(string), RSTRING_LEN(string));
+
+  if ( ret >= 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE channel_close(VALUE self)
+{
+  MallCopChannel *m_channel;
+  int ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_close(m_channel->libssh2_channel);
+
+  if ( ret >= 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE eof(VALUE self)
+{
+  MallCopChannel *m_channel;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  return libssh2_channel_eof(m_channel->libssh2_channel) ? Qtrue : Qfalse;
+}
+
+static VALUE send_eof(VALUE self)
+{
+  MallCopChannel *m_channel;
+  int ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_send_eof(m_channel->libssh2_channel);
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE flush(VALUE self)
+{
+  MallCopChannel *m_channel;
+  int ret;
+
+  Data_Get_Struct(self, MallCopChannel, m_channel);
+
+  ret = libssh2_channel_flush(m_channel->libssh2_channel);
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_channel->m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+void init_mallcop_channel()
+{
+  rb_cMallCopChannel = rb_define_class_under(rb_mMallCop, "Channel", rb_cObject);
+  rb_define_alloc_func(rb_cMallCopChannel, allocate);
+
+  rb_define_private_method(rb_cMallCopChannel, "native_eof?", eof, 0);
+  rb_define_private_method(rb_cMallCopChannel, "native_initialize", initialize, 0);
+  rb_define_private_method(rb_cMallCopChannel, "native_channel_close", channel_close, 0);
+  rb_define_private_method(rb_cMallCopChannel, "native_channel_read", channel_read, 0);
+  rb_define_private_method(rb_cMallCopChannel, "native_channel_write", channel_write, 1);
+  rb_define_private_method(rb_cMallCopChannel, "native_request_pty", request_pty, 1);
+  rb_define_private_method(rb_cMallCopChannel, "native_shell", shell, 0);
+  rb_define_private_method(rb_cMallCopChannel, "native_channel_exec", channel_exec, 1);
+  rb_define_private_method(rb_cMallCopChannel, "native_send_eof", send_eof, 0);
+  rb_define_private_method(rb_cMallCopChannel, "native_flush", flush, 0);
+}

data/ext/mallcop/extconf.rb

@@ -0,0 +1,16 @@
+require 'mkmf'
+
+# :stopdoc:
+
+dir_config 'libssh2', '/opt/local/include', '/opt/local/lib'
+
+def asplode missing
+  abort "#{missing} is missing. Please install libssh2."
+end
+
+asplode('libssh2.h')  unless find_header 'libssh2.h'
+asplode('libssh2') unless find_library 'ssh2', 'libssh2_session_init_ex'
+
+create_makefile 'mallcop/mallcop'
+
+# :startdoc:

data/ext/mallcop/mallcop.c

@@ -0,0 +1,33 @@
+#include <mallcop.h>
+
+VALUE rb_mMallCop;
+
+void mallcop_raise_last_error(MallCopSession *m_session, VALUE klass)
+{
+  char *errmsg;
+
+  libssh2_session_last_error(m_session->libssh2_session, &errmsg, NULL, 0);
+
+  rb_raise(klass, errmsg);
+}
+
+void Init_mallcop()
+{
+  int rc;
+
+  rb_mMallCop = rb_define_module("MallCop");
+
+  rb_define_const(rb_mMallCop, "ERROR_ALLOC",             INT2FIX(LIBSSH2_ERROR_ALLOC));
+  rb_define_const(rb_mMallCop, "ERROR_BANNER_SEND",       INT2FIX(LIBSSH2_ERROR_BANNER_SEND));
+  rb_define_const(rb_mMallCop, "ERROR_CHANNEL_FAILURE",   INT2FIX(LIBSSH2_ERROR_CHANNEL_FAILURE));
+  rb_define_const(rb_mMallCop, "ERROR_EAGAIN",            INT2FIX(LIBSSH2_ERROR_EAGAIN));
+  rb_define_const(rb_mMallCop, "ERROR_KEX_FAILURE",       INT2FIX(LIBSSH2_ERROR_KEX_FAILURE));
+  rb_define_const(rb_mMallCop, "ERROR_SOCKET_DISCONNECT", INT2FIX(LIBSSH2_ERROR_SOCKET_DISCONNECT));
+  rb_define_const(rb_mMallCop, "ERROR_SOCKET_NONE",       INT2FIX(LIBSSH2_ERROR_SOCKET_NONE));
+  rb_define_const(rb_mMallCop, "ERROR_SOCKET_SEND",       INT2FIX(LIBSSH2_ERROR_SOCKET_SEND));
+  rb_define_const(rb_mMallCop, "ERROR_PROTO",             INT2FIX(LIBSSH2_ERROR_PROTO));
+  rb_define_const(rb_mMallCop, "HASH_SHA1",               INT2FIX(LIBSSH2_HOSTKEY_HASH_SHA1));
+
+  init_mallcop_session();
+  init_mallcop_channel();
+}

data/ext/mallcop/mallcop.h

@@ -0,0 +1,31 @@
+#ifndef MALLCOP_H
+#define MALLCOP_H
+
+#include <ruby.h>
+#include <libssh2.h>
+
+#define BLOCK(stmt) while ((stmt) == LIBSSH2_ERROR_EAGAIN)
+
+extern VALUE rb_mMallCop;
+extern VALUE rb_cMallCopSession;
+extern VALUE rb_cMallCopChannel;
+
+typedef struct {
+  LIBSSH2_SESSION *libssh2_session;
+  int count;
+} MallCopSession;
+
+typedef struct {
+  LIBSSH2_CHANNEL *libssh2_channel;
+  MallCopSession *m_session;
+} MallCopChannel;
+
+void init_mallcop_channel();
+void init_mallcop_session();
+
+void mallcop_session_retain(MallCopSession *session);
+void mallcop_session_release(MallCopSession *session);
+
+void mallcop_raise_last_error(MallCopSession *m_session, VALUE klass);
+
+#endif

data/ext/mallcop/session.c

@@ -0,0 +1,228 @@
+#include <mallcop.h>
+
+VALUE rb_cMallCopSession;
+
+void mallcop_session_retain(MallCopSession *m_session)
+{
+  ++m_session->count;
+}
+
+void mallcop_session_release(MallCopSession *m_session)
+{
+  if ( --m_session->count == 0 ) {
+    BLOCK(libssh2_session_free( m_session->libssh2_session ));
+    free( m_session );
+  }
+}
+
+static void dealloc(MallCopSession *m_session)
+{
+  mallcop_session_release( m_session );
+}
+
+static VALUE allocate(VALUE klass)
+{
+  MallCopSession *m_session = calloc( 1, sizeof(MallCopSession) );
+  mallcop_session_retain(m_session);
+
+  return Data_Wrap_Struct(klass, 0, dealloc, m_session);
+}
+
+static VALUE initialize(VALUE self)
+{
+  MallCopSession *m_session;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  if ( m_session->libssh2_session ) {
+    rb_raise(rb_eRuntimeError, "the libssh2 session is already initialized");
+    return Qnil;
+  }
+
+  m_session->libssh2_session = libssh2_session_init();
+
+  if ( !m_session->libssh2_session ) {
+    rb_raise(rb_eRuntimeError, "session init failed");
+    return Qnil;
+  }
+
+  return Qtrue;
+}
+
+static VALUE set_blocking(VALUE self, VALUE blocking)
+{
+  MallCopSession *m_session;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  if ( !m_session->libssh2_session ) {
+    rb_raise(rb_eRuntimeError, "session not initialized yet");
+    return Qnil;
+  }
+
+  libssh2_session_set_blocking( m_session->libssh2_session, NUM2INT(blocking) );
+
+  return Qnil;
+}
+
+static VALUE start(VALUE self, VALUE r_fd)
+{
+  MallCopSession *m_session;
+  int ret;
+  int fd = NUM2INT(r_fd);
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  ret = libssh2_session_startup(m_session->libssh2_session, fd);
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  mallcop_raise_last_error(m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE hostkey_hash(VALUE self, VALUE hashtype)
+{
+  MallCopSession *m_session;
+  const char *key;
+  int keytype = NUM2INT(hashtype);
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  key = libssh2_hostkey_hash(m_session->libssh2_session, keytype);
+
+  if ( !key ) {
+    return Qnil;
+  }
+
+  if ( keytype == LIBSSH2_HOSTKEY_HASH_SHA1 ) {
+    return rb_str_new(key, 20);
+  }
+
+  return rb_str_new(key, 16);
+}
+
+static VALUE userauth_list(VALUE self, VALUE user)
+{
+  MallCopSession *m_session;
+  char * list;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  do {
+    list = libssh2_userauth_list(m_session->libssh2_session,
+            StringValuePtr(user), RSTRING_LEN(user));
+  } while ( !list && libssh2_session_last_errno(m_session->libssh2_session) == LIBSSH2_ERROR_EAGAIN );
+
+  return rb_str_new2(list);
+}
+
+static VALUE userauth_password(VALUE self, VALUE user, VALUE password)
+{
+  MallCopSession *m_session;
+  int ret;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  ret = libssh2_userauth_password(m_session->libssh2_session,
+            StringValuePtr(user), StringValuePtr(password));
+
+  if ( ret == 0 || ret == LIBSSH2_ERROR_EAGAIN ) {
+    return INT2FIX(ret);
+  }
+
+  ret = libssh2_session_last_errno(m_session->libssh2_session);
+
+  switch (ret) {
+    case LIBSSH2_ERROR_PASSWORD_EXPIRED:
+    case LIBSSH2_ERROR_AUTHENTICATION_FAILED:
+      return INT2FIX(ret);
+      break;
+  }
+
+  mallcop_raise_last_error(m_session, rb_eRuntimeError);
+  return Qnil;
+}
+
+static VALUE userauth_publickey_fromfile(VALUE self, VALUE user, VALUE public_key, VALUE private_key, VALUE password)
+{
+  MallCopSession *m_session;
+  int ret;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  BLOCK(ret = libssh2_userauth_publickey_fromfile(m_session->libssh2_session,
+                StringValuePtr(user), StringValuePtr(public_key),
+                StringValuePtr(private_key), StringValuePtr(password)));
+
+  if(ret) {
+    switch(ret) {
+      case LIBSSH2_ERROR_AUTHENTICATION_FAILED:
+      case LIBSSH2_ERROR_PUBLICKEY_UNVERIFIED:
+        return Qfalse;
+        break;
+      default:
+        rb_raise(rb_eRuntimeError, "authentication failed (%d)", ret);
+        break;
+    }
+  }
+
+  return Qtrue;
+}
+
+static VALUE disconnect(VALUE self, VALUE description)
+{
+  MallCopSession *m_session;
+  int ret;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  BLOCK(ret = libssh2_session_disconnect(m_session->libssh2_session,
+                StringValuePtr(description)));
+
+  return INT2NUM(ret);
+}
+
+static VALUE last_errno(VALUE self)
+{
+  MallCopSession *m_session;
+  int ret;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  ret = libssh2_session_last_errno(m_session->libssh2_session);
+
+  return INT2FIX(ret);
+}
+
+static VALUE last_errmsg(VALUE self)
+{
+  MallCopSession *m_session;
+  char *errmsg;
+  int errlen;
+
+  Data_Get_Struct(self, MallCopSession, m_session);
+
+  libssh2_session_last_error(m_session->libssh2_session, &errmsg, &errlen, 0);
+
+  return rb_str_new(errmsg, (long) errlen);
+}
+
+void init_mallcop_session()
+{
+  rb_cMallCopSession = rb_define_class_under(rb_mMallCop, "Session", rb_cObject);
+  rb_define_alloc_func(rb_cMallCopSession, allocate);
+
+  rb_define_private_method(rb_cMallCopSession, "native_initialize", initialize, 0);
+  rb_define_private_method(rb_cMallCopSession, "native_start", start, 1);
+  rb_define_private_method(rb_cMallCopSession, "native_set_blocking", set_blocking, 1);
+  rb_define_private_method(rb_cMallCopSession, "native_userauth_list", userauth_list, 1);
+  rb_define_private_method(rb_cMallCopSession, "native_last_errno", last_errno, 0);
+  rb_define_private_method(rb_cMallCopSession, "native_last_errmsg", last_errmsg, 0);
+  rb_define_private_method(rb_cMallCopSession, "native_userauth_password", userauth_password, 2);
+  rb_define_private_method(rb_cMallCopSession, "native_hostkey_hash", hostkey_hash, 1);
+  rb_define_private_method(rb_cMallCopSession, "native_userauth_publickey_fromfile", userauth_publickey_fromfile, 4);
+  rb_define_private_method(rb_cMallCopSession, "native_disconnect", disconnect, 1);
+}

data/lib/mallcop.rb

@@ -0,0 +1,48 @@
+require 'mallcop/mallcop'
+require 'mallcop/channel'
+require 'mallcop/session'
+require 'mallcop/shell'
+
+module MallCop
+  VERSION = '1.0.0'
+
+  # Raised when something happens with connecting to the server
+  class RuntimeError    < ::RuntimeError; end
+  class ConnectionError < StandardError ; end
+  class SessionError    < StandardError ; end
+  class ChannelError    < StandardError ; end
+
+  # Establishes a connection to the SSH server.
+  #
+  # Arguments
+  # =====
+  # +host+      The host to connect to
+  # +username+  The username to connect under.
+  #
+  # Optional:
+  # +options+
+  #   :port     The port to connect on (defaults to 22)
+  #   :password Use a password based authentication strategy.
+  def self.connect(host, username, options = {})
+    session = Session.start(host, options[:port] || 22)
+
+    unless session.authenticate username, options
+      raise ChannelError, "Could not authenticate user '#{username}'"
+    end
+
+    if block_given?
+      yield session
+      session.disconnect("Closing session")
+    end
+
+    session
+  end
+
+  def self.shell(host, username, options = {})
+    connect(host, username, options) do |session|
+      shell = Shell.new(session)
+      yield shell
+      shell.cleanup
+    end
+  end
+end

data/lib/mallcop/channel.rb

@@ -0,0 +1,93 @@
+module MallCop
+  class Channel
+
+    def initialize(session)
+      @session, @open = session, true
+
+      block { native_initialize }
+    end
+
+    def read
+      retval = ERROR_EAGAIN
+
+      until retval != ERROR_EAGAIN
+        @session.io_select
+        retval = read_nonblock
+      end
+
+      retval
+    end
+
+    def read_nonblock
+      raise ChannelError, "The channel has been closed" unless @open
+      native_channel_read
+    end
+
+    def write(cmd)
+      retval = ERROR_EAGAIN
+
+      until retval != ERROR_EAGAIN
+        @session.io_select(true)
+        retval = write_nonblock(cmd)
+      end
+
+      retval
+    end
+
+    def write_nonblock(cmd)
+      raise ChannelError, "The channel has been closed" unless @open
+      native_channel_write(cmd)
+    end
+
+    def eof?
+      native_eof?
+    end
+
+    def close
+      return true unless @open
+      block { native_channel_close }
+      true
+    ensure
+      @open = false
+    end
+
+    def request_pty(term)
+      block { native_request_pty(term) }
+      true
+    end
+
+    def shell
+      block { native_shell }
+      true
+    end
+
+    def channel_exec(cmd)
+      block { native_channel_exec(cmd) }
+      true
+    end
+
+    def send_eof
+      block { native_send_eof }
+      true
+    end
+
+    def flush
+      block { native_flush }
+      true
+    end
+
+  private
+
+    def block
+      retval = yield
+
+      until retval != ERROR_EAGAIN
+        @session.io_select
+        retval = yield
+      end
+
+      retval
+    end
+
+  end
+end

data/lib/mallcop/session.rb

@@ -0,0 +1,107 @@
+require 'socket'
+
+module MallCop
+  class Session
+
+    def self.start(host, port)
+      session = new(host, port)
+      session.start
+      session
+    end
+
+    def initialize(host, port)
+      native_initialize
+      native_set_blocking(0)
+      @host = host
+      @port = port
+    end
+
+    def start
+      block { native_start(socket.fileno) }
+      true
+    end
+
+    def authenticate(username, options)
+      allowed = authlist_for username
+
+      attempted = []
+
+      auth_methods.each do |method|
+        next unless allowed.include?(method)
+
+        attempted << method
+
+        if send "authenticate_via_#{method.downcase}", username, options
+          return true
+        end
+      end
+
+      false
+    end
+
+    def open_channel
+      Channel.new(self)
+    end
+
+    def hostkey_hash(hashtype)
+      native_hostkey_hash(hashtype)
+    end
+
+    def userauth_publickey_fromfile(user, public_key, private_key, password)
+      native_userauth_publickey_fromfile(user, public_key, private_key, password)
+    end
+
+    def disconnect(description)
+      native_disconnect(description)
+    end
+
+    def io_select(write = false)
+      write ?
+        select([], [socket], nil, nil) :
+        select([socket], nil, nil, nil)
+    end
+
+  private
+
+    def socket
+      @socket ||= Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0).tap do |s|
+        s.connect Socket.sockaddr_in(@port, @host)
+      end
+    rescue Errno::ECONNREFUSED => e
+      raise ConnectionError, "Cannot establish a connection to #{@host} on port #{@port}"
+    end
+
+    def auth_methods
+      %w(publickey hostbased password)
+    end
+
+    def authlist_for(username)
+      native_userauth_list(username).split ','
+    end
+
+    def authenticate_via_password(username, options)
+      return unless options[:password]
+      block { native_userauth_password(username, options[:password]) } == 0
+    end
+
+    def authenticate_via_publickey(username, options)
+      return unless options[:keys] && options[:keys].any?
+      # Not implemented yet
+    end
+    
+    def authenticate_via_hostbased(username, options)
+      # Not implemented yet
+    end
+
+    def block
+      retval = yield
+
+      until retval != ERROR_EAGAIN
+        io_select
+        retval = yield
+      end
+
+      retval
+    end
+  end
+end

data/lib/mallcop/shell.rb

@@ -0,0 +1,49 @@
+require 'digest/sha1'
+
+module MallCop
+  class Shell
+    attr_reader :channel
+
+    def initialize(session)
+      @session = session
+      @channel = session.open_channel
+
+      @channel.shell
+    end
+
+    def sh(cmd)
+      res, code = exec_with_status(cmd)
+      res
+    end
+
+    def cleanup
+      @channel.close
+    end
+
+  private
+
+    def exec_with_status(cmd)
+      separator = generate_separator
+      # An `echo ''` is appended for now, mostly because I don't know of
+      # a better way to do this.
+      cmd = cmd.dup.strip
+      cmd << ";" if cmd !~ /[;&]$/
+      cmd << %[ echo #{separator} $?\n]
+
+      @channel.write(cmd)
+
+      res = ""
+      until res =~ /#{separator} (\d+)\Z/m
+        res << @channel.read
+      end
+
+      return $`.strip, $1.to_i
+    end
+
+    def generate_separator
+      s = Digest::SHA1.hexdigest([channel.object_id, object_id, Time.now.to_i, Time.now.usec, rand(0xFFFFFFFF)].join(":"))
+      s << Digest::SHA1.hexdigest(s)
+    end
+
+  end
+end

metadata

@@ -0,0 +1,81 @@
+--- !ruby/object:Gem::Specification 
+name: mallcop
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Carl Lerche
+- Aaron Patterson
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-08-18 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Don't reinvent the wheel.
+email: 
+- me@carllerche.com
+executables: []
+
+extensions: 
+- ext/mallcop/extconf.rb
+extra_rdoc_files: []
+
+files: 
+- ext/mallcop/extconf.rb
+- ext/mallcop/channel.c
+- ext/mallcop/mallcop.c
+- ext/mallcop/session.c
+- ext/mallcop/mallcop.h
+- lib/mallcop.rb
+- lib/mallcop/channel.rb
+- lib/mallcop/session.rb
+- lib/mallcop/shell.rb
+- README.md
+- LICENSE
+has_rdoc: true
+homepage: http://github.com/carllerche/mallcop
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 23
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: mallcop
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Bindings to libssh2
+test_files: []
+