Fortified the d2c compiler by checking asprint() return values.

1 files changed, 9 insertions, 3 deletions

diff --git a/tools/d2c/rules/tokens.l b/tools/d2c/rules/tokens.l
index 6b14a85..7c6c020 100644
--- a/tools/d2c/rules/tokens.l
+++ b/tools/d2c/rules/tokens.l
@@ -55,9 +55,15 @@
 
 .                   {
                         char *msg;
-                        asprintf(&msg, "Unhandled token in d2c rules block: '%s'", yytext);
-                        YY_FATAL_ERROR(msg);
-                        free(msg);
+                        int ret;
+                        ret = asprintf(&msg, "Unhandled token in d2c rules block: '%s'", yytext);
+                        if (ret == -1)
+                            YY_FATAL_ERROR("Unhandled token in undisclosed d2c rules block");
+                        else
+                        {
+                            YY_FATAL_ERROR(msg);
+                            free(msg);
+                        }
                     }