Fixed flash utility for STM32F4

[fw/stlink] / src / test_sg.c

diff --git a/src/test_sg.c b/src/test_sg.c
index 9dcd3db0bbe0fb090967d50ce81bfeac16c29d26..34fbe540caa000c2665a87582b7578487b2f1816 100644 (file)
--- a/src/test_sg.c
+++ b/src/test_sg.c
@@ -6,6 +6,10 @@
 
 #include <stdio.h>
 #include <stdlib.h>
+#if CONFIG_USE_LIBSG
+#include <scsi/sg_lib.h>
+#include <scsi/sg_pt.h>
+#endif
 #include "stlink-common.h"
 
 int main(int argc, char *argv[]) {
@@ -208,4 +212,4 @@ int main(int argc, char *argv[]) {
 
        //fflush(stderr); fflush(stdout);
        return EXIT_SUCCESS;
-}
\ No newline at end of file
+}