1
0
mirror of https://github.com/fumiama/base16384.git synced 2026-06-27 12:10:24 +08:00

Merge pull request #10 from fumiama/master

sync
This commit is contained in:
源文雨
2022-10-17 12:58:12 +08:00
committed by GitHub
3 changed files with 28 additions and 20 deletions

View File

@@ -103,6 +103,7 @@ int main(int argc, char** argv) {
case base16384_err_write_file: perror("base16384_err_write_file"); break;
case base16384_err_open_input_file: perror("base16384_err_open_input_file"); break;
case base16384_err_map_input_file: perror("base16384_err_map_input_file"); break;
case base16384_err_read_file: perror("base16384_err_read_file"); break;
default: perror("base16384"); break;
}
return exitstat;

View File

@@ -20,6 +20,7 @@
#ifndef __cosmopolitan
#include <stdint.h>
#include <stdio.h>
#endif
// base16384_err_t is the return value of base16384_en/decode_file
@@ -31,6 +32,7 @@ enum base16384_err_t {
base16384_err_write_file,
base16384_err_open_input_file,
base16384_err_map_input_file,
base16384_err_read_file,
};
// base16384_err_t is the return value of base16384_en/decode_file
typedef enum base16384_err_t base16384_err_t;

45
file.c
View File

@@ -35,7 +35,7 @@
#ifdef __cosmopolitan
#define get_file_size(filepath) ((off_t)GetFileSize(filepath))
#else
static off_t get_file_size(const char* filepath) {
static inline off_t get_file_size(const char* filepath) {
struct stat statbuf;
return stat(filepath, &statbuf)?-1:statbuf.st_size;
}
@@ -70,7 +70,7 @@ base16384_err_t base16384_encode_file(const char* input, const char* output, cha
size_t cnt = 0;
fputc(0xFE, fpo);
fputc(0xFF, fpo);
while((cnt = fread(encbuf, sizeof(char), inputsize, fp))) {
while((cnt = fread(encbuf, sizeof(char), inputsize, fp)) > 0) {
int n = base16384_encode(encbuf, cnt, decbuf, outputsize);
if(fwrite(decbuf, n, 1, fpo) <= 0) {
return base16384_err_write_file;
@@ -115,7 +115,7 @@ base16384_err_t base16384_encode_fp(FILE* input, FILE* output, char* encbuf, cha
size_t cnt = 0;
fputc(0xFE, output);
fputc(0xFF, output);
while((cnt = fread(encbuf, sizeof(char), inputsize, input))) {
while((cnt = fread(encbuf, sizeof(char), inputsize, input)) > 0) {
int n = base16384_encode(encbuf, cnt, decbuf, outputsize);
if(fwrite(decbuf, n, 1, output) <= 0) {
return base16384_err_write_file;
@@ -125,17 +125,17 @@ base16384_err_t base16384_encode_fp(FILE* input, FILE* output, char* encbuf, cha
}
base16384_err_t base16384_encode_fd(int input, int output, char* encbuf, char* decbuf) {
if(!input) {
if(input < 0) {
return base16384_err_fopen_input_file;
}
if(!output) {
if(output < 0) {
return base16384_err_fopen_output_file;
}
off_t inputsize = BUFSIZ*1024/7*7;
int outputsize = base16384_encode_len(inputsize)+16;
size_t cnt = 0;
write(output, "\xfe\xff", 2);
while((cnt = read(input, encbuf, inputsize))) {
while((cnt = read(input, encbuf, inputsize)) > 0) {
int n = base16384_encode(encbuf, cnt, decbuf, outputsize);
if(write(output, decbuf, n) < n) {
return base16384_err_write_file;
@@ -152,7 +152,7 @@ base16384_err_t base16384_encode_fd(int input, int output, char* encbuf, char* d
#define skip_offset(input_file) ((input_file[0]==(char)0xFE)?2:0)
static int is_next_end(FILE* fp) {
static inline int is_next_end(FILE* fp) {
int ch = fgetc(fp);
if(ch == EOF) return 0;
if(ch == '=') return fgetc(fp);
@@ -188,7 +188,7 @@ base16384_err_t base16384_decode_file(const char* input, const char* output, cha
int cnt = 0;
int end = 0;
rm_head(fp);
while((cnt = fread(decbuf, sizeof(char), inputsize, fp))) {
while((cnt = fread(decbuf, sizeof(char), inputsize, fp)) > 0) {
if((end = is_next_end(fp))) {
decbuf[cnt++] = '=';
decbuf[cnt++] = end;
@@ -234,7 +234,7 @@ base16384_err_t base16384_decode_fp(FILE* input, FILE* output, char* encbuf, cha
int cnt = 0;
int end = 0;
rm_head(input);
while((cnt = fread(decbuf, sizeof(char), inputsize, input))) {
while((cnt = fread(decbuf, sizeof(char), inputsize, input)) > 0) {
if((end = is_next_end(input))) {
decbuf[cnt++] = '=';
decbuf[cnt++] = end;
@@ -246,7 +246,7 @@ base16384_err_t base16384_decode_fp(FILE* input, FILE* output, char* encbuf, cha
return base16384_err_ok;
}
static int is_next_end_fd(int fd) {
static inline int is_next_end_fd(int fd) {
char ch = 0;
read(fd, &ch, 1);
if(ch == '=') {
@@ -256,10 +256,10 @@ static int is_next_end_fd(int fd) {
}
base16384_err_t base16384_decode_fd(int input, int output, char* encbuf, char* decbuf) {
if(!input) {
if(input < 0) {
return base16384_err_fopen_input_file;
}
if(!output) {
if(output < 0) {
return base16384_err_fopen_output_file;
}
off_t inputsize = BUFSIZ*1024/8*8;
@@ -267,19 +267,24 @@ base16384_err_t base16384_decode_fd(int input, int output, char* encbuf, char* d
int cnt = 0;
int end = 0;
decbuf[0] = 0;
read(input, decbuf, 2);
if(read(input, decbuf, 2) < 2) {
return base16384_err_read_file;
}
if(decbuf[0] != (char)(0xfe)) cnt = 2;
while((cnt += read(input, decbuf+cnt, inputsize-cnt), cnt)) {
if((end = is_next_end_fd(input))) {
decbuf[cnt++] = '=';
decbuf[cnt++] = end;
return base16384_err_ok;
}
while((end = read(input, decbuf+cnt, inputsize-cnt)) > 0 || cnt > 0) {
if(end > 0) {
cnt += end;
if((end = is_next_end_fd(input))) {
decbuf[cnt++] = '=';
decbuf[cnt++] = end;
end = 0;
} else end = 1;
} else end = 0;
cnt = base16384_decode(decbuf, cnt, encbuf, outputsize);
if(write(output, encbuf, cnt) < cnt) {
return base16384_err_write_file;
}
cnt = 1;
cnt = end;
}
return base16384_err_ok;
}