mirror of
https://github.com/fumiama/base16384.git
synced 2026-06-16 02:50:24 +08:00
add v1 branch
This commit is contained in:
源文雨
312
base16384.c
312
base16384.c
@@ -1,225 +1,99 @@
|
||||
#ifndef __cosmopolitan
|
||||
//base16384.c
|
||||
//MIT fumiama 20200416
|
||||
#include <stdio.h>
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/stat.h>
|
||||
#include <fcntl.h>
|
||||
#include <time.h>
|
||||
#ifdef _WIN32
|
||||
#include <windows.h>
|
||||
#else
|
||||
#include <sys/mman.h>
|
||||
#endif
|
||||
#endif
|
||||
#include "base16384.h"
|
||||
#include "bitio.h"
|
||||
|
||||
#ifdef __cosmopolitan
|
||||
#define get_file_size(filepath) ((off_t)GetFileSize(filepath))
|
||||
#else
|
||||
static off_t get_file_size(const char* filepath) {
|
||||
struct stat statbuf;
|
||||
return stat(filepath, &statbuf)?-1:statbuf.st_size;
|
||||
}
|
||||
#endif
|
||||
#define HEAD 0x4E
|
||||
BIT outbuf;
|
||||
unsigned char buf[BUFSIZ/8*8+2];
|
||||
|
||||
char encbuf[BUFSIZ*1024/7*7];
|
||||
char decbuf[BUFSIZ*1024/8*8+2];
|
||||
|
||||
static int encode_file(const char* input, const char* output) {
|
||||
off_t inputsize;
|
||||
FILE* fp = NULL;
|
||||
FILE* fpo;
|
||||
if(*(uint16_t*)input == *(uint16_t*)"-") { // read from stdin
|
||||
inputsize = 0;
|
||||
fp = stdin;
|
||||
} else inputsize = get_file_size(input);
|
||||
if(inputsize < 0) {
|
||||
perror("Get file size error");
|
||||
return 1;
|
||||
}
|
||||
fpo = (*(uint16_t*)output == *(uint16_t*)"-")?stdout:fopen(output, "wb");
|
||||
if(!fpo) {
|
||||
perror("Fopen output file error");
|
||||
return 2;
|
||||
}
|
||||
if(!inputsize || inputsize > BUFSIZ*1024) { // stdin or big file, use encbuf & fread
|
||||
inputsize = BUFSIZ*1024/7*7;
|
||||
#ifdef _WIN32
|
||||
}
|
||||
#endif
|
||||
if(!fp) fp = fopen(input, "rb");
|
||||
if(!fp) {
|
||||
perror("Fopen input file error");
|
||||
return 3;
|
||||
}
|
||||
|
||||
int outputsize = encode_len(inputsize)+16;
|
||||
size_t cnt = 0;
|
||||
fputc(0xFE, fpo);
|
||||
fputc(0xFF, fpo);
|
||||
while((cnt = fread(encbuf, sizeof(char), inputsize, fp))) {
|
||||
int n = encode(encbuf, cnt, decbuf, outputsize);
|
||||
if(fwrite(decbuf, n, 1, fpo) <= 0) {
|
||||
perror("Write file error");
|
||||
return 4;
|
||||
}
|
||||
}
|
||||
/* 由操作系统负责释放资源
|
||||
fclose(fpo);
|
||||
fclose(fp);
|
||||
以缩短程序运行时间 */
|
||||
#ifndef _WIN32
|
||||
} else { // small file, use mmap & fwrite
|
||||
int fd = open(input, O_RDONLY);
|
||||
if(fd <= 0) {
|
||||
perror("Open input file error");
|
||||
return 5;
|
||||
}
|
||||
char *input_file = mmap(NULL, (size_t)inputsize, PROT_READ, MAP_PRIVATE, fd, 0);
|
||||
if(input_file == MAP_FAILED) {
|
||||
perror("Map input file error");
|
||||
return 6;
|
||||
}
|
||||
int outputsize = encode_len(inputsize)+16;
|
||||
fputc(0xFE, fpo);
|
||||
fputc(0xFF, fpo);
|
||||
int n = encode(input_file, (int)inputsize, decbuf, outputsize);
|
||||
if(fwrite(decbuf, n, 1, fpo) <= 0) {
|
||||
perror("Write file error");
|
||||
return 7;
|
||||
}
|
||||
munmap(input_file, (size_t)inputsize);
|
||||
/* 由操作系统负责释放资源
|
||||
fclose(fpo);
|
||||
close(fd);
|
||||
以缩短程序运行时间 */
|
||||
}
|
||||
#endif
|
||||
return 0;
|
||||
}
|
||||
|
||||
#define rm_head(fp) {\
|
||||
int ch = fgetc(fp);\
|
||||
if(ch == 0xFE) fgetc(fp);\
|
||||
else rewind(fp);\
|
||||
}
|
||||
|
||||
#define skip_offset(input_file) ((input_file[0]==(char)0xFE)?2:0)
|
||||
|
||||
static int is_next_end(FILE* fp) {
|
||||
int ch = fgetc(fp);
|
||||
if(ch == '=') return fgetc(fp);
|
||||
ungetc(ch, fp);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int decode_file(const char* input, const char* output) {
|
||||
off_t inputsize;
|
||||
FILE* fp = NULL;
|
||||
FILE* fpo;
|
||||
if(*(uint16_t*)input == *(uint16_t*)"-") { // read from stdin
|
||||
inputsize = 0;
|
||||
fp = stdin;
|
||||
} else inputsize = get_file_size(input);
|
||||
if(inputsize < 0) {
|
||||
perror("Get file size error");
|
||||
return 1;
|
||||
}
|
||||
fpo = (*(uint16_t*)output == *(uint16_t*)"-")?stdout:fopen(output, "wb");
|
||||
if(!fpo) {
|
||||
perror("Fopen output file error");
|
||||
return 2;
|
||||
}
|
||||
if(!inputsize || inputsize > BUFSIZ*1024) { // stdin or big file, use decbuf & fread
|
||||
inputsize = BUFSIZ*1024/8*8;
|
||||
#ifdef _WIN32
|
||||
}
|
||||
#endif
|
||||
if(!fp) fp = fopen(input, "rb");
|
||||
if(!fp) {
|
||||
perror("Fopen input file error");
|
||||
return 3;
|
||||
}
|
||||
int outputsize = decode_len(inputsize, 0)+16;
|
||||
int cnt = 0;
|
||||
int end = 0;
|
||||
rm_head(fp);
|
||||
while((cnt = fread(decbuf, sizeof(char), inputsize, fp))) {
|
||||
if((end = is_next_end(fp))) {
|
||||
decbuf[cnt++] = '=';
|
||||
decbuf[cnt++] = end;
|
||||
}
|
||||
if(fwrite(encbuf, decode(decbuf, cnt, encbuf, outputsize), 1, fpo) <= 0) {
|
||||
perror("Write file error");
|
||||
return 4;
|
||||
}
|
||||
}
|
||||
/* 由操作系统负责释放资源
|
||||
fclose(fpo);
|
||||
fclose(fp);
|
||||
以缩短程序运行时间 */
|
||||
#ifndef _WIN32
|
||||
} else { // small file, use mmap & fwrite
|
||||
int fd = open(input, O_RDONLY);
|
||||
if(fd <= 0) {
|
||||
perror("Open input file error");
|
||||
return 5;
|
||||
}
|
||||
char *input_file = mmap(NULL, (size_t)inputsize, PROT_READ, MAP_PRIVATE, fd, 0);
|
||||
if(input_file == MAP_FAILED) {
|
||||
perror("Map input file error");
|
||||
return 6;
|
||||
}
|
||||
int outputsize = decode_len(inputsize, 0)+16;
|
||||
int off = skip_offset(input_file);
|
||||
if(fwrite(encbuf, decode(input_file+off, inputsize-off, encbuf, outputsize), 1, fpo) <= 0) {
|
||||
perror("Write file error");
|
||||
return 7;
|
||||
}
|
||||
munmap(input_file, (size_t)inputsize);
|
||||
/* 由操作系统负责释放资源
|
||||
fclose(fpo);
|
||||
close(fd);
|
||||
以缩短程序运行时间 */
|
||||
}
|
||||
#endif
|
||||
return 0;
|
||||
}
|
||||
|
||||
#ifndef _WIN32
|
||||
unsigned long get_start_ms() {
|
||||
struct timespec ts;
|
||||
clock_gettime(CLOCK_MONOTONIC, &ts);
|
||||
return (ts.tv_sec * 1000 + ts.tv_nsec / 1000000);
|
||||
}
|
||||
#endif
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
if(argc != 4) {
|
||||
puts("Usage: -[e|d] <inputfile> <outputfile>");
|
||||
puts("\t-e encode");
|
||||
puts("\t-d decode");
|
||||
puts("\t<inputfile> pass - to read from stdin");
|
||||
puts("\t<outputfile> pass - to write to stdout");
|
||||
exit(EXIT_SUCCESS);
|
||||
static void push(int istrue, FILE *fpo, int offset){
|
||||
if(!~pushbit(&outbuf, istrue)){
|
||||
if(offset) for(int i = 0; i < BITBUFSIZE; i += 2) outbuf.b[i] += offset;
|
||||
fwrite(outbuf.b, sizeof(char), BITBUFSIZE, fpo);
|
||||
outbuf.p = 0;
|
||||
pushbit(&outbuf, istrue);
|
||||
}
|
||||
#ifdef _WIN32
|
||||
clock_t t = clock();
|
||||
#else
|
||||
unsigned long t = get_start_ms();
|
||||
#endif
|
||||
int exitstat = 0;
|
||||
switch(argv[1][1]) {
|
||||
case 'e': exitstat = encode_file(argv[2], argv[3]); break;
|
||||
case 'd': exitstat = decode_file(argv[2], argv[3]); break;
|
||||
default: break;
|
||||
}
|
||||
if(!exitstat && *(uint16_t*)(argv[3]) != *(uint16_t*)"-") {
|
||||
#ifdef _WIN32
|
||||
printf("spend time: %lums\n", clock() - t);
|
||||
#else
|
||||
printf("spend time: %lums\n", get_start_ms() - t);
|
||||
#endif
|
||||
}
|
||||
return exitstat;
|
||||
}
|
||||
|
||||
static void encode(FILE *fp, FILE *fpo){
|
||||
int cnt, prevcnt;
|
||||
|
||||
fputc(0xFE, fpo);
|
||||
fputc(0xFF, fpo);
|
||||
while((cnt = fread(buf, sizeof(char), BUFSIZ/7*7, fp))){
|
||||
prevcnt = cnt;
|
||||
for(int i = 0; i < cnt * 8; i++){
|
||||
if(!(outbuf.p % 16)){
|
||||
push(0, fpo, HEAD);
|
||||
push(0, fpo, HEAD);
|
||||
}
|
||||
push(buf[i / 8] & (128u >> (i % 8)), fpo, HEAD);
|
||||
}
|
||||
}
|
||||
if(outbuf.p){
|
||||
outbuf.b[outbuf.p / 8] &= ~(255u >> outbuf.p % 8);
|
||||
for(int i = 0; i < outbuf.p / 8 + 1; i += 2) outbuf.b[i] += HEAD;
|
||||
fwrite(outbuf.b, sizeof(char), (outbuf.p % 8)?outbuf.p / 8 + 1:outbuf.p / 8 , fpo);
|
||||
if(((outbuf.p % 8)?outbuf.p / 8 + 1:outbuf.p / 8) % 2) fputc(0, fpo);
|
||||
if(outbuf.p % 16) {
|
||||
fputc('=', fpo);
|
||||
fputc(prevcnt%7, fpo);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static uint8_t map[] = {0, 6, 12, 4, 10, 2, 8};
|
||||
|
||||
static void decode(FILE *fp, FILE *fpo){
|
||||
int cnt;
|
||||
int cnt8;
|
||||
while((cnt = fread(buf, sizeof(char), BUFSIZ/8*8, fp))){
|
||||
cnt8 = cnt * 8;
|
||||
if(buf[cnt-2] == '=') {
|
||||
cnt8 -= 2*8 + map[buf[cnt-1]&7];
|
||||
} else if(cnt == BUFSIZ && !feof(fp)) {
|
||||
int ch;
|
||||
if((ch=fgetc(fp)) != '=') ungetc(ch, fp);
|
||||
else cnt8 -= map[fgetc(fp)&7];
|
||||
}
|
||||
for(int i = ((buf[0] == 0xFE)?16:0); i < cnt8; i++){
|
||||
if(!(i % 16)){
|
||||
buf[i / 8] -= HEAD;
|
||||
i += 2;
|
||||
}
|
||||
push(buf[i / 8] & (128u >> (i % 8)), fpo, 0);
|
||||
}
|
||||
}
|
||||
if(outbuf.p) fwrite(outbuf.b, sizeof(char), (outbuf.p % 8)?outbuf.p / 8 + 1:outbuf.p / 8, fpo);
|
||||
}
|
||||
|
||||
#define CHOICE argv[1][1]
|
||||
#define READ argv[2]
|
||||
#define WRITE argv[3]
|
||||
|
||||
int main(int argc, char **argv){
|
||||
FILE *fp = NULL;
|
||||
FILE *fpo = NULL;
|
||||
|
||||
if(argc != 4){
|
||||
fputs("Usage: -[e|d] inputfile outputfile\n", stderr);
|
||||
fputs(" -e encode\n", stderr);
|
||||
fputs(" -d decode\n", stderr);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
fp = fopen(READ, "rb");
|
||||
fpo = fopen(WRITE, "wb");
|
||||
if(!fp || !fpo){
|
||||
fputs("IO error!", stderr);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
switch(CHOICE){
|
||||
case 'e': encode(fp, fpo); break;
|
||||
case 'd': decode(fp, fpo); break;
|
||||
default: break;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user