2 files changed, 10 insertions, 0 deletions
diff --git a/Tests/CustomCommand/generator.cxx b/Tests/CustomCommand/generator.cxx
index 52ffdc3c47..99da90ceb4 100644
--- a/Tests/CustomCommand/generator.cxx
+++ b/Tests/CustomCommand/generator.cxx
@@ -2,6 +2,11 @@
 
 int main(int argc, char *argv[])
 {
+  if ( argc < 2 )
+    {
+    fprintf(stderr, "Usage: %s <file>\n", argv[0]);
+    return 1;
+    }
   FILE *fp = fopen(argv[1],"w");
   
   fprintf(fp,"int generated() { return 3; }\n");
diff --git a/Tests/CustomCommand/wrapper.cxx b/Tests/CustomCommand/wrapper.cxx
index b472f7f94c..3a1149da5f 100644
--- a/Tests/CustomCommand/wrapper.cxx
+++ b/Tests/CustomCommand/wrapper.cxx
@@ -2,6 +2,11 @@
 
 int main(int argc, char *argv[])
 {
+  if ( argc < 2 )
+    {
+    fprintf(stderr, "Usage: %s <file>\n", argv[0]);
+    return 1;
+    }
   FILE *fp = fopen(argv[1],"w");
   
   fprintf(fp,"int wrapped() { return 5; }\n");