nortega/indivisible-legacy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't allow writing to the same file.

Browse Source
This commit is contained in:
Nicolás Ortega Froysa 2017-09-17 05:46:03 +02:00
parent c73bfd1501
commit 3348d52c49
No known key found for this signature in database
GPG Key ID: FEC70E3BAE2E69BF
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/files.c
View File

@ -26,6 +26,7 @@ int inputPrimes(char *file, List *list) {
assert(file); assert(file);
assert(list); assert(list);
printf("Loading primes from `%s'...\n", file);
FILE *in = fopen(file, "r"); FILE *in = fopen(file, "r");
if(!in) return 1; if(!in) return 1;
mpz_t n; mpz_t n;

5
src/main.c
View File

@ -21,6 +21,7 @@
#include <signal.h> #include <signal.h>
#include <stdbool.h> #include <stdbool.h>
#include <unistd.h> #include <unistd.h>
#include <string.h>
#include <gmp.h> #include <gmp.h>
#include <omp.h> #include <omp.h>
@ -114,6 +115,10 @@ int main(int argc, char *argv[]) {
printUsage(argv[0]); printUsage(argv[0]);
return 0; return 0;
} else if(efile && dfile) { } else if(efile && dfile) {
if(strcmp(efile, dfile) == 0) {
fprintf(stderr, "Error: file paths are the same.\n");
return 0;
}
int err = exportPrimes(efile, dfile, base); int err = exportPrimes(efile, dfile, base);
if(err) { if(err) {
if(err == 1) fprintf(stderr, "Error: failed to open files.\n"); if(err == 1) fprintf(stderr, "Error: failed to open files.\n");