|
|
|
@ -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;
|
|
|
|
|
}
|
|
|
|
|