From 74dedc49c70673da227651e6f15412445be4230b Mon Sep 17 00:00:00 2001
From: Igor Pashev <pashev.igor@gmail.com>
Date: Wed, 14 Sep 2011 01:34:46 +0400
Subject: Fixed segfault

---
 brainfuck.c | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/brainfuck.c b/brainfuck.c
index dc36340..d62c2a9 100644
--- a/brainfuck.c
+++ b/brainfuck.c
@@ -78,7 +78,7 @@ read_code()
 
     if (n >= allocated)
     {
-        allocated <<= 1;
+        allocated *= 2;
         code = (char *)realloc(code, allocated * sizeof(char));
     }
     code[n++] = '\0';
@@ -740,9 +740,16 @@ main(int argc, char **argv)
     {
         bf2c();
     }
-    else if (strcmp(print_as, "cow") == 0)
+    else if (print_as != NULL)
     {
-        bf2moo();
+        if (strcmp(print_as, "cow") == 0)
+        {
+            bf2moo();
+        }
+        else
+        {
+            fprintf(stderr, "Unknown argument to -p: %s\n", print_as);
+        }
     }
     else
     {
-- 
cgit v1.2.3