SSH module async

.gitignore

@@ -0,0 +1,5 @@
+.acton.lock
+.build
+out
+zig-cache
+zig-out

Build.act

@@ -0,0 +1,9 @@
+name = "acton_ssh"
+fingerprint = 0xb650ac2857701cb1
+zig_dependencies = {
+   "libssh": (
+      url="https://github.com/precla/libssh/archive/refs/heads/zig-build.tar.gz",
+      hash="libssh-0.11.0-kUQJaoJiKgDjZpTAyxxo6GamiMzj9LPxlEr2FacwOxIR",
+      artifacts=[ "ssh" ]
+  )
+}

src/netconf.h

@@ -0,0 +1,35 @@
+// NETCONF <hello> message
+#define NETCONF_HELLO_MSG \
+    "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" \
+    "<hello xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n" \
+    "  <capabilities>\n" \
+    "    <capability>urn:ietf:params:netconf:base:1.0</capability>\n" \
+    "  </capabilities>\n" \
+    "</hello>]]>]]>"
+
+#define NETCONF_GET_CONFIG_MSG \
+    "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" \
+    "<rpc xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\" message-id=\"10\">\n" \
+    "  <get-config>\n" \
+    "    <source>\n" \
+    "      <running/>\n" \
+    "    </source>\n" \
+    "  </get-config>\n" \
+    "</rpc>]]>]]>"
+
+#define NETCONF_GET_STATE \
+    "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" \
+    "<rpc xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\" message-id=\"10\">\n" \
+    "  <get>" \
+    "    <filter type=\"subtree\">" \
+    "      <interfaces xmlns=\"urn:ietf:params:xml:ns:yang:ietf-interfaces\"/>" \
+    "    </filter>" \
+    "  </get>" \
+    "</rpc>]]>]]>"
+
+// NETCONF <close-session> message
+#define NETCONF_CLOSE_SESSION_MSG \
+    "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" \
+    "<rpc message-id=\"9999\" xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n" \
+    "  <close-session/>\n" \
+    "</rpc>]]>]]>"

src/ssh.act

@@ -1,69 +1,125 @@
 import net
-import testing
+
 
 def version() -> str:
-    """Get the libssh version"""
+    """Get the acton-ssh version"""
     NotImplemented
 
-def _test_version():
-    testing.assertEqual("0.11.0", version())
 
 actor Client(cap: net.TCPConnectCap,
              host: str,
              username: str,
              on_connect: action(Client) -> None,
-             on_close: action(Client, str) -> None,
-             key: ?str=None,
+             on_receive: action(Client, bytes) -> None,
+             on_error: action(Client, str) -> None,
+             on_remote_close: ?action(Client) -> None,
              password: ?str=None,
-             port: u16=22,
-             ):
-    """SSH Client"""
+             key: ?str=None,
+             port: int=22,
+             subsystem: ?str=None):
+    """SSH Client
+
+    Opens an SSH connection. If subsystem is None, opens a shell channel.
+    If subsystem is set (e.g. "netconf"), opens a subsystem channel instead.
+    """
+
+    var _client: u64 = 0
 
-    # haha, this is really a pointer :P
-    var _ssh_session: u64 = 0
+    proc def _pin_affinity() -> None:
+        NotImplemented
+    _pin_affinity()
+
+    action def write(data: bytes) -> None:
+        """Write data to the SSH channel"""
+        NotImplemented
+
+    action def close(on_close: action(Client) -> None) -> None:
+        """Close the connection"""
+        NotImplemented
+
+    def reconnect():
+        close(_connect)
+
+    def _connect(c):
+        NotImplemented
 
     proc def _init() -> None:
         """Initialize the SSH client"""
         NotImplemented
     _init()
-    print("SSH Client connected")
 
-#    action def close(on_close: action(TLSConnection) -> None) -> None:
-#        """Close the connection"""
-#        NotImplemented
-#
-#    def reconnect():
-#        close(_connect)
-#
-#    def _connect(c):
-#        NotImplemented
-#
 
-# TODO: implement support for channels
-# AFAIK, all things over ssh are done via channels, so need some channel
-# primitive, maybe an actor per channel that then multiplexes into the Client
-# session? Prolly need some higher level wrappers for common things like
-# starting a shell or running a single command. SFTP / SCP would be nice too,
-# but for sometime in the future. Custom subsystems need to be supported too.
+actor main(env):
+    var _buf: bytes = b""
 
+    NETCONF_HELLO = b"""<?xml version="1.0" encoding="UTF-8"?>
+<hello xmlns="urn:ietf:params:xml:ns:netconf:base:1.0">
+  <capabilities>
+    <capability>urn:ietf:params:netconf:base:1.0</capability>
+  </capabilities>
+</hello>]]>]]>"""
 
+    NETCONF_GET_CONFIG = b"""<?xml version="1.0" encoding="UTF-8"?>
+<rpc xmlns="urn:ietf:params:xml:ns:netconf:base:1.0" message-id="10">
+  <get-config>
+    <source>
+      <running/>
+    </source>
+  </get-config>
+</rpc>]]>]]>"""
 
+    NETCONF_CLOSE = b"""<?xml version="1.0" encoding="UTF-8"?>
+<rpc message-id="9999" xmlns="urn:ietf:params:xml:ns:netconf:base:1.0">
+  <close-session/>
+</rpc>]]>]]>"""
+
+    var _state: int = 0  # 0=wait hello, 1=wait get-config reply, 2=wait close reply
 
-actor main(env):
     def on_connect(client: Client):
-        print("Connected")
+        print("Connected! Sending NETCONF hello...")
+        client.write(NETCONF_HELLO)
+
+    def on_receive(client: Client, data: bytes):
+        _buf += data
+        s = _buf.decode()
+        if "]]>]]>" in s:
+            print("=== NETCONF message ===")
+            print(s)
+            print("=======================")
+            _buf = b""
+
+            st = _state
+            if st == 0:
+                print("Got server hello, sending get-config...")
+                _state = 1
+                client.write(NETCONF_GET_CONFIG)
+            elif st == 1:
+                print("Got config, sending close-session...")
+                _state = 2
+                client.write(NETCONF_CLOSE)
+            elif st == 2:
+                print("Got close reply, done!")
+                def _on_close(c: Client):
+                    env.exit(0)
+                client.close(_on_close)
+
+    def on_error(client: Client, error: str):
+        print("Error:", error)
+        env.exit(1)
 
-    def on_close(client: Client, error: str):
-        print("Error", error)
+    def on_remote_close(client: Client):
+        print("Remote closed connection")
+        env.exit(0)
 
-    print(version())
     c = Client(
         net.TCPConnectCap(net.TCPCap(net.NetCap(env.cap))),
         "localhost",
         "foo",
         on_connect,
-        on_close,
+        on_receive,
+        on_error,
+        on_remote_close,
         password="bar",
-        port=2223,
+        port=830,
+        subsystem="netconf",
     )
-    env.exit(0)