cmd/gc: check malloc return value

Check the return value from malloc - do not assume that we were
allocated memory just because we asked for it.

Update #4415.

R=minux.ma, daniel.morsing, remyoudompheng, rsc
CC=golang-dev
https://golang.org/cl/6782100

diff --git a/src/cmd/gc/lex.c b/src/cmd/gc/lex.c
index ad8bdebf036174404474310115ba1ac16b5d6289..8e16747efd5fda5ad74afd88d655c5fe0856a632 100644 (file)
--- a/src/cmd/gc/lex.c
+++ b/src/cmd/gc/lex.c
@@ -1165,6 +1165,11 @@ l0:
        case '[':
                if(loophack || lstk != nil) {
                        h = malloc(sizeof *h);
+                       if(h == nil) {
+                               flusherrors();
+                               yyerror("out of memory");
+                               errorexit();
+                       }
                        h->v = loophack;
                        h->next = lstk;
                        lstk = h;

diff --git a/src/cmd/gc/sinit.c b/src/cmd/gc/sinit.c
index d1438f1003008d3bbc8a4fac320677cc32a86781..e8010099d211e60cf1d034e4c50be172d4562306 100644 (file)
--- a/src/cmd/gc/sinit.c
+++ b/src/cmd/gc/sinit.c
@@ -84,6 +84,11 @@ init1(Node *n, NodeList **out)
        }
        n->initorder = InitPending;
        l = malloc(sizeof *l);
+       if(l == nil) {
+               flusherrors();
+               yyerror("out of memory");
+               errorexit();
+       }
        l->next = initlist;
        l->n = n;
        l->end = nil;