RAHHH

backend/lib/python3.12/site-packages/twisted/conch/ssh/service.py

@@ -0,0 +1,56 @@
+# Copyright (c) Twisted Matrix Laboratories.
+# See LICENSE for details.
+
+"""
+The parent class for all the SSH services.  Currently implemented services
+are ssh-userauth and ssh-connection.
+
+Maintainer: Paul Swartz
+"""
+
+from typing import Dict
+
+from twisted.logger import Logger
+
+
+class SSHService:
+    # this is the ssh name for the service:
+    name: bytes = None  # type:ignore[assignment]
+
+    protocolMessages: Dict[int, str] = {}  # map #'s -> protocol names
+    transport = None  # gets set later
+
+    _log = Logger()
+
+    def serviceStarted(self):
+        """
+        called when the service is active on the transport.
+        """
+
+    def serviceStopped(self):
+        """
+        called when the service is stopped, either by the connection ending
+        or by another service being started
+        """
+
+    def logPrefix(self):
+        return "SSHService {!r} on {}".format(
+            self.name, self.transport.transport.logPrefix()
+        )
+
+    def packetReceived(self, messageNum, packet):
+        """
+        called when we receive a packet on the transport
+        """
+        # print self.protocolMessages
+        if messageNum in self.protocolMessages:
+            messageType = self.protocolMessages[messageNum]
+            f = getattr(self, "ssh_%s" % messageType[4:], None)
+            if f is not None:
+                return f(packet)
+        self._log.info(
+            "couldn't handle {messageNum} {packet!r}",
+            messageNum=messageNum,
+            packet=packet,
+        )
+        self.transport.sendUnimplemented()