Make generator exe more resilient.

pull/11425/merge
Jussi Pakkanen 2 years ago
parent 24ea1d3f19
commit 58e6fb40f6
  1. 12
      test cases/common/105 generatorcustom/gen.c

@ -22,9 +22,17 @@ int main(int argc, const char ** argv) {
fprintf(output, "#pragma once\n");
fprintf(output, "#define ");
char c;
int bytes_copied = 0;
int c;
while((c = fgetc(input)) != EOF) {
fputc(c, output);
if(fputc(c, output) == EOF) {
fprintf(stderr, "Writing to output file failed.\n");
return 1;
}
if(++bytes_copied > 10000) {
fprintf(stderr, "File copy stuck in an eternal loop!\n");
return 1;
}
}
fputc('\n', output);

Loading…
Cancel
Save