fumiama/base16384
1
0
Fork 0
mirror of https://github.com/fumiama/base16384.git synced 2026-06-27 04:00:46 +08:00
Code Activity

优化 perror 输出

Browse Source
This commit is contained in:
源文雨
2022-05-01 00:07:14 +08:00
parent ac0c22ea63
commit 0a43046c9d
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
base16384.c
View File

@@ -34,12 +34,12 @@ static int encode_file(const char* input, const char* output) {
fp = stdin; fp = stdin;
} else inputsize = get_file_size(input); } else inputsize = get_file_size(input);
if(inputsize < 0) { if(inputsize < 0) {
perror("Get file size error: "); perror("Get file size error");
return 1; return 1;
} }
fpo = (*(uint16_t*)output == *(uint16_t*)"-")?stdout:fopen(output, "wb"); fpo = (*(uint16_t*)output == *(uint16_t*)"-")?stdout:fopen(output, "wb");
if(!fpo) { if(!fpo) {
perror("Fopen output file error: "); perror("Fopen output file error");
return 2; return 2;
} }
if(!inputsize || inputsize > BUFSIZ*1024) { // stdin or big file, use encbuf & fread if(!inputsize || inputsize > BUFSIZ*1024) { // stdin or big file, use encbuf & fread
@@ -49,7 +49,7 @@ static int encode_file(const char* input, const char* output) {
#endif #endif
if(!fp) fp = fopen(input, "rb"); if(!fp) fp = fopen(input, "rb");
if(!fp) { if(!fp) {
perror("Fopen input file error: "); perror("Fopen input file error");
return 3; return 3;
} }
@@ -60,7 +60,7 @@ static int encode_file(const char* input, const char* output) {
while((cnt = fread(encbuf, sizeof(char), inputsize, fp))) { while((cnt = fread(encbuf, sizeof(char), inputsize, fp))) {
int n = encode(encbuf, cnt, decbuf, outputsize); int n = encode(encbuf, cnt, decbuf, outputsize);
if(fwrite(decbuf, n, 1, fpo) <= 0) { if(fwrite(decbuf, n, 1, fpo) <= 0) {
perror("Write file error: "); perror("Write file error");
return 4; return 4;
} }
} }
@@ -72,12 +72,12 @@ static int encode_file(const char* input, const char* output) {
} else { // small file, use mmap & fwrite } else { // small file, use mmap & fwrite
int fd = open(input, O_RDONLY); int fd = open(input, O_RDONLY);
if(fd <= 0) { if(fd <= 0) {
perror("Open input file error: "); perror("Open input file error");
return 5; return 5;
} }
char *input_file = mmap(NULL, (size_t)inputsize, PROT_READ, MAP_PRIVATE, fd, 0); char *input_file = mmap(NULL, (size_t)inputsize, PROT_READ, MAP_PRIVATE, fd, 0);
if(input_file == MAP_FAILED) { if(input_file == MAP_FAILED) {
perror("Map input file error: "); perror("Map input file error");
return 6; return 6;
} }
int outputsize = encode_len(inputsize)+16; int outputsize = encode_len(inputsize)+16;
@@ -85,7 +85,7 @@ static int encode_file(const char* input, const char* output) {
fputc(0xFF, fpo); fputc(0xFF, fpo);
int n = encode(input_file, (int)inputsize, decbuf, outputsize); int n = encode(input_file, (int)inputsize, decbuf, outputsize);
if(fwrite(decbuf, n, 1, fpo) <= 0) { if(fwrite(decbuf, n, 1, fpo) <= 0) {
perror("Write file error: "); perror("Write file error");
return 7; return 7;
} }
munmap(input_file, (size_t)inputsize); munmap(input_file, (size_t)inputsize);
@@ -122,12 +122,12 @@ static int decode_file(const char* input, const char* output) {
fp = stdin; fp = stdin;
} else inputsize = get_file_size(input); } else inputsize = get_file_size(input);
if(inputsize < 0) { if(inputsize < 0) {
perror("Get file size error: "); perror("Get file size error");
return 1; return 1;
} }
fpo = (*(uint16_t*)output == *(uint16_t*)"-")?stdout:fopen(output, "wb"); fpo = (*(uint16_t*)output == *(uint16_t*)"-")?stdout:fopen(output, "wb");
if(!fpo) { if(!fpo) {
perror("Fopen output file error: "); perror("Fopen output file error");
return 2; return 2;
} }
if(!inputsize || inputsize > BUFSIZ*1024) { // stdin or big file, use decbuf & fread if(!inputsize || inputsize > BUFSIZ*1024) { // stdin or big file, use decbuf & fread
@@ -137,7 +137,7 @@ static int decode_file(const char* input, const char* output) {
#endif #endif
if(!fp) fp = fopen(input, "rb"); if(!fp) fp = fopen(input, "rb");
if(!fp) { if(!fp) {
perror("Fopen input file error: "); perror("Fopen input file error");
return 3; return 3;
} }
int outputsize = decode_len(inputsize, 0)+16; int outputsize = decode_len(inputsize, 0)+16;
@@ -150,7 +150,7 @@ static int decode_file(const char* input, const char* output) {
decbuf[cnt++] = end; decbuf[cnt++] = end;
} }
if(fwrite(encbuf, decode(decbuf, cnt, encbuf, outputsize), 1, fpo) <= 0) { if(fwrite(encbuf, decode(decbuf, cnt, encbuf, outputsize), 1, fpo) <= 0) {
perror("Write file error: "); perror("Write file error");
return 4; return 4;
} }
} }
@@ -162,18 +162,18 @@ static int decode_file(const char* input, const char* output) {
} else { // small file, use mmap & fwrite } else { // small file, use mmap & fwrite
int fd = open(input, O_RDONLY); int fd = open(input, O_RDONLY);
if(fd <= 0) { if(fd <= 0) {
perror("Open input file error: "); perror("Open input file error");
return 5; return 5;
} }
char *input_file = mmap(NULL, (size_t)inputsize, PROT_READ, MAP_PRIVATE, fd, 0); char *input_file = mmap(NULL, (size_t)inputsize, PROT_READ, MAP_PRIVATE, fd, 0);
if(input_file == MAP_FAILED) { if(input_file == MAP_FAILED) {
perror("Map input file error: "); perror("Map input file error");
return 6; return 6;
} }
int outputsize = decode_len(inputsize, 0)+16; int outputsize = decode_len(inputsize, 0)+16;
int off = skip_offset(input_file); int off = skip_offset(input_file);
if(fwrite(encbuf, decode(input_file+off, inputsize-off, encbuf, outputsize), 1, fpo) <= 0) { if(fwrite(encbuf, decode(input_file+off, inputsize-off, encbuf, outputsize), 1, fpo) <= 0) {
perror("Write file error: "); perror("Write file error");
return 7; return 7;
} }
munmap(input_file, (size_t)inputsize); munmap(input_file, (size_t)inputsize);