scip/pcp
1
0
Fork 0
mirror of https://codeberg.org/scip/pcp.git synced 2025-12-17 03:50:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Revert "rm html stuff from repo, changes too often. enhanced pcpstream"

Browse Source 
This reverts commit 82b05b768e.
This commit is contained in:
TLINDEN
2014-02-21 20:35:10 +01:00
parent 82b05b768e
commit fe1ea9cb76
6 changed files with 21 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

138
libpcp/pcpstream.c
View File

@@ -23,14 +23,11 @@
Pcpstream *ps_init(void) {
Pcpstream *stream = ucmalloc(sizeof(Pcpstream));
stream->z = buffer_new(32, "Z85stream");
stream->b = NULL;
stream->fd = NULL;
stream->is_buffer = 0;
stream->eof = 0;
stream->err = 0;
stream->armor = 0;
stream->determine = 0;
return stream;
}
@@ -54,47 +51,15 @@ Pcpstream *ps_new_outbuffer() {
return stream;
}
void ps_armor(Pcpstream *stream) {
stream->armor = 1;
}
void ps_determine(Pcpstream *stream) {
stream->determine = 1;
}
int ps_end(Pcpstream *stream) {
return stream->eof;
}
int ps_err(Pcpstream *stream) {
return stream->err;
}
size_t ps_read(Pcpstream *stream, void *buf, size_t readbytes) {
size_t gotbytes = 0;
if(stream->is_buffer) {
/* a buffer stream */
/* check if there's enough space in our buffer */
if(buffer_left(stream->b) < readbytes)
readbytes = buffer_left(stream->b);
if(stream->armor == 1) {
size_t i = 0;
uint8_t c;
while (i < readbytes) {
c = buffer_get8(stream->b);
if(c != '\r' && c != '\n') {
buffer_add8(stream->z, c);
i++;
}
}
memcpy(buf, buffer_get(stream->z), buffer_size(stream->z));
gotbytes = buffer_size(stream->z);
buffer_clear(stream->z);
}
else
gotbytes = buffer_get_chunk(stream->b, buf, readbytes);
gotbytes = buffer_get_chunk(stream->b, buf, readbytes);
if(gotbytes == 0) {
/* this should not happen with buffers */
stream->eof = 1;
@@ -102,27 +67,7 @@ size_t ps_read(Pcpstream *stream, void *buf, size_t readbytes) {
}
}
else {
/* a FILE stream */
if(stream->armor == 1) {
size_t i = 0;
uint8_t c;
while (i < readbytes) {
gotbytes = fread(&c, 1, 1, stream->fd);
if(gotbytes == 0)
break;
if(c != '\r' && c != '\n') {
buffer_add8(stream->z, c);
i++;
}
}
memcpy(buf, buffer_get(stream->z), buffer_size(stream->z));
gotbytes = buffer_size(stream->z);
_dump("buf", buf, gotbytes);
buffer_clear(stream->z);
}
else
gotbytes = fread(buf, 1, readbytes, stream->fd);
gotbytes = fread(buf, 1, readbytes, stream->fd);
if(gotbytes == 0) {
if(feof(stream->fd) != 0)
stream->eof = 1;
@@ -131,82 +76,22 @@ size_t ps_read(Pcpstream *stream, void *buf, size_t readbytes) {
}
}
if(gotbytes > 0 && stream->determine && stream->firstread == 0) {
/* check if we need to decode input */
fprintf(stderr, "determine\n");
if(_buffer_is_binary(buf, gotbytes) == 0) {
fprintf(stderr, "is armored\n");
stream->armor = 1;
}
}
stream->firstread = 1;
if(gotbytes > 0 && stream->armor == 1) {
/* z85 decode buf */
size_t binlen;
unsigned char *bin = pcp_z85_decode(buf, &binlen);
if(bin == NULL) {
return 0;
}
memcpy(buf, bin, binlen);
_dump("decoded", buf, binlen);
free(bin);
return binlen;
}
return gotbytes;
}
size_t ps_write(Pcpstream *stream, void *buf, size_t writebytes) {
size_t donebytes = 0;
if(stream->armor == 1) {
/* z85 encode buf */
size_t padlen, zlen, i, pos;
unsigned char *padded = pcp_padfour(buf, writebytes, &padlen);
zlen = (padlen * 5 / 4);
char *z85 = ucmalloc(zlen);
zmq_z85_encode(z85, padded, padlen);
_dump(" orig", buf, writebytes);
_dump(" z85", z85, zlen);
pos = stream->linewr;
for(i=0; i<zlen; ++i) {
if(pos >= 71) {
buffer_add8(stream->z, '\r');
buffer_add8(stream->z, '\n');
pos = 1;
}
else
pos++;
buffer_add8(stream->z, z85[i]);
}
stream->linewr = pos;
_dump("n added", buffer_get(stream->z), buffer_size(stream->z));
}
else {
buffer_add(stream->z, buf, writebytes);
}
if(stream->is_buffer) {
buffer_add(stream->b, buffer_get(stream->z), buffer_size(stream->z));
donebytes = buffer_size(stream->z);
buffer_add(stream->b, buf, writebytes);
donebytes = writebytes;
}
else {
donebytes = fwrite(buffer_get(stream->z), 1, buffer_size(stream->z), stream->fd);
if(ferror(stream->fd) != 0 || donebytes < buffer_size(stream->z))
donebytes = fwrite(buf, 1, writebytes, stream->fd);
if(ferror(stream->fd) != 0 || donebytes < writebytes)
stream->err = 1;
}
writebytes = buffer_size(stream->z);
buffer_clear(stream->z);
return writebytes;
}
@@ -240,6 +125,14 @@ void ps_close(Pcpstream *stream) {
}
}
int ps_end(Pcpstream *stream) {
return stream->eof;
}
int ps_err(Pcpstream *stream) {
return stream->err;
}
size_t ps_tell(Pcpstream *stream) {
if(stream->is_buffer) {
if(stream->b->end > stream->b->offset)
@@ -255,4 +148,3 @@ size_t ps_tell(Pcpstream *stream) {
Buffer *ps_buffer(Pcpstream *stream) {
return stream->b;
}