shithub: riscv

Info  •  Files  •  Log  •  Branches

cc: fix include array overflow handling

--- a/sys/src/cmd/cc/lex.c

+++ b/sys/src/cmd/cc/lex.c

@@ -1535,12 +1535,12 @@

 			if(strcmp(p, include[i]) == 0)

 				break;

-		if(i >= ninclude)

+		if(i >= ninclude){

+			if(ninclude >= nelem(include)) {

+				diag(Z, "ninclude too small %d", nelem(include));

+				exits("ninclude");

+			}

 			include[ninclude++] = p;

-

-		if(ninclude > nelem(include)) {

-			diag(Z, "ninclude too small %d", nelem(include));

-			exits("ninclude");

 		}

 		if(e == 0)