massive commit to get umass working, only significant known bug is that changes don...

[goodfet] / client / USBClass.py

diff --git a/client/USBClass.py b/client/USBClass.py
new file mode 100644 (file)
index 0000000..daac236
--- /dev/null
+++ b/client/USBClass.py
@@ -0,0 +1,35 @@
+# USBClass.py
+#
+# Contains class definition for USBClass, intended as a base class (in the OO
+# sense) for implementing device classes (in the USB sense), eg, HID devices,
+# mass storage devices.
+
+class USBClass:
+    name = "generic USB device class"
+
+    # maps bRequest to handler function
+    request_handlers = { }
+
+    def __init__(self, verbose=0):
+        self.interface = None
+        self.verbose = verbose
+
+        self.setup_request_handlers()
+
+    def set_interface(self, interface):
+        self.interface = interface
+
+    def setup_request_handlers(self):
+        """To be overridden for subclasses to modify self.class_request_handlers"""
+        pass
+
+    def handle_class_request(self, req):
+        r = req.get_request()
+
+        if r in self.class_request_handlers:
+            self.class_request_handlers[r](req)
+        else:
+            print(self.name, "unhandled class request:", req)
+            print(self.name, "stalling")
+            self.device.stall_ep0()
+