[WIP] ssh backend

Author: marcy-terui

.gitignore

@@ -56,5 +56,3 @@ coverage.xml
 Session.vim
 .netrwhist
 *~
-
-test.py

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "specinfra"]
+	path = specinfra
+	url = [email protected]:libspecinfra/specinfra.git

.travis.yml

@@ -0,0 +1,10 @@
+language: rust
+rust:
+  - stable
+  - beta
+  - nightly
+matrix:
+  allow_failures:
+    - rust: nightly
+script:
+  - cd specinfra && cargo build --release

build.sh

@@ -0,0 +1,2 @@
+docker run --rm --user "$(id -u)":"$(id -g)" -v "$PWD/specinfra":/usr/src/specinfra -w /usr/src/specinfra rust cargo build --release
+cp specinfra/target/release/libspecinfra.so ./libspecinfra/

libspecinfra/__init__.py

@@ -8,6 +8,7 @@
 
 import sys
 import ctypes
+import os
 import libspecinfra.resource
 
 from libspecinfra.structures import SpecinfraS, BackendWrapperS
@@ -18,10 +19,11 @@ def load():
     global library
 
     if library is None:
+        libdir = os.path.dirname(os.path.abspath(__file__))
         prefix = {'win32': ''}.get(sys.platform, 'lib')
         extension = {'darwin': '.dylib', 'win32': '.dll'}.get(sys.platform, '.so')
-        library = ctypes.cdll.LoadLibrary(
-            '{}specinfra{}'.format(prefix, extension))
+        libpath = os.path.join(libdir, '{}specinfra{}'.format(prefix, extension))
+        library = ctypes.cdll.LoadLibrary(libpath)
 
     return library
 

libspecinfra/backend/__init__.py

@@ -1,6 +1,6 @@
 import ctypes
 import libspecinfra
-from libspecinfra.structures import BackendWrapperS
+from libspecinfra.structures import BackendWrapperS, BackendWrapperS ,SSHBuilderS
 
 
 class Direct(object):
@@ -16,12 +16,40 @@ def __exit__(self, exc_type, exc_value, traceback):
 
 class SSH(object):
 
-    def __init__(self, host):
+    def __init__(self, host, user=None, password=None, key_file=None, port=None):
         lib = libspecinfra.load()
-        lib.backend_ssh_new.argtypes = (ctypes.c_char_p,)
-        lib.backend_ssh_new.restype = ctypes.POINTER(BackendWrapperS)
-        lib.backend_ssh_free.argtypes = (ctypes.POINTER(BackendWrapperS),)
-        self.obj = lib.backend_ssh_new(host.encode('utf-8'))
+        ssh_builder_p = ctypes.POINTER(SSHBuilderS)
+        lib.backend_ssh_builder_new.argtypes = (ctypes.c_char_p,)
+        lib.backend_ssh_builder_new.restype = ssh_builder_p
+        ssh_builder = lib.backend_ssh_builder_new(host.encode('utf-8'))
+
+        if user is not None:
+            user = user.encode('utf-8')
+            lib.backend_ssh_builder_user.argtypes = (ssh_builder_p, ctypes.c_char_p,)
+            lib.backend_ssh_builder_user.restype = ssh_builder_p
+            ssh_builder = lib.backend_ssh_builder_user(ssh_builder, user)
+
+        if password is not None:
+            password = password.encode('utf-8')
+            lib.backend_ssh_builder_password.argtypes = (ssh_builder_p, ctypes.c_char_p,)
+            lib.backend_ssh_builder_password.restype = ssh_builder_p
+            ssh_builder = lib.backend_ssh_builder_password(ssh_builder, password)
+
+        if key_file is not None:
+            key_file = key_file.encode('utf-8')
+            lib.backend_ssh_builder_key_file.argtypes = (ssh_builder_p, ctypes.c_char_p,)
+            lib.backend_ssh_builder_key_file.restype = ssh_builder_p
+            ssh_builder = lib.backend_ssh_builder_key_file(ssh_builder, key_file)
+
+        if port is not None:
+            lib.backend_ssh_builder_port.argtypes = (ssh_builder_p, ctypes.c_uint32,)
+            lib.backend_ssh_builder_port.restype = ssh_builder_p
+            ssh_builder = lib.backend_ssh_builder_port(ssh_builder, port)
+
+        lib.backend_ssh_builder_finalize.argtypes = (ssh_builder_p,)
+        lib.backend_ssh_builder_finalize.restype = ctypes.POINTER(BackendWrapperS)
+        self.obj = lib.backend_ssh_builder_finalize(ssh_builder)
 
     def __exit__(self, exc_type, exc_value, traceback):
+        libspecinfra.load().backend_ssh_builder_free.argtypes = (ctypes.POINTER(SSHBuilderS),)
         self.lib.backend_ssh_free(self.obj)

libspecinfra/structures.py

@@ -5,3 +5,5 @@ class SpecinfraS(ctypes.Structure): pass;
 class BackendWrapperS(ctypes.Structure): pass;
 
 class FileS(ctypes.Structure): pass;
+
+class SSHBuilderS(ctypes.Structure): pass;

setup.py

@@ -2,6 +2,11 @@
 import os
 
 from setuptools import setup, find_packages
+from setuptools.dist import Distribution
+
+class BinaryDistribution(Distribution):
+    def is_pure(self):
+        return False
 
 description = 'Python bindings for libspecinfra.'
 
@@ -14,6 +19,9 @@
     author_email='[email protected]',
     url='https://github.com/libspecinfra/libspecinfra-python',
     packages=find_packages(exclude=['tests*', 'specinfra*']),
+    package_data={'libspecinfra': ['libspecinfra/libspecinfra.so']},
+    include_package_data=True,
+    distclass=BinaryDistribution,
     # rust extensions are not zip safe, just like C-extensions.
     zip_safe=False,
     license='MIT License',

specinfra

@@ -0,0 +1,15 @@
+import libspecinfra
+import libspecinfra.backend
+
+#backend = libspecinfra.backend.Direct()
+
+backend = libspecinfra.backend.SSH(
+    host='ec2-52-25-30-143.us-west-2.compute.amazonaws.com',
+    user='ubuntu',
+    key_file='/Users/marcy/.ssh/orthros.pem',
+    port=22)
+specinfra = libspecinfra.Specinfra(backend)
+f = specinfra.file('/etc/passwd')
+
+#print(oct(f.mode())) # => 0o6441
+print(oct(f.mode()))