special case buffer_get_token with token length 1 through memccpy

master
leitner 9 years ago
parent 9b2377c4be
commit 9cfa19396e

@ -2,10 +2,28 @@
#include "buffer.h"
#include "scan.h"
#include <string.h> /* for memccpy */
ssize_t buffer_get_token(buffer* b,char* x,size_t len,const char* charset,size_t setlen) {
size_t blen;
if ((ssize_t)len<0) len=(ssize_t)(((size_t)-1)>>1);
if (setlen==1) {
for (blen=0; blen<len; ) {
ssize_t n=buffer_feed(b);
char* d;
if (n<=0) return blen;
if (n>(ssize_t)(len-blen)) n=len-blen;
if ((d=memccpy(x+blen,b->x+b->p,(unsigned char)charset[0],n))) {
/* memccpy returns a pointer to the next char after matching
* char or NULL if it copied all bytes it was asked for */
b->p+=(d-(x+blen));
return d-x-1;
}
blen+=n;
b->p+=n;
}
} else {
for (blen=0;blen<len;++blen) {
register ssize_t r;
if ((r=buffer_getc(b,x))<0) return r;
@ -13,5 +31,6 @@ ssize_t buffer_get_token(buffer* b,char* x,size_t len,const char* charset,size_t
if (byte_chr(charset,setlen,*x)<setlen) break;
++x;
}
}
return (ssize_t)blen;
}

Loading…
Cancel
Save