commit
75e2133ada
GPG Key ID:
284488987C31F630
@ -0,0 +1 @@
|
|||||||
|
cpns
|
||||||
@ -0,0 +1,9 @@
|
|||||||
|
|
||||||
|
|
||||||
|
all: clean cpns
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm cpns
|
||||||
|
|
||||||
|
cpns:
|
||||||
|
gcc main.c -Wall -o $@ -lm
|
||||||
@ -0,0 +1,281 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <time.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <math.h>
|
||||||
|
|
||||||
|
#define __imdumb(one,two,three) strncpy(one,three,two)
|
||||||
|
|
||||||
|
int openOutputs(const char** ops, FILE*** fps, int* _cnt)
|
||||||
|
{
|
||||||
|
int ok =0;
|
||||||
|
register int cnt = 0;
|
||||||
|
|
||||||
|
for(;ops[cnt];cnt++); (void)0;
|
||||||
|
|
||||||
|
*_cnt = cnt;
|
||||||
|
if (cnt < 1) return -1;
|
||||||
|
else {
|
||||||
|
*fps = malloc(sizeof(FILE**)*cnt);
|
||||||
|
for (int i=0;i<cnt;i++) {
|
||||||
|
if(!((*fps)[i] = fopen(ops[i], "wb"))) {
|
||||||
|
printf("Warning: Could not open `%s' for writing.\n", ops[i]);
|
||||||
|
ok=1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ok;
|
||||||
|
}
|
||||||
|
|
||||||
|
void closeOutputs(FILE** fps, int cnt)
|
||||||
|
{
|
||||||
|
for(int i=0;i<cnt;i++)
|
||||||
|
if(fps[i]) fclose(fps[i]);
|
||||||
|
free(fps);
|
||||||
|
}
|
||||||
|
|
||||||
|
int ask(const char* thing)
|
||||||
|
{
|
||||||
|
printf(thing);
|
||||||
|
char* line=NULL;
|
||||||
|
size_t sz;
|
||||||
|
if (getline(&line, &sz, stdin)==0) {
|
||||||
|
int rt = (line[0] == 'y' || line[0] == 'Y');
|
||||||
|
free(line);
|
||||||
|
return rt;
|
||||||
|
} else return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifndef DEFAULT_BLOCKSIZE
|
||||||
|
#define DEFAULT_BLOCKSIZE 1024
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int BLOCKSIZE = DEFAULT_BLOCKSIZE;
|
||||||
|
|
||||||
|
int copyBlock(FILE* op, FILE** fps, int sz)
|
||||||
|
{
|
||||||
|
static unsigned char *buf=NULL;
|
||||||
|
|
||||||
|
if(!buf) buf = malloc(BLOCKSIZE);
|
||||||
|
|
||||||
|
int rd = fread(buf, 1, BLOCKSIZE, op);
|
||||||
|
|
||||||
|
if(rd<1) return rd;
|
||||||
|
|
||||||
|
for(register int i=0;i<sz;i++)
|
||||||
|
if(fps[i])
|
||||||
|
fwrite(buf, 1, rd, fps[i]);
|
||||||
|
|
||||||
|
return rd;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t sizeof_file(FILE* fp)
|
||||||
|
{
|
||||||
|
fseek(fp, 0, SEEK_END);
|
||||||
|
size_t ret = ftell(fp);
|
||||||
|
fseek(fp, 0, SEEK_SET);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define PROGRESS_SIZE 50
|
||||||
|
|
||||||
|
char* humanbytes(size_t sz)
|
||||||
|
{
|
||||||
|
static char buf[256];
|
||||||
|
size_t b, lsz;
|
||||||
|
size_t kb, mb,gb;
|
||||||
|
|
||||||
|
lsz = sz;
|
||||||
|
memset(buf,0,256);
|
||||||
|
|
||||||
|
gb = sz / (1024*1024*1024);
|
||||||
|
sz %= (1024*1024*1024);
|
||||||
|
#define _D(v, t) (((double)v) / ((double)t))
|
||||||
|
if(gb>0)
|
||||||
|
{
|
||||||
|
snprintf(buf, 255, "%.2f g", _D(lsz, (1024*2014*1024)));
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
mb = sz / (1024+1024);
|
||||||
|
sz %= (1024*1024);
|
||||||
|
|
||||||
|
if(mb>0)
|
||||||
|
{
|
||||||
|
snprintf(buf, 255, "%.2f m", _D(lsz, (1024*2014)));
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
kb = sz / 1024;
|
||||||
|
sz %= 1024;
|
||||||
|
|
||||||
|
if(kb>0)
|
||||||
|
{
|
||||||
|
snprintf(buf, 255, "%.2f k", _D(lsz, 1024));
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
b = sz;
|
||||||
|
|
||||||
|
snprintf(buf, 255, "%u b", (unsigned int)lsz);
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
char* secstr(double tm)
|
||||||
|
{
|
||||||
|
static char buf[256];
|
||||||
|
double sec = tm;
|
||||||
|
int min,hr;
|
||||||
|
min=hr=0;
|
||||||
|
|
||||||
|
while(sec>=60.0) { min +=1; sec-=6.0; }
|
||||||
|
while(min>=60) { hr +=1; min -= 60; }
|
||||||
|
|
||||||
|
char* _buf=buf;
|
||||||
|
int sz= 255;
|
||||||
|
memset(_buf,0,256);
|
||||||
|
|
||||||
|
if(hr>0) {
|
||||||
|
snprintf(_buf, sz, "%d h ", hr);
|
||||||
|
sz -= strlen(_buf);
|
||||||
|
_buf+=strlen(_buf);
|
||||||
|
}
|
||||||
|
if(min>0) {
|
||||||
|
snprintf(_buf, sz, "%d m ", min);
|
||||||
|
sz -= strlen(_buf);
|
||||||
|
_buf+=strlen(_buf);
|
||||||
|
}
|
||||||
|
snprintf(_buf, sz, "%.2f s", sec);
|
||||||
|
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
char* remain(size_t small, size_t large, double bps)
|
||||||
|
{
|
||||||
|
static char buf[256];
|
||||||
|
memset(buf,0,256);
|
||||||
|
if(bps==0)
|
||||||
|
__imdumb(buf, 255, "?");
|
||||||
|
else {
|
||||||
|
double vl = (large-small)/bps;
|
||||||
|
__imdumb(buf, 255, secstr(vl));
|
||||||
|
}
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct lastp {
|
||||||
|
int fnum;
|
||||||
|
int f2dp;
|
||||||
|
int b2dp;
|
||||||
|
char rem[256];
|
||||||
|
};
|
||||||
|
|
||||||
|
int lastp_equalp(const struct lastp *v1, const struct lastp *v2) {
|
||||||
|
return memcmp(v1, v2, sizeof(struct lastp)) == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void progress(size_t small, size_t large, size_t inc, time_t start, time_t elapsed)
|
||||||
|
{
|
||||||
|
static char* largesz = NULL;
|
||||||
|
static char befores[256];
|
||||||
|
|
||||||
|
static struct lastp *lastvls = NULL;
|
||||||
|
struct lastp thisvls;
|
||||||
|
|
||||||
|
if(!largesz)
|
||||||
|
largesz = strdup(humanbytes(large));
|
||||||
|
|
||||||
|
double frac = ((double)small)/((double)large);
|
||||||
|
double pct = (long)(frac*PROGRESS_SIZE);
|
||||||
|
|
||||||
|
double before = ((double)small)/((double)(elapsed-start));
|
||||||
|
if(lastvls == NULL) {
|
||||||
|
lastvls = malloc(sizeof(struct lastp));
|
||||||
|
memset(lastvls, 0, sizeof(struct lastp));
|
||||||
|
} else {
|
||||||
|
memset(&thisvls, 0, sizeof(struct lastp));
|
||||||
|
thisvls.fnum = (int) (frac*PROGRESS_SIZE);
|
||||||
|
thisvls.f2dp = (int)round(frac*100.00 *100.00);
|
||||||
|
thisvls.b2dp = (int)round(before*100.00);
|
||||||
|
__imdumb(thisvls.rem, 255, remain(small,large,before));
|
||||||
|
if (lastp_equalp(&thisvls, lastvls))
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
memset(befores, 0, 256);
|
||||||
|
__imdumb(befores, 255, humanbytes((size_t)before));
|
||||||
|
//printf("%c[2K", 27);
|
||||||
|
printf("\r[");
|
||||||
|
for(register int i=0;i<PROGRESS_SIZE;i++)
|
||||||
|
printf("%c", (i<pct?'#':' '));
|
||||||
|
printf("]: %.2f (%s / %s, (%s/s, ~%s left.) ", frac*100.0, humanbytes(small), largesz, befores, remain(small, large, before));
|
||||||
|
|
||||||
|
*lastvls = thisvls;
|
||||||
|
}
|
||||||
|
|
||||||
|
void flag(const char* inp)
|
||||||
|
{
|
||||||
|
if(!*inp) return;
|
||||||
|
switch(*inp) {
|
||||||
|
case 'l':
|
||||||
|
BLOCKSIZE += DEFAULT_BLOCKSIZE; break;
|
||||||
|
case 's':
|
||||||
|
BLOCKSIZE /= 2;
|
||||||
|
if(BLOCKSIZE < 1) BLOCKSIZE = 1; break;
|
||||||
|
default:
|
||||||
|
printf("Warning: Unknown flag `%c'\n", *inp);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
flag(inp+1);
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc,const char** argv)
|
||||||
|
{
|
||||||
|
const char* input = argv[1];
|
||||||
|
const char** output = argv+2;
|
||||||
|
int rv=1;
|
||||||
|
|
||||||
|
redo:
|
||||||
|
if (input == NULL || output[0]==NULL) {
|
||||||
|
printf("Usage: %s <input> <outputs...>\n", argv[0]);
|
||||||
|
} else {
|
||||||
|
if(output[0][0] == '-')
|
||||||
|
{
|
||||||
|
flag(output[0]+1);
|
||||||
|
output+=1;
|
||||||
|
goto redo;
|
||||||
|
}
|
||||||
|
FILE* finput = fopen(input, "rb");
|
||||||
|
if(!input) printf("Error: could not open `%s' for reading.\n", input);
|
||||||
|
else {
|
||||||
|
size_t input_sz = sizeof_file(finput);
|
||||||
|
FILE** foutput;
|
||||||
|
int ok, cnt;
|
||||||
|
if((ok = openOutputs(output, &foutput, &cnt))>=0) {
|
||||||
|
if(ok==1&& !ask("\nOne or more outputs failed to open, continue? (y/N) "))
|
||||||
|
printf("Error: aborted\n");
|
||||||
|
else {
|
||||||
|
printf("Will copy %s to %d files with a blocksize of %d\n", humanbytes(input_sz), cnt, BLOCKSIZE);
|
||||||
|
time_t start = time(NULL);
|
||||||
|
int rd;
|
||||||
|
size_t read=0;
|
||||||
|
while( (rd=copyBlock(finput, foutput, cnt))>0)
|
||||||
|
{
|
||||||
|
read+=rd;
|
||||||
|
progress(read, input_sz, rd,start, time(NULL));
|
||||||
|
}
|
||||||
|
time_t end = time(NULL);
|
||||||
|
|
||||||
|
char* dDone = strdup(humanbytes(read));
|
||||||
|
printf("\n\nCopied %s to %d files in %s. (~%s /s)\n", dDone, cnt, secstr(end-start), humanbytes(((double)read)/((double)(end-start))));
|
||||||
|
free(dDone);
|
||||||
|
rv = 0;
|
||||||
|
}
|
||||||
|
closeOutputs(foutput, cnt);
|
||||||
|
} else printf("Error: no output files could be opened\n");
|
||||||
|
fclose(finput);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return rv;
|
||||||
|
}
|
||||||
Loading…
Reference in new issue