Tests: Fix FILE stream leaks in helper programs

This commit is contained in:
Brad King
2024-10-23 19:12:51 -04:00
parent 6fdb15672d
commit be540dbe19
2 changed files with 11 additions and 7 deletions

View File

@@ -162,6 +162,7 @@ int main(int argc, char** argv)
jobserver = jobserver_auth(message);
if (jobserver == NULL) {
fclose(fp);
fprintf(stderr, "%s\n", message);
return 1;
}
@@ -171,6 +172,7 @@ int main(int argc, char** argv)
#else
result = posix(jobserver, message);
#endif
fclose(fp);
free(jobserver);
message[MAX_MESSAGE_LENGTH] = '\0';

View File

@@ -9,6 +9,8 @@ int main(int argc, char* argv[])
{
FILE* source;
FILE* target;
char buffer[500];
size_t n;
int i;
for (i = 1; i < argc; ++i) {
if (strcmp(argv[i], "-bad") == 0) {
@@ -22,13 +24,13 @@ int main(int argc, char* argv[])
return 1;
}
target = fopen(argv[argc - 2], "wb");
if (target != NULL) {
char buffer[500];
size_t n = fread(buffer, 1, sizeof(buffer), source);
fwrite(buffer, 1, n, target);
if (target == NULL) {
fclose(source);
fclose(target);
return 0;
return 1;
}
return 1;
n = fread(buffer, 1, sizeof(buffer), source);
fwrite(buffer, 1, n, target);
fclose(source);
fclose(target);
return 0;
}