1
0
Fork 0

2024-01-02 work

This commit is contained in:
dtb 2024-01-02 22:31:21 -07:00
parent ba1a1ed210
commit 9f9f41da3e
1 changed files with 69 additions and 39 deletions

View File

@ -8,6 +8,42 @@
STDIN_FILENO, STDOUT_FILENO */
#define ARGV0 (argv[0] == NULL ? "<no argv[0]>" : argv[0])
#define fdisnotstd(fd) \
((fd) != STDIN_FILENO \
&& (fd) != STDOUT_FILENO \
&& (fd) != STDERR_FILENO)
/* information that needs to be kept about each file in the "pipe" */
struct Io{
size_t index; /* index in io[] array */
char *fn; /* file name (may be stdin_name or stdout_name) */
int rec; /* records processed */
int prec; /* partial records processed */
int fd; /* file descriptor */
int seek; /* bytes to seek/skip (will be 0 after skippage) */
int bs; /* buffer size */
size_t bufuse; /* buffer usage */
char *buf; /* buffer */
};
static char stdin_name = "<stdin>";
static char stdout_name = "<stdout>";
static void
Io_setdefaults(struct Io io[2]){
io[0].bs = io[1].bs = 1024;
io[0].buf = io[1].buf = NULL;
io[0].fd = STDIN_FILENO;
io[0].fn = stdin_name;
io[1].fd = STDOUT_FILENO;
io[1].fn = stdout_name;
io[0].index = 0;
io[1].index = 1;
io[0].seek = io[1].seek = 0;
return;
}
/* read(2) but with a second chance if it returns zero */
static int
@ -32,12 +68,15 @@ parse(char *s){
: r;
}
/* s is the component (e.g. filename) that failed */
static int
oserr(char *argv0, char *s){
fprintf(stderr, "%s: %s: %s\n", argv0, s, strerror(errno));
return EX_OSERR;
}
/* diagnostic output
* dd(1p) does %d+%d\n%d+%d\n */
struct Io *
output(struct Io io[2]){
@ -50,7 +89,7 @@ static int
terminate(struct Io io[2]){
for(;; ++io){
if(io->fd != STDIN_FILENO && io->fd != STDOUT_FILENO)
if(fdisnotstd(io->fd))
close(io->fd);
free(io->buf);
if(io->index == 1)
@ -62,6 +101,7 @@ terminate(struct Io io[2]){
static int
usage(char *argv0){
fprintf(stderr, "Usage: %s (-af) (-c [count]\n"
"\t(-i [input file]) (-b [input block size])"
" (-s [input offset])\n"
@ -70,41 +110,20 @@ usage(char *argv0){
return EX_USAGE;
}
static char stdin_name = "<stdin>";
static char stdout_name = "<stdout>";
int main(int argc, char *argv[]){
struct Io{
size_t index;
char *fn;
int rec;
int prec;
int fd;
int seek;
int bs;
size_t bufuse;
char *buf;
} io[2]; /* { read, write } same as pipe(2) */
char align;
char noerror;
int c;
int count;
size_t i;
size_t p;
char noerror;
int p; /* pipe side */
struct Io io[2]; /* { read, write } same as pipe(2) */
/* defaults */
align = 0;
count = 0;
noerror = 0;
io[0].bs = io[1].bs = 1024;
io[0].buf = io[1].buf = NULL;
io[0].fd = STDIN_FILENO;
io[0].fn = stdin_name;
io[1].fd = STDOUT_FILENO;
io[1].fn = stdout_name;
io[0].index = 0;
io[1].index = 1;
io[0].seek = io[1].seek = 0;
Io_setdefaults(io);
if(argc > 0)
while((c = getopt(argc, argv, "ac:i:b:fs:o:B:S:")) != -1)
@ -112,14 +131,13 @@ int main(int argc, char *argv[]){
int t;
p = (c == 'o');
if( (t = open(optarg,
if( ((t = open(optarg,
c == 'i'
? O_RDONLY
: O_WRONLY | O_CREAT))
== -1)
|| (io[p].fd != STDIN_FILENO
&& io[p].fd != STDOUT_FILENO
&& fclose(io[p].f) == EOF){
|| (fdisnotstd(io[p].fd)
&& close(io[p].fd) != 0)){
terminate(io);
return oserr(argv[0], optarg);
}else{
@ -152,9 +170,8 @@ int main(int argc, char *argv[]){
terminate(io);
return EX_OSERR;
}else if(io[p].seek > 0){ /* seek */
if(io[p].fd == STDIN_FILENO
|| io[p].fd == STDOUT_FILENO
|| lseek(io[p].fd, io[p].seek, SEEK_SET) == -1){
if(!fdisnotstd(io[p].fd)
|| lseek(io[p].fd, io[p].seek, SEEK_SET) == -1){
if(p == 0){
do if((io[p].bufuse = _read(io[p].fd, io[p].buf,
io[p].bs < io[p].seek
@ -185,16 +202,29 @@ int main(int argc, char *argv[]){
io[0].prec = io[0].rec = io[1].prec = io[1].rec = 0;
do{ if( (io[0].bufuse
= fread(io[0].buf, sizeof io[0].buf, io[0].bs, io[0].f))
< ibs){
if(feof(io[0].f) != 0)
do{ /* read */ if((io[0].bufuse
= read(io[0].fd, io[0].buf, io[0].bs))
< io[0].bs){
if(io[0].bufuse == 0 && (!noerror
|| ((io[0].bufuse = read(io[0].fd, io[0].buf, io[0].bs))
== 0)))
break; /* that's all she wrote */
else{
io[0].prec += 1;
if(noerror){
}else
count = 1;
if(align){
memset(io[0].buf + io[0].bufuse, '\0',
io[0].bs - io[0].bufuse);
io[0].bufuse = io[0].bs;
}
}
}
while(io[0].bufuse > 0){
/* write */ while(io[0].bufuse > 0){
}
}while(count == 0 || --count > 0);
return terminate(output(io));