1 files changed, 13 insertions, 0 deletions

diff --git a/firmware/target/arm/imx31/gigabeat-s/usb-target.h b/firmware/target/arm/imx31/gigabeat-s/usb-target.h
index c93400ca0b..7931058241 100644
--- a/firmware/target/arm/imx31/gigabeat-s/usb-target.h
+++ b/firmware/target/arm/imx31/gigabeat-s/usb-target.h
@@ -30,4 +30,17 @@ void usb_init_device(void);
 /* Read the immediate state of the cable from the PMIC */
 bool usb_plugged(void);
 
+/** Sector read/write filters **/
+
+/* Filter some things in the MBR - see usb-gigabeat-s.c */
+void usb_fix_mbr(unsigned char *mbr);
+#define USBSTOR_READ_SECTORS_FILTER() \
+    ({ if (cur_cmd.sector == 0) \
+            usb_fix_mbr(cur_cmd.data[cur_cmd.data_select]); \
+    0; })
+
+/* Disallow MBR writes entirely since it was "fixed" in usb_fix_mbr */
+#define USBSTOR_WRITE_SECTORS_FILTER() \
+    ({ cur_cmd.sector != 0 ? 0 : -1; })
+
 #endif /* USB_TARGET */