Fix crash for empty table headers or row

OneDrive · Feb 19, 2016 · 70afff3 · 70afff3
1 parent 5767b3b
commit 70afff3
Showing 1 changed file with 21 additions and 5 deletions.
diff --git a/OSS/MarkdownDeep/TableSpec.cs b/OSS/MarkdownDeep/TableSpec.cs
@@ -227,15 +227,31 @@ public static TableSpec Parse(StringScanner p)
 
         #region RGregg extensions
 
-        public string[] ColumnHeaders { get { return Headers.ToArray(); } }
+        public string[] ColumnHeaders
+        {
+            get
+            {
+                if (null != Headers)
+                    return Headers.ToArray();
+                else
+                    return new string[0];
+            }
+        }
 
         public string[][] RowValues
         { 
             get
-            { 
-                var rowArrays = from row in Rows
-                                select row.ToArray();
-                return rowArrays.ToArray();
+            {
+                if (null != Rows)
+                {
+                    var rowArrays = from row in Rows
+                                    select row.ToArray();
+                    return rowArrays.ToArray();
+                }
+                else
+                {
+                    return new string[][] { new string[] { } };
+                }
             }
         }