From cfd5e1e6d19d94481ef84b4a16b41b2fae6c252e Mon Sep 17 00:00:00 2001
From: hathach <thach@tinyusb.org>
Date: Sun, 24 Jun 2018 22:30:44 +0700
Subject: [PATCH] clean up

---
 examples/device/nrf52840/src/tusb_config.h | 2 +-
 examples/obsolete/device/src/tusb_config.h | 2 +-
 src/class/msc/msc_device.c                 | 6 ++++++
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/examples/device/nrf52840/src/tusb_config.h b/examples/device/nrf52840/src/tusb_config.h
index aee6099b5..8e031ea3b 100644
--- a/examples/device/nrf52840/src/tusb_config.h
+++ b/examples/device/nrf52840/src/tusb_config.h
@@ -80,7 +80,7 @@
 // If not enabled, application must call tud_cdc_flush() periodically
 #define CFG_TUD_CDC_FLUSH_ON_SOF    1
 
-// Number of supported Logical Unit Number
+// Number of supported Logical Unit Number (At least 1)
 #define CFG_TUD_MSC_MAXLUN        1
 
 // Buffer size of Device Mass storage
diff --git a/examples/obsolete/device/src/tusb_config.h b/examples/obsolete/device/src/tusb_config.h
index 3904facbe..9dc9b3c2f 100644
--- a/examples/obsolete/device/src/tusb_config.h
+++ b/examples/obsolete/device/src/tusb_config.h
@@ -86,7 +86,7 @@
 #define CFG_TUD_CDC_FLUSH_ON_SOF  1
 
 
-// Number of supported Logical Unit Number
+// Number of supported Logical Unit Number (At least 1)
 #define CFG_TUD_MSC_MAXLUN        1
 
 // Buffer size of Device Mass storage
diff --git a/src/class/msc/msc_device.c b/src/class/msc/msc_device.c
index ec804dcc8..3adcbc985 100644
--- a/src/class/msc/msc_device.c
+++ b/src/class/msc/msc_device.c
@@ -48,6 +48,12 @@
 #include "msc_device.h"
 #include "device/usbd_pvt.h"
 
+VERIFY_STATIC(CFG_TUD_MSC_BUFSIZE < UINT16_MAX, "Size is not correct");
+
+#ifndef CFG_TUD_MSC_MAXLUN
+#define CFG_TUD_MSC_MAXLUN 1
+#endif
+
 //--------------------------------------------------------------------+
 // MACRO CONSTANT TYPEDEF
 //--------------------------------------------------------------------+